B Ç S]×Rã@s dZddlmZddlmZmZddlmZddlm Z m Z m Z m Z ddl mZddd „ZGd d „d ejƒZGd d „d ejƒZGdd„dejƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„dejƒZGdd„dejƒZGdd„dejƒZGdd„dejƒZdS) z' Test suite for the code in fixer_util é)Úsupporté)ÚNodeÚLeaf)Ú fixer_util)ÚAttrÚNameÚCallÚComma)ÚtokencCs0t |¡}xt|ƒD]}|jd}qWd|_|S)Nr)rZ parse_stringÚrangeÚchildrenÚparent)ÚcodeÚ strip_levelsZtreeÚi©rú(lib/python3.7/lib2to3/tests/test_util.pyÚparse s  rc@seZdZdd„ZdS)Ú MacroTestCasecCs0t|ttfƒrttjj|ƒ}| t|ƒ|¡dS)N) Ú isinstanceÚtupleÚlistrrZsymsZ simple_stmtÚ assertEqualÚstr)ÚselfÚnodeÚstringrrrÚ assertStrszMacroTestCase.assertStrN)Ú__name__Ú __module__Ú __qualname__rrrrrrsrc@s$eZdZdd„Zdd„Zdd„ZdS)Ú Test_is_tuplecCst t|dd¡S)Né)r)rÚis_tupler)rrrrrr$szTest_is_tuple.is_tuplecCsT| | d¡¡| | d¡¡| | d¡¡| | d¡¡| | d¡¡dS)Nz(a, b)z (a, (b, c))z((a, (b, c)),)z(a,)z())Ú assertTruer$)rrrrÚ test_valid s zTest_is_tuple.test_validcCs$| | d¡¡| | d¡¡dS)Nz(a)z('foo') % (b, c))Ú assertFalser$)rrrrÚ test_invalid'szTest_is_tuple.test_invalidN)rr r!r$r&r(rrrrr"sr"c@s$eZdZdd„Zdd„Zdd„ZdS)Ú Test_is_listcCst t|dd¡S)Nr#)r)rÚis_listr)rrrrrr*-szTest_is_list.is_listcCsT| | d¡¡| | d¡¡| | d¡¡| | d¡¡| | d¡¡dS)Nz[]z[a]z[a, b]z [a, [b, c]]z[[a, [b, c]],])r%r*)rrrrr&0s zTest_is_list.test_validcCs| | d¡¡dS)Nz[]+[])r'r*)rrrrr(7szTest_is_list.test_invalidN)rr r!r*r&r(rrrrr),sr)c@seZdZdd„Zdd„ZdS)Ú Test_AttrcCs@tddd}| ttdƒtdƒƒd¡| t|tdƒƒd¡dS)Nzfoo()r#)rÚaÚbza.bzfoo().b)rrrr)rZcallrrrÚtest<s zTest_Attr.testcCs&ttdƒtdƒƒ}| t|ƒt¡dS)Nr,r-)rrrÚtyper)rÚattrrrrÚ test_returnsBszTest_Attr.test_returnsN)rr r!r.r1rrrrr+;sr+c@seZdZdd„ZdS)Ú Test_NamecCs8| tdƒd¡| tdƒd¡| tdddd¡dS)Nr,z foo.foo().barr-)ÚprefixZba)rr)rrrrr.HszTest_Name.testN)rr r!r.rrrrr2Gsr2c@seZdZddd„Zdd„ZdS)Ú Test_CallNcCsJg}t|tƒr:x"|D]}| |¡| tƒ¡qW| ¡tt|ƒ||ƒS)zHelp the next test)rrÚappendr Úpopr r)rÚnameÚargsr3r ÚargrrrÚ_CallOs   zTest_Call._Callc CsÌdttjdƒttjdƒttjdƒgttjdƒttjdƒttjdƒttjdƒgttjdƒttjdddgg}| | d ¡d ¡| | d|d¡d ¡| | d |d¡d ¡| |jd|dddd¡dS)Nrr#éér-Újú )r3ÚAzA()zb(1,2,3)za.b().cza.b().c(1,3,2,4)Údz d(b, j))rr ÚNUMBERÚSTRINGrr:)rZkidsrrrr.Ys zTest_Call.test)NN)rr r!r:r.rrrrr4Ns r4c@s,eZdZdd„Zdd„Zdd„Zdd„Zd S) ÚTest_does_tree_importcCs<t ||¡}|r|Sx"|jD]}| ||¡}|r|SqWdS)N)rÚ find_bindingr Ú_find_bind_rec)rr7rÚcZchildrrrrEhs   z$Test_does_tree_import._find_bind_reccCs"t|ƒ}| d|¡}t |||¡S)NÚstart)rrErÚdoes_tree_import)rÚpackager7rrrrrrHqs z&Test_does_tree_import.does_tree_importcCs´d}xR|D]J\}}}| |||d|¡}| |¡| |||d|¡}| |¡q Wd}xR|D]J\}}}| |||d|¡}| |¡| |||d|¡}| |¡qbWdS)N))r,r,zfrom a import b)za.dr,zfrom a.d import b)zd.ar,zfrom d.a import b)Nr,zimport b)Nr,zimport b, c, dÚ ))r,r,zfrom a import a)Úxr,zfrom x import a)rKr,zfrom x import b, c, a, d)zx.br,zfrom x.b import a)zx.br,zfrom x.b import b, c, a, d)Nr,zimport a)Nr,zimport b, c, a, d)rHr'r%)rrZ failing_testsrIr7Úimport_ÚnZ passing_testsrrrÚtry_withws  zTest_does_tree_import.try_withcCs| d¡dS)Nzdef foo(): bar.baz() start=3)rN)rrrrÚtest_in_functionsz&Test_does_tree_import.test_in_functionN)rr r!rErHrNrOrrrrrCgs rCc@sÖeZdZd4dd„Zdd„Zdd„Zdd „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'„Zd(d)„Zd*d+„Zd,d-„Zd.d/„Zd0d1„Zd2d3„ZdS)5ÚTest_find_bindingNcCst |t|ƒ|¡S)N)rrDr)rr7rrIrrrrD”szTest_find_binding.find_bindingcCsp| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡dS)Nr,za = bz a = [b, c, d]z a = foo()za = foo().foo.foo[6][foo]zfoo = azfoo = (a, b, c))r%rDr')rrrrÚtest_simple_assignment—s z(Test_find_binding.test_simple_assignmentcCsp| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡dS)Nr,z(a,) = bz(a, b, c) = [b, c, d]z(c, (d, a), b) = foo()z(a, b) = foo().foo[6][foo]z(foo, b) = (b, a)z(foo, (b, c)) = (a, b, c))r%rDr')rrrrÚtest_tuple_assignmentŸs z'Test_find_binding.test_tuple_assignmentcCsp| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡dS)Nr,z[a] = bz[a, b, c] = [b, c, d]z[c, [d, a], b] = foo()z[a, b] = foo().foo[a][foo]z[foo, b] = (b, a)z[foo, [b, c]] = (a, b, c))r%rDr')rrrrÚtest_list_assignment§s z&Test_find_binding.test_list_assignmentcCsL| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡dS)Nr,z foo.a = 5z foo[a] = 5z foo(a) = 5z foo(a, b) = 5)r'rD)rrrrÚtest_invalid_assignments¯sz*Test_find_binding.test_invalid_assignmentscCsL| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡dS)Nr,zimport azimport b, c, a, dzimport bzimport b, c, d)r%rDr')rrrrÚtest_simple_importµsz$Test_find_binding.test_simple_importcCs”| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd ¡¡dS) Nr,zfrom x import azfrom a import azfrom x import b, c, a, dzfrom x.b import azfrom x.b import b, c, a, dzfrom a import bzfrom a.d import bzfrom d.a import b)r%rDr')rrrrÚtest_from_import»sz"Test_find_binding.test_from_importcCsL| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡dS)Nr,z import b as azimport b as a, c, a as f, dz import a as fzimport b, c as f, d as e)r%rDr')rrrrÚtest_import_asÅsz Test_find_binding.test_import_ascCs‚| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡dS) Nr,zfrom x import b as azfrom x import g as a, d as bzfrom x.b import t as azfrom x.b import g as a, dzfrom a import b as tzfrom a.d import b as tzfrom d.a import b as t)r%rDr')rrrrÚtest_from_import_asËsz%Test_find_binding.test_from_import_ascCsP| | dd¡¡| | dd¡¡| | ddd¡¡| | ddd¡¡dS)Nr-zimport bzimport b, c, drF)r%rDr')rrrrÚtest_simple_import_with_packageÔsz1Test_find_binding.test_simple_import_with_packagecCsà| | ddd¡¡| | ddd¡¡| | ddd¡¡| | ddd¡¡| | ddd¡¡| | dd d¡¡| | dd d¡¡| | dd d¡¡| | dd d ¡¡| | ddd ¡¡| | ddd¡¡dS)Nr,zfrom x import arKzfrom a import azfrom x import *zfrom x import b, c, a, dzfrom x.b import azx.bzfrom x.b import *zfrom x.b import b, c, a, dzfrom a import bzfrom a.d import bza.dzfrom d.a import bzfrom x.y import *za.b)r%rDr')rrrrÚtest_from_import_with_packageÚsz/Test_find_binding.test_from_import_with_packagecCs@| | ddd¡¡| | ddd¡¡| | ddd¡¡dS)Nr,zimport b.c as azb.cz import a as fÚf)r'rD)rrrrÚtest_import_as_with_packageçsz-Test_find_binding.test_import_as_with_packagecCs| | ddd¡¡| | ddd¡¡| | ddd¡¡| | ddd¡¡| | ddd¡¡| | ddd ¡¡| | ddd ¡¡dS) Nr,zfrom x import b as arKzfrom x import g as a, d as bzfrom x.b import t as azx.bzfrom x.b import g as a, dzfrom a import b as tr-Út)r'rD)rrrrÚ test_from_import_as_with_packageìsz2Test_find_binding.test_from_import_as_with_packagecCs˜| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡d }| | d|¡¡dS) Nr,z def a(): passzdef a(b, c, d): passzdef a(): b = 7zdef d(b, (c, a), e): passzdef d(a=7): passzdef d(a): passzdef d(): a = 7zG def d(): def a(): pass)r%rDr')rÚsrrrÚtest_function_deføsz#Test_find_binding.test_function_defcCsÎ| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd ¡¡| | dd ¡¡| | dd ¡¡d }| | d|¡¡dS) Nr,z class a: passzclass a(): passzclass a(b): passzclass a(b, c=8): passz class d: passzclass d(a): passzclass d(b, a=7): passzclass d(b, *a): passzclass d(b, **a): passzclass d: a = 7zK class d(): class a(): pass)r%rDr')rr_rrrÚtest_class_defsz Test_find_binding.test_class_defcCs‚| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡dS) Nr,zfor a in r: passzfor a, b in r: passzfor (a, b) in r: passzfor c, (a,) in r: passzfor c, (a, b) in r: passzfor c in r: a = czfor c in a: pass)r%rDr')rrrrÚtest_forszTest_find_binding.test_forcCs´d}| | d|¡¡d}| | d|¡¡d}| | d|¡¡d}| | d|¡¡d}| | d|¡¡d}| | d|¡¡d}| | d|¡¡d }| | d|¡¡dS) NzM for b in r: for a in b: passr,zP for b in r: for a, c in b: passzR for b in r: for (a, c) in b: passzP for b in r: for (a,) in b: passzU for b in r: for c, (a, d) in b: passzN for b in r: for c in b: a = 7zN for b in r: for c in b: d = azN for b in r: for c in a: d = 7)r%rDr')rr_rrrÚtest_for_nested"s z!Test_find_binding.test_for_nestedcCs(| | dd¡¡| | dd¡¡dS)Nr,zif b in r: a = czif a in r: d = e)r%rDr')rrrrÚtest_ifSszTest_find_binding.test_ifcCs0d}| | d|¡¡d}| | d|¡¡dS)NzL if b in r: if c in d: a = cr,zL if b in r: if c in d: c = a)r%rDr')rr_rrrÚtest_if_nestedWsz Test_find_binding.test_if_nestedcCs(| | dd¡¡| | dd¡¡dS)Nr,zwhile b in r: a = czwhile a in r: d = e)r%rDr')rrrrÚ test_whiledszTest_find_binding.test_whilecCs0d}| | d|¡¡d}| | d|¡¡dS)NzR while b in r: while c in d: a = cr,zR while b in r: while c in d: c = a)r%rDr')rr_rrrÚtest_while_nestedhsz#Test_find_binding.test_while_nestedcCs\d}| | d|¡¡d}| | d|¡¡d}| | d|¡¡d}| | d|¡¡dS)NzQ try: a = 6 except: b = 8r,zQ try: b = 8 except: a = 6zƒ try: b = 8 except KeyError: pass except: a = 6zQ try: b = 8 except: b = 6)r%rDr')rr_rrrÚtest_try_exceptusz!Test_find_binding.test_try_exceptcCsžd}| | d|¡¡d}| | d|¡¡d}| | d|¡¡d}| | d|¡¡d}| | d|¡¡d}| | d|¡¡d}| | d|¡¡dS) Nz› try: try: a = 6 except: pass except: b = 8r,z› try: b = 8 except: try: a = 6 except: passz› try: b = 8 except: try: pass except: a = 6zÕ try: try: b = 8 except KeyError: pass except: a = 6 except: passzÕ try: pass except: try: b = 8 except KeyError: pass except: a = 6zQ try: b = 8 except: b = 6a try: try: b = 8 except: c = d except: try: b = 6 except: t = 8 except: o = y)r%rDr')rr_rrrÚtest_try_except_nested”s    z(Test_find_binding.test_try_except_nestedcCs\d}| | d|¡¡d}| | d|¡¡d}| | d|¡¡d}| | d|¡¡dS)Nz| try: c = 6 except: b = 8 finally: a = 9r,zR try: b = 8 finally: a = 6zR try: b = 8 finally: b = 6z| try: b = 8 except: b = 9 finally: b = 6)r%rDr')rr_rrrÚtest_try_except_finallyász)Test_find_binding.test_try_except_finallycCsFd}| | d|¡¡d}| | d|¡¡d}| | d|¡¡dS)Nzú try: c = 6 except: b = 8 finally: try: a = 9 except: b = 9 finally: c = 9r,z try: b = 8 finally: try: pass finally: a = 6zž try: b = 8 finally: try: b = 6 finally: b = 7)r%rDr')rr_rrrÚtest_try_except_finally_nesteds   z0Test_find_binding.test_try_except_finally_nested)N)rr r!rDrQrRrSrTrUrVrWrXrYrZr\r^r`rarbrcrdrerfrgrhrirjrkrrrrrP“s2      1  M!rPc@s4eZdZdd„Zdd„Zdd„Zdd„Zd d „Zd S) ÚTest_touch_importcCs*tdƒ}t dd|¡| t|ƒd¡dS)Nz"""foo""" bar()Úfooz"""foo""" import foo bar() )rrÚ touch_importrr)rrrrrÚtest_after_docstring'sz&Test_touch_import.test_after_docstringcCs*tdƒ}t dd|¡| t|ƒd¡dS)Nz"""foo""" import bar bar()rmz'"""foo""" import bar import foo bar() )rrrnrr)rrrrrÚtest_after_imports,sz$Test_touch_import.test_after_importscCs*tdƒ}t dd|¡| t|ƒd¡dS)Nzbar()rmzimport foo bar() )rrrnrr)rrrrrÚtest_beginning1sz Test_touch_import.test_beginningcCs*tdƒ}t dd|¡| t|ƒd¡dS)Nzbar()ZhtmlÚescapezfrom html import escape bar() )rrrnrr)rrrrrrV6sz"Test_touch_import.test_from_importcCs*tdƒ}t dd|¡| t|ƒd¡dS)Nzbar()Zcgizimport cgi bar() )rrrnrr)rrrrrÚtest_name_import;sz"Test_touch_import.test_name_importN)rr r!rorprqrVrsrrrrrl%s rlc@seZdZdd„Zdd„ZdS)ÚTest_find_indentationcCs:tj}tdƒ}| ||ƒd¡tdƒ}| ||ƒd¡dS)Nznode()Ú)rÚfind_indentationrr)rÚfirrrrÚ test_nothingBs z"Test_find_indentation.test_nothingcCsntj}tdƒ}| ||ƒd¡| ||jdjdjdƒd¡tdƒ}| ||jdjdjdƒd¡dS)Nzdef f(): x()rurr<r#z zdef f(): x() y())rrvrrr )rrwrrrrÚ test_simpleIs "z!Test_find_indentation.test_simpleN)rr r!rxryrrrrrt@srtN)r)Ú__doc__rurZlib2to3.pytreerrZlib2to3rZlib2to3.fixer_utilrrr r Z lib2to3.pgen2r rZTestCaserr"r)r+r2r4rCrPrlrtrrrrÚs&     ,