blob: cdb9e7e411e407af15b4a92b6ebfdabc79de3c07 [file] [log] [blame]
Guido van Rossum7dab2422002-04-26 19:40:56 +00001import unittest
Michael W. Hudson0edc7a02005-07-12 10:21:19 +00002import sys
Guido van Rossum7dab2422002-04-26 19:40:56 +00003
Benjamin Petersonee8712c2008-05-20 21:35:26 +00004from test import support
Guido van Rossum7dab2422002-04-26 19:40:56 +00005
Guido van Rossum7dab2422002-04-26 19:40:56 +00006class G:
7 'Sequence using __getitem__'
8 def __init__(self, seqn):
9 self.seqn = seqn
10 def __getitem__(self, i):
11 return self.seqn[i]
12
13class I:
14 'Sequence using iterator protocol'
15 def __init__(self, seqn):
16 self.seqn = seqn
17 self.i = 0
18 def __iter__(self):
19 return self
Georg Brandla18af4e2007-04-21 15:47:16 +000020 def __next__(self):
Guido van Rossum7dab2422002-04-26 19:40:56 +000021 if self.i >= len(self.seqn): raise StopIteration
22 v = self.seqn[self.i]
23 self.i += 1
24 return v
25
26class Ig:
27 'Sequence using iterator protocol defined with a generator'
28 def __init__(self, seqn):
29 self.seqn = seqn
30 self.i = 0
31 def __iter__(self):
32 for val in self.seqn:
33 yield val
34
35class X:
36 'Missing __getitem__ and __iter__'
37 def __init__(self, seqn):
38 self.seqn = seqn
39 self.i = 0
Georg Brandla18af4e2007-04-21 15:47:16 +000040 def __next__(self):
Guido van Rossum7dab2422002-04-26 19:40:56 +000041 if self.i >= len(self.seqn): raise StopIteration
42 v = self.seqn[self.i]
43 self.i += 1
44 return v
45
46class E:
47 'Test propagation of exceptions'
48 def __init__(self, seqn):
49 self.seqn = seqn
50 self.i = 0
51 def __iter__(self):
52 return self
Georg Brandla18af4e2007-04-21 15:47:16 +000053 def __next__(self):
Raymond Hettingerffdb8bb2004-09-27 15:29:05 +000054 3 // 0
Guido van Rossum7dab2422002-04-26 19:40:56 +000055
56class N:
Georg Brandla18af4e2007-04-21 15:47:16 +000057 'Iterator missing __next__()'
Guido van Rossum7dab2422002-04-26 19:40:56 +000058 def __init__(self, seqn):
59 self.seqn = seqn
60 self.i = 0
61 def __iter__(self):
62 return self
63
64class EnumerateTestCase(unittest.TestCase):
65
66 enum = enumerate
Raymond Hettingere8b0f042003-05-28 14:05:34 +000067 seq, res = 'abc', [(0,'a'), (1,'b'), (2,'c')]
Guido van Rossum7dab2422002-04-26 19:40:56 +000068
69 def test_basicfunction(self):
Raymond Hettingere8b0f042003-05-28 14:05:34 +000070 self.assertEqual(type(self.enum(self.seq)), self.enum)
71 e = self.enum(self.seq)
Guido van Rossum7dab2422002-04-26 19:40:56 +000072 self.assertEqual(iter(e), e)
Raymond Hettingere8b0f042003-05-28 14:05:34 +000073 self.assertEqual(list(self.enum(self.seq)), self.res)
Guido van Rossum7dab2422002-04-26 19:40:56 +000074 self.enum.__doc__
75
76 def test_getitemseqn(self):
Raymond Hettingere8b0f042003-05-28 14:05:34 +000077 self.assertEqual(list(self.enum(G(self.seq))), self.res)
Guido van Rossum7dab2422002-04-26 19:40:56 +000078 e = self.enum(G(''))
Georg Brandla18af4e2007-04-21 15:47:16 +000079 self.assertRaises(StopIteration, next, e)
Guido van Rossum7dab2422002-04-26 19:40:56 +000080
81 def test_iteratorseqn(self):
Raymond Hettingere8b0f042003-05-28 14:05:34 +000082 self.assertEqual(list(self.enum(I(self.seq))), self.res)
Guido van Rossum7dab2422002-04-26 19:40:56 +000083 e = self.enum(I(''))
Georg Brandla18af4e2007-04-21 15:47:16 +000084 self.assertRaises(StopIteration, next, e)
Guido van Rossum7dab2422002-04-26 19:40:56 +000085
86 def test_iteratorgenerator(self):
Raymond Hettingere8b0f042003-05-28 14:05:34 +000087 self.assertEqual(list(self.enum(Ig(self.seq))), self.res)
Guido van Rossum7dab2422002-04-26 19:40:56 +000088 e = self.enum(Ig(''))
Georg Brandla18af4e2007-04-21 15:47:16 +000089 self.assertRaises(StopIteration, next, e)
Guido van Rossum7dab2422002-04-26 19:40:56 +000090
91 def test_noniterable(self):
Raymond Hettingere8b0f042003-05-28 14:05:34 +000092 self.assertRaises(TypeError, self.enum, X(self.seq))
Guido van Rossum7dab2422002-04-26 19:40:56 +000093
94 def test_illformediterable(self):
Thomas Wouters1034dad2006-04-15 09:16:16 +000095 self.assertRaises(TypeError, self.enum, N(self.seq))
Guido van Rossum7dab2422002-04-26 19:40:56 +000096
97 def test_exception_propagation(self):
Raymond Hettingere8b0f042003-05-28 14:05:34 +000098 self.assertRaises(ZeroDivisionError, list, self.enum(E(self.seq)))
99
100 def test_argumentcheck(self):
101 self.assertRaises(TypeError, self.enum) # no arguments
102 self.assertRaises(TypeError, self.enum, 1) # wrong type (not iterable)
Alexandre Vassalottie9f305f2008-05-16 04:39:54 +0000103 self.assertRaises(TypeError, self.enum, 'abc', 'a') # wrong type
104 self.assertRaises(TypeError, self.enum, 'abc', 2, 3) # too many arguments
Raymond Hettingere8b0f042003-05-28 14:05:34 +0000105
106 def test_tuple_reuse(self):
107 # Tests an implementation detail where tuple is reused
108 # whenever nothing else holds a reference to it
Raymond Hettingera690a992003-11-16 16:17:49 +0000109 self.assertEqual(len(set(map(id, list(enumerate(self.seq))))), len(self.seq))
110 self.assertEqual(len(set(map(id, enumerate(self.seq)))), min(1,len(self.seq)))
Guido van Rossum7dab2422002-04-26 19:40:56 +0000111
112class MyEnum(enumerate):
113 pass
114
115class SubclassTestCase(EnumerateTestCase):
116
117 enum = MyEnum
118
Raymond Hettingere8b0f042003-05-28 14:05:34 +0000119class TestEmpty(EnumerateTestCase):
120
121 seq, res = '', []
122
123class TestBig(EnumerateTestCase):
124
125 seq = range(10,20000,2)
Guido van Rossum801f0d72006-08-24 19:48:10 +0000126 res = list(zip(range(20000), seq))
Raymond Hettingere8b0f042003-05-28 14:05:34 +0000127
Raymond Hettinger85c20a42003-11-06 14:06:48 +0000128class TestReversed(unittest.TestCase):
129
130 def test_simple(self):
131 class A:
132 def __getitem__(self, i):
133 if i < 5:
134 return str(i)
135 raise StopIteration
136 def __len__(self):
137 return 5
Guido van Rossum805365e2007-05-07 22:24:25 +0000138 for data in 'abc', range(5), tuple(enumerate('abc')), A(), range(1,17,5):
Raymond Hettinger85c20a42003-11-06 14:06:48 +0000139 self.assertEqual(list(data)[::-1], list(reversed(data)))
140 self.assertRaises(TypeError, reversed, {})
Alexandre Vassalottibee32532008-05-16 18:15:12 +0000141 # don't allow keyword arguments
142 self.assertRaises(TypeError, reversed, [], a=1)
Raymond Hettinger85c20a42003-11-06 14:06:48 +0000143
Guido van Rossum805365e2007-05-07 22:24:25 +0000144 def test_range_optimization(self):
145 x = range(1)
Raymond Hettinger85c20a42003-11-06 14:06:48 +0000146 self.assertEqual(type(reversed(x)), type(iter(x)))
Raymond Hettingere8b0f042003-05-28 14:05:34 +0000147
Raymond Hettinger029dba52004-02-10 09:33:39 +0000148 def test_len(self):
Raymond Hettingerd2c36262004-03-10 08:32:47 +0000149 # This is an implementation detail, not an interface requirement
Raymond Hettinger6b27cda2005-09-24 21:23:05 +0000150 from test.test_iterlen import len
Guido van Rossum805365e2007-05-07 22:24:25 +0000151 for s in ('hello', tuple('hello'), list('hello'), range(5)):
Raymond Hettingeref9bf402004-03-10 10:10:42 +0000152 self.assertEqual(len(reversed(s)), len(s))
Raymond Hettingerff5dc0e2004-09-29 11:40:50 +0000153 r = reversed(s)
154 list(r)
155 self.assertEqual(len(r), 0)
156 class SeqWithWeirdLen:
157 called = False
158 def __len__(self):
159 if not self.called:
160 self.called = True
161 return 10
162 raise ZeroDivisionError
163 def __getitem__(self, index):
164 return index
165 r = reversed(SeqWithWeirdLen())
166 self.assertRaises(ZeroDivisionError, len, r)
167
168
169 def test_gc(self):
170 class Seq:
171 def __len__(self):
172 return 10
173 def __getitem__(self, index):
174 return index
175 s = Seq()
176 r = reversed(s)
177 s.r = r
178
179 def test_args(self):
180 self.assertRaises(TypeError, reversed)
181 self.assertRaises(TypeError, reversed, [], 'extra')
Raymond Hettinger029dba52004-02-10 09:33:39 +0000182
Michael W. Hudson0edc7a02005-07-12 10:21:19 +0000183 def test_bug1229429(self):
184 # this bug was never in reversed, it was in
185 # PyObject_CallMethod, and reversed_new calls that sometimes.
186 if not hasattr(sys, "getrefcount"):
187 return
188 def f():
189 pass
190 r = f.__reversed__ = object()
191 rc = sys.getrefcount(r)
192 for i in range(10):
193 try:
194 reversed(f)
195 except TypeError:
196 pass
197 else:
198 self.fail("non-callable __reversed__ didn't raise!")
199 self.assertEqual(rc, sys.getrefcount(r))
Tim Petersf5f32b42005-07-17 23:16:17 +0000200
Michael W. Hudson0edc7a02005-07-12 10:21:19 +0000201
Alexandre Vassalottie9f305f2008-05-16 04:39:54 +0000202class TestStart(EnumerateTestCase):
203
204 enum = lambda i: enumerate(i, start=11)
205 seq, res = 'abc', [(1, 'a'), (2, 'b'), (3, 'c')]
206
207
208class TestLongStart(EnumerateTestCase):
209
210 enum = lambda i: enumerate(i, start=sys.maxsize+1)
211 seq, res = 'abc', [(sys.maxsize+1,'a'), (sys.maxsize+2,'b'),
212 (sys.maxsize+3,'c')]
213
214
Raymond Hettingere8b0f042003-05-28 14:05:34 +0000215def test_main(verbose=None):
Raymond Hettinger85c20a42003-11-06 14:06:48 +0000216 testclasses = (EnumerateTestCase, SubclassTestCase, TestEmpty, TestBig,
217 TestReversed)
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000218 support.run_unittest(*testclasses)
Raymond Hettingere8b0f042003-05-28 14:05:34 +0000219
220 # verify reference counting
221 import sys
222 if verbose and hasattr(sys, "gettotalrefcount"):
223 counts = [None] * 5
Guido van Rossum805365e2007-05-07 22:24:25 +0000224 for i in range(len(counts)):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000225 support.run_unittest(*testclasses)
Raymond Hettingere8b0f042003-05-28 14:05:34 +0000226 counts[i] = sys.gettotalrefcount()
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000227 print(counts)
Guido van Rossum7dab2422002-04-26 19:40:56 +0000228
229if __name__ == "__main__":
Raymond Hettingere8b0f042003-05-28 14:05:34 +0000230 test_main(verbose=True)