def test_edge_disconnection_single_connection(self):
vertice1 = Vertice("one")
vertice2 = Vertice("two")
edge = Edge(vertice1, vertice2)
vertice1.add_neighbor(vertice2)
self.assertTrue(edge.connected())
self.assertTrue(vertice1.is_neighbor(vertice2))
edge.disconnect_vertices()
self.assertFalse(vertice1.is_neighbor(vertice2))
def test_edge_disconnect_idempotency(self):
vertice1 = Vertice("one")
vertice2 = Vertice("two")
edge = Edge(vertice1, vertice2)
edge.connect_vertices()
self.assertTrue(edge.connected())
edge.disconnect_vertices()
self.assertFalse(edge.connected())
edge.disconnect_vertices()
self.assertFalse(edge.connected())
def test_edge_disconnection_single_connection_other_way(self):
"""Should be equivalent to previous test"""
vertice1 = Vertice("one")
vertice2 = Vertice("two")
edge = Edge(vertice1, vertice2)
vertice2.add_neighbor(vertice1)
self.assertTrue(edge.connected())
self.assertTrue(vertice2.is_neighbor(vertice1))
edge.disconnect_vertices()
self.assertFalse(vertice2.is_neighbor(vertice1))