blob: 2a3b71fac4a5a00336854213d11ea2e40bec3878 [file] [log] [blame]
Guido van Rossum3bead091992-01-27 17:00:37 +00001# Python test set -- part 1, grammar.
2# This just tests whether the parser accepts them all.
3
Serhiy Storchakae7a4bb52019-02-19 08:30:15 +02004from test.support import check_syntax_error, check_syntax_warning
Yury Selivanov75445082015-05-11 22:57:16 -04005import inspect
Thomas Wouters89f507f2006-12-13 04:49:30 +00006import unittest
Jeremy Hylton7d3dff22001-10-10 01:45:02 +00007import sys
Serhiy Storchakad31e7732018-10-21 10:09:39 +03008import warnings
Thomas Wouters89f507f2006-12-13 04:49:30 +00009# testing import *
10from sys import *
Guido van Rossum3bead091992-01-27 17:00:37 +000011
Yury Selivanovf8cb8a12016-09-08 20:50:03 -070012# different import patterns to check that __annotations__ does not interfere
13# with import machinery
14import test.ann_module as ann_module
15import typing
16from collections import ChainMap
17from test import ann_module2
18import test
19
Brett Cannona721aba2016-09-09 14:57:09 -070020# These are shared with test_tokenize and other test modules.
21#
22# Note: since several test cases filter out floats by looking for "e" and ".",
23# don't add hexadecimal literals that contain "e" or "E".
24VALID_UNDERSCORE_LITERALS = [
25 '0_0_0',
26 '4_2',
27 '1_0000_0000',
28 '0b1001_0100',
29 '0xffff_ffff',
30 '0o5_7_7',
31 '1_00_00.5',
32 '1_00_00.5e5',
33 '1_00_00e5_1',
34 '1e1_0',
35 '.1_4',
36 '.1_4e1',
37 '0b_0',
38 '0x_f',
39 '0o_5',
40 '1_00_00j',
41 '1_00_00.5j',
42 '1_00_00e5_1j',
43 '.1_4j',
44 '(1_2.5+3_3j)',
45 '(.5_6j)',
46]
47INVALID_UNDERSCORE_LITERALS = [
48 # Trailing underscores:
49 '0_',
50 '42_',
51 '1.4j_',
52 '0x_',
53 '0b1_',
54 '0xf_',
55 '0o5_',
56 '0 if 1_Else 1',
57 # Underscores in the base selector:
58 '0_b0',
59 '0_xf',
60 '0_o5',
61 # Old-style octal, still disallowed:
62 '0_7',
63 '09_99',
64 # Multiple consecutive underscores:
65 '4_______2',
66 '0.1__4',
67 '0.1__4j',
68 '0b1001__0100',
69 '0xffff__ffff',
70 '0x___',
71 '0o5__77',
72 '1e1__0',
73 '1e1__0j',
74 # Underscore right before a dot:
75 '1_.4',
76 '1_.4j',
77 # Underscore right after a dot:
78 '1._4',
79 '1._4j',
80 '._5',
81 '._5j',
82 # Underscore right after a sign:
83 '1.0e+_1',
84 '1.0e+_1j',
85 # Underscore right before j:
86 '1.4_j',
87 '1.4e5_j',
88 # Underscore right before e:
89 '1_e1',
90 '1.4_e1',
91 '1.4_e1j',
92 # Underscore right after e:
93 '1e_1',
94 '1.4e_1',
95 '1.4e_1j',
96 # Complex cases with parens:
97 '(1+1.5_j_)',
98 '(1+1.5_j)',
99]
100
Florent Xicluna9b86b9a2010-03-19 19:00:44 +0000101
Thomas Wouters89f507f2006-12-13 04:49:30 +0000102class TokenTests(unittest.TestCase):
Guido van Rossum3bead091992-01-27 17:00:37 +0000103
Serhiy Storchakae7a4bb52019-02-19 08:30:15 +0200104 from test.support import check_syntax_error
Serhiy Storchakacf7303e2018-07-09 15:09:35 +0300105
Benjamin Petersonc8507bf2011-05-30 10:52:48 -0500106 def test_backslash(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +0000107 # Backslash means line continuation:
108 x = 1 \
109 + 1
Florent Xicluna9b86b9a2010-03-19 19:00:44 +0000110 self.assertEqual(x, 2, 'backslash for line continuation')
Guido van Rossum3bead091992-01-27 17:00:37 +0000111
Thomas Wouters89f507f2006-12-13 04:49:30 +0000112 # Backslash does not means continuation in comments :\
113 x = 0
Florent Xicluna9b86b9a2010-03-19 19:00:44 +0000114 self.assertEqual(x, 0, 'backslash ending comment')
Guido van Rossum3bead091992-01-27 17:00:37 +0000115
Benjamin Petersonc8507bf2011-05-30 10:52:48 -0500116 def test_plain_integers(self):
Florent Xicluna9b86b9a2010-03-19 19:00:44 +0000117 self.assertEqual(type(000), type(0))
118 self.assertEqual(0xff, 255)
119 self.assertEqual(0o377, 255)
120 self.assertEqual(2147483647, 0o17777777777)
121 self.assertEqual(0b1001, 9)
Georg Brandlfceab5a2008-01-19 20:08:23 +0000122 # "0x" is not a valid literal
123 self.assertRaises(SyntaxError, eval, "0x")
Christian Heimesa37d4c62007-12-04 23:02:19 +0000124 from sys import maxsize
125 if maxsize == 2147483647:
Florent Xicluna9b86b9a2010-03-19 19:00:44 +0000126 self.assertEqual(-2147483647-1, -0o20000000000)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000127 # XXX -2147483648
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000128 self.assertTrue(0o37777777777 > 0)
129 self.assertTrue(0xffffffff > 0)
130 self.assertTrue(0b1111111111111111111111111111111 > 0)
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000131 for s in ('2147483648', '0o40000000000', '0x100000000',
132 '0b10000000000000000000000000000000'):
Thomas Wouters89f507f2006-12-13 04:49:30 +0000133 try:
134 x = eval(s)
135 except OverflowError:
136 self.fail("OverflowError on huge integer literal %r" % s)
Christian Heimesa37d4c62007-12-04 23:02:19 +0000137 elif maxsize == 9223372036854775807:
Florent Xicluna9b86b9a2010-03-19 19:00:44 +0000138 self.assertEqual(-9223372036854775807-1, -0o1000000000000000000000)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000139 self.assertTrue(0o1777777777777777777777 > 0)
140 self.assertTrue(0xffffffffffffffff > 0)
141 self.assertTrue(0b11111111111111111111111111111111111111111111111111111111111111 > 0)
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000142 for s in '9223372036854775808', '0o2000000000000000000000', \
143 '0x10000000000000000', \
144 '0b100000000000000000000000000000000000000000000000000000000000000':
Thomas Wouters89f507f2006-12-13 04:49:30 +0000145 try:
146 x = eval(s)
147 except OverflowError:
148 self.fail("OverflowError on huge integer literal %r" % s)
149 else:
Christian Heimesa37d4c62007-12-04 23:02:19 +0000150 self.fail('Weird maxsize value %r' % maxsize)
Guido van Rossum3bead091992-01-27 17:00:37 +0000151
Benjamin Petersonc8507bf2011-05-30 10:52:48 -0500152 def test_long_integers(self):
Guido van Rossume2a383d2007-01-15 16:59:06 +0000153 x = 0
Guido van Rossume2a383d2007-01-15 16:59:06 +0000154 x = 0xffffffffffffffff
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000155 x = 0Xffffffffffffffff
156 x = 0o77777777777777777
157 x = 0O77777777777777777
Guido van Rossume2a383d2007-01-15 16:59:06 +0000158 x = 123456789012345678901234567890
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000159 x = 0b100000000000000000000000000000000000000000000000000000000000000000000
160 x = 0B111111111111111111111111111111111111111111111111111111111111111111111
Guido van Rossum3bead091992-01-27 17:00:37 +0000161
Benjamin Petersonc8507bf2011-05-30 10:52:48 -0500162 def test_floats(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +0000163 x = 3.14
164 x = 314.
165 x = 0.314
166 # XXX x = 000.314
167 x = .314
168 x = 3e14
169 x = 3E14
170 x = 3e-14
171 x = 3e+14
172 x = 3.e14
173 x = .3e14
174 x = 3.1e4
Guido van Rossum3bead091992-01-27 17:00:37 +0000175
Benjamin Petersonc4161622014-06-07 12:36:39 -0700176 def test_float_exponent_tokenization(self):
177 # See issue 21642.
178 self.assertEqual(1 if 1else 0, 1)
179 self.assertEqual(1 if 0else 0, 0)
180 self.assertRaises(SyntaxError, eval, "0 if 1Else 0")
181
Brett Cannona721aba2016-09-09 14:57:09 -0700182 def test_underscore_literals(self):
183 for lit in VALID_UNDERSCORE_LITERALS:
184 self.assertEqual(eval(lit), eval(lit.replace('_', '')))
185 for lit in INVALID_UNDERSCORE_LITERALS:
186 self.assertRaises(SyntaxError, eval, lit)
187 # Sanity check: no literal begins with an underscore
188 self.assertRaises(NameError, eval, "_0")
189
Serhiy Storchakacf7303e2018-07-09 15:09:35 +0300190 def test_bad_numerical_literals(self):
191 check = self.check_syntax_error
192 check("0b12", "invalid digit '2' in binary literal")
193 check("0b1_2", "invalid digit '2' in binary literal")
194 check("0b2", "invalid digit '2' in binary literal")
195 check("0b1_", "invalid binary literal")
196 check("0b", "invalid binary literal")
197 check("0o18", "invalid digit '8' in octal literal")
198 check("0o1_8", "invalid digit '8' in octal literal")
199 check("0o8", "invalid digit '8' in octal literal")
200 check("0o1_", "invalid octal literal")
201 check("0o", "invalid octal literal")
202 check("0x1_", "invalid hexadecimal literal")
203 check("0x", "invalid hexadecimal literal")
204 check("1_", "invalid decimal literal")
205 check("012",
206 "leading zeros in decimal integer literals are not permitted; "
207 "use an 0o prefix for octal integers")
208 check("1.2_", "invalid decimal literal")
209 check("1e2_", "invalid decimal literal")
210 check("1e+", "invalid decimal literal")
211
Benjamin Petersonc8507bf2011-05-30 10:52:48 -0500212 def test_string_literals(self):
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000213 x = ''; y = ""; self.assertTrue(len(x) == 0 and x == y)
214 x = '\''; y = "'"; self.assertTrue(len(x) == 1 and x == y and ord(x) == 39)
215 x = '"'; y = "\""; self.assertTrue(len(x) == 1 and x == y and ord(x) == 34)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000216 x = "doesn't \"shrink\" does it"
217 y = 'doesn\'t "shrink" does it'
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000218 self.assertTrue(len(x) == 24 and x == y)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000219 x = "does \"shrink\" doesn't it"
220 y = 'does "shrink" doesn\'t it'
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000221 self.assertTrue(len(x) == 24 and x == y)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000222 x = """
Guido van Rossumb6775db1994-08-01 11:34:53 +0000223The "quick"
224brown fox
225jumps over
226the 'lazy' dog.
227"""
Thomas Wouters89f507f2006-12-13 04:49:30 +0000228 y = '\nThe "quick"\nbrown fox\njumps over\nthe \'lazy\' dog.\n'
Florent Xicluna9b86b9a2010-03-19 19:00:44 +0000229 self.assertEqual(x, y)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000230 y = '''
Guido van Rossumb6775db1994-08-01 11:34:53 +0000231The "quick"
232brown fox
233jumps over
234the 'lazy' dog.
Thomas Wouters89f507f2006-12-13 04:49:30 +0000235'''
Florent Xicluna9b86b9a2010-03-19 19:00:44 +0000236 self.assertEqual(x, y)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000237 y = "\n\
Guido van Rossumb6775db1994-08-01 11:34:53 +0000238The \"quick\"\n\
239brown fox\n\
240jumps over\n\
241the 'lazy' dog.\n\
Thomas Wouters89f507f2006-12-13 04:49:30 +0000242"
Florent Xicluna9b86b9a2010-03-19 19:00:44 +0000243 self.assertEqual(x, y)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000244 y = '\n\
Guido van Rossumb6775db1994-08-01 11:34:53 +0000245The \"quick\"\n\
246brown fox\n\
247jumps over\n\
248the \'lazy\' dog.\n\
Thomas Wouters89f507f2006-12-13 04:49:30 +0000249'
Florent Xicluna9b86b9a2010-03-19 19:00:44 +0000250 self.assertEqual(x, y)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000251
Benjamin Petersonc8507bf2011-05-30 10:52:48 -0500252 def test_ellipsis(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +0000253 x = ...
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000254 self.assertTrue(x is Ellipsis)
Georg Brandldde00282007-03-18 19:01:53 +0000255 self.assertRaises(SyntaxError, eval, ".. .")
Thomas Wouters89f507f2006-12-13 04:49:30 +0000256
Benjamin Peterson758888d2011-05-30 11:12:38 -0500257 def test_eof_error(self):
258 samples = ("def foo(", "\ndef foo(", "def foo(\n")
259 for s in samples:
260 with self.assertRaises(SyntaxError) as cm:
261 compile(s, "<test>", "exec")
262 self.assertIn("unexpected EOF", str(cm.exception))
263
Yury Selivanovf8cb8a12016-09-08 20:50:03 -0700264var_annot_global: int # a global annotated is necessary for test_var_annot
265
266# custom namespace for testing __annotations__
267
268class CNS:
269 def __init__(self):
270 self._dct = {}
271 def __setitem__(self, item, value):
272 self._dct[item.lower()] = value
273 def __getitem__(self, item):
274 return self._dct[item]
275
276
Thomas Wouters89f507f2006-12-13 04:49:30 +0000277class GrammarTests(unittest.TestCase):
278
Serhiy Storchakae7a4bb52019-02-19 08:30:15 +0200279 from test.support import check_syntax_error, check_syntax_warning
Serhiy Storchaka07ca9af2018-02-04 10:53:48 +0200280
Thomas Wouters89f507f2006-12-13 04:49:30 +0000281 # single_input: NEWLINE | simple_stmt | compound_stmt NEWLINE
282 # XXX can't test in a script -- this rule is only used when interactive
283
284 # file_input: (NEWLINE | stmt)* ENDMARKER
285 # Being tested as this very moment this very module
286
287 # expr_input: testlist NEWLINE
288 # XXX Hard to test -- used only in calls to input()
289
Benjamin Petersonc8507bf2011-05-30 10:52:48 -0500290 def test_eval_input(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +0000291 # testlist ENDMARKER
292 x = eval('1, 0 or 1')
293
Yury Selivanovf8cb8a12016-09-08 20:50:03 -0700294 def test_var_annot_basics(self):
295 # all these should be allowed
296 var1: int = 5
297 var2: [int, str]
298 my_lst = [42]
299 def one():
300 return 1
301 int.new_attr: int
302 [list][0]: type
303 my_lst[one()-1]: int = 5
304 self.assertEqual(my_lst, [5])
305
306 def test_var_annot_syntax_errors(self):
307 # parser pass
308 check_syntax_error(self, "def f: int")
309 check_syntax_error(self, "x: int: str")
310 check_syntax_error(self, "def f():\n"
311 " nonlocal x: int\n")
312 # AST pass
313 check_syntax_error(self, "[x, 0]: int\n")
314 check_syntax_error(self, "f(): int\n")
315 check_syntax_error(self, "(x,): int")
316 check_syntax_error(self, "def f():\n"
317 " (x, y): int = (1, 2)\n")
318 # symtable pass
319 check_syntax_error(self, "def f():\n"
320 " x: int\n"
321 " global x\n")
322 check_syntax_error(self, "def f():\n"
323 " global x\n"
324 " x: int\n")
325
326 def test_var_annot_basic_semantics(self):
327 # execution order
328 with self.assertRaises(ZeroDivisionError):
329 no_name[does_not_exist]: no_name_again = 1/0
330 with self.assertRaises(NameError):
331 no_name[does_not_exist]: 1/0 = 0
332 global var_annot_global
333
334 # function semantics
335 def f():
336 st: str = "Hello"
337 a.b: int = (1, 2)
338 return st
339 self.assertEqual(f.__annotations__, {})
340 def f_OK():
341 x: 1/0
342 f_OK()
343 def fbad():
344 x: int
345 print(x)
346 with self.assertRaises(UnboundLocalError):
347 fbad()
348 def f2bad():
349 (no_such_global): int
350 print(no_such_global)
351 try:
352 f2bad()
353 except Exception as e:
354 self.assertIs(type(e), NameError)
355
356 # class semantics
357 class C:
Guido van Rossum015d8742016-09-11 09:45:24 -0700358 __foo: int
Yury Selivanovf8cb8a12016-09-08 20:50:03 -0700359 s: str = "attr"
360 z = 2
361 def __init__(self, x):
362 self.x: int = x
Guido van Rossum015d8742016-09-11 09:45:24 -0700363 self.assertEqual(C.__annotations__, {'_C__foo': int, 's': str})
Yury Selivanovf8cb8a12016-09-08 20:50:03 -0700364 with self.assertRaises(NameError):
365 class CBad:
366 no_such_name_defined.attr: int = 0
367 with self.assertRaises(NameError):
368 class Cbad2(C):
369 x: int
370 x.y: list = []
371
372 def test_var_annot_metaclass_semantics(self):
373 class CMeta(type):
374 @classmethod
375 def __prepare__(metacls, name, bases, **kwds):
376 return {'__annotations__': CNS()}
377 class CC(metaclass=CMeta):
378 XX: 'ANNOT'
379 self.assertEqual(CC.__annotations__['xx'], 'ANNOT')
380
381 def test_var_annot_module_semantics(self):
382 with self.assertRaises(AttributeError):
383 print(test.__annotations__)
384 self.assertEqual(ann_module.__annotations__,
385 {1: 2, 'x': int, 'y': str, 'f': typing.Tuple[int, int]})
386 self.assertEqual(ann_module.M.__annotations__,
387 {'123': 123, 'o': type})
388 self.assertEqual(ann_module2.__annotations__, {})
Yury Selivanovf8cb8a12016-09-08 20:50:03 -0700389
390 def test_var_annot_in_module(self):
391 # check that functions fail the same way when executed
392 # outside of module where they were defined
393 from test.ann_module3 import f_bad_ann, g_bad_ann, D_bad_ann
394 with self.assertRaises(NameError):
395 f_bad_ann()
396 with self.assertRaises(NameError):
397 g_bad_ann()
398 with self.assertRaises(NameError):
399 D_bad_ann(5)
400
401 def test_var_annot_simple_exec(self):
402 gns = {}; lns= {}
403 exec("'docstring'\n"
404 "__annotations__[1] = 2\n"
405 "x: int = 5\n", gns, lns)
406 self.assertEqual(lns["__annotations__"], {1: 2, 'x': int})
407 with self.assertRaises(KeyError):
408 gns['__annotations__']
409
410 def test_var_annot_custom_maps(self):
411 # tests with custom locals() and __annotations__
412 ns = {'__annotations__': CNS()}
413 exec('X: int; Z: str = "Z"; (w): complex = 1j', ns)
414 self.assertEqual(ns['__annotations__']['x'], int)
415 self.assertEqual(ns['__annotations__']['z'], str)
416 with self.assertRaises(KeyError):
417 ns['__annotations__']['w']
418 nonloc_ns = {}
419 class CNS2:
420 def __init__(self):
421 self._dct = {}
422 def __setitem__(self, item, value):
423 nonlocal nonloc_ns
424 self._dct[item] = value
425 nonloc_ns[item] = value
426 def __getitem__(self, item):
427 return self._dct[item]
428 exec('x: int = 1', {}, CNS2())
429 self.assertEqual(nonloc_ns['__annotations__']['x'], int)
430
431 def test_var_annot_refleak(self):
432 # complex case: custom locals plus custom __annotations__
433 # this was causing refleak
434 cns = CNS()
435 nonloc_ns = {'__annotations__': cns}
436 class CNS2:
437 def __init__(self):
438 self._dct = {'__annotations__': cns}
439 def __setitem__(self, item, value):
440 nonlocal nonloc_ns
441 self._dct[item] = value
442 nonloc_ns[item] = value
443 def __getitem__(self, item):
444 return self._dct[item]
445 exec('X: str', {}, CNS2())
446 self.assertEqual(nonloc_ns['__annotations__']['x'], str)
447
Ivan Levkivskyi62c35a82019-01-25 01:39:19 +0000448 def test_var_annot_rhs(self):
449 ns = {}
450 exec('x: tuple = 1, 2', ns)
451 self.assertEqual(ns['x'], (1, 2))
452 stmt = ('def f():\n'
453 ' x: int = yield')
454 exec(stmt, ns)
455 self.assertEqual(list(ns['f']()), [None])
456
Benjamin Petersonc8507bf2011-05-30 10:52:48 -0500457 def test_funcdef(self):
Neal Norwitzc1505362006-12-28 06:47:50 +0000458 ### [decorators] 'def' NAME parameters ['->' test] ':' suite
459 ### decorator: '@' dotted_name [ '(' [arglist] ')' ] NEWLINE
460 ### decorators: decorator+
461 ### parameters: '(' [typedargslist] ')'
462 ### typedargslist: ((tfpdef ['=' test] ',')*
Guido van Rossum1bc535d2007-05-15 18:46:22 +0000463 ### ('*' [tfpdef] (',' tfpdef ['=' test])* [',' '**' tfpdef] | '**' tfpdef)
Neal Norwitzc1505362006-12-28 06:47:50 +0000464 ### | tfpdef ['=' test] (',' tfpdef ['=' test])* [','])
Guido van Rossum1bc535d2007-05-15 18:46:22 +0000465 ### tfpdef: NAME [':' test]
Neal Norwitzc1505362006-12-28 06:47:50 +0000466 ### varargslist: ((vfpdef ['=' test] ',')*
Guido van Rossum1bc535d2007-05-15 18:46:22 +0000467 ### ('*' [vfpdef] (',' vfpdef ['=' test])* [',' '**' vfpdef] | '**' vfpdef)
Neal Norwitzc1505362006-12-28 06:47:50 +0000468 ### | vfpdef ['=' test] (',' vfpdef ['=' test])* [','])
Guido van Rossum1bc535d2007-05-15 18:46:22 +0000469 ### vfpdef: NAME
Thomas Wouters89f507f2006-12-13 04:49:30 +0000470 def f1(): pass
471 f1()
472 f1(*())
473 f1(*(), **{})
474 def f2(one_argument): pass
475 def f3(two, arguments): pass
Florent Xicluna9b86b9a2010-03-19 19:00:44 +0000476 self.assertEqual(f2.__code__.co_varnames, ('one_argument',))
477 self.assertEqual(f3.__code__.co_varnames, ('two', 'arguments'))
Thomas Wouters89f507f2006-12-13 04:49:30 +0000478 def a1(one_arg,): pass
479 def a2(two, args,): pass
480 def v0(*rest): pass
481 def v1(a, *rest): pass
482 def v2(a, b, *rest): pass
Thomas Wouters89f507f2006-12-13 04:49:30 +0000483
484 f1()
485 f2(1)
486 f2(1,)
487 f3(1, 2)
488 f3(1, 2,)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000489 v0()
490 v0(1)
491 v0(1,)
492 v0(1,2)
493 v0(1,2,3,4,5,6,7,8,9,0)
494 v1(1)
495 v1(1,)
496 v1(1,2)
497 v1(1,2,3)
498 v1(1,2,3,4,5,6,7,8,9,0)
499 v2(1,2)
500 v2(1,2,3)
501 v2(1,2,3,4)
502 v2(1,2,3,4,5,6,7,8,9,0)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000503
Thomas Wouters89f507f2006-12-13 04:49:30 +0000504 def d01(a=1): pass
505 d01()
506 d01(1)
507 d01(*(1,))
Yury Selivanov14acf5f2015-08-05 17:54:10 -0400508 d01(*[] or [2])
509 d01(*() or (), *{} and (), **() or {})
Thomas Wouters89f507f2006-12-13 04:49:30 +0000510 d01(**{'a':2})
Benjamin Petersonde12b792015-05-16 09:44:45 -0400511 d01(**{'a':2} or {})
Thomas Wouters89f507f2006-12-13 04:49:30 +0000512 def d11(a, b=1): pass
513 d11(1)
514 d11(1, 2)
515 d11(1, **{'b':2})
516 def d21(a, b, c=1): pass
517 d21(1, 2)
518 d21(1, 2, 3)
519 d21(*(1, 2, 3))
520 d21(1, *(2, 3))
521 d21(1, 2, *(3,))
522 d21(1, 2, **{'c':3})
523 def d02(a=1, b=2): pass
524 d02()
525 d02(1)
526 d02(1, 2)
527 d02(*(1, 2))
528 d02(1, *(2,))
529 d02(1, **{'b':2})
530 d02(**{'a': 1, 'b': 2})
531 def d12(a, b=1, c=2): pass
532 d12(1)
533 d12(1, 2)
534 d12(1, 2, 3)
535 def d22(a, b, c=1, d=2): pass
536 d22(1, 2)
537 d22(1, 2, 3)
538 d22(1, 2, 3, 4)
539 def d01v(a=1, *rest): pass
540 d01v()
541 d01v(1)
542 d01v(1, 2)
543 d01v(*(1, 2, 3, 4))
544 d01v(*(1,))
545 d01v(**{'a':2})
546 def d11v(a, b=1, *rest): pass
547 d11v(1)
548 d11v(1, 2)
549 d11v(1, 2, 3)
550 def d21v(a, b, c=1, *rest): pass
551 d21v(1, 2)
552 d21v(1, 2, 3)
553 d21v(1, 2, 3, 4)
554 d21v(*(1, 2, 3, 4))
555 d21v(1, 2, **{'c': 3})
556 def d02v(a=1, b=2, *rest): pass
557 d02v()
558 d02v(1)
559 d02v(1, 2)
560 d02v(1, 2, 3)
561 d02v(1, *(2, 3, 4))
562 d02v(**{'a': 1, 'b': 2})
563 def d12v(a, b=1, c=2, *rest): pass
564 d12v(1)
565 d12v(1, 2)
566 d12v(1, 2, 3)
567 d12v(1, 2, 3, 4)
568 d12v(*(1, 2, 3, 4))
569 d12v(1, 2, *(3, 4, 5))
570 d12v(1, *(2,), **{'c': 3})
571 def d22v(a, b, c=1, d=2, *rest): pass
572 d22v(1, 2)
573 d22v(1, 2, 3)
574 d22v(1, 2, 3, 4)
575 d22v(1, 2, 3, 4, 5)
576 d22v(*(1, 2, 3, 4))
577 d22v(1, 2, *(3, 4, 5))
578 d22v(1, *(2, 3), **{'d': 4})
Georg Brandld8b690f2008-05-16 17:28:50 +0000579
580 # keyword argument type tests
581 try:
582 str('x', **{b'foo':1 })
583 except TypeError:
584 pass
585 else:
586 self.fail('Bytes should not work as keyword argument names')
Thomas Wouters89f507f2006-12-13 04:49:30 +0000587 # keyword only argument tests
588 def pos0key1(*, key): return key
589 pos0key1(key=100)
590 def pos2key2(p1, p2, *, k1, k2=100): return p1,p2,k1,k2
591 pos2key2(1, 2, k1=100)
592 pos2key2(1, 2, k1=100, k2=200)
593 pos2key2(1, 2, k2=100, k1=200)
594 def pos2key2dict(p1, p2, *, k1=100, k2, **kwarg): return p1,p2,k1,k2,kwarg
595 pos2key2dict(1,2,k2=100,tokwarg1=100,tokwarg2=200)
596 pos2key2dict(1,2,tokwarg1=100,tokwarg2=200, k2=100)
597
Robert Collinsdf395992015-08-12 08:00:06 +1200598 self.assertRaises(SyntaxError, eval, "def f(*): pass")
599 self.assertRaises(SyntaxError, eval, "def f(*,): pass")
600 self.assertRaises(SyntaxError, eval, "def f(*, **kwds): pass")
601
Benjamin Peterson2d735bc2008-08-19 20:57:10 +0000602 # keyword arguments after *arglist
603 def f(*args, **kwargs):
604 return args, kwargs
Florent Xicluna9b86b9a2010-03-19 19:00:44 +0000605 self.assertEqual(f(1, x=2, *[3, 4], y=5), ((1, 3, 4),
Benjamin Peterson2d735bc2008-08-19 20:57:10 +0000606 {'x':2, 'y':5}))
Benjamin Peterson025e9eb2015-05-05 20:16:41 -0400607 self.assertEqual(f(1, *(2,3), 4), ((1, 2, 3, 4), {}))
Benjamin Peterson2d735bc2008-08-19 20:57:10 +0000608 self.assertRaises(SyntaxError, eval, "f(1, x=2, *(3,4), x=5)")
Benjamin Peterson025e9eb2015-05-05 20:16:41 -0400609 self.assertEqual(f(**{'eggs':'scrambled', 'spam':'fried'}),
610 ((), {'eggs':'scrambled', 'spam':'fried'}))
611 self.assertEqual(f(spam='fried', **{'eggs':'scrambled'}),
612 ((), {'eggs':'scrambled', 'spam':'fried'}))
Benjamin Peterson2d735bc2008-08-19 20:57:10 +0000613
Serhiy Storchaka0cc99c82018-01-04 10:36:35 +0200614 # Check ast errors in *args and *kwargs
615 check_syntax_error(self, "f(*g(1=2))")
616 check_syntax_error(self, "f(**g(1=2))")
617
Neal Norwitzc1505362006-12-28 06:47:50 +0000618 # argument annotation tests
619 def f(x) -> list: pass
Florent Xicluna9b86b9a2010-03-19 19:00:44 +0000620 self.assertEqual(f.__annotations__, {'return': list})
Zachary Warece17f762015-08-01 21:55:36 -0500621 def f(x: int): pass
Florent Xicluna9b86b9a2010-03-19 19:00:44 +0000622 self.assertEqual(f.__annotations__, {'x': int})
Pablo Galindoa0c01bf2019-05-31 15:19:50 +0100623 def f(x: int, /): pass
624 self.assertEqual(f.__annotations__, {'x': int})
625 def f(x: int = 34, /): pass
626 self.assertEqual(f.__annotations__, {'x': int})
Zachary Warece17f762015-08-01 21:55:36 -0500627 def f(*x: str): pass
Florent Xicluna9b86b9a2010-03-19 19:00:44 +0000628 self.assertEqual(f.__annotations__, {'x': str})
Zachary Warece17f762015-08-01 21:55:36 -0500629 def f(**x: float): pass
Florent Xicluna9b86b9a2010-03-19 19:00:44 +0000630 self.assertEqual(f.__annotations__, {'x': float})
Zachary Warece17f762015-08-01 21:55:36 -0500631 def f(x, y: 1+2): pass
Florent Xicluna9b86b9a2010-03-19 19:00:44 +0000632 self.assertEqual(f.__annotations__, {'y': 3})
Pablo Galindoa0c01bf2019-05-31 15:19:50 +0100633 def f(x, y: 1+2, /): pass
634 self.assertEqual(f.__annotations__, {'y': 3})
Zachary Warece17f762015-08-01 21:55:36 -0500635 def f(a, b: 1, c: 2, d): pass
Florent Xicluna9b86b9a2010-03-19 19:00:44 +0000636 self.assertEqual(f.__annotations__, {'b': 1, 'c': 2})
Pablo Galindoa0c01bf2019-05-31 15:19:50 +0100637 def f(a, b: 1, /, c: 2, d): pass
638 self.assertEqual(f.__annotations__, {'b': 1, 'c': 2})
Zachary Warece17f762015-08-01 21:55:36 -0500639 def f(a, b: 1, c: 2, d, e: 3 = 4, f=5, *g: 6): pass
Florent Xicluna9b86b9a2010-03-19 19:00:44 +0000640 self.assertEqual(f.__annotations__,
Zachary Warece17f762015-08-01 21:55:36 -0500641 {'b': 1, 'c': 2, 'e': 3, 'g': 6})
642 def f(a, b: 1, c: 2, d, e: 3 = 4, f=5, *g: 6, h: 7, i=8, j: 9 = 10,
643 **k: 11) -> 12: pass
Florent Xicluna9b86b9a2010-03-19 19:00:44 +0000644 self.assertEqual(f.__annotations__,
Zachary Warece17f762015-08-01 21:55:36 -0500645 {'b': 1, 'c': 2, 'e': 3, 'g': 6, 'h': 7, 'j': 9,
646 'k': 11, 'return': 12})
Pablo Galindoa0c01bf2019-05-31 15:19:50 +0100647 def f(a, b: 1, c: 2, d, e: 3 = 4, f: int = 5, /, *g: 6, h: 7, i=8, j: 9 = 10,
648 **k: 11) -> 12: pass
649 self.assertEqual(f.__annotations__,
650 {'b': 1, 'c': 2, 'e': 3, 'f': int, 'g': 6, 'h': 7, 'j': 9,
651 'k': 11, 'return': 12})
Yury Selivanov34ce99f2014-02-18 12:49:41 -0500652 # Check for issue #20625 -- annotations mangling
653 class Spam:
Zachary Warece17f762015-08-01 21:55:36 -0500654 def f(self, *, __kw: 1):
Yury Selivanov34ce99f2014-02-18 12:49:41 -0500655 pass
656 class Ham(Spam): pass
Benjamin Petersonbcfcfc52014-03-09 20:59:24 -0500657 self.assertEqual(Spam.f.__annotations__, {'_Spam__kw': 1})
658 self.assertEqual(Ham.f.__annotations__, {'_Spam__kw': 1})
Nick Coghlan71011e22007-04-23 11:05:01 +0000659 # Check for SF Bug #1697248 - mixing decorators and a return annotation
660 def null(x): return x
661 @null
662 def f(x) -> list: pass
Florent Xicluna9b86b9a2010-03-19 19:00:44 +0000663 self.assertEqual(f.__annotations__, {'return': list})
Nick Coghlan71011e22007-04-23 11:05:01 +0000664
Serhiy Storchaka64204de2016-06-12 17:36:24 +0300665 # test closures with a variety of opargs
Guido van Rossum0240b922007-02-26 21:23:50 +0000666 closure = 1
667 def f(): return closure
668 def f(x=1): return closure
669 def f(*, k=1): return closure
670 def f() -> int: return closure
Neal Norwitzc1505362006-12-28 06:47:50 +0000671
Robert Collinsdf395992015-08-12 08:00:06 +1200672 # Check trailing commas are permitted in funcdef argument list
673 def f(a,): pass
674 def f(*args,): pass
675 def f(**kwds,): pass
676 def f(a, *args,): pass
677 def f(a, **kwds,): pass
678 def f(*args, b,): pass
679 def f(*, b,): pass
680 def f(*args, **kwds,): pass
681 def f(a, *args, b,): pass
682 def f(a, *, b,): pass
683 def f(a, *args, **kwds,): pass
684 def f(*args, b, **kwds,): pass
685 def f(*, b, **kwds,): pass
686 def f(a, *args, b, **kwds,): pass
687 def f(a, *, b, **kwds,): pass
688
Benjamin Petersonc8507bf2011-05-30 10:52:48 -0500689 def test_lambdef(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +0000690 ### lambdef: 'lambda' [varargslist] ':' test
691 l1 = lambda : 0
Florent Xicluna9b86b9a2010-03-19 19:00:44 +0000692 self.assertEqual(l1(), 0)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000693 l2 = lambda : a[d] # XXX just testing the expression
Guido van Rossume2a383d2007-01-15 16:59:06 +0000694 l3 = lambda : [2 < x for x in [-1, 3, 0]]
Florent Xicluna9b86b9a2010-03-19 19:00:44 +0000695 self.assertEqual(l3(), [0, 1, 0])
Thomas Wouters89f507f2006-12-13 04:49:30 +0000696 l4 = lambda x = lambda y = lambda z=1 : z : y() : x()
Florent Xicluna9b86b9a2010-03-19 19:00:44 +0000697 self.assertEqual(l4(), 1)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000698 l5 = lambda x, y, z=2: x + y + z
Florent Xicluna9b86b9a2010-03-19 19:00:44 +0000699 self.assertEqual(l5(1, 2), 5)
700 self.assertEqual(l5(1, 2, 3), 6)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000701 check_syntax_error(self, "lambda x: x = 2")
Amaury Forgeot d'Arc35c86582008-06-17 21:11:29 +0000702 check_syntax_error(self, "lambda (None,): None")
Thomas Wouters89f507f2006-12-13 04:49:30 +0000703 l6 = lambda x, y, *, k=20: x+y+k
Florent Xicluna9b86b9a2010-03-19 19:00:44 +0000704 self.assertEqual(l6(1,2), 1+2+20)
705 self.assertEqual(l6(1,2,k=10), 1+2+10)
Guido van Rossumb31c7f71993-11-11 10:31:23 +0000706
Robert Collinsdf395992015-08-12 08:00:06 +1200707 # check that trailing commas are permitted
708 l10 = lambda a,: 0
709 l11 = lambda *args,: 0
710 l12 = lambda **kwds,: 0
711 l13 = lambda a, *args,: 0
712 l14 = lambda a, **kwds,: 0
713 l15 = lambda *args, b,: 0
714 l16 = lambda *, b,: 0
715 l17 = lambda *args, **kwds,: 0
716 l18 = lambda a, *args, b,: 0
717 l19 = lambda a, *, b,: 0
718 l20 = lambda a, *args, **kwds,: 0
719 l21 = lambda *args, b, **kwds,: 0
720 l22 = lambda *, b, **kwds,: 0
721 l23 = lambda a, *args, b, **kwds,: 0
722 l24 = lambda a, *, b, **kwds,: 0
723
Guido van Rossumb31c7f71993-11-11 10:31:23 +0000724
Thomas Wouters89f507f2006-12-13 04:49:30 +0000725 ### stmt: simple_stmt | compound_stmt
726 # Tested below
Georg Brandl52318d62006-09-06 07:06:08 +0000727
Benjamin Petersonc8507bf2011-05-30 10:52:48 -0500728 def test_simple_stmt(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +0000729 ### simple_stmt: small_stmt (';' small_stmt)* [';']
730 x = 1; pass; del x
731 def foo():
Ezio Melotti13925002011-03-16 11:05:33 +0200732 # verify statements that end with semi-colons
Thomas Wouters89f507f2006-12-13 04:49:30 +0000733 x = 1; pass; del x;
734 foo()
Georg Brandl52318d62006-09-06 07:06:08 +0000735
Guido van Rossumd8faa362007-04-27 19:54:29 +0000736 ### small_stmt: expr_stmt | pass_stmt | del_stmt | flow_stmt | import_stmt | global_stmt | access_stmt
Thomas Wouters89f507f2006-12-13 04:49:30 +0000737 # Tested below
Georg Brandl52318d62006-09-06 07:06:08 +0000738
Benjamin Petersonc8507bf2011-05-30 10:52:48 -0500739 def test_expr_stmt(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +0000740 # (exprlist '=')* exprlist
Victor Stinner15a30952016-02-08 22:45:06 +0100741 1
Thomas Wouters89f507f2006-12-13 04:49:30 +0000742 1, 2, 3
743 x = 1
744 x = 1, 2, 3
745 x = y = z = 1, 2, 3
746 x, y, z = 1, 2, 3
747 abc = a, b, c = x, y, z = xyz = 1, 2, (3, 4)
Guido van Rossum3bead091992-01-27 17:00:37 +0000748
Thomas Wouters89f507f2006-12-13 04:49:30 +0000749 check_syntax_error(self, "x + 1 = 1")
750 check_syntax_error(self, "a + 1 = b + 2")
Guido van Rossum3bead091992-01-27 17:00:37 +0000751
Nick Coghlan5b1fdc12014-06-16 19:48:02 +1000752 # Check the heuristic for print & exec covers significant cases
753 # As well as placing some limits on false positives
754 def test_former_statements_refer_to_builtins(self):
755 keywords = "print", "exec"
756 # Cases where we want the custom error
757 cases = [
758 "{} foo",
759 "{} {{1:foo}}",
760 "if 1: {} foo",
761 "if 1: {} {{1:foo}}",
762 "if 1:\n {} foo",
763 "if 1:\n {} {{1:foo}}",
764 ]
765 for keyword in keywords:
766 custom_msg = "call to '{}'".format(keyword)
767 for case in cases:
768 source = case.format(keyword)
769 with self.subTest(source=source):
770 with self.assertRaisesRegex(SyntaxError, custom_msg):
771 exec(source)
772 source = source.replace("foo", "(foo.)")
773 with self.subTest(source=source):
774 with self.assertRaisesRegex(SyntaxError, "invalid syntax"):
775 exec(source)
776
Benjamin Petersonc8507bf2011-05-30 10:52:48 -0500777 def test_del_stmt(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +0000778 # 'del' exprlist
779 abc = [1,2,3]
780 x, y, z = abc
781 xyz = x, y, z
Barry Warsaw7e3e1c12000-10-11 21:26:03 +0000782
Thomas Wouters89f507f2006-12-13 04:49:30 +0000783 del abc
784 del x, y, (z, xyz)
Barry Warsaw9182b452000-08-29 04:57:10 +0000785
Benjamin Petersonc8507bf2011-05-30 10:52:48 -0500786 def test_pass_stmt(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +0000787 # 'pass'
788 pass
Barry Warsaw9182b452000-08-29 04:57:10 +0000789
Thomas Wouters89f507f2006-12-13 04:49:30 +0000790 # flow_stmt: break_stmt | continue_stmt | return_stmt | raise_stmt
791 # Tested below
Barry Warsaw9182b452000-08-29 04:57:10 +0000792
Benjamin Petersonc8507bf2011-05-30 10:52:48 -0500793 def test_break_stmt(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +0000794 # 'break'
795 while 1: break
Barry Warsaw9182b452000-08-29 04:57:10 +0000796
Benjamin Petersonc8507bf2011-05-30 10:52:48 -0500797 def test_continue_stmt(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +0000798 # 'continue'
799 i = 1
800 while i: i = 0; continue
Barry Warsaw9182b452000-08-29 04:57:10 +0000801
Thomas Wouters89f507f2006-12-13 04:49:30 +0000802 msg = ""
803 while not msg:
804 msg = "ok"
805 try:
806 continue
807 msg = "continue failed to continue inside try"
808 except:
809 msg = "continue inside try called except block"
810 if msg != "ok":
811 self.fail(msg)
Barry Warsawefc92ee2000-08-21 15:46:50 +0000812
Thomas Wouters89f507f2006-12-13 04:49:30 +0000813 msg = ""
814 while not msg:
815 msg = "finally block not called"
816 try:
817 continue
818 finally:
819 msg = "ok"
820 if msg != "ok":
821 self.fail(msg)
Guido van Rossum3bead091992-01-27 17:00:37 +0000822
Thomas Wouters89f507f2006-12-13 04:49:30 +0000823 def test_break_continue_loop(self):
824 # This test warrants an explanation. It is a test specifically for SF bugs
825 # #463359 and #462937. The bug is that a 'break' statement executed or
826 # exception raised inside a try/except inside a loop, *after* a continue
827 # statement has been executed in that loop, will cause the wrong number of
828 # arguments to be popped off the stack and the instruction pointer reset to
829 # a very small number (usually 0.) Because of this, the following test
830 # *must* written as a function, and the tracking vars *must* be function
831 # arguments with default values. Otherwise, the test will loop and loop.
Guido van Rossum3bead091992-01-27 17:00:37 +0000832
Thomas Wouters89f507f2006-12-13 04:49:30 +0000833 def test_inner(extra_burning_oil = 1, count=0):
834 big_hippo = 2
835 while big_hippo:
836 count += 1
837 try:
838 if extra_burning_oil and big_hippo == 1:
839 extra_burning_oil -= 1
840 break
841 big_hippo -= 1
842 continue
843 except:
844 raise
845 if count > 2 or big_hippo != 1:
846 self.fail("continue then break in try/except in loop broken!")
847 test_inner()
Guido van Rossum3bead091992-01-27 17:00:37 +0000848
Benjamin Petersonc8507bf2011-05-30 10:52:48 -0500849 def test_return(self):
David Cuthbertfd97d1f2018-09-21 18:31:15 -0700850 # 'return' [testlist_star_expr]
Thomas Wouters89f507f2006-12-13 04:49:30 +0000851 def g1(): return
852 def g2(): return 1
David Cuthbertfd97d1f2018-09-21 18:31:15 -0700853 def g3():
854 z = [2, 3]
855 return 1, *z
856
Thomas Wouters89f507f2006-12-13 04:49:30 +0000857 g1()
858 x = g2()
David Cuthbertfd97d1f2018-09-21 18:31:15 -0700859 y = g3()
860 self.assertEqual(y, (1, 2, 3), "unparenthesized star expr return")
Thomas Wouters89f507f2006-12-13 04:49:30 +0000861 check_syntax_error(self, "class foo:return 1")
Guido van Rossum3bead091992-01-27 17:00:37 +0000862
Serhiy Storchaka7cc42c32018-01-02 02:38:35 +0200863 def test_break_in_finally(self):
864 count = 0
865 while count < 2:
866 count += 1
867 try:
868 pass
869 finally:
870 break
871 self.assertEqual(count, 1)
872
873 count = 0
874 while count < 2:
875 count += 1
876 try:
877 continue
878 finally:
879 break
880 self.assertEqual(count, 1)
881
882 count = 0
883 while count < 2:
884 count += 1
885 try:
886 1/0
887 finally:
888 break
889 self.assertEqual(count, 1)
890
891 for count in [0, 1]:
892 self.assertEqual(count, 0)
893 try:
894 pass
895 finally:
896 break
897 self.assertEqual(count, 0)
898
899 for count in [0, 1]:
900 self.assertEqual(count, 0)
901 try:
902 continue
903 finally:
904 break
905 self.assertEqual(count, 0)
906
907 for count in [0, 1]:
908 self.assertEqual(count, 0)
909 try:
910 1/0
911 finally:
912 break
913 self.assertEqual(count, 0)
914
Serhiy Storchakafe2bbb12018-03-18 09:56:52 +0200915 def test_continue_in_finally(self):
916 count = 0
917 while count < 2:
918 count += 1
919 try:
920 pass
921 finally:
922 continue
923 break
924 self.assertEqual(count, 2)
925
926 count = 0
927 while count < 2:
928 count += 1
929 try:
930 break
931 finally:
932 continue
933 self.assertEqual(count, 2)
934
935 count = 0
936 while count < 2:
937 count += 1
938 try:
939 1/0
940 finally:
941 continue
942 break
943 self.assertEqual(count, 2)
944
945 for count in [0, 1]:
946 try:
947 pass
948 finally:
949 continue
950 break
951 self.assertEqual(count, 1)
952
953 for count in [0, 1]:
954 try:
955 break
956 finally:
957 continue
958 self.assertEqual(count, 1)
959
960 for count in [0, 1]:
961 try:
962 1/0
963 finally:
964 continue
965 break
966 self.assertEqual(count, 1)
967
Serhiy Storchaka7cc42c32018-01-02 02:38:35 +0200968 def test_return_in_finally(self):
969 def g1():
970 try:
971 pass
972 finally:
973 return 1
974 self.assertEqual(g1(), 1)
975
976 def g2():
977 try:
978 return 2
979 finally:
980 return 3
981 self.assertEqual(g2(), 3)
982
983 def g3():
984 try:
985 1/0
986 finally:
987 return 4
988 self.assertEqual(g3(), 4)
989
Benjamin Petersonc8507bf2011-05-30 10:52:48 -0500990 def test_yield(self):
Nick Coghlan1f7ce622012-01-13 21:43:40 +1000991 # Allowed as standalone statement
992 def g(): yield 1
993 def g(): yield from ()
994 # Allowed as RHS of assignment
995 def g(): x = yield 1
996 def g(): x = yield from ()
997 # Ordinary yield accepts implicit tuples
998 def g(): yield 1, 1
999 def g(): x = yield 1, 1
1000 # 'yield from' does not
1001 check_syntax_error(self, "def g(): yield from (), 1")
1002 check_syntax_error(self, "def g(): x = yield from (), 1")
1003 # Requires parentheses as subexpression
1004 def g(): 1, (yield 1)
1005 def g(): 1, (yield from ())
1006 check_syntax_error(self, "def g(): 1, yield 1")
1007 check_syntax_error(self, "def g(): 1, yield from ()")
1008 # Requires parentheses as call argument
1009 def g(): f((yield 1))
1010 def g(): f((yield 1), 1)
1011 def g(): f((yield from ()))
1012 def g(): f((yield from ()), 1)
David Cuthbertfd97d1f2018-09-21 18:31:15 -07001013 # Do not require parenthesis for tuple unpacking
1014 def g(): rest = 4, 5, 6; yield 1, 2, 3, *rest
Serhiy Storchaka4642d5f2018-10-05 21:09:56 +03001015 self.assertEqual(list(g()), [(1, 2, 3, 4, 5, 6)])
Nick Coghlan1f7ce622012-01-13 21:43:40 +10001016 check_syntax_error(self, "def g(): f(yield 1)")
1017 check_syntax_error(self, "def g(): f(yield 1, 1)")
1018 check_syntax_error(self, "def g(): f(yield from ())")
1019 check_syntax_error(self, "def g(): f(yield from (), 1)")
1020 # Not allowed at top level
1021 check_syntax_error(self, "yield")
1022 check_syntax_error(self, "yield from")
1023 # Not allowed at class scope
Thomas Wouters89f507f2006-12-13 04:49:30 +00001024 check_syntax_error(self, "class foo:yield 1")
Nick Coghlan1f7ce622012-01-13 21:43:40 +10001025 check_syntax_error(self, "class foo:yield from ()")
Yury Selivanovf315c1c2015-07-23 09:10:44 +03001026 # Check annotation refleak on SyntaxError
1027 check_syntax_error(self, "def g(a:(yield)): pass")
Guido van Rossum3bead091992-01-27 17:00:37 +00001028
Serhiy Storchaka73a7e9b2017-12-01 06:54:17 +02001029 def test_yield_in_comprehensions(self):
1030 # Check yield in comprehensions
1031 def g(): [x for x in [(yield 1)]]
1032 def g(): [x for x in [(yield from ())]]
1033
Serhiy Storchaka07ca9af2018-02-04 10:53:48 +02001034 check = self.check_syntax_error
Serhiy Storchaka73a7e9b2017-12-01 06:54:17 +02001035 check("def g(): [(yield x) for x in ()]",
1036 "'yield' inside list comprehension")
1037 check("def g(): [x for x in () if not (yield x)]",
1038 "'yield' inside list comprehension")
1039 check("def g(): [y for x in () for y in [(yield x)]]",
1040 "'yield' inside list comprehension")
1041 check("def g(): {(yield x) for x in ()}",
1042 "'yield' inside set comprehension")
1043 check("def g(): {(yield x): x for x in ()}",
1044 "'yield' inside dict comprehension")
1045 check("def g(): {x: (yield x) for x in ()}",
1046 "'yield' inside dict comprehension")
1047 check("def g(): ((yield x) for x in ())",
1048 "'yield' inside generator expression")
1049 check("def g(): [(yield from x) for x in ()]",
1050 "'yield' inside list comprehension")
1051 check("class C: [(yield x) for x in ()]",
1052 "'yield' inside list comprehension")
1053 check("[(yield x) for x in ()]",
1054 "'yield' inside list comprehension")
1055
Benjamin Petersonc8507bf2011-05-30 10:52:48 -05001056 def test_raise(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +00001057 # 'raise' test [',' test]
Collin Winter828f04a2007-08-31 00:04:24 +00001058 try: raise RuntimeError('just testing')
Thomas Wouters89f507f2006-12-13 04:49:30 +00001059 except RuntimeError: pass
1060 try: raise KeyboardInterrupt
1061 except KeyboardInterrupt: pass
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00001062
Benjamin Petersonc8507bf2011-05-30 10:52:48 -05001063 def test_import(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +00001064 # 'import' dotted_as_names
1065 import sys
1066 import time, sys
1067 # 'from' dotted_name 'import' ('*' | '(' import_as_names ')' | import_as_names)
1068 from time import time
1069 from time import (time)
1070 # not testable inside a function, but already done at top of the module
1071 # from sys import *
1072 from sys import path, argv
1073 from sys import (path, argv)
1074 from sys import (path, argv,)
Tim Peters10fb3862001-02-09 20:17:14 +00001075
Benjamin Petersonc8507bf2011-05-30 10:52:48 -05001076 def test_global(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +00001077 # 'global' NAME (',' NAME)*
1078 global a
1079 global a, b
1080 global one, two, three, four, five, six, seven, eight, nine, ten
Thomas Wouters80d373c2001-09-26 12:43:39 +00001081
Benjamin Petersonc8507bf2011-05-30 10:52:48 -05001082 def test_nonlocal(self):
Benjamin Petersona933e522008-10-24 22:16:39 +00001083 # 'nonlocal' NAME (',' NAME)*
1084 x = 0
1085 y = 0
1086 def f():
1087 nonlocal x
1088 nonlocal x, y
1089
Benjamin Petersonc8507bf2011-05-30 10:52:48 -05001090 def test_assert(self):
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001091 # assertTruestmt: 'assert' test [',' test]
Thomas Wouters89f507f2006-12-13 04:49:30 +00001092 assert 1
1093 assert 1, 1
1094 assert lambda x:x
1095 assert 1, lambda x:x+1
Ezio Melotti6cc5bf72011-12-02 18:22:52 +02001096
1097 try:
1098 assert True
1099 except AssertionError as e:
1100 self.fail("'assert True' should not have raised an AssertionError")
1101
1102 try:
1103 assert True, 'this should always pass'
1104 except AssertionError as e:
1105 self.fail("'assert True, msg' should not have "
1106 "raised an AssertionError")
1107
1108 # these tests fail if python is run with -O, so check __debug__
1109 @unittest.skipUnless(__debug__, "Won't work if __debug__ is False")
1110 def testAssert2(self):
Thomas Wouters80d373c2001-09-26 12:43:39 +00001111 try:
Thomas Wouters89f507f2006-12-13 04:49:30 +00001112 assert 0, "msg"
Guido van Rossumb940e112007-01-10 16:19:56 +00001113 except AssertionError as e:
Florent Xicluna9b86b9a2010-03-19 19:00:44 +00001114 self.assertEqual(e.args[0], "msg")
Thomas Wouters89f507f2006-12-13 04:49:30 +00001115 else:
Ezio Melotti6cc5bf72011-12-02 18:22:52 +02001116 self.fail("AssertionError not raised by assert 0")
1117
1118 try:
1119 assert False
1120 except AssertionError as e:
1121 self.assertEqual(len(e.args), 0)
1122 else:
1123 self.fail("AssertionError not raised by 'assert False'")
1124
Serhiy Storchakae7a4bb52019-02-19 08:30:15 +02001125 self.check_syntax_warning('assert(x, "msg")',
1126 'assertion is always true')
Serhiy Storchakad31e7732018-10-21 10:09:39 +03001127 with warnings.catch_warnings():
Serhiy Storchakae7a4bb52019-02-19 08:30:15 +02001128 warnings.simplefilter('error', SyntaxWarning)
Serhiy Storchakad31e7732018-10-21 10:09:39 +03001129 compile('assert x, "msg"', '<testcase>', 'exec')
1130
Thomas Wouters80d373c2001-09-26 12:43:39 +00001131
Thomas Wouters89f507f2006-12-13 04:49:30 +00001132 ### compound_stmt: if_stmt | while_stmt | for_stmt | try_stmt | funcdef | classdef
1133 # Tested below
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001134
Benjamin Petersonc8507bf2011-05-30 10:52:48 -05001135 def test_if(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +00001136 # 'if' test ':' suite ('elif' test ':' suite)* ['else' ':' suite]
1137 if 1: pass
1138 if 1: pass
1139 else: pass
1140 if 0: pass
1141 elif 0: pass
1142 if 0: pass
1143 elif 0: pass
1144 elif 0: pass
1145 elif 0: pass
1146 else: pass
Guido van Rossum3bead091992-01-27 17:00:37 +00001147
Benjamin Petersonc8507bf2011-05-30 10:52:48 -05001148 def test_while(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +00001149 # 'while' test ':' suite ['else' ':' suite]
1150 while 0: pass
1151 while 0: pass
1152 else: pass
Guido van Rossum3bead091992-01-27 17:00:37 +00001153
Christian Heimes969fe572008-01-25 11:23:10 +00001154 # Issue1920: "while 0" is optimized away,
1155 # ensure that the "else" clause is still present.
1156 x = 0
1157 while 0:
1158 x = 1
1159 else:
1160 x = 2
Florent Xicluna9b86b9a2010-03-19 19:00:44 +00001161 self.assertEqual(x, 2)
Christian Heimes969fe572008-01-25 11:23:10 +00001162
Benjamin Petersonc8507bf2011-05-30 10:52:48 -05001163 def test_for(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +00001164 # 'for' exprlist 'in' exprlist ':' suite ['else' ':' suite]
1165 for i in 1, 2, 3: pass
1166 for i, j, k in (): pass
1167 else: pass
1168 class Squares:
1169 def __init__(self, max):
1170 self.max = max
1171 self.sofar = []
1172 def __len__(self): return len(self.sofar)
1173 def __getitem__(self, i):
1174 if not 0 <= i < self.max: raise IndexError
1175 n = len(self.sofar)
1176 while n <= i:
1177 self.sofar.append(n*n)
1178 n = n+1
1179 return self.sofar[i]
1180 n = 0
1181 for x in Squares(10): n = n+x
1182 if n != 285:
1183 self.fail('for over growing sequence')
Guido van Rossum3bead091992-01-27 17:00:37 +00001184
Thomas Wouters89f507f2006-12-13 04:49:30 +00001185 result = []
1186 for x, in [(1,), (2,), (3,)]:
1187 result.append(x)
1188 self.assertEqual(result, [1, 2, 3])
Guido van Rossum3bead091992-01-27 17:00:37 +00001189
Benjamin Petersonc8507bf2011-05-30 10:52:48 -05001190 def test_try(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +00001191 ### try_stmt: 'try' ':' suite (except_clause ':' suite)+ ['else' ':' suite]
1192 ### | 'try' ':' suite 'finally' ':' suite
Guido van Rossumb940e112007-01-10 16:19:56 +00001193 ### except_clause: 'except' [expr ['as' expr]]
Thomas Wouters89f507f2006-12-13 04:49:30 +00001194 try:
1195 1/0
1196 except ZeroDivisionError:
1197 pass
1198 else:
1199 pass
1200 try: 1/0
1201 except EOFError: pass
Guido van Rossumb940e112007-01-10 16:19:56 +00001202 except TypeError as msg: pass
Thomas Wouters89f507f2006-12-13 04:49:30 +00001203 except: pass
1204 else: pass
1205 try: 1/0
1206 except (EOFError, TypeError, ZeroDivisionError): pass
1207 try: 1/0
Guido van Rossumb940e112007-01-10 16:19:56 +00001208 except (EOFError, TypeError, ZeroDivisionError) as msg: pass
Thomas Wouters89f507f2006-12-13 04:49:30 +00001209 try: pass
1210 finally: pass
Jeremy Hyltonf828e2d2001-02-19 15:54:52 +00001211
Benjamin Petersonc8507bf2011-05-30 10:52:48 -05001212 def test_suite(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +00001213 # simple_stmt | NEWLINE INDENT NEWLINE* (stmt NEWLINE*)+ DEDENT
1214 if 1: pass
1215 if 1:
1216 pass
1217 if 1:
1218 #
1219 #
1220 #
1221 pass
1222 pass
1223 #
1224 pass
1225 #
Guido van Rossum3bead091992-01-27 17:00:37 +00001226
Benjamin Petersonc8507bf2011-05-30 10:52:48 -05001227 def test_test(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +00001228 ### and_test ('or' and_test)*
1229 ### and_test: not_test ('and' not_test)*
1230 ### not_test: 'not' not_test | comparison
1231 if not 1: pass
1232 if 1 and 1: pass
1233 if 1 or 1: pass
1234 if not not not 1: pass
1235 if not 1 and 1 and 1: pass
1236 if 1 and 1 or 1 and 1 and 1 or not 1 and 1: pass
Guido van Rossum3bead091992-01-27 17:00:37 +00001237
Benjamin Petersonc8507bf2011-05-30 10:52:48 -05001238 def test_comparison(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +00001239 ### comparison: expr (comp_op expr)*
1240 ### comp_op: '<'|'>'|'=='|'>='|'<='|'!='|'in'|'not' 'in'|'is'|'is' 'not'
1241 if 1: pass
1242 x = (1 == 1)
1243 if 1 == 1: pass
1244 if 1 != 1: pass
1245 if 1 < 1: pass
1246 if 1 > 1: pass
1247 if 1 <= 1: pass
1248 if 1 >= 1: pass
Serhiy Storchaka3bcbedc2019-01-18 07:47:48 +02001249 if x is x: pass
1250 if x is not x: pass
Thomas Wouters89f507f2006-12-13 04:49:30 +00001251 if 1 in (): pass
1252 if 1 not in (): pass
Serhiy Storchaka3bcbedc2019-01-18 07:47:48 +02001253 if 1 < 1 > 1 == 1 >= 1 <= 1 != 1 in 1 not in x is x is not x: pass
1254
1255 def test_comparison_is_literal(self):
1256 def check(test, msg='"is" with a literal'):
Serhiy Storchakae7a4bb52019-02-19 08:30:15 +02001257 self.check_syntax_warning(test, msg)
Serhiy Storchaka3bcbedc2019-01-18 07:47:48 +02001258
1259 check('x is 1')
1260 check('x is "thing"')
1261 check('1 is x')
1262 check('x is y is 1')
1263 check('x is not 1', '"is not" with a literal')
1264
1265 with warnings.catch_warnings():
Serhiy Storchakae7a4bb52019-02-19 08:30:15 +02001266 warnings.simplefilter('error', SyntaxWarning)
Serhiy Storchaka3bcbedc2019-01-18 07:47:48 +02001267 compile('x is None', '<testcase>', 'exec')
1268 compile('x is False', '<testcase>', 'exec')
1269 compile('x is True', '<testcase>', 'exec')
1270 compile('x is ...', '<testcase>', 'exec')
Guido van Rossum3bead091992-01-27 17:00:37 +00001271
Serhiy Storchaka62e44812019-02-16 08:12:19 +02001272 def test_warn_missed_comma(self):
1273 def check(test):
Serhiy Storchakae7a4bb52019-02-19 08:30:15 +02001274 self.check_syntax_warning(test, msg)
Serhiy Storchaka62e44812019-02-16 08:12:19 +02001275
1276 msg=r'is not callable; perhaps you missed a comma\?'
1277 check('[(1, 2) (3, 4)]')
1278 check('[(x, y) (3, 4)]')
1279 check('[[1, 2] (3, 4)]')
1280 check('[{1, 2} (3, 4)]')
1281 check('[{1: 2} (3, 4)]')
1282 check('[[i for i in range(5)] (3, 4)]')
1283 check('[{i for i in range(5)} (3, 4)]')
1284 check('[(i for i in range(5)) (3, 4)]')
1285 check('[{i: i for i in range(5)} (3, 4)]')
1286 check('[f"{x}" (3, 4)]')
1287 check('[f"x={x}" (3, 4)]')
1288 check('["abc" (3, 4)]')
1289 check('[b"abc" (3, 4)]')
1290 check('[123 (3, 4)]')
1291 check('[12.3 (3, 4)]')
1292 check('[12.3j (3, 4)]')
1293 check('[None (3, 4)]')
1294 check('[True (3, 4)]')
1295 check('[... (3, 4)]')
1296
1297 msg=r'is not subscriptable; perhaps you missed a comma\?'
1298 check('[{1, 2} [i, j]]')
1299 check('[{i for i in range(5)} [i, j]]')
1300 check('[(i for i in range(5)) [i, j]]')
1301 check('[(lambda x, y: x) [i, j]]')
1302 check('[123 [i, j]]')
1303 check('[12.3 [i, j]]')
1304 check('[12.3j [i, j]]')
1305 check('[None [i, j]]')
1306 check('[True [i, j]]')
1307 check('[... [i, j]]')
1308
1309 msg=r'indices must be integers or slices, not tuple; perhaps you missed a comma\?'
1310 check('[(1, 2) [i, j]]')
1311 check('[(x, y) [i, j]]')
1312 check('[[1, 2] [i, j]]')
1313 check('[[i for i in range(5)] [i, j]]')
1314 check('[f"{x}" [i, j]]')
1315 check('[f"x={x}" [i, j]]')
1316 check('["abc" [i, j]]')
1317 check('[b"abc" [i, j]]')
1318
1319 msg=r'indices must be integers or slices, not tuple;'
1320 check('[[1, 2] [3, 4]]')
1321 msg=r'indices must be integers or slices, not list;'
1322 check('[[1, 2] [[3, 4]]]')
1323 check('[[1, 2] [[i for i in range(5)]]]')
1324 msg=r'indices must be integers or slices, not set;'
1325 check('[[1, 2] [{3, 4}]]')
1326 check('[[1, 2] [{i for i in range(5)}]]')
1327 msg=r'indices must be integers or slices, not dict;'
1328 check('[[1, 2] [{3: 4}]]')
1329 check('[[1, 2] [{i: i for i in range(5)}]]')
1330 msg=r'indices must be integers or slices, not generator;'
1331 check('[[1, 2] [(i for i in range(5))]]')
1332 msg=r'indices must be integers or slices, not function;'
1333 check('[[1, 2] [(lambda x, y: x)]]')
1334 msg=r'indices must be integers or slices, not str;'
1335 check('[[1, 2] [f"{x}"]]')
1336 check('[[1, 2] [f"x={x}"]]')
1337 check('[[1, 2] ["abc"]]')
1338 msg=r'indices must be integers or slices, not'
1339 check('[[1, 2] [b"abc"]]')
1340 check('[[1, 2] [12.3]]')
1341 check('[[1, 2] [12.3j]]')
1342 check('[[1, 2] [None]]')
1343 check('[[1, 2] [...]]')
1344
1345 with warnings.catch_warnings():
Serhiy Storchakae7a4bb52019-02-19 08:30:15 +02001346 warnings.simplefilter('error', SyntaxWarning)
Serhiy Storchaka62e44812019-02-16 08:12:19 +02001347 compile('[(lambda x, y: x) (3, 4)]', '<testcase>', 'exec')
1348 compile('[[1, 2] [i]]', '<testcase>', 'exec')
1349 compile('[[1, 2] [0]]', '<testcase>', 'exec')
1350 compile('[[1, 2] [True]]', '<testcase>', 'exec')
1351 compile('[[1, 2] [1:2]]', '<testcase>', 'exec')
1352 compile('[{(1, 2): 3} [i, j]]', '<testcase>', 'exec')
1353
Benjamin Petersonc8507bf2011-05-30 10:52:48 -05001354 def test_binary_mask_ops(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +00001355 x = 1 & 1
1356 x = 1 ^ 1
1357 x = 1 | 1
Guido van Rossum3bead091992-01-27 17:00:37 +00001358
Benjamin Petersonc8507bf2011-05-30 10:52:48 -05001359 def test_shift_ops(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +00001360 x = 1 << 1
1361 x = 1 >> 1
1362 x = 1 << 1 >> 1
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001363
Benjamin Petersonc8507bf2011-05-30 10:52:48 -05001364 def test_additive_ops(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +00001365 x = 1
1366 x = 1 + 1
1367 x = 1 - 1 - 1
1368 x = 1 - 1 + 1 - 1 + 1
Guido van Rossum3bead091992-01-27 17:00:37 +00001369
Benjamin Petersonc8507bf2011-05-30 10:52:48 -05001370 def test_multiplicative_ops(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +00001371 x = 1 * 1
1372 x = 1 / 1
1373 x = 1 % 1
1374 x = 1 / 1 * 1 % 1
Guido van Rossum3bead091992-01-27 17:00:37 +00001375
Benjamin Petersonc8507bf2011-05-30 10:52:48 -05001376 def test_unary_ops(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +00001377 x = +1
1378 x = -1
1379 x = ~1
1380 x = ~1 ^ 1 & 1 | 1 & 1 ^ -1
1381 x = -1*1/1 + 1*1 - ---1*1
Guido van Rossum3bead091992-01-27 17:00:37 +00001382
Benjamin Petersonc8507bf2011-05-30 10:52:48 -05001383 def test_selectors(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +00001384 ### trailer: '(' [testlist] ')' | '[' subscript ']' | '.' NAME
1385 ### subscript: expr | [expr] ':' [expr]
Guido van Rossum3bead091992-01-27 17:00:37 +00001386
Thomas Wouters89f507f2006-12-13 04:49:30 +00001387 import sys, time
1388 c = sys.path[0]
1389 x = time.time()
1390 x = sys.modules['time'].time()
1391 a = '01234'
1392 c = a[0]
1393 c = a[-1]
1394 s = a[0:5]
1395 s = a[:5]
1396 s = a[0:]
1397 s = a[:]
1398 s = a[-5:]
1399 s = a[:-1]
1400 s = a[-4:-3]
1401 # A rough test of SF bug 1333982. http://python.org/sf/1333982
1402 # The testing here is fairly incomplete.
1403 # Test cases should include: commas with 1 and 2 colons
1404 d = {}
1405 d[1] = 1
1406 d[1,] = 2
1407 d[1,2] = 3
1408 d[1,2,3] = 4
1409 L = list(d)
Serhiy Storchaka0cc99c82018-01-04 10:36:35 +02001410 L.sort(key=lambda x: (type(x).__name__, x))
Florent Xicluna9b86b9a2010-03-19 19:00:44 +00001411 self.assertEqual(str(L), '[1, (1,), (1, 2), (1, 2, 3)]')
Guido van Rossum3bead091992-01-27 17:00:37 +00001412
Benjamin Petersonc8507bf2011-05-30 10:52:48 -05001413 def test_atoms(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +00001414 ### atom: '(' [testlist] ')' | '[' [testlist] ']' | '{' [dictsetmaker] '}' | NAME | NUMBER | STRING
1415 ### dictsetmaker: (test ':' test (',' test ':' test)* [',']) | (test (',' test)* [','])
Guido van Rossum3bead091992-01-27 17:00:37 +00001416
Thomas Wouters89f507f2006-12-13 04:49:30 +00001417 x = (1)
1418 x = (1 or 2 or 3)
1419 x = (1 or 2 or 3, 2, 3)
Guido van Rossum3bead091992-01-27 17:00:37 +00001420
Thomas Wouters89f507f2006-12-13 04:49:30 +00001421 x = []
1422 x = [1]
1423 x = [1 or 2 or 3]
1424 x = [1 or 2 or 3, 2, 3]
1425 x = []
Guido van Rossum3bead091992-01-27 17:00:37 +00001426
Thomas Wouters89f507f2006-12-13 04:49:30 +00001427 x = {}
1428 x = {'one': 1}
1429 x = {'one': 1,}
1430 x = {'one' or 'two': 1 or 2}
1431 x = {'one': 1, 'two': 2}
1432 x = {'one': 1, 'two': 2,}
1433 x = {'one': 1, 'two': 2, 'three': 3, 'four': 4, 'five': 5, 'six': 6}
Guido van Rossum3bead091992-01-27 17:00:37 +00001434
Thomas Wouters89f507f2006-12-13 04:49:30 +00001435 x = {'one'}
1436 x = {'one', 1,}
1437 x = {'one', 'two', 'three'}
1438 x = {2, 3, 4,}
1439
1440 x = x
1441 x = 'x'
1442 x = 123
1443
1444 ### exprlist: expr (',' expr)* [',']
1445 ### testlist: test (',' test)* [',']
1446 # These have been exercised enough above
1447
Benjamin Petersonc8507bf2011-05-30 10:52:48 -05001448 def test_classdef(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +00001449 # 'class' NAME ['(' [testlist] ')'] ':' suite
1450 class B: pass
1451 class B2(): pass
1452 class C1(B): pass
1453 class C2(B): pass
1454 class D(C1, C2, B): pass
1455 class C:
1456 def meth1(self): pass
1457 def meth2(self, arg): pass
1458 def meth3(self, a1, a2): pass
1459
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001460 # decorator: '@' dotted_name [ '(' [arglist] ')' ] NEWLINE
1461 # decorators: decorator+
1462 # decorated: decorators (classdef | funcdef)
1463 def class_decorator(x): return x
1464 @class_decorator
1465 class G: pass
1466
Benjamin Petersonc8507bf2011-05-30 10:52:48 -05001467 def test_dictcomps(self):
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001468 # dictorsetmaker: ( (test ':' test (comp_for |
1469 # (',' test ':' test)* [','])) |
1470 # (test (comp_for | (',' test)* [','])) )
1471 nums = [1, 2, 3]
1472 self.assertEqual({i:i+1 for i in nums}, {1: 2, 2: 3, 3: 4})
1473
Benjamin Petersonc8507bf2011-05-30 10:52:48 -05001474 def test_listcomps(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +00001475 # list comprehension tests
1476 nums = [1, 2, 3, 4, 5]
1477 strs = ["Apple", "Banana", "Coconut"]
1478 spcs = [" Apple", " Banana ", "Coco nut "]
1479
1480 self.assertEqual([s.strip() for s in spcs], ['Apple', 'Banana', 'Coco nut'])
1481 self.assertEqual([3 * x for x in nums], [3, 6, 9, 12, 15])
1482 self.assertEqual([x for x in nums if x > 2], [3, 4, 5])
1483 self.assertEqual([(i, s) for i in nums for s in strs],
1484 [(1, 'Apple'), (1, 'Banana'), (1, 'Coconut'),
1485 (2, 'Apple'), (2, 'Banana'), (2, 'Coconut'),
1486 (3, 'Apple'), (3, 'Banana'), (3, 'Coconut'),
1487 (4, 'Apple'), (4, 'Banana'), (4, 'Coconut'),
1488 (5, 'Apple'), (5, 'Banana'), (5, 'Coconut')])
1489 self.assertEqual([(i, s) for i in nums for s in [f for f in strs if "n" in f]],
1490 [(1, 'Banana'), (1, 'Coconut'), (2, 'Banana'), (2, 'Coconut'),
1491 (3, 'Banana'), (3, 'Coconut'), (4, 'Banana'), (4, 'Coconut'),
1492 (5, 'Banana'), (5, 'Coconut')])
1493 self.assertEqual([(lambda a:[a**i for i in range(a+1)])(j) for j in range(5)],
1494 [[1], [1, 1], [1, 2, 4], [1, 3, 9, 27], [1, 4, 16, 64, 256]])
1495
1496 def test_in_func(l):
1497 return [0 < x < 3 for x in l if x > 2]
1498
1499 self.assertEqual(test_in_func(nums), [False, False, False])
1500
1501 def test_nested_front():
1502 self.assertEqual([[y for y in [x, x + 1]] for x in [1,3,5]],
1503 [[1, 2], [3, 4], [5, 6]])
1504
1505 test_nested_front()
1506
1507 check_syntax_error(self, "[i, s for i in nums for s in strs]")
1508 check_syntax_error(self, "[x if y]")
1509
1510 suppliers = [
1511 (1, "Boeing"),
1512 (2, "Ford"),
1513 (3, "Macdonalds")
1514 ]
1515
1516 parts = [
1517 (10, "Airliner"),
1518 (20, "Engine"),
1519 (30, "Cheeseburger")
1520 ]
1521
1522 suppart = [
1523 (1, 10), (1, 20), (2, 20), (3, 30)
1524 ]
1525
1526 x = [
1527 (sname, pname)
1528 for (sno, sname) in suppliers
1529 for (pno, pname) in parts
1530 for (sp_sno, sp_pno) in suppart
1531 if sno == sp_sno and pno == sp_pno
1532 ]
1533
1534 self.assertEqual(x, [('Boeing', 'Airliner'), ('Boeing', 'Engine'), ('Ford', 'Engine'),
1535 ('Macdonalds', 'Cheeseburger')])
1536
Benjamin Petersonc8507bf2011-05-30 10:52:48 -05001537 def test_genexps(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +00001538 # generator expression tests
1539 g = ([x for x in range(10)] for x in range(1))
Georg Brandla18af4e2007-04-21 15:47:16 +00001540 self.assertEqual(next(g), [x for x in range(10)])
Thomas Wouters89f507f2006-12-13 04:49:30 +00001541 try:
Georg Brandla18af4e2007-04-21 15:47:16 +00001542 next(g)
Thomas Wouters89f507f2006-12-13 04:49:30 +00001543 self.fail('should produce StopIteration exception')
1544 except StopIteration:
1545 pass
1546
1547 a = 1
1548 try:
1549 g = (a for d in a)
Georg Brandla18af4e2007-04-21 15:47:16 +00001550 next(g)
Thomas Wouters89f507f2006-12-13 04:49:30 +00001551 self.fail('should produce TypeError')
1552 except TypeError:
1553 pass
1554
1555 self.assertEqual(list((x, y) for x in 'abcd' for y in 'abcd'), [(x, y) for x in 'abcd' for y in 'abcd'])
1556 self.assertEqual(list((x, y) for x in 'ab' for y in 'xy'), [(x, y) for x in 'ab' for y in 'xy'])
1557
1558 a = [x for x in range(10)]
1559 b = (x for x in (y for y in a))
1560 self.assertEqual(sum(b), sum([x for x in range(10)]))
1561
1562 self.assertEqual(sum(x**2 for x in range(10)), sum([x**2 for x in range(10)]))
1563 self.assertEqual(sum(x*x for x in range(10) if x%2), sum([x*x for x in range(10) if x%2]))
1564 self.assertEqual(sum(x for x in (y for y in range(10))), sum([x for x in range(10)]))
1565 self.assertEqual(sum(x for x in (y for y in (z for z in range(10)))), sum([x for x in range(10)]))
1566 self.assertEqual(sum(x for x in [y for y in (z for z in range(10))]), sum([x for x in range(10)]))
1567 self.assertEqual(sum(x for x in (y for y in (z for z in range(10) if True)) if True), sum([x for x in range(10)]))
1568 self.assertEqual(sum(x for x in (y for y in (z for z in range(10) if True) if False) if True), 0)
1569 check_syntax_error(self, "foo(x for x in range(10), 100)")
1570 check_syntax_error(self, "foo(100, x for x in range(10))")
1571
Benjamin Petersonc8507bf2011-05-30 10:52:48 -05001572 def test_comprehension_specials(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +00001573 # test for outmost iterable precomputation
1574 x = 10; g = (i for i in range(x)); x = 5
1575 self.assertEqual(len(list(g)), 10)
1576
1577 # This should hold, since we're only precomputing outmost iterable.
1578 x = 10; t = False; g = ((i,j) for i in range(x) if t for j in range(x))
1579 x = 5; t = True;
1580 self.assertEqual([(i,j) for i in range(10) for j in range(5)], list(g))
1581
1582 # Grammar allows multiple adjacent 'if's in listcomps and genexps,
1583 # even though it's silly. Make sure it works (ifelse broke this.)
1584 self.assertEqual([ x for x in range(10) if x % 2 if x % 3 ], [1, 5, 7])
1585 self.assertEqual(list(x for x in range(10) if x % 2 if x % 3), [1, 5, 7])
1586
1587 # verify unpacking single element tuples in listcomp/genexp.
1588 self.assertEqual([x for x, in [(4,), (5,), (6,)]], [4, 5, 6])
1589 self.assertEqual(list(x for x, in [(7,), (8,), (9,)]), [7, 8, 9])
1590
Benjamin Petersonf17ab892009-05-29 21:55:57 +00001591 def test_with_statement(self):
1592 class manager(object):
1593 def __enter__(self):
1594 return (1, 2)
1595 def __exit__(self, *args):
1596 pass
1597
1598 with manager():
1599 pass
1600 with manager() as x:
1601 pass
1602 with manager() as (x, y):
1603 pass
1604 with manager(), manager():
1605 pass
1606 with manager() as x, manager() as y:
1607 pass
1608 with manager() as x, manager():
1609 pass
1610
Benjamin Petersonc8507bf2011-05-30 10:52:48 -05001611 def test_if_else_expr(self):
Thomas Wouters89f507f2006-12-13 04:49:30 +00001612 # Test ifelse expressions in various cases
1613 def _checkeval(msg, ret):
1614 "helper to check that evaluation of expressions is done correctly"
Victor Stinnerc6ec54d2016-04-12 18:33:41 +02001615 print(msg)
Thomas Wouters89f507f2006-12-13 04:49:30 +00001616 return ret
1617
Nick Coghlan650f0d02007-04-15 12:05:43 +00001618 # the next line is not allowed anymore
1619 #self.assertEqual([ x() for x in lambda: True, lambda: False if x() ], [True])
Thomas Wouters89f507f2006-12-13 04:49:30 +00001620 self.assertEqual([ x() for x in (lambda: True, lambda: False) if x() ], [True])
1621 self.assertEqual([ x(False) for x in (lambda x: False if x else True, lambda x: True if x else False) if x(False) ], [True])
1622 self.assertEqual((5 if 1 else _checkeval("check 1", 0)), 5)
1623 self.assertEqual((_checkeval("check 2", 0) if 0 else 5), 5)
1624 self.assertEqual((5 and 6 if 0 else 1), 1)
1625 self.assertEqual(((5 and 6) if 0 else 1), 1)
1626 self.assertEqual((5 and (6 if 1 else 1)), 6)
1627 self.assertEqual((0 or _checkeval("check 3", 2) if 0 else 3), 3)
1628 self.assertEqual((1 or _checkeval("check 4", 2) if 1 else _checkeval("check 5", 3)), 1)
1629 self.assertEqual((0 or 5 if 1 else _checkeval("check 6", 3)), 5)
1630 self.assertEqual((not 5 if 1 else 1), False)
1631 self.assertEqual((not 5 if 0 else 1), 1)
1632 self.assertEqual((6 + 1 if 1 else 2), 7)
1633 self.assertEqual((6 - 1 if 1 else 2), 5)
1634 self.assertEqual((6 * 2 if 1 else 4), 12)
1635 self.assertEqual((6 / 2 if 1 else 3), 3)
1636 self.assertEqual((6 < 4 if 0 else 2), 2)
Jeremy Hylton7b03bad2006-02-28 17:46:23 +00001637
Benjamin Petersonf6489f92009-11-25 17:46:26 +00001638 def test_paren_evaluation(self):
1639 self.assertEqual(16 // (4 // 2), 8)
1640 self.assertEqual((16 // 4) // 2, 2)
1641 self.assertEqual(16 // 4 // 2, 2)
Serhiy Storchaka3bcbedc2019-01-18 07:47:48 +02001642 x = 2
1643 y = 3
1644 self.assertTrue(False is (x is y))
1645 self.assertFalse((False is x) is y)
1646 self.assertFalse(False is x is y)
Benjamin Petersonf6489f92009-11-25 17:46:26 +00001647
Benjamin Petersond51374e2014-04-09 23:55:56 -04001648 def test_matrix_mul(self):
1649 # This is not intended to be a comprehensive test, rather just to be few
1650 # samples of the @ operator in test_grammar.py.
1651 class M:
1652 def __matmul__(self, o):
1653 return 4
1654 def __imatmul__(self, o):
1655 self.other = o
1656 return self
1657 m = M()
1658 self.assertEqual(m @ m, 4)
1659 m @= 42
1660 self.assertEqual(m.other, 42)
1661
Yury Selivanov75445082015-05-11 22:57:16 -04001662 def test_async_await(self):
Yury Selivanov75445082015-05-11 22:57:16 -04001663 async def test():
1664 def sum():
Yury Selivanov8fb307c2015-07-22 13:33:45 +03001665 pass
Yury Selivanov75445082015-05-11 22:57:16 -04001666 if 1:
1667 await someobj()
1668
1669 self.assertEqual(test.__name__, 'test')
1670 self.assertTrue(bool(test.__code__.co_flags & inspect.CO_COROUTINE))
1671
1672 def decorator(func):
1673 setattr(func, '_marked', True)
1674 return func
1675
1676 @decorator
1677 async def test2():
1678 return 22
1679 self.assertTrue(test2._marked)
1680 self.assertEqual(test2.__name__, 'test2')
1681 self.assertTrue(bool(test2.__code__.co_flags & inspect.CO_COROUTINE))
1682
1683 def test_async_for(self):
1684 class Done(Exception): pass
1685
1686 class AIter:
Yury Selivanova6f6edb2016-06-09 15:08:31 -04001687 def __aiter__(self):
Yury Selivanov75445082015-05-11 22:57:16 -04001688 return self
1689 async def __anext__(self):
1690 raise StopAsyncIteration
1691
1692 async def foo():
1693 async for i in AIter():
1694 pass
1695 async for i, j in AIter():
1696 pass
1697 async for i in AIter():
1698 pass
1699 else:
1700 pass
1701 raise Done
1702
1703 with self.assertRaises(Done):
1704 foo().send(None)
1705
1706 def test_async_with(self):
1707 class Done(Exception): pass
1708
1709 class manager:
1710 async def __aenter__(self):
1711 return (1, 2)
1712 async def __aexit__(self, *exc):
1713 return False
1714
1715 async def foo():
1716 async with manager():
1717 pass
1718 async with manager() as x:
1719 pass
1720 async with manager() as (x, y):
1721 pass
1722 async with manager(), manager():
1723 pass
1724 async with manager() as x, manager() as y:
1725 pass
1726 async with manager() as x, manager():
1727 pass
1728 raise Done
1729
1730 with self.assertRaises(Done):
1731 foo().send(None)
1732
Guido van Rossum3bead091992-01-27 17:00:37 +00001733
Thomas Wouters89f507f2006-12-13 04:49:30 +00001734if __name__ == '__main__':
Zachary Ware38c707e2015-04-13 15:00:43 -05001735 unittest.main()