()
| 49 | |
| 50 | |
| 51 | def test_shuffle_larger_array(): |
| 52 | arr = da.random.random((15, 15, 15), chunks=(5, 5, 5)) |
| 53 | indexer = np.arange(0, 15) |
| 54 | np.random.shuffle(indexer) |
| 55 | indexer = [indexer[0:6], indexer[6:8], indexer[8:9], indexer[9:]] |
| 56 | indexer = list(map(list, indexer)) |
| 57 | take_indexer = list(flatten(indexer)) |
| 58 | assert_eq(shuffle(arr, indexer, axis=1), arr[..., take_indexer, :]) |
| 59 | |
| 60 | |
| 61 | def test_incompatible_indexer(darr): |