Merge pull request #10 from Ackuq/testing-improvements

Gracefully stop Spark in tests
Ackuq · Feb 16, 2022 · aade14c · aade14c
2 parents 6384412 + d182338
commit aade14c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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())