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