diff options
author | Raymond Hettinger <python@rcn.com> | 2014-06-24 21:39:27 -0700 |
---|---|---|
committer | Raymond Hettinger <python@rcn.com> | 2014-06-24 21:39:27 -0700 |
commit | ee205dc7d834ce33e245b46cecc9e15ea73aaefb (patch) | |
tree | c98f40eb7202c852c19f51a5fa7f169f597e6aa2 /Lib/test/test_itertools.py | |
parent | 7232986239fe0a5de3cfaf671d9e94af18ba8d8c (diff) | |
parent | 97d3555029b02fa4b432c1a1259b64aaa31e4e3e (diff) | |
download | cpython-git-ee205dc7d834ce33e245b46cecc9e15ea73aaefb.tar.gz |
merge
Diffstat (limited to 'Lib/test/test_itertools.py')
-rw-r--r-- | Lib/test/test_itertools.py | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/Lib/test/test_itertools.py b/Lib/test/test_itertools.py index 70517f0670..fea68163b3 100644 --- a/Lib/test/test_itertools.py +++ b/Lib/test/test_itertools.py @@ -967,6 +967,12 @@ class TestBasicOps(unittest.TestCase): self.assertEqual(take(2, copy.deepcopy(c)), list('a' * 2)) self.pickletest(repeat(object='a', times=10)) + def test_repeat_with_negative_times(self): + self.assertEqual(repr(repeat('a', -1)), "repeat('a', 0)") + self.assertEqual(repr(repeat('a', -2)), "repeat('a', 0)") + self.assertEqual(repr(repeat('a', times=-1)), "repeat('a', 0)") + self.assertEqual(repr(repeat('a', times=-2)), "repeat('a', 0)") + def test_map(self): self.assertEqual(list(map(operator.pow, range(3), range(1,7))), [0**1, 1**2, 2**3]) @@ -1741,8 +1747,15 @@ class LengthTransparency(unittest.TestCase): def test_repeat(self): self.assertEqual(operator.length_hint(repeat(None, 50)), 50) + self.assertEqual(operator.length_hint(repeat(None, 0)), 0) self.assertEqual(operator.length_hint(repeat(None), 12), 12) + def test_repeat_with_negative_times(self): + self.assertEqual(operator.length_hint(repeat(None, -1)), 0) + self.assertEqual(operator.length_hint(repeat(None, -2)), 0) + self.assertEqual(operator.length_hint(repeat(None, times=-1)), 0) + self.assertEqual(operator.length_hint(repeat(None, times=-2)), 0) + class RegressionTests(unittest.TestCase): def test_sf_793826(self): |