test_components.py 文件源码

def test_directed_no_self_edge(self):
        # Directed no self-edge
        g = sptgraph.create_spatio_temporal_graph(gen_graph(True), gen_signal(), False, verbose=False)

        # The graph has only 1 connected component
        h = components.find_connected_components(g)
        cc = components.create_component_sframe(h)
        comps = components.extract_components(h, cc)
        self.assertEqual(1, len(comps))

        # We remove the edge (7, 12) to create 2 weakly connected components
        nodes = g.vertices
        edges = g.edges.add_row_number('eid')
        to_remove = g.get_edges(7, 12)
        edges = edges[edges['eid'] != to_remove['eid'][0]]
        g = gl.SGraph(nodes, edges)

        h = components.find_connected_components(g)
        cc = components.create_component_sframe(h)
        comps = components.extract_components(h, cc)
        self.assertEqual(2, len(comps))

    # @unittest.skip('Skipping undirected self edge')