()
| 621 | |
| 622 | |
| 623 | def test_optimize(): |
| 624 | x = dask.delayed(inc)(1) |
| 625 | y = dask.delayed(inc)(x) |
| 626 | z = x + y |
| 627 | |
| 628 | x2, y2, z2, constant = optimize(x, y, z, 1) |
| 629 | assert constant == 1 |
| 630 | |
| 631 | # Same graphs for each |
| 632 | dsk = dict(x2.dask) |
| 633 | assert dict(y2.dask) == dsk |
| 634 | assert dict(z2.dask) == dsk |
| 635 | |
| 636 | # Computationally equivalent |
| 637 | assert dask.compute(x2, y2, z2) == dask.compute(x, y, z) |
| 638 | |
| 639 | |
| 640 | def test_optimize_nested(): |