a Ø þd°*ã@sRdZddlZddlZddlmZddlmZGdd„dejƒZedkrNe  ¡dS)zTests for yapf.comment_splicer.éN)Úcomment_splicer)Ú pytree_utilsc@s¨eZdZdd„Zd)dd„Zd*dd„Zd d „Zd d „Zd d„Zdd„Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd „Zd!d"„Zd#d$„Zd%d&„Zd'd(„ZdS)+ÚCommentSplicerTestcCs| |t |¡¡dS)N)Ú assertEqualrÚNodeName)ÚselfÚ expected_typeÚnode©r úg/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/yapftests/comment_splicer_test.pyÚ_AssertNodeTypesz"CommentSplicerTest._AssertNodeTypeNcCsXt |¡dkr.| d|jd¡|jdj}n| d|¡|j}|durT| ||¡dS)NZ simple_stmtÚCOMMENTr)rrr ÚchildrenÚvalueZassertIn)rr Ztext_in_commentZ node_valuer r r Ú_AssertNodeIsComments z'CommentSplicerTest._AssertNodeIsCommentécsJt‡fdd„| ¡DƒƒD]\}}||dkr|Sqtd |¡ƒ‚dS)Ncsg|]}t |¡ˆkr|‘qSr )rr)Ú.0Úc©Únamer r Ú (óz9CommentSplicerTest._FindNthChildNamed..rzNo Nth child for n={0})Ú enumerateZ pre_orderÚ RuntimeErrorÚformat)rr rÚnÚiÚchildr rr Ú_FindNthChildNamed&s ÿ   z%CommentSplicerTest._FindNthChildNamedcCs`d}t |¡}t |¡|jdjd}| d|¡| dt|jƒ¡|jd}| |d¡dS)Nzfoo = 1 # and a comment rZ expr_stmtééz# and a comment) rÚParseCodeToTreerÚSpliceCommentsrr rÚlenr)rÚcodeÚtreeÚexprÚ comment_noder r r ÚtestSimpleInline-s    z#CommentSplicerTest.testSimpleInlinecCsHt d¡}t |¡}t |¡| dt|jƒ¡|jd}|  |¡dS)Nz9 foo = 1 # first comment bar = 2 rr© ÚtextwrapÚdedentrr!rr"rr#rr)rr$r%r'r r r ÚtestSimpleSeparateLine:s     z)CommentSplicerTest.testSimpleSeparateLinecCsDt d¡}t |¡}t |¡| dt|jƒ¡|  |jd¡dS)NzP foo = 1 # first comment # second comment bar = 2 rrr))rr$r%r r r ÚtestTwoLineCommentIs    z%CommentSplicerTest.testTwoLineCommentcCsTt d¡}t |¡}t |¡|jdjd}| d|jd¡| |jd¡dS)Nz7 if x: # a comment foo = 1 rr ÚNEWLINEr© r*r+rr!rr"rr r©rr$r%Zif_suiter r r Ú!testCommentIsFirstChildInCompoundWs    z4CommentSplicerTest.testCommentIsFirstChildInCompoundcCsTt d¡}t |¡}t |¡|jdjd}| d|jd¡| |jd¡dS)Nz7 if x: foo = 1 # a comment rr ÚDEDENTéÿÿÿÿéþÿÿÿr/r0r r r Ú testCommentIsLastChildInCompoundgs    z3CommentSplicerTest.testCommentIsLastChildInCompoundcCs^t d¡}t |¡}t |¡|jd}| |d¡|jdjd}|jd}| |d¡dS)NzI bar = 1 # line comment foo = 1 # inline comment rú# line commentérr3ú# inline comment©r*r+rr!rr"rr©rr$r%Zsep_comment_noder&Zinline_comment_noder r r ÚtestInlineAfterSeparateLinews      z.CommentSplicerTest.testInlineAfterSeparateLinecCs^t d¡}t |¡}t |¡|jd}| |d¡|jdjd}|jd}| |d¡dS)NzI bar = 1 foo = 1 # inline comment # line comment r4r6rrr3r8r9r:r r r ÚtestSeparateLineAfterInline‰s      z.CommentSplicerTest.testSeparateLineAfterInlinecCsTt d¡}t |¡}t |¡| |jd¡|jdjd}| d|jd¡dS)NzA if bar: z = 1 # a comment j = 2 rrr r2r3) r*r+rr!rr"rrr r0r r r ÚtestCommentBeforeDedent›s    z*CommentSplicerTest.testCommentBeforeDedentcCsTt d¡}t |¡}t |¡|jdjd}| |jd¡| d|jd¡dS)NzU if foo: if bar: z = 1 # a comment y = 1 rr r4r2r3© r*r+rr!rr"rrr r0r r r ÚtestCommentBeforeDedentTwoLevelªs    z2CommentSplicerTest.testCommentBeforeDedentTwoLevelcCsTt d¡}t |¡}t |¡|jdjd}| |jd¡| d|jd¡dS)NzV if foo: if bar: z = 1 # comment 2 y = 1 rr r4r2r3r>r0r r r Ú1testCommentBeforeDedentTwoLevelImproperlyIndented»s    zDCommentSplicerTest.testCommentBeforeDedentTwoLevelImproperlyIndentedcCsœt d¡}t |¡}t |¡| |jdd¡|j|ddd}| |jdd¡|  d|jd ¡|j|dd d}| |jdd ¡|  d|jd ¡dS) Nz} if foo: if bar: z = 1 # comment 2 # comment 1 # comment 0 j = 2 rz # comment 0Zsuite)rr4z # comment 1r2r3r7z # comment 2) r*r+rr!rr"rrrr )rr$r%Z if_suite_1Z if_suite_2r r r Ú!testCommentBeforeDedentThreeLevelÏs  z4CommentSplicerTest.testCommentBeforeDedentThreeLevelcCsht d¡}t |¡}t |¡|jdjd}|jd}| |d¡|jd}|jd}| |d¡dS)Nz… class Foo: """docstring abc...""" # top-level comment def foo(): pass # another comment rr r4z # anotherz # top-levelr9)rr$r%Ú class_suiteZanother_commentÚfuncdefZtoplevel_commentr r r ÚtestCommentsInClassês       z&CommentSplicerTest.testCommentsInClasscCsXt d¡}t |¡}t |¡|jd}|jd}| |d¡|jd}| |d¡dS)Nzl # Block comment number 1 # Block comment number 2 def f(): pass rz# Block comment number 1rz# Block comment number 2r9)rr$r%rCZblock_comment_1Zblock_comment_2r r r ÚtestMultipleBlockCommentss       z,CommentSplicerTest.testMultipleBlockCommentscCsbt d¡}t |¡}t |¡|jd}|jd}|jd}| |d¡|jd}| |d¡dS)Nz¶ class Foo(object): # A comment for qux. def qux(self): pass # Interim comment. def mux(self): pass rérz# A comment for qux.rz# Interim comment.r9)rr$r%ZclassdefrBZ qux_commentZinterim_commentr r r ÚtestCommentsOnDedentss       z(CommentSplicerTest.testCommentsOnDedentscCsFt d¡}t |¡}t |¡| |dd¡}|jd}| |d¡dS)NzH foo( # Request fractions of an hour. 948.0/3600, 20) Útrailerrz# Request fractions of an hour.) r*r+rr!rr"rrr)rr$r%rHÚcommentr r r ÚtestExprComments+s     z#CommentSplicerTest.testExprCommentscCs‚t d¡}t |¡}t |¡| |dd¡}| |jdd¡| |dd¡}| |jdd¡| |dd¡}| |jd d ¡dS) NzV foo( # com 1 948.0/3600, # com 2 20 + 12 # com 3 ) rHrz# com 1Úarglistr7z# com 2Ú arith_exprr3z# com 3) r*r+rr!rr"rrr)rr$r%rHrKrLr r r ÚtestMultipleCommentsInOneExpr7s   z0CommentSplicerTest.testMultipleCommentsInOneExpr)N)r)Ú__name__Ú __module__Ú __qualname__r rrr(r,r-r1r5r;r<r=r?r@rArDrErGrJrMr r r r rs&   rÚ__main__) Ú__doc__r*ZunittestZ yapf.pytreerrZTestCaserrNÚmainr r r r Ús  6