relax water constraint test to support small variations from dependen…

…cy versions (#100)
IMMM-SFA · Dec 29, 2022 · 6ad55bc · 6ad55bc
1 parent 1fca428
commit 6ad55bc
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/mosartwmpy/_version.py b/mosartwmpy/_version.py
@@ -1 +1 @@
-__version__ = "0.5.0"
+__version__ = "0.5.1"
diff --git a/mosartwmpy/tests/test_calculate_water_constraints_by_farm.py b/mosartwmpy/tests/test_calculate_water_constraints_by_farm.py
@@ -1,4 +1,5 @@
 import unittest
+import numpy as np
 import pandas as pd
 import pkg_resources
 
@@ -65,7 +66,11 @@ def test_calculate_water_constraints_by_farm(self):
         90: 1867.5493916929483, 91: 871.2380665376529, 92: 551.7301491571899, 93: 12.085666626862842, 94: 0.0, 95: 0.0, 
         96: 2.757034402743735, 97: 2.059543992483289, 98: 0.0, 99: 0.0}
 
-        self.assertEqual(water_constraints_by_farm, expected)
+        np.testing.assert_almost_equal(
+            np.array(list(water_constraints_by_farm.values())),
+            np.array(list(expected.values())),
+            2
+        )
 
 
 if __name__ == '__main__':