1 parent c4a8d63 commit 421ed61Copy full SHA for 421ed61
1 file changed
tests/test_gpu_aamp_stimp.py
@@ -59,7 +59,7 @@ def test_gpu_aamp_stimp(T):
59
naive.replace_inf(ref_PAN)
60
naive.replace_inf(cmp_PAN)
61
62
- npt.assert_almost_equal(ref_PAN, cmp_PAN)
+ npt.assert_allclose(cmp_PAN, ref_PAN, atol=1.5e-07)
63
64
# Compare transformed pan
65
cmp_pan = pan.PAN_
@@ -76,4 +76,4 @@ def test_gpu_aamp_stimp(T):
76
naive.replace_inf(ref_pan)
77
naive.replace_inf(cmp_pan)
78
79
- npt.assert_almost_equal(ref_pan, cmp_pan)
+ npt.assert_allclose(cmp_pan, ref_pan, atol=1.5e-07)
0 commit comments