mirror of https://github.com/tqdm/tqdm.git
20 lines
452 B
Python
20 lines
452 B
Python
|
from __future__ import division
|
||
|
|
|
||
|
|
from time import sleep
|
||
|
|
|
||
|
|
from .tests_tqdm import importorskip, mark
|
||
|
|
|
||
|
|
pytestmark = mark.slow
|
||
|
|
|
||
|
|
|
||
|
|
def test_dask(capsys):
|
||
|
|
"""Test tqdm.dask.TqdmCallback"""
|
||
|
|
ProgressBar = importorskip('tqdm.dask').TqdmCallback
|
||
|
|
dask = importorskip('dask')
|
||
|
|
|
||
|
|
schedule = [dask.delayed(sleep)(i / 10) for i in range(5)]
|
||
|
|
with ProgressBar():
|
||
|
|
dask.compute(schedule)
|
||
|
|
_, err = capsys.readouterr()
|
||
|
|
assert '5/5' in err
|