TST: assert_allclose behavior with nans that are also infs

danra · danra · commit 260cddd45bdf · 2025-07-13T22:09:44.000-07:00
diff --git a/numpy/testing/tests/test_utils.py b/numpy/testing/tests/test_utils.py
@@ -1295,11 +1295,21 @@ def test_equal_nan(self):
         # Should not raise:
         assert_allclose(a, b, equal_nan=True)
 
+        a = np.array([complex(np.nan, np.inf)])
+        b = np.array([complex(np.nan, np.inf)])
+        assert_allclose(a, b, equal_nan=True)
+        b = np.array([complex(np.nan, -np.inf)])
+        assert_allclose(a, b, equal_nan=True)
+
     def test_not_equal_nan(self):
         a = np.array([np.nan])
         b = np.array([np.nan])
         assert_raises(AssertionError, assert_allclose, a, b, equal_nan=False)
 
+        a = np.array([complex(np.nan, np.inf)])
+        b = np.array([complex(np.nan, np.inf)])
+        assert_raises(AssertionError, assert_allclose, a, b, equal_nan=False)
+
     def test_equal_nan_default(self):
         # Make sure equal_nan default behavior remains unchanged. (All
         # of these functions use assert_array_compare under the hood.)
