ansible-later/testenv/lib/python2.7/site-packages/toolz/tests/test_recipes.py

28 lines
820 B
Python
Raw Normal View History

2019-04-23 11:04:27 +00:00
from toolz import first, identity, countby, partitionby
def iseven(x):
return x % 2 == 0
def test_countby():
assert countby(iseven, [1, 2, 3]) == {True: 1, False: 2}
assert countby(len, ['cat', 'dog', 'mouse']) == {3: 2, 5: 1}
assert countby(0, ('ab', 'ac', 'bc')) == {'a': 2, 'b': 1}
def test_partitionby():
assert list(partitionby(identity, [])) == []
vowels = "aeiou"
assert (list(partitionby(vowels.__contains__, "abcdefghi")) ==
[("a",), ("b", "c", "d"), ("e",), ("f", "g", "h"), ("i",)])
assert (list(map(first,
partitionby(identity,
[1, 1, 1, 2, 3, 3, 2, 2, 3]))) ==
[1, 2, 3, 2, 3])
assert ''.join(map(first,
partitionby(identity, "Khhhaaaaannnnn!!!!"))) == 'Khan!'