diff --git a/tests/test_aamp_stimp.py b/tests/test_aamp_stimp.py index 3e83724ab..aed0ef758 100644 --- a/tests/test_aamp_stimp.py +++ b/tests/test_aamp_stimp.py @@ -68,7 +68,7 @@ def test_aamp_stimp_1_percent(T): naive.replace_inf(ref_PAN) naive.replace_inf(cmp_PAN) - npt.assert_almost_equal(ref_PAN, cmp_PAN) + npt.assert_allclose(cmp_PAN, ref_PAN, atol=1.5e-07) # Compare transformed pan cmp_pan = pan.PAN_ @@ -85,7 +85,7 @@ def test_aamp_stimp_1_percent(T): naive.replace_inf(ref_pan) naive.replace_inf(cmp_pan) - npt.assert_almost_equal(ref_pan, cmp_pan) + npt.assert_allclose(cmp_pan, ref_pan, atol=1.5e-07) @pytest.mark.parametrize("T", T) @@ -126,7 +126,7 @@ def test_aamp_stimp_max_m(T): naive.replace_inf(ref_PAN) naive.replace_inf(cmp_PAN) - npt.assert_almost_equal(ref_PAN, cmp_PAN) + npt.assert_allclose(cmp_PAN, ref_PAN, atol=1.5e-07) # Compare transformed pan cmp_pan = pan.PAN_ @@ -143,7 +143,7 @@ def test_aamp_stimp_max_m(T): naive.replace_inf(ref_pan) naive.replace_inf(cmp_pan) - npt.assert_almost_equal(ref_pan, cmp_pan) + npt.assert_allclose(cmp_pan, ref_pan, atol=1.5e-07) @pytest.mark.parametrize("T", T) @@ -178,7 +178,7 @@ def test_aamp_stimp_100_percent(T): naive.replace_inf(ref_PAN) naive.replace_inf(cmp_PAN) - npt.assert_almost_equal(ref_PAN, cmp_PAN) + npt.assert_allclose(cmp_PAN, ref_PAN, atol=1.5e-07) # Compare transformed pan cmp_pan = pan.PAN_ @@ -195,7 +195,7 @@ def test_aamp_stimp_100_percent(T): naive.replace_inf(ref_pan) naive.replace_inf(cmp_pan) - npt.assert_almost_equal(ref_pan, cmp_pan) + npt.assert_allclose(cmp_pan, ref_pan, atol=1.5e-07) @pytest.mark.parametrize("T", T) @@ -226,7 +226,9 @@ def test_stimp_raw_mp(T): naive.replace_inf(ref_P_) naive.replace_inf(cmp_P_) - npt.assert_almost_equal(ref_P_, cmp_P_) + npt.assert_allclose( + cmp_P_.astype(np.float64), ref_P_.astype(np.float64), atol=1.5e-07 + ) @pytest.mark.filterwarnings("ignore:numpy.dtype size changed") @@ -264,7 +266,7 @@ def test_aamp_stimped(T, dask_cluster): naive.replace_inf(ref_PAN) naive.replace_inf(cmp_PAN) - npt.assert_almost_equal(ref_PAN, cmp_PAN) + npt.assert_allclose(cmp_PAN, ref_PAN, atol=1.5e-07) # Compare transformed pan cmp_pan = pan.PAN_ @@ -281,4 +283,4 @@ def test_aamp_stimped(T, dask_cluster): naive.replace_inf(ref_pan) naive.replace_inf(cmp_pan) - npt.assert_almost_equal(ref_pan, cmp_pan) + npt.assert_allclose(cmp_pan, ref_pan, atol=1.5e-07)