Skip to content

Commit 571cd81

Browse files
committed
Update test_strongly_connected.py
1 parent 3fcefbf commit 571cd81

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/digraph/test_strongly_connected.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,7 @@ def setUp(self):
130130

131131
def test_condensation(self):
132132
# Call the condensation function
133-
condensed_graph, node_map = rustworkx.condensation(self.graph)
133+
condensed_graph = rustworkx.condensation(self.graph)
134134

135135
# Check the number of nodes (two cycles should be condensed into one node each)
136136
self.assertEqual(
@@ -158,6 +158,7 @@ def test_condensation_with_sccs_argument(self):
158158
sccs = rustworkx.strongly_connected_components(self.graph)
159159
# Call condensation with explicit sccs argument
160160
condensed_graph = rustworkx.condensation(self.graph, sccs=sccs)
161+
condensed_graph.attrs["node_map"]
161162

162163
# Check the number of nodes (should match SCC count)
163164
self.assertEqual(len(condensed_graph.node_indices()), len(sccs))

0 commit comments

Comments
 (0)