Merge pull request #1042 from gnoack/patch-1

5-ary and 6-ary izip

Author: JelleZijlstra

stdlib/2/itertools.pyi

@@ -47,6 +47,8 @@ _T1 = TypeVar('_T1')
 _T2 = TypeVar('_T2')
 _T3 = TypeVar('_T3')
 _T4 = TypeVar('_T4')
+_T5 = TypeVar('_T5')
+_T6 = TypeVar('_T6')
 
 @overload
 def imap(func: Callable[[_T1], _S], iter1: Iterable[_T1]) -> Iterator[_S]: ...
@@ -71,7 +73,20 @@ def izip(iter1: Iterable[_T1], iter2: Iterable[_T2],
 @overload
 def izip(iter1: Iterable[_T1], iter2: Iterable[_T2], iter3: Iterable[_T3],
          iter4: Iterable[_T4]) -> Iterator[Tuple[_T1, _T2,
-                                           _T3, _T4]]: ...  # TODO more than four iterables
+                                           _T3, _T4]]: ...
+@overload
+def izip(iter1: Iterable[_T1], iter2: Iterable[_T2],
+         iter3: Iterable[_T3], iter4: Iterable[_T4],
+         iter5: Iterable[_T5]) -> Iterator[Tuple[_T1, _T2,
+                                                 _T3, _T4, _T5]]: ...
+@overload
+def izip(iter1: Iterable[_T1], iter2: Iterable[_T2],
+         iter3: Iterable[_T3], iter4: Iterable[_T4],
+         iter5: Iterable[_T5], iter6: Iterable[_T6]) -> Iterator[Tuple[_T1, _T2, _T3,
+                                                                 _T4, _T5, _T6]]: ...
+# TODO more than 6 iterables
+# TODO should have a fallback for the n-ary variant
+
 def izip_longest(*p: Iterable[Any],
                  fillvalue: Any = ...) -> Iterator[Any]: ...