Use ValidationError in tests

ImperialCollegeLondon · Jan 17, 2024 · 975ae9e · 975ae9e
1 parent 3251159
commit 975ae9e
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/tests/measurement/test_models.py b/tests/measurement/test_models.py
@@ -1,6 +1,7 @@
 from datetime import datetime
 
 import pytz
+from django.core.exceptions import ValidationError
 from django.test import TestCase
 from model_bakery import baker
 
@@ -95,11 +96,11 @@ def test_clean(self):
 
         # But fails for the other two
         self.model.report_type = ReportType.DAILY
-        with self.assertRaises(ValueError):
+        with self.assertRaises(ValidationError):
             self.model.clean()
 
         self.model.report_type = ReportType.MONTLY
-        with self.assertRaises(ValueError):
+        with self.assertRaises(ValidationError):
             self.model.clean()
 
         # We check 'used_for_monthly' compatibility
@@ -111,9 +112,9 @@ def test_clean(self):
 
         # But not for the other two
         self.model.report_type = ReportType.HOURLY
-        with self.assertRaises(ValueError):
+        with self.assertRaises(ValidationError):
             self.model.clean()
 
         self.model.report_type = ReportType.MONTLY
-        with self.assertRaises(ValueError):
+        with self.assertRaises(ValidationError):
             self.model.clean()