o DDiȟ@sddlmZmZddlZddlmZddlmZddlm Z ddl m Z GdddeZ d e d ejfd d ZGd ddeZdS))CallableOptionalN)parse_expression) CSTNodeTest) CodeRange) data_providerc@seZdZeeeeddfeejee dee dfdedddeejee dee dfej e dd d edd deejee dee dfd edd feejeje de ddeje deddfd eddfeejee deje deddfd eddfeejeje de ddee dfdeddfeejee dee dfeje de ddee deje de ddfdeddfeejeje de ddeje de ddfeje de ddee deje de ddfdeddfeejee dee deje de ddeje de ddfeje de ddee deje de ddfdedded d!feejee d"d#edd$feejee d"eje de ddee deje de ddfd%edd&feejee dee deje de ddeje de ddfee d"eje de ddee deje de ddfd'edd(feejee d)d*edd+feejee d"ee d)d,edd-fejeje dd fe d.eeje dd edeje dd/fd0d1ed2d3ff 4d_d5ejd6ed7eed8d4fd9d:Zed;d<d=fd>d<d?fd@d<dAfdBd<dAfdCd<dAfdDd<dEfdFd<dGfdHd<dIfdJd<dKfdLd<dMfdNd<dMfdOd<dMfdPd<dQfdRd<dSfdTd<dUfdVd<dUfdWd<dUfdXd<dUfdYd<dUfdZd<dUffd[egejfd\ed8d4fd]d^Zd4S)`LambdaCreationTest5 lambda: 5barbazposonly_paramslambda bar, baz, /: 5nodecode whitespace_afterr posonly_indzlambda bar, baz, / : 5paramslambda bar, baz: 5"one"defaultlambda bar = "one", baz = 5: 5lambda bar, baz = 5: 5 kwonly_paramslambda *, bar = "one", baz: 5firstsecondbiz"two")rr!9lambda first, second, *, bar = "one", baz, biz = "two": 51.01.5Elambda first = 1.0, second = 1.5, *, bar = "one", baz, biz = "two": 5thirdfourthTlambda first, second, third = 1.0, fourth = 1.5, *, bar = "one", baz, biz = "two": 5)r)r.Trstar_arglambda *params: 5r1r!0lambda *params, bar = "one", baz, biz = "two": 5rr1r!Zlambda first, second, third = 1.0, fourth = 1.5, *params, bar = "one", baz, biz = "two": 5kwparams star_kwarglambda **kwparams: 5r1r9lambda *params, **kwparams: 5 whitespace_before)lparwhitespace_after_lambdarcolonbodyrpar( lambda : 5 ))r.)r. NrrpositionreturncCs|j|||ddS)N)expected_position) validate_nodeselfrrrHrNX/var/www/Datamplify/venv/lib/python3.10/site-packages/libcst/_nodes/tests/test_lambda.py test_validsuzLambdaCreationTest.test_validcC2tjtjttdfdtdtfdS)Nargrr )r@)cstLambda ParametersParamNameInteger LeftParenrNrNrNrO  zLambdaCreationTest.zleft paren without right parencCrQ)NrRrr )rD)rSrTrUrVrWrX RightParenrNrNrNrOrZr[zright paren without left parencC2tjtjttdfdtdtddS)NrRr r rArSrTrUrVrWrXSimpleWhitespacerNrNrNrOrZr[zat least one space after lambdacCr])NrRrr r^r_r`rNrNrNrOrZ!r[cCs<tjtjtjtdtddfdtdtddS)NrRr rrr^r_r`rNrNrNrOrZ)scCs.ttjtjtdtddtdS)Nr )equalr8r )rSrTrUrVrW AssignEqualrXrNrNrNrOrZ3s z3Must have a default when specifying an AssignEqual.cC*ttjtjtddddtdS)Nr z***starr8r rSrTrUrVrWrXrNrNrNrOrZ<z0Must specify either '', '\*' or '\*\*' for star.c Cs@ttjtjtdtddttdfdtdS)Nr rrrr rSrTrUrVrW SimpleStringrXrNrNrNrOrZCszCCannot have param without defaults following a param with defaults.cCsttjtdtdS)Nr0r )rSrTrU ParamStarrXrNrNrNrOrZQsz9Must have at least one kwonly param if ParamStar is used.cC,ttjtjtdddfdtdS)Nr *rerr rgrNrNrNrOrZWszExpecting a star prefix of ''cCs4ttjtjtdtdddfdtdS)Nr rrm)rrfrr rirNrNrNrOrZ^s cCrl)Nr rmrer r rgrNrNrNrOrZms cCrd)Nr **rer0r rgrNrNrNrOrZvrhzExpecting a star prefix of '\*'cCrd)Nr rmrer8r rgrNrNrNrOrZ}rhz!Expecting a star prefix of '\*\*'c CBtjtjtjtdttddfdtdtddS)NrRstr annotationr r r^r_rSrTrUrVrW AnnotationrXrarNrNrNrOrZz*Lambda params cannot have type annotationsc Cro)NrRrprqrr r^r_rsrNrNrNrOrZruc CsJtjtjtjtdtdttddfdtdtddS)NrRr rp)rrrrr^r_)rSrTrUrVrWrXrtrarNrNrNrOrZs c C@tjtjtjtdttdddtdtddS)NrRrprqr0r r^r_rsrNrNrNrOrZc Cro)NrRrprqr r r^r_rsrNrNrNrOrZruc Crv)NrRrprqr8r r^r_rsrNrNrNrOrZrwget_node expected_recCs|||dSN)assert_invalid)rMrxryrNrNrO test_invalidsXzLambdaCreationTest.test_invalidrz)__name__ __module__ __qualname__rrSrTrUrXrVrW ParamSlashrarjFloatrrYColonr\CSTNoderprrPrr|rNrNrNrOrs0           s                 U rrrIcCstj|tjdddS)Nz3.8)python_version)config)rSrPartialParserConfig)rrNrNrO_parse_expression_force_38s rc@s eZdZeeeeddfejejeje ddej e dddeje ddd fd ede dd d fejejeje de d e dej e dddeje dede ddfd ede dd dfejejeje ddej e dddeje dede ddfd ede dd dfejejeeje de d e dej e dddeje ddd fdede dd dfejejeje ddej e dddeje ddej e dddfeeje de d e dej e dddeje ddej e dddeje de de ddfdede dd dfejejeje dede dej e dddeje dede dej e dddfeeje de d e dej e dddeje ddej e dddeje de de ddfdede dd dfejejeje ddej e dddeje ddej e dddeje dede dej e dddeje dede dej e dddfeeje de d e dej e dddeje ddej e dddeje de de ddfdede dd dfejejeje d d!d d"ede dd d#fejejeje d d!ej e dddeje de d e dej e dddeje ddej e dddeje de de ddfdede dd d$fejejeje ddej e dddeje ddej e dddeje dede dej e dddeje dede dej e dddfeje d d!ej e dddeje de d e dej e dddeje ddej e dddeje de de ddfdede dd d%fejejeje d&d'd d(ede dd d)fejejeje d d!ej e dddeje d&d'd d*ede dd d+fejeje ddfeeje d,e dd-edeje dd.fd/d0fejejeje d1d!d d"ede dd d2fejejeje d3d'd d(ede dd d4fejejeje e de dd5eje d1dd gdede dd d6fejejeed7ed8eje d9e d9e dd:d;dejd?ed@eedAd=fdBdCZeejejeje ddej e dddeje ddej e dddfedDede dd dEdFf =dId>ejd?ed@eedAd=fdGdHZd=S)JLambdaParserTestr r r r^rr)rfcommar rerr_rr)rrbrfr)rrbrfrrr3r"r#r$r%r&r5r'r(r)r*r+r,r-rrmr0r2r4r6r7rnr8r:r;r<r=)r?rr>)r@rrBrCrDrEargszlambda*args: 5kwargszlambda**kwargs: 5)rzlambda*,args: 5rN)rrCrB_)targetiterr?)eltfor_inz[lambda:()for _ in _]NrrrHrIcC|||t|dSrz)rKrrLrNrNrOrPskzLambdaParserTest.test_validrrrcCrrz)rKrrLrNrNrO test_valid_38s"zLambdaParserTest.test_valid_38rz)r}r~rrrSrTrUrXrVrWCommararjrcrkrrYrr\ListCompTupleCompForrrprrrPrrrNrNrNrOrs    +/4  /38 " =AF  #( " AEJ       i  r)typingrrlibcstrSrlibcst._nodes.tests.baserlibcst.metadatarlibcst.testing.utilsrrrpBaseExpressionrrrNrNrNrOs    T