o ?n0jG$ã@shdZddlZddlZddlmZddlmZGdd„dejƒZGdd„dejƒZ e d kr2e  ¡dSdS) z Tests for yapf.pytree_unwrapper.éN)Ú pytree_utils)Úyapf_test_helperc@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#S)$ÚPytreeUnwrapperTestcCs>g}|D]}dd„|jDƒ}| |j|f¡q| ||¡dS)zåCheck that the given LogicalLines match expectations. Args: llines: list of LogicalLine list_of_expected: list of (depth, values) pairs. Non-semantic tokens are filtered out from the expected values. cSsg|] }|jtjvr|j‘qS©)ÚnamerÚNONSEMANTIC_TOKENSÚvalue©Ú.0Úftrrúf/home/wildlama/miniconda3/envs/lam_a2e/lib/python3.10/site-packages/yapftests/pytree_unwrapper_test.pyÚ $s  þz:PytreeUnwrapperTest._CheckLogicalLines..N)ÚtokensÚappendÚdepthÚ assertEqual)ÚselfÚllinesÚlist_of_expectedÚactualÚllineÚfiltered_valuesrrr Ú_CheckLogicalLinessþz&PytreeUnwrapperTest._CheckLogicalLinescCs@t d¡}t |¡}| |dgd¢fddgfdgd¢fg¡dS)Nz1 x = 1 # a comment y = 2 r©Úxú=Ú1ú # a comment©ÚyrÚ2©ÚtextwrapÚdedentrÚParseAndUnwrapr©rÚcoderrrr ÚtestSimpleFileScope-ó     ýz'PytreeUnwrapperTest.testSimpleFileScopecCó.t d¡}t |¡}| |dgd¢fg¡dS)Nz$ y = (1 + x) r)rrú(rú+rú)r!r%rrr ÚtestSimpleMultilineStatement:s    ÿz0PytreeUnwrapperTest.testSimpleMultilineStatementcCs8t d¡}t |¡}| |dgd¢fdgd¢fg¡dS)Nz. x = 1 # a comment y = 2 r)rrrrrr!r%rrr ÚtestFileScopeWithInlineCommentDs     þz2PytreeUnwrapperTest.testFileScopeWithInlineCommentcCsBt d¡}t |¡}| |dgd¢fdgd¢fdgd¢fg¡dS)Nz5 if foo: x = 1 y = 2 r©ÚifÚfooú:érrr!r%rrr Ú testSimpleIfOs      ýz PytreeUnwrapperTest.testSimpleIfc CsJt d¡}t |¡}| |ddgfdgd¢fdgd¢fdgd¢fg¡dS)NzE # c1 if foo: # c2 x = 1 y = 2 rú# c1)r0r1r2ú# c2r3rrr!r%rrr ÚtestSimpleIfWithComments\s      üz,PytreeUnwrapperTest.testSimpleIfWithCommentsc CsRt d¡}t |¡}| |dgd¢fddgfdgd¢fddgfdgd¢fg¡dS) NzX if foo: # c1 x = 1 # c2 # c3 y = 2 rr/r3r5)rrrr6ú# c3rr!r%rrr ÚtestIfWithCommentsInsideks      ûz,PytreeUnwrapperTest.testIfWithCommentsInsidec Csht d¡}t |¡}| |dgd¢fdgd¢fdgd¢fdgd¢fddd gfdd gfdgd ¢fg¡dS) Nzu if x: x = 1 # c1 elif y: # c2 y = 1 else: # c3 z = 1 r©r0rr2r3©rrrr5)Úelifrr2r6)rrrÚelser2r8)Úzrrr!r%rrr ÚtestIfElifElse|s         ùz"PytreeUnwrapperTest.testIfElifElsec Cs^t d¡}t |¡}| |dgd¢fdgd¢fdgd¢fddgfdgd ¢fdgd ¢fg¡dS) Nzj if x: x = 1 # c1 while t: # c2 j = 1 k = 1 rr:r3r;)ÚwhileÚtr2ér6)Újrr)Úkrrr!r%rrr ÚtestNestedCompoundTwoLevel‘s        úz.PytreeUnwrapperTest.testNestedCompoundTwoLevelcCó@t d¡}t |¡}| |dgd¢fddgfdgd¢fg¡dS)Nz? while x > 1: # c1 # c2 x = 1 r)r@rú>rr2r5r3r6rr!r%rrr ÚtestSimpleWhile¤r(z#PytreeUnwrapperTest.testSimpleWhilecCs~t d¡}t |¡}| |dddgfddgfdddgfddgfdddgfddgfdddgfddgfdd dgfddgfg ¡dS) NzŠ try: pass except: pass except: pass else: pass finally: pass rÚtryr2r3ÚpassÚexceptr=Úfinallyr!r%rrr Ú testSimpleTry±s        öz!PytreeUnwrapperTest.testSimpleTrycCs@t d¡}t |¡}| |dgd¢fddgfdddgfg¡dS)Nz< def foo(x): # c1 # c2 return x r©Údefr1r*rr,r2r5r3r6Úreturnrr!r%rrr ÚtestSimpleFuncdefÌr(z%PytreeUnwrapperTest.testSimpleFuncdefc Cs\t d¡}t |¡}| |dgd¢fddgfdddgfdgd¢fdd gfdddgfg¡dS) Nzq def foo(x): # c1 # c2 return x def bar(): # c3 # c4 return x rrNr3r6rPr)rOÚbarr*r,r2r8z# c4r!r%rrr ÚtestTwoFuncDefsÙs       úz#PytreeUnwrapperTest.testTwoFuncDefscCrF)Nz: class Klass: # c1 # c2 p = 1 r)ÚclassZKlassr2r5r3r6)Úprrr!r%rrr ÚtestSimpleClassDefír(z&PytreeUnwrapperTest.testSimpleClassDefcCs8t d¡}t |¡}| |dgd¢fdddgfg¡dS)Nz" def f(): return 37 r©rOÚfr*r,r2r3rPZ37r!r%rrr ÚtestSingleLineStmtInFuncús     þz,PytreeUnwrapperTest.testSingleLineStmtInFunccCsFt d¡}t |¡}| |ddgfddgfdgd¢fddgfg¡dS)NzR # Comment #1 # Comment #2 def f(): pass rz # Comment #1z # Comment #2rWr3rJr!r%rrr ÚtestMultipleCommentss    üz(PytreeUnwrapperTest.testMultipleCommentscCr))NzW a = [ 'a', 'b', 'c', # hello world ] r) Úarú[z'a'ú,z'b'r]z'c'r]z # hello worldú]r!r%rrr ÚtestSplitListWithComments  z,PytreeUnwrapperTest.testSplitListWithCommentN)Ú__name__Ú __module__Ú __qualname__rr'r-r.r4r7r9r?rErHrMrQrSrVrYrZr_rrrr rs$        rc@s,eZdZdd„Zdd„Zdd„Zdd„Zd S) ÚMatchBracketsTestc Cs®g}|D]}dd„|jDƒ}|r| |¡qt|ƒD]:\}}||}|s1|D] }| |d¡q&q|D] \} } | || d|| d¡| || d|| d¡q3qdS)a*Check that the tokens have the expected matching bracket. Arguments: llines: list of LogicalLine. list_of_expected: list of (index, index) pairs. The matching brackets at the indexes need to match. Non-semantic tokens are filtered out from the expected values. cSs"g|] }|jtjvr||jf‘qSr)rrrZmatching_bracketr rrr r /s  þz.r3rN)rrÚ enumerateZ assertIsNoner) rrrrrrÚindexZ bracket_listrÚ open_bracketZ close_bracketrrr Ú_CheckMatchingBrackets$s& ÿ €ÿ þúz(MatchBracketsTest._CheckMatchingBracketscCs,t d¡}t |¡}| |gd¢gg¡dS)NzD def foo(a, b=['w','d'], c=[42, 37]): pass ))rBé)éé )éé©r"r#rr$rgr%rrr ÚtestFunctionDef?s   þz!MatchBracketsTest.testFunctionDefcCs.t d¡}t |¡}| |dgdggg¡dS)Nz@ @bar() def foo(a, b, c): pass )rBé©rBérmr%rrr Ú testDecoratorJs   ýzMatchBracketsTest.testDecoratorcCs*t d¡}t |¡}| |dggg¡dS)Nz1 class A(B, C, D): pass rprmr%rrr Ú testClassDefWs   þzMatchBracketsTest.testClassDefN)r`rarbrgrnrrrsrrrr rc"s  rcÚ__main__) Ú__doc__r"ZunittestZ yapf.pytreerZ yapftestsrZYAPFTestrrcr`Úmainrrrr Ús    A ÿ