add FlatMap generator example

tandav · Jul 3, 2023 · a32f664 · a32f664
1 parent dca99fd
commit a32f664
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/docs/reference.md b/docs/reference.md
@@ -141,6 +141,9 @@ Same as `FilterKeys` but for `v` in `(k, v)` pairs
 >>> [range(0, 5), range(100, 105)] | FlatMap(yield_even) | Pipe(list)
 [0, 2, 4, 100, 102, 104]
 
+>>> [range(0, 5), range(100, 105)] | FlatMap(lambda it: (x for x in it if x % 2 == 0)) | Pipe(list)
+[0, 2, 4, 100, 102, 104]
+
 ```
 
 ## FlatMapValues

diff --git a/tests/pipe_test.py b/tests/pipe_test.py
@@ -116,6 +116,7 @@ def test_filter_values(it, f, expected):
         ([0, 2, 3, 0, 4], range, [0, 1, 0, 1, 2, 0, 1, 2, 3]),
         ([2, 3, 4], lambda x: [(x, x), (x, x)], [(2, 2), (2, 2), (3, 3), (3, 3), (4, 4), (4, 4)]),
         ([range(0, 5), range(100, 105)], yield_even, [0, 2, 4, 100, 102, 104]),
+        ([range(0, 5), range(100, 105)], lambda it: (x for x in it if x % 2 == 0), [0, 2, 4, 100, 102, 104]),
     ],
 )
 def test_flat_map(it, f, expected):