def test_random_flip(code, func, target):
img = np.array(
[[1, 1, 1],
[0, 1, 1],
[0, 0, 1]], dtype=np.uint8)
img = convert_2d_to_target_format([img], target=target)
assert np.array_equal(F.random_flip(img, code), func(img))
def test_random_flip_float(code, func, target):
img = np.array(
[[0.4, 0.4, 0.4],
[0.0, 0.4, 0.4],
[0.0, 0.0, 0.4]], dtype=np.float32)
img = convert_2d_to_target_format([img], target=target)
assert_array_almost_equal_nulp(F.random_flip(img, code), func(img))
def test_random_flip(code, func, target):
img = np.array(
[[1, 1, 1],
[0, 1, 1],
[0, 0, 1]], dtype=np.uint8)
if target == 'image':
img = convert_2d_to_3d(img)
assert np.array_equal(F.random_flip(img, code), func(img))
def test_random_flip(code, func, target):
img = np.array([[1, 1, 1], [0, 1, 1], [0, 0, 1]], dtype=np.uint8)
if target == 'image':
img = convert_2d_to_3d(img)
assert np.array_equal(F.random_flip(img, code), func(img))
