diff --git a/tests/run/numpy_math.pyx b/tests/run/numpy_math.pyx index eafd23a..4a15522 100644 --- a/tests/run/numpy_math.pyx +++ b/tests/run/numpy_math.pyx @@ -37,8 +37,8 @@ def test_fp_classif(): assert not npmath.isnan(d_zero) assert not npmath.isnan(f_zero) - assert npmath.isinf(npmath.INFINITY) == 1 - assert npmath.isinf(-npmath.INFINITY) == -1 + assert npmath.isinf(npmath.INFINITY) != 0 + assert npmath.isinf(-npmath.INFINITY) != 0 assert npmath.isnan(npmath.NAN) assert npmath.signbit(npmath.copysign(1., -1.))