Mercurial > repos > shellac > sam_consensus_v3
view env/lib/python3.9/site-packages/networkx/readwrite/json_graph/tests/test_tree.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 source
import json import pytest import networkx as nx from networkx.readwrite.json_graph import tree_data, tree_graph class TestTree: def test_graph(self): G = nx.DiGraph() G.add_nodes_from([1, 2, 3], color="red") G.add_edge(1, 2, foo=7) G.add_edge(1, 3, foo=10) G.add_edge(3, 4, foo=10) H = tree_graph(tree_data(G, 1)) nx.is_isomorphic(G, H) def test_graph_attributes(self): G = nx.DiGraph() G.add_nodes_from([1, 2, 3], color="red") G.add_edge(1, 2, foo=7) G.add_edge(1, 3, foo=10) G.add_edge(3, 4, foo=10) H = tree_graph(tree_data(G, 1)) assert H.nodes[1]["color"] == "red" d = json.dumps(tree_data(G, 1)) H = tree_graph(json.loads(d)) assert H.nodes[1]["color"] == "red" def test_exception(self): with pytest.raises(nx.NetworkXError): G = nx.MultiDiGraph() G.add_node(0) attrs = dict(id="node", children="node") tree_data(G, 0, attrs)