From 1c6f53ac7943ecdd953a2e421942beab90834e6d Mon Sep 17 00:00:00 2001 From: Hyukjin Kwon Date: Sun, 7 Apr 2024 15:08:05 +0900 Subject: [PATCH] Drop the tables after tests finished --- .../tests/connect/streaming/test_parity_listener.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/python/pyspark/sql/tests/connect/streaming/test_parity_listener.py b/python/pyspark/sql/tests/connect/streaming/test_parity_listener.py index 412f49a3960b6..f5ffa0154df1c 100644 --- a/python/pyspark/sql/tests/connect/streaming/test_parity_listener.py +++ b/python/pyspark/sql/tests/connect/streaming/test_parity_listener.py @@ -122,8 +122,16 @@ def verify(test_listener, table_postfix): # Remove again to verify this won't throw any error self.spark.streams.removeListener(test_listener) - verify(TestListenerV1(), "_v1") - verify(TestListenerV2(), "_v2") + with self.table( + "listener_start_events_v1", + "listener_progress_events_v1", + "listener_terminated_events_v1", + "listener_start_events_v2", + "listener_progress_events_v2", + "listener_terminated_events_v2", + ): + verify(TestListenerV1(), "_v1") + verify(TestListenerV2(), "_v2") def test_accessing_spark_session(self): spark = self.spark