o JAf×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)ÚtokencCs,t |¡}t|ƒD]}|jd}q d|_|S)Nr)rZ parse_stringÚrangeÚchildrenÚparent)ÚcodeÚ strip_levelsZtreeÚi©rú./usr/lib/python3.10/lib2to3/tests/test_util.pyÚparse s   rc@óeZdZdd„ZdS)Ú MacroTestCasecCs0t|ttfƒrttjj|ƒ}| t|ƒ|¡dS©N) Ú isinstanceÚtupleÚlistrrZsymsZ simple_stmtÚ assertEqualÚstr)ÚselfÚnodeÚstringrrrÚ assertStrszMacroTestCase.assertStrN)Ú__name__Ú __module__Ú __qualname__r rrrrró rc@ó$eZdZdd„Zdd„Zdd„ZdS)Ú Test_is_tuplecCót t|dd¡S©Né©r)rÚis_tupler©rrrrrr+ózTest_is_tuple.is_tuplecCóT| | d¡¡| | d¡¡| | d¡¡| | d¡¡| | d¡¡dS)Nz(a, b)z (a, (b, c))z((a, (b, c)),)z(a,)z())Ú assertTruer+©rrrrÚ test_valid ó zTest_is_tuple.test_validcCs$| | d¡¡| | d¡¡dS)Nz(a)z('foo') % (b, c))Ú assertFalser+r0rrrÚ test_invalid'szTest_is_tuple.test_invalidN)r!r"r#r+r1r4rrrrr&ó r&c@r%)Ú Test_is_listcCr'r()rÚis_listrr,rrrr7-r-zTest_is_list.is_listcCr.)Nz[]z[a]z[a, b]z [a, [b, c]]z[[a, [b, c]],])r/r7r0rrrr10r2zTest_is_list.test_validcCs| | d¡¡dS)Nz[]+[])r3r7r0rrrr47szTest_is_list.test_invalidN)r!r"r#r7r1r4rrrrr6,r5r6c@óeZdZdd„Zdd„ZdS)Ú Test_AttrcCs@tddd}| ttdƒtdƒƒd¡| t|tdƒƒd¡dS)Nzfoo()r)r*ÚaÚbúa.bzfoo().b)rr rr)rÚcallrrrÚtest<s zTest_Attr.testcCs&ttdƒtdƒƒ}| t|ƒt¡dS)Nr:r;)rrrÚtyper)rÚattrrrrÚ test_returnsBszTest_Attr.test_returnsN)r!r"r#r>rArrrrr9;s r9c@r)Ú Test_NamecCs8| tdƒd¡| tdƒd¡| tdddd¡dS)Nr:z foo.foo().barr;©ÚprefixZba)r rr0rrrr>HszTest_Name.testN)r!r"r#r>rrrrrBGr$rBc@seZdZddd„Zdd„ZdS)Ú Test_CallNcCsFg}t|tƒr|D] }| |¡| tƒ¡q | ¡tt|ƒ||ƒS)zHelp the next test)rrÚappendr Úpopr r)rÚnameÚargsrDr Ú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ú rCÚAzA()zb(1,2,3)za.b().cza.b().c(1,3,2,4)Údz d(b, j))rr ÚNUMBERÚSTRINGr rK)rZkidsrrrr>Ys ÿÿû zTest_Call.test)NN)r!r"r#rKr>rrrrrENs  rEc@s,eZdZdd„Zdd„Zdd„Zdd„Zd S) ÚTest_does_tree_importcCs<t ||¡}|r |S|jD]}| ||¡}|r|Sq dSr)rÚ find_bindingr Ú_find_bind_rec)rrHrÚcÚchildrrrrVhs   þz$Test_does_tree_import._find_bind_reccCs"t|ƒ}| d|¡}t |||¡S)NÚstart)rrVrÚdoes_tree_import)rÚpackagerHrrrrrrZqs z&Test_does_tree_import.does_tree_importcCs¬d}|D]%\}}}| |||d|¡}| |¡| |||d|¡}| |¡qd}|D]%\}}}| |||d|¡}| |¡| |||d|¡}| |¡q.dS)N))r:r:úfrom a import b)úa.dr:úfrom a.d import b)zd.ar:úfrom d.a import b)Nr:úimport b)Nr:úimport b, c, dÚ ))r:r:úfrom a import a)Úxr:úfrom x import a)rdr:úfrom x import b, c, a, d)úx.br:úfrom x.b import a)rgr:úfrom x.b import b, c, a, d)Nr:úimport a)Nr:úimport b, c, a, d)rZr3r/)rrZ failing_testsr[rHÚimport_ÚnZ passing_testsrrrÚtry_withws    üzTest_does_tree_import.try_withcCs| d¡dS)Nzdef foo(): bar.baz() start=3)rnr0rrrÚtest_in_functionsz&Test_does_tree_import.test_in_functionN)r!r"r#rVrZrnrorrrrrTgs   rTc@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|ƒ|¡Sr)rrUr)rrHrr[rrrrU”r-zTest_find_binding.find_bindingcCóp| | 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/rUr3r0rrrÚtest_simple_assignment—ó z(Test_find_binding.test_simple_assignmentcCrq)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)rrr0rrrÚtest_tuple_assignmentŸrtz'Test_find_binding.test_tuple_assignmentcCrq)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)rrr0rrrÚtest_list_assignment§rtz&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©r3rUr0rrrÚtest_invalid_assignments¯óz*Test_find_binding.test_invalid_assignmentscCóL| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡dS)Nr:rjrkr`rarrr0rrrÚtest_simple_importµryz$Test_find_binding.test_simple_importcCs”| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd ¡¡dS) Nr:rercrfrhrir\r^r_rrr0rrrÚtest_from_import»sz"Test_find_binding.test_from_importcCrz)Nr:z import b as azimport b as a, c, a as f, dú import a as fzimport b, c as f, d as errr0rrrÚtest_import_asÅryz Test_find_binding.test_import_ascCs‚| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡| | dd¡¡dS) Nr:úfrom x import b as aúfrom x import g as a, d as búfrom x.b import t as aúfrom x.b import g as a, dúfrom a import b as tzfrom a.d import b as tzfrom d.a import b as trrr0rrrÚtest_from_import_asËóz%Test_find_binding.test_from_import_ascCsP| | dd¡¡| | dd¡¡| | ddd¡¡| | ddd¡¡dS)Nr;r`rarWrrr0rrrÚ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:rerdrczfrom x import *rfrhrgzfrom x.b import *rir\r^r]r_zfrom x.y import *r<rrr0rrrÚtest_from_import_with_packageÚsz/Test_find_binding.test_from_import_with_packagecCs@| | ddd¡¡| | ddd¡¡| | ddd¡¡dS)Nr:zimport b.c as azb.cr}Úfrwr0rrrÚtest_import_as_with_packageçsz-Test_find_binding.test_import_as_with_packagecCs| | ddd¡¡| | ddd¡¡| | ddd¡¡| | ddd¡¡| | ddd¡¡| | ddd ¡¡| | ddd ¡¡dS) Nr:rrdr€rrgr‚rƒr;Útrwr0rrrÚ 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(): passrr©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(): passrrrŒ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: passrrr0rrrÚtest_forr…zTest_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 = 7rrrŒrrrÚtest_for_nested"s z!Test_find_binding.test_for_nestedcCó(| | dd¡¡| | dd¡¡dS)Nr:zif b in r: a = czif a in r: d = errr0rrrÚtest_ifSózTest_find_binding.test_ifcCó0d}| | 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 = arrrŒrrrÚtest_if_nestedWóz Test_find_binding.test_if_nestedcCr’)Nr:zwhile b in r: a = czwhile a in r: d = errr0rrrÚ test_whiledr”zTest_find_binding.test_whilecCr•)NzR while b in r: while c in d: a = cr:zR while b in r: while c in d: c = arrrŒrrrÚtest_while_nestedhr—z#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 = 6úQ try: b = 8 except: b = 6rrrŒ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 = 6rša try: try: b = 8 except: c = d except: try: b = 6 except: t = 8 except: o = yrrrŒ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 = 6rrrŒ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 = 7rrrŒrrrÚtest_try_except_finally_nesteds  z0Test_find_binding.test_try_except_finally_nestedr)r!r"r#rUrsrurvrxr{r|r~r„r†r‡r‰r‹rŽrrr‘r“r–r˜r™r›rœrržrrrrrp“s4      1  M !rpc@s4eZdZdd„Zdd„Zdd„Zdd„Zd d „Zd S) ÚTest_touch_importcCó*tdƒ}t dd|¡| t|ƒd¡dS)Nz"""foo""" bar()Úfooz"""foo""" import foo bar() ©rrZ touch_importrr©rrrrrÚtest_after_docstring'óz&Test_touch_import.test_after_docstringcCr )Nz"""foo""" import bar bar()r¡z'"""foo""" import bar import foo bar() r¢r£rrrÚtest_after_imports,r¥z$Test_touch_import.test_after_importscCr )Núbar()r¡zimport foo bar() r¢r£rrrÚtest_beginning1r¥z Test_touch_import.test_beginningcCs*tdƒ}t dd|¡| t|ƒd¡dS)Nr§ZhtmlÚescapezfrom html import escape bar() r¢r£rrrr|6r¥z"Test_touch_import.test_from_importcCr )Nr§Zcgizimport cgi bar() r¢r£rrrÚtest_name_import;r¥z"Test_touch_import.test_name_importN)r!r"r#r¤r¦r¨r|rªrrrrrŸ%s  rŸc@r8)Ú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()r¬rrMr)z zdef f(): x() y())rr­rrr r®rrrÚ test_simpleIs "&z!Test_find_indentation.test_simpleN)r!r"r#r°r±rrrrr«@s r«N)r)Ú__doc__r¬rZlib2to3.pytreerrZlib2to3rZlib2to3.fixer_utilrrr r Z lib2to3.pgen2r rZTestCaserr&r6r9rBrErTrprŸr«rrrrÚs(      ,