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