diff options
author | Raymond Hettinger <python@rcn.com> | 2008-07-19 23:58:47 (GMT) |
---|---|---|
committer | Raymond Hettinger <python@rcn.com> | 2008-07-19 23:58:47 (GMT) |
commit | f1f46f0350390cf86ec015f9f60eda41a8210033 (patch) | |
tree | 6bfc82fedbc8676fefe4e74b30b19363fb3accf5 /Lib/test | |
parent | 39e0eb766f02e18711d7eac9f948754a1ee569e3 (diff) | |
download | cpython-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.py | 72 |
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 |