test_sptgraphfast.py 文件源码

def test_create_signal_graph(self):
        node_signal = sptgraph.create_node_signal(gen_signal(), 'baseID', 'layer', False)

        for is_directed in [True, False]:
            g = utils.networkx_to_graphlab(gen_graph(is_directed))
            sg = sptgraph.merge_signal_on_graph(g, node_signal, 'baseID', 'layer', use_fast=False, verbose=False)

            # Node 5 is never activated
            self.assertEqual(4, len(sg.vertices))

            if not is_directed:
                self.assertEqual(8, len(sg.edges))
            else:
                self.assertEqual(4, len(sg.edges))

            actual_columns = set(sg.vertices.column_names())
            expected_columns = {'layers', 'node_weight', '__id'}
            self.assertItemsEqual(expected_columns, actual_columns)

    # @unittest.skip('Skipping aggregate layers')