a Ø þdG$ã@sddZddlZddlZddlmZddlmZGdd„dejƒZGdd„dejƒZ e d kr`e  ¡dS) 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úh/var/www/html/stable-diffusion-webui/venv/lib/python3.9/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-s    ýz'PytreeUnwrapperTest.testSimpleFileScopecCs.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)r.r/r0ú# c2r1rrr!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-r1r3)rrrr4ú# 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©r.rr0r1©rrrr3)Úelifrr0r4)rrrÚelser0r6)Ú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 rr8r1r9)ÚwhileÚtr0ér4)Újrr)Úkrrr!r%rrr ÚtestNestedCompoundTwoLevel‘s       úz.PytreeUnwrapperTest.testNestedCompoundTwoLevelcCs@t d¡}t |¡}| |dgd¢fddgfdgd¢fg¡dS)Nz? while x > 1: # c1 # c2 x = 1 r)r>rú>rr0r3r1r4rr!r%rrr ÚtestSimpleWhile¤s    ýz#PytreeUnwrapperTest.testSimpleWhilecCs~t d¡}t |¡}| |dddgfddgfdddgfddgfdddgfddgfdddgfddgfdd dgfddgfg ¡dS) NzŠ try: pass except: pass except: pass else: pass finally: pass rÚtryr0r1Ú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©Údefr/r(rr*r0r3r1r4Úreturnrr!r%rrr ÚtestSimpleFuncdefÌs    ý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 rrKr1r4rMr)rLÚbarr(r*r0r6z# c4r!r%rrr ÚtestTwoFuncDefsÙs     úz#PytreeUnwrapperTest.testTwoFuncDefscCs@t d¡}t |¡}| |dgd¢fddgfdgd¢fg¡dS)Nz: class Klass: # c1 # c2 p = 1 r)ÚclassZKlassr0r3r1r4)Úprrr!r%rrr ÚtestSimpleClassDefís    ýz&PytreeUnwrapperTest.testSimpleClassDefcCs8t d¡}t |¡}| |dgd¢fdddgfg¡dS)Nz" def f(): return 37 r©rLÚfr(r*r0r1rMZ37r!r%rrr ÚtestSingleLineStmtInFuncús     þz,PytreeUnwrapperTest.testSingleLineStmtInFunccCsFt d¡}t |¡}| |ddgfddgfdgd¢fddgfg¡dS)NzR # Comment #1 # Comment #2 def f(): pass rz # Comment #1z # Comment #2rTr1rGr!r%rrr ÚtestMultipleCommentss   üz(PytreeUnwrapperTest.testMultipleCommentscCs.t d¡}t |¡}| |dgd¢fg¡dS)NzW a = [ 'a', 'b', 'c', # hello world ] r) Úarú[z'a'ú,z'b'rZz'c'rZz # hello worldú]r!r%rrr ÚtestSplitListWithComments  z,PytreeUnwrapperTest.testSplitListWithCommentN)Ú__name__Ú __module__Ú __qualname__rr'r+r,r2r5r7r=rCrErJrNrPrSrVrWr\rrrr rs"        rc@s,eZdZdd„Zdd„Zdd„Zdd„Zd S) ÚMatchBracketsTestc Cs®g}|D]"}dd„|jDƒ}|r| |¡qt|ƒD]t\}}||}|sb|D]}| |d¡qLq4|D]@\} } | || d|| d¡| || d|| d¡qfq4dS)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.r1rN)rrÚ enumerateZ assertIsNoner) rrrrrrÚindexZ bracket_listrZ 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 ))r@é)éé )éé©r"r#rr$rcr%rrr ÚtestFunctionDef?s   þz!MatchBracketsTest.testFunctionDefcCs.t d¡}t |¡}| |dgdggg¡dS)Nz@ @bar() def foo(a, b, c): pass )r@é©r@érir%rrr Ú testDecoratorJs  ýzMatchBracketsTest.testDecoratorcCs*t d¡}t |¡}| |dggg¡dS)Nz1 class A(B, C, D): pass rlrir%rrr Ú testClassDefWs   þzMatchBracketsTest.testClassDefN)r]r^r_rcrjrnrorrrr r`"s  r`Ú__main__) Ú__doc__r"ZunittestZ yapf.pytreerZ yapftestsrZYAPFTestrr`r]Úmainrrrr Ús   A