| 627 | |
| 628 | |
| 629 | def test_array_bag_delayed(): |
| 630 | np = pytest.importorskip("numpy") |
| 631 | da = pytest.importorskip("dask.array") |
| 632 | |
| 633 | arr1 = np.arange(100).reshape((10, 10)) |
| 634 | arr2 = arr1.dot(arr1.T) |
| 635 | darr1 = da.from_array(arr1, chunks=(5, 5)) |
| 636 | darr2 = da.from_array(arr2, chunks=(5, 5)) |
| 637 | b = db.from_sequence([1, 2, 3]) |
| 638 | seq = [arr1, arr2, darr1, darr2, b] |
| 639 | out = delayed(sum)([i.sum() for i in seq]) |
| 640 | assert out.compute() == 2 * arr1.sum() + 2 * arr2.sum() + sum([1, 2, 3]) |
| 641 | |
| 642 | |
| 643 | def test_delayed_picklable(): |