Fix failing tests

ImperialCollegeLondon · Jan 25, 2024 · 09b8e2d · 09b8e2d
1 parent 6c99dc2
commit 09b8e2d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/measurement/test_models.py b/tests/measurement/test_models.py
@@ -72,7 +72,7 @@ def test_query_ordering(self):
 
 class TestReport(TestCase):
     def setUp(self) -> None:
-        from measurement.models import Report
+        from measurement.models import Report, ReportType
         from station.models import Station
         from variable.models import Variable
 
@@ -83,7 +83,7 @@ def setUp(self) -> None:
             station=station,
             variable=variable,
             value=42,
-            report_type=Report.HOURLY,
+            report_type=ReportType.HOURLY,
             completeness=1,
         )