diff --git a/python/tests/utils.py b/python/tests/utils.py index d44e958..bf91196 100644 --- a/python/tests/utils.py +++ b/python/tests/utils.py @@ -24,7 +24,7 @@ def setUp(self) -> None: self.pit_context = PitContext(self.sql_context) def tearDown(self) -> None: - return super().tearDown() + self.spark.stop() def _assertFieldsEqual(self, a: StructField, b: StructField): self.assertEqual(a.name.lower(), b.name.lower())