def test_clone_vertex(self): vertices = [1, 2, 3, 4, 5] g = Graph(vertices) edges = [(1, 2), (1, 3), (2, 4), (2, 5), (4, 5)] g.add_edges(edges) g.clone_vertex(g.index(2), 6) self.assertSparseGraphEqual((vertices + [6], edges + [(1, 6), (2, 6), (4, 6), (5, 6)]), g.as_sparse_graph()) g = Graph(vertices) g.add_edges(edges) g.clone_vertex(g.index(2), 3) self.assertSparseGraphEqual((vertices, edges + [(2, 3), (3, 4), (3, 5)]), g.as_sparse_graph())
def test_sparse(self): g = Graph([7, 5, 9, 4, 2]) g.add_edges([(7, 5), (5, 2), (4, 9), (4, 7), (7, 9)]) expectedVertices = [2, 4, 5, 7, 9] expectedEdges = [(5, 7), (2, 5), (4, 9), (4, 7), (7, 9)] self.assertSparseGraphEqual((expectedVertices, expectedEdges), g.as_sparse_graph())
def test_add_vertices(self): vertices = [1, 2, 3, 4, 5] g = Graph(vertices) h = Graph() h.add_vertices(vertices) self.assertSparseGraphEqual(g.as_sparse_graph(), h.as_sparse_graph())