From 5d7c36e64b7cc6f2d94ceb383b97b3d657827d91 Mon Sep 17 00:00:00 2001 From: viknesh-ai Date: Sat, 15 Aug 2026 00:11:25 +0530 Subject: [PATCH] Migrate tests/test_aamp_stimp.py from npt.assert_almost_equal to npt.assert_allclose --- tests/test_aamp_stimp.py | 20 +++++++++++--------- 1 file changed, 11 insertions(+), 9 deletions(-) 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)