Tests for the unparse.py script in the Tools/parser directory.Nctj|5}|cdddS#1swxYwYdS)znRead and return the contents of a Python source file (as a string), taking into account the file encoding.N)tokenizeopenread)filenamestreams (/usr/lib/python3.11/test/test_unparse.py read_pyfiler s x F{{}}s 6::zQdef f(): for x in range(10): break else: y = 2 z = 3 zIdef g(): while True: break else: y = 2 z = 3 zQfrom . import fred from .. import barney from .australia import shrimp as prawns zzdef f(): x = 1 def g(): nonlocal x x = 2 y = 7 def h(): nonlocal x, y zOtry: 1 / 0 except ZeroDivisionError as e: raise ArithmeticError from e z@f1(arg) @f2 class Foo: pass z=if cond1: suite1 elif cond2: suite2 else: suite3 z,if cond1: suite1 elif cond2: suite2 zctry: suite1 except ex1: suite2 except ex2: suite3 else: suite4 finally: suite5 zetry: suite1 except* ex1: suite2 except* ex2: suite3 else: suite4 finally: suite5 zwith f(): suite1 zwith f() as x: suite1 z$with f() as x, g() as y: suite1 )zclass foo: zdef foo(): zasync def foo(): c<eZdZdZdZefdZddZddZddZ dS) ASTTestCasec|tj|tjtffd ||dS)Nct|t|ur4 t|dt|t|tjr>|jD]4}t |||}t |||}||ur ||5dSt|tr t||dD]\}}||dS#t$rRt|}t|} || krd} || z } nd} | |z } | d| d} | dwxYw||kr |d|dSdS) Nz is not T)strictFirstSecondz list contains z additional elements.z != ) typefail isinstanceastAST_fieldsgetattrlistzip ValueErrorlenfailureException)abmissingfieldvalue1value2node1node2len1len2whatdiffmsgselftraverse_compares r r-z4ASTTestCase.assertASTEqual..traverse_comparesAwwd1gg%% T!WW;;Q;;<<<!SW%% -Y99E$Qw77F$Qw77FV++((888 99At$$ -?(+Aq(>(>(>77 u((666677! ? ? ?q66Dq66Dd{{&#d{'#d{!MM$MMMC//44$> ?a Q++a++,,,,,s $C((AE)assertIsInstancerrobject)r,ast1ast2r-s` @r assertASTEqualzASTTestCase.assertASTEquals{ dCG,,, dCG,,, ,288 - - - - - - -: t$$$$$c |||5tj|fi|}tj|}tj|fi|}|||ddddS#1swxYwYdS)N)code1ast_parse_kwargs)subTestrparseunparser2)r,r5kwargsr0code2r1s r check_ast_roundtripzASTTestCase.check_ast_roundtrips \\\ ? ? , ,9U--f--DK%%E9U--f--D   d + + +  , , , , , , , , , , , , , , , , , ,sAA44A8;A8c||5||tj|ddddS#1swxYwYdS)N)node)r7 assertRaisesrr9)r,r>raisess r check_invalidzASTTestCase.check_invalids \\t\ $ $ 9 9   fck4 8 8 8 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9s"AA  A Nc^|p|}tjtj|}||fSN)rr9r8r,r5r;s r get_sourcezASTTestCase.get_sources. CIe,,--e|r3c|||\}}|||5|||ddddS#1swxYwYdSN)r5r;)rEr7 assertEqualrDs r check_src_roundtripzASTTestCase.check_src_roundtripsue44 u \\U\ 3 3 + +   UE * * * + + + + + + + + + + + + + + + + + +AAAc|||\}}|||5|||ddddS#1swxYwYdSrG)rEr7assertNotEqualrDs r check_src_dont_roundtripz$ASTTestCase.check_src_dont_roundtripsue44 u \\U\ 3 3 . .   u - - - . . . . . . . . . . . . . . . . . .rJrC) __name__ __module__ __qualname__r2r<rrArErIrMr3r r r s%%%%%%N,,,*49999 ++++ ......r3r c(eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!d Z"d!Z#d"Z$d#Z%d$Z&d%Z'd&Z(d'Z)d(Z*d)Z+d*Z,d+Z-d,Z.d-Z/d.Z0d/Z1d0S)1UnparseTestCasec|d|d|d|d|d|d|d|d|d |d |d |d |d |d|d|ddS)Nzf'a'zf'{{}}'zf'{{5}}'z f'{{5}}5'z f'X{{}}X'zf'{a}'z f'{ {1:2}}'zf'a{a}a'z f'a{a}{a}a'z f'a{a}a{a}a'zf'{a!r}x{a!s}12{{}}{a!a}'z f'{a:10}'zf'{a:100_000{10}}'z f'{a!r:10}'z f'{a:a{b}10}'zRf'a{b}{c!s}{d!r}{e!a}{f:a}{g:a{b}}{h!s:a}{j!s:{a}b}{k!s:a{b}c}{l!a:{b}c{d}}{x+y=}'r<r,s r test_fstringszUnparseTestCase.test_fstringssW   (((   +++   ,,,   ---   ---   ***   ///   ,,,   ///   000   !<===   ---   !5666   ///   111   <     r3c|d|d|d|ddS)Nz f'{f"{0}"*3}'z f'{f"{y}"*3}'zf''zf"""'end' "quote\""""rUrVs r test_fstrings_special_charsz+UnparseTestCase.test_fstrings_special_charss[   !5666   !5666   +++   !=>>>>>r3c,|d|d|d|d|d|d|ddS)Nz f'''{"'"}'''%f'''-{f"""*{f"+{f'.{x}.'}+"}*"""}-'''z4f'''-{f"""*{f"+{f'.{x}.'}+"}*"""}-'single quote\''''zf"""{''' '''}"""zf"""{g(''' ''')}"""z f"a\r\nb"f"\u2028{'x'}"rUrVs r test_fstrings_complicatedz)UnparseTestCase.test_fstrings_complicateds   !3444   !RSSS   !bccc   !:;;;   !=>>>   !2333   !677777r3c|d|d|ddS)Nzu'foo'zr'foo'zb'foo'rUrVs r test_stringszUnparseTestCase.test_stringssD   ***   ***   *****r3c0|ddS)Nz del x, y, zrUrVs r test_del_statementz"UnparseTestCase.test_del_statements   /////r3cZ|d|ddS)Nz45 << 2z13 >> 7rUrVs r test_shiftszUnparseTestCase.test_shiftss0   +++   +++++r3c:|tdSrC)r<for_elserVs r test_for_elsezUnparseTestCase.test_for_elses   *****r3c:|tdSrC)r< while_elserVs r test_while_elsezUnparseTestCase.test_while_else   ,,,,,r3c|d|d|d|d|ddS)Nz(-1)**7z(-1.)**8z(-1j)**6znot True or FalsezTrue or not FalserUrVs r test_unary_parensz!UnparseTestCase.test_unary_parenssn   +++   ,,,   ,,,   !4555   !455555r3c0|ddS)Nz 3 .__abs__()rUrVs r test_integer_parensz#UnparseTestCase.test_integer_parenss   00000r3c|d|d|d|ddS)N1e1000z-1e10001e1000jz-1e1000jrUrVs r test_huge_floatzUnparseTestCase.test_huge_float sX   ***   +++   +++   ,,,,,r3c |tjtjtjt dtjddS)Nnanvaluez1e1000 - 1e1000)r2rr8r9ConstantfloatrVs r test_nanzUnparseTestCase.test_nans\  Ick#,U5\\"B"B"BCC D D I' ( (     r3c|td|tddS)Nil)r<strrVs r test_min_intzUnparseTestCase.test_min_ints<   Z111   Z11111r3c|d|d|d|ddS)N7jz-7j0jz-0jrUrVs r test_imaginary_literalsz'UnparseTestCase.test_imaginary_literalssX   &&&   '''   &&&   '''''r3c0|ddS)Nz(lambda: int)()rUrVs r test_lambda_parenthesesz'UnparseTestCase.test_lambda_parentheses    !233333r3cZ|d|ddS)Nz 1 < 4 <= 5za is b is c is not drUrVs r test_chained_comparisonsz(UnparseTestCase.test_chained_comparisons#s1   ...   !788888r3c|d|d|d|d|d|d|d|d|d |d |d |d dS) Nz def f(): passzdef f(a): passzdef f(b = 2): passzdef f(a, b): passzdef f(a, b = 2): passzdef f(a = 5, b = 2): passzdef f(*, a = 1, b = 2): passzdef f(*, a = 1, b): passzdef f(*, a, b = 2): passz&def f(a, b = None, *, c, **kwds): passz'def f(a=2, *args, c=5, d, **kwds): passzdef f(*args, **kwargs): passrUrVs r test_function_argumentsz'UnparseTestCase.test_function_arguments's   111   !1222   !5666   !4555   !8999   !<===   !?@@@   !;<<<   !;<<<   !IJJJ   !JKKK   !?@@@@@r3c:|tdSrC)r<relative_importrVs r test_relative_importz$UnparseTestCase.test_relative_import5   11111r3c:|tdSrC)r< nonlocal_exrVs r test_nonlocalzUnparseTestCase.test_nonlocal8   -----r3c:|tdSrC)r< raise_fromrVs r test_raise_fromzUnparseTestCase.test_raise_from;rjr3c0|ddS)Nzb'123'rUrVs r test_byteszUnparseTestCase.test_bytes>s   *****r3c|d|d|d|d|ddS)Nzdef f(a : int): passzdef f(a: int = 5): passzdef f(*args: [int]): passzdef f(**kwargs: dict): passzdef f() -> None: passrUrVs r test_annotationsz UnparseTestCase.test_annotationsAsq   !7888   !:;;;   !<===   !>???   !899999r3c0|ddS)Nz{'a', 'b', 'c'}rUrVs r test_set_literalz UnparseTestCase.test_set_literalHrr3c |tjtjtjgtjddS)N)eltsz{*()})r2rr8r9SetrVs r test_empty_setzUnparseTestCase.test_empty_setKsU  Ick#'r"2"2"233 4 4 Ig       r3c0|ddS)Nz{x for x in range(5)}rUrVs r test_set_comprehensionz&UnparseTestCase.test_set_comprehensionQs   !899999r3c0|ddS)Nz{x: x*x for x in range(10)}rUrVs r test_dict_comprehensionz'UnparseTestCase.test_dict_comprehensionTs   !>?????r3c:|tdSrC)r<class_decoratorrVs r test_class_decoratorsz%UnparseTestCase.test_class_decoratorsWrr3c0|ddS)Nz(class A(metaclass=type, *[], **{}): passrUrVs r test_class_definitionz%UnparseTestCase.test_class_definitionZs   !KLLLLLr3cn|t|tdSrC)r<elif1elif2rVs r test_elifszUnparseTestCase.test_elifs]s0   '''   '''''r3c:|tdSrC)r<try_except_finallyrVs r test_try_except_finallyz'UnparseTestCase.test_try_except_finallyas   !344444r3c:|tdSrC)r<try_except_star_finallyrVs r test_try_except_star_finallyz,UnparseTestCase.test_try_except_star_finallyds   !899999r3c|d|d|d|ddS)Nza, *b, c = seqza, (*b, c) = seqza, *b[0], c = seqza, *(b, c) = seqrUrVs r test_starred_assignmentz'UnparseTestCase.test_starred_assignmentgs\   !1222   !3444   !4555   !344444r3c:|tdSrC)r< with_simplerVs r test_with_simplez UnparseTestCase.test_with_simplemrr3c:|tdSrC)r<with_asrVs r test_with_aszUnparseTestCase.test_with_asps   )))))r3c:|tdSrC)r<with_two_itemsrVs r test_with_two_itemsz#UnparseTestCase.test_with_two_itemsss   00000r3cZ|d|ddS)Nz{**{'y': 2}, 'x': 1}z{**{'y': 2}, **{'x': 1}}rUrVs r test_dict_unpacking_in_dictz+UnparseTestCase.test_dict_unpacking_in_dictvs2   !<===   !@AAAAAr3c|d|d|d|d|d|d|d|d|d |d |d |d |d |d|d|d|d|d|d|d|d|d|d|ddS)Nza[i]za[i,]za[i, j]za[(*a,)]za[*a]za[b, *a]za[*a, c]z a[b, *a, c] a[*a, *a]z a[b, *a, *a]z a[*a, b, *a]z a[*a, *a, b]za[b, *a, *a, c]z a[(a:=b)]z a[(a:=b,c)]a[()]za[i:j]za[:j]za[i:]za[i:j:k]za[:j:k]za[i::k]za[i:j,]z a[i:j, k]rUrVs r test_sliceszUnparseTestCase.test_slices{s   (((   )))   +++   ,,,   )))   ,,,   ,,,   ///   ---   000   000   000   !2333   ---   ///   )))   ***   )))   )))   ,,,   +++   +++   +++   -----r3c~|tjdtjddS)NXid)exccause)rArRaiseNamerVs r test_invalid_raisez"UnparseTestCase.test_invalid_raises8 39SX5E5E5EFFFGGGGGr3c|tjtjdtjdgdS)Ntestrru)values)rAr JoinedStrrrwrVs r test_invalid_fstring_valuez*UnparseTestCase.test_invalid_fstring_values]  MH'''Lv...        r3c||tjtjddS)Nz\\ru)rArFormattedValuerwrVs r test_invalid_fstring_backslashz.UnparseTestCase.test_invalid_fstring_backslashs7 3-CLv4N4N4NOOOPPPPPr3cV|tjddS)Nru)rAr YieldFromrVs r test_invalid_yield_fromz'UnparseTestCase.test_invalid_yield_froms) 3=t44455555r3cPtjdtjdg}|tj|dtjdtjdgd}|tj|ddS)Nmodx)name)modulenameszfrom mod import x)rrlevel)r ImportFromaliasrHr9)r,trees r test_import_from_level_nonez+UnparseTestCase.test_import_from_level_nones~U39#3F3F3F2GHHH T**,?@@@~U39#3F3F3F2GtTTT T**,?@@@@@r3cFd}|D]}|d|ddS)N) zthis ends with double quote"z"this includes a """triple quote""" z\r z\t z\nz \r \t \nz$""">>> content = """blabla""" <<<"""z foo\n\x00z' \'\'\'""" ""\'\' \'u🐍⛎𩸽üéş^\\X\\BB⟿z'''rU)r, docstrings docstrings r test_docstringszUnparseTestCase.test_docstringssG $ ; ;I  $ $%99%9%9%9 : : : : ; ;r3c|tjddd|tjddddS)N))rvkindz(1,))rz (1, 2, 3))rIrrwrVs r test_constant_tuplesz$UnparseTestCase.test_constant_tupless]   Dt!D!D!DfMMM   Lyt 4 4 4k     r3c>dD]}||ddS)N)z () -> intz(int, int) -> intz=(Callable[complex], More[Complex(call.to_typevar())]) -> None func_type)moderU)r, function_types r test_function_typez"UnparseTestCase.test_function_types?  F FM  $ $] $ E E E E  F Fr3c>dD]}||ddS)N) z a = 5 # type:za = 5 # type: intza = 5 # type: int and morez!def x(): # type: () -> None passz.def x(y): # type: (int) -> None and more passz'async def x(): # type: () -> None passz4async def x(y): # type: (int) -> None and more passzfor x in y: # type: int passz#async for x in y: # type: int passzwith x(): # type: int passz!async with x(): # type: int passT type_commentsrUr, statements r test_type_commentsz"UnparseTestCase.test_type_commentss?  D DI  $ $Yd $ C C C C D Dr3c>dD]}||ddS)N) za = 5 # type: ignoreza = 5 # type: ignore and morezdef x(): # type: ignore passz'def x(y): # type: ignore and more passz#async def x(): # type: ignore passz-async def x(y): # type: ignore and more passz for x in y: # type: ignore passz&async for x in y: # type: ignore passzwith x(): # type: ignore passz$async with x(): # type: ignore passTrrUrs r test_type_ignorez UnparseTestCase.test_type_ignores?  D DI  $ $Yd $ C C C C D Dr3N)2rNrOrPrWrYr]r_rarcrfrirlrnrrryr|rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrQr3r rSrSs   *???888+++ 000,,,+++---666111---    222((( 444999 A A A222...---+++:::444   :::@@@222MMM(((555:::555 ...***111BBB ...6HHH   QQQ666AAA ;;;(   FFFDDD D D D D Dr3rScNeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d S) CosmeticTestCasezATest if there are cosmetic issues caused by unnecessary additionsc|d|d|d|d|d|d|d|d|d |d |d |d |d |d|d|d|d|d|d|d|d|d|d|d|d|d|d|ddS)Nz(a := b)zawait xz x if x else yz lambda x: xz1 + 1z 1 + 2 / 3z (1 + 2) / 3z(1 + 2) * 3 + 4 * (5 + 2)z(1 + 2) * 3 + 4 * (5 + 2) ** 2z~xzx and yz x and y and zzx and (y and x)z(x and y) and zz(x ** y) ** z ** qzx >> yzx << yzx >> y and x >> zzx + y - z * q ^ t ** kzP * V if P and V else n * R * Tz"lambda P, V, n: P * V == n * R * Tzflag & (other | foo)z not x == yz x == (not y)zyield xz yield from xzcall((yield x))zreturn x + (yield x)rIrVs r test_simple_expressions_parensz/CosmeticTestCase.test_simple_expressions_parenssD   ,,,   +++   111   ///   )))   ---   ///   !<===   !ABBB   &&&   +++   111   !2333   !2333   !5666   ***   ***   !4555   !9:::   !BCCC   !EFFF   !7888   ...   000   +++   000   !2333   !788888r3c|d|d|d|d|d|d|d|d|d |d dS) Nzclass X: passzclass X(A): passzclass X(A, B, C, D): passzclass X(x=y): passzclass X(metaclass=z): passzclass X(x=y, z=d): passzclass X(A, x=y): passzclass X(A, **kw): passzclass X(*args): passz"class X(*args, **kwargs): passrrVs r test_class_bases_and_keywordsz.CosmeticTestCase.test_class_bases_and_keywordss   !5666   !8999   !ABBB   !:;;;   !BCCC   !?@@@   !=>>>   !>???   !<===   !FGGGGGr3c|d|d|d|d|ddS)Nr[r\zf'{x}\n'zf'''{""" """}\n'''zf'''{f"""{x} """}\n'''rrVs r rWzCosmeticTestCase.test_fstringssp   !RSSS   !6777   ---   !=>>>   !ABBBBBr3cZd}tD] }|D]}|||!dS)N) z"""simple doc string"""z7"""A more complex one with some newlines"""z,"""Foo bar baz empty newline"""z"""With some """z"""Foo "bar" baz """z"""\r"""z""""""z """'''"""z """''''''"""u""""🐍⛎𩸽üéş^\\X\\BB⟿"""z"""end in single 'quote'"""z'''end in double "quote"'''z#"""almost end in double "quote".""")docstring_prefixesrI)r,rprefixrs r rz CosmeticTestCase.test_docstrings%sa &) A AF' A A ((F)?I)?)?@@@@ A A Ar3cd}tD]7}|D]2}||}||||38dS)N)za = """false"""z("""false""" + """unless its optimized"""z1 + 1 """false"""z"f"""no, top level but f-fstring""")rr<rM)r,docstrings_negativernegativesrcs r test_docstrings_negative_casesz/CosmeticTestCase.test_docstrings_negative_cases=sv  ) 3 3F/ 3 3 +++((-----c2222  3 3 3r3czdD]}||ddD]}||ddS)N)+-~1)notz 1r)r,rs r test_unary_op_factorz%CosmeticTestCase.test_unary_op_factorOsd% 3 3F  $ $\\\ 2 2 2 2 4 4F  $ $]]] 3 3 3 3 4 4r3cR|d|d|d|d|d|d|d|d|d |d |d |d |d |ddS)Nrza[1]za[1, 2]za[*a,]za[1, *a]za[*a, 2]z a[1, *a, 2]rz a[1, *a, *a]z a[*a, 1, *a]z a[*a, *a, 1]za[1, *a, *a, 2]z a[1:2, *a]z a[*a, 1:2]rrVs r rzCosmeticTestCase.test_slicesUs#   )))   (((   +++   ***   ,,,   ,,,   ///   ---   000   000   000   !2333   ...   .....r3cV|d|d|d|d|d|d|d|ddS) Nzlambda: somethingzfour = lambda: 2 + 2zlambda x: x * 2zsquare = lambda n: n ** 2zlambda x, y: x + yzadd = lambda x, y: x + yz lambda x, y, /, z, q, *, u: Nonezlambda x, *y, **z: NonerrVs r test_lambda_parametersz'CosmeticTestCase.test_lambda_parametershs   !4555   !7888   !2333   !<===   !5666   !;<<<   !CDDD   !:;;;;;r3cdD]b\}}dD]Z}|||5|||dddn #1swxYwY[cdS)N))zsingle assignmentz{target} = foo)zmultiple assignmentz{target} = {target} = bar)zfor loopzfor {target} in foo: pass)zasync for loopz#async for {target} in foo: pass)rza,za, bza, *b, cz a, (b, c), dza, (b, c, d), *eza, (b, *c, d), eza, (b, *c, (d, e), f), gz[a]z[a, b]z [a, *b, c]z[a, [b, c], d]z[a, [b, c, d], *e]z[a, [b, *c, d], e]z[a, [b, *c, [d, e], f], g]z a, [b, c], dz[a, b, (c, d), (e, f)]za, b, [*c], d, e) source_typetarget)r)r7rIformat)r,rsourcers r test_star_expr_assign_targetz-CosmeticTestCase.test_star_expr_assign_targetrs$  K K K  K K(\\k&\IIKK,,V]]&]-I-IJJJKKKKKKKKKKKKKKK) K  K Ks*AA A c,|d|d|d|d|d|d|ddS)Nz() = []z[] = ()z'() = [a] = c, = [d] = e, f = () = g = hz a = b = c = dza, b = c, d = e, f = gz[a, b] = [c, d] = [e, f] = gza, b = [c, d] = e, f = grrVs r %test_star_expr_assign_target_multiplez6CosmeticTestCase.test_star_expr_assign_target_multiples   +++   +++   !JKKK   111   !9:::   !?@@@   !;<<<< test_ast.py test_patma.pytest_syntax.pytest_compile.pytest_fstring.pytest_grammar.pytest_asdl_parser.pyNcjjSdjD}tjdsBfd|D}t t j|d}t||z}|_|S)Ncg|]H}|dD]0}|jd|1IS)z*.pybad)globr startswithresolve).0 directoryitems r z3DirectoryTestCase.files_to_test..si   !v..  9''..  LLNN    r3cpuc0h|]}|jjv|SrQ)rrun_always_files)r,r.clss r z2DirectoryTestCase.files_to_test..s5#F#F#FD#'90D#D#D$(#D#D#Dr3 ) _files_to_testtest_directoriesrsupportis_resource_enabledsetrandomsampler)r3itemstests_to_run_alwayss` r files_to_testzDirectoryTestCase.files_to_tests   )% %   1   |//66 6#F#F#F#FE#F#F#F  eR0011E!4455E# r3cB|D]}tjjr$t d|||5t|}||dddn #1swxYwYdS)NzTesting )r) r?rr8verboseprintabsoluter7r r<)r,r.rs r test_fileszDirectoryTestCase.test_filess&&(( 1 1D|# 4222333t,, 1 1$T**((000 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1  1 1s"%BB B )rNrOrPrpathlibPath__file__parentlib_dirr7r2r6 classmethodr?rDrQr3r rrsDDgl8$$+d2G6!12)))N[<11111r3r__main__)runittest test.supportrrEr;rrr rerhrrrrrrrrrrrrTestCaser rSrrrNmainrQr3r rPsDD                @.@.@.@.@.(#@.@.@.DiDiDiDiDiDkiDiDiDX f=f=f=f=f={f=f=f=T1111111111 111111h zHMOOOOOr3