summaryrefslogtreecommitdiffstats
path: root/tests/tests_dask.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/tests_dask.py')
-rw-r--r--tests/tests_dask.py20
1 files changed, 20 insertions, 0 deletions
diff --git a/tests/tests_dask.py b/tests/tests_dask.py
new file mode 100644
index 0000000..8bf4b64
--- /dev/null
+++ b/tests/tests_dask.py
@@ -0,0 +1,20 @@
+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(desc="computing"):
+ dask.compute(schedule)
+ _, err = capsys.readouterr()
+ assert "computing: " in err
+ assert '5/5' in err