graph_test.py 文件源码

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