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)