assert_array_equals

Author: benjeffery

python/tests/test_highlevel.py

@@ -5507,7 +5507,7 @@ def test_basic_individual_nodes(self, tmp_path):
 
         result = ts.individual_nodes
         assert result.shape == (2, 2)
-        assert np.array_equal(result, [[0, 1], [2, 3]])
+        assert_array_equal(result, [[0, 1], [2, 3]])
 
     def test_variable_ploidy(self, tmp_path):
         tables = tskit.TableCollection(sequence_length=100)
@@ -5536,7 +5536,7 @@ def test_variable_ploidy(self, tmp_path):
         expected = np.array(
             [[0, 1, -1], [2, -1, -1], [3, 4, 5]]  # Diploid  # Haploid  # Triploid
         )
-        assert np.array_equal(result, expected)
+        assert_array_equal(result, expected)
 
     def test_no_individuals(self):
         tables = tskit.TableCollection(sequence_length=100)
@@ -5546,7 +5546,7 @@ def test_no_individuals(self):
         result = ts.individual_nodes
         expected = np.array([], dtype=np.int32).reshape(0, 0)
         assert result.shape == (0, 0)
-        assert np.array_equal(result, expected)
+        assert_array_equal(result, expected)
 
     def test_no_nodes_with_individuals(self):
         tables = tskit.TableCollection(sequence_length=100)
@@ -5558,7 +5558,7 @@ def test_no_nodes_with_individuals(self):
         result = ts.individual_nodes
         expected = np.array([[]])
         assert result.shape == (1, 0)
-        assert np.array_equal(result, expected)
+        assert_array_equal(result, expected)
 
     def test_individual_with_no_nodes(self):
         tables = tskit.TableCollection(sequence_length=100)
@@ -5571,7 +5571,7 @@ def test_individual_with_no_nodes(self):
         result = ts.individual_nodes
         expected = np.array([[0], [-1]])
         assert result.shape == (2, 1)
-        assert np.array_equal(result, expected)
+        assert_array_equal(result, expected)
 
     def test_mixed_sample_status(self):
         tables = tskit.TableCollection(sequence_length=100)
@@ -5583,4 +5583,4 @@ def test_mixed_sample_status(self):
         result = ts.individual_nodes
         expected = np.array([[0, 1]])
         assert result.shape == (1, 2)
-        assert np.array_equal(result, expected)
+        assert_array_equal(result, expected)