Update test_centrality.py

xgi-org · Nov 12, 2024 · 0e5d138 · 0e5d138
1 parent 420477d
commit 0e5d138
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/algorithms/test_centrality.py b/tests/algorithms/test_centrality.py
@@ -113,8 +113,9 @@ def test_node_edge_centrality():
     assert abs(c[0] - c[1]) < 1e-6
 
     H = xgi.load_xgi_data("email-enron").cleanup()
-    c = H.edges.node_edge_centrality.asnumpy()
-    assert len(c) == H.num_nodes
+    c = xgi.node_edge_centrality(H)
+    assert len(c[0]) == H.num_nodes
+    assert len(c[1]) == H.num_edges
 
 
 def test_line_vector_centrality():