def f(): if futures_of(self): yield wait(self) raise gen.Return(self)
async def test_await(c, s, a, b): x = dask.delayed(inc)(1) x = await x.persist() assert x.key in s.tasks assert a.data or b.data assert all(f.done() for f in futures_of(x))