using assert_allclose rather assert_array_equal

NOAA-National-Severe-Storms-Laboratory · Mar 12, 2022 · 5a19b60 · 5a19b60
1 parent fe31121
commit 5a19b60
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/test_object_id.py b/tests/test_object_id.py
@@ -61,11 +61,12 @@ def test_single_threshold(self):
         areas = [region.area for region in object_props]
 
         # Test that the areas are right. 
-        np.testing.assert_array_equal(true_areas, areas) 
+        np.testing.assert_allclose(true_areas, areas, atol=5)  
 
         # Test that the labels are right. 
         np.testing.assert_array_equal(np.unique(storm_labels), np.array([0,1,2,3,4]))
 
+
     def test_missing_watershed_params(self):
         """ Check that watershed does proceed if parameters are missing """
         pass
@@ -94,7 +95,7 @@ def  test_watershed_method(self):
         areas = [region.area for region in object_props]
 
         # Test that the areas are right. 
-        np.testing.assert_array_equal(true_areas, areas) 
+        np.testing.assert_allclose(true_areas, areas, atol=5) 
 
         # Test that the labels are right. 
         np.testing.assert_array_equal(np.unique(storm_labels), np.array([0,1,2,3,4,5]))
@@ -129,7 +130,7 @@ def test_iterative_watershed_method(self):
         areas = [region.area for region in object_props]
 
         # Test that the areas are right. 
-        np.testing.assert_array_equal(true_areas, areas) 
+        np.testing.assert_allclose(true_areas, areas, atol=5) 
 
         # Test that the labels are right. 
         np.testing.assert_array_equal(np.unique(storm_labels), np.array([0,1,2,3,4,5]))