def test_get_edge(self):
root = Vertice('root')
g1 = Graph(root)
v1 = Vertice('v1')
g1.add_vertice(v1)
edge = Edge('e1', root, v1)
g1.add_edge(edge)
self.assertEqual(edge, g1.get_edge('e1'))
def test_get_edge(self):
g1 = Graph()
v1 = Vertice('v1')
v2 = Vertice('v2')
g1.add_vertice(v1)
g1.add_vertice(v2)
edge = Edge('e1', v1, v2)
g1.add_edge(edge)
self.assertEqual(edge, g1.get_edge('e1'))