def test_TaskProgress(c, s, a, b):
tp = TaskProgress(s)
futures = c.map(slowinc, range(10), delay=0.001)
yield wait(futures)
tp.update()
d = dict(tp.source.data)
assert all(len(L) == 1 for L in d.values())
assert d['name'] == ['slowinc']
futures2 = c.map(dec, range(5))
yield wait(futures2)
tp.update()
d = dict(tp.source.data)
assert all(len(L) == 2 for L in d.values())
assert d['name'] == ['slowinc', 'dec']
del futures, futures2
while s.tasks:
yield gen.sleep(0.01)
tp.update()
assert not tp.source.data['all']
def test_TaskProgress_empty(c, s, a, b):
tp = TaskProgress(s)
tp.update()
futures = [c.submit(inc, i, key='f-' + 'a' * i) for i in range(20)]
yield wait(futures)
tp.update()
del futures
while s.tasks:
yield gen.sleep(0.01)
tp.update()
assert not any(len(v) for v in tp.source.data.values())