wip

uktrade · Mar 22, 2024 · aa2987d · aa2987d
1 parent f48bbcf
commit aa2987d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/end_of_month/test/test_end_of_month_process.py b/end_of_month/test/test_end_of_month_process.py
@@ -114,7 +114,7 @@ def test_end_of_month_mar(self):
 
 class TestReadArchivedForecastTest:
     @pytest.fixture(autouse=True)
-    def _setup(self):
+    def _setup(self, db):
         self.archived_figure = [0 for _ in range(16)]
         self.init_data = MonthlyFigureSetup()
         self.init_data.setup_forecast()
@@ -333,7 +333,7 @@ def test_end_of_month_mar(self):
 
 class TestReadArchivedBudgetTest:
     @pytest.fixture(autouse=True)
-    def _setup(self):
+    def _setup(self, db):
         self.archived_figure = [0 for _ in range(16)]
         self.init_data = MonthlyFigureSetup()
         self.init_data.setup_budget()

diff --git a/end_of_month/test/test_outturn_variance.py b/end_of_month/test/test_outturn_variance.py
@@ -8,7 +8,7 @@
 
 class TestReadMonthlyVarianceTest:
     @pytest.fixture(autouse=True)
-    def _setup(self):
+    def _setup(self, db):
         self.archived_figure = [0 for _ in range(16)]
         self.init_data = MonthlyFigureSetup()
         self.init_data.setup_forecast()