def test_simple(self): g = Graph(3) self.assertEqual(g.find_all_distances(0), [-1, -1]) g.connect(0, 1) self.assertEqual(g.find_all_distances(1), [6, -1])
def test_failing(self): g = Graph(4) g.connect(3, 1) g.connect(0, 1) g.connect(0, 2) self.assertEqual(g.find_all_distances(0), [6, 6, 12])