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