B u9a<3у@sRddlZddlZddlmZGddДdeГZGddДdejГZedkrNeабdS)щN)┌BytecodeTestCasec@sЬeZdZddДZddДZddДZddДZd d ДZddДZd dДZ ddДZ ddДZddДZddДZ ddДZddДZddДZddДZdd ДZd!d"ДZd#d$ДZd%S)&┌ TestTranformscCs0ddД}|а|dб|а|dб|а|dбdS)NcSs|dks ~dS)Nщй)┌xrr·$/usr/lib/python3.7/test_peepholer.py┌unot sz%TestTranforms.test_unot..unotZ UNARY_NOT┌POP_JUMP_IF_FALSEZPOP_JUMP_IF_TRUE)┌assertNotInBytecode┌assertInBytecode)┌selfrrrr┌ test_unotszTestTranforms.test_unotcCs0x*dD]"\}}t|ddГ}|а|d|бqWdS)N))z not a is bzis not)z not a in bznot in)znot a is not b┌is)znot a not in b┌in┌┌singleZ COMPARE_OP)┌compiler)r┌lineZcmp_op┌coderrr┌test_elim_inversion_of_is_or_insz-TestTranforms.test_elim_inversion_of_is_or_incCs|ddД}ddД}ddД}x<|df|df|dffD]"\}}|а|d б|а|d |бq0WddД}|а|d б|а|d dбdS)NcSsd}d}|S)Nr)rrrr┌fsz0TestTranforms.test_global_as_constant..fcSsd}|S)NTr)rrrr┌g!sz0TestTranforms.test_global_as_constant..gcSsd}|S)NFr)rrrr┌h$sz0TestTranforms.test_global_as_constant..hTFZLOAD_GLOBAL┌ LOAD_CONSTcSsdS)z1Adding a docstring made this test fail in Py2.5.0Nrrrrrr,s)r r)rrrr┌func┌elemrrr┌test_global_as_constants z%TestTranforms.test_global_as_constantcCs@ddД}xdD]}|а||бqWxdD]}|а||бq(WdS)NcSs xqWtS)N)┌listrrrrr5sz'TestTranforms.test_while_one..f)rr )┌ JUMP_ABSOLUTE)r r)rrrrrr┌test_while_one3s zTestTranforms.test_while_onecCsFx@dD]8\}}t|ddГ}|а||б|а|dб|а|dбqWdS)N))za, = a,r)za, b = a, bZROT_TWO)za, b, c = a, b, cZ ROT_THREErr┌BUILD_TUPLEZUNPACK_TUPLE)rrr )rrrrrrr┌test_pack_unpack>szTestTranforms.test_pack_unpackcCsМx6dD].\}}t|ddГ}|а|d|б|а|dбqWttttdГГГddГ}|а|dбddДtа|бDГ}|аt |Гd бd dД}dS)N))z a = 1,2,3)щrщ)z ("a","b","c"))┌a┌b┌c)z a,b,c = 1,2,3)r"rr#)z(None, 1, None))Nr"N)z((1, 2), 3, 4)))r"rr#щrrrr i'cSsg|]}|jdkr|СqS)r)┌opname)┌.0┌instrrrr· YszETestTranforms.test_folding_of_tuples_of_constants..rcdSs╘ddddddddd d ddddddddd d ddddddddd d ddddddddd d ddddddddd d ddddddddd d ddddddddd d ddddddddd d ddddddddd d ddddddddd d gdfdS)Nrr"rr#r'щщщщщ rrrrr┌crater`szATestTranforms.test_folding_of_tuples_of_constants..crater) rrr ┌repr┌tuple┌range┌dis┌get_instructions┌assertEqual┌len)rrrrZload_constsr1rrr┌#test_folding_of_tuples_of_constantsIsz1TestTranforms.test_folding_of_tuples_of_constantscCs<x6dD].\}}t|ddГ}|а|d|б|а|dбqWdS)N))za in [1,2,3])r"rr#)za not in ["a","b","c"])r$r%r&)za in [None, 1, None])Nr"N)za not in [(1, 2), 3, 4]))r"rr#r'rrrZ BUILD_LIST)rrr )rrrrrrr┌"test_folding_of_lists_of_constantsns z0TestTranforms.test_folding_of_lists_of_constantsc Cs╓xДdtdddhГfdtdddhГfd tddhГfd tdddhГfd tdddhГffD].\}}t|ddГ}|а|dб|а|d|бqTWddД}ddД}|а|dГб|а|dГб|а|dГб|а|dГбdS)Nza in {1,2,3}r"rr#za not in {"a","b","c"}r$r&r%za in {None, 1, None}za not in {(1, 2), 3, 4})r"rr'za in {1, 2, 3, 3, 2, 1}rrZ BUILD_SETrcSs|dkS)N>r"rr#r)r$rrrrИsz:TestTranforms.test_folding_of_sets_of_constants..fcSs|dkS)N>r"rr#r)r$rrrrЛsz:TestTranforms.test_folding_of_sets_of_constants..g)┌ frozensetrr rZ assertTrue)rrrrrrrrr┌!test_folding_of_sets_of_constantszsz/TestTranforms.test_folding_of_sets_of_constantscCsxPdD]H\}}t|ddГ}|а|d|бx$tа|бD]}|а|jаdббq4WqWtdddГ}|а|ddб|а|ddбtd ddГ}|а|dd б|аdd |jбtdddГ}|а|dd б|аdd >|jбtdddГ}|а|dd б|аdd |jбdS)N))z a = 2+3+4r0)z"@"*4z@@@@)za="abc" + "def"Zabcdef)za = 3**4щQ)za = 3*4щ)z a = 13//4r#)za = 14%4r)za = 2+3r,)za = 13-4r0)za = (12,13)[1]щ )za = 13 << 2щ4)za = 13 >> 2r#)z a = 13 & 7r,)z a = 13 ^ 7щ )z a = 13 | 7щrrr┌BINARY_za=2+"b"rr%za="x"*10000i'rz a=1<<1000iшr"z a=2**1000) rrr5r6┌assertFalser(┌ startswithZassertNotIn┌ co_consts)rrrrr*rrr┌#test_folding_of_binops_on_constantsХs$z1TestTranforms.test_folding_of_binops_on_constantscCsОtdddГ}|а|ddб|а|dбtdddГ}|а|ddб|а|dбtd ddГ}|а|dd б|а|dбtdddГ}|а|dбdS)Nz"foo"[0]rrrrZ BINARY_SUBSCRu "aя┐┐"[1]uя┐┐u "ЁТНЕ"[0]uЁТНЕz "fuu"[10])rrr )rrrrr┌test_binary_subscr_on_unicode╝sz+TestTranforms.test_binary_subscr_on_unicodecCs╛xPdD]H\}}t|ddГ}|а|d|бx$tа|бD]}|а|jаdббq4WqWddД}x$tа|бD]}|а|jаdббqfWx8dD]0\}}}t|ddГ}|а|d|б|а||бqЖWdS) N))z-0.5gр┐)z-0.0gА)z -(1.0-1.0)gА)z-0r)z~-2r")z+1r"rrr┌UNARY_cSsdS)NgАrrrrr┌negzero▐szDTestTranforms.test_folding_of_unaryops_on_constants..negzero))z-"abc"┌abcZUNARY_NEGATIVE)z~"abc"rKZUNARY_INVERT)rrr5r6rDr(rE)rrrrr*rJr(rrr┌%test_folding_of_unaryops_on_constants╧sz3TestTranforms.test_folding_of_unaryops_on_constantscCs>ddД}|а|ddбddДtа|бDГ}|аt|ГdбdS)NcSs|S)Nr)rrrrrяsz/TestTranforms.test_elim_extra_return..frcSsg|]}|jdkr|СqS)┌RETURN_VALUE)r()r)r*rrrr+Єsz8TestTranforms.test_elim_extra_return..r")r r5r6r7r8)rr┌returnsrrr┌test_elim_extra_returnэsz$TestTranforms.test_elim_extra_returncCsHddД}|а|dб|а|dбddДtа|бDГ}|аt|ГdбdS)NcSs|r|S|S)Nr)ZcondZ true_valueZfalse_valuerrrr°sz1TestTranforms.test_elim_jump_to_return..f┌JUMP_FORWARDrcSsg|]}|jdkr|СqS)rM)r()r)r*rrrr+№sz:TestTranforms.test_elim_jump_to_return..r)r r5r6r7r8)rrrNrrr┌test_elim_jump_to_returnЎs z&TestTranforms.test_elim_jump_to_returncCsHddД}|а|dб|а|dбddДtа|бDГ}|аt|ГdбdS)NcSs(|rdS|rdSxdSx|r dSdSdS)Nr"rr#r'r,r-r)┌cond1┌cond2rrrrsz5TestTranforms.test_elim_jump_after_return1..frPrcSsg|]}|jdkr|СqS)rM)r()r)r*rrrr+ sz>TestTranforms.test_elim_jump_after_return1..r-)r r5r6r7r8)rrrNrrr┌test_elim_jump_after_return1s z*TestTranforms.test_elim_jump_after_return1cCs`ddД}|а|dбddДtа|бDГ}|аt|ГdбddДtа|бDГ}|аt|ГdбdS) NcSsx|rdSqWdS)Nr'r)rRrSrrrrsz5TestTranforms.test_elim_jump_after_return2..frPcSsg|]}|jdkr|СqS)r)r()r)r*rrrr+sz>TestTranforms.test_elim_jump_after_return2..r"cSsg|]}|jdkr|СqS)rM)r()r)r*rrrr+sr)r r5r6r7r8)rrrNrrr┌test_elim_jump_after_return2sz*TestTranforms.test_elim_jump_after_return2cCsddД}|а|dбdS)NcSsddЬddД}|S)Nr)┌returncSsdS)Nrrrrrr!szBTestTranforms.test_make_function_doesnt_bail..f..gr)rrrrr sz7TestTranforms.test_make_function_doesnt_bail..fZ BINARY_ADD)r )rrrrr┌test_make_function_doesnt_bailsz,TestTranforms.test_make_function_doesnt_bailc Cs~ddddddddd g }xb|D]Z}t|d dГ}xHtа|бD]:}|а|jаdбб|а|jаd бб|а|jаdббq8WqWdS)Nz3 * -5z-3 * 5z2 * (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}rrrIrCZBUILD_)rr5r6rDr(rE)rZexprs┌err*rrr┌test_constant_folding&s z#TestTranforms.test_constant_foldingcCsddД}|а|ГdбdS)NcSsdsdrdSdS)NTFr"rrrrrrr;sz=TestTranforms.test_condition_with_binop_with_bools..fr")r7)rrrrr┌$test_condition_with_binop_with_bools:sz2TestTranforms.test_condition_with_binop_with_boolsN)┌__name__┌ __module__┌__qualname__r rrrr!r9r:r<rGrHrLrOrQrTrUrWrYrZrrrrrs$ %' rc@seZdZddДZdS)┌TestBugletsc Cs(ddД}|аtбП|ГWdQRXdS)NcSsddh\}}||fS)Nr"r)r┌yrrrrHsz%TestBuglets.test_bug_11510..f)ZassertRaises┌ ValueError)rrrrr┌test_bug_11510DszTestBuglets.test_bug_11510N)r[r\r]rarrrrr^Bsr^┌__main__) r5ZunittestZtest.bytecode_helperrrZTestCaser^r[┌mainrrrr┌s>