B u9aÍÎã=@sþddlZddlZddlZddlZddlZddlZddlmZdd„Zddddd d d d d ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3g/Z d4gZ dd5d6d7d8d9d:d;dd?d@dAdBdCdDdEdFdGdHdIdJdKgZ GdLdM„dMej ƒZ GdNdO„dOej ƒZGdPdQ„dQej ƒZGdRdS„dSej ƒZdTdU„ZdVdWgfdVdXgfdVdYdZd[d\gdggdgfd]ggdfgfdVdYdZd[d\gdggdgfd^ggdfgfdVdYdZd[d\d_gdggdgfd`ggdfgfdVdYdZd[d\d_gdggddagfdbggdfgfdVdYdZd[d\gdcggdgfddggdfgfdVdYdZd[d\gdggdegfdfggdfgfdVdYdZd[d\d_dgdhdidjgdkdlgdmgdndodpdqdrgdsfdtduggfgfdvggdfgfdVdwdZdxggdyggfgfdVdwdZdxggdzggfgfdVdwdZdxd{ggdfggfgfdVdYdZd[d\gdggdgfd|ggdfgfdVd}dZd~gfgfdVddZd€gdfgfdVd‚gfdVdƒdZd„d…d†ggfgfdVd‡dZdˆdyggfgfdVd‰dZdŠd‹ggfgfdVdŒdZdgdŽgfgfdVdŒdZddgdgfgfdVd‘dZd’d“d”d•ggfdfgfdVd–dZd—gd˜d™dšdd›gfgggfgfdVd–dZd—gggd›gfgfdVdœgfdVddZdžgfgfdVdŸdZd d¡gdfgfdVd¢dZdEgfgfdVd£gfdVd¤gfdVdƒdZd„d…d¥ggfgfdVdƒdZd„d…d¦ggfgfdVdƒdZd§d¨d©dªgd«fd¬ddggfgfdVd­dZd®d¯d§d°d±d²gdsfd³d§d´dµd¶gd«fd·gdfgffgfdVd­dZd¸d¯d§d°d±d²gdsfd³d§d´dµd¶gd«fd·gdfgffgfdVd­dZd¸d¯d§d°d±d²gdsfd³d§d¹dºd»gd«fd¼gdfgffgfdVd­dZd¸d½d§d¾d¿dÀgdsfd³d§dÁdÂdÃgd«fdÄgdfgffgfdVd­dZdÅdZdÆdÇd³dÈdÉgdfd³dÊdËdÌgdfgffgfdVd­dZdÅdZdÆdÇd³d§d´dÍdÎgd«fdÏgdfgffgfdVd­dZdÐdZdÑd³dÒdÓdÔgdfgffgfdVd­dZdÐdZdÑd³d§dÕdÒdÖgd«fd×gdfgffgfdVdØdZd[d\gdggdgfdÙd­dÚdÛdÚd’dÜdÝggfffggdfgfdVdØdZd[d\gdggdgfdÞdßdàdádâgdãgfggdfgfdVdØdZd[d\gdggdgfdädßdågdægfggdfgfdVd­dZdtdZddçgdtdèdgdégfdêgffgfdVd­dZdëdZdìd¯dëd°dídégfdsfdîgffgfdVdØdZd[d\gdggdgfd­dßd®dïdðd³dñdògdófgffggdfgfg/ZdôdõgfgZdöd÷dødZdùdúdûgffdüdýd÷dþdZd\gdggdgfdÿffd÷dtdZdgdgffd÷dtdZggffd÷dëdZdgffd÷dtdZdgdgffd÷d®d¯dÆd³dddgdfgffd÷d¸d¯dÆd³dddgdfgffd÷ddZd d d gdd gffd÷d’dZd ddd gddgffdddddd÷dqdZdddgdsffd÷dqdZgdsffd÷d§dZd ddgdsffd÷d§d¯dddgdsffd÷d§dZgdsffd÷d’dZddggffgZeƒdS(éN)ÚsupportcCs¢|dkst|tttfƒr|St|tƒr4dd„|DƒS|jjg}t|dƒrdt|dƒrd| |j |j f¡|j dkrvt |ƒSx"|j D]}| t t||ƒƒ¡q~Wt |ƒS)NcSsg|] }t|ƒ‘qS©)Úto_tuple)Ú.0Úerrú/usr/lib/python3.7/test_ast.pyú szto_tuple..ÚlinenoÚ col_offset)Ú isinstanceÚstrÚintÚcomplexÚlistÚ __class__Ú__name__ÚhasattrÚappendr r Ú_fieldsÚtuplerÚgetattr)ÚtÚresultÚfrrrr s    rÚNonez'module docstring'z def f(): passzdef f(): 'function docstring'zdef f(a): passzdef f(a=0): passzdef f(*args): passzdef f(**kwargs): passzGdef f(a, b=1, c=None, d=[], e={}, *args, f=42, **kwargs): 'doc for f()'z class C:passz class C: 'docstring for class C'zclass C(object): passzdef f():return 1zdel vzv = 1zv += 1zfor v in v:passz while v:passz if v:passzwith x as y: passzwith x as y, z as q: passzraise Exception('string')z$try: pass except Exception: passztry: pass finally: passzassert vz import syszfrom sys import vzglobal vÚ1Úpasszfor v in v:breakzfor v in v:continuezfor a,b in c: passz[(a,b) for a,b in c]z((a,b) for a,b in c)z((a,b) for (a,b) in c)zL( ( Aa , Bb ) for Aa , Bb in Cc )z"{a : b for w in x for m in p if g}z{a : b for v,w in x}z{r for l in x if g}z{r for l,m in x}z3async def f(): 'async function' await something()z,async def f(): async for e in i: 1 else: 2z$async def f(): async with a as b: 1z{**{1:2}, 2:3}z {*{1, 2}, 3}z$async def f(): [i async for b in c]z1+2za and bza + bznot vz lambda:Nonez{ 1:2 }z{}z{None,}z&{ 1 : 2 }z[a for b in c if d]z(a for b in c if d)z 1 < 2 < 3zf(1,2,c=3,*d,**e)Z10z'string'za.bza[b:c]Úvz[1,2,3]z[]z1,2,3z(1,2,3)z()za.b.c.d(a.b[1:2])c@s²eZdZdd„Zdd„Zdd„Zdd„Zd d „Zd d „Zd d„Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zdd „Zd!d"„Zd#d$„Zd%d&„Zejd'd(„ƒZd)S)*Ú AST_TestscCs®t|tjƒr|jdkrdSt|tjtjtjfƒrV|j|jf}|  ||k¡|j|jf}xR|jD]H}t ||ƒ}t|t ƒr’x.|D]}|  ||¡q|Wq^|dk r^|  ||¡q^WdS)N) r ÚastÚASTrÚexprÚstmtZ excepthandlerr r Ú assertTruerrÚ_assertTrueorder)ÚselfZast_nodeZ parent_posZnode_posÚnameÚvalueZchildrrrr$És      zAST_Tests._assertTrueorderc Cs|t ¡}| |jd¡d|_| |jd¡| |jdd¡| t¡ |jWdQRX| t ¡t d¡WdQRXdS)Nré*Úfoobaré) rr Ú assertEqualrr)Ú__dict__Ú assertRaisesÚAttributeErrorÚvarargÚ TypeError)r%ÚxrrrÚtest_AST_objectsØs  zAST_Tests.test_AST_objectscCsLGdd„dƒ}t ¡}|ƒ|_||j_t |j¡}~t ¡| |ƒ¡dS)Nc@s eZdZdS)z0AST_Tests.test_AST_garbage_collection..XN)rÚ __module__Ú __qualname__rrrrÚXçsr5) rr r1ÚaÚweakrefÚrefrZ gc_collectÚ assertIsNone)r%r5r6r8rrrÚtest_AST_garbage_collectionæs z%AST_Tests.test_AST_garbage_collectionc Cs¶x°ttdfttdfttdffD]\}}}x„t||ƒD]v\}}|jd|d2t|d|t j ƒ}|  t |ƒ|¡|  |d¡WdQRX|jd||d t|d|ƒWdQRXq4WqWdS) NÚexecÚsingleÚevalZparsing)ÚactionÚinputú?)rrZ compiling)r>r?Úkind)Ú exec_testsÚ exec_resultsÚ single_testsÚsingle_resultsÚ eval_testsÚ eval_resultsÚzipÚsubTestÚcompilerÚ PyCF_ONLY_ASTr+rr$)r%r?ÚoutputrAÚiÚoZast_treerrrÚ test_snippetsñs zAST_Tests.test_snippetscCs<t d¡jdjj}| |j¡| |j¡| |j¡dS)Nzx[::]r) rÚparseÚbodyr'Úslicer9ÚupperÚlowerÚstep)r%ZslcrrrÚ test_sliceýs  zAST_Tests.test_slicecCs t d¡jd}| |j¡dS)Nzfrom . import yr)rrPrQr9Úmodule)r%ZimrrrÚtest_from_importszAST_Tests.test_from_importcCs>t d¡}| |jdtj¡d ¡|jd_t|ddƒdS)Nzfrom __future__ import divisionrz __future__ zr;)rrPZassertIsInstancerQÚ ImportFromÚstriprWrJ)r%ÚmodrrrÚ!test_non_interned_future_from_asts z+AST_Tests.test_non_interned_future_from_astcCs|| ttjtjƒ¡| ttjtjƒ¡| ttjtjƒ¡| ttjtjƒ¡| ttjtjƒ¡| ttj tjƒ¡dS)N) r#Ú issubclassrÚForr"ÚNamer!r Ú comprehensionZGt)r%rrrÚtest_base_classes s zAST_Tests.test_base_classescCs^xXtj ¡D]J\}}t|tƒr |dkr |d ¡r |ƒ}t|tjƒr | t|jƒt ¡q WdS)Nr r) rr,Úitemsr ÚtypeÚisupperr r+rr)r%r&Úitemr1rrrÚtest_field_attr_existences  z#AST_Tests.test_field_attr_existencec CsTt ¡}| |jd¡| t¡ |jWdQRXtjtddƒŽ}| |jd¡dS)N)Úargsr/Ú kwonlyargsÚ kw_defaultsÚkwargÚdefaultséér*)rÚ argumentsr+rr-r.r/Úrange)r%r1rrrÚtest_argumentss  zAST_Tests.test_argumentscCs t ¡}d|_| |jd¡dS)Niš)rÚNumrr+)r%r1rrrÚtest_field_attr_writable'sz"AST_Tests.test_field_attr_writablec Cst ¡}| |jd¡| t¡ |jWdQRXt d¡}| |jd¡| t¡ |jWdQRX| t¡ |jWdQRXtjdd}| |jd¡tjddd}| |jd¡| |jd¡| |jd¡| t tjdd¡|jt tjdddddS)N)Únr(r*)r rrl) rrqr+rr-r.rsr r)r0)r%r1rrrÚtest_classattrs-s$     zAST_Tests.test_classattrscCs(t d¡g}t |¡}| |j|¡dS)Nr()rrqÚModuler+rQ)r%rQr1rrrÚ test_moduleHs  zAST_Tests.test_modulec Csžt ¡}| |jd¡d|_| |jd¡t d¡}t d¡}t ¡}t |||¡}| |j|¡| |j|¡| |j |¡t ddd¡}| |jd¡| |jd¡| |j d¡tjddddd}| |jd¡| |jd¡| |j d¡| |j d¡|  t tjdddd¡|j t tjddddddtjddddd }| |jd¡| |jd¡| |j d¡| |j d¡tjdddd d }| |jd ¡dS) N)ÚleftÚopÚrightérlér*r)r é)rwrxryr r()Ú foobarbaz) rÚBinOpr+rr}rqÚAddrwrxryr r-r0)r%r1Zn1Zn3ZaddoprrrÚtest_nodeclassesMs:  zAST_Tests.test_nodeclassescCst ¡}| |jd¡dS)Nr)rZSubr+r)r%r1rrrÚtest_no_fieldsyszAST_Tests.test_no_fieldsc Cs¢ddl}|g}yddl}| |¡Wntk r8YnXdddg}xX|D]P}xJ|D]B}x‰sz*AST_Tests.test_pickling..) ÚpickleÚcPicklerÚ ImportErrorrBÚloadsÚdumpsr+r) r%rƒZmodsr„Z protocolsr[ZprotocolrZast2rrrÚ test_pickling~s   zAST_Tests.test_picklingc Csbtddd}t tjtjf|Žf|Žg¡}| t¡}t|ddƒWdQRX| dt |j ƒ¡dS)Nr*r{)r r zr;zbut got <_ast.expr) ÚdictrruÚExprr!r-r0rJÚassertInr Ú exception)r%ÚposÚmÚcmrrrÚtest_invalid_sums   zAST_Tests.test_invalid_sumc Cs`t t t dt ¡¡¡g¡}t |¡| t¡}t|ddƒWdQRX|  dt |j ƒ¡dS)Nr(zr;zidentifier must be of type str) rrurŠr_ÚLoadÚfix_missing_locationsr-r0rJr‹r rŒ)r%rŽrrrrÚtest_invalid_identitifer”s   z"AST_Tests.test_invalid_identitiferc CsVt d¡}d|jdjdj_| t¡}t|ddƒWdQRX| dt|j ƒ¡dS)Nzdef f(): yield from g()rzr;zfield value is required) rrPrQr'r-Ú ValueErrorrJr‹r rŒ)r%Zempty_yield_fromrrrrÚtest_empty_yield_from›s   zAST_Tests.test_empty_yield_fromc Cs>ddl}dd„}t |d|¡| ttjd¡WdQRXdS)NrcWsdS)Nr)rgrrrÚ bad_normalize¨sz0AST_Tests.test_issue31592..bad_normalizeÚ normalizeuÏ•)Ú unicodedatarZ swap_attrr-r0rrP)r%r˜r–rrrÚtest_issue31592£szAST_Tests.test_issue31592N)rr3r4r$r2r:rOrVrXr\rarfrprrrtrvr€rrˆrr“r•rÚ cpython_onlyr™rrrrrÇs(   ,rc@s|eZdZdd„Zdd„Zdd„Zdd„Zd d „Zd d „Zd d„Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„ZdS)ÚASTHelpers_TestcCs6t d¡}tdddtjƒ}| t |¡t |¡¡dS)Nz foo(1 + 1)z r;)rrPrJrKr+Údump)r%r6ÚbrrrÚ test_parse°s zASTHelpers_Test.test_parsec CsTy ddWnBtk rN| t¡}t d¡WdQRX| |jj¡YnXdS)Nrlrz'\U')Ú Exceptionr-Ú SyntaxErrorrÚ literal_evalZassertIsNotNonerŒÚ __context__)r%rrrrÚtest_parse_in_errorµs   z#ASTHelpers_Test.test_parse_in_errorcCsLt d¡}| t |¡d¡| tj|ddd¡| tj|ddd¡dS) Nzspam(eggs, "and cheese")z‡Module(body=[Expr(value=Call(func=Name(id='spam', ctx=Load()), args=[Name(id='eggs', ctx=Load()), Str(s='and cheese')], keywords=[]))])F)Zannotate_fieldszYModule([Expr(Call(Name('spam', Load()), [Name('eggs', Load()), Str('and cheese')], []))])T)Úinclude_attributesaModule(body=[Expr(value=Call(func=Name(id='spam', ctx=Load(), lineno=1, col_offset=0), args=[Name(id='eggs', ctx=Load(), lineno=1, col_offset=5), Str(s='and cheese', lineno=1, col_offset=11)], keywords=[], lineno=1, col_offset=0), lineno=1, col_offset=0)]))rrPr+rœ)r%ÚnoderrrÚ test_dump½s  zASTHelpers_Test.test_dumpcCsBtjddd}t t d¡|jj¡|j_| tj|ddd¡dS)Nz1 + 1r=)Úmoder*T)r¤z‡Expression(body=BinOp(left=Num(n=1, lineno=1, col_offset=0), op=Add(), right=Num(n=2, lineno=1, col_offset=4), lineno=1, col_offset=0)))rrPÚ copy_locationrqrQryr+rœ)r%ÚsrcrrrÚtest_copy_locationÐsz"ASTHelpers_Test.test_copy_locationc Csft d¡}|j t t t dt ¡¡t d¡gg¡¡¡|  |t  |¡¡|  tj |ddd¡dS)Nz write("spam")ÚspamZeggsT)r¤a|Module(body=[Expr(value=Call(func=Name(id='write', ctx=Load(), lineno=1, col_offset=0), args=[Str(s='spam', lineno=1, col_offset=6)], keywords=[], lineno=1, col_offset=0), lineno=1, col_offset=0), Expr(value=Call(func=Name(id='spam', ctx=Load(), lineno=1, col_offset=0), args=[Str(s='eggs', lineno=1, col_offset=0)], keywords=[], lineno=1, col_offset=0), lineno=1, col_offset=0)])) rrPrQrrŠÚCallr_r‘ÚStrr+r’rœ)r%r©rrrÚtest_fix_missing_locationsÙs  z*ASTHelpers_Test.test_fix_missing_locationscCs|tjddd}| tj|dd|¡| tj|ddd¡tjddd}| tj|jdd|j¡| tj|ddd¡dS) Nz1 + 1r=)r§r{)rsT)r¤z‡Expression(body=BinOp(left=Num(n=1, lineno=4, col_offset=0), op=Add(), right=Num(n=1, lineno=4, col_offset=4), lineno=4, col_offset=0)))rrPr+Zincrement_linenorœrQ)r%r©rrrÚtest_increment_linenoész%ASTHelpers_Test.test_increment_linenocCsHtjddd}tt |j¡ƒ}| | d¡jd¡| |ggdœ¡dS)Nzfoo()r=)r§ÚfuncÚfoo)Úkeywordsrg)rrPr‰Z iter_fieldsrQr+ÚpopÚid)r%r¥ÚdrrrÚtest_iter_fieldsúsz ASTHelpers_Test.test_iter_fieldscCs†tjddd}| ttt |j¡ƒƒd¡t |j¡}| t|ƒjd¡| t|ƒj d¡| t|ƒj d¡| t  t|ƒ¡d¡dS) Nzspam(23, 42, eggs='leek')r=)r§r|r«ér(z(keyword(arg='eggs', value=Str(s='leek'))) rrPr+ÚlenrZiter_child_nodesrQÚnextr´rsrœ)r%r¥ÚiteratorrrrÚtest_iter_child_nodess z%ASTHelpers_Test.test_iter_child_nodescCs†t d¡}| t |¡d¡t d¡}| t |jd¡d¡t d¡}| t |jd¡d¡t d¡}| t |jd¡d¡dS)Nz"""line one line two"""zline one line twoz&class foo: """line one line two"""rz&def foo(): """line one line two"""z#async def foo(): """spam ham"""zspam ham)rrPr+Ú get_docstringrQ)r%r¥rrrÚtest_get_docstring s     z"ASTHelpers_Test.test_get_docstringcCs.| t t d¡¡¡t d¡}| t |¡¡t d¡}| t |¡¡t d¡}| t |jd¡¡t d¡}| t |jd¡¡t d¡}| t |jd¡¡t d¡}| t |jd¡¡t d¡}| t |jd¡¡t d ¡}| t |jd¡¡t d ¡}| t |jd¡¡dS) NÚzx = "not docstring"zdef foo(): passzclass foo: passrz class foo: x = "not docstring"z class foo: def bar(self): passz def foo(): x = "not docstring"zasync def foo(): passz&async def foo(): x = "not docstring")r9rr¼rPrQ)r%r¥rrrÚtest_get_docstring_nones&         z'ASTHelpers_Test.test_get_docstring_nonecCs0| t d¡dddg¡| t d¡ddi¡| t d¡d ¡| t d ¡dddh¡| t d ¡d ¡| ttjd ¡| t d¡d¡| t d¡d¡| t d¡d¡| t d¡d¡| t d¡d¡| t d¡d¡| tt d¡ƒd¡| ttjd¡| ttjd¡| ttjd¡dS)Nz [1, 2, 3]rlr*r{z {"foo": 42}r±r(z(True, False, None))TFNz {1, 2, 3}zb"hi"shizfoo()Ú6éz+6z-6iúÿÿÿz3.25g @z+3.25z-3.25g Àz-0.0z++6z+Truez2+3)r+rr¡r-r”Úrepr)r%rrrÚtest_literal_eval3s z!ASTHelpers_Test.test_literal_evalcCs>| t d¡d¡| t d¡d¡| t d¡d¡| t d¡d¡| t d ¡d ¡| t d ¡d ¡| t d ¡d¡| t d¡d¡| t d¡d¡| t d¡d¡| t d¡d¡| t d¡d¡| t d¡d ¡| ttjd¡| ttjd¡| ttjd¡| ttjd¡| ttjd¡dS)NZ6jy@z-6jy€Àz6.75jy@z-6.75jy€Àz3+6jy@@z-3+6jyÀ@z3-6jy@Àz-3-6jyÀÀz 3.25+6.75jy @@z -3.25+6.75jy À@z 3.25-6.75jy @Àz -3.25-6.75jy ÀÀz(3+6j)z-6j+3z-6j+3jz3+-6jz3+(0+6j)z-(3+6j))r+rr¡r-r”)r%rrrÚtest_literal_eval_complexEs$z)ASTHelpers_Test.test_literal_eval_complexc Csbtjdtjddgddddg}t |¡}| t¡}t|ddƒWdQRX| dt|j ƒ¡dS)NÚtimeÚsleep)r&)rWÚnamesÚlevelr r Útestr;zinvalid integer value: None) rrYÚaliasrur-r”rJr‹r rŒ)r%rQr[rrrrÚtest_bad_integerZs    z ASTHelpers_Test.test_bad_integercCsTtjdtjddgddddg}t |¡}t|ddƒ}i}t||ƒ| d|¡dS)NrÅrÆ)r&r)rWrÇrÈr r rÉr;)rrYrÊrurJr;r‹)r%rQr[ÚcodeÚnsrrrÚtest_level_as_nonees     z"ASTHelpers_Test.test_level_as_noneN)rr3r4ržr£r¦rªr®r¯r¶r»r½r¿rÃrÄrËrÎrrrrr›®s   r›c@sŽeZdZd`edœdd„Zdaedœdd„Zdbdd „Zd d „Zd d „Zdd„Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zd d!„Zd"d#„Zd$d%„Zd&d'„Zd(d)„Zd*d+„Zd,d-„Zd.d/„Zd0d1„Zd2d3„Zd4d5„Zd6d7„Zd8d9„Zd:d;„Zdd?„Z!d@dA„Z"dBdC„Z#dDdE„Z$dFdG„Z%dHdI„Z&dJdK„Z'dLdM„Z(dNdO„Z)dPdQ„Z*dRdS„Z+dTdU„Z,dVdW„Z-dXdY„Z.dZd[„Z/d\d]„Z0d^d_„Z1dS)cÚASTValidatorTestsNr;)Úexcc CsVd|_|_t |¡| |¡}t|d|ƒWdQRX|dk rR| |t|jƒ¡dS)Nrz) r r rr’r-rJr‹r rŒ)r%r[Úmsgr§rÐrrrrr[ss    zASTValidatorTests.modcCs&t t |¡g¡}|j|||ddS)N)rÐ)rrurŠr[)r%r¥rÑrÐr[rrrr!{szASTValidatorTests.exprcCst |g¡}| ||¡dS)N)rrur[)r%r"rÑr[rrrr"s zASTValidatorTests.stmtc CsTt t t dt ¡¡¡g¡}| |dd¡t t dt ¡¡¡}| |dd¡dS)Nr1zmust have Load contextr<r=)rÚ InteractiverŠr_ÚStorer[Ú Expression)r%rŽrrrrvƒszASTValidatorTests.test_modulec sd‡fdd„ }t dt dt ¡¡¡g}|||ddƒ|||ddƒ||t d¡gdd ƒ||t d ¡gd d ƒt dt dt ¡¡¡g}|||t dt ¡¡gd dƒt dt dt ¡¡¡t dt dt ¡¡¡g}|||dt dt ¡¡gddƒdS)NcsL|dkr g}|dkrg}|dkr$g}|dkr0g}t ||||||¡}ˆ|ƒS)N)rrn)rgr/rhrjrkri)ÚfacrrrnŠs z5ASTValidatorTests._check_arguments..argumentsr1)rgzmust have Load context)rhr{)rkz"more positional defaults than argsr|)riz3length of kwonlyargs is not the same as kw_defaults)rgrkr6rÚy)rhri)NNNNNN)rÚargr_rÓrqr‘)r%rÕZcheckrnrgr)rÕrÚ_check_arguments‰s$ z"ASTValidatorTests._check_argumentsc Cs®t gdggdg¡}t d|ggd¡}| |d¡t d|t ¡gt dt ¡¡gd¡}| |d¡t d|t ¡ggt dt ¡¡¡}| |d¡dd„}| ||j¡dS)Nr1zempty body on FunctionDefzmust have Load contextcSst d|t ¡ggd¡S)Nr1)rÚ FunctionDefÚPass)rgrrrrÕ²sz+ASTValidatorTests.test_funcdef..fac)rrnrÙr"rÚr_rÓrØ)r%r6rrÕrrrÚ test_funcdef¨s    zASTValidatorTests.test_funcdefc Csœd dd„}| |t dt ¡¡gdd¡| |t dt dt ¡¡¡gdd¡| |gdd¡| |dgdd ¡| |t dt ¡¡gd d¡dS) NcSsH|dkr g}|dkrg}|dkr*t ¡g}|dkr6g}t d||||¡S)NZmyclass)rrÚÚClassDef)Úbasesr²rQÚdecorator_listrrrÚcls·s  z,ASTValidatorTests.test_classdef..clsr1)rÝzmust have Load context)r²)rQzempty body on ClassDefzNone disallowed)rÞ)NNNN)r"rr_rÓÚkeyword)r%rßrrrÚ test_classdef¶s "zASTValidatorTests.test_classdefc CsJ| t g¡d¡| t dg¡d¡| t t dt ¡¡g¡d¡dS)Nzempty targets on DeletezNone disallowedr1zmust have Del context)r"rÚDeleter_r‘)r%rrrÚ test_deleteËszASTValidatorTests.test_deletec Cs| t gt d¡¡d¡| t dgt d¡¡d¡| t t dt ¡¡gt d¡¡d¡| t t dt ¡¡gt dt ¡¡¡d¡dS)Nr{zempty targets on AssignzNone disallowedr1zmust have Store contextrÖzmust have Load context)r"rÚAssignrqr_r‘rÓ)r%rrrÚ test_assignÑs"zASTValidatorTests.test_assignc Cspt t dt ¡¡t ¡t dt ¡¡¡}| |d¡t t dt ¡¡t ¡t dt ¡¡¡}| |d¡dS)Nr1rÖzmust have Store contextzmust have Load context)rÚ AugAssignr_r‘rr"rÓ)r%ZaugrrrÚtest_augassignÚs  z ASTValidatorTests.test_augassignc CsÜt dt ¡¡}t dt ¡¡}t ¡}| t ||gg¡d¡| t t dt ¡¡||gg¡d¡| t |t dt ¡¡|gg¡d¡t t dt ¡¡¡}| t |||gg¡d¡| t |||g|g¡d¡dS)Nr1rÖzempty body on Forzmust have Store contextzmust have Load context)rr_rÓr‘rÚr"r^rŠ)r%r1rÖÚprrrrÚtest_forâs  zASTValidatorTests.test_forc Cs~| t t d¡gg¡d¡| t t dt ¡¡t ¡gg¡d¡| t t d¡t ¡gt t dt ¡¡¡g¡d¡dS)Nr{zempty body on Whiler1zmust have Load context)r"rÚWhilerqr_rÓrÚrŠ)r%rrrÚ test_whileïs "zASTValidatorTests.test_whilec Csº| t t d¡gg¡d¡t t dt ¡¡t ¡gg¡}| |d¡t t d¡t t dt ¡¡¡gg¡}| |d¡t t d¡t ¡gt t dt ¡¡¡g¡}| |d¡dS)Nr{zempty body on Ifr1zmust have Load context)r"rÚIfrqr_rÓrÚrŠ)r%rMrrrÚtest_if÷s  ( zASTValidatorTests.test_ifcCs°t ¡}| t g|g¡d¡t t d¡d¡}| t |gg¡d¡t t dt ¡¡d¡}| t |g|g¡d¡t t d¡t dt ¡¡¡}| t |g|g¡d¡dS)Nzempty items on Withr{zempty body on Withr1zmust have Load contextzmust have Store context) rrÚr"ÚWithÚwithitemrqr_rÓr‘)r%rèrMrrrÚ test_withszASTValidatorTests.test_withcCspt dt d¡¡}| |d¡t t dt ¡¡d¡}| |d¡t t d¡t dt ¡¡¡}| |d¡dS)Nr{z!Raise with cause but no exceptionr1zmust have Load contextr|)rÚRaiserqr"r_rÓ)r%ÚrrrrÚ test_raise s   zASTValidatorTests.test_raisec Csxt ¡}t ggg|g¡}| |d¡t t t dt ¡¡¡ggg|g¡}| |d¡t |gggg¡}| |d¡t |gg|g|g¡}| |d¡t |gt ddg¡ggg¡}| |d¡t t dt ¡¡d|g¡g}| t |g|gg¡d¡t dd|g¡g}t |g|t t dt ¡¡¡g|g¡}| |d¡t |g||gt t dt ¡¡¡g¡}| |d¡dS)Nzempty body on Tryr1zmust have Load contextz-Try has neither except handlers nor finalbodyz%Try has orelse but no except handlerszempty body on ExceptHandlerrÖ)rrÚÚTryr"rŠr_rÓÚ ExceptHandler)r%rèrrrrrÚtest_trys$ &    ( (zASTValidatorTests.test_tryc CsT| t t dt ¡¡d¡d¡t t dt ¡¡t dt ¡¡¡}| |d¡dS)Nr1zmust have Load contextrÖ)r"rÚAssertr_rÓr‘)r%ZassrtrrrÚ test_assert's zASTValidatorTests.test_assertcCs| t g¡d¡dS)Nzempty names on Import)r"rÚImport)r%rrrÚ test_import.szASTValidatorTests.test_importcCs>t dt dd¡gd¡}| |d¡| t dgd¡d¡dS)Nr1iÖÿÿÿzNegative ImportFrom levelrzempty names on ImportFrom)rrYrÊr")r%ZimprrrÚtest_importfrom1s z!ASTValidatorTests.test_importfromcCs| t g¡d¡dS)Nzempty names on Global)r"rÚGlobal)r%rrrÚ test_global6szASTValidatorTests.test_globalcCs| t g¡d¡dS)Nzempty names on Nonlocal)r"rZNonlocal)r%rrrÚ test_nonlocal9szASTValidatorTests.test_nonlocalcCs&t t dt ¡¡¡}| |d¡dS)Nr1zmust have Load context)rrŠr_rÓr")r%rrrrÚ test_expr<szASTValidatorTests.test_exprc Csœt t ¡g¡}| |d¡t t ¡t d¡g¡}| |d¡t t ¡t d¡dg¡}| |d¡t t ¡t d¡t dt ¡¡g¡}| |d¡dS)Nzless than 2 valuesr{r|zNone disallowedr1zmust have Load context)rÚBoolOpÚAndr!rqr_rÓ)r%rrrrÚ test_boolop@s   &zASTValidatorTests.test_boolopcCs,t t ¡t dt ¡¡¡}| |d¡dS)Nr1zmust have Load context)rÚUnaryOpÚNotr_rÓr!)r%ÚurrrÚ test_unaryopJszASTValidatorTests.test_unaryopc CsNt gdggdg¡}| t |t dt ¡¡¡d¡dd„}| ||j¡dS)Nr1zmust have Load contextcSst |t dt ¡¡¡S)Nr1)rÚLambdar_r‘)rgrrrrÕRsz*ASTValidatorTests.test_lambda..fac)rrnr!rr_rÓrØ)r%r6rÕrrrÚ test_lambdaNs zASTValidatorTests.test_lambdacCs\t dt ¡¡}t dt ¡¡}x6|||f|||f|||ffD]}| tj|Žd¡q>WdS)Nr1rÖzmust have Load context)rr_r‘rÓr!ZIfExp)r%ÚlÚsrgrrrÚ test_ifexpVs"zASTValidatorTests.test_ifexpcCsRt gt dt ¡¡g¡}| |d¡t t dt ¡¡gdg¡}| |d¡dS)Nr1zsame number of keys as valueszNone disallowed)rÚDictr_r‘r!)r%rµrrrÚ test_dict\s zASTValidatorTests.test_dictcCs<| t dg¡d¡t t dt ¡¡g¡}| |d¡dS)NzNone disallowedr1zmust have Load context)r!rÚSetr_rÓ)r%r rrrÚtest_setbszASTValidatorTests.test_setc Csü| |gƒd¡t t dt ¡¡t dt ¡¡gd¡}| ||gƒd¡t t dt ¡¡t dt ¡¡gd¡}| ||gƒd¡t dt ¡¡}t dt ¡¡}t ||dgd¡}| ||gƒd¡t ||t dt ¡¡gd¡}| ||gƒd¡dS)Nz comprehension with no generatorsr1rzmust have Store contextzmust have Load contextrÖzNone disallowed)r!rr`r_r‘rÓ)r%rÕÚgr1rÖrrrÚ_check_comprehensiongsz&ASTValidatorTests._check_comprehensioncsbt t dt ¡¡t dt ¡¡gd¡}| ˆt dt ¡¡|gƒd¡‡fdd„}| |¡dS)Nr1rzmust have Load contextcsˆt dt ¡¡|ƒS)Nr1)rr_rÓ)Zgens)rÕrrÚwrap{sz,ASTValidatorTests._simple_comp..wrap)rr`r_rÓr‘r!r)r%rÕrrr)rÕrÚ _simple_compvs  zASTValidatorTests._simple_compcCs| tj¡dS)N)rrÚListComp)r%rrrÚ test_listcompszASTValidatorTests.test_listcompcCs| tj¡dS)N)rrÚSetComp)r%rrrÚ test_setcomp‚szASTValidatorTests.test_setcompcCs| tj¡dS)N)rrÚ GeneratorExp)r%rrrÚtest_generatorexp…sz#ASTValidatorTests.test_generatorexpcCs¦t t dt ¡¡t dt ¡¡gd¡}t t dt ¡¡t dt ¡¡|g¡}| |d¡t t dt ¡¡t dt ¡¡|g¡}| |d¡dd„}| |¡dS)NrÖrèrr1zmust have Load contextcSs.t dt ¡¡}t dt ¡¡}t |||¡S)Nr1rÖ)rr_r‘ÚDictComp)ÚcompsÚkrrrrÚfactory‘sz0ASTValidatorTests.test_dictcomp..factory)rr`r_rÓr‘rr!r)r%rÚcrrrrÚ test_dictcompˆs  zASTValidatorTests.test_dictcompc Cs@| t t dt ¡¡¡d¡| t t dt ¡¡¡d¡dS)Nr1zmust have Load)r!rZYieldr_rÓZ YieldFrom)r%rrrÚ test_yield—szASTValidatorTests.test_yieldcCs¼t dt ¡¡}t |t ¡gg¡}| |d¡t |t ¡gt d¡t d¡g¡}| |d¡t t d¡t ¡g|g¡}|j|dtdt |t ¡gt d¡g¡}|j|dtddS) Nr1zno comparatorsr|rzz,different number of comparators and operandsZblahz non-numeric)rÐ)rr_r‘ÚCompareZInr!rqr0)r%rwÚcomprrrÚ test_compare›s $ zASTValidatorTests.test_comparecCs¶t dt ¡¡}t dt ¡¡g}t dt dt ¡¡¡g}t t dt ¡¡||¡}| |d¡t |dg|¡}| |d¡t dt dt ¡¡¡g}t |||¡}| |d¡dS)Nr1rÖÚwÚzzmust have Load contextzNone disallowed)rr_r‘ràr¬rÓr!)r%r°rgr²ZcallZ bad_keywordsrrrÚ test_call¦s  zASTValidatorTests.test_callcCshGdd„dtƒ}Gdd„dtƒ}Gdd„dtƒ}x2dd|ƒ|ƒ|ƒfD]}|jt |¡d td qFWdS) Nc@s eZdZdS)z*ASTValidatorTests.test_num..subintN)rr3r4rrrrÚsubint³sr'c@s eZdZdS)z,ASTValidatorTests.test_num..subfloatN)rr3r4rrrrÚsubfloatµsr(c@s eZdZdS)z.ASTValidatorTests.test_num..subcomplexN)rr3r4rrrrÚ subcomplex·sr)Ú0Zhelloz non-numeric)rÐ)r Úfloatrr!rrqr0)r%r'r(r)ÚobjrrrÚtest_num²s zASTValidatorTests.test_numcCs.t t dt ¡¡dt ¡¡}| |d¡dS)Nr1rÖzmust have Load context)rÚ Attributer_rÓr‘r!)r%ÚattrrrrÚtest_attribute¼sz ASTValidatorTests.test_attributec Cs$t t dt ¡¡t t d¡¡t ¡¡}| |d¡t dt ¡¡}t |t t dt ¡¡¡t ¡¡}| |d¡t dt ¡¡}xH|ddfd|dfdd|ffD](}tj|Ž}| t ||t ¡¡d¡q¤Wt  g¡}| t ||t ¡¡d¡t  t |¡g¡}| t ||t ¡¡d¡dS)Nr1r{zmust have Load contextrÖzempty dims on ExtSlice) rÚ Subscriptr_rÓZIndexrqr‘r!ÚSliceZExtSlice)r%Úsubr1r rgÚslrrrÚtest_subscriptÀs      "   z ASTValidatorTests.test_subscriptc CsNt t t dt ¡¡t ¡¡gt ¡¡}t |gt d¡¡}| |d¡dS)Nr1r|zmust have Store context) rÚListÚStarredr_r‘rÓrärqr")r%rwÚassignrrrÚ test_starredÒs  zASTValidatorTests.test_starredcCs@| |dgt ¡ƒd¡| |t dt ¡¡gt ¡ƒd¡dS)NzNone disallowedr1zmust have Load context)r!rr‘r_rÓ)r%rÕrrrÚ _sequenceØszASTValidatorTests._sequencecCs| tj¡dS)N)r:rr6)r%rrrÚ test_listÝszASTValidatorTests.test_listcCs| tj¡dS)N)r:rÚTuple)r%rrrÚ test_tupleàszASTValidatorTests.test_tuplecCs| t d¡d¡dS)Nr|z&singleton must be True, False, or None)r!rÚ NameConstant)r%rrrÚtest_nameconstantãsz#ASTValidatorTests.test_nameconstantc CsŠtj tj¡}dd„t |¡Dƒ}| ddg¡xT|D]L}tj ||¡}t|ddd}|  ¡}WdQRXt  ||¡}t ||dƒq6WdS) NcSsg|]}| d¡r|‘qS)z.py)Úendswith)rÚfnrrrrèsz;ASTValidatorTests.test_stdlib_validates..ztest/test_grammar.pyztest/test_unpack_ex.pyròzutf-8)Úencodingr;) ÚosÚpathÚdirnamerÚ__file__ÚlistdirÚextendÚjoinÚopenÚreadrPrJ)r%ZstdlibZtestsrWrAÚfpÚsourcer[rrrÚtest_stdlib_validatesæs  z'ASTValidatorTests.test_stdlib_validates)Nr;)N)N)2rr3r4r”r[r!r"rvrØrÛrárãrårçrérërírðrórörørúrûrýrþrÿrrrr r rrrrrrrr r#r&r-r0r5r9r:r;r=r?rNrrrrrÏqs\          rÏc@s^eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z e j dd„ƒZ dd„ZdS)Ú ConstantTestsz$Tests on the ast.Constant node type.cCs\t d¡}|jdj}tj|d}t ||¡||jd_t|ddƒ}i}t||ƒ|dS)Nzx = 123r)r'zr;r1)rrPrQr'ÚConstantr¨rJr;)r%r'Útreer¥Únew_noderÌrÍrrrÚcompile_constantõs       zConstantTests.compile_constantc Cs<| t¡}| dddg¡WdQRX| t|jƒd¡dS)Nrlr*r{z%got an invalid type in Constant: list)r-r0rSr+r rŒ)r%rrrrÚtest_validations  zConstantTests.test_validationc CsNxHdddtdtƒfD]2}|j|d| |¡}| ||¡WdQRXqWdS)NFTó)Úconst)ÚEllipsisÚ frozensetrIrSZassertIs)r%rVr'rrrÚtest_singletons s zConstantTests.test_singletonsc Cs’d}tdhƒ}x$tdƒD]}|df}t|dhƒ}qWddddd td ƒtd ƒ||f }x:|D]2}|j|d | |¡}| ||¡WdQRXqXWdS) N)rlrlr{r*é{gÀ^@yÀ^@ÚunicodesbytesrrX)r')rXrorrIrSr+)r%Z nested_tupleZnested_frozensetrÈÚvaluesr'rrrrÚ test_valuess    zConstantTests.test_valuesc Cszt d¡}|jdjd}tjdd}t ||¡||jdjd<| t¡}t|ddƒWdQRX|  t |j ƒd¡dS)Nzx = 1rrl)r'Ústringr;z6expression which can't be assigned to in Store context) rrPrQZtargetsrPr¨r-r”rJr+r rŒ)r%rQÚtargetÚ new_targetrrrrÚtest_assign_to_constants     z%ConstantTests.test_assign_to_constantcCs t d¡}| t |¡d¡dS)Nz'docstring' x = 1Z docstring)rrPr+r¼)r%rQrrrr½,s z ConstantTests.test_get_docstringcCs>t|ddƒ}g}x(t |¡D]}|jdkr| |j¡qW|S)Nzr;Z LOAD_CONST)rJÚdisZget_instructionsÚopnamerZargval)r%rQÚcoÚconstsZinstrrrrÚget_load_const0s   zConstantTests.get_load_constc CsÆdddddddddg }d  d d „|Dƒ¡}|d 7}| tdf¡t |¡}| | |¡|¡xRt|j|ƒD]B\}}t |tj ƒsŒt t  |¡ƒ‚tj |d }t ||j¡||_qjW| | |¡|¡dS)NTFé|g@y@r[sbytes)rlr*r{Ú cSsg|]}d |¡‘qS)zx={!r})Úformat)rrVrrrrEsz1ConstantTests.test_load_const..z x = ...)r')rIrHrWrrPr+rfrHrQr räÚAssertionErrorrœrPr¨r')r%rerÌrQr8rVrRrrrÚtest_load_const:s*     zConstantTests.test_load_constcCslt d¡}|jdj}tjdd}t ||j¡||_tjdd}t ||j¡||_| t  |¡d¡dS)Nz1 + 2ré )r'y4@y$@4@) rrPrQr'rPr¨rwryr+r¡)r%rQZbinopZnew_leftZ new_rightrrrrÃWs    zConstantTests.test_literal_evalN)rr3r4Ú__doc__rSrTrYr]rar½rfrršrkrÃrrrrrOòs rOcCsžtdkr dStjdd…dgkr’xdtdftdftdffD]J\}}t|dƒx,|D]$}t |d|¡}td t |ƒfƒqRWtd ƒq8Wtd ƒt ‚t   ¡dS) NÚ__main__rlz-gr;r<r=z _results = [r@z%r,ú]zmain()) rÚsysÚargvrBrDrFÚprintrrPrÚ SystemExitÚunittestÚmain)Z statementsrAZ statementrQrrrrufs   ruru)rŠ)rlr)r>)rlrN)rŠ)rlr)r­)rlrzmodule docstringrÙ)rlrrrn)rÚ)rlé )rŠ)rlrv)r­)rlrvzfunction docstring)r×)rlrÁr6N)rÚ)rlrl)rq)rlér)rÚ)rlé )r×)rlrmrgN)rÚ)rlé)r×)rlrwÚkwargsN)rÚ)rlé)r×)rlrvrN)r×)rlryrN)r×)rlérµN)r×)rlérN)r×)rlé#rgN)r×)rlé)rN)rq)rlé+r()r×)rlé1rzN)rq)rlé rl)r>)rléNr6)rlé)r‘r )rlé)rŠ)rlé:)r­)rlr†z doc for f()rÜÚC)rÚ)rlrw)rŠ)rlrv)r­)rlrvzdocstring for class C)r_)rlrwÚobject)r‘)ZReturn)rlrw)rq)rlérlrâ)r_)rlr|r)ZDelrä)r_)rlrr)rÓ)rq)rlr|rl)ræ)rlr)r_)rlrr)rÓ)r)rq)rlrzrlr^)r_)rlr|r)rÓ)r_)rlrvr)r‘)rÚ)rlr‚rê)r_)rlrÁr)r‘rì)r_)rlr{r)r‘)rÚ)rlrzrî)rï)r_)rlrzr1)r‘)r_)rlrlrÖ)rÓ)rÚ)rlé )rï)r_)rlrŠr%)r‘)r_)rléÚq)rÓ)rÚ)rlérñr¬)rlrÁ)r_)rlrÁrŸ)r‘)r­)rlrƒr^rô)rÚ)r*r*rõ)r{r)r_)r{rmrŸ)r‘)rÚ)r|r*)r÷)rlr)r_)rlrmr)r‘Nrù)rÊrpNrYrp)rÊrNrü)rŠ)rlr)rq)rlrrl)rÚ)rlr)ZBreak)rlr‚)ZContinue)rlr‚r<)rlr|)r_)rlr|r6)rÓ)r_)rlrÁr)rÓ)rÓ)r_)rlr‚r)r‘rŠr)rlrl)rlr*)r_)rlr*r6)r‘)r_)rlr|r)r‘r`)rlr‚)r_)rlr‚r6)rÓ)r_)rlrŠr)rÓ)r_)rlr‹r)r‘r)rlrx)r_)rlrxr6)rÓ)r_)rlryr)rÓ)r_)rlér)r‘)r*r|)r{r|)r_)r{r|ÚAa)r‘)r_)rzrmÚBb)r‘)rwr|)r_)rwr|r)rÓ)r_)rlr|r)rÓ)r_)rlrlZCc)r‘r)r_)rlrlr6)r‘)r_)rlrzr)r‘)r_)rlr‚r$)rÓ)r_)rlrƒr1)r‘)r_)rlr|rŽ)rÓ)r_)rlérè)r‘)r_)rlé r)r‘)r_)rlr‚r)rÓ)r_)rlrŠr$)rÓ)r_)rlr‹r1)r‘r)r_)rlrlrò)r‘)r_)rlrmr )rÓ)r_)rlrxr1)r‘)r_)rlr{r)r‘)rlrm)r_)rlrvrŽ)rÓ)r_)rlryr1)r‘ZAsyncFunctionDef)rŠ)r*rl)r­)r*rlzasync function)r{rlZAwait)r{rm)r_)r{rmZ something)r‘ZAsyncFor)r*rl)r_)r*r‚r)rÓ)r_)r*rƒrM)r‘)rŠ)r*é)rq)r*r“rl)rŠ)r{rm)rq)r{rmr*Z AsyncWith)rï)r_)r*rxr6)r‘)r_)r*r{r)rÓ)rŠ)r*rŽ)rq)r*rŽrl)rq)rlrlr*)rlr{)rq)rlrÁr*)rq)rlrxr{rr7)rq)rlr{rl)rq)rlrlr{)r*r*)r_)r*r*rM)r‘)r_)r*ryr)rÓ)r_)r*r“r)r‘rlrÒ)rŠ)rlr)r~)rlr)rq)rlrrl)r)rq)rlr*r*)rÔ)r>)rlrNrÔr)r)r_)rlrr6)r‘)r_)rlrÁr)r‘)rÔ)r~)rlr)r_)rlrr6)r‘)r)r_)rlr|r)r‘)rÔ)r)rlr)r)r_)rlr|r)r‘r)r>)rlrmN)rq)rlr*rl)rq)rlr|r*)r>)rlrlN)rq)r*rÁrl)rq)r|rlr*)r_)rlrmr)rÓ)r_)rlrxr)r‘)r_)rlr{rµ)r‘r!)rq)rlrrl)ZLt)rq)rlrwr{)r_)rlrr)r‘)r7)rlrl)r_)rlr‚rµ)r‘)r‘)ràr)rq)rlrwr{)ràN)r_)rlr‰r)r‘)rÔ)rq)rlrrl)rÔ)r­)rlrr^)rÔ)r.)rlr)r_)rlrr6)r‘r)r‘)rÔ)r1)rlr)r_)rlrr6)r‘)r2)r_)rlr*r)r‘)r_)rlr|r)r‘N)r‘)rÔ)r_)rlrr)r‘)rq)rlrlrl)rq)rlr{r*)rq)rlrzr{)rq)rlr*r*)rq)rlr|r{)r.)rlr)r.)rlr)r.)rlr)r_)rlrr6)r‘r)r‘r)r‘rµ)r‘)r1)rlrw)r.)rlrw)r_)rlrwr6)r‘r)r‘)r2)rq)rlrxrl)rq)rlryr*N)r‘)rrbrCrprtr7rÉrrrBrDrFZTestCaserr›rÏrOrurCrErGrrrrÚsD  hDt&&(*&&L&$"::::20$.>62**D ""$&