Add take() to examples. Tighten the islice() example

1 files changed, 7 insertions, 1 deletions

diff --git a/Lib/test/test_itertools.py b/Lib/test/test_itertools.py
index 846a690..db7e3bd 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}