def test_groupedby_failure1():
with pytest.raises(_hf.FailIter.EXC_TYP, match=_hf.FailIter.EXC_MSG):
groupedby(_hf.FailIter(), key=len)
def test_groupedby_failure3():
# keep func fails
with pytest.raises(TypeError):
groupedby([T(1), T(2), T(3)], key=lambda x: x,
keep=lambda x: T(x.value + 'a'))
def test_groupedby_reduce3():
assert groupedby(map(T, range(500)), key=lambda x: T(x.value % 5),
reduce=operator.add,
reducestart=T(0))
def test_groupedby_reduce4():
# reduce=None is identical to no reduce
assert groupedby([T(1), T(1), T(2), T(3)], lambda x: x,
reduce=None) == {T(1): [T(1), T(1)],
T(2): [T(2)], T(3): [T(3)]}
def test_groupedby_reduce1():
assert groupedby([(T('a'), T(1)), (T('a'), T(2)), (T('b'), T(5))],
key=operator.itemgetter(0),
keep=operator.itemgetter(1),
reduce=operator.add) == {T('a'): T(3), T('b'): T(5)}
def test_groupedby_reduce2():
assert groupedby([(T('a'), T(1)), (T('a'), T(2)), (T('b'), T(5))],
key=operator.itemgetter(0),
reduce=lambda x, y: x + y[1],
reducestart=T(0)) == {T('a'): T(3), T('b'): T(5)}
def test_groupedby_normal3():
# generator
assert groupedby((i for i in [T('a'), T('ab'), T('abc')]),
key=lambda x: x.value[0]
) == {'a': toT(['a', 'ab', 'abc'])}
def test_groupedby_failure7():
# reduce function fails
with pytest.raises(TypeError):
groupedby(map(T, [1, 2, 3, 4, 'a']), lambda x: True,
reduce=operator.add)
def test_groupedby_empty1():
assert groupedby([], key=lambda x: x) == {}
def test_groupedby_normal2():
assert groupedby([T('a'), T('ba'), T('ab'), T('abc'), T('b')],
key=lambda x: x.value[0]
) == {'a': toT(['a', 'ab', 'abc']),
'b': toT(['ba', 'b'])}
def test_groupedby_failure10():
# Changing next method
with pytest.raises(_hf.CacheNext.EXC_TYP, match=_hf.CacheNext.EXC_MSG):
groupedby(_hf.CacheNext(1), iteration_utilities.return_True)
def test_groupedby_failure9():
# too few arguments
with pytest.raises(TypeError):
groupedby()
def test_groupedby_failure8():
# Test that a failing iterator doesn't raise a SystemError
with pytest.raises(_hf.FailNext.EXC_TYP, match=_hf.FailNext.EXC_MSG):
groupedby(_hf.FailNext(), bool)
def test_groupedby_failure4():
# unhashable
with pytest.raises(TypeError):
groupedby([{T('a'): T(10)}], key=lambda x: x)
def test_groupedby_keep1():
assert groupedby([T('a'), T('ba'), T('ab'), T('abc'), T('b')],
key=lambda x: x.value[0],
keep=len) == {'a': [1, 2, 3], 'b': [2, 1]}
def test_groupedby_failure5():
# no reduce but reducestart
with pytest.raises(TypeError):
groupedby(toT(range(10)), lambda x: x, reducestart=T(0))
def test_groupedby_failure6():
# reduce function fails with reducestart
with pytest.raises(TypeError):
groupedby(map(T, range(10)), lambda x: x.value % 2 == 0,
reduce=operator.add, reducestart=T('a'))
