micropython-lib/itertools/test_itertools.py

12 wiersze
331 B
Python

import itertools
assert list(itertools.islice(list(range(10)), 4)) == [0, 1, 2, 3]
assert list(itertools.islice(list(range(10)), 2, 6)) == [2, 3, 4, 5]
assert list(itertools.islice(list(range(10)), 2, 6, 2)) == [2, 4]
def g():
while True:
yield 123
assert list(itertools.islice(g(), 5)) == [123, 123, 123, 123, 123]