diff env/lib/python3.9/site-packages/networkx/readwrite/json_graph/tests/test_jit.py @ 0:4f3585e2f14b draft default tip

"planemo upload commit 60cee0fc7c0cda8592644e1aad72851dec82c959"

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/env/lib/python3.9/site-packages/networkx/readwrite/json_graph/tests/test_jit.py	Mon Mar 22 18:12:50 2021 +0000
@@ -0,0 +1,64 @@
+import json
+import pytest
+import networkx as nx
+from networkx.readwrite.json_graph import jit_data, jit_graph
+
+
+class TestJIT:
+    def test_jit(self):
+        G = nx.Graph()
+        G.add_node("Node1", node_data="foobar")
+        G.add_node("Node3", node_data="bar")
+        G.add_node("Node4")
+        G.add_edge("Node1", "Node2", weight=9, something="isSomething")
+        G.add_edge("Node2", "Node3", weight=4, something="isNotSomething")
+        G.add_edge("Node1", "Node2")
+        d = jit_data(G)
+        K = jit_graph(json.loads(d))
+        assert nx.is_isomorphic(G, K)
+
+    def test_jit_2(self):
+        G = nx.Graph()
+        G.add_node(1, node_data=3)
+        G.add_node(3, node_data=0)
+        G.add_edge(1, 2, weight=9, something=0)
+        G.add_edge(2, 3, weight=4, something=3)
+        G.add_edge(1, 2)
+        d = jit_data(G)
+        K = jit_graph(json.loads(d))
+        assert nx.is_isomorphic(G, K)
+
+    def test_jit_directed(self):
+        G = nx.DiGraph()
+        G.add_node(1, node_data=3)
+        G.add_node(3, node_data=0)
+        G.add_edge(1, 2, weight=9, something=0)
+        G.add_edge(2, 3, weight=4, something=3)
+        G.add_edge(1, 2)
+        d = jit_data(G)
+        K = jit_graph(json.loads(d), create_using=nx.DiGraph())
+        assert nx.is_isomorphic(G, K)
+
+    def test_jit_multi_directed(self):
+        G = nx.MultiDiGraph()
+        G.add_node(1, node_data=3)
+        G.add_node(3, node_data=0)
+        G.add_edge(1, 2, weight=9, something=0)
+        G.add_edge(2, 3, weight=4, something=3)
+        G.add_edge(1, 2)
+        pytest.raises(nx.NetworkXNotImplemented, jit_data, G)
+
+        H = nx.DiGraph(G)
+        d = jit_data(H)
+        K = jit_graph(json.loads(d), create_using=nx.MultiDiGraph())
+        assert nx.is_isomorphic(H, K)
+        K.add_edge(1, 2)
+        assert not nx.is_isomorphic(H, K)
+        assert nx.is_isomorphic(G, K)
+
+    def test_jit_round_trip(self):
+        G = nx.Graph()
+        d = nx.jit_data(G)
+        H = jit_graph(json.loads(d))
+        K = jit_graph(d)
+        assert nx.is_isomorphic(H, K)