Skip to content

Commit 40c0318

Browse files
committed
add test_map_buffersize_on_multiple_infinite_iterables
1 parent b2bd1ad commit 40c0318

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

Lib/test/test_concurrent_futures/executor.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -110,7 +110,17 @@ def test_map_buffersize_on_infinite_iterable(self):
110110
self.assertEqual(next(res, None), "0")
111111
self.assertEqual(next(res, None), "1")
112112
self.assertEqual(next(res, None), "2")
113-
self.assertEqual(next(res, None), "3")
113+
114+
def test_map_buffersize_on_multiple_infinite_iterables(self):
115+
res = self.executor.map(
116+
add,
117+
itertools.count(),
118+
itertools.count(),
119+
buffersize=2
120+
)
121+
self.assertEqual(next(res, None), 0)
122+
self.assertEqual(next(res, None), 2)
123+
self.assertEqual(next(res, None), 4)
114124

115125
def test_map_buffersize_on_empty_iterable(self):
116126
res = self.executor.map(str, [], buffersize=2)

0 commit comments

Comments
 (0)