?d#dZddlZddlZddlmZddlmZddlmZddlmZGddejZ e d k(rejyy) z%Tests for yapf.blank_line_calculator.N) reformatter)style)yapf_api)yapf_test_helpercjeZdZedZdZdZdZdZdZ dZ dZ d Z d Z d Zd Zd ZdZy)BasicBlankLineCalculatorTestcRtjtjy)N)rSetGlobalStyleCreateYapfStyle)clss F/usr/lib/python3/dist-packages/yapftests/blank_line_calculator_test.py setUpClassz'BasicBlankLineCalculatorTest.setUpClasss ..01ctjd}tjd}tj|}|j |t j |y)Nz; @bork() def foo(): pass z: @bork() def foo(): pass textwrapdedentrParseAndUnwrapassertCodeEqualrReformatselfunformatted_codeexpected_formatted_codelliness r testDecoratorsz+BasicBlankLineCalculatorTest.testDecorators s[(   'oo/    , ,-= >F0+2F2Fv2NOrctjd}tjd}tj|}|j |t j |y)Nz import sys @bork() def foo(): pass @fork() class moo(object): @bar() @baz() def method(self): pass z import sys @bork() def foo(): pass @fork() class moo(object): @bar() @baz() def method(self): pass rrs r testComplexDecoratorsz2BasicBlankLineCalculatorTest.testComplexDecorators/s[(  'oo/  " , ,-= >F0+2F2Fv2NOrctjd}tjd}tj|}|j |t j |y)Na> def foo(): pass top_level_code = True class moo(object): def method_1(self): pass ivar_a = 42 ivar_b = 13 def method_2(self): pass try: raise Error except Error as error: pass aG def foo(): pass top_level_code = True class moo(object): def method_1(self): pass ivar_a = 42 ivar_b = 13 def method_2(self): pass try: raise Error except Error as error: pass rrs r testCodeAfterFunctionsAndClassesz=BasicBlankLineCalculatorTest.testCodeAfterFunctionsAndClassesSs[(   'oo/  2 , ,-= >F0+2F2Fv2NOrctjd}tjd}tj|}|j |t j |y)Na2 # This is the first comment # And it's multiline # This is the second comment def foo(): pass # multiline before a # class definition # This is the second comment class qux(object): pass # An attached comment. class bar(object): '''class docstring''' # Comment attached to # function def foo(self): '''Another docstring.''' # Another multiline # comment pass a6 # This is the first comment # And it's multiline # This is the second comment def foo(): pass # multiline before a # class definition # This is the second comment class qux(object): pass # An attached comment. class bar(object): '''class docstring''' # Comment attached to # function def foo(self): '''Another docstring.''' # Another multiline # comment pass rrs r testCommentSpacingz/BasicBlankLineCalculatorTest.testCommentSpacings\(  :'oo /  B , ,-= >F0+2F2Fv2NOrctjd}tj|}|j |t j |y)Nzu class foo(object): # pylint: disable=invalid-name def f(self): pass rrcoders r testCommentBeforeMethodz4BasicBlankLineCalculatorTest.testCommentBeforeMethodsB ??  D , ,T 2F{33F;F0+2F2Fv2NOrctjd}tjd}tj|ddg\}}|j |||j |y)Nz def A(): pass def B(): # 4 pass # 5 def C(): pass def D(): # 9 pass # 10 def E(): pass z def A(): pass def B(): # 4 pass # 5 def C(): pass def D(): # 9 pass # 10 def E(): pass )) linesrrr FormatCoder assertTruerrrr%changeds r testLinesOnRangeBoundaryz5BasicBlankLineCalculatorTest.testLinesOnRangeBoundarysi (  'oo/  "'' 13MD'0$7OOGrctjd}tjd}tj|dg\}}|j |||j |y)Nzx def A(): pass def B(): # 6 pass # 7 def C(): pass )r4)rrrr7r assertFalser9s r testLinesRangeBoundaryNotOutsidez=BasicBlankLineCalculatorTest.testLinesRangeBoundaryNotOutside=sf (  'oo /  ''(8IMD'0$7Wrctjd}tjd}tj|dg\}}|j |||j |y)Nzy def A(): pass def B(): # 6 pass # 7 def C(): pass zx def A(): pass def B(): # 6 pass # 7 def C(): pass )r1r2r4r6r9s r testLinesRangeRemovez1BasicBlankLineCalculatorTest.testLinesRangeRemove^sd(  'oo /  ''(8IMD'0$7OOGrctjd}tjd}tj|dg\}}|j |||j |y)Nzz def A(): pass def B(): # 7 pass # 8 def C(): pass zy def A(): pass def B(): # 7 pass # 8 def C(): pass )r=r2r4r6r9s r testLinesRangeRemoveSomez5BasicBlankLineCalculatorTest.testLinesRangeRemoveSomesd(   'oo/  ''(8IMD'0$7OOGrN)__name__ __module__ __qualname__ classmethodrrrr r"r&r(r*r,r.r;r@rBrDrr rrsa22 P"PH+PZ@PD = ==*="P0#JB D"rr__main__) __doc__runittest yapf.yapflibrrr yapftestsrYAPFTestrrEmainrIrr rQsL,$!&H#3#<#<HV  z(--/r