B 18™\×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)rÚ parse_stringÚrangeÚchildrenÚparent)ÚcodeÚ strip_levelsÚtreeÚi©rú[/oak/stanford/groups/akundaje/marinovg/programs/Python-3.7.3/Lib/lib2to3/tests/test_util.pyÚparse s  rc@seZdZdd„ZdS)Ú MacroTestCasecCs0t|ttfƒrttjj|ƒ}| t|ƒ|¡dS)N) Ú isinstanceÚtupleÚlistrrÚsymsÚ simple_stmtÚ assertEqualÚstr)ÚselfÚnodeÚstringrrrÚ assertStrszMacroTestCase.assertStrN)Ú__name__Ú __module__Ú __qualname__r"rrrrrsrc@s$eZdZdd„Zdd„Zdd„ZdS)Ú Test_is_tuplecCst t|dd¡S)Né)r)rÚis_tupler)rr!rrrr(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)r#r$r%r(r*r,rrrrr&sr&c@s$eZdZdd„Zdd„Zdd„ZdS)Ú Test_is_listcCst t|dd¡S)Nr')r)rÚis_listr)rr!rrrr.-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)r#r$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)rr"rr)rÚcallrrrÚtest<s zTest_Attr.testcCs&ttdƒtdƒƒ}| t|ƒt¡dS)Nr0r1)rrrÚtyper)rÚattrrrrÚ test_returnsBszTest_Attr.test_returnsN)r#r$r%r3r6rrrrr/;sr/c@seZdZdd„ZdS)Ú Test_NamecCs8| tdƒd¡| tdƒd¡| tdddd¡dS)Nr0z foo.foo().barr1)ÚprefixÚba)r"r)rrrrr3HszTest_Name.testN)r#r$r%r3rrrrr7Gsr7c@seZdZddd„Zdd„ZdS)Ú Test_CallNcCsJg}t|tƒr:x"|D]}| |¡| tƒ¡qW| ¡tt|ƒ||ƒS)zHelp the next test)rrÚappendr Úpopr r)rÚnameÚargsr8rÚ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'éér1Újú )r8ÚAzA()zb(1,2,3)za.b().cza.b().c(1,3,2,4)Údz d(b, j))rr ÚNUMBERÚSTRINGr"r@)rÚkidsrrrr3Ys zTest_Call.test)NN)r#r$r%r@r3rrrrr:Ns r:c@s,eZdZdd„Zdd„Zdd„Zdd„Zd S) ÚTest_does_tree_importcCs<t ||¡}|r|Sx"|jD]}| ||¡}|r|SqWdS)N)rÚ find_bindingrÚ_find_bind_rec)rr=r ÚcÚchildrrrrLhs   z$Test_does_tree_import._find_bind_reccCs"t|ƒ}| d|¡}t |||¡S)NÚstart)rrLrÚdoes_tree_import)rÚpackager=r!r rrrrPqs z&Test_does_tree_import.does_tree_importcCs´d}xR|D]J\}}}| |||d|¡}| |¡| |||d|¡}| |¡q Wd}xR|D]J\}}}| |||d|¡}| |¡| |||d|¡}| |¡qbWdS)N))r0r0zfrom a import b)za.dr0zfrom a.d import b)zd.ar0zfrom d.a import b)Nr0zimport b)Nr0zimport b, c, dÚ ))r0r0zfrom a import a)Úxr0zfrom x import a)rSr0zfrom x import b, c, a, d)zx.br0zfrom x.b import a)zx.br0zfrom x.b import b, c, a, d)Nr0zimport a)Nr0zimport b, c, a, d)rPr+r))rr!Z failing_testsrQr=Úimport_ÚnZ passing_testsrrrÚtry_withws  zTest_does_tree_import.try_withcCs| d¡dS)Nzdef foo(): bar.baz() start=3)rV)rrrrÚtest_in_functionsz&Test_does_tree_import.test_in_functionN)r#r$r%rLrPrVrWrrrrrJgs rJc@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)rrKr)rr=r!rQrrrrK”szTest_find_binding.find_bindingcCsp| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡dS)Nr0za = bz a = [b, c, d]z a = foo()za = foo().foo.foo[6][foo]zfoo = azfoo = (a, b, c))r)rKr+)rrrrÚtest_simple_assignment—s z(Test_find_binding.test_simple_assignmentcCsp| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡dS)Nr0z(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)rKr+)rrrrÚtest_tuple_assignmentŸs z'Test_find_binding.test_tuple_assignmentcCsp| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡dS)Nr0z[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)rKr+)rrrrÚtest_list_assignment§s z&Test_find_binding.test_list_assignmentcCsL| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡dS)Nr0z foo.a = 5z foo[a] = 5z foo(a) = 5z foo(a, b) = 5)r+rK)rrrrÚtest_invalid_assignments¯sz*Test_find_binding.test_invalid_assignmentscCsL| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡dS)Nr0zimport azimport b, c, a, dzimport bzimport b, c, d)r)rKr+)rrrrÚtest_simple_importµsz$Test_find_binding.test_simple_importcCs”| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd ¡¡dS) Nr0zfrom 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)rKr+)rrrrÚtest_from_import»sz"Test_find_binding.test_from_importcCsL| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡dS)Nr0z import b as azimport b as a, c, a as f, dz import a as fzimport b, c as f, d as e)r)rKr+)rrrrÚtest_import_asÅsz Test_find_binding.test_import_ascCs‚| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡dS) Nr0zfrom 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)rKr+)rrrrÚtest_from_import_asËsz%Test_find_binding.test_from_import_ascCsP| | dd¡¡| | dd¡¡| | ddd¡¡| | ddd¡¡dS)Nr1zimport bzimport b, c, drM)r)rKr+)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)Nr0zfrom x import arSzfrom 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)rKr+)rrrrÚtest_from_import_with_packageÚsz/Test_find_binding.test_from_import_with_packagecCs@| | ddd¡¡| | ddd¡¡| | ddd¡¡dS)Nr0zimport b.c as azb.cz import a as fÚf)r+rK)rrrrÚtest_import_as_with_packageçsz-Test_find_binding.test_import_as_with_packagecCs| | ddd¡¡| | ddd¡¡| | ddd¡¡| | ddd¡¡| | ddd¡¡| | ddd ¡¡| | ddd ¡¡dS) Nr0zfrom x import b as arSzfrom 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 tr1Út)r+rK)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) Nr0z 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)rKr+)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) Nr0z 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)rKr+)rrgrrrÚtest_class_defsz Test_find_binding.test_class_defcCs‚| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡dS) Nr0zfor 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)rKr+)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: passr0zP 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)rKr+)rrgrrrÚtest_for_nested"s z!Test_find_binding.test_for_nestedcCs(| | dd¡¡| | dd¡¡dS)Nr0zif b in r: a = czif a in r: d = e)r)rKr+)rrrrÚtest_ifSszTest_find_binding.test_ifcCs0d}| | d|¡¡d}| | d|¡¡dS)NzL if b in r: if c in d: a = cr0zL if b in r: if c in d: c = a)r)rKr+)rrgrrrÚtest_if_nestedWsz Test_find_binding.test_if_nestedcCs(| | dd¡¡| | dd¡¡dS)Nr0zwhile b in r: a = czwhile a in r: d = e)r)rKr+)rrrrÚ test_whiledszTest_find_binding.test_whilecCs0d}| | d|¡¡d}| | d|¡¡dS)NzR while b in r: while c in d: a = cr0zR while b in r: while c in d: c = a)r)rKr+)rrgrrrÚtest_while_nestedhsz#Test_find_binding.test_while_nestedcCs\d}| | d|¡¡d}| | d|¡¡d}| | d|¡¡d}| | d|¡¡dS)NzQ try: a = 6 except: b = 8r0zQ try: b = 8 except: a = 6zƒ try: b = 8 except KeyError: pass except: a = 6zQ try: b = 8 except: b = 6)r)rKr+)rrgrrrÚ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 = 8r0z› 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)rKr+)rrgrrrÚ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 = 9r0zR try: b = 8 finally: a = 6zR try: b = 8 finally: b = 6z| try: b = 8 except: b = 9 finally: b = 6)r)rKr+)rrgrrrÚ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 = 9r0z try: b = 8 finally: try: pass finally: a = 6zž try: b = 8 finally: try: b = 6 finally: b = 7)r)rKr+)rrgrrrÚtest_try_except_finally_nesteds   z0Test_find_binding.test_try_except_finally_nested)N)r#r$r%rKrYrZr[r\r]r^r_r`rarbrdrfrhrirjrkrlrmrnrorprqrrrsrrrrrX“s2      1  M!rXc@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)rr rrrÚtest_after_docstring'sz&Test_touch_import.test_after_docstringcCs*tdƒ}t dd|¡| t|ƒd¡dS)Nz"""foo""" import bar bar()ruz'"""foo""" import bar import foo bar() )rrrvrr)rr rrrÚtest_after_imports,sz$Test_touch_import.test_after_importscCs*tdƒ}t dd|¡| t|ƒd¡dS)Nzbar()ruzimport foo bar() )rrrvrr)rr rrrÚtest_beginning1sz Test_touch_import.test_beginningcCs*tdƒ}t dd|¡| t|ƒd¡dS)Nzbar()ÚhtmlÚescapezfrom html import escape bar() )rrrvrr)rr rrrr^6sz"Test_touch_import.test_from_importcCs*tdƒ}t dd|¡| t|ƒd¡dS)Nzbar()Zcgizimport cgi bar() )rrrvrr)rr rrrÚtest_name_import;sz"Test_touch_import.test_name_importN)r#r$r%rwrxryr^r|rrrrrt%s rtc@seZdZdd„Zdd„ZdS)ÚTest_find_indentationcCs:tj}tdƒ}| ||ƒd¡tdƒ}| ||ƒd¡dS)Nznode()Ú)rÚfind_indentationrr)rÚfir rrrÚ test_nothingBs z"Test_find_indentation.test_nothingcCsntj}tdƒ}| ||ƒd¡| ||jdjdjdƒd¡tdƒ}| ||jdjdjdƒd¡dS)Nzdef f(): x()r~rrBr'z zdef f(): x() y())rrrrr)rr€r rrrÚ test_simpleIs "z!Test_find_indentation.test_simpleN)r#r$r%rr‚rrrrr}@sr}N)r)Ú__doc__r~rZlib2to3.pytreerrÚlib2to3rZlib2to3.fixer_utilrrr r Z lib2to3.pgen2r rÚTestCaserr&r-r/r7r:rJrXrtr}rrrrÚs&     ,