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())
