def test_remove_in_edge(self): edge = Mock() vertex = BaseVertex('spam') vertex.add_in_edge(edge) vertex.remove_in_edge(edge) self.assertFalse(vertex._in_edges) self.assertFalse(vertex.edges)
def test_add_in_edge(self): vertex = BaseVertex('spam') edge = Mock() vertex.add_in_edge(edge) self.assertTrue(edge in vertex._in_edges) self.assertTrue(edge in vertex.edges)