test_hierarchy: compat with new NetworkX exception

ricklupton · Jul 26, 2024 · fd076f0 · fd076f0
1 parent b2807ae
commit fd076f0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_hierarchy.py b/test/test_hierarchy.py
@@ -21,5 +21,5 @@ def test_hierarchy():
     assert h('East Anglia') == "location in ['Cambridge', 'Ely', 'Escape \"']"
     assert h('*') == None
 
-    with pytest.raises(KeyError):
+    with pytest.raises((KeyError, nx.NetworkXError)):
         h('unknown')