async def test_dont_optimize_out(c, s, a, b):
da = pytest.importorskip("dask.array")
x = da.arange(10, chunks=(5, ))
y = x.map_blocks(lambda x: x + 1)
z = y.map_blocks(lambda x: 2 * x)
w = z.sum()
await c.compute(w, resources={tuple(y.__dask_keys__()): {"A": 1}})
for key in map(tokey, y.__dask_keys__()):
assert "executing" in str(a.story(key))
def test_dont_optimize_out(c, s, a, b):
da = pytest.importorskip('dask.array')
x = da.arange(10, chunks=(5,))
y = x.map_blocks(lambda x: x + 1)
z = y.map_blocks(lambda x: 2 * x)
w = z.sum()
yield c.compute(w, resources={tuple(y.__dask_keys__()): {'A': 1}},)
for key in map(tokey, y.__dask_keys__()):
assert 'executing' in str(a.story(key))
def test_dont_optimize_out(c, s, a, b):
da = pytest.importorskip('dask.array')
x = da.arange(10, chunks=(5,))
y = x.map_blocks(lambda x: x + 1)
z = y.map_blocks(lambda x: 2 * x)
w = z.sum()
yield c.compute(w, resources={tuple(y.__dask_keys__()): {'A': 1}},)
for key in map(tokey, y.__dask_keys__()):
assert 'executing' in str(a.story(key))