diff options
author | Raymond Hettinger <python@rcn.com> | 2007-10-04 00:20:27 (GMT) |
---|---|---|
committer | Raymond Hettinger <python@rcn.com> | 2007-10-04 00:20:27 (GMT) |
commit | 50e90e265fc3205fc6cc0193d3cebe08e9859b0a (patch) | |
tree | e32432fef1d15a607dfff20638fe36c0db463e6c /Lib/test/test_itertools.py | |
parent | 8f6693701c1899168f5bca96549ae2f2f590624e (diff) | |
download | cpython-50e90e265fc3205fc6cc0193d3cebe08e9859b0a.zip cpython-50e90e265fc3205fc6cc0193d3cebe08e9859b0a.tar.gz cpython-50e90e265fc3205fc6cc0193d3cebe08e9859b0a.tar.bz2 |
itertools.count() no longer limited to sys.maxint.
Diffstat (limited to 'Lib/test/test_itertools.py')
-rw-r--r-- | Lib/test/test_itertools.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/Lib/test/test_itertools.py b/Lib/test/test_itertools.py index 6c362ad..3a370d9 100644 --- a/Lib/test/test_itertools.py +++ b/Lib/test/test_itertools.py @@ -52,9 +52,12 @@ class TestBasicOps(unittest.TestCase): self.assertEqual(zip('abc',count()), [('a', 0), ('b', 1), ('c', 2)]) self.assertEqual(zip('abc',count(3)), [('a', 3), ('b', 4), ('c', 5)]) self.assertEqual(take(2, zip('abc',count(3))), [('a', 3), ('b', 4)]) + self.assertEqual(take(2, zip('abc',count(-1))), [('a', -1), ('b', 0)]) + self.assertEqual(take(2, zip('abc',count(-3))), [('a', -3), ('b', -2)]) self.assertRaises(TypeError, count, 2, 3) self.assertRaises(TypeError, count, 'a') - self.assertRaises(OverflowError, list, islice(count(maxsize-5), 10)) + self.assertEqual(list(islice(count(maxsize-5), 10)), range(maxsize-5, maxsize+5)) + self.assertEqual(list(islice(count(-maxsize-5), 10)), range(-maxsize-5, -maxsize+5)) c = count(3) self.assertEqual(repr(c), 'count(3)') c.next() @@ -63,6 +66,8 @@ class TestBasicOps(unittest.TestCase): self.assertEqual(repr(c), 'count(-9)') c.next() self.assertEqual(c.next(), -8) + for i in (-sys.maxint-5, -sys.maxint+5 ,-10, -1, 0, 10, sys.maxint-5, sys.maxint+5): + self.assertEqual(repr(count(i)), 'count(%r)' % i) def test_cycle(self): self.assertEqual(take(10, cycle('abc')), list('abcabcabca')) |