From 3567a876c7c86d443b3da18fb183ed677f260881 Mon Sep 17 00:00:00 2001
From: Raymond Hettinger <python@rcn.com>
Date: Sat, 28 Jun 2003 05:44:36 +0000
Subject: [PATCH] Add take() to examples.  Tighten the islice() example

---
 Doc/lib/libitertools.tex   | 5 ++++-
 Lib/test/test_itertools.py | 8 +++++++-
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/Doc/lib/libitertools.tex b/Doc/lib/libitertools.tex
index e146c6c989c..4f025e3c9b7 100644
--- a/Doc/lib/libitertools.tex
+++ b/Doc/lib/libitertools.tex
@@ -314,7 +314,7 @@ Check 1202 is for $823.14
 
 >>> reportlines = ['EuroPython', 'Roster', '', 'alex', '', 'laura',
                   '', 'martin', '', 'walter', '', 'samuele']
->>> for name in islice(reportlines, 3, len(reportlines), 2):
+>>> for name in islice(reportlines, 3, None, 2):
 ...    print name.title()
 ...
 Alex
@@ -380,4 +380,7 @@ from building blocks.
 ...         result = result[1:] + (elem,)
 ...         yield result
 
+>>> def take(n, seq):
+...     return list(islice(seq, n))
+
 \end{verbatim}
diff --git a/Lib/test/test_itertools.py b/Lib/test/test_itertools.py
index 846a690836f..db7e3bdba2e 100644
--- a/Lib/test/test_itertools.py
+++ b/Lib/test/test_itertools.py
@@ -392,7 +392,7 @@ Check 1202 is for $823.14
 27
 
 >>> reportlines = ['EuroPython', 'Roster', '', 'alex', '', 'laura', '', 'martin', '', 'walter', '', 'samuele']
->>> for name in islice(reportlines, 3, len(reportlines), 2):
+>>> for name in islice(reportlines, 3, None, 2):
 ...    print name.title()
 ...
 Alex
@@ -449,6 +449,9 @@ Samuele
 ...         result = result[1:] + (elem,)
 ...         yield result
 
+>>> def take(n, seq):
+...     return list(islice(seq, n))
+
 This is not part of the examples but it tests to make sure the definitions
 perform as purported.
 
@@ -494,6 +497,9 @@ False
 >>> dotproduct([1,2,3], [4,5,6])
 32
 
+>>> take(10, count())
+[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
+
 """
 
 __test__ = {'libreftest' : libreftest}