From 8ea3a8728a5d1decd907296620bddbf0e538cffc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Radek=20Hu=C5=A1ek?= <husek@iuuk.mff.cuni.cz>
Date: Thu, 29 Jul 2021 21:40:41 +0200
Subject: [PATCH] parmap: add test case

---
 parmap.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/parmap.py b/parmap.py
index 2ecaa54..f2ff81c 100644
--- a/parmap.py
+++ b/parmap.py
@@ -83,6 +83,16 @@ def parmap(f, X, nprocs = None, chunksize = 1, chunks_in_flight = None,
 
     >>> list(parmap(lambda x: x+1, range(10)))
     [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
+
+    >>> from time import sleep
+    >>> def w(d):
+    ...   sleep(d)
+    ...   while d < 3:
+    ...     yield d
+    ...     sleep(1)
+    ...     d += 1
+    >>> list(parmap(w, [0, 0.5], nprocs=2, multimap=True, in_order=False, out_chunksize=1))
+    [0, 0.5, 1, 1.5, 2, 2.5]
     """
     if nprocs is None:
       nprocs = multiprocessing.cpu_count()
-- 
GitLab