From 18c6e3ab1da1ca1f7c035ccf695a5f4739244415 Mon Sep 17 00:00:00 2001 From: Shing Zhan Date: Mon, 18 Sep 2023 14:19:30 +0100 Subject: [PATCH] Use allclose instead of almost_equal --- python/tests/test_imputation.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/tests/test_imputation.py b/python/tests/test_imputation.py index 96135fe90f..15c858f001 100644 --- a/python/tests/test_imputation.py +++ b/python/tests/test_imputation.py @@ -694,10 +694,10 @@ def test_get_weights(genotyped_pos, ungenotyped_pos, expected_weights, expected_ actual_weights, actual_idx = tests.beagle.get_weights( genotyped_pos, ungenotyped_pos ) - np.testing.assert_array_almost_equal(actual_weights, expected_weights) + np.testing.assert_allclose(actual_weights, expected_weights) np.testing.assert_array_equal(actual_idx, expected_idx) actual_weights, actual_idx = tests.beagle_numba.get_weights( genotyped_pos, ungenotyped_pos ) - np.testing.assert_array_almost_equal(actual_weights, expected_weights) + np.testing.assert_allclose(actual_weights, expected_weights) np.testing.assert_array_equal(actual_idx, expected_idx)