def test_remove_out_edge(self): edge = Mock() vertex = BaseVertex('spam') vertex.add_out_edge(edge) vertex.remove_out_edge(edge) self.assertFalse(vertex._out_edges) self.assertFalse(vertex.edges)
def test_add_out_edge(self): vertex = BaseVertex('spam') edge = Mock() vertex.add_out_edge(edge) self.assertTrue(edge in vertex._out_edges) self.assertTrue(edge in vertex.edges)