tests_dask.py 467 B

123456789101112131415161718
  1. from time import sleep
  2. from .tests_tqdm import importorskip, mark
  3. pytestmark = mark.slow
  4. def test_dask(capsys):
  5. """Test tqdm.dask.TqdmCallback"""
  6. ProgressBar = importorskip('tqdm.dask').TqdmCallback
  7. dask = importorskip('dask')
  8. schedule = [dask.delayed(sleep)(i / 10) for i in range(5)]
  9. with ProgressBar(desc="computing"):
  10. dask.compute(schedule)
  11. _, err = capsys.readouterr()
  12. assert "computing: " in err
  13. assert '5/5' in err