Python code coverage for Lib/test/test_pep380.py

#	count	content
1	n/a	# -- coding: utf-8 --
2	n/a
3	n/a	"""
4	n/a	Test suite for PEP 380 implementation
5	n/a
6	n/a	adapted from original tests written by Greg Ewing
7	n/a	see <http://www.cosc.canterbury.ac.nz/greg.ewing/python/yield-from/YieldFrom-Python3.1.2-rev5.zip>
8	n/a	"""
9	n/a
10	n/a	import unittest
11	n/a	import io
12	n/a	import sys
13	n/a	import inspect
14	n/a	import parser
15	n/a
16	n/a	from test.support import captured_stderr, disable_gc, gc_collect
17	n/a
18	n/a	class TestPEP380Operation(unittest.TestCase):
19	n/a	"""
20	n/a	Test semantics.
21	n/a	"""
22	n/a
23	n/a	def test_delegation_of_initial_next_to_subgenerator(self):
24	n/a	"""
25	n/a	Test delegation of initial next() call to subgenerator
26	n/a	"""
27	n/a	trace = []
28	n/a	def g1():
29	n/a	trace.append("Starting g1")
30	n/a	yield from g2()
31	n/a	trace.append("Finishing g1")
32	n/a	def g2():
33	n/a	trace.append("Starting g2")
34	n/a	yield 42
35	n/a	trace.append("Finishing g2")
36	n/a	for x in g1():
37	n/a	trace.append("Yielded %s" % (x,))
38	n/a	self.assertEqual(trace,[
39	n/a	"Starting g1",
40	n/a	"Starting g2",
41	n/a	"Yielded 42",
42	n/a	"Finishing g2",
43	n/a	"Finishing g1",
44	n/a	])
45	n/a
46	n/a	def test_raising_exception_in_initial_next_call(self):
47	n/a	"""
48	n/a	Test raising exception in initial next() call
49	n/a	"""
50	n/a	trace = []
51	n/a	def g1():
52	n/a	try:
53	n/a	trace.append("Starting g1")
54	n/a	yield from g2()
55	n/a	finally:
56	n/a	trace.append("Finishing g1")
57	n/a	def g2():
58	n/a	try:
59	n/a	trace.append("Starting g2")
60	n/a	raise ValueError("spanish inquisition occurred")
61	n/a	finally:
62	n/a	trace.append("Finishing g2")
63	n/a	try:
64	n/a	for x in g1():
65	n/a	trace.append("Yielded %s" % (x,))
66	n/a	except ValueError as e:
67	n/a	self.assertEqual(e.args[0], "spanish inquisition occurred")
68	n/a	else:
69	n/a	self.fail("subgenerator failed to raise ValueError")
70	n/a	self.assertEqual(trace,[
71	n/a	"Starting g1",
72	n/a	"Starting g2",
73	n/a	"Finishing g2",
74	n/a	"Finishing g1",
75	n/a	])
76	n/a
77	n/a	def test_delegation_of_next_call_to_subgenerator(self):
78	n/a	"""
79	n/a	Test delegation of next() call to subgenerator
80	n/a	"""
81	n/a	trace = []
82	n/a	def g1():
83	n/a	trace.append("Starting g1")
84	n/a	yield "g1 ham"
85	n/a	yield from g2()
86	n/a	yield "g1 eggs"
87	n/a	trace.append("Finishing g1")
88	n/a	def g2():
89	n/a	trace.append("Starting g2")
90	n/a	yield "g2 spam"
91	n/a	yield "g2 more spam"
92	n/a	trace.append("Finishing g2")
93	n/a	for x in g1():
94	n/a	trace.append("Yielded %s" % (x,))
95	n/a	self.assertEqual(trace,[
96	n/a	"Starting g1",
97	n/a	"Yielded g1 ham",
98	n/a	"Starting g2",
99	n/a	"Yielded g2 spam",
100	n/a	"Yielded g2 more spam",
101	n/a	"Finishing g2",
102	n/a	"Yielded g1 eggs",
103	n/a	"Finishing g1",
104	n/a	])
105	n/a
106	n/a	def test_raising_exception_in_delegated_next_call(self):
107	n/a	"""
108	n/a	Test raising exception in delegated next() call
109	n/a	"""
110	n/a	trace = []
111	n/a	def g1():
112	n/a	try:
113	n/a	trace.append("Starting g1")
114	n/a	yield "g1 ham"
115	n/a	yield from g2()
116	n/a	yield "g1 eggs"
117	n/a	finally:
118	n/a	trace.append("Finishing g1")
119	n/a	def g2():
120	n/a	try:
121	n/a	trace.append("Starting g2")
122	n/a	yield "g2 spam"
123	n/a	raise ValueError("hovercraft is full of eels")
124	n/a	yield "g2 more spam"
125	n/a	finally:
126	n/a	trace.append("Finishing g2")
127	n/a	try:
128	n/a	for x in g1():
129	n/a	trace.append("Yielded %s" % (x,))
130	n/a	except ValueError as e:
131	n/a	self.assertEqual(e.args[0], "hovercraft is full of eels")
132	n/a	else:
133	n/a	self.fail("subgenerator failed to raise ValueError")
134	n/a	self.assertEqual(trace,[
135	n/a	"Starting g1",
136	n/a	"Yielded g1 ham",
137	n/a	"Starting g2",
138	n/a	"Yielded g2 spam",
139	n/a	"Finishing g2",
140	n/a	"Finishing g1",
141	n/a	])
142	n/a
143	n/a	def test_delegation_of_send(self):
144	n/a	"""
145	n/a	Test delegation of send()
146	n/a	"""
147	n/a	trace = []
148	n/a	def g1():
149	n/a	trace.append("Starting g1")
150	n/a	x = yield "g1 ham"
151	n/a	trace.append("g1 received %s" % (x,))
152	n/a	yield from g2()
153	n/a	x = yield "g1 eggs"
154	n/a	trace.append("g1 received %s" % (x,))
155	n/a	trace.append("Finishing g1")
156	n/a	def g2():
157	n/a	trace.append("Starting g2")
158	n/a	x = yield "g2 spam"
159	n/a	trace.append("g2 received %s" % (x,))
160	n/a	x = yield "g2 more spam"
161	n/a	trace.append("g2 received %s" % (x,))
162	n/a	trace.append("Finishing g2")
163	n/a	g = g1()
164	n/a	y = next(g)
165	n/a	x = 1
166	n/a	try:
167	n/a	while 1:
168	n/a	y = g.send(x)
169	n/a	trace.append("Yielded %s" % (y,))
170	n/a	x += 1
171	n/a	except StopIteration:
172	n/a	pass
173	n/a	self.assertEqual(trace,[
174	n/a	"Starting g1",
175	n/a	"g1 received 1",
176	n/a	"Starting g2",
177	n/a	"Yielded g2 spam",
178	n/a	"g2 received 2",
179	n/a	"Yielded g2 more spam",
180	n/a	"g2 received 3",
181	n/a	"Finishing g2",
182	n/a	"Yielded g1 eggs",
183	n/a	"g1 received 4",
184	n/a	"Finishing g1",
185	n/a	])
186	n/a
187	n/a	def test_handling_exception_while_delegating_send(self):
188	n/a	"""
189	n/a	Test handling exception while delegating 'send'
190	n/a	"""
191	n/a	trace = []
192	n/a	def g1():
193	n/a	trace.append("Starting g1")
194	n/a	x = yield "g1 ham"
195	n/a	trace.append("g1 received %s" % (x,))
196	n/a	yield from g2()
197	n/a	x = yield "g1 eggs"
198	n/a	trace.append("g1 received %s" % (x,))
199	n/a	trace.append("Finishing g1")
200	n/a	def g2():
201	n/a	trace.append("Starting g2")
202	n/a	x = yield "g2 spam"
203	n/a	trace.append("g2 received %s" % (x,))
204	n/a	raise ValueError("hovercraft is full of eels")
205	n/a	x = yield "g2 more spam"
206	n/a	trace.append("g2 received %s" % (x,))
207	n/a	trace.append("Finishing g2")
208	n/a	def run():
209	n/a	g = g1()
210	n/a	y = next(g)
211	n/a	x = 1
212	n/a	try:
213	n/a	while 1:
214	n/a	y = g.send(x)
215	n/a	trace.append("Yielded %s" % (y,))
216	n/a	x += 1
217	n/a	except StopIteration:
218	n/a	trace.append("StopIteration")
219	n/a	self.assertRaises(ValueError,run)
220	n/a	self.assertEqual(trace,[
221	n/a	"Starting g1",
222	n/a	"g1 received 1",
223	n/a	"Starting g2",
224	n/a	"Yielded g2 spam",
225	n/a	"g2 received 2",
226	n/a	])
227	n/a
228	n/a	def test_delegating_close(self):
229	n/a	"""
230	n/a	Test delegating 'close'
231	n/a	"""
232	n/a	trace = []
233	n/a	def g1():
234	n/a	try:
235	n/a	trace.append("Starting g1")
236	n/a	yield "g1 ham"
237	n/a	yield from g2()
238	n/a	yield "g1 eggs"
239	n/a	finally:
240	n/a	trace.append("Finishing g1")
241	n/a	def g2():
242	n/a	try:
243	n/a	trace.append("Starting g2")
244	n/a	yield "g2 spam"
245	n/a	yield "g2 more spam"
246	n/a	finally:
247	n/a	trace.append("Finishing g2")
248	n/a	g = g1()
249	n/a	for i in range(2):
250	n/a	x = next(g)
251	n/a	trace.append("Yielded %s" % (x,))
252	n/a	g.close()
253	n/a	self.assertEqual(trace,[
254	n/a	"Starting g1",
255	n/a	"Yielded g1 ham",
256	n/a	"Starting g2",
257	n/a	"Yielded g2 spam",
258	n/a	"Finishing g2",
259	n/a	"Finishing g1"
260	n/a	])
261	n/a
262	n/a	def test_handing_exception_while_delegating_close(self):
263	n/a	"""
264	n/a	Test handling exception while delegating 'close'
265	n/a	"""
266	n/a	trace = []
267	n/a	def g1():
268	n/a	try:
269	n/a	trace.append("Starting g1")
270	n/a	yield "g1 ham"
271	n/a	yield from g2()
272	n/a	yield "g1 eggs"
273	n/a	finally:
274	n/a	trace.append("Finishing g1")
275	n/a	def g2():
276	n/a	try:
277	n/a	trace.append("Starting g2")
278	n/a	yield "g2 spam"
279	n/a	yield "g2 more spam"
280	n/a	finally:
281	n/a	trace.append("Finishing g2")
282	n/a	raise ValueError("nybbles have exploded with delight")
283	n/a	try:
284	n/a	g = g1()
285	n/a	for i in range(2):
286	n/a	x = next(g)
287	n/a	trace.append("Yielded %s" % (x,))
288	n/a	g.close()
289	n/a	except ValueError as e:
290	n/a	self.assertEqual(e.args[0], "nybbles have exploded with delight")
291	n/a	self.assertIsInstance(e.__context__, GeneratorExit)
292	n/a	else:
293	n/a	self.fail("subgenerator failed to raise ValueError")
294	n/a	self.assertEqual(trace,[
295	n/a	"Starting g1",
296	n/a	"Yielded g1 ham",
297	n/a	"Starting g2",
298	n/a	"Yielded g2 spam",
299	n/a	"Finishing g2",
300	n/a	"Finishing g1",
301	n/a	])
302	n/a
303	n/a	def test_delegating_throw(self):
304	n/a	"""
305	n/a	Test delegating 'throw'
306	n/a	"""
307	n/a	trace = []
308	n/a	def g1():
309	n/a	try:
310	n/a	trace.append("Starting g1")
311	n/a	yield "g1 ham"
312	n/a	yield from g2()
313	n/a	yield "g1 eggs"
314	n/a	finally:
315	n/a	trace.append("Finishing g1")
316	n/a	def g2():
317	n/a	try:
318	n/a	trace.append("Starting g2")
319	n/a	yield "g2 spam"
320	n/a	yield "g2 more spam"
321	n/a	finally:
322	n/a	trace.append("Finishing g2")
323	n/a	try:
324	n/a	g = g1()
325	n/a	for i in range(2):
326	n/a	x = next(g)
327	n/a	trace.append("Yielded %s" % (x,))
328	n/a	e = ValueError("tomato ejected")
329	n/a	g.throw(e)
330	n/a	except ValueError as e:
331	n/a	self.assertEqual(e.args[0], "tomato ejected")
332	n/a	else:
333	n/a	self.fail("subgenerator failed to raise ValueError")
334	n/a	self.assertEqual(trace,[
335	n/a	"Starting g1",
336	n/a	"Yielded g1 ham",
337	n/a	"Starting g2",
338	n/a	"Yielded g2 spam",
339	n/a	"Finishing g2",
340	n/a	"Finishing g1",
341	n/a	])
342	n/a
343	n/a	def test_value_attribute_of_StopIteration_exception(self):
344	n/a	"""
345	n/a	Test 'value' attribute of StopIteration exception
346	n/a	"""
347	n/a	trace = []
348	n/a	def pex(e):
349	n/a	trace.append("%s: %s" % (e.__class__.__name__, e))
350	n/a	trace.append("value = %s" % (e.value,))
351	n/a	e = StopIteration()
352	n/a	pex(e)
353	n/a	e = StopIteration("spam")
354	n/a	pex(e)
355	n/a	e.value = "eggs"
356	n/a	pex(e)
357	n/a	self.assertEqual(trace,[
358	n/a	"StopIteration: ",
359	n/a	"value = None",
360	n/a	"StopIteration: spam",
361	n/a	"value = spam",
362	n/a	"StopIteration: spam",
363	n/a	"value = eggs",
364	n/a	])
365	n/a
366	n/a
367	n/a	def test_exception_value_crash(self):
368	n/a	# There used to be a refcount error when the return value
369	n/a	# stored in the StopIteration has a refcount of 1.
370	n/a	def g1():
371	n/a	yield from g2()
372	n/a	def g2():
373	n/a	yield "g2"
374	n/a	return [42]
375	n/a	self.assertEqual(list(g1()), ["g2"])
376	n/a
377	n/a
378	n/a	def test_generator_return_value(self):
379	n/a	"""
380	n/a	Test generator return value
381	n/a	"""
382	n/a	trace = []
383	n/a	def g1():
384	n/a	trace.append("Starting g1")
385	n/a	yield "g1 ham"
386	n/a	ret = yield from g2()
387	n/a	trace.append("g2 returned %s" % (ret,))
388	n/a	ret = yield from g2(42)
389	n/a	trace.append("g2 returned %s" % (ret,))
390	n/a	yield "g1 eggs"
391	n/a	trace.append("Finishing g1")
392	n/a	def g2(v = None):
393	n/a	trace.append("Starting g2")
394	n/a	yield "g2 spam"
395	n/a	yield "g2 more spam"
396	n/a	trace.append("Finishing g2")
397	n/a	if v:
398	n/a	return v
399	n/a	for x in g1():
400	n/a	trace.append("Yielded %s" % (x,))
401	n/a	self.assertEqual(trace,[
402	n/a	"Starting g1",
403	n/a	"Yielded g1 ham",
404	n/a	"Starting g2",
405	n/a	"Yielded g2 spam",
406	n/a	"Yielded g2 more spam",
407	n/a	"Finishing g2",
408	n/a	"g2 returned None",
409	n/a	"Starting g2",
410	n/a	"Yielded g2 spam",
411	n/a	"Yielded g2 more spam",
412	n/a	"Finishing g2",
413	n/a	"g2 returned 42",
414	n/a	"Yielded g1 eggs",
415	n/a	"Finishing g1",
416	n/a	])
417	n/a
418	n/a	def test_delegation_of_next_to_non_generator(self):
419	n/a	"""
420	n/a	Test delegation of next() to non-generator
421	n/a	"""
422	n/a	trace = []
423	n/a	def g():
424	n/a	yield from range(3)
425	n/a	for x in g():
426	n/a	trace.append("Yielded %s" % (x,))
427	n/a	self.assertEqual(trace,[
428	n/a	"Yielded 0",
429	n/a	"Yielded 1",
430	n/a	"Yielded 2",
431	n/a	])
432	n/a
433	n/a
434	n/a	def test_conversion_of_sendNone_to_next(self):
435	n/a	"""
436	n/a	Test conversion of send(None) to next()
437	n/a	"""
438	n/a	trace = []
439	n/a	def g():
440	n/a	yield from range(3)
441	n/a	gi = g()
442	n/a	for x in range(3):
443	n/a	y = gi.send(None)
444	n/a	trace.append("Yielded: %s" % (y,))
445	n/a	self.assertEqual(trace,[
446	n/a	"Yielded: 0",
447	n/a	"Yielded: 1",
448	n/a	"Yielded: 2",
449	n/a	])
450	n/a
451	n/a	def test_delegation_of_close_to_non_generator(self):
452	n/a	"""
453	n/a	Test delegation of close() to non-generator
454	n/a	"""
455	n/a	trace = []
456	n/a	def g():
457	n/a	try:
458	n/a	trace.append("starting g")
459	n/a	yield from range(3)
460	n/a	trace.append("g should not be here")
461	n/a	finally:
462	n/a	trace.append("finishing g")
463	n/a	gi = g()
464	n/a	next(gi)
465	n/a	with captured_stderr() as output:
466	n/a	gi.close()
467	n/a	self.assertEqual(output.getvalue(), '')
468	n/a	self.assertEqual(trace,[
469	n/a	"starting g",
470	n/a	"finishing g",
471	n/a	])
472	n/a
473	n/a	def test_delegating_throw_to_non_generator(self):
474	n/a	"""
475	n/a	Test delegating 'throw' to non-generator
476	n/a	"""
477	n/a	trace = []
478	n/a	def g():
479	n/a	try:
480	n/a	trace.append("Starting g")
481	n/a	yield from range(10)
482	n/a	finally:
483	n/a	trace.append("Finishing g")
484	n/a	try:
485	n/a	gi = g()
486	n/a	for i in range(5):
487	n/a	x = next(gi)
488	n/a	trace.append("Yielded %s" % (x,))
489	n/a	e = ValueError("tomato ejected")
490	n/a	gi.throw(e)
491	n/a	except ValueError as e:
492	n/a	self.assertEqual(e.args[0],"tomato ejected")
493	n/a	else:
494	n/a	self.fail("subgenerator failed to raise ValueError")
495	n/a	self.assertEqual(trace,[
496	n/a	"Starting g",
497	n/a	"Yielded 0",
498	n/a	"Yielded 1",
499	n/a	"Yielded 2",
500	n/a	"Yielded 3",
501	n/a	"Yielded 4",
502	n/a	"Finishing g",
503	n/a	])
504	n/a
505	n/a	def test_attempting_to_send_to_non_generator(self):
506	n/a	"""
507	n/a	Test attempting to send to non-generator
508	n/a	"""
509	n/a	trace = []
510	n/a	def g():
511	n/a	try:
512	n/a	trace.append("starting g")
513	n/a	yield from range(3)
514	n/a	trace.append("g should not be here")
515	n/a	finally:
516	n/a	trace.append("finishing g")
517	n/a	try:
518	n/a	gi = g()
519	n/a	next(gi)
520	n/a	for x in range(3):
521	n/a	y = gi.send(42)
522	n/a	trace.append("Should not have yielded: %s" % (y,))
523	n/a	except AttributeError as e:
524	n/a	self.assertIn("send", e.args[0])
525	n/a	else:
526	n/a	self.fail("was able to send into non-generator")
527	n/a	self.assertEqual(trace,[
528	n/a	"starting g",
529	n/a	"finishing g",
530	n/a	])
531	n/a
532	n/a	def test_broken_getattr_handling(self):
533	n/a	"""
534	n/a	Test subiterator with a broken getattr implementation
535	n/a	"""
536	n/a	class Broken:
537	n/a	def __iter__(self):
538	n/a	return self
539	n/a	def __next__(self):
540	n/a	return 1
541	n/a	def __getattr__(self, attr):
542	n/a	1/0
543	n/a
544	n/a	def g():
545	n/a	yield from Broken()
546	n/a
547	n/a	with self.assertRaises(ZeroDivisionError):
548	n/a	gi = g()
549	n/a	self.assertEqual(next(gi), 1)
550	n/a	gi.send(1)
551	n/a
552	n/a	with self.assertRaises(ZeroDivisionError):
553	n/a	gi = g()
554	n/a	self.assertEqual(next(gi), 1)
555	n/a	gi.throw(AttributeError)
556	n/a
557	n/a	with captured_stderr() as output:
558	n/a	gi = g()
559	n/a	self.assertEqual(next(gi), 1)
560	n/a	gi.close()
561	n/a	self.assertIn('ZeroDivisionError', output.getvalue())
562	n/a
563	n/a	def test_exception_in_initial_next_call(self):
564	n/a	"""
565	n/a	Test exception in initial next() call
566	n/a	"""
567	n/a	trace = []
568	n/a	def g1():
569	n/a	trace.append("g1 about to yield from g2")
570	n/a	yield from g2()
571	n/a	trace.append("g1 should not be here")
572	n/a	def g2():
573	n/a	yield 1/0
574	n/a	def run():
575	n/a	gi = g1()
576	n/a	next(gi)
577	n/a	self.assertRaises(ZeroDivisionError,run)
578	n/a	self.assertEqual(trace,[
579	n/a	"g1 about to yield from g2"
580	n/a	])
581	n/a
582	n/a	def test_attempted_yield_from_loop(self):
583	n/a	"""
584	n/a	Test attempted yield-from loop
585	n/a	"""
586	n/a	trace = []
587	n/a	def g1():
588	n/a	trace.append("g1: starting")
589	n/a	yield "y1"
590	n/a	trace.append("g1: about to yield from g2")
591	n/a	yield from g2()
592	n/a	trace.append("g1 should not be here")
593	n/a
594	n/a	def g2():
595	n/a	trace.append("g2: starting")
596	n/a	yield "y2"
597	n/a	trace.append("g2: about to yield from g1")
598	n/a	yield from gi
599	n/a	trace.append("g2 should not be here")
600	n/a	try:
601	n/a	gi = g1()
602	n/a	for y in gi:
603	n/a	trace.append("Yielded: %s" % (y,))
604	n/a	except ValueError as e:
605	n/a	self.assertEqual(e.args[0],"generator already executing")
606	n/a	else:
607	n/a	self.fail("subgenerator didn't raise ValueError")
608	n/a	self.assertEqual(trace,[
609	n/a	"g1: starting",
610	n/a	"Yielded: y1",
611	n/a	"g1: about to yield from g2",
612	n/a	"g2: starting",
613	n/a	"Yielded: y2",
614	n/a	"g2: about to yield from g1",
615	n/a	])
616	n/a
617	n/a	def test_returning_value_from_delegated_throw(self):
618	n/a	"""
619	n/a	Test returning value from delegated 'throw'
620	n/a	"""
621	n/a	trace = []
622	n/a	def g1():
623	n/a	try:
624	n/a	trace.append("Starting g1")
625	n/a	yield "g1 ham"
626	n/a	yield from g2()
627	n/a	yield "g1 eggs"
628	n/a	finally:
629	n/a	trace.append("Finishing g1")
630	n/a	def g2():
631	n/a	try:
632	n/a	trace.append("Starting g2")
633	n/a	yield "g2 spam"
634	n/a	yield "g2 more spam"
635	n/a	except LunchError:
636	n/a	trace.append("Caught LunchError in g2")
637	n/a	yield "g2 lunch saved"
638	n/a	yield "g2 yet more spam"
639	n/a	class LunchError(Exception):
640	n/a	pass
641	n/a	g = g1()
642	n/a	for i in range(2):
643	n/a	x = next(g)
644	n/a	trace.append("Yielded %s" % (x,))
645	n/a	e = LunchError("tomato ejected")
646	n/a	g.throw(e)
647	n/a	for x in g:
648	n/a	trace.append("Yielded %s" % (x,))
649	n/a	self.assertEqual(trace,[
650	n/a	"Starting g1",
651	n/a	"Yielded g1 ham",
652	n/a	"Starting g2",
653	n/a	"Yielded g2 spam",
654	n/a	"Caught LunchError in g2",
655	n/a	"Yielded g2 yet more spam",
656	n/a	"Yielded g1 eggs",
657	n/a	"Finishing g1",
658	n/a	])
659	n/a
660	n/a	def test_next_and_return_with_value(self):
661	n/a	"""
662	n/a	Test next and return with value
663	n/a	"""
664	n/a	trace = []
665	n/a	def f(r):
666	n/a	gi = g(r)
667	n/a	next(gi)
668	n/a	try:
669	n/a	trace.append("f resuming g")
670	n/a	next(gi)
671	n/a	trace.append("f SHOULD NOT BE HERE")
672	n/a	except StopIteration as e:
673	n/a	trace.append("f caught %s" % (repr(e),))
674	n/a	def g(r):
675	n/a	trace.append("g starting")
676	n/a	yield
677	n/a	trace.append("g returning %s" % (r,))
678	n/a	return r
679	n/a	f(None)
680	n/a	f(42)
681	n/a	self.assertEqual(trace,[
682	n/a	"g starting",
683	n/a	"f resuming g",
684	n/a	"g returning None",
685	n/a	"f caught StopIteration()",
686	n/a	"g starting",
687	n/a	"f resuming g",
688	n/a	"g returning 42",
689	n/a	"f caught StopIteration(42,)",
690	n/a	])
691	n/a
692	n/a	def test_send_and_return_with_value(self):
693	n/a	"""
694	n/a	Test send and return with value
695	n/a	"""
696	n/a	trace = []
697	n/a	def f(r):
698	n/a	gi = g(r)
699	n/a	next(gi)
700	n/a	try:
701	n/a	trace.append("f sending spam to g")
702	n/a	gi.send("spam")
703	n/a	trace.append("f SHOULD NOT BE HERE")
704	n/a	except StopIteration as e:
705	n/a	trace.append("f caught %r" % (e,))
706	n/a	def g(r):
707	n/a	trace.append("g starting")
708	n/a	x = yield
709	n/a	trace.append("g received %s" % (x,))
710	n/a	trace.append("g returning %s" % (r,))
711	n/a	return r
712	n/a	f(None)
713	n/a	f(42)
714	n/a	self.assertEqual(trace,[
715	n/a	"g starting",
716	n/a	"f sending spam to g",
717	n/a	"g received spam",
718	n/a	"g returning None",
719	n/a	"f caught StopIteration()",
720	n/a	"g starting",
721	n/a	"f sending spam to g",
722	n/a	"g received spam",
723	n/a	"g returning 42",
724	n/a	"f caught StopIteration(42,)",
725	n/a	])
726	n/a
727	n/a	def test_catching_exception_from_subgen_and_returning(self):
728	n/a	"""
729	n/a	Test catching an exception thrown into a
730	n/a	subgenerator and returning a value
731	n/a	"""
732	n/a	trace = []
733	n/a	def inner():
734	n/a	try:
735	n/a	yield 1
736	n/a	except ValueError:
737	n/a	trace.append("inner caught ValueError")
738	n/a	return 2
739	n/a
740	n/a	def outer():
741	n/a	v = yield from inner()
742	n/a	trace.append("inner returned %r to outer" % v)
743	n/a	yield v
744	n/a	g = outer()
745	n/a	trace.append(next(g))
746	n/a	trace.append(g.throw(ValueError))
747	n/a	self.assertEqual(trace,[
748	n/a	1,
749	n/a	"inner caught ValueError",
750	n/a	"inner returned 2 to outer",
751	n/a	2,
752	n/a	])
753	n/a
754	n/a	def test_throwing_GeneratorExit_into_subgen_that_returns(self):
755	n/a	"""
756	n/a	Test throwing GeneratorExit into a subgenerator that
757	n/a	catches it and returns normally.
758	n/a	"""
759	n/a	trace = []
760	n/a	def f():
761	n/a	try:
762	n/a	trace.append("Enter f")
763	n/a	yield
764	n/a	trace.append("Exit f")
765	n/a	except GeneratorExit:
766	n/a	return
767	n/a	def g():
768	n/a	trace.append("Enter g")
769	n/a	yield from f()
770	n/a	trace.append("Exit g")
771	n/a	try:
772	n/a	gi = g()
773	n/a	next(gi)
774	n/a	gi.throw(GeneratorExit)
775	n/a	except GeneratorExit:
776	n/a	pass
777	n/a	else:
778	n/a	self.fail("subgenerator failed to raise GeneratorExit")
779	n/a	self.assertEqual(trace,[
780	n/a	"Enter g",
781	n/a	"Enter f",
782	n/a	])
783	n/a
784	n/a	def test_throwing_GeneratorExit_into_subgenerator_that_yields(self):
785	n/a	"""
786	n/a	Test throwing GeneratorExit into a subgenerator that
787	n/a	catches it and yields.
788	n/a	"""
789	n/a	trace = []
790	n/a	def f():
791	n/a	try:
792	n/a	trace.append("Enter f")
793	n/a	yield
794	n/a	trace.append("Exit f")
795	n/a	except GeneratorExit:
796	n/a	yield
797	n/a	def g():
798	n/a	trace.append("Enter g")
799	n/a	yield from f()
800	n/a	trace.append("Exit g")
801	n/a	try:
802	n/a	gi = g()
803	n/a	next(gi)
804	n/a	gi.throw(GeneratorExit)
805	n/a	except RuntimeError as e:
806	n/a	self.assertEqual(e.args[0], "generator ignored GeneratorExit")
807	n/a	else:
808	n/a	self.fail("subgenerator failed to raise GeneratorExit")
809	n/a	self.assertEqual(trace,[
810	n/a	"Enter g",
811	n/a	"Enter f",
812	n/a	])
813	n/a
814	n/a	def test_throwing_GeneratorExit_into_subgen_that_raises(self):
815	n/a	"""
816	n/a	Test throwing GeneratorExit into a subgenerator that
817	n/a	catches it and raises a different exception.
818	n/a	"""
819	n/a	trace = []
820	n/a	def f():
821	n/a	try:
822	n/a	trace.append("Enter f")
823	n/a	yield
824	n/a	trace.append("Exit f")
825	n/a	except GeneratorExit:
826	n/a	raise ValueError("Vorpal bunny encountered")
827	n/a	def g():
828	n/a	trace.append("Enter g")
829	n/a	yield from f()
830	n/a	trace.append("Exit g")
831	n/a	try:
832	n/a	gi = g()
833	n/a	next(gi)
834	n/a	gi.throw(GeneratorExit)
835	n/a	except ValueError as e:
836	n/a	self.assertEqual(e.args[0], "Vorpal bunny encountered")
837	n/a	self.assertIsInstance(e.__context__, GeneratorExit)
838	n/a	else:
839	n/a	self.fail("subgenerator failed to raise ValueError")
840	n/a	self.assertEqual(trace,[
841	n/a	"Enter g",
842	n/a	"Enter f",
843	n/a	])
844	n/a
845	n/a	def test_yield_from_empty(self):
846	n/a	def g():
847	n/a	yield from ()
848	n/a	self.assertRaises(StopIteration, next, g())
849	n/a
850	n/a	def test_delegating_generators_claim_to_be_running(self):
851	n/a	# Check with basic iteration
852	n/a	def one():
853	n/a	yield 0
854	n/a	yield from two()
855	n/a	yield 3
856	n/a	def two():
857	n/a	yield 1
858	n/a	try:
859	n/a	yield from g1
860	n/a	except ValueError:
861	n/a	pass
862	n/a	yield 2
863	n/a	g1 = one()
864	n/a	self.assertEqual(list(g1), [0, 1, 2, 3])
865	n/a	# Check with send
866	n/a	g1 = one()
867	n/a	res = [next(g1)]
868	n/a	try:
869	n/a	while True:
870	n/a	res.append(g1.send(42))
871	n/a	except StopIteration:
872	n/a	pass
873	n/a	self.assertEqual(res, [0, 1, 2, 3])
874	n/a	# Check with throw
875	n/a	class MyErr(Exception):
876	n/a	pass
877	n/a	def one():
878	n/a	try:
879	n/a	yield 0
880	n/a	except MyErr:
881	n/a	pass
882	n/a	yield from two()
883	n/a	try:
884	n/a	yield 3
885	n/a	except MyErr:
886	n/a	pass
887	n/a	def two():
888	n/a	try:
889	n/a	yield 1
890	n/a	except MyErr:
891	n/a	pass
892	n/a	try:
893	n/a	yield from g1
894	n/a	except ValueError:
895	n/a	pass
896	n/a	try:
897	n/a	yield 2
898	n/a	except MyErr:
899	n/a	pass
900	n/a	g1 = one()
901	n/a	res = [next(g1)]
902	n/a	try:
903	n/a	while True:
904	n/a	res.append(g1.throw(MyErr))
905	n/a	except StopIteration:
906	n/a	pass
907	n/a	# Check with close
908	n/a	class MyIt(object):
909	n/a	def __iter__(self):
910	n/a	return self
911	n/a	def __next__(self):
912	n/a	return 42
913	n/a	def close(self_):
914	n/a	self.assertTrue(g1.gi_running)
915	n/a	self.assertRaises(ValueError, next, g1)
916	n/a	def one():
917	n/a	yield from MyIt()
918	n/a	g1 = one()
919	n/a	next(g1)
920	n/a	g1.close()
921	n/a
922	n/a	def test_delegator_is_visible_to_debugger(self):
923	n/a	def call_stack():
924	n/a	return [f[3] for f in inspect.stack()]
925	n/a
926	n/a	def gen():
927	n/a	yield call_stack()
928	n/a	yield call_stack()
929	n/a	yield call_stack()
930	n/a
931	n/a	def spam(g):
932	n/a	yield from g
933	n/a
934	n/a	def eggs(g):
935	n/a	yield from g
936	n/a
937	n/a	for stack in spam(gen()):
938	n/a	self.assertTrue('spam' in stack)
939	n/a
940	n/a	for stack in spam(eggs(gen())):
941	n/a	self.assertTrue('spam' in stack and 'eggs' in stack)
942	n/a
943	n/a	def test_custom_iterator_return(self):
944	n/a	# See issue #15568
945	n/a	class MyIter:
946	n/a	def __iter__(self):
947	n/a	return self
948	n/a	def __next__(self):
949	n/a	raise StopIteration(42)
950	n/a	def gen():
951	n/a	nonlocal ret
952	n/a	ret = yield from MyIter()
953	n/a	ret = None
954	n/a	list(gen())
955	n/a	self.assertEqual(ret, 42)
956	n/a
957	n/a	def test_close_with_cleared_frame(self):
958	n/a	# See issue #17669.
959	n/a	#
960	n/a	# Create a stack of generators: outer() delegating to inner()
961	n/a	# delegating to innermost(). The key point is that the instance of
962	n/a	# inner is created first: this ensures that its frame appears before
963	n/a	# the instance of outer in the GC linked list.
964	n/a	#
965	n/a	# At the gc.collect call:
966	n/a	# - frame_clear is called on the inner_gen frame.
967	n/a	# - gen_dealloc is called on the outer_gen generator (the only
968	n/a	# reference is in the frame's locals).
969	n/a	# - gen_close is called on the outer_gen generator.
970	n/a	# - gen_close_iter is called to close the inner_gen generator, which
971	n/a	# in turn calls gen_close, and gen_yf.
972	n/a	#
973	n/a	# Previously, gen_yf would crash since inner_gen's frame had been
974	n/a	# cleared (and in particular f_stacktop was NULL).
975	n/a
976	n/a	def innermost():
977	n/a	yield
978	n/a	def inner():
979	n/a	outer_gen = yield
980	n/a	yield from innermost()
981	n/a	def outer():
982	n/a	inner_gen = yield
983	n/a	yield from inner_gen
984	n/a
985	n/a	with disable_gc():
986	n/a	inner_gen = inner()
987	n/a	outer_gen = outer()
988	n/a	outer_gen.send(None)
989	n/a	outer_gen.send(inner_gen)
990	n/a	outer_gen.send(outer_gen)
991	n/a
992	n/a	del outer_gen
993	n/a	del inner_gen
994	n/a	gc_collect()
995	n/a
996	n/a	def test_send_tuple_with_custom_generator(self):
997	n/a	# See issue #21209.
998	n/a	class MyGen:
999	n/a	def __iter__(self):
1000	n/a	return self
1001	n/a	def __next__(self):
1002	n/a	return 42
1003	n/a	def send(self, what):
1004	n/a	nonlocal v
1005	n/a	v = what
1006	n/a	return None
1007	n/a	def outer():
1008	n/a	v = yield from MyGen()
1009	n/a	g = outer()
1010	n/a	next(g)
1011	n/a	v = None
1012	n/a	g.send((1, 2, 3, 4))
1013	n/a	self.assertEqual(v, (1, 2, 3, 4))
1014	n/a
1015	n/a
1016	n/a	if __name__ == '__main__':
1017	n/a	unittest.main()