18 lines
467 B
Python
18 lines
467 B
Python
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(desc="computing"):
|
|
dask.compute(schedule)
|
|
_, err = capsys.readouterr()
|
|
assert "computing: " in err
|
|
assert '5/5' in err
|