samwip

uktrade · Mar 22, 2024 · 816badb · 816badb
1 parent d782b2d
commit 816badb
Show file tree

Hide file tree

Showing 2 changed files with 9 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
@@ -112,7 +112,9 @@ def test_end_of_month_mar(self):
 
 
 class ReadArchivedForecastTest(TestCase):
-    archived_figure = []
+    @classmethod
+    def setUpTestData(cls):
+        cls.archived_figure = []
 
     def setUp(self):
         self.init_data = MonthlyFigureSetup()
@@ -334,7 +336,9 @@ def test_end_of_month_mar(self):
 
 
 class ReadArchivedBudgetTest(TestCase):
-    archived_figure = []
+    @classmethod
+    def setUpTestData(cls):
+        cls.archived_figure = []
 
     def setUp(self):
         self.init_data = MonthlyFigureSetup()

diff --git a/end_of_month/test/test_outturn_variance.py b/end_of_month/test/test_outturn_variance.py
@@ -7,7 +7,9 @@
 
 
 class ReadMonthlyVarianceTest(TestCase):
-    archived_figure = []
+    @classmethod
+    def setUpTestData(cls):
+        cls.archived_figure = []
 
     def setUp(self):
         self.init_data = MonthlyFigureSetup()