test: exercise node_link_data with invalid data

roedoejet · Sep 11, 2024 · b5e2258 · b5e2258
1 parent b10f22e
commit b5e2258
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/g2p/tests/test_network.py b/g2p/tests/test_network.py
@@ -167,6 +167,24 @@ def test_node_link_data(self):
         graph = node_link_graph(self.data)
         self.assertEqual(node_link_data(graph), self.data)
 
+    def test_node_link_graph_errors(self):
+        with self.assertRaises(ValueError):
+            node_link_graph({**self.data, "directed": False})
+        with self.assertRaises(ValueError):
+            node_link_graph({**self.data, "multigraph": True})
+        with self.assertRaises(ValueError):
+            node_link_graph({**self.data, "nodes": "not a list"})
+        with self.assertRaises(ValueError):
+            node_link_graph({**self.data, "links": "not a list"})
+        with self.assertRaises(ValueError):
+            data = self.data.copy()
+            del data["nodes"]
+            node_link_graph(data)
+        with self.assertRaises(ValueError):
+            data = self.data.copy()
+            del data["links"]
+            node_link_graph(data)
+
     def test_no_duplicates(self):
         graph = DiGraph()
         graph.add_edge("a", "b")