B v9a°!ã@s|ddlZddlZddlZddlZddlZddlmZddlmZddl m Z m Z e ƒGdd„dej ƒZ edkrxe ¡dS)éN)Úsupport)Úassert_python_ok)Ú scriptsdirÚskip_if_missingc@sreZdZej ed¡Zdd„Zdd„Z dd„Z dd „Z d d „Z d d „Z dd„Zdd„Zdd„Zdd„Zdd„ZdS)Ú PindentTestsz pindent.pyc Cs@t|ƒ.}t|ƒ}| | ¡| ¡¡WdQRXWdQRXdS)N)ÚopenÚ assertEqualÚ readlines)ÚselfZfn1Zfn2Úf1Úf2©r ú"/usr/lib/python3.7/test_pindent.pyÚassertFileEqualszPindentTests.assertFileEqualc GsJtjtj|jf|tjtjdd}| |¡\}}WdQRX| |¡|S)NT)ÚstdinÚstdoutZuniversal_newlines)Ú subprocessÚPopenÚsysÚ executableÚscriptÚPIPEZ communicateZ assertIsNone)r ÚsourceÚargsÚprocÚoutÚerrr r rÚpindents  zPindentTests.pindentcCsd dd„| ¡Dƒ¡dS)NÚ css|]}| ¡VqdS)N)Úlstrip)Ú.0Úliner r rú !sz+PindentTests.lstriplines..)ÚjoinÚ splitlines)r Údatar r rÚ lstriplines szPindentTests.lstriplinesc CsPd|_t ¡6}tj |d¡}t|jƒ}| ¡}WdQRXt|dƒ}|  |¡WdQRXt |jd|ƒ\}}}|  |d¡|  |d¡|d}|  tj  |¡¡t|ƒ}|  | ¡|¡WdQRXt|ƒ}| ¡} WdQRXt| ddƒ|  | | d¡|¡|  | |d¡| ¡t |jd|ƒ\}}}|  |d¡|  |d¡t|ƒ}|  | ¡| ¡WdQRXt|ƒ}|  | ¡|¡WdQRX| |¡} t|dƒ}|  | ¡WdQRXt |jd|ƒ\}}}|  |d¡|  |d¡t|ƒ}|  | ¡| ¡WdQRXt|ƒ}| ¡} WdQRXt| ddƒ|  | | d¡| ¡WdQRXdS) Nz_test.pyÚwz-dóú~Úexecz-cz-r)ZmaxDiffrZtemp_dirÚosÚpathr#rrÚreadÚwriterrZ assertTrueÚexistsÚcompilerr&) r Z directoryZ data_pathÚfÚclosedZrcrrZbackupÚcleanÚbrokenZindentedr r rÚ test_selftest#sL                   zPindentTests.test_selftestc CsP| | |d¡|¡| | |d¡|¡| |¡}| | |dddd¡|¡dS)Nz-cz-dz-rz-ez-sÚ4)rrr&)r r3r2r4r r rÚ pindent_testNs zPindentTests.pindent_testcCs$t d¡}t d¡}| ||¡dS)Na: if a: pass if a: pass else: pass if a: pass elif: pass else: pass while a: break while a: break else: pass for i in a: break for i in a: break else: pass try: pass finally: pass try: pass except TypeError: pass except ValueError: pass else: pass try: pass except TypeError: pass except ValueError: pass finally: pass with a: pass class A: pass def f(): pass a` if a: pass # end if if a: pass else: pass # end if if a: pass elif: pass else: pass # end if while a: break # end while while a: break else: pass # end while for i in a: break # end for for i in a: break else: pass # end for try: pass finally: pass # end try try: pass except TypeError: pass except ValueError: pass else: pass # end try try: pass except TypeError: pass except ValueError: pass finally: pass # end try with a: pass # end with class A: pass # end class A def f(): pass # end def f )ÚtextwrapÚdedentr7)r r3r2r r rÚtest_statementsTs ?LzPindentTests.test_statementscCs$t d¡}t d¡}| ||¡dS)Nz÷ def foobar(a, b): if a == b: a = a+1 elif a < b: b = b-1 if b > a: a = a-1 else: print 'oops!' aJ def foobar(a, b): if a == b: a = a+1 elif a < b: b = b-1 if b > a: a = a-1 # end if else: print 'oops!' # end if # end def foobar )r8r9r7)r r3r2r r rÚtest_multilevelås   zPindentTests.test_multilevelc Cs¾t d¡}t d¡}| | |d¡|¡| | |d¡|¡| |¡}| | |dddd¡|¡t d ¡}t d ¡}| | |d¡|¡| | |d¡|¡| |¡}| | |d¡|¡dS) Nz\ if a: if b: pass z if a: if b: pass # end if # end if z-cz-dz-rz-ez-sÚ9zD if a: if b: pass zo if a: if b: pass # end if # end if )r8r9rrr&)r r3r2r4r r rÚtest_preserve_indentsÿs   z"PindentTests.test_preserve_indentscCs@t d¡}t d¡}| | |d¡|¡| | |d¡|¡dS)Nzƒ class\ \ A: def \ f: pass z¸ class\ \ A: def \ f: pass # end def f # end class A z-cz-d)r8r9rr)r r3r2r r rÚtest_escaped_newline!s  z!PindentTests.test_escaped_newlinecCs$t d¡}t d¡}| ||¡dS)Nz4 if a: pass zI if a: pass # end if )r8r9r7)r r3r2r r rÚtest_empty_line9s zPindentTests.test_empty_linecCs$t d¡}t d¡}| ||¡dS)Nz# if a: pass z8 if a: pass # end if )r8r9r7)r r3r2r r rÚ test_onelineGs zPindentTests.test_onelineN)Ú__name__Ú __module__Ú __qualname__r+r,r#rrrrr&r5r7r:r;r=r>r?r@r r r rrs +"rÚ__main__)r+rZunittestrr8ZtestrZtest.support.script_helperrZtest.test_toolsrrZTestCaserrAÚmainr r r rÚs  D