Example #1
0
    def test_change_io(self):
        """ Test set_io"""

        ins = [dict(name="in1"), dict(name="in2")]
        outs = [dict(name="out1"), dict(name="out2")]
        sg = CompositeNode(ins, outs)
        assert sg.get_nb_input() == 2
        assert sg.get_nb_output() == 2

        ins = [dict(name="in1")]
        outs = [dict(name="out1"), dict(name="out2"), dict(name="out3")]

        sg.set_io(ins, outs)
        assert sg.get_nb_input() == 1
        assert sg.get_nb_output() == 3