def test_dask_rolling(axis, window, center):
import dask.array as da
x = np.array(np.random.randn(100, 40), dtype=float)
dx = da.from_array(x, chunks=[(6, 30, 30, 20, 14), 8])
expected = rolling_window(x,
axis=axis,
window=window,
center=center,
fill_value=np.nan)
actual = rolling_window(dx,
axis=axis,
window=window,
center=center,
fill_value=np.nan)
assert isinstance(actual, da.Array)
assert_array_equal(actual, expected)
assert actual.shape == expected.shape
# we need to take care of window size if chunk size is small
# window/2 should be smaller than the smallest chunk size.
with pytest.raises(ValueError):
rolling_window(dx,
axis=axis,
window=100,
center=center,
fill_value=np.nan)
def test_dask_rolling(axis, window, center):
import dask.array as da
x = np.array(np.random.randn(100, 40), dtype=float)
dx = da.from_array(x, chunks=[(6, 30, 30, 20, 14), 8])
expected = rolling_window(x, axis=axis, window=window, center=center,
fill_value=np.nan)
actual = rolling_window(dx, axis=axis, window=window, center=center,
fill_value=np.nan)
assert isinstance(actual, da.Array)
assert_array_equal(actual, expected)
assert actual.shape == expected.shape
# we need to take care of window size if chunk size is small
# window/2 should be smaller than the smallest chunk size.
with pytest.raises(ValueError):
rolling_window(dx, axis=axis, window=100, center=center,
fill_value=np.nan)
