B 18\=@sddlZddlZddlZddlZddlZddlZddlmZddZddddd 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 GdLdMdMej Z GdNdOdOej ZGdPdQdQej ZGdRdSdSej ZdTdUZdVdWgfdVdXgfdVdYdZd[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~gfgfdVddZdgdfgfdVdgfdVddZdddggfgfdVddZddyggfgfdVddZddggfgfdVddZdgdgfgfdVddZddgdgfgfdVddZddddggfdfgfdVddZdgdddddgfgggfgfdVddZdgggdgfgfdVdgfdVddZdgfgfdVddZddgdfgfdVddZdEgfgfdVdgfdVdgfdVddZdddggfgfdVddZdddggfgfdVddZddddgdfdddggfgfdVddZddddddgdsfdddddgdfdgdfgffgfdVddZddddddgdsfdddddgdfdgdfgffgfdVddZddddddgdsfdddddgdfdgdfgffgfdVddZddddddgdsfdddddgdfdgdfgffgfdVddZddZdddddgdfddddgdfgffgfdVddZddZdddddddgdfdgdfgffgfdVddZddZdddddgdfgffgfdVddZddZddddddgdfdgdfgffgfdVddZd[d\gdggdgfddddddddggfffggdfgfdVddZd[d\gdggdgfdddddgdgfggdfgfdVddZd[d\gdggdgfdddgdgfggdfgfdVddZdtdZddgdtddgdgfdgffgfdVddZddZddddddgfdsfdgffgfdVddZd[d\gdggdgfddddddddgdfgffggdfgfg/ZddgfgZddddZdddgffdddddZd\gdggdgfdffddtdZdgdgffddtdZggffdddZdgffddtdZdgdgffddddddddgdfgffddddddddgdfgffdddZd d d gdd gffdddZd ddd gddgffdddddddqdZdddgdsffddqdZgdsffdddZd ddgdsffddddddgdsffdddZgdsffdddZddggffgZedS(N)supportcCs|dkst|tttfr|St|tr4dd|DS|jjg}t|drdt|drd||j |j f|j dkrvt |Sx"|j D]}|t t||q~Wt |S)NcSsg|] }t|qS)to_tuple).0errQ/oak/stanford/groups/akundaje/marinovg/programs/Python-3.7.3/Lib/test/test_ast.py szto_tuple..lineno col_offset) isinstancestrintcomplexlist __class____name__hasattrappendr r _fieldstuplergetattr)tresultfrrrr s    rNonez'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 v1passzfor 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)10z'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@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zejd'd(Zd)S)* AST_TestscCst|tjr|jdkrdSt|tjtjtjfrV|j|jf}| ||k|j|jf}xR|jD]H}t ||}t|t rx.|D]}| ||q|Wq^|dk r^| ||q^WdS)N) r astASTrexprstmtZ excepthandlerr r assertTruerr_assertTrueorder)selfZast_nodeZ parent_posZnode_posnamevalueZchildrrrr%s      zAST_Tests._assertTrueorderc Cs|t}||jdd|_||jd||jdd|t |jWdQRX|t tdWdQRXdS)Nr*foobar) r r! assertEqualrr*__dict__ assertRaisesAttributeErrorvararg TypeError)r&xrrrtest_AST_objectss  zAST_Tests.test_AST_objectscCsLGddd}t}||_||j_t|j}~t||dS)Nc@s eZdZdS)z0AST_Tests.test_AST_garbage_collection..XN)r __module__ __qualname__rrrrXsr6) r r!r2aweakrefrefr gc_collect assertIsNone)r&r6r7r9rrrtest_AST_garbage_collections z%AST_Tests.test_AST_garbage_collectionc CsxttdfttdfttdffD]\}}}xt||D]v\}}|jd|d2t|d|t j }| t ||| |dWdQRX|jd||d t|d|WdQRXq4WqWdS) NexecsingleevalZparsing)actioninput?)rrZ compiling)r@rAkind) exec_tests exec_results single_testssingle_results eval_tests eval_resultszipsubTestcompiler PyCF_ONLY_ASTr,rr%)r&rAoutputrCioZast_treerrr test_snippetss zAST_Tests.test_snippetscCs<tdjdjj}||j||j||jdS)Nzx[::]r) r parsebodyr(slicer;upperlowerstep)r&Zslcrrr test_slices  zAST_Tests.test_slicecCs tdjd}||jdS)Nzfrom . import yr)r rRrSr;module)r&Zimrrrtest_from_importszAST_Tests.test_from_importcCs>td}||jdtjd|jd_t|dddS)Nzfrom __future__ import divisionrz __future__ zr=)r rRassertIsInstancerS ImportFromstriprYrL)r&modrrr!test_non_interned_future_from_asts z+AST_Tests.test_non_interned_future_from_astcCs||ttjtj|ttjtj|ttjtj|ttjtj|ttjtj|ttj tjdS)N) r$ issubclassr Forr#Namer"r! comprehensionZGt)r&rrrtest_base_classes s zAST_Tests.test_base_classescCs^xXtjD]J\}}t|tr |dkr |dr |}t|tjr |t|jt q WdS)Nr!r) r r-itemsr typeisupperr!r,rr)r&r'itemr2rrrtest_field_attr_existences  z#AST_Tests.test_field_attr_existencec CsTt}||jd|t |jWdQRXtjtdd}||jddS)N)argsr0 kwonlyargs kw_defaultskwargdefaultsr+)r argumentsr,rr.r/r0range)r&r2rrrtest_argumentss  zAST_Tests.test_argumentscCs t}d|_||jddS)Ni)r Numrr,)r&r2rrrtest_field_attr_writable'sz"AST_Tests.test_field_attr_writablec Cst}||jd|t |jWdQRXtd}||jd|t |jWdQRX|t |jWdQRXtjdd}||jdtjddd}||jd||jd||jd|t tjdd|jt tjdddddS)N)nr)r+)r rro) r rtr,rr.r/rvr r*r1)r&r2rrrtest_classattrs-s$     zAST_Tests.test_classattrscCs(tdg}t|}||j|dS)Nr))r rtModuler,rS)r&rSr2rrr test_moduleHs  zAST_Tests.test_modulec Cst}||jdd|_||jdtd}td}t}t|||}||j|||j|||j |tddd}||jd||jd||j dtjddddd}||jd||jd||j d||j d| t tjdddd|j t tjddddddtjddddd }||jd||jd||j d||j dtjdddd d }||jd dS) N)leftoprightror+r)r )rzr{r|r r)) foobarbaz) r BinOpr,rrrtAddrzr{r|r r.r1)r&r2Zn1Zn3Zaddoprrrtest_nodeclassesMs:  zAST_Tests.test_nodeclassescCst}||jddS)Nr)r ZSubr,r)r&r2rrrtest_no_fieldsyszAST_Tests.test_no_fieldsc Csddl}|g}yddl}||Wntk r8YnXdddg}xX|D]P}xJ|D]B}xsz*AST_Tests.test_pickling..) picklecPickler ImportErrorrDloadsdumpsr,r) r&rZmodsrZ protocolsr^protocolr Zast2rrr test_pickling~s   zAST_Tests.test_picklingc Csbtddd}ttjtjf|f|g}|t}t|ddWdQRX|dt |j dS)Nr+r~)r r zr=zbut got <_ast.expr) dictr rxExprr"r.r1rLassertInr exception)r&posmcmrrrtest_invalid_sums   zAST_Tests.test_invalid_sumc Cs`tttdtg}t||t}t|ddWdQRX| dt |j dS)Nr)zr=zidentifier must be of type str) r rxrrbLoadfix_missing_locationsr.r1rLrr r)r&rrrrrtest_invalid_identitifers   z"AST_Tests.test_invalid_identitiferc CsVtd}d|jdjdj_|t}t|ddWdQRX|dt|j dS)Nzdef f(): yield from g()rzr=zfield value is required) r rRrSr(r. ValueErrorrLrr r)r&Zempty_yield_fromrrrrtest_empty_yield_froms   zAST_Tests.test_empty_yield_fromc Cs>ddl}dd}t|d||ttjdWdQRXdS)NrcWsdS)Nr)rjrrr bad_normalizesz0AST_Tests.test_issue31592..bad_normalize normalizeuϕ) unicodedatar swap_attrr.r1r rR)r&rrrrrtest_issue31592szAST_Tests.test_issue31592N)rr4r5r%r3r<rQrXrZr_rdrirsrurwryrrrrrrr cpython_onlyrrrrrrs(   ,rc@s|eZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZdS)ASTHelpers_TestcCs6td}tdddtj}|t|t|dS)Nz foo(1 + 1)z r=)r rRrLrMr,dump)r&r7brrr test_parses zASTHelpers_Test.test_parsec CsTy ddWnBtk rN|t}tdWdQRX||jjYnXdS)Nrorz'\U') Exceptionr. SyntaxErrorr literal_evalassertIsNotNoner __context__)r&rrrrtest_parse_in_errors   z#ASTHelpers_Test.test_parse_in_errorcCsLtd}|t|d|tj|ddd|tj|ddddS) Nzspam(eggs, "and cheese")zModule(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)]))r rRr,r)r&noderrr test_dumps  zASTHelpers_Test.test_dumpcCsBtjddd}ttd|jj|j_|tj|ddddS)Nz1 + 1r?)moder+T)rzExpression(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)))r rR copy_locationrtrSr|r,r)r&srcrrrtest_copy_locationsz"ASTHelpers_Test.test_copy_locationc Csftd}|jtttdttdgg| |t || tj |ddddS)Nz write("spam")spamZeggsT)ra|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)])) r rRrSrrCallrbrStrr,rr)r&rrrrtest_fix_missing_locationss  z*ASTHelpers_Test.test_fix_missing_locationscCs|tjddd}|tj|dd||tj|dddtjddd}|tj|jdd|j|tj|ddddS) Nz1 + 1r?)rr~)rvT)rzExpression(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)))r rRr,Zincrement_linenorrS)r&rrrrtest_increment_linenosz%ASTHelpers_Test.test_increment_linenocCsHtjddd}tt|j}||djd||ggddS)Nzfoo()r?)rfuncfoo)keywordsrj)r rRrZ iter_fieldsrSr,popid)r&rdrrrtest_iter_fieldssz ASTHelpers_Test.test_iter_fieldscCstjddd}|ttt|jdt|j}|t|jd|t|j d|t|j d|t t|ddS) Nzspam(23, 42, eggs='leek')r?)rrrr)z(keyword(arg='eggs', value=Str(s='leek'))) r rRr,lenrZiter_child_nodesrSnextrrvr)r&riteratorrrrtest_iter_child_nodess z%ASTHelpers_Test.test_iter_child_nodescCstd}|t|dtd}|t|jddtd}|t|jddtd}|t|jdddS)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)r rRr, get_docstringrS)r&rrrrtest_get_docstring s     z"ASTHelpers_Test.test_get_docstringcCs.|ttdtd}|t|td}|t|td}|t|jdtd}|t|jdtd}|t|jdtd}|t|jdtd}|t|jdtd }|t|jdtd }|t|jddS) Nzx = "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")r;r rrRrS)r&rrrrtest_get_docstring_nones&         z'ASTHelpers_Test.test_get_docstring_nonecCs0|tddddg|tdddi|tdd |td dddh|td d |ttjd |tdd|tdd|tdd|tdd|tdd|tdd|ttdd|ttjd|ttjd|ttjddS)Nz [1, 2, 3]ror+r~z {"foo": 42}rr)z(True, False, None))TFNz {1, 2, 3}zb"hi"shizfoo()6z+6z-6iz3.25g @z+3.25z-3.25g z-0.0z++6z+Truez2+3)r,r rr.rrepr)r&rrrtest_literal_eval3s z!ASTHelpers_Test.test_literal_evalcCs>|tdd|tdd|tdd|tdd|td d |td d |td d|tdd|tdd|tdd|tdd|tdd|tdd |ttjd|ttjd|ttjd|ttjd|ttjddS)NZ6jy@z-6jyz6.75jy@z-6.75jyz3+6jy@@z-3+6jy@z3-6jy@z-3-6jyz 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,r rr.r)r&rrrtest_literal_eval_complexEs$z)ASTHelpers_Test.test_literal_eval_complexc Csbtjdtjddgddddg}t|}|t}t|ddWdQRX|dt|j dS)Ntimesleep)r')rYnameslevelr r testr=zinvalid integer value: None) r r\aliasrxr.rrLrr r)r&rSr^rrrrtest_bad_integerZs    z ASTHelpers_Test.test_bad_integercCsTtjdtjddgddddg}t|}t|dd}i}t|||d|dS)Nrr)r'r)rYrrr r rr=)r r\rrxrLr=r)r&rSr^codensrrrtest_level_as_nonees     z"ASTHelpers_Test.test_level_as_noneN)rr4r5rrrrrrrrrrrrrrrrrrrs   rc@seZdZd`edddZdaedddZdbdd Zd d Zd d ZddZ ddZ ddZ ddZ ddZ ddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3Zd4d5Zd6d7Zd8d9Zd:d;Zdd?Z!d@dAZ"dBdCZ#dDdEZ$dFdGZ%dHdIZ&dJdKZ'dLdMZ(dNdOZ)dPdQZ*dRdSZ+dTdUZ,dVdWZ-dXdYZ.dZd[Z/d\d]Z0d^d_Z1dS)cASTValidatorTestsNr=)excc CsVd|_|_t|||}t|d|WdQRX|dk rR||t|jdS)Nrz) r r r rr.rLrr r)r&r^msgrrrrrrr^ss    zASTValidatorTests.modcCs&tt|g}|j|||ddS)N)r)r rxrr^)r&rrrr^rrrr"{szASTValidatorTests.exprcCst|g}|||dS)N)r rxr^)r&r#rr^rrrr#s zASTValidatorTests.stmtc CsTtttdtg}||ddttdt}||dddS)Nr2zmust have Load contextr>r?)r InteractiverrbStorer^ Expression)r&rrrrryszASTValidatorTests.test_modulec sdfdd }tdtdtg}|||dd|||dd||tdgdd ||td gd d tdtdtg}|||tdtgd dtdtdttdtdtg}|||dtdtgdddS)NcsL|dkr g}|dkrg}|dkr$g}|dkr0g}t||||||}|S)N)r rq)rjr0rkrmrnrl)facrrrqs z5ASTValidatorTests._check_arguments..argumentsr2)rjzmust have Load context)rkr~)rnz"more positional defaults than argsr)rlz3length of kwonlyargs is not the same as kw_defaults)rjrnr7ry)rkrl)NNNNNN)r argrbrrtr)r&rcheckrqrjr)rr_check_argumentss$ z"ASTValidatorTests._check_argumentsc Cstgdggdg}td|ggd}||dtd|tgtdtgd}||dtd|tggtdt}||ddd}|||jdS)Nr2zempty body on FunctionDefzmust have Load contextcSstd|tggdS)Nr2)r FunctionDefPass)rjrrrrsz+ASTValidatorTests.test_funcdef..fac)r rqrr#rrbrr)r&r7rrrrr test_funcdefs    zASTValidatorTests.test_funcdefc Csd dd}||tdtgdd||tdtdtgdd||gdd||dgdd ||tdtgd ddS) NcSsH|dkr g}|dkrg}|dkr*tg}|dkr6g}td||||S)NZmyclass)r rClassDef)basesrrSdecorator_listrrrclss  z,ASTValidatorTests.test_classdef..clsr2)rzmust have Load context)r)rSzempty body on ClassDefzNone disallowed)r)NNNN)r#r rbrkeyword)r&rrrr test_classdefs "zASTValidatorTests.test_classdefc CsJ|tgd|tdgd|ttdtgddS)Nzempty targets on DeletezNone disallowedr2zmust have Del context)r#r Deleterbr)r&rrr test_deleteszASTValidatorTests.test_deletec Cs|tgtdd|tdgtdd|ttdtgtdd|ttdtgtdtddS)Nr~zempty targets on AssignzNone disallowedr2zmust have Store contextrzmust have Load context)r#r Assignrtrbrr)r&rrr test_assigns"zASTValidatorTests.test_assignc Cspttdtttdt}||dttdtttdt}||ddS)Nr2rzmust have Store contextzmust have Load context)r AugAssignrbrrr#r)r&Zaugrrrtest_augassigns  z ASTValidatorTests.test_augassignc Cstdt}tdt}t}|t||ggd|ttdt||ggd|t|tdt|ggdttdt}|t|||ggd|t|||g|gddS)Nr2rzempty body on Forzmust have Store contextzmust have Load context)r rbrrrr#rar)r&r2rprrrrtest_fors  zASTValidatorTests.test_forc Cs~|ttdggd|ttdttggd|ttdtgttdtgddS)Nr~zempty body on Whiler2zmust have Load context)r#r Whilertrbrrr)r&rrr test_whiles "zASTValidatorTests.test_whilec Cs|ttdggdttdttgg}||dttdttdtgg}||dttdtgttdtg}||ddS)Nr~zempty body on Ifr2zmust have Load context)r#r Ifrtrbrrr)r&rOrrrtest_ifs  ( zASTValidatorTests.test_ifcCst}|tg|gdttdd}|t|ggdttdtd}|t|g|gdttdtdt}|t|g|gddS)Nzempty items on Withr~zempty body on Withr2zmust have Load contextzmust have Store context) r rr#Withwithitemrtrbrr)r&rrOrrr test_withszASTValidatorTests.test_withcCsptdtd}||dttdtd}||dttdtdt}||ddS)Nr~z!Raise with cause but no exceptionr2zmust have Load contextr)r Raisertr#rbr)r&rrrr test_raise s   zASTValidatorTests.test_raisec Csxt}tggg|g}||dtttdtggg|g}||dt|gggg}||dt|gg|g|g}||dt|gtddgggg}||dttdtd|gg}|t|g|ggdtdd|gg}t|g|ttdtg|g}||dt|g||gttdtg}||ddS)Nzempty body on Tryr2zmust have Load contextz-Try has neither except handlers nor finalbodyz%Try has orelse but no except handlerszempty body on ExceptHandlerr)r rTryr#rrbr ExceptHandler)r&rrrrrrtest_trys$ &    ( (zASTValidatorTests.test_tryc CsT|ttdtddttdttdt}||ddS)Nr2zmust have Load contextr)r#r Assertrbrr)r&Zassrtrrr test_assert's zASTValidatorTests.test_assertcCs|tgddS)Nzempty names on Import)r#r Import)r&rrr test_import.szASTValidatorTests.test_importcCs>tdtddgd}||d|tdgdddS)Nr2izNegative ImportFrom levelrzempty names on ImportFrom)r r\rr#)r&imprrrtest_importfrom1s z!ASTValidatorTests.test_importfromcCs|tgddS)Nzempty names on Global)r#r Global)r&rrr test_global6szASTValidatorTests.test_globalcCs|tgddS)Nzempty names on Nonlocal)r#r ZNonlocal)r&rrr test_nonlocal9szASTValidatorTests.test_nonlocalcCs&ttdt}||ddS)Nr2zmust have Load context)r rrbrr#)r&rrrr test_expr<szASTValidatorTests.test_exprc Csttg}||dtttdg}||dtttddg}||dtttdtdtg}||ddS)Nzless than 2 valuesr~rzNone disallowedr2zmust have Load context)r BoolOpAndr"rtrbr)r&rrrr test_boolop@s   &zASTValidatorTests.test_boolopcCs,tttdt}||ddS)Nr2zmust have Load context)r UnaryOpNotrbrr")r&urrr test_unaryopJszASTValidatorTests.test_unaryopc CsNtgdggdg}|t|tdtddd}|||jdS)Nr2zmust have Load contextcSst|tdtS)Nr2)r Lambdarbr)rjrrrrRsz*ASTValidatorTests.test_lambda..fac)r rqr"rrbrr)r&r7rrrr test_lambdaNs zASTValidatorTests.test_lambdacCs\tdt}tdt}x6|||f|||f|||ffD]}|tj|dq>WdS)Nr2rzmust have Load context)r rbrrr"ZIfExp)r&lsrjrrr test_ifexpVs"zASTValidatorTests.test_ifexpcCsRtgtdtg}||dttdtgdg}||ddS)Nr2zsame number of keys as valueszNone disallowed)r Dictrbrr")r&rrrr test_dict\s zASTValidatorTests.test_dictcCs<|tdgdttdtg}||ddS)NzNone disallowedr2zmust have Load context)r"r Setrbr)r&rrrrtest_setbszASTValidatorTests.test_setc Cs||gdttdttdtgd}|||gdttdttdtgd}|||gdtdt}tdt}t||dgd}|||gdt||tdtgd}|||gddS)Nz comprehension with no generatorsr2rzmust have Store contextzmust have Load contextrzNone disallowed)r"r rcrbrr)r&rgr2rrrr_check_comprehensiongsz&ASTValidatorTests._check_comprehensioncsbttdttdtgd}|tdt|gdfdd}||dS)Nr2rzmust have Load contextcstdt|S)Nr2)r rbr)Zgens)rrrwrap{sz,ASTValidatorTests._simple_comp..wrap)r rcrbrrr"r)r&rrrr)rr _simple_compvs  zASTValidatorTests._simple_compcCs|tjdS)N)rr ListComp)r&rrr test_listcompszASTValidatorTests.test_listcompcCs|tjdS)N)rr SetComp)r&rrr test_setcompszASTValidatorTests.test_setcompcCs|tjdS)N)rr GeneratorExp)r&rrrtest_generatorexpsz#ASTValidatorTests.test_generatorexpcCsttdttdtgd}ttdttdt|g}||dttdttdt|g}||ddd}||dS)Nrrrr2zmust have Load contextcSs.tdt}tdt}t|||S)Nr2r)r rbrDictComp)compskrrrrfactorysz0ASTValidatorTests.test_dictcomp..factory)r rcrbrrr"r"r)r&rcr%rrr test_dictcomps  zASTValidatorTests.test_dictcompc Cs@|ttdtd|ttdtddS)Nr2zmust have Load)r"r ZYieldrbrZ YieldFrom)r&rrr test_yieldszASTValidatorTests.test_yieldcCstdt}t|tgg}||dt|tgtdtdg}||dttdtg|g}|j|dtdt|tgtdg}|j|dtddS) Nr2zno comparatorsrr}z,different number of comparators and operandsZblahz non-numeric)r)r rbrCompareZInr"rtr1)r&rzcomprrr test_compares $ zASTValidatorTests.test_comparecCstdt}tdtg}tdtdtg}ttdt||}||dt|dg|}||dtdtdtg}t|||}||ddS)Nr2rwzzmust have Load contextzNone disallowed)r rbrrrrr")r&rrjrcallZ bad_keywordsrrr test_calls  zASTValidatorTests.test_callcCshGdddt}Gdddt}Gdddt}x2dd|||fD]}|jt|d td qFWdS) Nc@s eZdZdS)z*ASTValidatorTests.test_num..subintN)rr4r5rrrrsubintsr0c@s eZdZdS)z,ASTValidatorTests.test_num..subfloatN)rr4r5rrrrsubfloatsr1c@s eZdZdS)z.ASTValidatorTests.test_num..subcomplexN)rr4r5rrrr subcomplexsr20Zhelloz non-numeric)r)r floatrr"r rtr1)r&r0r1r2objrrrtest_nums zASTValidatorTests.test_numcCs.ttdtdt}||ddS)Nr2rzmust have Load context)r Attributerbrrr")r&attrrrrtest_attributesz ASTValidatorTests.test_attributec Cs$ttdtttdt}||dtdt}t|ttdtt}||dtdt}xH|ddfd|dfdd|ffD](}tj|}|t||tdqWt g}|t||tdt t|g}|t||tddS)Nr2r~zmust have Load contextrzempty dims on ExtSlice) r SubscriptrbrZIndexrtrr"SliceZExtSlice)r&subr2rrjslrrrtest_subscripts      "   z ASTValidatorTests.test_subscriptc CsNtttdttgt}t|gtd}||ddS)Nr2rzmust have Store context) r ListStarredrbrrrrtr#)r&rzassignrrr test_starreds  zASTValidatorTests.test_starredcCs@||dgtd||tdtgtddS)NzNone disallowedr2zmust have Load context)r"r rrbr)r&rrrr _sequenceszASTValidatorTests._sequencecCs|tjdS)N)rCr r?)r&rrr test_listszASTValidatorTests.test_listcCs|tjdS)N)rCr Tuple)r&rrr test_tupleszASTValidatorTests.test_tuplecCs|tdddS)Nrz&singleton must be True, False, or None)r"r NameConstant)r&rrrtest_nameconstantsz#ASTValidatorTests.test_nameconstantc Cstjtj}ddt|D}|ddgxT|D]L}tj||}t|ddd}| }WdQRXt ||}t ||dq6WdS) NcSsg|]}|dr|qS)z.py)endswith)rfnrrrrsz;ASTValidatorTests.test_stdlib_validates..ztest/test_grammar.pyztest/test_unpack_ex.pyrzutf-8)encodingr=) ospathdirnamer __file__listdirextendjoinopenreadrRrL)r&stdlibtestsrYrJfpsourcer^rrrtest_stdlib_validatess  z'ASTValidatorTests.test_stdlib_validates)Nr=)N)N)2rr4r5rr^r"r#ryrrrrrrrrrrrrrrrrrrr rrrrrrrrrr!r'r(r+r/r6r9r>rBrCrDrFrHrYrrrrrqs\          rc@s^eZdZdZddZddZddZdd Zd d Zd d Z ddZ e j ddZ ddZdS) ConstantTestsz$Tests on the ast.Constant node type.cCs\td}|jdj}tj|d}t||||jd_t|dd}i}t|||dS)Nzx = 123r)r(zr=r2)r rRrSr(ConstantrrLr=)r&r(treernew_noderrrrrcompile_constants       zConstantTests.compile_constantc Cs<|t}|dddgWdQRX|t|jddS)Nror+r~z%got an invalid type in Constant: list)r.r1r^r,r r)r&rrrrtest_validations  zConstantTests.test_validationc CsNxHdddtdtfD]2}|j|d||}|||WdQRXqWdS)NFT)const)Ellipsis frozensetrKr^assertIs)r&rar(rrrtest_singletons s zConstantTests.test_singletonsc Csd}tdh}x$tdD]}|df}t|dh}qWddddd td td ||f }x:|D]2}|j|d ||}|||WdQRXqXWdS) N)roror~r+{g^@y^@unicodesbytesrrc)r()rcrrrrKr^r,)r&Z nested_tupleZnested_frozensetrvaluesr(rrrr test_valuess    zConstantTests.test_valuesc Csztd}|jdjd}tjdd}t||||jdjd<|t}t|ddWdQRX| t |j ddS)Nzx = 1rro)r(stringr=z6expression which can't be assigned to in Store context) r rRrSZtargetsr[rr.rrLr,r r)r&r\target new_targetrrrrtest_assign_to_constants     z%ConstantTests.test_assign_to_constantcCs td}|t|ddS)Nz'docstring' x = 1Z docstring)r rRr,r)r&r\rrrr,s z ConstantTests.test_get_docstringcCs>t|dd}g}x(t|D]}|jdkr||jqW|S)Nzr= LOAD_CONST)rLdisget_instructionsopnamerargval)r&r\coconstsinstrrrrget_load_const0s   zConstantTests.get_load_constc Csdddddddddg }d d d |D}|d 7}|tdft|}||||xRt|j|D]B\}}t |tj st t |tj |d }t||j||_qjW||||dS)NTF|g@y@rgsbytes)ror+r~ cSsg|]}d|qS)zx={!r})format)rrarrrrEsz1ConstantTests.test_load_const..z x = ...)r()rRrQrbr rRr,rvrJrSr rAssertionErrorrr[rr()r&rtrr\rArar]rrrtest_load_const:s*     zConstantTests.test_load_constcCsltd}|jdj}tjdd}t||j||_tjdd}t||j||_|t |ddS)Nz1 + 2r )r(y4@y$@4@) r rRrSr(r[rrzr|r,r)r&r\ZbinopZnew_leftZ new_rightrrrrWs    zConstantTests.test_literal_evalN)rr4r5__doc__r^r_rerirmrrvrrr{rrrrrrZs rZcCstdkr dStjdddgkrxdtdftdftdffD]J\}}t|dx,|D]$}t|d|}td t |fqRWtd q8Wtd t t dS) N__main__roz-gr=r>r?z _results = [rBz%r,]zmain()) rsysargvrDrFrHprintr rRr SystemExitunittestmain)Z statementsrC statementr\rrrrfs   rrx)r)ror)rG)rorN)r)ror)r)rorzmodule docstringr)rorrrq)r)ro )r)ror)r)rorzfunction docstring)r)rorr7N)r)ror|)rt)ror)r)ro )r)rorprjN)r)ro)r)rorkwargsN)r)ro)r)rorrN)r)rorr&N)r)rorN)r)rorN)r)ro#rjN)r)ro)rN)rt)ro+r))r)ro1rN)rt)ro ro)rG)roNr?)ro)rr)ro)r)ro:)r)rorz doc for f()rC)r)ror)r)ror)r)rorzdocstring for class C)rb)rorobject)r)ZReturn)ror)rt)roror)rb)rorr)ZDelr)rb)rorr)r)rt)rorro)r)ror)rb)rorr)r)r)rt)ror}rora)rb)rorr)r)rb)rorr)r)r)rorr)rb)rorr)rr)rb)ror~r)r)r)ror}r)r)rb)ror}r2)r)rb)ror|r)r)r)ro )r)rb)rorr-)r)rb)roq)r)r)rorr)ror)rb)rorr)r)r)rorrjr)r)r+r+r)r~r)rb)r~rpr)r)r)rr+)r)ror)rb)rorpr)rNr)rrNr\r)rrNr)r)ror)rt)rorro)r)ror)ZBreak)ror)ZContinue)rorrE)ror)rb)rorr7)r)rb)rorr)r)r)rb)rorr&)rrr)roro)ror+)rb)ror+r7)r)rb)rorr)rrc)ror)rb)rorr7)r)rb)rorr)r)rb)rorr&)rr )ror)rb)rorr7)r)rb)rorr)r)rb)ror&)r)r+r)r~r)rb)r~rAa)r)rb)r}rpBb)r)rr)rb)rrr)r)rb)r|rr)r)rb)r|r|ZCc)rr")rb)roror7)r)rb)ror}r)r)rb)rorr,)r)rb)rorr2)r)rb)rorr)r)rb)ror)r)rb)ro r)r)rb)rorr)r)rb)rorr,)r)rb)rorr2)rr)rb)roror)r)rb)rorpr)r)rb)rorr2)r)rb)rorr)r)rorp)rb)rorr)r)rb)rorr2)rZAsyncFunctionDef)r)r+ro)r)r+rozasync function)r~roZAwait)r~rp)rb)r~rpZ something)rZAsyncFor)r+ro)rb)r+rr)r)rb)r+rrO)r)r)r+)rt)r+rro)r)r~rp)rt)r~rpr+Z AsyncWith)r)rb)r+rr7)r)rb)r+rr)r)r)r+r)rt)r+rro)rt)ror|r+)ror~)rt)rorr+)rt)rorr~rr@)rt)ror~ro)rt)ror|r~)r+r+)rb)r+r+rO)r)rb)r+rr)r)rb)r+rr&)rror)r)ror)r)ror)rt)rorro)r)rt)ror+r+)r)rG)rorNrr)r )rb)rorr7)r)rb)rorr)r)r)r)ror)rb)rorr7)r)r)rb)rorr)r)r)r )ror)r )rb)rorr)rr)rG)rorpN)rt)ror+ro)rt)rorr+)rG)roroN)rt)r+rro)rt)rr|r+)rb)rorpr)r)rb)rorr&)r)rb)rorr)rr))rt)rorro)ZLt)rt)rorr~)rb)rorr)r)r@)ror|)rb)rorr)r)r)rr&)rt)rorr~)rN)rb)rorr)r)r)rt)rorr|)r)r)rorrj)r)r7)ror)rb)rorr7)rr)r)r)r:)ror)rb)rorr7)r)r;)rb)ror+r)r)rb)rorr&)rN)r)r)rb)rorr)r)rt)rororo)rt)ror~r+)rt)ror}r~)rt)ror+r+)rt)rorr~)r7)ror)r7)ror)r7)ror)rb)rorr7)rr)rr&)rr)r)r:)ror)r7)ror)rb)rorr7)rr)r)r;)rt)rorro)rt)rorr+N)r)r rorLrrr8rrrrDrFrHTestCaserrrrZrrErGrIrrrrsD  hDt&&(*&&L&$"::::20$.>62**D ""$&