""cd}tj|D]}|j|kr|dz }t|dr|j}|jD]%}t|dr|t ||z }&|S)Nrr co_code)disget_instructionsopnamehasattrr co_constscount_instr_recursively)rrcountinstrcs rrrs E%a(( <6 ! ! QJEq* J [88 1i  8 ,Q77 7E Lrc eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZejddZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!dZ"d Z#d!Z$d"Z%d#Z&d$Z'd%S)& TestTranformsc .ttj|}d|D}|D]}d|jvr ||j}|jdvr4||jd|jd|jd|j|jdvr?|jdkr4||jd|jd|jd|jd|jvr=d|jvr4||jd|jd|jd|jdS) Nci|] }|j| S)offset.0rs r z4TestTranforms.check_jump_targets..(sAAA55<AAArJUMP_) JUMP_ABSOLUTE JUMP_FORWARDz at z jumps to RETURN_VALUE_OR_POPJUMP_IF_)listrrrargvalfailr#)selfcode instructionstargetsrtgts rcheck_jump_targetsz TestTranforms.check_jump_targets&sC06677 AALAAA! D DEel**%,'Cz>>> U\CCu|CC&)jCC69jCCDDD AAA n,, U\CCu|CC&)jCC69jCCDDDEL((Z3:-E-E U\CCu|CC&)jCC69jCCDDD D DrcR tj|}ttj|}t d|D}t d|D}||d||t|j dS)Nc3&K|] }|dV dSrNr"r%ts r z-TestTranforms.check_lnotab..A&00A1Q4000000rc3&K|] }|dV dSr8r"r9s rr;z-TestTranforms.check_lnotab..Br<rr) r_get_code_objectr-findlinestartsminmaxassertGreaterEqual assertLesslenr)r0r1lnotab min_bytecode max_bytecodes r check_lnotabzTestTranforms.check_lnotab;s:#D))c(..//0000000 0000000   a000  c$,&7&788888rcd}||d||d||d||d||dS)Nc|dks~dSdS)Nr"xs runotz%TestTranforms.test_unot..unotKs66AA6r UNARY_NOTPOP_JUMP_FORWARD_IF_FALSEPOP_JUMP_BACKWARD_IF_FALSEPOP_JUMP_FORWARD_IF_TRUEassertNotInBytecodeassertInBytecoderH)r0rNs r test_unotzTestTranforms.test_unotIs      {333   'BCCC   'CDDD d$>??? $rcdD]q\}}}||5t|dd}||||||dddn #1swxYwYrdS)N))z not a is bIS_OPr)znot a is not brXr)z not a in b CONTAINS_OPr)znot a not in brYrlinesingle)subTestcompilerUrH)r0r[cmp_opinvertr1s rtest_elim_inversion_of_is_or_inz-TestTranforms.test_elim_inversion_of_is_or_inTs% ( ( D&& 4(( ( (tR22%%dFF;;;!!$''' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (s>A))A- 0A- cd}d}d}|df|df|dffD]u\}}||5||d||d|||dddn #1swxYwYvd }||d||dd||dS) Ncd}d}|SNr"rLs rrz0TestTranforms.test_global_as_constant..fbsAAHrc d}|S)NTr"rLs rgz0TestTranforms.test_global_as_constant..gfs AHrc d}|S)NFr"rLs rhz0TestTranforms.test_global_as_constant..his AHrTF)func LOAD_GLOBAL LOAD_CONSTc dSrer"r"rrrz0TestTranforms.test_global_as_constant..fss ?4r)r^rTrUrH)r0rrgrirjelems rtest_global_as_constantz%TestTranforms.test_global_as_constant`s\         t9q$i!U< ( (JD$4(( ( (((}===%%dL$???!!$''' ( ( ( ( ( ( ( ( ( ( ( ( ( ( (       M222 at444 !sAA??B B cd}dD]}|||dD]}|||||dS)Nc re)r-r"rrrz'TestTranforms.test_while_one..f}s  r)rlPOP_JUMP_IF_FALSE) JUMP_BACKWARDrS)r0rrns rtest_while_onezTestTranforms.test_while_one{s}   8 . .D  $ $Q - - - -& + +D  ! !!T * * * * !rcBdD]\}}||5t|dd}|||||d||d||dddn #1swxYwYdS)N))za, = a,rl)z a, b = a, bSWAP)za, b, c = a, b, crvrZr\r] BUILD_TUPLEUNPACK_SEQUENCEr^r_rUrTrHr0r[rnr1s rtest_pack_unpackzTestTranforms.test_pack_unpacks ( (JD$ 4(( ( (tBx00%%dD111((}===((/@AAA!!$'''  ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (sA)BB B c dD]\}}||5t|dd}||d|||d||dddn #1swxYwYtt t tddd}||ddtj |D}| t|d ||d }||dS) N))z a = 1,2,3rrK)z ("a","b","c")abr)z a,b,c = 1,2,3r})z(None, 1, None)NrN)z((1, 2), 3, 4)rrKr~rZr\r]rlrw'c(g|]}|jdk |S)rlrr$s r zETestTranforms.test_folding_of_tuples_of_constants..s.???!&!=!=!=!=!=rrKcgdfdS)N)drrrKr~r rrrKr~rrrrrrrrrKr~rrrrrrrrrKr~rrrrrrrrrKr~rrrrrrrrrKr~rrrrrrrrrKr~rrrrrrrrrKr~rrrrrrrrrKr~rrrrrrrrrKr~rrrrrrr"r"rrcraterzATestTranforms.test_folding_of_tuples_of_constants..craters*        r) r^r_rUrTrHreprtuplerangerr assertEqualrD)r0r[rnr1 load_constsrs r#test_folding_of_tuples_of_constantsz1TestTranforms.test_folding_of_tuples_of_constantss ( (JD$4(( ( (tBx00%%dL$???((}===!!$'''  ( ( ( ( ( ( ( ( ( ( ( ( ( ( (tE%,,//00H==   }555??#*>t*D*D???  [))1--- $    &!!!!!AA>>B B cdD]\}}||5t|dd}||d|||d||dddn #1swxYwYdS)N))z a in [1,2,3]r})za not in ["a","b","c"]r)za in [None, 1, None]r)za not in [(1, 2), 3, 4]rrZr\r]rl BUILD_LISTryrzs r"test_folding_of_lists_of_constantsz0TestTranforms.test_folding_of_lists_of_constantss ( (JD$4(( ( (tR22%%dL$???((|<<<!!$'''  ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (rc  dthdfdthdfdtddhfdthdfd thdffD]\}}|| 5t|d d }||d ||d|||dddn #1swxYwYd}d}||d||d ||||d ||d||dS)Nz a in {1,2,3}rrKr~za not in {"a","b","c"}>rrrza in {None, 1, None}rza not in {(1, 2), 3, 4}>rr~rza in {1, 2, 3, 3, 2, 1}rZr\r] BUILD_SETrlc |dvSNrr"rs rrz:TestTranforms.test_folding_of_sets_of_constants..fs  > !rc |dvSrr"rs rrgz:TestTranforms.test_folding_of_sets_of_constants..gs I% %rr~r) frozensetr^r_rTrUrH assertTrue)r0r[rnr1rrgs r!test_folding_of_sets_of_constantsz/TestTranforms.test_folding_of_sets_of_constantssYyyy11 2 %y'A'A B #Y4y%9%9 : & ...(A(A B & )))(<(< =  ( (JD$4(( ( (tR22(({;;;%%dL$???!!$'''  ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( " " " & & & ! AAaDD!!! ! AAaDD!!! ! !s8ACC C c>dD]\}}||5t|dd}||d|tj|D]/}||jd0||dddn #1swxYwYtddd}||dd||dd ||td dd}||dd | d d z|j ||td dd}||dd| ddz|j ||tddd}||dd| ddz|j ||dS)N))z a = 2+3+4r)z"@"*4z@@@@)za="abc" + "def"abcdef)za = 3**4Q)za = 3*4 )z a = 13//4r~)za = 14%4rK)za = 2+3r)za = 13-4r)za = (12,13)[1] )z a = 13 << 24)z a = 13 >> 2r~)z a = 13 & 7r)z a = 13 ^ 7 )z a = 13 | 7rZr\r]rlBINARY_za=2+"b"rKrz a="x"*10000rrMz a=1<<1000irz a=2**1000) r^r_rUrr assertFalser startswithrH assertNotInr)r0r[rnr1rs r#test_folding_of_binops_on_constantsz1TestTranforms.test_folding_of_binops_on_constantssm ( (JD$"4(( ( (tR22%%dL$??? 1$77IIE$$U\%<%.negzero?s:r))z-"abc"abcUNARY_NEGATIVE)z~"abc"r UNARY_INVERT) r^r_rUrrrrrrH)r0r[rnr1rrrs r%test_folding_of_unaryops_on_constantsz3TestTranforms.test_folding_of_unaryops_on_constants.sL ( (JD$4(( ( (tR22%%dL$??? 1$77HHE$$U\%<%> ? ? ? ? '"""#  ( ( D$4(( ( (tR22%%dL$???%%dF333!!$'''  ( ( ( ( ( ( ( ( ( ( ( ( ( ( (  ( (s%BB,,B0 3B0 2AFF F cd}||dddtj|D}|t |d||dS)Nc|Srer"rLs rrz/TestTranforms.test_elim_extra_return..fSsHrrlc(g|]}|jdk |Sr*rr$s rrz8TestTranforms.test_elim_extra_return..V.===U"\^;;;;;rr)rTrrrrDrHr0rreturnss rtest_elim_extra_returnz$TestTranforms.test_elim_extra_returnQs      L$777==c&:1&=&==== Wq))) !rzVFollowing gh-92228 the return has two predecessors and that prevents jump elimination.c:d}||||d||ddtj|D}|t |d||dS)Nc|r|n|Srer")cond true_value false_values rrz1TestTranforms.test_elim_jump_to_return..f_s"&%JJ$ &rr)r(c(g|]}|jdk |Srrr$s rrz:TestTranforms.test_elim_jump_to_return..frrrK)r5rTrrrrDrHrs rtest_elim_jump_to_returnz&TestTranforms.test_elim_jump_to_return[s & & & """   N333   O444==c&:1&=&==== Wq))) !rc`d}||||dS)Ncptr tstrtdSdSt dSre)rrdfoobazr"rrrz6TestTranforms.test_elim_jump_to_uncond_jump..fms@ EEEEErr5rHr0rs rtest_elim_jump_to_uncond_jumpz+TestTranforms.test_elim_jump_to_uncond_jumpksA    """ !rc`d}||||dS)NcJ|r tstrt}|dSdSre)rrrrs rrz7TestTranforms.test_elim_jump_to_uncond_jump2..fzsA A      rrrs rtest_elim_jump_to_uncond_jump2z,TestTranforms.test_elim_jump_to_uncond_jump2xsA    """ !rcd}|||||t|ddd}|||||t|ddd}||||||d||d||dd}||||||d||d||d dS) Nc|o|o|Srer"rs rrz7TestTranforms.test_elim_jump_to_uncond_jump3..fsW1 rJUMP_IF_FALSE_OR_POPrKc|p|p|Srer"rs rrz7TestTranforms.test_elim_jump_to_uncond_jump3..fsV! rJUMP_IF_TRUE_OR_POPc|r|p|Srer"rs rrz7TestTranforms.test_elim_jump_to_uncond_jump3..fsW1 rrPc|s|o|Srer"rs rrz7TestTranforms.test_elim_jump_to_uncond_jump3..fsV! rrR)r5rHrrrTrUrs rtest_elim_jump_to_uncond_jump3z,TestTranforms.test_elim_jump_to_uncond_jump3s    """ ! 04JKKQOOO    """ ! 04IJJANNN    """ !   $:;;; a!6777 a!<===    """ !   $9::: a!7888 a!;<<<<.fs  QQ QQ qrr)r(c(g|]}|jdk |Srrr$s rrz>TestTranforms.test_elim_jump_after_return1..rrr)rTrrassertLessEqualrDrHrs rtest_elim_jump_after_return1z*TestTranforms.test_elim_jump_after_return1s      N333   O444==c&:1&=&==== S\\1--- !rcbd}||d||dS)Ncdd}|S)NreturnrKcdSrer"r"rrrgzBTestTranforms.test_make_function_doesnt_bail..f..gsr)rrKr")rgs rrz7TestTranforms.test_make_function_doesnt_bail..fs    Hr BINARY_OP)rTrHrs rtest_make_function_doesnt_bailz,TestTranforms.test_make_function_doesnt_bailsC      K000 !rcgd}|D]}||5t|dd}tj|D]}||jd||jd||jd||dddn #1swxYwYdS)N) z3 * -5z-3 * 5z 2 * (3 * 4)z (2 * 3) * 4z (-1, 2, 3)z (1, -2, 3)z (1, 2, -3)z(1, 2, -3) * 6z:lambda x: x in {(3 * -5) + (-1 - 6), (1, -2, 3) * 2, None})er\r]rrBUILD_)r^r_rrrrrrH)r0exprsrr1rs rtest_constant_foldingz#TestTranforms.test_constant_foldingsH     ( (A"" ( (q"h// 1$77HHE$$U\%<%.containtestsA; rrrrrrH)r0rs rtest_in_literal_listz"TestTranforms.test_in_literal_listsL    0lKKQOOO +&&&&&rc~d}|t|dd||dS)Nc(ttfD]}dSrerrrLs rforloopz8TestTranforms.test_iterate_literal_list..forloops"V    rrrr)r0rs rtest_iterate_literal_listz'TestTranforms.test_iterate_literal_listsN    0,GGKKK '"""""rcrd}||d||dS)Nc dS)NTFrr"r"rrrz=TestTranforms.test_condition_with_binop_with_bools..f  qrr)rrHrs r$test_condition_with_binop_with_boolsz2TestTranforms.test_condition_with_binop_with_boolssI    a    !rcrd}||d||dS)Nc|rnndSdS)NTFr"rLs rrz3TestTranforms.test_if_with_if_expression..fs   t5rT)rrHrs rtest_if_with_if_expressionz(TestTranforms.test_if_with_if_expressionsG    $    !rc6d}||dS)Nc dS)Nrr~r"rLs rrz+TestTranforms.test_trailing_nops..frr)rHrs rtest_trailing_nopsz TestTranforms.test_trailing_nopss-    !rc>d}|t|ddd}|t|ddd}|t|ddd}|t|dddS)Nc$dtDS)Nc0g|]}t|}|Sr"rr%rMys rrz[TestTranforms.test_assignment_idiom_in_comprehensions..listcomp.."111!1Q44QA111rrr"rrlistcompzGTestTranforms.test_assignment_idiom_in_comprehensions..listcomp11q111 1rFOR_ITERrc$dtDS)Nc0h|]}t|}|Sr"r r s r zYTestTranforms.test_assignment_idiom_in_comprehensions..setcomp..r rrr"rrsetcompzFTestTranforms.test_assignment_idiom_in_comprehensions..setcomprrc$dtDS)Nc2i|]}t|}||Sr"r r s rr&z[TestTranforms.test_assignment_idiom_in_comprehensions..dictcomp..s$444QaddAq444rrr"rrdictcompzGTestTranforms.test_assignment_idiom_in_comprehensions..dictcomps44!444 4rc$dtDS)Nc38K|]}t|}|VdSrer r s rr;zYTestTranforms.test_assignment_idiom_in_comprehensions..genexpr.. s,11!1Q44QA111111rrr"rrgenexprzFTestTranforms.test_assignment_idiom_in_comprehensions..genexpr rr)rr)r0r rrrs r'test_assignment_idiom_in_comprehensionsz5TestTranforms.test_assignment_idiom_in_comprehensionss 2 2 2 0:FFJJJ 2 2 2 0*EEqIII 5 5 5 0:FFJJJ 2 2 2 0*EEqIIIIIrc d}gtddtddtddttdtd fd }gtd d d d }|D]\}}|D]\}}tt|dzD]}t ||D]} dd| z|z|z|z} || |5| |z} t| dd|i} || | d| dddn #1swxYwYdS)Nz-+ #0)r\1234uабвгsra)i.duioxX)|EJ@g|EJ duifegFEGinffegFEG)r\130)r\.z.0z.2)r\.40)r%r'r%r\)fmtvaluez % (x,)rMzfmt = ) rfloatrrDrjoinr^evalr) r0flags testcases width_precsr*suffixwidthprecrspecr)s1s2s rtest_format_combinationsz&TestTranforms.test_format_combinationss -u 5 5 ]H - - 2K @ @ eEllU5\\M2H = =   _&; < <    ' B BME6* B B ts5zzA~..BBA ,UA 6 6BB!BGGDMM1E9D@6I!\\c\??BB!$uB!%&7&7&7#u!F!FB ,,R[[[AAABBBBBBBBBBBBBBBBB B B Bs15D2 2D69D6cd}||dd||ddd||ddd||ddd||d dd ||dd d ||d d d ||dd d||ddd||ddd||dddddS)Nc (dtt|D}t|dkrd|dzdz}ndd|zdz}t|d|t t ||S) Ncg|] }d|dz S)rMrr")r%is rrzBTestTranforms.test_format_misc..format..'s":::!I!II:::rr(rz,)z, )z % )rrDr,r-dictzip)r)valuesvarsargss rformatz.TestTranforms.test_format_misc..format&s::uS[['9'9:::D4yyA~~T!W}t+TYYt__,s23++T++T#dF2C2C-D-DEE Erstringzx = %s!rz x = 1234!zx = %d!zx = %x!zx = 4d2!zx = %f!zx = 1234.000000!r zx = 1234.5678901!zx = 1234.567890!z x = %s%% %%%%z x = 1234% %%z%% %sz x = %% %s!zx = %s, y = %dr"zx = 12, y = 34)r)r0rDs rtest_format_misczTestTranforms.test_format_misc%s F F F ))8444  400+>>>  400+>>>  400*===  4002DEEE  <88:MNNN  <88:LMMM  <88+FFF 66GGG  733\BBB  0"b99;KLLLLLrc|td5tddddn #1swxYwY|td5tdddddddn #1swxYwY|td5td d d idddn #1swxYwY|td5td d d idddn #1swxYwY|td5td d d idddn #1swxYwY|td5tdd d idddn #1swxYwY|td5tdd didddn #1swxYwY|td5tdd didddn #1swxYwY|td5tdd didddn #1swxYwY|td5tdd didddn #1swxYwY|td5tddgddddn #1swxYwY|td5tddddgdddddS#1swxYwYdS)Nz¬ enough arguments for format stringz '%s' % ()z4not all arguments converted during string formattingz '%s' % (x, y)rrKrMr zincomplete formatz '%s%' % (x,)rMrz'%s%%%' % (x,)z '%s%z' % (x,)zunsupported format characterz'%s%z' % (x, 5)z"a real number is required, not strz '%d' % (x,)rz!an integer is required, not floatz '%x' % (x,)g ףp=J@zan integer is required, not strzmust be real number, not strz '%f' % (x,)z'%s, %s' % (x, *y)r~)assertRaisesRegex TypeErrorr- ValueErrorr0s rtest_format_errorsz TestTranforms.test_format_errors:s?  # #I<>>                      # #IJLL 4 4 "2"2 3 3 3 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 # #J0C D D . . #t - - - . . . . . . . . . . . . . . .  # #J0C D D 0 0 !C; / / / 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0  # #I<>> / / 3+ . . . / / / / / / / / / / / / / / / # #J0N O O 1 1 "S$K 0 0 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1  # #I/S T T / / f . . . / / / / / / / / / / / / / / /  # #I/R S S 0 0 g / / / 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0  # #I/P Q Q / / f . . . / / / / / / / / / / / / / / /  # #I/M N N / / f . . . / / / / / / / / / / / / / / /  # #I<>> : : %QR'8'8 9 9 9 : : : : : : : : : : : : : : : # #IJLL > > %QaV'<'< = = = > > > > > > > > > > > > > > > > > >s8<<A>>BB$CC C)DD D .E  EE3FFF8GGG=HH #H I!!I%(I%J&&J*-J* K,,K03K0L55L9<L9c8d}||ddS)Nc||}}||}}dSrer"rs rrz5TestTranforms.test_static_swaps_unpack_two..fZsaqAaqAAArrvrTrs rtest_static_swaps_unpack_twoz*TestTranforms.test_static_swaps_unpack_twoYs/      F+++++rc8d}||ddS)NcN|||}}}|||}}}|||}}}|||}}}|||}}}|||}}}dSrer"rs rrz7TestTranforms.test_static_swaps_unpack_three..f`sTA!qAA!qAA!qAA!qAA!qAA!qAAArrvrQrs rtest_static_swaps_unpack_threez,TestTranforms.test_static_swaps_unpack_three_s/      F+++++rctdddD]d\}}}d|d|d|d}||5t|}||ddddn #1swxYwYedS) N_a_b_cz{'a': z, 'b': z, 'c': }rv)rr^rrT)r0rrrr r1s rtest_static_swaps_match_mappingz-TestTranforms.test_static_swaps_match_mappingistT400 7 7GAq!999!99A999Gg&& 7 77@@((v666 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7s&A++A/ 2A/ cgd}tdddD]t\}}}|D]k}||||}||5t|}||ddddn #1swxYwYludS)N)z C({}, {}, {})zC({}, {}, c={})zC({}, b={}, c={})zC(a={}, b={}, c={})rWrXrYrv)rrDr^rrT)r0formsrrrformr r1s rtest_static_swaps_match_classz+TestTranforms.test_static_swaps_match_classps    tT400 ; ;GAq! ; ;++aA..\\'**;;;GDDD,,T6:::;;;;;;;;;;;;;;; ; ; ;s &A>>B B cXhd}gd}tdddD]\}}}|D]}||||}||5t|}||vr||dn||ddddn #1swxYwYdS)N>*_, b, ca, *_, ca, b, *_)z {}, {}, {}z {}, {}, *{}z {}, *{}, {}z *{}, {}, {}rWrXrYrv)rrDr^rrUrT) r0swapsr]rrrr^r r1s r test_static_swaps_match_sequencez.TestTranforms.test_static_swaps_match_sequence~s444KKKtT400 ? ?GAq! ? ?++aA..\\'**??;GDDD%''--dF;;;;00v>>>??????????????? ? ? ?sABB! $B! N)(__name__ __module__ __qualname__r5rHrVrbrortr{rrrrrrrunittestskiprrrrrrrrrrrrrr8rGrNrRrUr[r_rer"rrrr$sSDDD* 9 9 9     ( ( (6    ( ( ('"'"'"R ( ( (>+ + + Z   .!(!(!(FX]9::  ::       !=!=!=F$(((,''' ###    J J JBBB.MMM*>>>>,,, ,,,777 ; ; ; ? ? ? ? ?rrc&eZdZdZdZdZdZdS) TestBugletscd}|t5|ddddS#1swxYwYdS)Ncddh\}}||fS)Nrr"rIs rrz%TestBuglets.test_bug_11510..fsq6DAqa4Kr) assertRaisesrLrs rtest_bug_11510zTestBuglets.test_bug_11510s     z * *   ACCC                  s 6::cftdD] } t#tpt$rYwxYwdS)Nr)r Exception)r0r<s rtest_bpo_42057zTestBuglets.test_bpo_42057sPr  A )       s ..c(tddddS)Nzwhile True or spam: passr r_rMs rtest_bpo_45773_pop_jump_if_truez+TestBuglets.test_bpo_45773_pop_jump_if_trues*Hf=====rc(tddddS)Nzwhile True or not spam: passrur rvrMs r test_bpo_45773_pop_jump_if_falsez,TestBuglets.test_bpo_45773_pop_jump_if_falses.&AAAAArN)rfrgrhrprsrwryr"rrrlrlsU>>>BBBBBrrl__main__)r itertoolsrrrritest.support.bytecode_helperrrrrTestCaserlrfmainr"rrrs ++++++++999999 # # #   g ?g ?g ?g ?g ?$g ?g ?g ?TBBBBB(#BBB4 zHMOOOOOr