graph_test.py 文件源码

def testSCC(self):
        a = Vertex('a')        
        b = Vertex('b')        
        c = Vertex('c')        
        d = Vertex('d')        
        e = Vertex('e')        
        f = Vertex('f')        
        g = Vertex('g')        
        h = Vertex('h')        
        vertices = [a, b, c, d, e, f, g, h]
        edges = [(e, a), (a, b), (b, c), (d, c), (c, d), (b, e), (e, f), (b, f), (g, f), (f, g), (c, g), (g, h), (h, h)]    
        G = Graph(vertices, edges)
        G.strongly_connected_components()
        self.assertEquals(a.cc, 1)
        self.assertEquals(b.cc, 1)
        self.assertEquals(c.cc, 2)
        self.assertEquals(d.cc, 2)
        self.assertEquals(e.cc, 1)
        self.assertEquals(f.cc, 3)
        self.assertEquals(g.cc, 3)
        self.assertEquals(h.cc, 4)