-

cool-RR · cool-RR · commit c57618ca37ae · 2014-10-22T16:16:33.000+03:00
diff --git a/source_py2/test_python_toolbox/test_combi/test_extensive.py b/source_py2/test_python_toolbox/test_combi/test_extensive.py
@@ -83,7 +83,7 @@ def __iter__(self):
         if (self.is_recurrent and self.is_combination):
             def make_iterator():
                 crap = set()
-                for item in map(self.perm_type, self._iter()):
+                for item in itertools.imap(self.perm_type, self._iter()):
                     fc = nifty_collections.FrozenBag(item)
                     if fc in crap:
                         continue
@@ -92,7 +92,7 @@ def make_iterator():
                         crap.add(fc)
             iterator = make_iterator()
         else:
-            iterator = iter(map(self.perm_type, self._iter()))
+            iterator = iter(itertools.imap(self.perm_type, self._iter()))
         if self.slice_:
             return itertools.islice(iterator, self.slice_.start,
                                     self.slice_.stop)
@@ -148,7 +148,6 @@ class FruityPerm(FruityMixin, Perm): pass
 class FruityComb(FruityMixin, Comb): pass
 class FruityTuple(FruityMixin, tuple): pass
 
-
 def _check_variation_selection(variation_selection, perm_space_type,
                                iterable_or_length_and_sequence, domain_to_cut,
                                n_elements, is_combination, purified_fixed_map,
@@ -264,7 +263,7 @@ def _check_variation_selection(variation_selection, perm_space_type,
         # really not.
     
     for i, (perm, brute_perm_tuple) in enumerate(
-                      itertools.islice(zip(perm_space, brute_perm_space), 10)):
+           itertools.islice(itertools.izip(perm_space, brute_perm_space), 10)):
         
         assert tuple(perm) == brute_perm_tuple
         assert perm in perm_space
