def test_non_tensor_input():
shape = (2, 4)
population = [[1, 2, 3, 4], [1, 2, 3, 4]]
inputs = Input(shape)
inputs.assign(population)
res = inputs.get()
assert B.is_tensor(res)
def test_2d():
shape = (128, 64, 64)
population = B.randint(1, 10, shape=shape)
inputs = Input(shape)
inputs.assign(population)
assert B.tensor_equal(inputs.get(), population)
def test_call_vs_get():
shape = (128, 64)
population = B.randint(1, 10, shape=shape)
inputs = Input(shape)
inputs.assign(population)
assert B.tensor_equal(inputs.get(), inputs.call(''))