From 47c4ef54bcd2f51dc2bf0bc345f111889d4418e1 Mon Sep 17 00:00:00 2001 From: Paul Sokolovsky Date: Tue, 25 Aug 2015 21:18:25 +0300 Subject: [PATCH] itertools: test_itertools.py: Test for islice() on generator. --- itertools/test_itertools.py | 6 ++++++ 1 file changed, 6 insertions(+) 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]