Use np.testing

szhan · Sep 13, 2023 · 0486b0f · 0486b0f
1 parent 0279630
commit 0486b0f
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/python/tests/test_imputation.py b/python/tests/test_imputation.py
@@ -484,8 +484,7 @@ def test_compare_imputed_alleles(input_ref, input_query, expected):
         imputed_alleles[i], _ = tests.beagle.run_beagle(
             input_ref, input_query[i], pos, miscall_rate=0.0001, ne=10.0
         )
-        num_diff = np.sum(imputed_alleles[i] != expected_subset[i])
-        assert num_diff == 0, f"Wrongly imputed alleles: {num_diff}"
+        np.testing.assert_array_equal(imputed_alleles[i], expected_subset[i])
 
 
 @pytest.mark.parametrize(