()
| 610 | |
| 611 | |
| 612 | def test_optimize(): |
| 613 | x = dask.delayed(inc)(1) |
| 614 | y = dask.delayed(inc)(x) |
| 615 | z = x + y |
| 616 | |
| 617 | x2, y2, z2, constant = optimize(x, y, z, 1) |
| 618 | assert constant == 1 |
| 619 | |
| 620 | # Same graphs for each |
| 621 | dsk = dict(x2.dask) |
| 622 | assert dict(y2.dask) == dsk |
| 623 | assert dict(z2.dask) == dsk |
| 624 | |
| 625 | # Computationally equivalent |
| 626 | assert dask.compute(x2, y2, z2) == dask.compute(x, y, z) |
| 627 | |
| 628 | |
| 629 | def test_optimize_nested(): |