def test_digraph2(self):
data = nx.read_dimacs("networkx/readwrite/tests/dimacs_digraph1")
nx.write_dimacs(data[0],data[1],data[2],"networkx/readwrite/tests/dimacs_digraph1_written")
data_read = nx.read_dimacs("networkx/readwrite/tests/dimacs_digraph1_written")
assert_equal(sorted(data[0].nodes()),sorted(data_read[0].nodes()))
assert_equal(sorted(data[0].edges(data=True)),sorted(data_read[0].edges(data=True)))
assert_equal(data[1],data_read[1])
assert_equal(data[2],data_read[2])
def test_dimacs_digraph1(self):
#AK generator, small graph
#http://www.avglab.com/andrew/CATS/gens/ak/
[G,s,t] = nx.read_dimacs("networkx/algorithms/flow/tests/dimacs_digraph1")
value = nx.ford_fulkerson(G,s,t)[0]
compare_value_flows_vanilla_push_relabel(G,s,t,value)
compare_value_flows_relabel_to_front(G,s,t,value)
compare_value_flows_highest_label(G,s,t,value)
def test_digraph1(self):
G=nx.DiGraph()
edges=[(3,4),(3,6),(4,5),(4,6),(5,8),(5,6),(6,7),(7,8),(9,10),(10,11),
(11,12),(12,13),(13,14),(9,14),(10,13),(1,3),(1,9),(8,2),
(14,2)]
capacities=[3,1,2,1,1,1,3,3,2,2,2,2,2,1,1,1000000,1000000,
1000000,1000000]
s='1'
t='2'
edges_with_data=[(str(u),str(v),{'capacity':float(capacity)})
for (u,v), capacity in zip(edges,capacities)]
G.add_edges_from(edges_with_data)
data = nx.read_dimacs("networkx/readwrite/tests/dimacs_digraph1")
assert_equal(sorted(G.nodes()),sorted(data[0].nodes()))
assert_equal(sorted(G.edges(data=True)),sorted(data[0].edges(data=True)))
assert_equal(s,data[1])
assert_equal(t,data[2])
