o noi{@sddlmZmZddlZddlmZddlmZmZm Z ddl m Z ddl m Z GdddeZd ed ejfd d ZGd ddeZdS))AnyCallableN)parse_statement) CSTNodeTestDummyIndentedBlockparse_statement_as) CodeRange) data_providerc@seZdZegeedeee fddejedeee fe eddde ddd ejedeee fe d d dejedeee fe e ed d de ddd eedeje ede edfdee fddeedeje ede ede ede ed fdee fddeedejej ededdej ededdfdee fddeedejej ede ededdej ede ed eddfdee fddeedeje ede edfdee fddeedeje ede ede ede ed fdee fddeedejej ededdej ededdfdee fddeedejej ede ededdej ede ed eddfdee fddeedeje ede edej ede ed eddfdee fddeedejej ededde edfd ee fd!deedejej ede ededde ede ed ej ed"e eded#dfd ee fd$deedeje ed%e ed&fej ede ededde ede ed ej ed"e eded#dfd'ee fd(deedejej ed%ed)dej ed&ed*dfej ede ededde ede ed ej ed"e eded#dfd'ee fd+deedeje ed%e ed&ej ed,ed)dej ed-ed*dfej ede ededde ede ed ej ed"e eded#dfd'ee fd.deedeje ed/d0ee fd1deedeje ed/e edej ede ededde ede ed ej ed"e eded#dfd2ee fd3deedeje ed%e ed&ej ed,ed)dej ed-ed*dfe ed/e edej ede ededde ede ed ej ed"e eded#dfd4ee fd5deedeje ed6d7ee fd8deedeje ed/e ed6d9ee fd:deedeje ed/e ede ed6e ed d9ee fd;deedeje edfe edfd<ee fd=deedeje edfe edd>ee fd?deedeje edfe edfd@ee fdAdeedeje edfe eddBee fdCdeedeee feedfdDdeedeee feeedeedeedEffdFdeedeee feeedeedEfeeedeedGffdHe dIdJd edKeedeee feedfdLdedKeedeeee ffeedfdMe dNdOd ejedeee fejedPdQfdRdSe dTdUd ejeejedVdQfejedWejededWedXdYdZfejed[dQfej edXd\edXededXedXeej edXedXedd]edWee fd^ d_e d`dad ejedWejededWedXdYdZdbe ddcd eedeee feejejedddddfdedeedeee feeedfeedgfdhdeedeee feejdefefdifdjdeje ed%e ed&ej ed,ed)dej ed-ed*dfe ed/e edej ede ededde ede ed ej ed"e eded#dfd4dke ddld ej ed,dmed)dndoe ddpd ej ed,dqedWdrdse ddtd ejejdduejej ejdvdudwgdejeje gdxgdxejejdKdudydzd{e dd|d Rd}ed~dfddZ eejejdduejgej dqedfej ej!edgduedmdddee fde"ddejejdduejgej dqedfej ej!ej#edej$e%ed e&dej$ej%eddqedmde&de$e%e'gdduedmdddee fde"ddejedeee fe(e)e*edfdde"dejedeee fe(e)ej*eded ej+edWd\dej&edWd\e)e,edej&edWd\e)e-edfdde"dejedeee fej(ej)ej*ededej+edWejefdddde&edWde)e-ededXe&edXfdedXdde"dfd}ed~dfddZ.edddfdddfdddfdddfdddfdddfdddfdddfdddfdddfdddfdddfdddfdddffde/gej0fde1d~dfddZ2dS)FunctionDefCreationTestfoodef foo(): pass nodecodestrreturnsdef foo() -> str: pass )r)r)rrexpected_position asynchronousasync def foo(): pass intrrasync def foo() -> int: pass )rbarbazparamsdef foo(bar, baz): pass "def foo(bar: str, baz: int): pass "one"default5$def foo(bar = "one", baz = 5): pass .def foo(bar: str = "one", baz: int = 5): pass )posonly_paramsdef foo(bar, baz, /): pass %def foo(bar: str, baz: int, /): pass 'def foo(bar = "one", baz = 5, /): pass 1def foo(bar: str = "one", baz: int = 5, /): pass &def foo(bar: str, baz: int = 5): pass  kwonly_params#def foo(*, bar = "one", baz): pass biz"two"?def foo(*, bar: str = "one", baz: int, biz: str = "two"): pass firstsecond)r!r1Ndef foo(first, second, *, bar: str = "one", baz: int, biz: str = "two"): pass 1.01.5Zdef foo(first = 1.0, second = 1.5, *, bar: str = "one", baz: int, biz: str = "two"): pass thirdfourthidef foo(first, second, third = 1.0, fourth = 1.5, *, bar: str = "one", baz: int, biz: str = "two"): pass r!star_argdef foo(*params): pass r@r1Jdef foo(*params: str, bar: str = "one", baz: int, biz: str = "two"): pass r!r@r1tdef foo(first, second, third = 1.0, fourth = 1.5, *params: str, bar: str = "one", baz: int, biz: str = "two"): pass kwparams star_kwargdef foo(**kwparams): pass r@rH#def foo(*params, **kwparams): pass -def foo(*params: str, **kwparams: int): pass r*r!def foo(bar, /, baz): pass )r*r@def foo(bar, /, *baz): pass )r*r1def foo(bar, /, *, baz): pass )r*rHdef foo(bar, /, **baz): pass @bar def foo(): pass '123'!@bar(baz, '123') def foo(): pass '456'(@bar('123') @baz('456') def foo(): pass )r)rWz z @bar def foo(): pass z% @bar def foo(): pass ))rW # leading commentcomment leading_lines"# leading comment def foo(): pass )rYr)rYrX# What an amazing decorator  funcwhitespace_after_funcwhitespace_before_argswhitespace_after_at decorator# What a great functionwhitespace_afterwhitespace_before_indicatorwhitespace_after_indicator annotation r` decoratorslines_after_decoratorsrwhitespace_after_defnamewhitespace_after_namewhitespace_before_paramsr!rwhitespace_before_colonbodyf # What an amazing decorator @ bar ( ) # What a great function async def foo ( ) -> str : pass )r)r}%z @ bar ( ) )r )rfz@bar()() def foo(): pass abz@a + b def foo(): pass lparrparz@(bar) def foo(): pass zdfirst, second, third = 1.0, fourth = 1.5, *params: str, bar: str = "one", baz: int, biz: str = "two")rd)starr&z third = 1.0)rr}*)rwhitespace_after_starz* third)r)valueparam1)rw)r{) last_line)rwr!r{ryzdef foo( param1): pass )rWkwargsreturnNcKs&d|vr |d|jdi|dS)N native_only)pop validate_nodeselfrrrT/home/ubuntu/.local/lib/python3.10/site-packages/libcst/_nodes/tests/test_funcdef.py test_valids\ z"FunctionDefCreationTest.test_validrp)rrwrr)r!r@)rwr!r{zdef foo(*a: *b): pass )rparserrtuplecommaTs)rrr)rslicez'def foo(*a: *tuple[int,*Ts,...]): pass T)type_parameterszdef foo[T](): pass )rrr)boundcolonKWz#def foo[T: int, *Ts, **KW](): pass T) empty_linesindent)whitespace_beforern)paramrPS)r whitespace_after_type_parametersz'def foo[T : str ,** PS ,] (): pass cKs|jdi|dS)Nr)rrrrrtest_valid_nativesz)FunctionDefCreationTest.test_valid_nativecCs6ttjdtftfdtttfS)Nr r)cst FunctionDefName LeftParen RightParen ParametersSimpleStatementSuitePassrrrrs z FunctionDefCreationTest.zCannot have parens around NamecCs6tjtdtttftjtdddS)Nr rrmr)rrrrrr AsynchronousSimpleWhitespacerrrrrszone space after AsynchronouscCs.tjtdtttftddS)Nr r)rv)rrrrrrrrrrrrs z one space between 'def' and namecCs<ttdtjtjtdtddttfS)Nr r)equalrG)rrrrParam AssignEqualrrrrrrrsz3Must have a default when specifying an AssignEqual.cC8ttdtjtjtddddttfS)Nr rz***rrGrrrrrrrrrrrr z0Must specify either '', '\*' or '\*\*' for star.c CsNttdtjtjtdtddttdfdttfS)Nr rr$r%r rrrrr SimpleStringrrrrrrrszCCannot have param without defaults following a param with defaults.c CsPttdtjtjtdtddfttdfdttfS)Nr rr$r%rMrrrrrrscC,ttdtjtdttfS)Nr r?)rrrr ParamStarrrrrrrr z9Must have at least one kwonly param if ParamStar is used.cCr)Nr ) posonly_ind)rrrr ParamSlashrrrrrrrrz;Must have at least one posonly param if ParamSlash is used.cC:ttdtjtjtdddfdttfS)Nr rrrr rrrrrrs zExpecting a star prefix of ''c CsBttdtjtjtdtdddfdttfS)Nr rr$rr&rr rrrrrrs cCr)Nr rrrr0rrrrrrscCr)Nr r**rr?rrrrrrrzExpecting a star prefix of '\*'cCr)Nr rrrrGrrrrrrrz!Expecting a star prefix of '\*\*'get_node expected_recCs|||dSN)assert_invalid)rrrrrr test_invalid}sz$FunctionDefCreationTest.test_invalid)3__name__ __module__ __qualname__r rrrrrr AnnotationrrrrIntegerFloat DecoratorCallArgr IndentedBlockSimpleStatementLine EmptyLineCommentrBinaryOperationAddrrParenthesizedWhitespacerrStarredElementr SubscriptSubscriptElementIndexCommaEllipsisTypeParameters TypeParamTypeVarColon TypeVarTuple ParamSpecrrCSTNoderrrrrrr s" !-8 F Vl z   / A Qh<E_  %1=IU^q  #0 $Yf p ~ +09  [    "&,                r rrcCs.tj|tjddd}t|tjstd|S)Nz3.8python_version)configz=This function is expecting to parse compound statements only!)rrPartialParserConfig isinstanceBaseCompoundStatement ValueError)r statementrrr_parse_statement_force_38s  rc)@seZdZeeedeee fdfejedeee fej ede ddddfejedeee fe dd fejedeee fe ej ed e ddd d feedejej ed deje dddej edddfdee fdfeedejej ed ej ede dddeje dddej edej ed e ddddfdee fdfeedejej ed eeddeje dddej edeedddfdee fdfeedejej ed ej ede ddeeddeje dddej edej ed e ddeedddfdee fdfeedejej ed ej ede ddddeje dddej edej ed e ddeedddfdee fdfeedejeej ed eeddeje dddej eddddfd ee fd!feedejeej ed ej ede ddeeddeje dddej edej ed e ddddeje dddej ed"ej ede ddeed#ddfd ee fd$feedejej ed%dddeje ddd&ej ed'dddeje ddd&feej ed ej ede ddeeddeje dddej edej ed e ddddeje dddej ed"ej ede ddeed#ddfd(ee fd)feedejej ed%deed*deje ddd+ej ed'deed,deje ddd+feej ed ej ede ddeeddeje dddej edej ed e ddddeje dddej ed"ej ede ddeed#ddfd(ee fd-feedejej ed%dddeje ddd&ej ed'dddeje ddd&ej ed.deed*deje ddd+ej ed/deed,deje ddd+feej ed ej ede ddeeddeje dddej edej ed e ddddeje dddej ed"ej ede ddeed#ddfd(ee fd0feedejej ed1ddd2d3d4ee fd5feedejej ed1ej ede dddd2eje dddej ed ej ede ddeeddeje dddej edej ed e ddddeje dddej ed"ej ede ddeed#ddfd ee fd6feedejej ed%dddeje ddd&ej ed'dddeje ddd&ej ed.deed*deje ddd+ej ed/deed,deje ddd+fej ed1ej ede dddd2eje dddej ed ej ede ddeeddeje dddej edej ed e ddddeje dddej ed"ej ede ddeed#ddfd(ee fd7feedejej ed8ddd9d3d:ee fd;feedejej ed1ddd2eje ddd&ej ed8ddd9d3d<ee fd=feedejej ed1ej ede dddd2eje dddej ed8ej ed e dddd9dd<ee fd>feedeee feed fd?feedeee feeed ejeddeje ddd@ejedAddBffdCfeedeee feeed ejedAddBfeeedejedDddBffdEfejedeee fejedFdGfdHdIfejeejedJdGfeje dejed e de dKdLdMfejedNdGfej e dKde dKede dKe dKeej e dKe dKeddOe dee fdP dQffdRejdSedTdfdUdVZeeedejej ed deje dddej eddeje dddfedWee fdXdYeedejej ed deje dddej eddeje dddfeje dddWee fdZd[d\eedejej ed ej ede dddeje dddej edej ed e dddeje dddfedWee fd]dYeedejej ed ededeje ddd^ej edededeje ddd^fedWee fd_dYeedejej ed ej ede ddededeje ddd^ej edej ed e ddededeje ddd^fedWee fd`dYeedejej ed deje dddfejeje dddaej edddfdbee fdcdYeedejej ed deje dddfejeje dddaej edd2dddee fdedYeedejej ed deje dddfejeje dddaejeje dddaej edddfdfee fdgdYeedejej ed deje dddfejeje dddaej edd9ddhee fdidYf dRejdSedjedTdfdkdlZedmedndod[dpdmedqdod[dpdmedrdod[dpdmedsdodtdpfdjedTdfdudvZ edSdwidSdxidSdyidSdzidSd{idSd|idSd}idSd~idSdidSdif djedTdfddZ!dS)FunctionDefParserTestr r rrcrrrrrrrrrrrm)rrrrr r"r#r$)rr&rrr')rr&rr(r)N)r&rrr/rrBr2r3r4r5r6)rrr&rrr7rDr8r9)rrrr&rrr:r;r<r=r>r!r)rrr&rr?rArCrErFrrGrIrJrKrLrR)keywordrrS)rrTrUrVr\r]r_rarbrdrerirlrorsr|rrrcC|||tdSr)rr)rrrrrrr!s}z FunctionDefParserTest.test_valid)r*rr+r zdef foo(bar, baz, / ): pass T)rrrr,)r&rrrr-r.r)r*rr!rN)r*rr@rO)r*rr@r1rP)r*rrHrQrcKrr)rr)rrrrrrr test_valid_38sz#FunctionDefParserTest.test_valid_38zasync def foo(): passz3.7r)rrexpect_successz3.6z3.5z3.3FcKs*|dds |d|jdi|dS)NrTz+parse errors are disabled for native parserr)getskipTest assert_parsesrrrr test_versionss  z#FunctionDefParserTest.test_versionszA[:*b]zA[*b:]zA[*b:*b]z A[*(1:2)]zA[*:]zA[:*]zA[**b]zdef f(x: *b): passzdef f(**x: *b): passzx: *bcKs|jdi|dtddS)NF)rrr)rrrrrrtest_parse_errorsz&FunctionDefParserTest.test_parse_error)"rrrr rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr s      $ #("' .27 BFK DHM   (VZ_  7;@   ( _ch     $| #(   #',    r)typingrrlibcstrrlibcst._nodes.tests.baserrrlibcst.metadatarlibcst.testing.utilsr r rrrrrrrrs"