Python XLayer.subgraph_data Examples

Example #1

File: test_xlayer.py Project: williamyang4978/pyxir

    def test_xlayer_subgraph_data(self):

        X_sub = XLayer(name='x_sub', type=['Sub'])

        X = XLayer(subgraph="xp0", subgraph_data=[X_sub])

        assert isinstance(X.target, str)

        sg_data = X.subgraph_data
        assert len(sg_data) == 1
        assert sg_data[0].name == 'x_sub'
        assert sg_data[0].type == ['Sub']

        sg_data[0].type[0] = 'Sub2'
        assert sg_data[0].name == 'x_sub'
        assert sg_data[0].type == ['Sub2']

        X.subgraph_data = [sg_data[0], XLayer(name='x_sub3', type=['Sub3'])]
        assert len(X.subgraph_data) == 2
        assert X.subgraph_data[1].name == 'x_sub3'
        assert X.subgraph_data[1].type == ['Sub3']

Example #2

File: test_xlayer.py Project: eghasemi89/pyxir

    def test_xlayer_subgraph_data(self):

        X_sub = XLayer(name="x_sub", type=["Sub"])

        X = XLayer(subgraph="xp0", subgraph_data=[X_sub])

        assert isinstance(X.target, str)

        sg_data = X.subgraph_data
        assert len(sg_data) == 1
        assert sg_data[0].name == "x_sub"
        assert sg_data[0].type == ["Sub"]

        sg_data[0].type[0] = "Sub2"
        assert sg_data[0].name == "x_sub"
        assert sg_data[0].type == ["Sub2"]

        X.subgraph_data = [sg_data[0], XLayer(name="x_sub3", type=["Sub3"])]
        assert len(X.subgraph_data) == 2
        assert X.subgraph_data[1].name == "x_sub3"
        assert X.subgraph_data[1].type == ["Sub3"]