def test_layer_with_kwargs(): l = ConstructionWrapper.create('FooLayerImpl', name='foo', a=2, b=3) descr = get_layer_description(l.layer) assert descr == { '@type': 'Foo', '@outgoing_connections': {}, 'a': 2, 'b': 3 }
def test_get_layer_description(): l = ConstructionWrapper.create('FooLayerImpl', name='foo') l2 = ConstructionWrapper.create('FooLayerImpl', name='bar') l3 = ConstructionWrapper.create('FooLayerImpl', name='baz') _ = l >> l2 _ = l >> l3 descr = get_layer_description(l.layer) assert descr == { '@type': 'Foo', '@outgoing_connections': { 'default': ['bar.default', 'baz.default'] } }
def test_get_layer_description_named_inputs_outputs(): l = ConstructionWrapper.create('FooLayerImpl', name='foo') l2 = ConstructionWrapper.create('FooLayerImpl', name='bar') l3 = ConstructionWrapper.create('FooLayerImpl', name='baz') _ = l - 'out1' >> l2 _ = l >> 'A' - l3 descr = get_layer_description(l.layer) assert descr == { '@type': 'Foo', '@outgoing_connections': { 'default': ['baz.A'], 'out1': ['bar.default'] } }