()
| 4106 | |
| 4107 | |
| 4108 | def test_map_blocks_delayed(): |
| 4109 | x = da.ones((10, 10), chunks=(5, 5)) |
| 4110 | y = np.ones((5, 5)) |
| 4111 | |
| 4112 | z = x.map_blocks(add, y, dtype=x.dtype) |
| 4113 | z.dask.validate() |
| 4114 | dask.optimize(z)[0].dask.validate() |
| 4115 | |
| 4116 | yy = delayed(y) |
| 4117 | zz = x.map_blocks(add, yy, dtype=x.dtype) |
| 4118 | zz.dask.validate() |
| 4119 | dask.optimize(zz)[0].dask.validate() |
| 4120 | |
| 4121 | assert_eq(z, zz) |
| 4122 | |
| 4123 | assert yy.key in zz.dask |
| 4124 | |
| 4125 | |
| 4126 | def test_no_chunks(): |