;eR ddlmZddlmZmZddlmZddlmZm Z m Z m Z ddl m Z ddZGdd ejZGd d ejZGd d ejZGddeZGddeZGddeZGddejZGddejZGddejZGddejZdS))support)NodeLeaf) fixer_util)AttrNameCallComma)tokencztj|}t|D]}|jd}d|_|S)Nr)r parse_stringrangechildrenparent)code strip_levelstreeis ./usr/lib/python3.11/lib2to3/tests/test_util.pyparser sE   % %D <   }QDK KceZdZdZdS) MacroTestCasect|ttfrttjj|}|t||dSN) isinstancetuplelistrrsyms simple_stmt assertEqualstr)selfnodestrings r assertStrzMacroTestCase.assertStrsN dUDM * * ; 3T::D TF+++++rN)__name__ __module__ __qualname__r'rrrrs#,,,,,rrc eZdZdZdZdZdS) Test_is_tuplecHtjt|dSNr)ris_tuplerr$r&s rr2zTest_is_tuple.is_tuples "5a#@#@#@AAArc||d||d||d||d||ddS)Nz(a, b)z (a, (b, c))z((a, (b, c)),)z(a,)z()) assertTruer2r$s r test_validzTest_is_tuple.test_valid s  h//000  m44555  &677888  f--...  d++,,,,,rc||d||ddS)Nz(a)z('foo') % (b, c)) assertFalser2r6s r test_invalidzTest_is_tuple.test_invalid'sI u--... '9::;;;;;rN)r(r)r*r2r7r:r+rrr-r-sDBBB---<<<<zTest_is_list.is_list-s !%Q"?"?"?@@@rc||d||d||d||d||ddS)Nz[]z[a]z[a, b]z [a, [b, c]]z[[a, [b, c]],])r5r>r6s rr7zTest_is_list.test_valid0s  T**+++  U++,,,  X..///  ]33444  %56677777rcV||ddS)Nz[]+[])r9r>r6s rr:zTest_is_list.test_invalid7s( g../////rN)r(r)r*r>r7r:r+rrr<r<,sDAAA88800000rr<ceZdZdZdZdS) Test_Attrctdd}|ttdtdd|t|tdddS)Nzfoo()r0r1aba.bzfoo().b)rr'rr )r$calls rtestzTest_Attr.test<shW1--- tDIItCyy115999 tD$s)),,i88888rcttdtd}|t|tdS)NrDrE)rr r"typer)r$attrs r test_returnszTest_Attr.test_returnsBs?DIItCyy)) dT*****rN)r(r)r*rHrLr+rrrBrB;s2999 +++++rrBceZdZdZdS) Test_Namec|tdd|tdd|tddddS)NrDz foo.foo().barrEprefixba)r'r r6s rrHzTest_Name.testHs_ tCyy#&&& tO,,o>>> tC,,,d33333rN)r(r)r*rHr+rrrNrNGs#44444rrNceZdZddZdZdS) Test_CallNc  g}t|trO|D]8}|||t9|t t |||Sr)rrappendr popr r )r$nameargsrQrargs r_CallzTest_Call._CallOs{  dD ! !  ) )$$$(((( LLNNNDJJ&111rc Ndttjdttjdttjdgttjdttjdttjdttjdgttjdttjddgg}||d d ||d|dd ||d |dd ||d|ddddS)Nrr0rEj rPAzA()zb(1,2,3)za.b().cza.b().c(1,3,2,4)dz d(b, j))rr NUMBERSTRINGr'r[)r$kidss rrHzTest_Call.testYs>elA&&U\1(=(=elA&&(elA&&U\1(=(=elA&&U\1(=(=?elC(($u|S*M*M*MN  tzz#... tzz#tAw//<<< tzz)T!W557IJJJ tzz#tAwsz;;ZHHHHHr)NN)r(r)r*r[rHr+rrrTrTNs<2222 I I I I IrrTc&eZdZdZdZdZdZdS)Test_does_tree_importctj||}|r|S|jD]}|||}|r|cSdSr)r find_bindingr_find_bind_rec)r$rXr%cchilds rrjz$Test_does_tree_import._find_bind_rechs^  #D$ / / Qh]  E##D%00AN(((N  rcxt|}|d|}tj|||S)Nstart)rrjrdoes_tree_import)r$packagerXr&r%s rroz&Test_does_tree_import.does_tree_importqs7V}}""7D11*7D$???rcd}|D]j\}}}||||dz|z}||||||dz|z}||kd}|D]j\}}}||||dz|z}||||||dz|z}||kdS)N))rDrDfrom a import b)a.drDfrom a.d import b)zd.arDfrom d.a import b)NrDimport b)NrDimport b, c, d ))rDrDfrom a import a)xrDfrom x import a)rzrDfrom x import b, c, a, d)x.brDfrom x.b import a)r}rDfrom x.b import b, c, a, d)NrDimport a)NrDimport b, c, a, d)ror9r5)r$r& failing_testsrprXimport_n passing_testss rtry_withzTest_does_tree_import.try_withws8 '4  "GT7%%gtWt^f5LMMA   Q   %%gtVd]W5LMMA   Q    ; '4   "GT7%%gtWt^f5LMMA OOA   %%gtVd]W5LMMA OOA       rc0|ddS)Nzdef foo(): bar.baz() start=3)rr6s rtest_in_functionz&Test_does_tree_import.test_in_functions :;;;;;rN)r(r)r*rjrorrr+rrrgrggsS@@@ 2<<<<>> **3 ==>>> **3@@AAAAArcN||dd||dd||dd||dddS)NrDrrrvrwrr6s rtest_simple_importz$Test_find_binding.test_simple_imports ))#z::;;; ))#/BCCDDD **3 ;;<<< **30@AABBBBBrc||dd||dd||dd||dd||dd||dd||dd||dd dS) NrDr{ryr|r~rrrrtrurr6s rtest_from_importz"Test_find_binding.test_from_imports& ))#/@AABBB ))#/@AABBB ))#/IJJKKK ))#/BCCDDD ))#/KLLMMM **30ABBCCC **30CDDEEE **30CDDEEEEErcN||dd||dd||dd||dddS)NrDz import b as azimport b as a, c, a as f, d import a as fzimport b, c as f, d as err6s rtest_import_asz Test_find_binding.test_import_ass ))#??@@@ ))#/LMMNNN **3@@AAA **30JKKLLLLLrcD||dd||dd||dd||dd||dd||dd||dddS) NrDfrom x import b as afrom x import g as a, d as bfrom x.b import t as afrom x.b import g as a, dfrom a import b as tzfrom a.d import b as tzfrom d.a import b as trr6s rtest_from_import_asz%Test_find_binding.test_from_import_ass ))#/EFFGGG ))#/MNNOOO ))#/GHHIII ))#/JKKLLL **30FGGHHH **30HIIJJJ **30HIIJJJJJrcR||dd||dd||ddd||ddddS)NrErvrwrkrr6s rtest_simple_import_with_packagez1Test_find_binding.test_simple_import_with_packages ))#z::;;; ))#/?@@AAA **3 C@@AAA **30@#FFGGGGGrc||ddd||ddd||ddd||ddd||ddd||dd d||dd d||dd d||dd d ||ddd ||ddddS)NrDr{rzryzfrom x import *r|r~r}zfrom x.b import *rrrrtrsruzfrom x.y import *rFrr6s rtest_from_import_with_packagez/Test_find_binding.test_from_import_with_packages ))#/@#FFGGG ))#/@#FFGGG ))#/@#FFGGG ))#/I3OOPPP ))#/BEJJKKK ))#/BEJJKKK ))#/KUSSTTT **30A3GGHHH **30CUKKLLL **30CUKKLLL **30CUKKLLLLLrc||ddd||ddd||ddddS)NrDzimport b.c as azb.crfrr6s rtest_import_as_with_packagez-Test_find_binding.test_import_as_with_packages{ **30A5IIJJJ **3EEFFF **3EEFFFFFrcR||ddd||ddd||ddd||ddd||ddd||ddd ||ddd dS) NrDrrzrrr}rrrEtrr6s r test_from_import_as_with_packagez2Test_find_binding.test_from_import_as_with_packages **30FLLMMM **30NPSTTUUU **30H%PPQQQ **30KUSSTTT **30FLLMMM **30FLLMMM **30FLLMMMMMrc||dd||dd||dd||dd||dd||dd||ddd }||d|dS) NrDz 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$ss rtest_function_defz#Test_find_binding.test_function_defs1 ))#??@@@ ))#/EFFGGG ))#/?@@AAA **30KLLMMM **30BCCDDD **30@AABBB **30@AABBB  **32233333rc||dd||dd||dd||dd||dd||dd||dd||dd ||dd ||dd d }||d|dS) NrDz 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(): passrrs rtest_class_defz Test_find_binding.test_class_defs ))#??@@@ ))#/@AABBB ))#/ABBCCC ))#/FGGHHH **3@@AAA **30BCCDDD **30GHHIII **30FGGHHH **30GHHIII **30@AABBB  **32233333rcD||dd||dd||dd||dd||dd||dd||dddS) NrDzfor 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: passrr6s rtest_forzTest_find_binding.test_fors ))#/ABBCCC ))#/DEEFFF ))#/FGGHHH ))#/GHHIII ))#/IJJKKK ))#/BCCDDD **30BCCDDDDDrcd}||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: passrDzP 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 = 7rrs rtest_for_nestedz!Test_find_binding.test_for_nested"sd  ))#q11222  ))#q11222  ))#q11222  ))#q11222  ))#q11222  ))#q11222  **322333  **32233333rc||dd||dddS)NrDzif b in r: a = czif a in r: d = err6s rtest_ifzTest_find_binding.test_ifSsP ))#/ABBCCC **30BCCDDDDDrcd}||d|d}||d|dS)NzL if b in r: if c in d: a = crDzL if b in r: if c in d: c = arrs rtest_if_nestedz Test_find_binding.test_if_nestedW`  ))#q11222  **32233333rc||dd||dddS)NrDzwhile b in r: a = czwhile a in r: d = err6s r test_whilezTest_find_binding.test_whiledsP ))#/DEEFFF **30EFFGGGGGrcd}||d|d}||d|dS)NzR while b in r: while c in d: a = crDzR while b in r: while c in d: c = arrs rtest_while_nestedz#Test_find_binding.test_while_nestedhrrc^d}||d|d}||d|d}||d|d}||d|dS)NzQ try: a = 6 except: b = 8rDzQ try: b = 8 except: a = 6z try: b = 8 except KeyError: pass except: a = 6Q try: b = 8 except: b = 6rrs rtest_try_exceptz!Test_find_binding.test_try_exceptus  ))#q11222  ))#q11222  ))#q11222  **32233333rc`d}||d|d}||d|d}||d|d}||d|d}||d|d}||d|d}||d|dS) Nz try: try: a = 6 except: pass except: b = 8rDz 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 = 6ra try: try: b = 8 except: c = d except: try: b = 6 except: t = 8 except: o = yrrs rtest_try_except_nestedz(Test_find_binding.test_try_except_nesteds9  ))#q11222  ))#q11222  ))#q11222  ))#q11222  ))#q11222  **322333  **32233333rc^d}||d|d}||d|d}||d|d}||d|dS)Nz| try: c = 6 except: b = 8 finally: a = 9rDzR try: b = 8 finally: a = 6zR try: b = 8 finally: b = 6z| try: b = 8 except: b = 9 finally: b = 6rrs rtest_try_except_finallyz)Test_find_binding.test_try_except_finallys  ))#q11222  ))#q11222  **322333  **32233333rcd}||d|d}||d|d}||d|dS)Nz try: c = 6 except: b = 8 finally: try: a = 9 except: b = 9 finally: c = 9rDz try: b = 8 finally: try: pass finally: a = 6z try: b = 8 finally: try: b = 6 finally: b = 7rrs rtest_try_except_finally_nestedz0Test_find_binding.test_try_except_finally_nesteds  ))#q11222  ))#q11222  **32233333rr)r(r)r*rirrrrrrrrrrrrrrrrrrrrrrrrr+rrrrsEEEEDDDNNNNNNBBB CCC FFFMMM KKKHHH M M MGGG N N N 4 4 4444$EEE/4/4/4bEEE 4 4 4HHH 4 4 4444>K4K4K4Z444B!4!4!4!4!4rrc,eZdZdZdZdZdZdZdS)Test_touch_importctd}tjdd||t |ddS)Nz"""foo""" bar()fooz"""foo""" import foo bar() rr touch_importr"r#r$r%s rtest_after_docstringz&Test_touch_import.test_after_docstring'sH'((eT222 T$FGGGGGrctd}tjdd||t |ddS)Nz"""foo""" import bar bar()rz'"""foo""" import bar import foo bar() rrs rtest_after_importsz$Test_touch_import.test_after_imports,sH344eT222 T$RSSSSSrctd}tjdd||t |ddS)Nbar()rzimport foo bar() rrs rtest_beginningz Test_touch_import.test_beginning1EW~~eT222 T$;<<<<rsq-&%%%%%%%666666666666,,,,,G$,,, < < < <