import itertools from itertools import * import math def lzip(*args): return list(zip(*args)) def onearg(x): 'Test function of one argument' return 2*x def errfunc(*args): 'Test function that raises an error' raise ValueError() def gen3(): 'Non-restartable source sequence' for i in (0, 1, 2): yield i def isEven(x): 'Test predicate' return x%2==0 def isOdd(x): 'Test predicate' return x%2==1 def tupleize(*args): return args def irange(n): for i in range(n): yield i def take(n, seq): 'Convenience function for partially consuming a long of infinite iterable' return list(islice(seq, n)) def testR(r): return r[0] def testR2(r): return r[2] def underten(x): return x<10 @test def test_combinations(): assert list(itertools.combinations('ABCD', 2)) == [['A','B'], ['A','C'], ['A','D'], ['B','C'], ['B','D'], ['C','D']] test_intermediate = itertools.combinations('ABCD', 2) next(test_intermediate) assert list(test_intermediate) == [['A','C'], ['A','D'], ['B','C'], ['B','D'], ['C','D']] assert list(itertools.combinations(range(4), 3)) == [[0, 1, 2], [0, 1, 3], [0, 2, 3], [1, 2, 3]] test_intermediat = itertools.combinations(range(4), 3) next(test_intermediat) assert list(test_intermediat) == [[0, 1, 3], [0, 2, 3], [1, 2, 3]] @test def test_combinations_with_replacement(): assert list(itertools.combinations_with_replacement(range(3), 3)) == [[0, 0, 0], [0, 0, 1], [0, 0, 2], [0, 1, 1], [0, 1, 2], [0, 2, 2], [1, 1, 1], [1, 1, 2], [1, 2, 2], [2, 2, 2]] assert list(itertools.combinations_with_replacement('ABC', 2)) == [['A', 'A'], ['A', 'B'], ['A', 'C'], ['B', 'B'], ['B', 'C'], ['C', 'C']] test_intermediate = itertools.combinations_with_replacement('ABC', 2) next(test_intermediate) assert list(test_intermediate) == [['A', 'B'], ['A', 'C'], ['B', 'B'], ['B', 'C'], ['C', 'C']] @test def test_islice(): ra100 = range(100) ra = range(10) assert list(itertools.islice(iter('ABCDEFG'), 0, 2, 1)) == ['A', 'B'] assert list(itertools.islice(iter(ra100), 10, 20, 3)) == [10, 13, 16, 19] assert list(itertools.islice(iter(ra100), 10, 3, 20)) == [] assert list(itertools.islice(iter(ra100), 10, 20, 1)) == [10, 11, 12, 13, 14, 15, 16, 17, 18, 19] assert list(itertools.islice(iter(ra100), 10, 10, 1)) == [] assert list(itertools.islice(iter(ra100), 10, 3, 1)) == [] # stop=len(iterable) assert list(itertools.islice(iter(ra), 0, 10, 1)) == [0, 1, 2, 3, 4, 5, 6, 7, 8, 9] assert list(itertools.islice(iter(ra), 2, 10, 1)) == [2, 3, 4, 5, 6, 7, 8, 9] assert list(itertools.islice(iter(ra), 1, 10, 2)) == [1, 3, 5, 7, 9] try: list(itertools.islice(iter(ra), -5, 10, 1)) assert False except ValueError: pass @test def test_count(): # infinite loop here assert take(3, itertools.count(3.25, 1.0)) == [3.25, 4.25, 5.25] assert take(3, zip('abc', itertools.count(3.25, 1.0))) == [('a', 3.25), ('b', 4.25), ('c', 5.25)] assert take(2, zip('abc', itertools.count(-1.0, 1.0))) == [('a', -1.0), ('b', 0.0)] assert take(2, zip('abc', itertools.count(-3.0, 1.0))) == [('a', -3.0), ('b', -2.0)] @test def test_repeat(): assert list(itertools.repeat('a', 3)) == ['a', 'a', 'a'] assert list(itertools.repeat(1, 10)) == [1, 1, 1, 1, 1, 1, 1, 1, 1, 1] assert list(itertools.repeat('a', -1)) == [] assert len(list(itertools.repeat('a', 0))) == 0 @test def test_cycle(): assert take(10, zip('zzzzzzzzzzz', itertools.cycle(iter('abc')))) == [('z', 'a'), ('z', 'b'), ('z', 'c'), ('z', 'a'), ('z', 'b'), ('z', 'c'), ('z', 'a'), ('z', 'b'), ('z', 'c'), ('z', 'a')] assert take(10, zip('zzzzzzzzzzz', itertools.cycle(['a', 'b']))) == [('z', 'a'), ('z', 'b'), ('z', 'a'), ('z', 'b'), ('z', 'a'), ('z', 'b'), ('z', 'a'), ('z', 'b'), ('z', 'a'), ('z', 'b')] @test def test_compress(): assert list(itertools.compress('ABCDEF', [1, 0, 1, 0, 1, 1])) == ['A', 'C', 'E', 'F'] assert list(itertools.compress('ABCDEF', [1, 1, 1, 1, 1, 1])) == ['A', 'B', 'C', 'D', 'E', 'F'] assert list(itertools.compress('ABCDEF', [1, 0, 1])) == ['A', 'C'] assert list(itertools.compress('ABC', [0, 1, 1, 1, 1, 1])) == ['B', 'C'] @test def test_dropwhile(): data = [1, 3, 5, 20, 2, 4, 6, 8] assert list(itertools.dropwhile(underten, data)) == [20, 2, 4, 6, 8] @test def test_takewhile(): data = [1, 3, 5, 20, 2, 4, 6, 8] assert list(itertools.takewhile(underten, data)) == [1, 3, 5] @test def test_filterfalse(): assert list(itertools.filterfalse(isEven, range(10))) == [1, 3, 5, 7, 9] assert list(itertools.filterfalse(lambda x: bool(x), [0, 1, 0, 2, 0])) == [0, 0, 0] @test def test_permutations(): assert list(itertools.permutations(range(3), 2)) == [[0, 1], [0,2], [1,0], [1,2], [2,0], [2,1]] for n in range(3): values = [5*x-12 for x in range(n)] for r in range(n+2): result = list(itertools.permutations(values, r)) if r > n: # right number of perms assert len(result) == 0 # factorial is not yet implemented in math # else: fact(n) / fact(n - r) @test def test_accumulate(): # addition assert list(itertools.accumulate(range(10), int.__add__, initial=0)) == [0, 0, 1, 3, 6, 10, 15, 21, 28, 36, 45] assert list(itertools.accumulate([7], int.__add__, initial=0)) == [0, 7] # iterable of length 1 assert list(itertools.accumulate(range(10), int.__add__)) == [0, 1, 3, 6, 10, 15, 21, 28, 36, 45] assert list(itertools.accumulate([7], int.__add__)) == [7] # iterable of length 1 assert list(itertools.accumulate('abc', str.__add__, initial='')) == ['', 'a', 'ab', 'abc'] assert list(itertools.accumulate([''], str.__add__, initial=str(0))) == ['0', '0'] # multiply assert list(itertools.accumulate(range(10), int.__mul__, initial=0)) == [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0] assert list(itertools.accumulate([1, 2, 3, 4, 5], int.__mul__, initial=1)) == [1, 1, 2, 6, 24, 120] assert list(itertools.accumulate([7], int.__mul__)) == [7] # pass @test def test_chain(): assert list(itertools.chain('abc', 'def')) == ['a', 'b', 'c', 'd', 'e', 'f'] assert list(itertools.chain('abc')) == ['a', 'b', 'c'] assert list(itertools.chain('a', 'b', 'c')) == ['a', 'b', 'c'] assert list(itertools.chain(['abc', 'def'])) == ['abc', 'def'] assert list(itertools.chain(['abc'])) == ['abc'] assert list(itertools.chain(['a', 'b', 'c'])) == ['a', 'b', 'c'] @test def test_starmap(): assert list(itertools.starmap(math.pow, [(2.0, 5.0), (3.0, 2.0), (10.0, 3.0)])) == [32.0, 9.0, 1000.0] assert list(itertools.starmap(math.pow, [(0.0, 1.0), (1.0, 2.0), (2.0, 3.0)])) == [0.0**1.0, 1.0**2.0, 2.0**3.0] @test def test_groupby(): def key_str(s: str): return s assert list(itertools.groupby(iter('AAAABBBCCDAABBC'), key_str)) == [('A', ['A', 'A', 'A', 'A']), ('B', ['B', 'B', 'B']), ('C', ['C', 'C']), ('D', ['D']), ('A', ['A', 'A']), ('B', ['B', 'B']), ('C', ['C'])] @test def test_zip_longest(): assert list(itertools.zip_longest('ABCDE', '12345', fillvalue='-')) == [('A', '1'), ('B', '2'), ('C', '3'), ('D', '4'), ('E', '5')] assert list(itertools.zip_longest('ABCDE', '123', fillvalue='-')) == [('A', '1'), ('B', '2'), ('C', '3'), ('D', '-'), ('E', '-')] assert list(itertools.zip_longest('123', 'ABCDE', fillvalue='-')) == [('1', 'A'), ('2', 'B'), ('3', 'C'), ('-', 'D'), ('-', 'E')] assert list(itertools.zip_longest('', 'ABCDE', fillvalue='-')) == [('-', 'A'), ('-', 'B'), ('-', 'C'), ('-', 'D'), ('-', 'E')] assert list(itertools.zip_longest('ABCDE', '', fillvalue='-')) == [('A', '-'), ('B', '-'), ('C', '-'), ('D', '-'), ('E', '-')] assert not list(itertools.zip_longest('', '', fillvalue='-')) @test def test_zip_test(): assert list(zip()) == [] assert list(zip((1, 2))) == [(1,), (2,)] assert list(zip([1, 2], ['a', 'b'], (False, True))) == [(1, 'a', False), (2, 'b', True)] test_combinations() test_combinations_with_replacement() test_islice() test_count() test_repeat() test_cycle() test_compress() test_dropwhile() test_takewhile() test_filterfalse() test_permutations() test_accumulate() test_chain() test_starmap() test_groupby() test_zip_longest() test_zip_test() # Updated tests lifted from CPython test suite @test def test_accumulate_from_cpython(): assert list(accumulate(range(10))) == [0, 1, 3, 6, 10, 15, 21, 28, 36, 45] assert list(accumulate(iterable=range(10))) == [0, 1, 3, 6, 10, 15, 21, 28, 36, 45] assert list(accumulate('abc')) == ['a', 'ab', 'abc'] assert list(accumulate(List[float]())) == [] assert list(accumulate([7])) == [7] s = [2, 8, 9, 5, 7, 0, 3, 4, 1, 6] assert list(accumulate(s, min)) == [2, 2, 2, 2, 2, 0, 0, 0, 0, 0] assert list(accumulate(s, max)) == [2, 8, 9, 9, 9, 9, 9, 9, 9, 9] mul = lambda a, b: a * b assert list(accumulate(s, mul)) == [2, 16, 144, 720, 5040, 0, 0, 0, 0, 0] # assert list(accumulate([10, 5, 1], initial=None)) == [10, 15, 16] assert list(accumulate([10, 5, 1], initial=100)) == [100, 110, 115, 116] assert list(accumulate([10, 5, 1], initial=100.5)) == [100.5, 110.5, 115.5, 116.5] assert list(accumulate(List[int](), initial=100)) == [100] test_accumulate_from_cpython() @test def test_chain_from_cpython(): assert list(chain('abc', 'def')) == list('abcdef') assert list(chain('abc')) == list('abc') assert list(chain('')) == [] assert list(take(4, chain('abc', 'def'))) == list('abcd') test_chain_from_cpython() @test def test_chain_from_iterable_from_cpython(): assert list(chain.from_iterable(['abc', 'def'])) == list('abcdef') assert list(chain.from_iterable(['abc'])) == list('abc') assert list(chain.from_iterable([''])) == [] assert take(4, chain.from_iterable(['abc', 'def'])) == list('abcd') test_chain_from_iterable_from_cpython() @test def test_combinations_from_cpython(): from math import factorial as fact err = False try: list(combinations('abc', -2)) assert False except ValueError: err = True assert err assert list(combinations('abc', 32)) == [] # r > n assert list(combinations('ABCD', 2)) == [['A','B'], ['A','C'], ['A','D'], ['B','C'], ['B','D'], ['C','D']] assert list(combinations(range(4), 3)) == [[0,1,2], [0,1,3], [0,2,3], [1,2,3]] for n in range(7): values = [5*x-12 for x in range(n)] for r in range(n+2): result = list(combinations(values, r)) assert len(result) == (0 if r>n else fact(n) // fact(r) // fact(n-r)) assert len(result) == len(set(result)) # no repeats #assert result == sorted(result) # lexicographic order for c in result: assert len(c) == r # r-length combinations assert len(set(c)) == r # no duplicate elements assert list(c) == sorted(c) # keep original ordering assert all(e in values for e in c) # elements taken from input iterable assert list(c) == [e for e in values if e in c] # comb is a subsequence of the input iterable test_combinations_from_cpython() @test def test_combinations_with_replacement_from_cpython(): cwr = combinations_with_replacement err = False try: list(cwr('abc', -2)) assert False except ValueError: err = True assert err assert list(cwr('ABC', 2)) == [['A','A'], ['A','B'], ['A','C'], ['B','B'], ['B','C'], ['C','C']] def numcombs(n, r): from math import factorial as fact if not n: return 0 if r else 1 return fact(n+r-1) // fact(r) // fact(n-1) for n in range(7): values = [5*x-12 for x in range(n)] for r in range(n+2): result = list(cwr(values, r)) regular_combs = list(combinations(values, r)) assert len(result) == numcombs(n, r) assert len(result) == len(set(result)) # no repeats #assert result == sorted(result) # lexicographic order if n == 0 or r <= 1: assert result == regular_combs # cases that should be identical else: assert set(result) >= set(regular_combs) for c in result: assert len(c) == r # r-length combinations noruns = [k for k,v in groupby(c)] # combo without consecutive repeats assert len(noruns) == len(set(noruns)) # no repeats other than consecutive assert list(c) == sorted(c) # keep original ordering assert all(e in values for e in c) # elements taken from input iterable assert noruns == [e for e in values if e in c] # comb is a subsequence of the input iterable test_combinations_with_replacement_from_cpython() @test def test_permutations_from_cpython(): from math import factorial as fact err = False try: list(permutations('abc', -2)) assert False except ValueError: err = True assert err assert list(permutations('abc', 32)) == [] assert list(permutations(range(3), 2)) == [[0,1], [0,2], [1,0], [1,2], [2,0], [2,1]] for n in range(7): values = [5*x-12 for x in range(n)] for r in range(n+2): result = list(permutations(values, r)) assert len(result) == (0 if r>n else fact(n) // fact(n-r)) # right number of perms assert len(result) == len(set(result)) # no repeats #assert result == sorted(result) # lexicographic order for p in result: assert len(p) == r # r-length permutations assert len(set(p)) == r # no duplicate elements assert all(e in values for e in p) # elements taken from input iterable if r == n: assert result == list(permutations(values, None)) # test r as None assert result == list(permutations(values)) # test default r test_permutations_from_cpython() @extend class List: def __lt__(self, other: List[T]): if len(self) != len(other): return len(self) < len(other) for a,b in zip(self, other): if a < b: return True if a > b: return False return False def __le__(self, other: List[T]): if len(self) != len(other): return len(self) < len(other) for a,b in zip(self, other): if a < b: return True if a > b: return False return True def __gt__(self, other: List[T]): if len(self) != len(other): return len(self) < len(other) for a,b in zip(self, other): if a > b: return True if a < b: return False return False def __ge__(self, other: List[T]): if len(self) != len(other): return len(self) < len(other) for a,b in zip(self, other): if a > b: return True if a < b: return False return True @test def test_combinatorics_from_cpython(): # Test relationships between product(), permutations(), # combinations() and combinations_with_replacement(). from math import factorial as fact for n in range(6): s = 'ABCDEFG'[:n] for r in range(8): prod = list(product(s, repeat=r)) cwr = list(combinations_with_replacement(s, r)) perm = list(permutations(s, r)) comb = list(combinations(s, r)) # Check size assert len(prod) == n**r assert len(cwr) == ((fact(n+r-1) // fact(r)// fact(n-1)) if n else (0 if r else 1)) assert len(perm) == (0 if r>n else fact(n) // fact(n-r)) assert len(comb) == (0 if r>n else fact(n) // fact(r) // fact(n-r)) # Check lexicographic order without repeated tuples assert prod == sorted(set(prod)) assert cwr == sorted(set(cwr)) assert perm == sorted(set(perm)) assert comb == sorted(set(comb)) # Check interrelationships assert cwr == [t for t in prod if sorted(t)==list(t)] # cwr: prods which are sorted assert perm == [t for t in prod if len(set(t))==r] # perm: prods with no dups assert comb == [t for t in perm if sorted(t)==list(t)] # comb: perms that are sorted assert comb == [t for t in cwr if len(set(t))==r] # comb: cwrs without dups assert comb == list(filter(set(cwr).__contains__, perm)) # comb: perm that is a cwr assert comb == list(filter(set(perm).__contains__, cwr)) # comb: cwr that is a perm assert comb == sorted(set(cwr) & set(perm)) # comb: both a cwr and a perm test_combinatorics_from_cpython() @test def test_compress_from_cpython(): assert list(compress(data='ABCDEF', selectors=[1,0,1,0,1,1])) == list('ACEF') assert list(compress('ABCDEF', [1,0,1,0,1,1])) == list('ACEF') assert list(compress('ABCDEF', [0,0,0,0,0,0])) == list('') assert list(compress('ABCDEF', [1,1,1,1,1,1])) == list('ABCDEF') assert list(compress('ABCDEF', [1,0,1])) == list('AC') assert list(compress('ABC', [0,1,1,1,1,1])) == list('BC') n = 10000 data = chain.from_iterable(repeat(range(6), n)) selectors = chain.from_iterable(repeat((0, 1))) assert list(compress(data, selectors)) == [1,3,5] * n test_compress_from_cpython() @test def test_count_from_cpython(): assert lzip('abc',count()) == [('a', 0), ('b', 1), ('c', 2)] assert lzip('abc',count(3)) == [('a', 3), ('b', 4), ('c', 5)] assert take(2, lzip('abc',count(3))) == [('a', 3), ('b', 4)] assert take(2, zip('abc',count(-1))) == [('a', -1), ('b', 0)] assert take(2, zip('abc',count(-3))) == [('a', -3), ('b', -2)] assert take(3, count(3.25)) == [3.25, 4.25, 5.25] test_count_from_cpython() @test def test_count_with_stride_from_cpython(): assert lzip('abc',count(2,3)) == [('a', 2), ('b', 5), ('c', 8)] assert lzip('abc',count(start=2,step=3)) == [('a', 2), ('b', 5), ('c', 8)] assert lzip('abc',count(step=-1)) == [('a', 0), ('b', -1), ('c', -2)] assert lzip('abc',count(2,0)) == [('a', 2), ('b', 2), ('c', 2)] assert lzip('abc',count(2,1)) == [('a', 2), ('b', 3), ('c', 4)] assert lzip('abc',count(2,3)) == [('a', 2), ('b', 5), ('c', 8)] assert take(3, count(2., 1.25)) == [2., 3.25, 4.5] test_count_with_stride_from_cpython() @test def test_cycle_from_cpython(): assert take(10, cycle('abc')) == list('abcabcabca') assert list(cycle('')) == [] assert list(islice(cycle(gen3()),10)) == [0,1,2,0,1,2,0,1,2,0] test_cycle_from_cpython() @test def test_groupby_from_cpython(): # Check whether it accepts arguments correctly assert [] == list(groupby(List[int]())) assert [] == list(groupby(List[int](), key=lambda a: a)) # Check normal input if 1: s = [(0, 10, 20), (0, 11,21), (0,12,21), (1,13,21), (1,14,22), (2,15,22), (3,16,23), (3,17,23)] if 1: dup = [] for k, g in groupby(s, lambda r:r[0]): for elem in g: assert k == elem[0] dup.append(elem) assert s == dup # Check nested case if 1: dup = [] for k, g in groupby(s, testR): for ik, ig in groupby(g, testR2): for elem in ig: assert k == elem[0] assert ik == elem[2] dup.append(elem) assert s == dup # Check case where inner iterator is not used keys = [k for k, g in groupby(s, testR)] expectedkeys = set([r[0] for r in s]) assert set(keys) == expectedkeys assert len(keys) == len(expectedkeys) if 1: # Exercise pipes and filters style s = 'abracadabra' if 1: # sort s | uniq r = [k for k, g in groupby(sorted(s))] assert r == ['a', 'b', 'c', 'd', 'r'] if 1: # sort s | uniq -d r = [k for k, g in groupby(sorted(s)) if list(islice(g,1,2))] assert r == ['a', 'b', 'r'] if 1: # sort s | uniq -c r = [(len(list(g)), k) for k, g in groupby(sorted(s))] assert r == [(5, 'a'), (2, 'b'), (1, 'c'), (1, 'd'), (2, 'r')] if 1: # sort s | uniq -c | sort -rn | head -3 r = sorted([(len(list(g)) , k) for k, g in groupby(sorted(s))], reverse=True)[:3] assert r == [(5, 'a'), (2, 'r'), (2, 'b')] test_groupby_from_cpython() @test def test_filter_from_cpython(): assert list(filter(isEven, range(6))) == [0,2,4] # assert list(filter(None, [0,1,0,2,0])) == [1,2] # TODO assert list(filter(lambda x: bool(x), [0,1,0,2,0])) == [1,2] assert take(4, filter(isEven, count())) == [0,2,4,6] test_filter_from_cpython() @test def test_filterfalse_from_cpython(): assert list(filterfalse(isEven, range(6))) == [1,3,5] # assert list(filter(None, [0,1,0,2,0])) == [0,0,0] # TODO assert list(filterfalse(lambda x: bool(x), [0,1,0,2,0])) == [0,0,0] assert take(4, filterfalse(isEven, count())) == [1,3,5,7] test_filterfalse_from_cpython() @test def test_zip_from_cpython(): ans = [(x,y) for x, y in zip('abc',count())] assert ans == [('a', 0), ('b', 1), ('c', 2)] assert list(zip('abc', range(6))) == lzip('abc', range(6)) assert list(zip('abcdef', range(3))) == lzip('abcdef', range(3)) assert take(3,zip('abcdef', count())) == lzip('abcdef', range(3)) assert list(zip('abcdef')) == lzip('abcdef') assert list(zip()) == lzip() assert [pair for pair in zip('abc', 'def')] == lzip('abc', 'def') test_zip_from_cpython() @test def test_ziplongest_from_cpython(): for args in ( (range(1000), range(2000,2100), range(3000,3050)), (range(1000), range(0), range(3000,3050), range(1200), range(1500)), (range(1000), range(0), range(3000,3050), range(1200), range(1500), range(0)), ): target = [tuple(arg[i] if i < len(arg) else None for arg in args) for i in range(max(map(len, args)))] assert str(list(zip_longest(*args))) == str(target) target2 = [[(-999 if e is None else ~e) for e in t] for t in target] # Replace None fills with 'X' assert list(zip_longest(*args, fillvalue=-999)) == target2 assert str(list(zip_longest('abc', range(6)))) == "[('a', 0), ('b', 1), ('c', 2), (None, 3), (None, 4), (None, 5)]" assert str(list(zip_longest(range(6), 'abc'))) == "[(0, 'a'), (1, 'b'), (2, 'c'), (3, None), (4, None), (5, None)]" test_ziplongest_from_cpython() @test def test_product_from_cpython(): for args, result in ( #((), [()]), # zero iterables # TODO (('ab',), [('a',), ('b',)]), # one iterable ((range(2), range(3)), [(0,0), (0,1), (0,2), (1,0), (1,1), (1,2)]), # two iterables ((range(0), range(2), range(3)), List[Tuple[int,int,int]]()), # first iterable with zero length ((range(2), range(0), range(3)), List[Tuple[int,int,int]]()), # middle iterable with zero length ((range(2), range(3), range(0)), List[Tuple[int,int,int]]()), # last iterable with zero length ): assert list(product(*args)) == result assert len(list(product(range(7), range(7), range(7), range(7), range(7), range(7)))) == 7**6 test_product_from_cpython() @test def test_repeat_from_cpython(): assert list(repeat(object='a', times=3)) == ['a', 'a', 'a'] assert lzip(range(3),repeat('a')) == [(0, 'a'), (1, 'a'), (2, 'a')] assert list(repeat('a', 3)) == ['a', 'a', 'a'] assert take(3, repeat('a')) == ['a', 'a', 'a'] assert list(repeat('a', 0)) == [] assert list(repeat('a', -3)) == [] test_repeat_from_cpython() @test def test_map_from_cpython(): power = lambda a,b: a ** b assert list(map(power, range(3), range(1,7))) == [0**1, 1**2, 2**3] assert list(map(tupleize, 'abc', range(5))) == [('a',0),('b',1),('c',2)] assert list(map(tupleize, 'abc', count())) == [('a',0),('b',1),('c',2)] assert take(2,map(tupleize, 'abc', count())) == [('a',0),('b',1)] assert list(map(tupleize, List[int]())) == [] test_map_from_cpython() @test def test_starmap_from_cpython(): power = lambda a,b: a ** b assert list(starmap(power, zip(range(3), range(1,7)))) == [0**1, 1**2, 2**3] assert take(3, starmap(power, zip(count(), count(1)))) == [0**1, 1**2, 2**3] #assert list(starmap(tupleize, List[int]())) == [] # TODO assert list(starmap(power, [(4,5)])) == [4**5] test_starmap_from_cpython() @test def test_islice_from_cpython(): for args in ( # islice(args) should agree with range(args) (10, 20, 3), (10, 3, 20), (10, 20), (10, 10), (10, 3), (20,) ): assert list(islice(range(100), *args)) == list(range(*args)) for args, tgtargs in ( # Stop when seqn is exhausted ((10, 110, 3), ((10, 100, 3))), ((10, 110), ((10, 100))), ((110,), (100,)) ): assert list(islice(range(100), *args)) == list(range(*tgtargs)) # Test stop=None assert list(islice(range(10), None)) == list(range(10)) assert list(islice(range(10), None, None)) == list(range(10)) assert list(islice(range(10), None, None, None)) == list(range(10)) assert list(islice(range(10), 2, None)) == list(range(2, 10)) assert list(islice(range(10), 1, None, 2)) == list(range(1, 10, 2)) test_islice_from_cpython() @test def test_takewhile_from_cpython(): data = [1, 3, 5, 20, 2, 4, 6, 8] assert list(takewhile(underten, data)) == [1, 3, 5] assert list(takewhile(underten, List[int]())) == [] t = takewhile(lambda x: bool(x), [1, 1, 1, 0, 0, 0]) assert list(t) == [1, 1, 1] test_takewhile_from_cpython() @test def test_dropwhile_from_cpython(): data = [1, 3, 5, 20, 2, 4, 6, 8] assert list(dropwhile(underten, data)) == [20, 2, 4, 6, 8] assert list(dropwhile(underten, List[int]())) == [] test_dropwhile_from_cpython() @test def test_tee_from_cpython(): import random n = 200 a, b = tee(List[int]()) # test empty iterator assert list(a) == [] assert list(b) == [] a, b = tee(irange(n)) # test 100% interleaved assert lzip(a,b) == lzip(range(n), range(n)) a, b = tee(irange(n)) # test 0% interleaved assert list(a) == list(range(n)) assert list(b) == list(range(n)) a, b = tee(irange(n)) # test dealloc of leading iterator for i in range(100): assert next(a) == i assert list(b) == list(range(n)) a, b = tee(irange(n)) # test dealloc of trailing iterator for i in range(100): assert next(a) == i assert list(a) == list(range(100, n)) for j in range(5): # test randomly interleaved order = [0]*n + [1]*n random.shuffle(order) lists = ([], []) its = tee(irange(n)) for i in order: value = next(its[i]) lists[i].append(value) assert lists[0] == list(range(n)) assert lists[1] == list(range(n)) # test long-lagged and multi-way split a, b, c = tee(range(2000), 3) for i in range(100): assert next(a) == i assert list(b) == list(range(2000)) assert [next(c), next(c)] == list(range(2)) assert list(a) == list(range(100,2000)) assert list(c) == list(range(2,2000)) test_tee_from_cpython()