tests: add test_partial_iteration_on_streams_using_concurrency

ebonnal · Dec 30, 2023 · 03a00a7 · 03a00a7
1 parent dfc6f08
commit 03a00a7
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/test_stream.py b/tests/test_stream.py
@@ -251,6 +251,12 @@ def test_flatten(self, concurrency) -> None:
             )
         )
 
+    def test_partial_iteration_on_streams_using_concurrency(self) -> None:
+        next(iter(Stream(src).batch().flatten(concurrency=2)))
+        next(iter(Stream(src).map(identity, concurrency=2)))
+        next(iter(Stream(src).do(identity, concurrency=2)))
+
+
     def test_flatten_typing(self) -> None:
         flattened_iterator_stream: Stream[str] = (
             Stream("abc".__iter__).map(iter).flatten()