NXOntology.n_edges

nxontology/ontology.py

@@ -246,7 +246,16 @@ def n_nodes(self) -> int:
         """
         Total number of nodes in the graph.
         """
-        return len(self.graph)
+        # int wrapper is solely to satisfy mypy
+        return int(self.graph.number_of_nodes())
+
+    @property
+    def n_edges(self) -> int:
+        """
+        Total number of edges in the graph.
+        """
+        # int wrapper is solely to satisfy mypy
+        return int(self.graph.number_of_edges())
 
     def set_graph_attributes(
         self,

nxontology/tests/ontology_test.py

@@ -12,6 +12,14 @@
 from nxontology.ontology import Node_Info, NXOntology, Similarity, SimilarityIC
 
 
+def test_n_nodes(metal_nxo: NXOntology[str]) -> None:
+    assert metal_nxo.n_nodes == 8
+
+
+def test_n_edges(metal_nxo: NXOntology[str]) -> None:
+    assert metal_nxo.n_edges == 9
+
+
 def test_add_node(metal_nxo: NXOntology[str]) -> None:
     assert "brass" not in metal_nxo.graph
     metal_nxo.add_node("brass", color="#b5a642")