updated tests to new graph_elements api

jkminder · Apr 8, 2024 · 892b2aa · 892b2aa
1 parent 5305534
commit 892b2aa
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/unit/neo4j/test_graph_elements.py b/tests/unit/neo4j/test_graph_elements.py
@@ -92,8 +92,7 @@ def test_nodes():
     assert hash(n1) != hash(n3)
 
     # test non existing attribute
-    with pytest.raises(KeyError):
-        n1["non_existing"]
+    assert n1["non_existing"] is None
 
 def test_relationships():
     n1 = Node("test", id=1)
@@ -138,8 +137,7 @@ def test_relationships():
     assert hash(r1) != hash(r2)
 
     # test non existing attribute
-    with pytest.raises(KeyError):
-        r1["non_existing"]
+    r1["non_existing"] is None
 
 
 def test_nodes_from_attributes():