def test_skiperator_skip_negative(self):
# negative value of skip is the same as skip=0
g = _skiperator([1, 2, 3], skip=-1)
result = []
for i in range(10):
result.append(next(g))
assert result == [1, 2, 3, 1, 2, 3, 1, 2, 3, 1]
def test_skiperator_empty_skip(self):
g = _skiperator([], skip=2)
with pytest.raises(StopIteration):
next(g)
def test_skiperator_basic(self):
g = _skiperator([1, 2, 3])
result = []
for i in range(10):
result.append(next(g))
assert result == [1, 2, 3, 1, 2, 3, 1, 2, 3, 1]
def test_skiperator_short_skip(self):
g = _skiperator([1, 2, 3], skip=2)
result = []
for i in range(2):
result.append(next(g))
assert result == [3, 1]
def test_skiperator_one_value(self):
g = _skiperator([1])
result = []
for i in range(10):
result.append(next(g))
assert result == [1, 1, 1, 1, 1, 1, 1, 1, 1, 1]
def test_skiperator_skip_more(self):
g = _skiperator([1, 2, 3], skip=4)
result = []
for i in range(10):
result.append(next(g))
assert result == [2, 3, 1, 2, 3, 1, 2, 3, 1, 2]