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