def main(prt=sys.stdout):
"""Plot tiny graph"""
fin_graphs = ['tinyG.txt', 'mediumG.txt']
for fin_graph in fin_graphs:
graph = Graph(cli_get_fin(join(TEST_DIR, fin_graph)))
prt.write("{}\n".format(graph))
fout_png = fin_graph.replace('txt', 'png')
graph.wr_png(fout_png)
def test_1(prt=sys.stdout):
adjtxtblk = """
A: E B
B: E A F
C: D F
D: C G H
E: A B
F: C G B
G: D H F
H: G D """
# Convert adjacency list in text block to array, fmt=[V, E, edge_pairs]
g = Graph(adjtxt=adjtxtblk)
prt.write("{}\n".format(g))
g.wr_png("Graph_test_1.png")
def test_1(prt=sys.stdout):
"""Test that Graph is implemented properly"""
adjtxtblk = """
A: E B
B: E A F
C: D F
D: C G H
E: A B
F: C G B
G: D H F
H: G D """
# Convert adjacency list in text block to array, fmt=[V, E, edge_pairs]
graph = Graph(adjtxt=adjtxtblk)
assert graph.num_nodes == 8
assert graph.num_edges == 10
prt.write("{}\n".format(graph))
graph.wr_png("Graph_test_1.png")
def run(a, prt=sys.stdout):
g = Graph(a)
prt.write("{}\n".format(g))
g.wr_png("Graph_test_0.png")
def _run(data, prt=sys.stdout):
graph = Graph(data)
prt.write("{}\n".format(graph))
graph.wr_png("Graph_test_0.png")
return graph
