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