Python code coverage for Lib/test/test_enumerate.py

#	count	content
1	n/a	import unittest
2	n/a	import operator
3	n/a	import sys
4	n/a	import pickle
5	n/a
6	n/a	from test import support
7	n/a
8	n/a	class G:
9	n/a	'Sequence using __getitem__'
10	n/a	def __init__(self, seqn):
11	n/a	self.seqn = seqn
12	n/a	def __getitem__(self, i):
13	n/a	return self.seqn[i]
14	n/a
15	n/a	class I:
16	n/a	'Sequence using iterator protocol'
17	n/a	def __init__(self, seqn):
18	n/a	self.seqn = seqn
19	n/a	self.i = 0
20	n/a	def __iter__(self):
21	n/a	return self
22	n/a	def __next__(self):
23	n/a	if self.i >= len(self.seqn): raise StopIteration
24	n/a	v = self.seqn[self.i]
25	n/a	self.i += 1
26	n/a	return v
27	n/a
28	n/a	class Ig:
29	n/a	'Sequence using iterator protocol defined with a generator'
30	n/a	def __init__(self, seqn):
31	n/a	self.seqn = seqn
32	n/a	self.i = 0
33	n/a	def __iter__(self):
34	n/a	for val in self.seqn:
35	n/a	yield val
36	n/a
37	n/a	class X:
38	n/a	'Missing __getitem__ and __iter__'
39	n/a	def __init__(self, seqn):
40	n/a	self.seqn = seqn
41	n/a	self.i = 0
42	n/a	def __next__(self):
43	n/a	if self.i >= len(self.seqn): raise StopIteration
44	n/a	v = self.seqn[self.i]
45	n/a	self.i += 1
46	n/a	return v
47	n/a
48	n/a	class E:
49	n/a	'Test propagation of exceptions'
50	n/a	def __init__(self, seqn):
51	n/a	self.seqn = seqn
52	n/a	self.i = 0
53	n/a	def __iter__(self):
54	n/a	return self
55	n/a	def __next__(self):
56	n/a	3 // 0
57	n/a
58	n/a	class N:
59	n/a	'Iterator missing __next__()'
60	n/a	def __init__(self, seqn):
61	n/a	self.seqn = seqn
62	n/a	self.i = 0
63	n/a	def __iter__(self):
64	n/a	return self
65	n/a
66	n/a	class PickleTest:
67	n/a	# Helper to check picklability
68	n/a	def check_pickle(self, itorg, seq):
69	n/a	for proto in range(pickle.HIGHEST_PROTOCOL + 1):
70	n/a	d = pickle.dumps(itorg, proto)
71	n/a	it = pickle.loads(d)
72	n/a	self.assertEqual(type(itorg), type(it))
73	n/a	self.assertEqual(list(it), seq)
74	n/a
75	n/a	it = pickle.loads(d)
76	n/a	try:
77	n/a	next(it)
78	n/a	except StopIteration:
79	n/a	self.assertFalse(seq[1:])
80	n/a	continue
81	n/a	d = pickle.dumps(it, proto)
82	n/a	it = pickle.loads(d)
83	n/a	self.assertEqual(list(it), seq[1:])
84	n/a
85	n/a	class EnumerateTestCase(unittest.TestCase, PickleTest):
86	n/a
87	n/a	enum = enumerate
88	n/a	seq, res = 'abc', [(0,'a'), (1,'b'), (2,'c')]
89	n/a
90	n/a	def test_basicfunction(self):
91	n/a	self.assertEqual(type(self.enum(self.seq)), self.enum)
92	n/a	e = self.enum(self.seq)
93	n/a	self.assertEqual(iter(e), e)
94	n/a	self.assertEqual(list(self.enum(self.seq)), self.res)
95	n/a	self.enum.__doc__
96	n/a
97	n/a	def test_pickle(self):
98	n/a	self.check_pickle(self.enum(self.seq), self.res)
99	n/a
100	n/a	def test_getitemseqn(self):
101	n/a	self.assertEqual(list(self.enum(G(self.seq))), self.res)
102	n/a	e = self.enum(G(''))
103	n/a	self.assertRaises(StopIteration, next, e)
104	n/a
105	n/a	def test_iteratorseqn(self):
106	n/a	self.assertEqual(list(self.enum(I(self.seq))), self.res)
107	n/a	e = self.enum(I(''))
108	n/a	self.assertRaises(StopIteration, next, e)
109	n/a
110	n/a	def test_iteratorgenerator(self):
111	n/a	self.assertEqual(list(self.enum(Ig(self.seq))), self.res)
112	n/a	e = self.enum(Ig(''))
113	n/a	self.assertRaises(StopIteration, next, e)
114	n/a
115	n/a	def test_noniterable(self):
116	n/a	self.assertRaises(TypeError, self.enum, X(self.seq))
117	n/a
118	n/a	def test_illformediterable(self):
119	n/a	self.assertRaises(TypeError, self.enum, N(self.seq))
120	n/a
121	n/a	def test_exception_propagation(self):
122	n/a	self.assertRaises(ZeroDivisionError, list, self.enum(E(self.seq)))
123	n/a
124	n/a	def test_argumentcheck(self):
125	n/a	self.assertRaises(TypeError, self.enum) # no arguments
126	n/a	self.assertRaises(TypeError, self.enum, 1) # wrong type (not iterable)
127	n/a	self.assertRaises(TypeError, self.enum, 'abc', 'a') # wrong type
128	n/a	self.assertRaises(TypeError, self.enum, 'abc', 2, 3) # too many arguments
129	n/a
130	n/a	@support.cpython_only
131	n/a	def test_tuple_reuse(self):
132	n/a	# Tests an implementation detail where tuple is reused
133	n/a	# whenever nothing else holds a reference to it
134	n/a	self.assertEqual(len(set(map(id, list(enumerate(self.seq))))), len(self.seq))
135	n/a	self.assertEqual(len(set(map(id, enumerate(self.seq)))), min(1,len(self.seq)))
136	n/a
137	n/a	class MyEnum(enumerate):
138	n/a	pass
139	n/a
140	n/a	class SubclassTestCase(EnumerateTestCase):
141	n/a
142	n/a	enum = MyEnum
143	n/a
144	n/a	class TestEmpty(EnumerateTestCase):
145	n/a
146	n/a	seq, res = '', []
147	n/a
148	n/a	class TestBig(EnumerateTestCase):
149	n/a
150	n/a	seq = range(10,20000,2)
151	n/a	res = list(zip(range(20000), seq))
152	n/a
153	n/a	class TestReversed(unittest.TestCase, PickleTest):
154	n/a
155	n/a	def test_simple(self):
156	n/a	class A:
157	n/a	def __getitem__(self, i):
158	n/a	if i < 5:
159	n/a	return str(i)
160	n/a	raise StopIteration
161	n/a	def __len__(self):
162	n/a	return 5
163	n/a	for data in 'abc', range(5), tuple(enumerate('abc')), A(), range(1,17,5):
164	n/a	self.assertEqual(list(data)[::-1], list(reversed(data)))
165	n/a	self.assertRaises(TypeError, reversed, {})
166	n/a	# don't allow keyword arguments
167	n/a	self.assertRaises(TypeError, reversed, [], a=1)
168	n/a
169	n/a	def test_range_optimization(self):
170	n/a	x = range(1)
171	n/a	self.assertEqual(type(reversed(x)), type(iter(x)))
172	n/a
173	n/a	def test_len(self):
174	n/a	for s in ('hello', tuple('hello'), list('hello'), range(5)):
175	n/a	self.assertEqual(operator.length_hint(reversed(s)), len(s))
176	n/a	r = reversed(s)
177	n/a	list(r)
178	n/a	self.assertEqual(operator.length_hint(r), 0)
179	n/a	class SeqWithWeirdLen:
180	n/a	called = False
181	n/a	def __len__(self):
182	n/a	if not self.called:
183	n/a	self.called = True
184	n/a	return 10
185	n/a	raise ZeroDivisionError
186	n/a	def __getitem__(self, index):
187	n/a	return index
188	n/a	r = reversed(SeqWithWeirdLen())
189	n/a	self.assertRaises(ZeroDivisionError, operator.length_hint, r)
190	n/a
191	n/a
192	n/a	def test_gc(self):
193	n/a	class Seq:
194	n/a	def __len__(self):
195	n/a	return 10
196	n/a	def __getitem__(self, index):
197	n/a	return index
198	n/a	s = Seq()
199	n/a	r = reversed(s)
200	n/a	s.r = r
201	n/a
202	n/a	def test_args(self):
203	n/a	self.assertRaises(TypeError, reversed)
204	n/a	self.assertRaises(TypeError, reversed, [], 'extra')
205	n/a
206	n/a	@unittest.skipUnless(hasattr(sys, 'getrefcount'), 'test needs sys.getrefcount()')
207	n/a	def test_bug1229429(self):
208	n/a	# this bug was never in reversed, it was in
209	n/a	# PyObject_CallMethod, and reversed_new calls that sometimes.
210	n/a	def f():
211	n/a	pass
212	n/a	r = f.__reversed__ = object()
213	n/a	rc = sys.getrefcount(r)
214	n/a	for i in range(10):
215	n/a	try:
216	n/a	reversed(f)
217	n/a	except TypeError:
218	n/a	pass
219	n/a	else:
220	n/a	self.fail("non-callable __reversed__ didn't raise!")
221	n/a	self.assertEqual(rc, sys.getrefcount(r))
222	n/a
223	n/a	def test_objmethods(self):
224	n/a	# Objects must have __len__() and __getitem__() implemented.
225	n/a	class NoLen(object):
226	n/a	def __getitem__(self, i): return 1
227	n/a	nl = NoLen()
228	n/a	self.assertRaises(TypeError, reversed, nl)
229	n/a
230	n/a	class NoGetItem(object):
231	n/a	def __len__(self): return 2
232	n/a	ngi = NoGetItem()
233	n/a	self.assertRaises(TypeError, reversed, ngi)
234	n/a
235	n/a	class Blocked(object):
236	n/a	def __getitem__(self, i): return 1
237	n/a	def __len__(self): return 2
238	n/a	__reversed__ = None
239	n/a	b = Blocked()
240	n/a	self.assertRaises(TypeError, reversed, b)
241	n/a
242	n/a	def test_pickle(self):
243	n/a	for data in 'abc', range(5), tuple(enumerate('abc')), range(1,17,5):
244	n/a	self.check_pickle(reversed(data), list(data)[::-1])
245	n/a
246	n/a
247	n/a	class EnumerateStartTestCase(EnumerateTestCase):
248	n/a
249	n/a	def test_basicfunction(self):
250	n/a	e = self.enum(self.seq)
251	n/a	self.assertEqual(iter(e), e)
252	n/a	self.assertEqual(list(self.enum(self.seq)), self.res)
253	n/a
254	n/a
255	n/a	class TestStart(EnumerateStartTestCase):
256	n/a
257	n/a	enum = lambda self, i: enumerate(i, start=11)
258	n/a	seq, res = 'abc', [(11, 'a'), (12, 'b'), (13, 'c')]
259	n/a
260	n/a
261	n/a	class TestLongStart(EnumerateStartTestCase):
262	n/a
263	n/a	enum = lambda self, i: enumerate(i, start=sys.maxsize+1)
264	n/a	seq, res = 'abc', [(sys.maxsize+1,'a'), (sys.maxsize+2,'b'),
265	n/a	(sys.maxsize+3,'c')]
266	n/a
267	n/a
268	n/a	if __name__ == "__main__":
269	n/a	unittest.main()