dafRdZddlmZddlmZmZddlmZddlm Z m Z m Z m Z ddl mZddZGd d ej ZGd d ej ZGd dej ZGddeZGddeZGddeZGddej ZGddej ZGddej ZGddej Zy)z' Test suite for the code in fixer_util )support)NodeLeaf) fixer_util)AttrNameCallComma)tokenc|tj|}t|D]}|jd}d|_|S)Nr)r parse_stringrangechildrenparent)code strip_levelstreeis 6/root/Python-3.12.4/Lib/test/test_lib2to3/test_util.pyparser sB    %D <  }}Q DK KceZdZdZy) MacroTestCasect|ttfr$ttj j |}|jt||yN) isinstancetuplelistrrsyms simple_stmt assertEqualstr)selfnodestrings r assertStrzMacroTestCase.assertStrs: dUDM * 33T:D TF+rN)__name__ __module__ __qualname__r'rrrrs,rrceZdZdZdZdZy) Test_is_tuplecBtjt|dSNr)ris_tuplerr$r&s rr2zTest_is_tuple.is_tuples""5a#@AArcD|j|jd|j|jd|j|jd|j|jd|j|jdy)Nz(a, b)z (a, (b, c))z((a, (b, c)),)z(a,)z()) assertTruer2r$s r test_validzTest_is_tuple.test_valid sl  h/0  m45  &678  f-.  d+,rc|j|jd|j|jdy)Nz(a)z('foo') % (b, c)) assertFalser2r6s r test_invalidzTest_is_tuple.test_invalid's1 u-. '9:;rN)r(r)r*r2r7r:r+rrr-r-sB-zTest_is_list.is_list-s!!%Q"?@@rcD|j|jd|j|jd|j|jd|j|jd|j|jdy)Nz[]z[a]z[a, b]z [a, [b, c]]z[[a, [b, c]],])r5r>r6s rr7zTest_is_list.test_valid0sl  T*+  U+,  X./  ]34  %567rcD|j|jdy)Nz[]+[])r9r>r6s rr:zTest_is_list.test_invalid7s g./rN)r(r)r*r>r7r:r+rrr<r<,sA80rr<ceZdZdZdZy) Test_Attrctdd}|jttdtdd|jt|tddy)Nzfoo()r0r1aba.bzfoo().b)rr'rr )r$calls rtestzTest_Attr.test<sDW1- tDItCy159 tD$s),i8rc~ttdtd}|jt|ty)NrDrE)rr r"typer)r$attrs r test_returnszTest_Attr.test_returnsBs*DItCy) dT*rN)r(r)r*rHrLr+rrrBrB;s 9 +rrBceZdZdZy) Test_Namec|jtdd|jtdd|jtdddy)NrDz foo.foo().barrEprefixba)r'r r6s rrHzTest_Name.testHs< tCy#& tO,o> tC,d3rN)r(r)r*rHr+rrrNrNGs4rrNceZdZddZdZy) Test_CallNcg}t|trA|D],}|j||jt.|j t t |||S)zHelp the next test)rrappendr popr r )r$nameargsrQrargs r_CallzTest_Call._CallOsX dD ! )$( ) LLNDJ&11rc dttjdttjdttjdgttjdttjdttjdttjdgttjdttjddgg}|j |j d d |j |j d|dd |j |j d |dd |j |j d|dddy)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 ellA&U\\1(=ellA&(ellA&U\\1(=ellA&U\\1(=?ellC($u||S*MN  tzz#. tzz#tAw/< tzz)T!W57IJ tzz#tAwsz;ZHr)NN)r(r)r*r[rHr+rrrTrTNs 2 IrrTc$eZdZdZdZdZdZy)Test_does_tree_importctj||}|r|S|jD]}|j||}|s|cSyr)r find_bindingr_find_bind_rec)r$rXr%cchilds rrjz$Test_does_tree_import._find_bind_rechsI  # #D$ / Qh]] E##D%0A( rcjt|}|jd|}tj|||S)Nstart)rrjrdoes_tree_import)r$packagerXr&r%s rroz&Test_does_tree_import.does_tree_importqs2V}""7D1**7D$??rcd}|D]Z\}}}|j|||dz|z}|j||j|||dz|z}|j|\d}|D]Z\}}}|j|||dz|z}|j||j|||dz|z}|j|\y)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^f5LMA   Q %%gtVd]W5LMA   Q   ; '4  "GT7%%gtWt^f5LMA OOA %%gtVd]W5LMA OOA   rc&|jdy)Nzdef foo(): bar.baz() start=3)rr6s rtest_in_functionz&Test_does_tree_import.test_in_functions :;rN)r(r)r*rjrorrr+rrrgrggs@ 2 **3 => **3@Arc |j|jdd|j|jdd|j|jdd|j|jddy)NrDrrrvrwrr6s rtest_simple_importz$Test_find_binding.test_simple_importsl ))#z:; ))#/BCD **3 ;< **30@ABrc|j|jdd|j|jdd|j|jdd|j|jdd|j|jdd|j|jdd|j|jdd|j|jdd y) NrDr{ryr|r~rrrrtrurr6s rtest_from_importz"Test_find_binding.test_from_imports ))#/@AB ))#/@AB ))#/IJK ))#/BCD ))#/KLM **30ABC **30CDE **30CDErc |j|jdd|j|jdd|j|jdd|j|jddy)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_assl ))#?@ ))#/LMN **3@A **30JKLrc|j|jdd|j|jdd|j|jdd|j|jdd|j|jdd|j|jdd|j|jddy) 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 ))#/EFG ))#/MNO ))#/GHI ))#/JKL **30FGH **30HIJ **30HIJrc|j|jdd|j|jdd|j|jddd|j|jdddy)NrErvrwrkrr6s rtest_simple_import_with_packagez1Test_find_binding.test_simple_import_with_packagesp ))#z:; ))#/?@A **3 C@A **30@#FGrc|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|jdddy)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> ))#/@#FG ))#/@#FG ))#/@#FG ))#/I3OP ))#/BEJK ))#/BEJK ))#/KUST **30A3GH **30CUKL **30CUKL **30CUKLrc|j|jddd|j|jddd|j|jdddy)NrDzimport b.c as azb.crfrr6s rtest_import_as_with_packagez-Test_find_binding.test_import_as_with_packagesZ **30A5IJ **3EF **3EFrc|j|jddd|j|jddd|j|jddd|j|jddd|j|jddd|j|jddd |j|jddd y) NrDrrzrrr}rrrEtrr6s r test_from_import_as_with_packagez2Test_find_binding.test_from_import_as_with_packages **30FLM **30NPSTU **30H%PQ **30KUST **30FLM **30FLM **30FLMrc|j|jdd|j|jdd|j|jdd|j|jdd|j|jdd|j|jdd|j|jddd }|j|jd|y) 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_defs ))#?@ ))#/EFG ))#/?@A **30KLM **30BCD **30@AB **30@AB  **323rc|j|jdd|j|jdd|j|jdd|j|jdd|j|jdd|j|jdd|j|jdd|j|jdd |j|jdd |j|jdd d }|j|jd|y) 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_defs4 ))#?@ ))#/@AB ))#/ABC ))#/FGH **3@A **30BCD **30GHI **30FGH **30GHI **30@AB  **323rc|j|jdd|j|jdd|j|jdd|j|jdd|j|jdd|j|jdd|j|jddy) 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 ))#/ABC ))#/DEF ))#/FGH ))#/GHI ))#/IJK ))#/BCD **30BCDrc4d}|j|jd|d}|j|jd|d}|j|jd|d}|j|jd|d}|j|jd|d}|j|jd|d}|j|jd|d }|j|jd|y) 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"s  ))#q12  ))#q12  ))#q12  ))#q12  ))#q12  ))#q12  **323  **323rc|j|jdd|j|jddy)NrDzif b in r: a = czif a in r: d = err6s rtest_ifzTest_find_binding.test_ifSs8 ))#/ABC **30BCDrcd}|j|jd|d}|j|jd|y)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_nestedWH  ))#q12  **323rc|j|jdd|j|jddy)NrDzwhile b in r: a = czwhile a in r: d = err6s r test_whilezTest_find_binding.test_whileds8 ))#/DEF **30EFGrcd}|j|jd|d}|j|jd|y)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_nestedhrrcd}|j|jd|d}|j|jd|d}|j|jd|d}|j|jd|y)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  ))#q12  ))#q12  ))#q12  **323rcd}|j|jd|d}|j|jd|d}|j|jd|d}|j|jd|d}|j|jd|d}|j|jd|d}|j|jd|y) 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_nesteds  ))#q12  ))#q12  ))#q12  ))#q12  ))#q12  **323  **323rcd}|j|jd|d}|j|jd|d}|j|jd|d}|j|jd|y)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  ))#q12  ))#q12  **323  **323rcd}|j|jd|d}|j|jd|d}|j|jd|y)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_nestedsj  ))#q12  ))#q12  **323rr)r(r)r*rirrrrrrrrrrrrrrrrrrrrrrrrr+rrrrsEDNNB C FM KH MG N 44$E/4bE 4H 44>K4Z4B!4rrc*eZdZdZdZdZdZdZy)Test_touch_importc~td}tjdd||jt |dy)Nz"""foo""" bar()fooz"""foo""" import foo bar() rr touch_importr"r#r$r%s rtest_after_docstringz&Test_touch_import.test_after_docstring's4'(eT2 T$FGrc~td}tjdd||jt |dy)Nz"""foo""" import bar bar()rz'"""foo""" import bar import foo bar() rrs rtest_after_importsz$Test_touch_import.test_after_imports,s434eT2 T$RSrc~td}tjdd||jt |dy)Nbar()rzimport foo bar() rrs rtest_beginningz Test_touch_import.test_beginning12W~eT2 T$;rs-&66,G$$,