def test_transpose_float(input_shape, expected_shape):
img = np.random.uniform(low=0.0, high=1.0, size=input_shape).astype('float32')
transposed = F.transpose(img)
assert transposed.shape == expected_shape
def test_transpose(input_shape, expected_shape):
img = np.random.randint(low=0, high=256, size=input_shape, dtype=np.uint8)
transposed = F.transpose(img)
assert transposed.shape == expected_shape
def test_transpose(input_shape, expected_shape):
img = np.random.randint(low=0, high=256, size=input_shape, dtype=np.uint8)
transposed = F.transpose(img)
assert transposed.shape == expected_shape
