master
/ miniconda3 / pkgs / tqdm-4.65.0-py311h92b7b1e_0 / info / test / tests / tests_dask.py

tests_dask.py @a663dc1 — raw · history · blame 

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
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