blob: 1069606cffba1264e8777147156e26d03d7f88b8 [file] [log] [blame]
Fred Drake79ca79d2000-08-21 22:30:53 +00001import parser
Benjamin Petersoneeed0c72008-11-03 15:18:30 +00002import os
Fred Drake58422e52001-06-04 03:56:24 +00003import unittest
Martin v. Löwis66e26632008-03-18 13:16:05 +00004import sys
Barry Warsaw04f357c2002-07-23 19:04:11 +00005from test import test_support
Fred Drake79ca79d2000-08-21 22:30:53 +00006
7#
8# First, we test that we can generate trees from valid source fragments,
9# and that these valid trees are indeed allowed by the tree-loading side
10# of the parser module.
11#
12
Fred Drake58422e52001-06-04 03:56:24 +000013class RoundtripLegalSyntaxTestCase(unittest.TestCase):
Guido van Rossum32c2ae72002-08-22 19:45:32 +000014
Fred Drake58422e52001-06-04 03:56:24 +000015 def roundtrip(self, f, s):
16 st1 = f(s)
17 t = st1.totuple()
18 try:
Fred Drake6e4f2c02001-07-17 19:33:25 +000019 st2 = parser.sequence2st(t)
Anthony Baxterc2a5a632004-08-02 06:10:11 +000020 except parser.ParserError, why:
21 self.fail("could not roundtrip %r: %s" % (s, why))
Fred Drake79ca79d2000-08-21 22:30:53 +000022
Fred Drake58422e52001-06-04 03:56:24 +000023 self.assertEquals(t, st2.totuple(),
24 "could not re-generate syntax tree")
Fred Drake28f739a2000-08-25 22:42:40 +000025
Fred Drake58422e52001-06-04 03:56:24 +000026 def check_expr(self, s):
27 self.roundtrip(parser.expr, s)
Fred Drake28f739a2000-08-25 22:42:40 +000028
Benjamin Peterson084ce7a2008-10-31 02:26:20 +000029 def test_flags_passed(self):
30 # The unicode literals flags has to be passed from the paser to AST
31 # generation.
32 suite = parser.suite("from __future__ import unicode_literals; x = ''")
33 code = suite.compile()
34 scope = {}
35 exec code in scope
36 self.assertTrue(isinstance(scope["x"], unicode))
37
Fred Drake58422e52001-06-04 03:56:24 +000038 def check_suite(self, s):
39 self.roundtrip(parser.suite, s)
Fred Drake28f739a2000-08-25 22:42:40 +000040
Fred Drakecf580c72001-07-17 03:01:29 +000041 def test_yield_statement(self):
Tim Peters496563a2002-04-01 00:28:59 +000042 self.check_suite("def f(): yield 1")
Phillip J. Eby0d6615f2005-08-02 00:46:46 +000043 self.check_suite("def f(): yield")
44 self.check_suite("def f(): x += yield")
45 self.check_suite("def f(): x = yield 1")
46 self.check_suite("def f(): x = y = yield 1")
47 self.check_suite("def f(): x = yield")
48 self.check_suite("def f(): x = y = yield")
49 self.check_suite("def f(): 1 + (yield)*2")
50 self.check_suite("def f(): (yield 1)*2")
Tim Peters496563a2002-04-01 00:28:59 +000051 self.check_suite("def f(): return; yield 1")
52 self.check_suite("def f(): yield 1; return")
53 self.check_suite("def f():\n"
Fred Drakecf580c72001-07-17 03:01:29 +000054 " for x in range(30):\n"
55 " yield x\n")
Phillip J. Eby0d6615f2005-08-02 00:46:46 +000056 self.check_suite("def f():\n"
57 " if (yield):\n"
58 " yield x\n")
Fred Drakecf580c72001-07-17 03:01:29 +000059
Fred Drake58422e52001-06-04 03:56:24 +000060 def test_expressions(self):
61 self.check_expr("foo(1)")
62 self.check_expr("[1, 2, 3]")
63 self.check_expr("[x**3 for x in range(20)]")
64 self.check_expr("[x**3 for x in range(20) if x % 3]")
Neal Norwitzd3a91622006-04-12 05:27:46 +000065 self.check_expr("[x**3 for x in range(20) if x % 2 if x % 3]")
66 self.check_expr("list(x**3 for x in range(20))")
67 self.check_expr("list(x**3 for x in range(20) if x % 3)")
68 self.check_expr("list(x**3 for x in range(20) if x % 2 if x % 3)")
Fred Drake58422e52001-06-04 03:56:24 +000069 self.check_expr("foo(*args)")
70 self.check_expr("foo(*args, **kw)")
71 self.check_expr("foo(**kw)")
72 self.check_expr("foo(key=value)")
73 self.check_expr("foo(key=value, *args)")
74 self.check_expr("foo(key=value, *args, **kw)")
75 self.check_expr("foo(key=value, **kw)")
76 self.check_expr("foo(a, b, c, *args)")
77 self.check_expr("foo(a, b, c, *args, **kw)")
78 self.check_expr("foo(a, b, c, **kw)")
Benjamin Petersonbd6a05f2008-08-19 22:06:11 +000079 self.check_expr("foo(a, *args, keyword=23)")
Fred Drake58422e52001-06-04 03:56:24 +000080 self.check_expr("foo + bar")
Michael W. Hudson5e83b7a2003-01-29 14:20:23 +000081 self.check_expr("foo - bar")
82 self.check_expr("foo * bar")
83 self.check_expr("foo / bar")
84 self.check_expr("foo // bar")
Fred Drake58422e52001-06-04 03:56:24 +000085 self.check_expr("lambda: 0")
86 self.check_expr("lambda x: 0")
87 self.check_expr("lambda *y: 0")
88 self.check_expr("lambda *y, **z: 0")
89 self.check_expr("lambda **z: 0")
90 self.check_expr("lambda x, y: 0")
91 self.check_expr("lambda foo=bar: 0")
92 self.check_expr("lambda foo=bar, spaz=nifty+spit: 0")
93 self.check_expr("lambda foo=bar, **z: 0")
94 self.check_expr("lambda foo=bar, blaz=blat+2, **z: 0")
95 self.check_expr("lambda foo=bar, blaz=blat+2, *y, **z: 0")
96 self.check_expr("lambda x, *y, **z: 0")
Raymond Hettinger354433a2004-05-19 08:20:33 +000097 self.check_expr("(x for x in range(10))")
98 self.check_expr("foo(x for x in range(10))")
Fred Drake79ca79d2000-08-21 22:30:53 +000099
Fred Drake58422e52001-06-04 03:56:24 +0000100 def test_print(self):
101 self.check_suite("print")
102 self.check_suite("print 1")
103 self.check_suite("print 1,")
104 self.check_suite("print >>fp")
105 self.check_suite("print >>fp, 1")
106 self.check_suite("print >>fp, 1,")
Fred Drake79ca79d2000-08-21 22:30:53 +0000107
Fred Drake58422e52001-06-04 03:56:24 +0000108 def test_simple_expression(self):
109 # expr_stmt
110 self.check_suite("a")
Fred Drake79ca79d2000-08-21 22:30:53 +0000111
Fred Drake58422e52001-06-04 03:56:24 +0000112 def test_simple_assignments(self):
113 self.check_suite("a = b")
114 self.check_suite("a = b = c = d = e")
Fred Drake28f739a2000-08-25 22:42:40 +0000115
Fred Drake58422e52001-06-04 03:56:24 +0000116 def test_simple_augmented_assignments(self):
117 self.check_suite("a += b")
118 self.check_suite("a -= b")
119 self.check_suite("a *= b")
120 self.check_suite("a /= b")
Michael W. Hudson5e83b7a2003-01-29 14:20:23 +0000121 self.check_suite("a //= b")
Fred Drake58422e52001-06-04 03:56:24 +0000122 self.check_suite("a %= b")
123 self.check_suite("a &= b")
124 self.check_suite("a |= b")
125 self.check_suite("a ^= b")
126 self.check_suite("a <<= b")
127 self.check_suite("a >>= b")
128 self.check_suite("a **= b")
Fred Drakee3fb18c2001-01-07 06:02:19 +0000129
Fred Drake58422e52001-06-04 03:56:24 +0000130 def test_function_defs(self):
131 self.check_suite("def f(): pass")
132 self.check_suite("def f(*args): pass")
133 self.check_suite("def f(*args, **kw): pass")
134 self.check_suite("def f(**kw): pass")
135 self.check_suite("def f(foo=bar): pass")
136 self.check_suite("def f(foo=bar, *args): pass")
137 self.check_suite("def f(foo=bar, *args, **kw): pass")
138 self.check_suite("def f(foo=bar, **kw): pass")
Fred Drakee3fb18c2001-01-07 06:02:19 +0000139
Fred Drake58422e52001-06-04 03:56:24 +0000140 self.check_suite("def f(a, b): pass")
141 self.check_suite("def f(a, b, *args): pass")
142 self.check_suite("def f(a, b, *args, **kw): pass")
143 self.check_suite("def f(a, b, **kw): pass")
144 self.check_suite("def f(a, b, foo=bar): pass")
145 self.check_suite("def f(a, b, foo=bar, *args): pass")
146 self.check_suite("def f(a, b, foo=bar, *args, **kw): pass")
147 self.check_suite("def f(a, b, foo=bar, **kw): pass")
Fred Drakee3fb18c2001-01-07 06:02:19 +0000148
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000149 self.check_suite("@staticmethod\n"
150 "def f(): pass")
151 self.check_suite("@staticmethod\n"
152 "@funcattrs(x, y)\n"
153 "def f(): pass")
154 self.check_suite("@funcattrs()\n"
155 "def f(): pass")
156
Brett Cannonf4189912005-04-09 02:30:16 +0000157 def test_class_defs(self):
158 self.check_suite("class foo():pass")
Tim Peterse8906822005-04-20 17:45:13 +0000159
Fred Drake58422e52001-06-04 03:56:24 +0000160 def test_import_from_statement(self):
161 self.check_suite("from sys.path import *")
162 self.check_suite("from sys.path import dirname")
Anthony Baxter1a4ddae2004-08-31 10:07:13 +0000163 self.check_suite("from sys.path import (dirname)")
164 self.check_suite("from sys.path import (dirname,)")
Fred Drake58422e52001-06-04 03:56:24 +0000165 self.check_suite("from sys.path import dirname as my_dirname")
Anthony Baxter1a4ddae2004-08-31 10:07:13 +0000166 self.check_suite("from sys.path import (dirname as my_dirname)")
167 self.check_suite("from sys.path import (dirname as my_dirname,)")
Fred Drake58422e52001-06-04 03:56:24 +0000168 self.check_suite("from sys.path import dirname, basename")
Anthony Baxter1a4ddae2004-08-31 10:07:13 +0000169 self.check_suite("from sys.path import (dirname, basename)")
170 self.check_suite("from sys.path import (dirname, basename,)")
Fred Drake58422e52001-06-04 03:56:24 +0000171 self.check_suite(
172 "from sys.path import dirname as my_dirname, basename")
173 self.check_suite(
Anthony Baxter1a4ddae2004-08-31 10:07:13 +0000174 "from sys.path import (dirname as my_dirname, basename)")
175 self.check_suite(
176 "from sys.path import (dirname as my_dirname, basename,)")
177 self.check_suite(
Fred Drake58422e52001-06-04 03:56:24 +0000178 "from sys.path import dirname, basename as my_basename")
Anthony Baxter1a4ddae2004-08-31 10:07:13 +0000179 self.check_suite(
180 "from sys.path import (dirname, basename as my_basename)")
181 self.check_suite(
182 "from sys.path import (dirname, basename as my_basename,)")
Benjamin Petersoneeed0c72008-11-03 15:18:30 +0000183 self.check_suite("from .bogus import x")
Fred Drakee3fb18c2001-01-07 06:02:19 +0000184
Fred Drake58422e52001-06-04 03:56:24 +0000185 def test_basic_import_statement(self):
186 self.check_suite("import sys")
187 self.check_suite("import sys as system")
188 self.check_suite("import sys, math")
189 self.check_suite("import sys as system, math")
190 self.check_suite("import sys, math as my_math")
Fred Drake79ca79d2000-08-21 22:30:53 +0000191
Neal Norwitz9caf9c02003-02-10 01:54:06 +0000192 def test_pep263(self):
193 self.check_suite("# -*- coding: iso-8859-1 -*-\n"
194 "pass\n")
195
196 def test_assert(self):
197 self.check_suite("assert alo < ahi and blo < bhi\n")
198
Benjamin Peterson9e233a52008-11-24 04:19:49 +0000199 def test_with(self):
200 self.check_suite("with open('x'): pass\n")
201 self.check_suite("with open('x') as f: pass\n")
202
Jeremy Hylton60e96f62006-08-22 20:46:00 +0000203 def test_position(self):
204 # An absolutely minimal test of position information. Better
205 # tests would be a big project.
206 code = "def f(x):\n return x + 1\n"
207 st1 = parser.suite(code)
208 st2 = st1.totuple(line_info=1, col_info=1)
209
210 def walk(tree):
211 node_type = tree[0]
212 next = tree[1]
213 if isinstance(next, tuple):
214 for elt in tree[1:]:
215 for x in walk(elt):
216 yield x
217 else:
218 yield tree
Tim Peters147f9ae2006-08-25 22:05:39 +0000219
Jeremy Hylton60e96f62006-08-22 20:46:00 +0000220 terminals = list(walk(st2))
221 self.assertEqual([
222 (1, 'def', 1, 0),
223 (1, 'f', 1, 4),
224 (7, '(', 1, 5),
225 (1, 'x', 1, 6),
226 (8, ')', 1, 7),
227 (11, ':', 1, 8),
228 (4, '', 1, 9),
229 (5, '', 2, -1),
230 (1, 'return', 2, 4),
231 (1, 'x', 2, 11),
232 (14, '+', 2, 13),
233 (2, '1', 2, 15),
234 (4, '', 2, 16),
235 (6, '', 2, -1),
236 (4, '', 2, -1),
237 (0, '', 2, -1)],
238 terminals)
239
240
Fred Drake79ca79d2000-08-21 22:30:53 +0000241#
242# Second, we take *invalid* trees and make sure we get ParserError
243# rejections for them.
244#
245
Fred Drake58422e52001-06-04 03:56:24 +0000246class IllegalSyntaxTestCase(unittest.TestCase):
Guido van Rossum32c2ae72002-08-22 19:45:32 +0000247
Fred Drake58422e52001-06-04 03:56:24 +0000248 def check_bad_tree(self, tree, label):
249 try:
Fred Drake6e4f2c02001-07-17 19:33:25 +0000250 parser.sequence2st(tree)
Fred Drake58422e52001-06-04 03:56:24 +0000251 except parser.ParserError:
252 pass
253 else:
254 self.fail("did not detect invalid tree for %r" % label)
Fred Drake79ca79d2000-08-21 22:30:53 +0000255
Fred Drake58422e52001-06-04 03:56:24 +0000256 def test_junk(self):
257 # not even remotely valid:
258 self.check_bad_tree((1, 2, 3), "<junk>")
259
Fred Drakecf580c72001-07-17 03:01:29 +0000260 def test_illegal_yield_1(self):
Guido van Rossum32c2ae72002-08-22 19:45:32 +0000261 # Illegal yield statement: def f(): return 1; yield 1
Fred Drakecf580c72001-07-17 03:01:29 +0000262 tree = \
263 (257,
264 (264,
265 (285,
266 (259,
267 (1, 'def'),
268 (1, 'f'),
269 (260, (7, '('), (8, ')')),
270 (11, ':'),
271 (291,
272 (4, ''),
273 (5, ''),
274 (264,
275 (265,
276 (266,
277 (272,
278 (275,
279 (1, 'return'),
280 (313,
281 (292,
282 (293,
283 (294,
284 (295,
285 (297,
286 (298,
287 (299,
288 (300,
289 (301,
290 (302, (303, (304, (305, (2, '1')))))))))))))))))),
291 (264,
292 (265,
293 (266,
294 (272,
295 (276,
296 (1, 'yield'),
297 (313,
298 (292,
299 (293,
300 (294,
301 (295,
302 (297,
303 (298,
304 (299,
305 (300,
306 (301,
307 (302,
308 (303, (304, (305, (2, '1')))))))))))))))))),
309 (4, ''))),
310 (6, ''))))),
311 (4, ''),
312 (0, ''))))
313 self.check_bad_tree(tree, "def f():\n return 1\n yield 1")
314
315 def test_illegal_yield_2(self):
Guido van Rossum32c2ae72002-08-22 19:45:32 +0000316 # Illegal return in generator: def f(): return 1; yield 1
Fred Drakecf580c72001-07-17 03:01:29 +0000317 tree = \
318 (257,
319 (264,
320 (265,
321 (266,
322 (278,
323 (1, 'from'),
324 (281, (1, '__future__')),
325 (1, 'import'),
326 (279, (1, 'generators')))),
327 (4, ''))),
328 (264,
329 (285,
330 (259,
331 (1, 'def'),
332 (1, 'f'),
333 (260, (7, '('), (8, ')')),
334 (11, ':'),
335 (291,
336 (4, ''),
337 (5, ''),
338 (264,
339 (265,
340 (266,
341 (272,
342 (275,
343 (1, 'return'),
344 (313,
345 (292,
346 (293,
347 (294,
348 (295,
349 (297,
350 (298,
351 (299,
352 (300,
353 (301,
354 (302, (303, (304, (305, (2, '1')))))))))))))))))),
355 (264,
356 (265,
357 (266,
358 (272,
359 (276,
360 (1, 'yield'),
361 (313,
362 (292,
363 (293,
364 (294,
365 (295,
366 (297,
367 (298,
368 (299,
369 (300,
370 (301,
371 (302,
372 (303, (304, (305, (2, '1')))))))))))))))))),
373 (4, ''))),
374 (6, ''))))),
375 (4, ''),
376 (0, ''))))
377 self.check_bad_tree(tree, "def f():\n return 1\n yield 1")
378
Fred Drake58422e52001-06-04 03:56:24 +0000379 def test_print_chevron_comma(self):
Guido van Rossum32c2ae72002-08-22 19:45:32 +0000380 # Illegal input: print >>fp,
Fred Drake58422e52001-06-04 03:56:24 +0000381 tree = \
382 (257,
383 (264,
384 (265,
385 (266,
386 (268,
387 (1, 'print'),
388 (35, '>>'),
389 (290,
390 (291,
391 (292,
392 (293,
393 (295,
394 (296,
395 (297,
396 (298, (299, (300, (301, (302, (303, (1, 'fp')))))))))))))),
397 (12, ','))),
398 (4, ''))),
399 (0, ''))
400 self.check_bad_tree(tree, "print >>fp,")
401
402 def test_a_comma_comma_c(self):
Guido van Rossum32c2ae72002-08-22 19:45:32 +0000403 # Illegal input: a,,c
Fred Drake58422e52001-06-04 03:56:24 +0000404 tree = \
405 (258,
406 (311,
407 (290,
408 (291,
409 (292,
410 (293,
411 (295,
412 (296,
413 (297,
414 (298, (299, (300, (301, (302, (303, (1, 'a')))))))))))))),
415 (12, ','),
416 (12, ','),
417 (290,
418 (291,
419 (292,
420 (293,
421 (295,
422 (296,
423 (297,
424 (298, (299, (300, (301, (302, (303, (1, 'c'))))))))))))))),
425 (4, ''),
426 (0, ''))
427 self.check_bad_tree(tree, "a,,c")
428
429 def test_illegal_operator(self):
Guido van Rossum32c2ae72002-08-22 19:45:32 +0000430 # Illegal input: a $= b
Fred Drake58422e52001-06-04 03:56:24 +0000431 tree = \
432 (257,
433 (264,
434 (265,
435 (266,
436 (267,
437 (312,
438 (291,
439 (292,
440 (293,
441 (294,
442 (296,
443 (297,
444 (298,
445 (299,
446 (300, (301, (302, (303, (304, (1, 'a'))))))))))))))),
447 (268, (37, '$=')),
448 (312,
449 (291,
450 (292,
451 (293,
452 (294,
453 (296,
454 (297,
455 (298,
456 (299,
457 (300, (301, (302, (303, (304, (1, 'b'))))))))))))))))),
458 (4, ''))),
459 (0, ''))
460 self.check_bad_tree(tree, "a $= b")
Fred Drake79ca79d2000-08-21 22:30:53 +0000461
Neal Norwitz9caf9c02003-02-10 01:54:06 +0000462 def test_malformed_global(self):
463 #doesn't have global keyword in ast
464 tree = (257,
465 (264,
466 (265,
467 (266,
468 (282, (1, 'foo'))), (4, ''))),
469 (4, ''),
Tim Petersf2715e02003-02-19 02:35:07 +0000470 (0, ''))
Neal Norwitz9caf9c02003-02-10 01:54:06 +0000471 self.check_bad_tree(tree, "malformed global ast")
Fred Drake79ca79d2000-08-21 22:30:53 +0000472
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000473
474class CompileTestCase(unittest.TestCase):
475
476 # These tests are very minimal. :-(
477
478 def test_compile_expr(self):
479 st = parser.expr('2 + 3')
480 code = parser.compilest(st)
481 self.assertEquals(eval(code), 5)
482
483 def test_compile_suite(self):
484 st = parser.suite('x = 2; y = x + 3')
485 code = parser.compilest(st)
486 globs = {}
487 exec code in globs
488 self.assertEquals(globs['y'], 5)
489
490 def test_compile_error(self):
491 st = parser.suite('1 = 3 + 4')
492 self.assertRaises(SyntaxError, parser.compilest, st)
493
Guido van Rossumb6ac23c2007-07-18 17:19:14 +0000494 def test_compile_badunicode(self):
495 st = parser.suite('a = u"\U12345678"')
496 self.assertRaises(SyntaxError, parser.compilest, st)
497 st = parser.suite('a = u"\u1"')
498 self.assertRaises(SyntaxError, parser.compilest, st)
499
Facundo Batistafc2d0102008-02-23 12:01:13 +0000500class ParserStackLimitTestCase(unittest.TestCase):
501 """try to push the parser to/over it's limits.
502 see http://bugs.python.org/issue1881 for a discussion
503 """
504 def _nested_expression(self, level):
505 return "["*level+"]"*level
506
507 def test_deeply_nested_list(self):
508 e = self._nested_expression(99)
509 st = parser.expr(e)
510 st.compile()
511
512 def test_trigger_memory_error(self):
513 e = self._nested_expression(100)
Martin v. Löwis66e26632008-03-18 13:16:05 +0000514 print >>sys.stderr, "Expecting 's_push: parser stack overflow' in next line"
Facundo Batistafc2d0102008-02-23 12:01:13 +0000515 self.assertRaises(MemoryError, parser.expr, e)
516
Fred Drake2e2be372001-09-20 21:33:42 +0000517def test_main():
Walter Dörwald21d3a322003-05-01 17:45:56 +0000518 test_support.run_unittest(
519 RoundtripLegalSyntaxTestCase,
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000520 IllegalSyntaxTestCase,
521 CompileTestCase,
Facundo Batistafc2d0102008-02-23 12:01:13 +0000522 ParserStackLimitTestCase,
Walter Dörwald21d3a322003-05-01 17:45:56 +0000523 )
Fred Drake2e2be372001-09-20 21:33:42 +0000524
525
526if __name__ == "__main__":
527 test_main()