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]