graph_test.py 文件源码

def testSquareGraph(self):
        a = Vertex(1)
        b = Vertex(2)
        c = Vertex(3)
        d = Vertex(4)
        G = Graph([a, b, c, d], [(a, b), (a, c), (c, d)])
        sqrt = G.square()
        self.assertEquals(sqrt.vertices, {a, b, c, d})
        self.assertEquals(sqrt.edges, {(a, b), (a, c), (a, d), (c, d)})
        self.assertEquals(sqrt.adj[a], {b, c, d})
        self.assertEquals(sqrt.adj[b], set())
        self.assertEquals(sqrt.adj[c], {d})
        self.assertEquals(sqrt.adj[d], set())
        a = Vertex(1)
        b = Vertex(2)
        c = Vertex(3)
        G = Graph([a, b, c], [(a, b), (b, c), (a, c)])
        sqrt = G.square()
        self.assertEquals(G, sqrt)