Skip to content

Commit 421ed61

Browse files
authored
Migrate tests/test_gpu_aamp_stimp.py from npt.assert_almost_equal to npt.assert_allclose (#1192)
1 parent c4a8d63 commit 421ed61

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

‎tests/test_gpu_aamp_stimp.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ def test_gpu_aamp_stimp(T):
5959
naive.replace_inf(ref_PAN)
6060
naive.replace_inf(cmp_PAN)
6161

62-
npt.assert_almost_equal(ref_PAN, cmp_PAN)
62+
npt.assert_allclose(cmp_PAN, ref_PAN, atol=1.5e-07)
6363

6464
# Compare transformed pan
6565
cmp_pan = pan.PAN_
@@ -76,4 +76,4 @@ def test_gpu_aamp_stimp(T):
7676
naive.replace_inf(ref_pan)
7777
naive.replace_inf(cmp_pan)
7878

79-
npt.assert_almost_equal(ref_pan, cmp_pan)
79+
npt.assert_allclose(cmp_pan, ref_pan, atol=1.5e-07)

0 commit comments

Comments
 (0)