Merge pull request #9 from ourstudio-se/8-filter_map_concat-needs-to-copy-input-if-iterator

8 filter map concat needs to copy input if iterator

Author: znittzel

maz/__init__.py

@@ -319,6 +319,7 @@ def __init__(
         self.fmap_function = fmap_function
 
     def __call__(self, objects: typing.Iterable[typing.Any]) -> typing.Iterable[typing.Any]:
+        objects_t, objects_f = itertools.tee(objects)
         return map(
             operator.itemgetter(1),
             sorted(
@@ -330,7 +331,7 @@ def __call__(self, objects: typing.Iterable[typing.Any]) -> typing.Iterable[typi
                                 self.filter_predicate,
                                 operator.itemgetter(1),
                             ),
-                            enumerate(objects)
+                            enumerate(objects_t)
                         ),
                     ),
 
@@ -342,7 +343,7 @@ def __call__(self, objects: typing.Iterable[typing.Any]) -> typing.Iterable[typi
                                 self.filter_predicate,
                                 operator.itemgetter(1),
                             ),
-                            enumerate(objects)
+                            enumerate(objects_f)
                         ),
                     )
                 ),

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = maz
-version = 0.0.4
+version = 0.0.5
 author = ourstudio
 author_email = rikard@ourstudio.se
 description = Functional programming tools.

tests/test_maz.py

@@ -105,13 +105,13 @@ def __init__(self, id: str, n: int):
         def __eq__(self, o):
             return (self.id == o.id) and (self.n == o.n)
 
-    items = [
+    items = iter([
         Var("a", 1),
         Var("b", 2),
         Var("c", 3),
         Var("d", 4),
         Var("e", 5),
-    ]
+    ])
 
     fmc_fn = maz.filter_map_concat(
         lambda x: x.n > 2,