def test_remove_edge_duplicate_correct(self): graph = Graph([(1, 2, 1), (1, 2, 1), (1, 2, 2), (2, 3, 1), (3, 1, 1)]) graph.remove_edge(1, 2, 1) expected = [ Edge(*args) for args in [(1, 2, 1), (1, 2, 2), (2, 3, 1), (3, 1, 1)] ] self.assertEqual(expected, graph.all_edges)
def test_remove_edge_correct(self): graph = Graph(self.edges) graph.remove_edge(1, 4, 4) expected = [ Edge(*args) for args in [(1, 2, 4), (2, 4, 1), (2, 3, 4), (3, 4, 4)] ] self.assertEqual(expected, graph.all_edges)