Mercurial > repos > shellac > sam_consensus_v3
diff env/lib/python3.9/site-packages/networkx/readwrite/json_graph/tests/test_jit.py @ 0:4f3585e2f14b draft default tip
"planemo upload commit 60cee0fc7c0cda8592644e1aad72851dec82c959"
author | shellac |
---|---|
date | Mon, 22 Mar 2021 18:12:50 +0000 |
parents | |
children |
line wrap: on
line diff
--- /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)