From 69c726edd24cd52125a1136ce88fe9ebb72245b5 Mon Sep 17 00:00:00 2001 From: Jeong YunWon Date: Wed, 15 Jun 2022 00:27:51 +0900 Subject: [PATCH] Add infinite test for itertools.chain.from_iterable This is safe, but not tested --- Lib/test/test_itertools.py | 1 + 1 file changed, 1 insertion(+) diff --git a/Lib/test/test_itertools.py b/Lib/test/test_itertools.py index 238afbbd883d3a..f469bfe185e65b 100644 --- a/Lib/test/test_itertools.py +++ b/Lib/test/test_itertools.py @@ -180,6 +180,7 @@ def test_chain_from_iterable(self): self.assertEqual(list(chain.from_iterable([''])), []) self.assertEqual(take(4, chain.from_iterable(['abc', 'def'])), list('abcd')) self.assertRaises(TypeError, list, chain.from_iterable([2, 3])) + self.assertEqual(list(islice(chain.from_iterable(repeat(range(5))), 2)), [0, 1]) def test_chain_reducible(self): for oper in [copy.deepcopy] + picklecopiers: