itertools.count() no longer limited to sys.maxint.
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 @@
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 @@
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'))