parsemis_wrapper_test.py 文件源码

def test_multidigraph_is_subgraph_with_labels(self):
        G = nx.MultiDiGraph()
        G.add_edges_from(
            [
                (1, 2, {'label': 'a'}),
                (1, 2, {'label': 'f'}),
                (2, 3, {'label': 'c'}),
                (3, 4, {'label': 'd'}),
                (2, 4, {'label': 'b'})
            ]
        )

        valid_subgraph_a = nx.MultiDiGraph()
        valid_subgraph_a.add_edge(1, 2, label='a')

        valid_subgraph_b = nx.MultiDiGraph()
        valid_subgraph_b.add_edge(1, 2, label='f')

        valid_subgraph_c = nx.MultiDiGraph()
        valid_subgraph_c.add_edge(3, 4, label='d')

        invalid_subgraph_a = nx.MultiDiGraph()
        invalid_subgraph_a.add_edge(1, 2, label='b')

        invalid_subgraph_b = nx.MultiDiGraph()
        invalid_subgraph_b.add_edge(2, 1, label='a')

        invalid_subgraph_c = nx.MultiDiGraph()
        invalid_subgraph_c.add_edge(1, 4, label='a')

        self.assertTrue(ParsemisMiner.is_subgraph(G, valid_subgraph_a))
        self.assertTrue(ParsemisMiner.is_subgraph(G, valid_subgraph_b))
        self.assertTrue(ParsemisMiner.is_subgraph(G, valid_subgraph_c))

        self.assertFalse(ParsemisMiner.is_subgraph(G, invalid_subgraph_a))
        self.assertFalse(ParsemisMiner.is_subgraph(G, invalid_subgraph_b))
        self.assertFalse(ParsemisMiner.is_subgraph(G, invalid_subgraph_c))