Fix support for iterators using "repeat"

Author: rhettinger

more_itertools/more.py

@@ -4198,7 +4198,7 @@ def nth_product(index, *iterables, repeat=1):
 
     ``IndexError`` will be raised if the given *index* is invalid.
     """
-    pools = tuple(map(tuple, reversed(iterables * repeat)))
+    pools = tuple(map(tuple, reversed(iterables))) * repeat
     ns = tuple(map(len, pools))
 
     c = reduce(mul, ns)
@@ -4362,7 +4362,7 @@ def product_index(element, *iterables, repeat=1):
     of *args*.
     """
     elements = tuple(element)
-    pools = tuple(map(tuple, iterables * repeat))
+    pools = tuple(map(tuple, iterables)) * repeat
     if len(elements) != len(pools):
         raise ValueError('element is not a product of args')
 

tests/test_more.py

@@ -4864,6 +4864,10 @@ def test_repeat(self):
             mi.nth_product(123, 'AB', 'CD', 'EFG', repeat=2),
             mi.nth_product(123, 'AB', 'CD', 'EFG', 'AB', 'CD', 'EFG'),
         )
+        self.assertEqual(
+            mi.nth_product(123, iter('AB'), iter('CD'), iter('EFG'), repeat=2),
+            mi.nth_product(123, 'AB', 'CD', 'EFG', 'AB', 'CD', 'EFG'),
+        )
 
 
 class NthCombinationWithReplacementTests(TestCase):
@@ -4982,6 +4986,12 @@ def test_repeat(self):
             mi.product_index(target, 'AB', 'CD', 'EFG', repeat=2),
             mi.product_index(target, 'AB', 'CD', 'EFG', 'AB', 'CD', 'EFG'),
         )
+        self.assertEqual(
+            mi.product_index(
+                iter(target), iter('AB'), iter('CD'), iter('EFG'), repeat=2
+            ),
+            mi.product_index(target, 'AB', 'CD', 'EFG', 'AB', 'CD', 'EFG'),
+        )
 
 
 class CombinationIndexTests(TestCase):