summaryrefslogtreecommitdiffstats
path: root/Lib/test/test_itertools.py
diff options
context:
space:
mode:
authorRaymond Hettinger <python@rcn.com>2016-03-07 02:11:38 (GMT)
committerRaymond Hettinger <python@rcn.com>2016-03-07 02:11:38 (GMT)
commite525ee3b483889313b62b09b4c9103d18dd9cbb7 (patch)
tree8530e435acbe1737a06eb7fe0439181215e593bb /Lib/test/test_itertools.py
parentd66dd5ce68cbf4a33c385439d5eeb2bff4e860f1 (diff)
downloadcpython-e525ee3b483889313b62b09b4c9103d18dd9cbb7.zip
cpython-e525ee3b483889313b62b09b4c9103d18dd9cbb7.tar.gz
cpython-e525ee3b483889313b62b09b4c9103d18dd9cbb7.tar.bz2
Document another recipe for itertools: all_equal(). Inspired by David Beazley.
Diffstat (limited to 'Lib/test/test_itertools.py')
-rw-r--r--Lib/test/test_itertools.py8
1 files changed, 8 insertions, 0 deletions
diff --git a/Lib/test/test_itertools.py b/Lib/test/test_itertools.py
index 08c9f39..f940852 100644
--- a/Lib/test/test_itertools.py
+++ b/Lib/test/test_itertools.py
@@ -2014,6 +2014,11 @@ Samuele
... "Returns the nth item or a default value"
... return next(islice(iterable, n, None), default)
+>>> def all_equal(iterable):
+... "Returns True if all the elements are equal to each other"
+... g = groupby(iterable)
+... return next(g, True) and not next(g, False)
+
>>> def quantify(iterable, pred=bool):
... "Count how many times the predicate is true"
... return sum(map(pred, iterable))
@@ -2127,6 +2132,9 @@ perform as purported.
>>> nth('abcde', 9) is None
True
+>>> [all_equal(s) for s in ('', 'A', 'AAAA', 'AAAB', 'AAABA')]
+[True, True, True, False, False]
+
>>> quantify(range(99), lambda x: x%2==0)
50