mirror of https://github.com/tqdm/tqdm.git
tests: add `dask`
This commit is contained in:
parent
28823f4288
commit
3cab8835ad
|
@ -25,6 +25,7 @@ dependencies:
|
|||
- flake8-comprehensions
|
||||
- coverage
|
||||
# extras
|
||||
- dask # dask
|
||||
- matplotlib # gui
|
||||
- numpy # pandas, keras, contrib.tenumerate
|
||||
- pandas
|
||||
|
|
|
@ -0,0 +1,19 @@
|
|||
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
|
Loading…
Reference in New Issue