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()))