def test_ghost():
x = np.arange(64).reshape((8, 8))
d = da.from_array(x, chunks=(4, 4))
g = ghost(d, depth={0: 2, 1: 1}, boundary={0: 100, 1: "reflect"})
assert g.chunks == ((8, 8), (6, 6))
expected = np.array(
[
[100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100],
[100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100],
[0, 0, 1, 2, 3, 4, 3, 4, 5, 6, 7, 7],
[8, 8, 9, 10, 11, 12, 11, 12, 13, 14, 15, 15],
[16, 16, 17, 18, 19, 20, 19, 20, 21, 22, 23, 23],
[24, 24, 25, 26, 27, 28, 27, 28, 29, 30, 31, 31],
[32, 32, 33, 34, 35, 36, 35, 36, 37, 38, 39, 39],
[40, 40, 41, 42, 43, 44, 43, 44, 45, 46, 47, 47],
[16, 16, 17, 18, 19, 20, 19, 20, 21, 22, 23, 23],
[24, 24, 25, 26, 27, 28, 27, 28, 29, 30, 31, 31],
[32, 32, 33, 34, 35, 36, 35, 36, 37, 38, 39, 39],
[40, 40, 41, 42, 43, 44, 43, 44, 45, 46, 47, 47],
[48, 48, 49, 50, 51, 52, 51, 52, 53, 54, 55, 55],
[56, 56, 57, 58, 59, 60, 59, 60, 61, 62, 63, 63],
[100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100],
[100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100],
]
)
assert eq(g, expected)
assert same_keys(g, ghost(d, depth={0: 2, 1: 1}, boundary={0: 100, 1: "reflect"}))
g = ghost(d, depth={0: 2, 1: 1}, boundary={0: 100})
assert g.chunks == ((8, 8), (5, 5))
def test_ghost():
x = np.arange(64).reshape((8, 8))
d = da.from_array(x, chunks=(4, 4))
g = ghost(d, depth={0: 2, 1: 1}, boundary={0: 100, 1: 'reflect'})
assert g.chunks == ((8, 8), (6, 6))
expected = np.array(
[[100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100],
[100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100],
[0, 0, 1, 2, 3, 4, 3, 4, 5, 6, 7, 7],
[8, 8, 9, 10, 11, 12, 11, 12, 13, 14, 15, 15],
[16, 16, 17, 18, 19, 20, 19, 20, 21, 22, 23, 23],
[24, 24, 25, 26, 27, 28, 27, 28, 29, 30, 31, 31],
[32, 32, 33, 34, 35, 36, 35, 36, 37, 38, 39, 39],
[40, 40, 41, 42, 43, 44, 43, 44, 45, 46, 47, 47],
[16, 16, 17, 18, 19, 20, 19, 20, 21, 22, 23, 23],
[24, 24, 25, 26, 27, 28, 27, 28, 29, 30, 31, 31],
[32, 32, 33, 34, 35, 36, 35, 36, 37, 38, 39, 39],
[40, 40, 41, 42, 43, 44, 43, 44, 45, 46, 47, 47],
[48, 48, 49, 50, 51, 52, 51, 52, 53, 54, 55, 55],
[56, 56, 57, 58, 59, 60, 59, 60, 61, 62, 63, 63],
[100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100],
[100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100]])
assert eq(g, expected)
g = ghost(d, depth={0: 2, 1: 1}, boundary={0: 100})
assert g.chunks == ((8, 8), (5, 5))
def test_nearest_ghost():
a = np.arange(144).reshape(12, 12).astype(float)
darr = da.from_array(a, chunks=(6, 6))
garr = ghost(darr, depth={0: 5, 1: 5}, boundary={0: "nearest", 1: "nearest"})
tarr = trim_internal(garr, {0: 5, 1: 5})
assert_array_almost_equal(tarr, a)
def test_nearest_ghost():
a = np.arange(144).reshape(12, 12).astype(float)
darr = da.from_array(a, chunks=(6, 6))
garr = ghost(darr, depth={0: 5, 1: 5},
boundary={0: 'nearest', 1: 'nearest'})
tarr = trim_internal(garr, {0: 5, 1: 5})
assert_array_almost_equal(tarr, a)
def test_depth_greater_than_boundary_length():
expected = np.arange(100).reshape(10, 10)
darr = da.from_array(expected, chunks=(5, 5))
depth = {0: 8, 1: 7}
reflected = ghost(darr, depth=depth, boundary='reflect')
nearest = ghost(darr, depth=depth, boundary='nearest')
periodic = ghost(darr, depth=depth, boundary='periodic')
constant = ghost(darr, depth=depth, boundary=42)
result = trim_internal(reflected, depth)
assert_array_equal(result, expected)
result = trim_internal(nearest, depth)
assert_array_equal(result, expected)
result = trim_internal(periodic, depth)
assert_array_equal(result, expected)
result = trim_internal(constant, depth)
assert_array_equal(result, expected)
def test_depth_greater_than_boundary_length():
expected = np.arange(100).reshape(10, 10)
darr = da.from_array(expected, chunks=(5, 5))
depth = {0: 8, 1: 7}
reflected = ghost(darr, depth=depth, boundary="reflect")
nearest = ghost(darr, depth=depth, boundary="nearest")
periodic = ghost(darr, depth=depth, boundary="periodic")
constant = ghost(darr, depth=depth, boundary=42)
result = trim_internal(reflected, depth)
assert_array_equal(result, expected)
result = trim_internal(nearest, depth)
assert_array_equal(result, expected)
result = trim_internal(periodic, depth)
assert_array_equal(result, expected)
result = trim_internal(constant, depth)
assert_array_equal(result, expected)
def test_depth_equals_boundary_length():
expected = np.arange(100).reshape(10, 10)
darr = da.from_array(expected, chunks=(5, 5))
depth = {0: 5, 1: 5}
reflected = ghost(darr, depth=depth, boundary='reflect')
nearest = ghost(darr, depth=depth, boundary='nearest')
periodic = ghost(darr, depth=depth, boundary='periodic')
constant = ghost(darr, depth=depth, boundary=42)
result = trim_internal(reflected, depth)
assert_array_equal(result, expected)
result = trim_internal(nearest, depth)
assert_array_equal(result, expected)
result = trim_internal(periodic, depth)
assert_array_equal(result, expected)
result = trim_internal(constant, depth)
assert_array_equal(result, expected)
def test_one_chunk_along_axis():
a = np.arange(2 * 9).reshape(2, 9)
darr = da.from_array(a, chunks=((2, ), (2, 2, 2, 3)))
g = ghost(darr, depth=0, boundary=0)
assert a.shape == g.shape
def test_one_chunk_along_axis():
a = np.arange(2 * 9).reshape(2, 9)
darr = da.from_array(a, chunks=((2,), (2, 2, 2, 3)))
g = ghost(darr, depth=0, boundary=0)
assert a.shape == g.shape
