def get_default_graph(): default_graph = Graph() default_graph.add_vertex(0) default_graph.add_vertex(1) default_graph.add_vertex(2) default_graph.add_vertex(3) default_graph.add_vertex(4) default_graph.add_vertex(5) default_graph.add_vertex(6) default_graph.add_vertex(7) default_graph.add_vertex(8) default_graph.add_vertex(9) default_graph.add_vertex(10) default_graph.add_vertex(11) default_graph.add_edge(0, 1, False) default_graph.add_edge(0, 3, False) default_graph.add_edge(1, 2, False) default_graph.add_edge(2, 4, False) default_graph.add_edge(4, 9, False) default_graph.add_edge(2, 4, False) default_graph.add_edge(3, 6, False) default_graph.add_edge(7, 8, False) default_graph.get_all_connected() #print(default_graph.breath_first(0, 7)) # default_graph.breath_first(0, 9) drawn_tree = BokehGraph(default_graph) drawn_tree.draw_graph()
def get_random_graph(num_verts, num_edges): graph = Graph() for vert_id in range(0, num_verts): graph.add_vertex(vert_id) all_edges = [] for i in range(0, num_verts): for j in range(0, num_verts): if i < j: all_edges.append((i, j)) random.shuffle(all_edges) random_edges = all_edges[:num_edges] for edge in random_edges: graph.add_edge(edge[0], edge[1], False) graph.get_all_connected() drawn_tree = BokehGraph(graph) drawn_tree.draw_graph()