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)