def create_graph(graph_data): is_graph = graph_data[0] is 'G' graph = Graph() for vertex in graph_data[1].split(','): graph.addVertex(vertex) counter = 0 for word in graph_data[2:]: counter += 1 if is_graph: graph.addEdge(word[3], word[1], word[5:].replace(')', '')) graph.addEdge(word[1], word[3], word[5:].replace(')', '')) else: graph.addEdge(word[1], word[3], word[5:].replace(')', '')) return graph, counter
def test_add_edge(self): graph = Graph() lst = ["A", "B", "C", "D"] [graph.addVertex(vert) for vert in lst] graph.addEdge('A', 'B')