o DDi6@s|ddlmZddlZddlmZmZddlmZddlm Z m Z m Z ddl m Z ddlmZddlmZGd d d e ZdS) )AnyNparse_statementPartialParserConfig) MaybeSentinel) CSTNodeTestDummyIndentedBlockparse_statement_as) is_native) CodeRange) data_providerc@sreZdZUdZeed<eee e e dfe e fdeedddeje e e dfe e fedd d d d eje d eeeje e e dfe e fedfedddd d ee e e d e e e dfe e fddd eej e e d ejeddde e e dfe e fded ee e e dee dfe e fded ee e e dejedededdfe e fded edee e e dfe e fddedddedee e e dfeee ffd dedd!deje e e dfe e fejed"d#fd$d%eed&d'deje e e deje ded(ed(d)fe e fed(ed(d*d+eedd,deje ej e derdneferdnefd-fe e ferenej erenej edd.d/eedd0dejej e e d ejej!ej"ejdd1dej#dd1d2gd3ejd4d1d5ddej e e dedfe e fejeddejedd6d-d7erWendedd8df d9e$d:dfd;d<Z%ed=d d>d?d@d dAd?dBd dCd?dDd dEd?dFd dEd?fd9e$d:dfdGdHZ&edIe'dJdKd3dLdIe'dMdKdNdLfd9e$d:dfdOdPZ(dSdQdRZ)dS)TWithTestimaxDiff context_mgrzwith context_mgr(): pass )r)r)nodecodeparserexpected_position) asynchronouszasync with context_mgr(): pass cCt|tdddS)Nz3.7python_versionconfigrrrV/var/www/Datamplify/venv/lib/python3.10/site-packages/libcst/_nodes/tests/test_with.py( zWithTest.)rrrfooz4async def foo(): async with context_mgr(): pass cCr)Nz3.6rrrrrrrr=r barzwith foo(), bar(): pass N whitespace_aftercommactxz with context_mgr() as ctx: pass r)whitespace_after_aswhitespace_before_aszwith context_mgr()as(): pass z z with context_mgr(): pass )r)rz% with context_mgr(): pass ) z# leading comment)comment) leading_linesz+# leading comment with context_mgr(): pass )r.r)r.rz )r+r*)whitespace_after_withwhitespace_before_colonz%with context_mgr() as ctx : pass )r$lparrpar)r6r7r2zwith(context_mgr()): pass )r)value) whitespacer0newlineTz ) first_line empty_linesindent last_linewhitespace_beforez$with ( foo(), bar(), ): pass )r.kwargsreturncK|jdi|dSNr) validate_nodeselfrCrrr test_valids_zWithTest.test_validc Cs tdtttffSrF)cstWith IndentedBlockSimpleStatementLinePassrrrrrsz0A With statement must have at least one WithItem)get_node expected_rec CsFttjttdtjtdddfttt ffS)Nr!r#r$r&) rKrLWithItemCallNameCommaSimpleWhitespacerMrNrOrrrrrszJThe last WithItem in an unparenthesized With cannot have a trailing comma.cCs6tjtttdfttftddS)Nrr))r2)rKrLrRrSrTSimpleStatementSuiterOrVrrrrr s z/Must have at least one space after with keywordcC<tjtttdfttftdtdS)Nrr))r2r6) rKrLrRrSrTrWrOrV LeftParenrrrrr z;Do not mix concrete LeftParen/RightParen with MaybeSentinelcCrX)Nrr))r2r7) rKrLrRrSrTrWrOrV RightParenrrrrrrZcKrErF)assert_invalidrHrrr test_invalids4zWithTest.test_invalidzwith a, b: passz3.1r)rrexpect_successz3.0FcKs0tr|dds|d|jdi|dS)Nr^Tz+parse errors are disabled for native parserr)r getskipTest assert_parsesrHrrr test_versions*s zWithTest.test_versionscCstjtjttdtjtddtjttdtdfttftj t ddtj t ddd}t g}| ||ddS) Nr!r$r&r"r#r@r5zwith ( foo(), bar(), ): pass )rKrLrRrSrTrUParenthesizedWhitespacerWrOrYrVr[Module assertEqual code_for_node)rIrmodulerrrtest_adding_parens=s"   zWithTest.test_adding_parens)rDN)*__name__ __module__ __qualname__rint__annotations__r rKrLrRrSrTrWrOrr Asynchronous FunctionDef ParametersrMrUrVAsNameTuplerrN EmptyLineCommentr rYr[rDEFAULTrcTrailingWhitespaceNewlinerrJr]r rbrhrrrrr s            ^   3r )typingrlibcstrKrrlibcst._maybe_sentinelrlibcst._nodes.tests.baserrr libcst._parser.entrypointsr libcst.metadatar libcst.testing.utilsr r rrrrs