def test_undirected_graph_remove_edge(self): """ Remove edges from an undirected graph """ g = UndirectedGraph() g.add_vertex(v_val='v0') g.add_vertex(v_val='v1') g.add_vertex(v_val='v2') g.add_edge(('v0', 'v1')) g.add_edge(('v0', 'v2')) g.remove_edge(('v0', 'v1')) self.assertFalse(g.has_edge(('v0', 'v1'))) self.assertFalse(g.has_edge(('v1', 'v0'))) self.assertTrue(g.has_edge(('v0', 'v2'))) g.remove_edge(('v0', 'v2')) self.assertFalse(g.has_edge(('v0', 'v1'))) self.assertFalse(g.has_edge(('v1', 'v0'))) self.assertFalse(g.has_edge(('v0', 'v2')))