()
| 219 | |
| 220 | |
| 221 | def test_fuse_roots(): |
| 222 | x = da.ones(10, chunks=(2,)) |
| 223 | y = da.zeros(10, chunks=(2,)) |
| 224 | z = (x + 1) + (2 * y**2) |
| 225 | (zz,) = dask.optimize(z) |
| 226 | # assert len(zz.dask) == 5 |
| 227 | assert sum(map(dask.istask, zz.dask.values())) == 5 # there are some aliases |
| 228 | assert_eq(zz, z) |
| 229 | |
| 230 | |
| 231 | def test_fuse_roots_annotations(): |