fix: tests

xgi-org · Oct 16, 2023 · c3e6c1d · c3e6c1d
1 parent ee380a9
commit c3e6c1d
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/tests/drawing/test_draw.py b/tests/drawing/test_draw.py
@@ -232,12 +232,16 @@ def test_draw_simplices(edgelist8):
     S = xgi.SimplicialComplex(edgelist8)
 
     fig, ax = plt.subplots()
-    ax = xgi.draw_simplices(S, ax=ax)
+    ax, collections = xgi.draw_simplices(S, ax=ax)
+    (dyad_collection, edge_collection) = collections
 
     # number of elements
-    assert len(ax.lines) == 18  # dyads
-    assert len(ax.patches) == 3  # hyperedges
-    assert len(ax.collections) == 0  # nodes
+    assert len(ax.lines) == 0 
+    assert len(ax.patches) == 0 
+    assert len(ax.collections) == 2 
+
+    assert len(dyad_collection.get_paths()) == 16  # dyads
+    assert len(edge_collection.get_paths()) == 3  # other hyperedges
 
     # zorder
     for line in ax.lines:  # dyads