müYøRã@s~dZddlmZddlZddlmZmZddlm Z ddl m Z m Z m Z mZddlmZdd d „ZGd d „d ejƒZGd d„dejƒ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éN)ÚNodeÚLeaf)Ú fixer_util)ÚAttrÚNameÚCallÚComma)ÚtokencCs@tj|ƒ}x!t|ƒD]}|jd}qWd|_|S)Nr)rZ parse_stringÚrangeÚchildrenÚparent)ÚcodeÚ strip_levelsZtreeÚi©rúJ/build/mts/release/bora-5583004/publish/lin64/lib64/python3.5/test_util.pyÚparses  rc@seZdZdd„ZdS)Ú MacroTestCasecCsDt|ttfƒr*ttjj|ƒ}|jt|ƒ|ƒdS)N) Ú isinstanceÚtupleÚlistrrZsymsZ simple_stmtÚ assertEqualÚstr)ÚselfÚnodeÚstringrrrÚ assertStrszMacroTestCase.assertStrN)Ú__name__Ú __module__Ú __qualname__rrrrrrs rc@s4eZdZdd„Zdd„Zdd„ZdS)Ú Test_is_tuplecCstjt|ddƒƒS)Nré)rÚis_tupler)rrrrrr$ szTest_is_tuple.is_tuplecCsr|j|jdƒƒ|j|jdƒƒ|j|jdƒƒ|j|jdƒƒ|j|jdƒƒdS)Nz(a, b)z (a, (b, c))z((a, (b, c)),)z(a,)z())Ú assertTruer$)rrrrÚ test_valid#s zTest_is_tuple.test_validcCs0|j|jdƒƒ|j|jdƒƒdS)Nz(a)z('foo') % (b, c))Ú assertFalser$)rrrrÚ test_invalid*szTest_is_tuple.test_invalidN)rr r!r$r&r(rrrrr"s   r"c@s4eZdZdd„Zdd„Zdd„ZdS)Ú Test_is_listcCstjt|ddƒƒS)Nrr#)rÚis_listr)rrrrrr*0szTest_is_list.is_listcCsr|j|jdƒƒ|j|jdƒƒ|j|jdƒƒ|j|jdƒƒ|j|jdƒƒdS)Nz[]z[a]z[a, b]z [a, [b, c]]z[[a, [b, c]],])r%r*)rrrrr&3s zTest_is_list.test_validcCs|j|jdƒƒdS)Nz[]+[])r'r*)rrrrr(:szTest_is_list.test_invalidN)rr r!r*r&r(rrrrr)/s   r)c@s(eZdZdd„Zdd„ZdS)Ú Test_AttrcCsZtdddƒ}|jttdƒtdƒƒdƒ|jt|tdƒƒdƒdS)Nzfoo()rr#ÚaÚbza.bzfoo().b)rrrr)rÚcallrrrÚtest?s%zTest_Attr.testcCs5ttdƒtdƒƒ}|jt|ƒtƒdS)Nr,r-)rrrÚtyper)rÚattrrrrÚ test_returnsEszTest_Attr.test_returnsN)rr r!r/r2rrrrr+>s  r+c@seZdZdd„ZdS)Ú Test_NamecCsL|jtdƒdƒ|jtdƒdƒ|jtdddƒdƒdS)Nr,z foo.foo().barÚprefixr-Zba)rr)rrrrr/KszTest_Name.testN)rr r!r/rrrrr3Js r3c@s.eZdZdddd„Zdd„ZdS)Ú Test_CallNcCscg}t|tƒrMx+|D]#}|j|ƒ|jtƒƒqW|jƒtt|ƒ||ƒS)zHelp the next test)rrÚappendr Úpopr r)rÚnameÚargsr4r ÚargrrrÚ_CallRs   zTest_Call._Callc Cs"dttjdƒttjdƒttjdƒgttjdƒttjdƒttjdƒttjdƒgttjdƒttjdddƒgg}|j|jd ƒd ƒ|j|jd|dƒd ƒ|j|jd |dƒd ƒ|j|jd|dddƒdƒdS)Nrr#éér-Újr4ú ÚAzA()zb(1,2,3)za.b().cza.b().c(1,3,2,4)Údz d(b, j))rr ÚNUMBERÚSTRINGrr;)rZkidsrrrr/\s!-  zTest_Call.test)rr r!r;r/rrrrr5Qs  r5c@s@eZdZdd„Zdd„Zdd„Zdd„Zd S) ÚTest_does_tree_importcCsPtj||ƒ}|r|Sx-|jD]"}|j||ƒ}|r&|Sq&WdS)N)rÚ find_bindingr Ú_find_bind_rec)rr8rÚcZchildrrrrFksz$Test_does_tree_import._find_bind_reccCs1t|ƒ}|jd|ƒ}tj|||ƒS)NÚstart)rrFrÚdoes_tree_import)rÚpackager8rrrrrrIts z&Test_does_tree_import.does_tree_importc Csìd}xk|D]c\}}}|j|||d |ƒ}|j|ƒ|j|||d |ƒ}|j|ƒq Wd }xk|D]c\}}}|j|||d |ƒ}|j|ƒ|j|||d |ƒ}|j|ƒqWdS)!Nr,úfrom a import búa.dúfrom a.d import búd.aúfrom d.a import búimport búimport b, c, dÚ úfrom a import aÚxúfrom x import aúfrom x import b, c, a, dúx.búfrom x.b import aúfrom x.b import b, c, a, dúimport aúimport b, c, a, d©r,r,rK©rLr,rM©rNr,rO©Nr,rP©Nr,rQ)r\r]r^r_r`©r,r,rS©rTr,rU©rTr,rV©rWr,rX©rWr,rY©Nr,rZ©Nr,r[)rarbrcrdrerfrg)rIr'r%)rrZ failing_testsrJr8Úimport_ÚnZ passing_testsrrrÚtry_withzs,  zTest_does_tree_import.try_withcCs|jdƒdS)Nzdef foo(): bar.baz() start=3)rj)rrrrÚtest_in_function“sz&Test_does_tree_import.test_in_functionN)rr r!rFrIrjrkrrrrrDjs   rDc@s?eZdZddd„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)4ÚTest_find_bindingNcCstj|t|ƒ|ƒS)N)rrEr)rr8rrJrrrrE—szTest_find_binding.find_bindingcCsš|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒdS)Nr,za = bz a = [b, c, d]z a = foo()za = foo().foo.foo[6][foo]zfoo = azfoo = (a, b, c))r%rEr')rrrrÚtest_simple_assignmentšs z(Test_find_binding.test_simple_assignmentcCsš|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ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%rEr')rrrrÚtest_tuple_assignment¢s z'Test_find_binding.test_tuple_assignmentcCsš|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ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%rEr')rrrrÚtest_list_assignmentªs z&Test_find_binding.test_list_assignmentcCsh|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒdS)Nr,z foo.a = 5z foo[a] = 5z foo(a) = 5z foo(a, b) = 5)r'rE)rrrrÚtest_invalid_assignments²sz*Test_find_binding.test_invalid_assignmentscCsh|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒdS)Nr,zimport azimport b, c, a, dzimport bzimport b, c, d)r%rEr')rrrrÚtest_simple_import¸sz$Test_find_binding.test_simple_importcCsÌ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jdd ƒƒ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%rEr')rrrrÚtest_from_import¾sz"Test_find_binding.test_from_importcCsh|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ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%rEr')rrrrÚtest_import_asÈsz Test_find_binding.test_import_ascCs³|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ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%rEr')rrrrÚtest_from_import_asÎsz%Test_find_binding.test_from_import_ascCsn|j|jddƒƒ|j|jddƒƒ|j|jdddƒƒ|j|jdddƒƒdS)Nr-zimport bzimport b, c, drG)r%rEr')rrrrÚtest_simple_import_with_package×sz1Test_find_binding.test_simple_import_with_packagecCs8|j|jdddƒƒ|j|jdddƒƒ|j|jdddƒƒ|j|jdddƒƒ|j|jdddƒƒ|j|jdd dƒƒ|j|jdd dƒƒ|j|jdd dƒƒ|j|jdd d ƒƒ|j|jddd ƒƒ|j|jdddƒƒdS)Nr,zfrom x import arTzfrom 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%rEr')rrrrÚtest_from_import_with_packageÝsz/Test_find_binding.test_from_import_with_packagecCsX|j|jdddƒƒ|j|jdddƒƒ|j|jdddƒƒdS)Nr,zimport b.c as azb.cz import a as fÚf)r'rE)rrrrÚtest_import_as_with_packageêsz-Test_find_binding.test_import_as_with_packagecCsÈ|j|jdddƒƒ|j|jdddƒƒ|j|jdddƒƒ|j|jdddƒƒ|j|jdddƒƒ|j|jddd ƒƒ|j|jddd ƒƒdS) Nr,zfrom x import b as arTzfrom 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'rE)rrrrÚ test_from_import_as_with_packageïsz2Test_find_binding.test_from_import_as_with_packagecCsÒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒd }|j|jd|ƒƒ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%rEr')rÚsrrrÚtest_function_defûsz#Test_find_binding.test_function_defcCs|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jdd ƒƒ|j|jdd ƒƒ|j|jdd ƒƒd }|j|jd|ƒƒ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%rEr')rr{rrrÚtest_class_def sz Test_find_binding.test_class_defcCs³|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ|j|jddƒƒ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%rEr')rrrrÚtest_forszTest_find_binding.test_forcCsüd}|j|jd|ƒƒd}|j|jd|ƒƒd}|j|jd|ƒƒd}|j|jd|ƒƒd}|j|jd|ƒƒd}|j|jd|ƒƒd}|j|jd|ƒƒd }|j|jd|ƒƒ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%rEr')rr{rrrÚtest_for_nested%s z!Test_find_binding.test_for_nestedcCs6|j|jddƒƒ|j|jddƒƒdS)Nr,zif b in r: a = czif a in r: d = e)r%rEr')rrrrÚtest_ifVszTest_find_binding.test_ifcCsBd}|j|jd|ƒƒd}|j|jd|ƒƒdS)NzL if b in r: if c in d: a = cr,zL if b in r: if c in d: c = a)r%rEr')rr{rrrÚtest_if_nestedZsz Test_find_binding.test_if_nestedcCs6|j|jddƒƒ|j|jddƒƒdS)Nr,zwhile b in r: a = czwhile a in r: d = e)r%rEr')rrrrÚ test_whilegszTest_find_binding.test_whilecCsBd}|j|jd|ƒƒd}|j|jd|ƒƒdS)NzR while b in r: while c in d: a = cr,zR while b in r: while c in d: c = a)r%rEr')rr{rrrÚtest_while_nestedksz#Test_find_binding.test_while_nestedcCs€d}|j|jd|ƒƒd}|j|jd|ƒƒd}|j|jd|ƒƒd}|j|jd|ƒƒ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%rEr')rr{rrrÚtest_try_exceptxsz!Test_find_binding.test_try_exceptcCsÝd}|j|jd|ƒƒd}|j|jd|ƒƒd}|j|jd|ƒƒd}|j|jd|ƒƒd}|j|jd|ƒƒd}|j|jd|ƒƒd}|j|jd|ƒƒ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%rEr')rr{rrrÚtest_try_except_nested—s    z(Test_find_binding.test_try_except_nestedcCs€d}|j|jd|ƒƒd}|j|jd|ƒƒd}|j|jd|ƒƒd}|j|jd|ƒƒ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%rEr')rr{rrrÚtest_try_except_finallyäsz)Test_find_binding.test_try_except_finallycCsad}|j|jd|ƒƒd}|j|jd|ƒƒd}|j|jd|ƒƒ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%rEr')rr{rrrÚtest_try_except_finally_nesteds   z0Test_find_binding.test_try_except_finally_nested)rr r!rErmrnrorprqrrrsrtrurvrxrzr|r}r~rr€rr‚rƒr„r…r†r‡rrrrrl–s2            1    M !rlc@sLeZdZdd„Zdd„Zdd„Zdd„Zd d „Zd S) ÚTest_touch_importcCs9tdƒ}tjdd|ƒ|jt|ƒdƒdS)Nz"""foo""" bar()Úfooz"""foo""" import foo bar() )rrÚ touch_importrr)rrrrrÚtest_after_docstring*s z&Test_touch_import.test_after_docstringcCs9tdƒ}tjdd|ƒ|jt|ƒdƒdS)Nz"""foo""" import bar bar()r‰z'"""foo""" import bar import foo bar() )rrrŠrr)rrrrrÚtest_after_imports/s z$Test_touch_import.test_after_importscCs9tdƒ}tjdd|ƒ|jt|ƒdƒdS)Nzbar()r‰zimport foo bar() )rrrŠrr)rrrrrÚtest_beginning4s z Test_touch_import.test_beginningcCs9tdƒ}tjdd|ƒ|jt|ƒdƒdS)Nzbar()ZhtmlÚescapezfrom html import escape bar() )rrrŠrr)rrrrrrr9s z"Test_touch_import.test_from_importcCs9tdƒ}tjdd|ƒ|jt|ƒdƒdS)Nzbar()Zcgizimport cgi bar() )rrrŠrr)rrrrrÚtest_name_import>s z"Test_touch_import.test_name_importN)rr r!r‹rŒrrrrrrrrrˆ(s     rˆc@s(eZdZdd„Zdd„ZdS)ÚTest_find_indentationcCsQtj}tdƒ}|j||ƒdƒtdƒ}|j||ƒdƒdS)Nznode()Ú)rÚfind_indentationrr)rÚfirrrrÚ test_nothingEs    z"Test_find_indentation.test_nothingcCs‘tj}tdƒ}|j||ƒdƒ|j||jdjdjdƒdƒtdƒ}|j||jdjdjdƒdƒdS)Nzdef f(): x()r‘rr=r#z zdef f(): x() y())rr’rrr )rr“rrrrÚ test_simpleLs   + z!Test_find_indentation.test_simpleN)rr r!r”r•rrrrrCs  r)Ú__doc__r‘rZos.pathÚosZlib2to3.pytreerrZlib2to3rZlib2to3.fixer_utilrrr r Z lib2to3.pgen2r rZTestCaserr"r)r+r3r5rDrlrˆrrrrrÚs$ "  ,ÿ“