Python code coverage for Lib/test/test_codeop.py

#	count	content
1	n/a	"""
2	n/a	Test cases for codeop.py
3	n/a	Nick Mathewson
4	n/a	"""
5	n/a	import unittest
6	n/a	from test.support import is_jython
7	n/a
8	n/a	from codeop import compile_command, PyCF_DONT_IMPLY_DEDENT
9	n/a	import io
10	n/a
11	n/a	if is_jython:
12	n/a	import sys
13	n/a
14	n/a	def unify_callables(d):
15	n/a	for n,v in d.items():
16	n/a	if hasattr(v, '__call__'):
17	n/a	d[n] = True
18	n/a	return d
19	n/a
20	n/a	class CodeopTests(unittest.TestCase):
21	n/a
22	n/a	def assertValid(self, str, symbol='single'):
23	n/a	'''succeed iff str is a valid piece of code'''
24	n/a	if is_jython:
25	n/a	code = compile_command(str, "<input>", symbol)
26	n/a	self.assertTrue(code)
27	n/a	if symbol == "single":
28	n/a	d,r = {},{}
29	n/a	saved_stdout = sys.stdout
30	n/a	sys.stdout = io.StringIO()
31	n/a	try:
32	n/a	exec(code, d)
33	n/a	exec(compile(str,"<input>","single"), r)
34	n/a	finally:
35	n/a	sys.stdout = saved_stdout
36	n/a	elif symbol == 'eval':
37	n/a	ctx = {'a': 2}
38	n/a	d = { 'value': eval(code,ctx) }
39	n/a	r = { 'value': eval(str,ctx) }
40	n/a	self.assertEqual(unify_callables(r),unify_callables(d))
41	n/a	else:
42	n/a	expected = compile(str, "<input>", symbol, PyCF_DONT_IMPLY_DEDENT)
43	n/a	self.assertEqual(compile_command(str, "<input>", symbol), expected)
44	n/a
45	n/a	def assertIncomplete(self, str, symbol='single'):
46	n/a	'''succeed iff str is the start of a valid piece of code'''
47	n/a	self.assertEqual(compile_command(str, symbol=symbol), None)
48	n/a
49	n/a	def assertInvalid(self, str, symbol='single', is_syntax=1):
50	n/a	'''succeed iff str is the start of an invalid piece of code'''
51	n/a	try:
52	n/a	compile_command(str,symbol=symbol)
53	n/a	self.fail("No exception raised for invalid code")
54	n/a	except SyntaxError:
55	n/a	self.assertTrue(is_syntax)
56	n/a	except OverflowError:
57	n/a	self.assertTrue(not is_syntax)
58	n/a
59	n/a	def test_valid(self):
60	n/a	av = self.assertValid
61	n/a
62	n/a	# special case
63	n/a	if not is_jython:
64	n/a	self.assertEqual(compile_command(""),
65	n/a	compile("pass", "<input>", 'single',
66	n/a	PyCF_DONT_IMPLY_DEDENT))
67	n/a	self.assertEqual(compile_command("\n"),
68	n/a	compile("pass", "<input>", 'single',
69	n/a	PyCF_DONT_IMPLY_DEDENT))
70	n/a	else:
71	n/a	av("")
72	n/a	av("\n")
73	n/a
74	n/a	av("a = 1")
75	n/a	av("\na = 1")
76	n/a	av("a = 1\n")
77	n/a	av("a = 1\n\n")
78	n/a	av("\n\na = 1\n\n")
79	n/a
80	n/a	av("def x():\n pass\n")
81	n/a	av("if 1:\n pass\n")
82	n/a
83	n/a	av("\n\nif 1: pass\n")
84	n/a	av("\n\nif 1: pass\n\n")
85	n/a
86	n/a	av("def x():\n\n pass\n")
87	n/a	av("def x():\n pass\n \n")
88	n/a	av("def x():\n pass\n \n")
89	n/a
90	n/a	av("pass\n")
91	n/a	av("3**3\n")
92	n/a
93	n/a	av("if 9==3:\n pass\nelse:\n pass\n")
94	n/a	av("if 1:\n pass\n if 1:\n pass\n else:\n pass\n")
95	n/a
96	n/a	av("#a\n#b\na = 3\n")
97	n/a	av("#a\n\n \na=3\n")
98	n/a	av("a=3\n\n")
99	n/a	av("a = 9+ \\\n3")
100	n/a
101	n/a	av("3**3","eval")
102	n/a	av("(lambda z: \n z**3)","eval")
103	n/a
104	n/a	av("9+ \\\n3","eval")
105	n/a	av("9+ \\\n3\n","eval")
106	n/a
107	n/a	av("\n\na**3","eval")
108	n/a	av("\n \na**3","eval")
109	n/a	av("#a\n#b\na**3","eval")
110	n/a
111	n/a	av("\n\na = 1\n\n")
112	n/a	av("\n\nif 1: a=1\n\n")
113	n/a
114	n/a	av("if 1:\n pass\n if 1:\n pass\n else:\n pass\n")
115	n/a	av("#a\n\n \na=3\n\n")
116	n/a
117	n/a	av("\n\na**3","eval")
118	n/a	av("\n \na**3","eval")
119	n/a	av("#a\n#b\na**3","eval")
120	n/a
121	n/a	av("def f():\n try: pass\n finally: [x for x in (1,2)]\n")
122	n/a	av("def f():\n pass\n#foo\n")
123	n/a	av("@a.b.c\ndef f():\n pass\n")
124	n/a
125	n/a	def test_incomplete(self):
126	n/a	ai = self.assertIncomplete
127	n/a
128	n/a	ai("(a **")
129	n/a	ai("(a,b,")
130	n/a	ai("(a,b,(")
131	n/a	ai("(a,b,(")
132	n/a	ai("a = (")
133	n/a	ai("a = {")
134	n/a	ai("b + {")
135	n/a
136	n/a	ai("if 9==3:\n pass\nelse:")
137	n/a	ai("if 9==3:\n pass\nelse:\n")
138	n/a	ai("if 9==3:\n pass\nelse:\n pass")
139	n/a	ai("if 1:")
140	n/a	ai("if 1:\n")
141	n/a	ai("if 1:\n pass\n if 1:\n pass\n else:")
142	n/a	ai("if 1:\n pass\n if 1:\n pass\n else:\n")
143	n/a	ai("if 1:\n pass\n if 1:\n pass\n else:\n pass")
144	n/a
145	n/a	ai("def x():")
146	n/a	ai("def x():\n")
147	n/a	ai("def x():\n\n")
148	n/a
149	n/a	ai("def x():\n pass")
150	n/a	ai("def x():\n pass\n ")
151	n/a	ai("def x():\n pass\n ")
152	n/a	ai("\n\ndef x():\n pass")
153	n/a
154	n/a	ai("a = 9+ \\")
155	n/a	ai("a = 'a\\")
156	n/a	ai("a = '''xy")
157	n/a
158	n/a	ai("","eval")
159	n/a	ai("\n","eval")
160	n/a	ai("(","eval")
161	n/a	ai("(\n\n\n","eval")
162	n/a	ai("(9+","eval")
163	n/a	ai("9+ \\","eval")
164	n/a	ai("lambda z: \\","eval")
165	n/a
166	n/a	ai("if True:\n if True:\n if True: \n")
167	n/a
168	n/a	ai("@a(")
169	n/a	ai("@a(b")
170	n/a	ai("@a(b,")
171	n/a	ai("@a(b,c")
172	n/a	ai("@a(b,c,")
173	n/a
174	n/a	ai("from a import (")
175	n/a	ai("from a import (b")
176	n/a	ai("from a import (b,")
177	n/a	ai("from a import (b,c")
178	n/a	ai("from a import (b,c,")
179	n/a
180	n/a	ai("[");
181	n/a	ai("[a");
182	n/a	ai("[a,");
183	n/a	ai("[a,b");
184	n/a	ai("[a,b,");
185	n/a
186	n/a	ai("{");
187	n/a	ai("{a");
188	n/a	ai("{a:");
189	n/a	ai("{a:b");
190	n/a	ai("{a:b,");
191	n/a	ai("{a:b,c");
192	n/a	ai("{a:b,c:");
193	n/a	ai("{a:b,c:d");
194	n/a	ai("{a:b,c:d,");
195	n/a
196	n/a	ai("a(")
197	n/a	ai("a(b")
198	n/a	ai("a(b,")
199	n/a	ai("a(b,c")
200	n/a	ai("a(b,c,")
201	n/a
202	n/a	ai("a[")
203	n/a	ai("a[b")
204	n/a	ai("a[b,")
205	n/a	ai("a[b:")
206	n/a	ai("a[b:c")
207	n/a	ai("a[b:c:")
208	n/a	ai("a[b:c:d")
209	n/a
210	n/a	ai("def a(")
211	n/a	ai("def a(b")
212	n/a	ai("def a(b,")
213	n/a	ai("def a(b,c")
214	n/a	ai("def a(b,c,")
215	n/a
216	n/a	ai("(")
217	n/a	ai("(a")
218	n/a	ai("(a,")
219	n/a	ai("(a,b")
220	n/a	ai("(a,b,")
221	n/a
222	n/a	ai("if a:\n pass\nelif b:")
223	n/a	ai("if a:\n pass\nelif b:\n pass\nelse:")
224	n/a
225	n/a	ai("while a:")
226	n/a	ai("while a:\n pass\nelse:")
227	n/a
228	n/a	ai("for a in b:")
229	n/a	ai("for a in b:\n pass\nelse:")
230	n/a
231	n/a	ai("try:")
232	n/a	ai("try:\n pass\nexcept:")
233	n/a	ai("try:\n pass\nfinally:")
234	n/a	ai("try:\n pass\nexcept:\n pass\nfinally:")
235	n/a
236	n/a	ai("with a:")
237	n/a	ai("with a as b:")
238	n/a
239	n/a	ai("class a:")
240	n/a	ai("class a(")
241	n/a	ai("class a(b")
242	n/a	ai("class a(b,")
243	n/a	ai("class a():")
244	n/a
245	n/a	ai("[x for")
246	n/a	ai("[x for x in")
247	n/a	ai("[x for x in (")
248	n/a
249	n/a	ai("(x for")
250	n/a	ai("(x for x in")
251	n/a	ai("(x for x in (")
252	n/a
253	n/a	def test_invalid(self):
254	n/a	ai = self.assertInvalid
255	n/a	ai("a b")
256	n/a
257	n/a	ai("a @")
258	n/a	ai("a b @")
259	n/a	ai("a ** @")
260	n/a
261	n/a	ai("a = ")
262	n/a	ai("a = 9 +")
263	n/a
264	n/a	ai("def x():\n\npass\n")
265	n/a
266	n/a	ai("\n\n if 1: pass\n\npass")
267	n/a
268	n/a	ai("a = 9+ \\\n")
269	n/a	ai("a = 'a\\ ")
270	n/a	ai("a = 'a\\\n")
271	n/a
272	n/a	ai("a = 1","eval")
273	n/a	ai("a = (","eval")
274	n/a	ai("]","eval")
275	n/a	ai("())","eval")
276	n/a	ai("[}","eval")
277	n/a	ai("9+","eval")
278	n/a	ai("lambda z:","eval")
279	n/a	ai("a b","eval")
280	n/a
281	n/a	ai("return 2.3")
282	n/a	ai("if (a == 1 and b = 2): pass")
283	n/a
284	n/a	ai("del 1")
285	n/a	ai("del (1,)")
286	n/a	ai("del [1]")
287	n/a	ai("del '1'")
288	n/a
289	n/a	ai("[i for i in range(10)] = (1, 2, 3)")
290	n/a
291	n/a	def test_filename(self):
292	n/a	self.assertEqual(compile_command("a = 1\n", "abc").co_filename,
293	n/a	compile("a = 1\n", "abc", 'single').co_filename)
294	n/a	self.assertNotEqual(compile_command("a = 1\n", "abc").co_filename,
295	n/a	compile("a = 1\n", "def", 'single').co_filename)
296	n/a
297	n/a
298	n/a	if __name__ == "__main__":
299	n/a	unittest.main()