def test_traverse():
"""To test the traverse implementation we call gc.collect() while instances
of all the C objects are still valid."""
acc = iteration_utilities.accumulate([])
app = iteration_utilities.applyfunc(lambda x: x, 1)
cha = iteration_utilities.chained(int, float)
cla = iteration_utilities.clamp([], 0, 1)
com = iteration_utilities.complement(int)
con = iteration_utilities.constant(1)
dee = iteration_utilities.deepflatten([])
dup = iteration_utilities.duplicates([])
fli = iteration_utilities.flip(int)
gro = iteration_utilities.grouper([], 2)
ine = iteration_utilities.intersperse([], 1)
iik = iteration_utilities.ItemIdxKey(10, 2)
ite = iteration_utilities.iter_except(int, TypeError)
mer = iteration_utilities.merge([])
nth = iteration_utilities.nth(1)
pac = iteration_utilities.packed(int)
par = iteration_utilities.partial(int, 10)
rep = iteration_utilities.replicate([], 3)
rou = iteration_utilities.roundrobin([])
see = iteration_utilities.Seen()
sid = iteration_utilities.sideeffects([], lambda x: x)
spl = iteration_utilities.split([], lambda x: True)
sta = iteration_utilities.starfilter(lambda x: True, [])
suc = iteration_utilities.successive([])
tab = iteration_utilities.tabulate(int)
une = iteration_utilities.unique_everseen([])
unj = iteration_utilities.unique_justseen([])
gc.collect()
def test_replicate_failure_lengthhint2():
# This only checks for overflow if the length_hint is above PY_SSIZE_T_MAX
of_it = _hf.OverflowLengthHint(toT([1, 2, 3]), sys.maxsize + 1)
it = replicate(of_it, 3)
with pytest.raises(OverflowError):
operator.length_hint(it)
with pytest.raises(OverflowError):
list(it)
def test_replicate_failure_lengthhint1():
f_it = _hf.FailLengthHint(toT([1, 2, 3]))
it = replicate(f_it, 3)
with pytest.raises(_hf.FailLengthHint.EXC_TYP,
match=_hf.FailLengthHint.EXC_MSG):
operator.length_hint(it)
with pytest.raises(_hf.FailLengthHint.EXC_TYP,
match=_hf.FailLengthHint.EXC_MSG):
list(it)
def test_replicate_failure_lengthhint3():
# It is also possible that the length_hint overflows when the length is
# below maxsize but "times * length" is above maxsize.
# In this case length = maxsize / 2 but times = 3
of_it = _hf.OverflowLengthHint(toT([1, 2, 3]), sys.maxsize // 2)
it = replicate(of_it, 3)
with pytest.raises(OverflowError):
operator.length_hint(it)
with pytest.raises(OverflowError):
list(it)
def test_replicate_attributes1():
# Key+reverse function tests
it = replicate(toT(range(5)), 3)
assert it.times == 3
assert it.timescurrent == 0
with pytest.raises(AttributeError):
it.current
assert next(it) == T(0)
assert it.times == 3
assert it.timescurrent == 1
assert it.current == T(0)
def test_replicate_failure_lengthhint4():
# There is also the possibility that "length * times" does not overflow
# but adding the "times - timescurrent" afterwards will overflow.
# That's a bit tricky, but it seems that ((2**x-1) // 10) * 10 + 9 > 2**x-1
# is true for x=15, 31, 63 and 127 so it's possible by setting the times to
# 10 and the length to sys.maxsize // 10. The 9 are because the first item
# is already popped and should be replicated 9 more times.
of_it = _hf.OverflowLengthHint(toT([1, 2, 3]), sys.maxsize // 10)
it = replicate(of_it, 10)
next(it)
with pytest.raises(OverflowError):
operator.length_hint(it)
with pytest.raises(OverflowError):
list(it)
def test_replicate_failure_setstate2():
# setstate has an invalid second item in "state" < 0
mg = replicate(toT(range(5)), 3)
with pytest.raises(ValueError):
mg.__setstate__((None, -1))
def test_replicate_failure_setstate1():
# "state" is not a tuple
mg = replicate(toT(range(5)), 3)
with pytest.raises(TypeError):
mg.__setstate__([None, 0])
def test_replicate_failure5():
# iterator throws an exception different from StopIteration
with pytest.raises(_hf.FailNext.EXC_TYP, match=_hf.FailNext.EXC_MSG):
list(replicate(_hf.FailNext(), 2))
def test_replicate_failure4():
# second argument <= 1
with pytest.raises(ValueError):
replicate([T(1), T(2)], 1)
def test_replicate_failure2():
with pytest.raises(_hf.FailIter.EXC_TYP, match=_hf.FailIter.EXC_MSG):
replicate(_hf.FailIter(), 2)
def test_replicate_failure1():
# not enough arguments
with pytest.raises(TypeError):
replicate([T(1), T(2)])
def test_replicate_normal2():
# using a generator
assert list(replicate((i for i in toT([1, 2])), 2)) == toT([1, 1, 2, 2])
def test_replicate_normal1():
assert list(replicate([T(1), T(2)], 3)) == toT([1, 1, 1, 2, 2, 2])
def test_replicate_empty1():
assert list(replicate([], 3)) == []
def test_replicate_pickle2(protocol):
# normal
rpl = replicate([T(1), T(2)], 3)
assert next(rpl) == T(1)
x = pickle.dumps(rpl, protocol=protocol)
assert list(pickle.loads(x)) == toT([1, 1, 2, 2, 2])
def test_replicate_copy1():
_hf.iterator_copy(replicate([T(1), T(2)], 3))
def test_replicate_lengthhint1():
it = replicate([T(1), T(2)], 3)
_hf.check_lengthhint_iteration(it, 6)
