Test ignored arguments for rank.

mcmtroffaes · Sep 18, 2024 · a7af3b3 · a7af3b3
1 parent badc20e
commit a7af3b3
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/test/test_matrix_rank.py b/test/test_matrix_rank.py
@@ -47,3 +47,23 @@ def test_matrix_rank_1(
         basis_cols,
         rank,
     )
+
+
+def test_matrix_rank_2() -> None:
+    # check ignored_cols and ignored_rows
+    assert cdd.matrix_rank(
+        cdd.matrix_from_array([[0, 1], [1, 0]]), ignored_rows={0}
+    ) == ({1}, {0}, 1)
+    assert cdd.matrix_rank(
+        cdd.matrix_from_array([[0, 1], [1, 0]]), ignored_cols={0}
+    ) == ({0}, {1}, 1)
+    assert cdd.matrix_rank(
+        cdd.matrix_from_array([[1, 0, 0], [0, 0, 1], [0, 1, 0]]),
+        ignored_rows={1},
+        ignored_cols={2},
+    ) == ({0, 2}, {0, 1}, 2)
+    assert cdd.matrix_rank(
+        cdd.matrix_from_array([[1, 1, 0], [0, 0, 1], [2, 2, 0], [0, 1, 0]]),
+        ignored_rows={1},
+        ignored_cols={2},
+    ) == ({0, 3}, {0, 1}, 2)