Skip to content

Commit c4a8d63

Browse files
authored
Migrate tests/test_gpu_aampdist.py from npt.assert_almost_equal to npt.assert_allclose (#1193)
1 parent a9fcfbc commit c4a8d63

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

‎tests/test_gpu_aampdist.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,6 @@ def test_gpu_aampdist(T_A, T_B):
4444
m = 3
4545
for p in [1.0, 2.0, 3.0]:
4646
ref_mpdist = naive.aampdist(T_A, T_B, m, p=p)
47-
comp_mpdist = gpu_aampdist(T_A, T_B, m, p=p)
47+
cmp_mpdist = gpu_aampdist(T_A, T_B, m, p=p)
4848

49-
npt.assert_almost_equal(ref_mpdist, comp_mpdist)
49+
npt.assert_allclose(cmp_mpdist, ref_mpdist, atol=1.5e-07)

0 commit comments

Comments
 (0)