| 681 | |
| 682 | |
| 683 | def test_persist_nested(): |
| 684 | a = delayed(1) + 5 |
| 685 | b = a + 1 |
| 686 | c = a + 2 |
| 687 | result = persist({"a": a, "b": [1, 2, b]}, (c, 2)) |
| 688 | assert isinstance(result[0]["a"], Delayed) |
| 689 | assert isinstance(result[0]["b"][2], Delayed) |
| 690 | assert isinstance(result[1][0], Delayed) |
| 691 | assert compute(*result) == ({"a": 6, "b": [1, 2, 7]}, (8, 2)) |
| 692 | |
| 693 | res = persist([a, b], c, traverse=False) |
| 694 | assert res[0][0] is a |
| 695 | assert res[0][1] is b |
| 696 | assert res[1].compute() == 8 |
| 697 | |
| 698 | |
| 699 | def test_persist_delayed(): |