def test_cull(): a = {"x": 1, "y": (inc, "x")} hg = HighLevelGraph({"a": a}, {"a": set()}) culled_by_x = hg.cull({"x"}) assert dict(culled_by_x) == {"x": 1} culled_by_y = hg.cull({"y"}) assert dict(culled_by_y) == a
def test_cull(): a = {"x": 1, "y": (inc, "x")} hg = HighLevelGraph({"a": a}, {"a": set()}) culled_by_x = hg.cull({"x"}) assert dict(culled_by_x) == {"x": 1} # parameter is the raw output of __dask_keys__() culled_by_y = hg.cull([[["y"]]]) assert dict(culled_by_y) == a
def test_cull(): a = {"x": 1, "y": (inc, "x")} layers = {"a": BasicLayer(a)} dependencies = {"a": set()} hg = HighLevelGraph(layers, dependencies) culled_by_x = hg.cull({"x"}) assert dict(culled_by_x) == {"x": 1} culled_by_y = hg.cull({"y"}) assert dict(culled_by_y) == a