TST: Update exp test to check for spurious over and underflow flag

pmli · Feb 3, 2022 · e3cb0a7 · e3cb0a7
1 parent 0d1d76e
commit e3cb0a7
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/numpy/core/tests/test_umath.py b/numpy/core/tests/test_umath.py
@@ -995,12 +995,13 @@ def test_exp_strides(self):
 
 class TestSpecialFloats:
     def test_exp_values(self):
-        x = [np.nan,  np.nan, np.inf, 0.]
-        y = [np.nan, -np.nan, np.inf, -np.inf]
-        for dt in ['f', 'd', 'g']:
-            xf = np.array(x, dtype=dt)
-            yf = np.array(y, dtype=dt)
-            assert_equal(np.exp(yf), xf)
+        with np.errstate(under='raise', over='raise'):
+            x = [np.nan,  np.nan, np.inf, 0.]
+            y = [np.nan, -np.nan, np.inf, -np.inf]
+            for dt in ['f', 'd', 'g']:
+                xf = np.array(x, dtype=dt)
+                yf = np.array(y, dtype=dt)
+                assert_equal(np.exp(yf), xf)
 
     # See: https://github.com/numpy/numpy/issues/19192
     @pytest.mark.xfail(