def test_xy_to_array_origin(): h, w = 3, 5 array = np.arange(h * w).reshape(h, w, 1) out = xy_to_array_origin(array_to_xy_origin(array.copy())) assert np.allclose(out, array)