def test_nested_sequential_network(): current_node = InputNode("foo", shape=(3, 4, 5)) for name in map(str, range(10)): current_node = SequentialNode("sequential" + name, [current_node, IdentityNode("identity" + name)]) network = current_node.network() fn = network.function(["foo"], ["sequential9"]) x = np.random.rand(3, 4, 5).astype(fX) assert np.allclose(fn(x), x)
def test_identity_network(): input_node = InputNode("foo", shape=(3, 4, 5)) network = input_node.network() fn = network.function(["foo"], ["foo"]) x = np.random.rand(3, 4, 5).astype(fX) assert np.allclose(fn(x), x)