def test_disconnect(self): a = BaseVertex('a') b = BaseVertex('b') edge = BaseEdge(a, b) edge.disconnect() self.assertFalse(edge in a.edges) self.assertFalse(edge in b.edges)
def test_set_dst(self): a = BaseVertex('a') b = BaseVertex('b') c = BaseVertex('c') edge = BaseEdge(a, b) edge.dst = c self.assertEqual(edge.dst, c) self.assertFalse(edge in b._in_edges) self.assertTrue(edge in c._in_edges)
def test_set_src(self): a = BaseVertex('a') b = BaseVertex('b') c = BaseVertex('c') edge = BaseEdge(a, b) edge.src = c self.assertEqual(edge.src, c) self.assertFalse(edge in a._out_edges) self.assertTrue(edge in c._out_edges)