Using assert_allclose for better numerical comparisons.

RE:natcap#1641
phargogh · Oct 16, 2024 · 63fb888 · 63fb888
1 parent ab90c46
commit 63fb888
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/test_urban_nature_access.py b/tests/test_urban_nature_access.py
@@ -412,7 +412,7 @@ def test_split_urban_nature(self):
         )
         for fieldname, expected_value in expected_values.items():
             numpy.testing.assert_allclose(
-                admin_feature.GetField(fieldname), expected_value)
+                admin_feature.GetField(fieldname), expected_value, rtol=1e-6)
 
         # The sum of the under-and-oversupplied populations should be equal
         # to the total population count.
@@ -603,8 +603,8 @@ def test_radii_by_pop_group(self):
             set(defn.GetName() for defn in summary_layer.schema),
             set(expected_field_values.keys()))
         for fieldname, expected_value in expected_field_values.items():
-            self.assertAlmostEqual(
-                expected_value, summary_feature.GetField(fieldname))
+            numpy.testing.assert_allclose(
+                expected_value, summary_feature.GetField(fieldname), rtol=1e-6)
 
         output_dir = os.path.join(args['workspace_dir'], 'output')
         self._assert_urban_nature(os.path.join(
@@ -679,8 +679,8 @@ def test_radii_by_pop_group_exponential_kernal(self):
             set(defn.GetName() for defn in summary_layer.schema),
             set(expected_field_values.keys()))
         for fieldname, expected_value in expected_field_values.items():
-            self.assertAlmostEqual(
-                expected_value, summary_feature.GetField(fieldname))
+            numpy.testing.assert_allclose(
+                expected_value, summary_feature.GetField(fieldname), rtol=1e-6)
 
         output_dir = os.path.join(args['workspace_dir'], 'output')
         self._assert_urban_nature(os.path.join(