diff options
| -rw-r--r-- | numpy/core/tests/test_umath.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/numpy/core/tests/test_umath.py b/numpy/core/tests/test_umath.py index bc8927302..720bd38ce 100644 --- a/numpy/core/tests/test_umath.py +++ b/numpy/core/tests/test_umath.py @@ -1095,6 +1095,13 @@ class TestPower: assert_raises(ValueError, np.power, a, minusone) assert_raises(ValueError, np.power, one, b) assert_raises(ValueError, np.power, one, minusone) + + def test_float_to_inf_power(self): + for dt in [np.float32, np.float64]: + a = np.array([1,1,2,2,-2,-2,np.inf,-np.inf], dt) + b = np.array([np.inf, -np.inf, np.inf, -np.inf, np.inf, -np.inf, np.inf, -np.inf], dt) + r = np.array([1,1,np.inf,0,np.inf,0,np.inf,0], dt) + assert_equal(np.power(a,b), r) class TestFloat_power: |