def test_shuffle_slice(size, chunks): x = da.random.randint(0, 1000, size=size, chunks=chunks) index = np.arange(len(x)) np.random.shuffle(index) a = x[index] b = shuffle_slice(x, index) assert_eq(a, b)
def permutation(self, x): from dask.array.slicing import shuffle_slice if isinstance(x, numbers.Number): x = arange(x, chunks="auto") index = np.arange(len(x)) self._numpy_state.shuffle(index) return shuffle_slice(x, index)