summaryrefslogtreecommitdiffstats
path: root/Lib/test
diff options
context:
space:
mode:
authorRaymond Hettinger <python@rcn.com>2008-07-19 23:58:47 (GMT)
committerRaymond Hettinger <python@rcn.com>2008-07-19 23:58:47 (GMT)
commitf1f46f0350390cf86ec015f9f60eda41a8210033 (patch)
tree6bfc82fedbc8676fefe4e74b30b19363fb3accf5 /Lib/test
parent39e0eb766f02e18711d7eac9f948754a1ee569e3 (diff)
downloadcpython-f1f46f0350390cf86ec015f9f60eda41a8210033.zip
cpython-f1f46f0350390cf86ec015f9f60eda41a8210033.tar.gz
cpython-f1f46f0350390cf86ec015f9f60eda41a8210033.tar.bz2
Clean-up itertools docs and recipes.
Diffstat (limited to 'Lib/test')
-rw-r--r--Lib/test/test_itertools.py72
1 files changed, 17 insertions, 55 deletions
diff --git a/Lib/test/test_itertools.py b/Lib/test/test_itertools.py
index 2d07d8d..1a4e936 100644
--- a/Lib/test/test_itertools.py
+++ b/Lib/test/test_itertools.py
@@ -1185,52 +1185,32 @@ Samuele
[22]
[25, 26, 27, 28]
->>> def take(n, seq):
-... return list(islice(seq, n))
+>>> def take(n, iterable):
+... "Return first n items of the iterable as a list"
+... return list(islice(iterable, n))
->>> def enumerate(iterable):
-... return izip(count(), iterable)
+>>> def enumerate(iterable, start=0):
+... return izip(count(start), iterable)
->>> def tabulate(function):
+>>> def tabulate(function, start=0):
... "Return function(0), function(1), ..."
-... return imap(function, count())
-
->>> def iteritems(mapping):
-... return izip(mapping.iterkeys(), mapping.itervalues())
+... return imap(function, count(start))
>>> def nth(iterable, n):
-... "Returns the nth item"
+... "Returns the nth item or empty list"
... return list(islice(iterable, n, n+1))
->>> def all(seq, pred=None):
-... "Returns True if pred(x) is true for every element in the iterable"
-... for elem in ifilterfalse(pred, seq):
-... return False
-... return True
-
->>> def any(seq, pred=None):
-... "Returns True if pred(x) is true for at least one element in the iterable"
-... for elem in ifilter(pred, seq):
-... return True
-... return False
-
->>> def no(seq, pred=None):
-... "Returns True if pred(x) is false for every element in the iterable"
-... for elem in ifilter(pred, seq):
-... return False
-... return True
-
->>> def quantify(seq, pred=None):
-... "Count how many times the predicate is true in the sequence"
-... return sum(imap(pred, seq))
-
->>> def padnone(seq):
+>>> def quantify(iterable, pred=bool):
+... "Count how many times the predicate is true"
+... return sum(imap(pred, iterable))
+
+>>> def padnone(iterable):
... "Returns the sequence elements and then returns None indefinitely"
-... return chain(seq, repeat(None))
+... return chain(iterable, repeat(None))
->>> def ncycles(seq, n):
-... "Returns the sequence elements n times"
-... return chain(*repeat(seq, n))
+>>> def ncycles(iterable, n):
+... "Returns the seqeuence elements n times"
+... return chain(*repeat(iterable, n))
>>> def dotproduct(vec1, vec2):
... return sum(imap(operator.mul, vec1, vec2))
@@ -1315,24 +1295,6 @@ perform as purported.
>>> nth('abcde', 3)
['d']
->>> all([2, 4, 6, 8], lambda x: x%2==0)
-True
-
->>> all([2, 3, 6, 8], lambda x: x%2==0)
-False
-
->>> any([2, 4, 6, 8], lambda x: x%2==0)
-True
-
->>> any([1, 3, 5, 9], lambda x: x%2==0,)
-False
-
->>> no([1, 3, 5, 9], lambda x: x%2==0)
-True
-
->>> no([1, 2, 5, 9], lambda x: x%2==0)
-False
-
>>> quantify(xrange(99), lambda x: x%2==0)
50