Add test

Gurobi · Oct 18, 2023 · f8bf258 · f8bf258
1 parent 9ed1faa
commit f8bf258
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tests/test_bipartite_matching.py b/tests/test_bipartite_matching.py
@@ -126,6 +126,17 @@ def test_random(self):
         self.assertEqual(matching.shape, adjacency.shape)
         self.assert_is_unweighted_matching(matching)
 
+    def test_random_csr_matrix(self):
+        # Property test for matchings on random graphs
+        adjacency, nodes1, nodes2 = random_bipartite(n1=8, n2=7, p=0.5, seed=98634)
+
+        matching = maximum_bipartite_matching(sp.csr_matrix(adjacency), nodes1, nodes2)
+
+        self.assertIsInstance(matching, sp.spmatrix)
+        self.assertIsNot(matching, adjacency)
+        self.assertEqual(matching.shape, adjacency.shape)
+        self.assert_is_unweighted_matching(matching)
+
     def test_random_csr_array(self):
         # Property test for matchings on random graphs
         adjacency, nodes1, nodes2 = random_bipartite(n1=8, n2=7, p=0.5, seed=98634)