def test_values(loop):
with cluster() as (c, [a, b]):
with Executor(('127.0.0.1', c['port']), loop=loop) as e:
L = [e.submit(inc, i) for i in range(5)]
wait(L)
p = MultiProgressWidget(L)
p.start()
assert set(p.all_keys) == {'inc'}
assert len(p.all_keys['inc']) == 5
assert p.status == 'finished'
assert not p.pc.is_running()
assert p.texts['inc'].value == '5 / 5'
assert p.bars['inc'].value == 1.0
x = e.submit(throws, 1)
p = MultiProgressWidget([x])
p.start()
assert p.status == 'error'
assert p.bars[p.func(x.key)].value == 1.0
def f(c, a, b):
s = Scheduler((c.ip, c.port), loop=loop)
yield s._sync_center()
done = s.start()
sched, report = Queue(), Queue(); s.handle_queues(sched, report)
msg = yield report.get(); assert msg['op'] == 'stream-start'
s.update_graph(dsk={'x-1': (inc, 1),
'x-2': (inc, 'x-1'),
'x-3': (inc, 'x-2'),
'y-1': (dec, 'x-3'),
'y-2': (dec, 'y-1'),
'e': (throws, 'y-2'),
'other': (inc, 123)},
keys=['e'])
p = MultiProgressWidget(['e'], scheduler=s)
assert p.keys == {'x': {'x-1', 'x-2', 'x-3'},
'y': {'y-1', 'y-2'},
'e': {'e'}}
while True:
msg = yield report.get()
if msg['op'] == 'key-in-memory' and msg['key'] == 'x-3':
break
assert p.keys == {'x': set(),
'y': {'y-1', 'y-2'},
'e': {'e'}}
p._update()
assert p.bars['x'].value == 1.0
assert p.bars['y'].value == 0.0
assert p.bars['e'].value == 0.0
assert '3 / 3' in p.texts['x'].value
assert '0 / 2' in p.texts['y'].value
assert '0 / 1' in p.texts['e'].value
while True:
msg = yield report.get()
if msg['op'] == 'key-in-memory' and msg['key'] == 'y-2':
break
p._update()
assert p.bars['x'].value == 1.0
assert p.bars['y'].value == 1.0
assert p.bars['e'].value == 0.0
assert p.keys == {'x': set(),
'y': set(),
'e': {'e'}}
while True:
msg = yield report.get()
if msg['op'] == 'task-erred' and msg['key'] == 'e':
break
assert p.bars['x'].bar_style == 'success'
assert p.bars['y'].bar_style == 'success'
assert p.bars['e'].bar_style == 'danger'
assert p.status == 'error'
sched.put_nowait({'op': 'close'})
yield done