Use allclose instead of almost_equal

szhan · Sep 18, 2023 · 18c6e3a · 18c6e3a
1 parent 8de7de7
commit 18c6e3a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)