Node_Info error if node does not exist

closes #5
related-sciences · Dec 23, 2020 · fa2ea18 · fa2ea18
1 parent 2bf6579
commit fa2ea18
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/nxontology/ontology.py b/nxontology/ontology.py
@@ -247,6 +247,8 @@ class Node_Info(Freezable):
  """
 
  def __init__(self, nxo: NXOntology, node: Node):
+ if node not in nxo.graph:
+ raise NodeNotFound(f"{node} not in graph.")
  self.nxo = nxo
  self.node = node
 

diff --git a/nxontology/tests/ontology_test.py b/nxontology/tests/ontology_test.py
@@ -97,6 +97,11 @@ def test_node_info_gold(metal_nxo_frozen: NXOntology) -> None:
  assert gold_info.n_ancestors == 4
 
 
+def test_node_info_not_found(metal_nxo_frozen: NXOntology) -> None:
+ with pytest.raises(NodeNotFound, match="not-a-metal not in graph"):
+ metal_nxo_frozen.node_info("not-a-metal")
+
+
 def test_intrinsic_ic_unscaled(metal_nxo_frozen: NXOntology) -> None:
  assert metal_nxo_frozen.n_nodes == 8
  # number of descendants per node including self