123456789101112131415161718 |
- 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
|