def test_remap(self): network = { "directed": True, "graph": { "username": "******", }, "nodes": [ { "id": "A" }, { "id": "B" } ], "links": [ { "source": "A", "target": "B" } ], "multigraph": False } remapped = batch_project.remap_links(network) G = json_graph.node_link_graph(remapped, multigraph=False, directed=True) self.assertTrue(G.edge['A']['B'] is not None)
def test_remap2(self): network = { "directed": True, "graph": { "username": "******", }, "nodes": [{ "id": "A", "fooA": "barA" }, { "id": "B", "fooB": "barB" }, { "id": "C", "fooC": "barC" }, { "id": "D", "fooD": "barD", 'source': 'A' }, { "id": "E", "fooE": "barE" }], "links": [{ "source": "A", "target": "B" }, { "source": "A", "target": "D" }, { "source": "B", "target": "C" }, { "source": "C", "target": "D", "split": True }, { "source": "D", "target": "E", "foo": "bar" }], "multigraph": False } remapped = batch_project.remap_links(network) G = json_graph.node_link_graph(remapped, multigraph=False, directed=True) G = batch_project.separate_paths(G) for node_id in G.nodes(): preds = [ pred for pred in G.predecessors(node_id) if not getValue(G[pred][node_id], 'donor', False) ] self.assertTrue(len(preds) < 2) self.assertEqual(7, len(G.nodes()))