def test_custom_edge_weight(self):
vertice1 = Vertice("one")
vertice2 = Vertice("two")
edge = Edge(vertice1, vertice2, 5)
self.assertEqual(5, edge.weight())
def test_default_edge_weight(self):
vertice1 = Vertice("one")
vertice2 = Vertice("two")
edge = Edge(vertice1, vertice2)
self.assertEqual(0, edge.weight())
