diff options
author | Raymond Hettinger <python@rcn.com> | 2003-10-26 15:34:50 (GMT) |
---|---|---|
committer | Raymond Hettinger <python@rcn.com> | 2003-10-26 15:34:50 (GMT) |
commit | d591f666de4c06cd36624fd513d5bc7d8c031ec8 (patch) | |
tree | 69b9a23154c1b1053173bf25421a1cc2c4ce19e0 /Lib/test/test_itertools.py | |
parent | f0c5aec85f30ed16d1a8b4d3113bd056881bbf6f (diff) | |
download | cpython-d591f666de4c06cd36624fd513d5bc7d8c031ec8.zip cpython-d591f666de4c06cd36624fd513d5bc7d8c031ec8.tar.gz cpython-d591f666de4c06cd36624fd513d5bc7d8c031ec8.tar.bz2 |
Replace the window() example with pairwise() which demonstrates tee().
Diffstat (limited to 'Lib/test/test_itertools.py')
-rw-r--r-- | Lib/test/test_itertools.py | 23 |
1 files changed, 10 insertions, 13 deletions
diff --git a/Lib/test/test_itertools.py b/Lib/test/test_itertools.py index e12aa41..07219a9 100644 --- a/Lib/test/test_itertools.py +++ b/Lib/test/test_itertools.py @@ -623,16 +623,10 @@ Samuele ... else: ... return starmap(func, repeat(args, times)) ->>> def window(seq, n=2): -... "Returns a sliding window (of width n) over data from the iterable" -... " s -> (s0,s1,...s[n-1]), (s1,s2,...,sn), ... " -... it = iter(seq) -... result = tuple(islice(it, n)) -... if len(result) == n: -... yield result -... for elem in it: -... result = result[1:] + (elem,) -... yield result +>>> def pairwise(iterable): +... "s -> (s0,s1), (s1,s2), (s2, s3), ..." +... a, b = tee(iterable) +... return izip(a, islice(b, 1, None)) This is not part of the examples but it tests to make sure the definitions perform as purported. @@ -681,10 +675,13 @@ False >>> take(5, imap(int, repeatfunc(random.random))) [0, 0, 0, 0, 0] ->>> list(window('abc')) -[('a', 'b'), ('b', 'c')] +>>> list(pairwise('abcd')) +[('a', 'b'), ('b', 'c'), ('c', 'd')] ->>> list(window('abc',5)) +>>> list(pairwise([])) +[] + +>>> list(pairwise('a')) [] >>> list(islice(padnone('abc'), 0, 6)) |