718 lines
23 KiB
Python
718 lines
23 KiB
Python
"""Test suite for 2to3's parser and grammar files.
|
||
|
||
This is the place to add tests for changes to 2to3's grammar, such as those
|
||
merging the grammars for Python 2 and 3. In addition to specific tests for
|
||
parts of the grammar we've changed, we also make sure we can parse the
|
||
test_grammar.py files from both Python 2 and Python 3.
|
||
"""
|
||
|
||
# Testing imports
|
||
from . import support
|
||
from .support import driver, driver_no_print_statement
|
||
|
||
# Python imports
|
||
import difflib
|
||
import importlib
|
||
import operator
|
||
import os
|
||
import pickle
|
||
import shutil
|
||
import subprocess
|
||
import sys
|
||
import tempfile
|
||
import test.support
|
||
import unittest
|
||
|
||
# Local imports
|
||
from lib2to3.pgen2 import driver as pgen2_driver
|
||
from lib2to3.pgen2 import tokenize
|
||
from ..pgen2.parse import ParseError
|
||
from lib2to3.pygram import python_symbols as syms
|
||
|
||
|
||
class TestDriver(support.TestCase):
|
||
|
||
def test_formfeed(self):
|
||
s = """print 1\n\x0Cprint 2\n"""
|
||
t = driver.parse_string(s)
|
||
self.assertEqual(t.children[0].children[0].type, syms.print_stmt)
|
||
self.assertEqual(t.children[1].children[0].type, syms.print_stmt)
|
||
|
||
|
||
class TestPgen2Caching(support.TestCase):
|
||
def test_load_grammar_from_txt_file(self):
|
||
pgen2_driver.load_grammar(support.grammar_path, save=False, force=True)
|
||
|
||
def test_load_grammar_from_pickle(self):
|
||
# Make a copy of the grammar file in a temp directory we are
|
||
# guaranteed to be able to write to.
|
||
tmpdir = tempfile.mkdtemp()
|
||
try:
|
||
grammar_copy = os.path.join(
|
||
tmpdir, os.path.basename(support.grammar_path))
|
||
shutil.copy(support.grammar_path, grammar_copy)
|
||
pickle_name = pgen2_driver._generate_pickle_name(grammar_copy)
|
||
|
||
pgen2_driver.load_grammar(grammar_copy, save=True, force=True)
|
||
self.assertTrue(os.path.exists(pickle_name))
|
||
|
||
os.unlink(grammar_copy) # Only the pickle remains...
|
||
pgen2_driver.load_grammar(grammar_copy, save=False, force=False)
|
||
finally:
|
||
shutil.rmtree(tmpdir)
|
||
|
||
@unittest.skipIf(sys.executable is None, 'sys.executable required')
|
||
def test_load_grammar_from_subprocess(self):
|
||
tmpdir = tempfile.mkdtemp()
|
||
tmpsubdir = os.path.join(tmpdir, 'subdir')
|
||
try:
|
||
os.mkdir(tmpsubdir)
|
||
grammar_base = os.path.basename(support.grammar_path)
|
||
grammar_copy = os.path.join(tmpdir, grammar_base)
|
||
grammar_sub_copy = os.path.join(tmpsubdir, grammar_base)
|
||
shutil.copy(support.grammar_path, grammar_copy)
|
||
shutil.copy(support.grammar_path, grammar_sub_copy)
|
||
pickle_name = pgen2_driver._generate_pickle_name(grammar_copy)
|
||
pickle_sub_name = pgen2_driver._generate_pickle_name(
|
||
grammar_sub_copy)
|
||
self.assertNotEqual(pickle_name, pickle_sub_name)
|
||
|
||
# Generate a pickle file from this process.
|
||
pgen2_driver.load_grammar(grammar_copy, save=True, force=True)
|
||
self.assertTrue(os.path.exists(pickle_name))
|
||
|
||
# Generate a new pickle file in a subprocess with a most likely
|
||
# different hash randomization seed.
|
||
sub_env = dict(os.environ)
|
||
sub_env['PYTHONHASHSEED'] = 'random'
|
||
code = """
|
||
from lib2to3.pgen2 import driver as pgen2_driver
|
||
pgen2_driver.load_grammar(%r, save=True, force=True)
|
||
""" % (grammar_sub_copy,)
|
||
msg = ("lib2to3 package is deprecated and may not be able "
|
||
"to parse Python 3.10+")
|
||
cmd = [sys.executable,
|
||
f'-Wignore:{msg}:PendingDeprecationWarning',
|
||
'-c', code]
|
||
subprocess.check_call( cmd, env=sub_env)
|
||
self.assertTrue(os.path.exists(pickle_sub_name))
|
||
|
||
with open(pickle_name, 'rb') as pickle_f_1, \
|
||
open(pickle_sub_name, 'rb') as pickle_f_2:
|
||
self.assertEqual(
|
||
pickle_f_1.read(), pickle_f_2.read(),
|
||
msg='Grammar caches generated using different hash seeds'
|
||
' were not identical.')
|
||
finally:
|
||
shutil.rmtree(tmpdir)
|
||
|
||
def test_load_packaged_grammar(self):
|
||
modname = __name__ + '.load_test'
|
||
class MyLoader:
|
||
def get_data(self, where):
|
||
return pickle.dumps({'elephant': 19})
|
||
class MyModule:
|
||
__file__ = 'parsertestmodule'
|
||
__spec__ = importlib.util.spec_from_loader(modname, MyLoader())
|
||
sys.modules[modname] = MyModule()
|
||
self.addCleanup(operator.delitem, sys.modules, modname)
|
||
g = pgen2_driver.load_packaged_grammar(modname, 'Grammar.txt')
|
||
self.assertEqual(g.elephant, 19)
|
||
|
||
|
||
class GrammarTest(support.TestCase):
|
||
def validate(self, code):
|
||
support.parse_string(code)
|
||
|
||
def invalid_syntax(self, code):
|
||
try:
|
||
self.validate(code)
|
||
except ParseError:
|
||
pass
|
||
else:
|
||
raise AssertionError("Syntax shouldn't have been valid")
|
||
|
||
|
||
class TestMatrixMultiplication(GrammarTest):
|
||
def test_matrix_multiplication_operator(self):
|
||
self.validate("a @ b")
|
||
self.validate("a @= b")
|
||
|
||
|
||
class TestYieldFrom(GrammarTest):
|
||
def test_yield_from(self):
|
||
self.validate("yield from x")
|
||
self.validate("(yield from x) + y")
|
||
self.invalid_syntax("yield from")
|
||
|
||
|
||
class TestAsyncAwait(GrammarTest):
|
||
def test_await_expr(self):
|
||
self.validate("""async def foo():
|
||
await x
|
||
""")
|
||
|
||
self.validate("""async def foo():
|
||
[i async for i in b]
|
||
""")
|
||
|
||
self.validate("""async def foo():
|
||
{i for i in b
|
||
async for i in a if await i
|
||
for b in i}
|
||
""")
|
||
|
||
self.validate("""async def foo():
|
||
[await i for i in b if await c]
|
||
""")
|
||
|
||
self.validate("""async def foo():
|
||
[ i for i in b if c]
|
||
""")
|
||
|
||
self.validate("""async def foo():
|
||
|
||
def foo(): pass
|
||
|
||
def foo(): pass
|
||
|
||
await x
|
||
""")
|
||
|
||
self.validate("""async def foo(): return await a""")
|
||
|
||
self.validate("""def foo():
|
||
def foo(): pass
|
||
async def foo(): await x
|
||
""")
|
||
|
||
self.invalid_syntax("await x")
|
||
self.invalid_syntax("""def foo():
|
||
await x""")
|
||
|
||
self.invalid_syntax("""def foo():
|
||
def foo(): pass
|
||
async def foo(): pass
|
||
await x
|
||
""")
|
||
|
||
def test_async_var(self):
|
||
self.validate("""async = 1""")
|
||
self.validate("""await = 1""")
|
||
self.validate("""def async(): pass""")
|
||
|
||
def test_async_for(self):
|
||
self.validate("""async def foo():
|
||
async for a in b: pass""")
|
||
|
||
def test_async_with(self):
|
||
self.validate("""async def foo():
|
||
async with a: pass""")
|
||
|
||
self.invalid_syntax("""def foo():
|
||
async with a: pass""")
|
||
|
||
def test_async_generator(self):
|
||
self.validate(
|
||
"""async def foo():
|
||
return (i * 2 async for i in arange(42))"""
|
||
)
|
||
self.validate(
|
||
"""def foo():
|
||
return (i * 2 async for i in arange(42))"""
|
||
)
|
||
|
||
|
||
class TestRaiseChanges(GrammarTest):
|
||
def test_2x_style_1(self):
|
||
self.validate("raise")
|
||
|
||
def test_2x_style_2(self):
|
||
self.validate("raise E, V")
|
||
|
||
def test_2x_style_3(self):
|
||
self.validate("raise E, V, T")
|
||
|
||
def test_2x_style_invalid_1(self):
|
||
self.invalid_syntax("raise E, V, T, Z")
|
||
|
||
def test_3x_style(self):
|
||
self.validate("raise E1 from E2")
|
||
|
||
def test_3x_style_invalid_1(self):
|
||
self.invalid_syntax("raise E, V from E1")
|
||
|
||
def test_3x_style_invalid_2(self):
|
||
self.invalid_syntax("raise E from E1, E2")
|
||
|
||
def test_3x_style_invalid_3(self):
|
||
self.invalid_syntax("raise from E1, E2")
|
||
|
||
def test_3x_style_invalid_4(self):
|
||
self.invalid_syntax("raise E from")
|
||
|
||
|
||
# Modelled after Lib/test/test_grammar.py:TokenTests.test_funcdef issue2292
|
||
# and Lib/test/text_parser.py test_list_displays, test_set_displays,
|
||
# test_dict_displays, test_argument_unpacking, ... changes.
|
||
class TestUnpackingGeneralizations(GrammarTest):
|
||
def test_mid_positional_star(self):
|
||
self.validate("""func(1, *(2, 3), 4)""")
|
||
|
||
def test_double_star_dict_literal(self):
|
||
self.validate("""func(**{'eggs':'scrambled', 'spam':'fried'})""")
|
||
|
||
def test_double_star_dict_literal_after_keywords(self):
|
||
self.validate("""func(spam='fried', **{'eggs':'scrambled'})""")
|
||
|
||
def test_double_star_expression(self):
|
||
self.validate("""func(**{'a':2} or {})""")
|
||
self.validate("""func(**() or {})""")
|
||
|
||
def test_star_expression(self):
|
||
self.validate("""func(*[] or [2])""")
|
||
|
||
def test_list_display(self):
|
||
self.validate("""[*{2}, 3, *[4]]""")
|
||
|
||
def test_set_display(self):
|
||
self.validate("""{*{2}, 3, *[4]}""")
|
||
|
||
def test_dict_display_1(self):
|
||
self.validate("""{**{}}""")
|
||
|
||
def test_dict_display_2(self):
|
||
self.validate("""{**{}, 3:4, **{5:6, 7:8}}""")
|
||
|
||
def test_complex_star_expression(self):
|
||
self.validate("func(* [] or [1])")
|
||
|
||
def test_complex_double_star_expression(self):
|
||
self.validate("func(**{1: 3} if False else {x: x for x in range(3)})")
|
||
|
||
def test_argument_unpacking_1(self):
|
||
self.validate("""f(a, *b, *c, d)""")
|
||
|
||
def test_argument_unpacking_2(self):
|
||
self.validate("""f(**a, **b)""")
|
||
|
||
def test_argument_unpacking_3(self):
|
||
self.validate("""f(2, *a, *b, **b, **c, **d)""")
|
||
|
||
def test_trailing_commas_1(self):
|
||
self.validate("def f(a, b): call(a, b)")
|
||
self.validate("def f(a, b,): call(a, b,)")
|
||
|
||
def test_trailing_commas_2(self):
|
||
self.validate("def f(a, *b): call(a, *b)")
|
||
self.validate("def f(a, *b,): call(a, *b,)")
|
||
|
||
def test_trailing_commas_3(self):
|
||
self.validate("def f(a, b=1): call(a, b=1)")
|
||
self.validate("def f(a, b=1,): call(a, b=1,)")
|
||
|
||
def test_trailing_commas_4(self):
|
||
self.validate("def f(a, **b): call(a, **b)")
|
||
self.validate("def f(a, **b,): call(a, **b,)")
|
||
|
||
def test_trailing_commas_5(self):
|
||
self.validate("def f(*a, b=1): call(*a, b=1)")
|
||
self.validate("def f(*a, b=1,): call(*a, b=1,)")
|
||
|
||
def test_trailing_commas_6(self):
|
||
self.validate("def f(*a, **b): call(*a, **b)")
|
||
self.validate("def f(*a, **b,): call(*a, **b,)")
|
||
|
||
def test_trailing_commas_7(self):
|
||
self.validate("def f(*, b=1): call(*b)")
|
||
self.validate("def f(*, b=1,): call(*b,)")
|
||
|
||
def test_trailing_commas_8(self):
|
||
self.validate("def f(a=1, b=2): call(a=1, b=2)")
|
||
self.validate("def f(a=1, b=2,): call(a=1, b=2,)")
|
||
|
||
def test_trailing_commas_9(self):
|
||
self.validate("def f(a=1, **b): call(a=1, **b)")
|
||
self.validate("def f(a=1, **b,): call(a=1, **b,)")
|
||
|
||
def test_trailing_commas_lambda_1(self):
|
||
self.validate("f = lambda a, b: call(a, b)")
|
||
self.validate("f = lambda a, b,: call(a, b,)")
|
||
|
||
def test_trailing_commas_lambda_2(self):
|
||
self.validate("f = lambda a, *b: call(a, *b)")
|
||
self.validate("f = lambda a, *b,: call(a, *b,)")
|
||
|
||
def test_trailing_commas_lambda_3(self):
|
||
self.validate("f = lambda a, b=1: call(a, b=1)")
|
||
self.validate("f = lambda a, b=1,: call(a, b=1,)")
|
||
|
||
def test_trailing_commas_lambda_4(self):
|
||
self.validate("f = lambda a, **b: call(a, **b)")
|
||
self.validate("f = lambda a, **b,: call(a, **b,)")
|
||
|
||
def test_trailing_commas_lambda_5(self):
|
||
self.validate("f = lambda *a, b=1: call(*a, b=1)")
|
||
self.validate("f = lambda *a, b=1,: call(*a, b=1,)")
|
||
|
||
def test_trailing_commas_lambda_6(self):
|
||
self.validate("f = lambda *a, **b: call(*a, **b)")
|
||
self.validate("f = lambda *a, **b,: call(*a, **b,)")
|
||
|
||
def test_trailing_commas_lambda_7(self):
|
||
self.validate("f = lambda *, b=1: call(*b)")
|
||
self.validate("f = lambda *, b=1,: call(*b,)")
|
||
|
||
def test_trailing_commas_lambda_8(self):
|
||
self.validate("f = lambda a=1, b=2: call(a=1, b=2)")
|
||
self.validate("f = lambda a=1, b=2,: call(a=1, b=2,)")
|
||
|
||
def test_trailing_commas_lambda_9(self):
|
||
self.validate("f = lambda a=1, **b: call(a=1, **b)")
|
||
self.validate("f = lambda a=1, **b,: call(a=1, **b,)")
|
||
|
||
|
||
# Adapted from Python 3's Lib/test/test_grammar.py:GrammarTests.testFuncdef
|
||
class TestFunctionAnnotations(GrammarTest):
|
||
def test_1(self):
|
||
self.validate("""def f(x) -> list: pass""")
|
||
|
||
def test_2(self):
|
||
self.validate("""def f(x:int): pass""")
|
||
|
||
def test_3(self):
|
||
self.validate("""def f(*x:str): pass""")
|
||
|
||
def test_4(self):
|
||
self.validate("""def f(**x:float): pass""")
|
||
|
||
def test_5(self):
|
||
self.validate("""def f(x, y:1+2): pass""")
|
||
|
||
def test_6(self):
|
||
self.validate("""def f(a, (b:1, c:2, d)): pass""")
|
||
|
||
def test_7(self):
|
||
self.validate("""def f(a, (b:1, c:2, d), e:3=4, f=5, *g:6): pass""")
|
||
|
||
def test_8(self):
|
||
s = """def f(a, (b:1, c:2, d), e:3=4, f=5,
|
||
*g:6, h:7, i=8, j:9=10, **k:11) -> 12: pass"""
|
||
self.validate(s)
|
||
|
||
def test_9(self):
|
||
s = """def f(
|
||
a: str,
|
||
b: int,
|
||
*,
|
||
c: bool = False,
|
||
**kwargs,
|
||
) -> None:
|
||
call(c=c, **kwargs,)"""
|
||
self.validate(s)
|
||
|
||
def test_10(self):
|
||
s = """def f(
|
||
a: str,
|
||
) -> None:
|
||
call(a,)"""
|
||
self.validate(s)
|
||
|
||
def test_11(self):
|
||
s = """def f(
|
||
a: str = '',
|
||
) -> None:
|
||
call(a=a,)"""
|
||
self.validate(s)
|
||
|
||
def test_12(self):
|
||
s = """def f(
|
||
*args: str,
|
||
) -> None:
|
||
call(*args,)"""
|
||
self.validate(s)
|
||
|
||
def test_13(self):
|
||
self.validate("def f(a: str, b: int) -> None: call(a, b)")
|
||
self.validate("def f(a: str, b: int,) -> None: call(a, b,)")
|
||
|
||
def test_14(self):
|
||
self.validate("def f(a: str, *b: int) -> None: call(a, *b)")
|
||
self.validate("def f(a: str, *b: int,) -> None: call(a, *b,)")
|
||
|
||
def test_15(self):
|
||
self.validate("def f(a: str, b: int=1) -> None: call(a, b=1)")
|
||
self.validate("def f(a: str, b: int=1,) -> None: call(a, b=1,)")
|
||
|
||
def test_16(self):
|
||
self.validate("def f(a: str, **b: int) -> None: call(a, **b)")
|
||
self.validate("def f(a: str, **b: int,) -> None: call(a, **b,)")
|
||
|
||
def test_17(self):
|
||
self.validate("def f(*a: str, b: int=1) -> None: call(*a, b=1)")
|
||
self.validate("def f(*a: str, b: int=1,) -> None: call(*a, b=1,)")
|
||
|
||
def test_18(self):
|
||
self.validate("def f(*a: str, **b: int) -> None: call(*a, **b)")
|
||
self.validate("def f(*a: str, **b: int,) -> None: call(*a, **b,)")
|
||
|
||
def test_19(self):
|
||
self.validate("def f(*, b: int=1) -> None: call(*b)")
|
||
self.validate("def f(*, b: int=1,) -> None: call(*b,)")
|
||
|
||
def test_20(self):
|
||
self.validate("def f(a: str='', b: int=2) -> None: call(a=a, b=2)")
|
||
self.validate("def f(a: str='', b: int=2,) -> None: call(a=a, b=2,)")
|
||
|
||
def test_21(self):
|
||
self.validate("def f(a: str='', **b: int) -> None: call(a=a, **b)")
|
||
self.validate("def f(a: str='', **b: int,) -> None: call(a=a, **b,)")
|
||
|
||
|
||
# Adapted from Python 3's Lib/test/test_grammar.py:GrammarTests.test_var_annot
|
||
class TestVarAnnotations(GrammarTest):
|
||
def test_1(self):
|
||
self.validate("var1: int = 5")
|
||
|
||
def test_2(self):
|
||
self.validate("var2: [int, str]")
|
||
|
||
def test_3(self):
|
||
self.validate("def f():\n"
|
||
" st: str = 'Hello'\n"
|
||
" a.b: int = (1, 2)\n"
|
||
" return st\n")
|
||
|
||
def test_4(self):
|
||
self.validate("def fbad():\n"
|
||
" x: int\n"
|
||
" print(x)\n")
|
||
|
||
def test_5(self):
|
||
self.validate("class C:\n"
|
||
" x: int\n"
|
||
" s: str = 'attr'\n"
|
||
" z = 2\n"
|
||
" def __init__(self, x):\n"
|
||
" self.x: int = x\n")
|
||
|
||
def test_6(self):
|
||
self.validate("lst: List[int] = []")
|
||
|
||
|
||
class TestExcept(GrammarTest):
|
||
def test_new(self):
|
||
s = """
|
||
try:
|
||
x
|
||
except E as N:
|
||
y"""
|
||
self.validate(s)
|
||
|
||
def test_old(self):
|
||
s = """
|
||
try:
|
||
x
|
||
except E, N:
|
||
y"""
|
||
self.validate(s)
|
||
|
||
|
||
class TestStringLiterals(GrammarTest):
|
||
prefixes = ("'", '"',
|
||
"r'", 'r"', "R'", 'R"',
|
||
"u'", 'u"', "U'", 'U"',
|
||
"b'", 'b"', "B'", 'B"',
|
||
"f'", 'f"', "F'", 'F"',
|
||
"ur'", 'ur"', "Ur'", 'Ur"',
|
||
"uR'", 'uR"', "UR'", 'UR"',
|
||
"br'", 'br"', "Br'", 'Br"',
|
||
"bR'", 'bR"', "BR'", 'BR"',
|
||
"rb'", 'rb"', "Rb'", 'Rb"',
|
||
"rB'", 'rB"', "RB'", 'RB"',)
|
||
|
||
def test_lit(self):
|
||
for pre in self.prefixes:
|
||
single = "{p}spamspamspam{s}".format(p=pre, s=pre[-1])
|
||
self.validate(single)
|
||
triple = "{p}{s}{s}eggs{s}{s}{s}".format(p=pre, s=pre[-1])
|
||
self.validate(triple)
|
||
|
||
|
||
# Adapted from Python 3's Lib/test/test_grammar.py:GrammarTests.testAtoms
|
||
class TestSetLiteral(GrammarTest):
|
||
def test_1(self):
|
||
self.validate("""x = {'one'}""")
|
||
|
||
def test_2(self):
|
||
self.validate("""x = {'one', 1,}""")
|
||
|
||
def test_3(self):
|
||
self.validate("""x = {'one', 'two', 'three'}""")
|
||
|
||
def test_4(self):
|
||
self.validate("""x = {2, 3, 4,}""")
|
||
|
||
|
||
# Adapted from Python 3's Lib/test/test_unicode_identifiers.py and
|
||
# Lib/test/test_tokenize.py:TokenizeTest.test_non_ascii_identifiers
|
||
class TestIdentifier(GrammarTest):
|
||
def test_non_ascii_identifiers(self):
|
||
self.validate("Örter = 'places'\ngrün = 'green'")
|
||
self.validate("蟒 = a蟒 = 锦蛇 = 1")
|
||
self.validate("µ = aµ = µµ = 1")
|
||
self.validate("𝔘𝔫𝔦𝔠𝔬𝔡𝔢 = a_𝔘𝔫𝔦𝔠𝔬𝔡𝔢 = 1")
|
||
|
||
|
||
class TestNumericLiterals(GrammarTest):
|
||
def test_new_octal_notation(self):
|
||
self.validate("""0o7777777777777""")
|
||
self.invalid_syntax("""0o7324528887""")
|
||
|
||
def test_new_binary_notation(self):
|
||
self.validate("""0b101010""")
|
||
self.invalid_syntax("""0b0101021""")
|
||
|
||
|
||
class TestClassDef(GrammarTest):
|
||
def test_new_syntax(self):
|
||
self.validate("class B(t=7): pass")
|
||
self.validate("class B(t, *args): pass")
|
||
self.validate("class B(t, **kwargs): pass")
|
||
self.validate("class B(t, *args, **kwargs): pass")
|
||
self.validate("class B(t, y=9, *args, **kwargs,): pass")
|
||
|
||
|
||
class TestParserIdempotency(support.TestCase):
|
||
|
||
"""A cut-down version of pytree_idempotency.py."""
|
||
|
||
def parse_file(self, filepath):
|
||
if test.support.verbose:
|
||
print(f"Parse file: {filepath}")
|
||
with open(filepath, "rb") as fp:
|
||
encoding = tokenize.detect_encoding(fp.readline)[0]
|
||
self.assertIsNotNone(encoding,
|
||
"can't detect encoding for %s" % filepath)
|
||
with open(filepath, "r", encoding=encoding) as fp:
|
||
source = fp.read()
|
||
try:
|
||
tree = driver.parse_string(source)
|
||
except ParseError:
|
||
try:
|
||
tree = driver_no_print_statement.parse_string(source)
|
||
except ParseError as err:
|
||
self.fail('ParseError on file %s (%s)' % (filepath, err))
|
||
new = str(tree)
|
||
if new != source:
|
||
print(diff_texts(source, new, filepath))
|
||
self.fail("Idempotency failed: %s" % filepath)
|
||
|
||
def test_all_project_files(self):
|
||
for filepath in support.all_project_files():
|
||
with self.subTest(filepath=filepath):
|
||
self.parse_file(filepath)
|
||
|
||
def test_extended_unpacking(self):
|
||
driver.parse_string("a, *b, c = x\n")
|
||
driver.parse_string("[*a, b] = x\n")
|
||
driver.parse_string("(z, *y, w) = m\n")
|
||
driver.parse_string("for *z, m in d: pass\n")
|
||
|
||
|
||
class TestLiterals(GrammarTest):
|
||
|
||
def validate(self, s):
|
||
driver.parse_string(support.dedent(s) + "\n\n")
|
||
|
||
def test_multiline_bytes_literals(self):
|
||
s = """
|
||
md5test(b"\xaa" * 80,
|
||
(b"Test Using Larger Than Block-Size Key "
|
||
b"and Larger Than One Block-Size Data"),
|
||
"6f630fad67cda0ee1fb1f562db3aa53e")
|
||
"""
|
||
self.validate(s)
|
||
|
||
def test_multiline_bytes_tripquote_literals(self):
|
||
s = '''
|
||
b"""
|
||
<?xml version="1.0" encoding="UTF-8"?>
|
||
<!DOCTYPE plist PUBLIC "-//Apple Computer//DTD PLIST 1.0//EN">
|
||
"""
|
||
'''
|
||
self.validate(s)
|
||
|
||
def test_multiline_str_literals(self):
|
||
s = """
|
||
md5test("\xaa" * 80,
|
||
("Test Using Larger Than Block-Size Key "
|
||
"and Larger Than One Block-Size Data"),
|
||
"6f630fad67cda0ee1fb1f562db3aa53e")
|
||
"""
|
||
self.validate(s)
|
||
|
||
|
||
class TestNamedAssignments(GrammarTest):
|
||
"""Also known as the walrus operator."""
|
||
|
||
def test_named_assignment_if(self):
|
||
driver.parse_string("if f := x(): pass\n")
|
||
|
||
def test_named_assignment_while(self):
|
||
driver.parse_string("while f := x(): pass\n")
|
||
|
||
def test_named_assignment_generator(self):
|
||
driver.parse_string("any((lastNum := num) == 1 for num in [1, 2, 3])\n")
|
||
|
||
def test_named_assignment_listcomp(self):
|
||
driver.parse_string("[(lastNum := num) == 1 for num in [1, 2, 3]]\n")
|
||
|
||
|
||
class TestPositionalOnlyArgs(GrammarTest):
|
||
|
||
def test_one_pos_only_arg(self):
|
||
driver.parse_string("def one_pos_only_arg(a, /): pass\n")
|
||
|
||
def test_all_markers(self):
|
||
driver.parse_string(
|
||
"def all_markers(a, b=2, /, c, d=4, *, e=5, f): pass\n")
|
||
|
||
def test_all_with_args_and_kwargs(self):
|
||
driver.parse_string(
|
||
"""def all_markers_with_args_and_kwargs(
|
||
aa, b, /, _cc, d, *args, e, f_f, **kwargs,
|
||
):
|
||
pass\n""")
|
||
|
||
def test_lambda_soup(self):
|
||
driver.parse_string(
|
||
"lambda a, b, /, c, d, *args, e, f, **kw: kw\n")
|
||
|
||
def test_only_positional_or_keyword(self):
|
||
driver.parse_string("def func(a,b,/,*,g,e=3): pass\n")
|
||
|
||
|
||
class TestPickleableException(unittest.TestCase):
|
||
def test_ParseError(self):
|
||
err = ParseError('msg', 2, None, (1, 'context'))
|
||
for proto in range(pickle.HIGHEST_PROTOCOL + 1):
|
||
err2 = pickle.loads(pickle.dumps(err, protocol=proto))
|
||
self.assertEqual(err.args, err2.args)
|
||
self.assertEqual(err.msg, err2.msg)
|
||
self.assertEqual(err.type, err2.type)
|
||
self.assertEqual(err.value, err2.value)
|
||
self.assertEqual(err.context, err2.context)
|
||
|
||
|
||
def diff_texts(a, b, filename):
|
||
a = a.splitlines()
|
||
b = b.splitlines()
|
||
return difflib.unified_diff(a, b, filename, filename,
|
||
"(original)", "(reserialized)",
|
||
lineterm="")
|
||
|
||
|
||
if __name__ == '__main__':
|
||
unittest.main()
|