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