diff --git a/itertools/test_itertools.py b/itertools/test_itertools.py index 51a29fd4..eee1caed 100644 --- a/itertools/test_itertools.py +++ b/itertools/test_itertools.py @@ -3,3 +3,9 @@ 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]