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)