def test_swap_tiles_on_image_with_empty_tiles(): img = np.array([[1, 1, 1, 1], [2, 2, 2, 2], [3, 3, 3, 3], [4, 4, 4, 4]], dtype=np.uint8) result_img = F.swap_tiles_on_image(img, []) assert np.array_equal(img, result_img)
def test_swap_tiles_on_image_with_non_empty_tiles(): img = np.array([[1, 1, 1, 1], [2, 2, 2, 2], [3, 3, 3, 3], [4, 4, 4, 4]], dtype=np.uint8) tiles = np.array([[0, 0, 2, 2, 2, 2], [2, 2, 0, 0, 2, 2]]) target = np.array([[3, 3, 1, 1], [4, 4, 2, 2], [3, 3, 1, 1], [4, 4, 2, 2]], dtype=np.uint8) result_img = F.swap_tiles_on_image(img, tiles) assert np.array_equal(result_img, target)