Function test_annotations_survive_optimization

dask/tests/test_delayed.py:880–896

()

def test_annotations_survive_optimization():
    with dask.annotate(foo="bar"):
        graph = HighLevelGraph.from_collections(
            "b",
            {"a": 1, "b": (inc, "a"), "c": (inc, "b")},
            [],
        )
        d = Delayed("b", graph)

    assert type(d.dask) is HighLevelGraph
    assert len(d.dask.layers) == 1
    assert len(d.dask.layers["b"]) == 3
    assert d.dask.layers["b"].annotations == {"foo": "bar"}
    optimized = collections_to_expr([d]).optimize()
    assert optimized.__dask_annotations__() == {
        "foo": dict.fromkeys(optimized.__dask_graph__(), "bar")
    }


def test_delayed_function_attributes_forwarded():