()
| 321 | |
| 322 | |
| 323 | def test_take(): |
| 324 | chunks, dsk = take("y-y", "x", [(20, 20, 20, 20)], [5, 1, 47, 3], axis=0) |
| 325 | assert len(dsk) == 6 |
| 326 | assert chunks == ((4,),) |
| 327 | |
| 328 | chunks, dsk = take("y-y", "x", [(20, 20, 20, 20), (20, 20)], [5, 1, 47, 3], axis=0) |
| 329 | assert len(dsk) == 9 |
| 330 | assert chunks == ((4,), (20, 20)) |
| 331 | |
| 332 | |
| 333 | def test_take_sorted(): |