Merge branch 'main' into gomes/reduce-count-in-asserts

Energinet-DataHub · Nov 29, 2024 · c15f276 · c15f276
2 parents db954f5 + 444da5c
commit c15f276
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/source/databricks/calculation_engine/tests/features/conftest.py b/source/databricks/calculation_engine/tests/features/conftest.py
@@ -24,6 +24,13 @@
 from package.calculation.calculation_output import CalculationOutput
 
 
+@pytest.fixture(scope="module", autouse=True)
+def clear_cache(spark: SparkSession) -> None:
+    yield
+    # Clear the cache after each test module to avoid memory issues
+    spark.catalog.clearCache()
+
+
 @pytest.fixture(scope="module")
 def actual_and_expected(
     request: FixtureRequest,