a zeP@sZddlZddlZddlmZddZGdddeZGdddejZed krVe dS) N)BytecodeTestCasecCs^d}t|D]}|j|kr|d7}qt|dr6|j}|jD]}t|dr<|t||7}q<|S)Nr__code__co_code)disget_instructionsopnamehasattrr co_constscount_instr_recursively)frcountinstrcrG/opt/bitninja-python-dojo/embedded/lib/python3.9/test/test_peepholer.pyr s     r c@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd5d6Zd7d8Zd9S): TestTranformsc Cstt|}dd|D}|D]}d|jvr0q ||j}|jdvrl||jd|jd|jd|j|jdvr|jdkr||jd|jd|jd|jd|jvr d |jvr ||jd|jd|jd|jq dS) NcSsi|] }|j|qSr)offset.0rrrr z4TestTranforms.check_jump_targets..ZJUMP_) JUMP_ABSOLUTE JUMP_FORWARDz at z jumps to RETURN_VALUEZ_OR_POPZJUMP_IF_)listrrrZargvalZfailr)selfcodeZ instructionstargetsrZtgtrrrcheck_jump_targetss4    z TestTranforms.check_jump_targetscCs^t|}tt|}tdd|D}tdd|D}||d||t|j dS)z0Check that the lnotab byte offsets are sensible.css|]}|dVqdSrNrrtrrr 1rz-TestTranforms.check_lnotab..css|]}|dVqdSr rr!rrrr#2rrN) rZ_get_code_objectrZfindlinestartsminmaxZassertGreaterEqualZ assertLesslenr)rrZlnotabZ min_bytecodeZ max_bytecoderrr check_lnotab+s   zTestTranforms.check_lnotabcCs:dd}||d||d||d||dS)NcSs|dks ~dS)Nrxrrrunot;sz%TestTranforms.test_unot..unotZ UNARY_NOTPOP_JUMP_IF_FALSEPOP_JUMP_IF_TRUEassertNotInBytecodeassertInBytecoder')rr+rrr test_unot9s    zTestTranforms.test_unotcCs8dD].\}}}t|dd}||||||qdS)N))z not a is bIS_OPr)znot a is not br2r)z not a in b CONTAINS_OPr)znot a not in br3rsingle)compiler0r')rlineZcmp_opinvertrrrrtest_elim_inversion_of_is_or_inCs z-TestTranforms.test_elim_inversion_of_is_or_incCsdd}dd}dd}|df|df|dffD],\}}||d ||d |||q.d d}||d ||d d||dS) NcSs d}d}|SNrr)rrrr Psz0TestTranforms.test_global_as_constant..fcSsd}|S)NTrr)rrrgTsz0TestTranforms.test_global_as_constant..gcSsd}|S)NFrr)rrrhWsz0TestTranforms.test_global_as_constant..hTFZ LOAD_GLOBAL LOAD_CONSTcSsdS)z1Adding a docstring made this test fail in Py2.5.0Nrrrrrr `sr.)rr r;r<funcelemrrrtest_global_as_constantNs   z%TestTranforms.test_global_as_constantcCsBdd}dD]}|||q dD]}|||q"||dS)NcSsqtSr:)rrrrrr jsz'TestTranforms.test_while_one..f)r=r,rr.)rr r?rrrtest_while_onehs zTestTranforms.test_while_onecCsLdD]B\}}t|dd}|||||d||d||qdS)N))za, = a,r=)z a, b = a, bZROT_TWO)za, b, c = a, b, cZ ROT_THREEr4r5 BUILD_TUPLEZ UNPACK_TUPLEr6r0r/r'rr7r?rrrrtest_pack_unpackts      zTestTranforms.test_pack_unpackcCsdD]8\}}t|dd}||d|||d||qttttddd}||dddt|D}| t |d ||d d }||dS) N))z a = 1,2,3rr()z ("a","b","c")abr)z a,b,c = 1,2,3rG)z(None, 1, None)NrN)z((1, 2), 3, 4)rr(rHr4r5r=rC'cSsg|]}|jdkr|qS)r=rrrrr s zETestTranforms.test_folding_of_tuples_of_constants..r(cSsgdfdS)N)drrr(rHrO rrr(rHrOrSrTrUrVrWrrr(rHrOrSrTrUrVrWrrr(rHrOrSrTrUrVrWrrr(rHrOrSrTrUrVrWrrr(rHrOrSrTrUrVrWrrr(rHrOrSrTrUrVrWrrr(rHrOrSrTrUrVrWrrr(rHrOrSrTrUrVrWrrr(rHrOrSrTrUrVrWrrrrrcraterszATestTranforms.test_folding_of_tuples_of_constants..crater) r6r0r/r'reprtuplerangerr assertEqualr&)rr7r?rZ load_constsrXrrr#test_folding_of_tuples_of_constantss       z1TestTranforms.test_folding_of_tuples_of_constantscCsBdD]8\}}t|dd}||d|||d||qdS)N))z a in [1,2,3]rG)za not in ["a","b","c"]rI)za in [None, 1, None]rL)za not in [(1, 2), 3, 4]rMr4r5r= BUILD_LISTrDrErrr"test_folding_of_lists_of_constantss    z0TestTranforms.test_folding_of_lists_of_constantscCsdthdfdthdfdtddhfdthdfd thdffD]8\}}t|d d }||d ||d |||qJdd}dd}||d||d ||||d ||d||dS)Nz a in {1,2,3}rr(rHza not in {"a","b","c"}>rKrJrza in {None, 1, None}rza not in {(1, 2), 3, 4}>rNrHrOza in {1, 2, 3, 3, 2, 1}r4r5Z BUILD_SETr=cSs|dvSNr`rrJrrrr sz:TestTranforms.test_folding_of_sets_of_constants..fcSs|dvSrarrbrrrr;sz:TestTranforms.test_folding_of_sets_of_constants..grHrO) frozensetr6r/r0r' assertTrue)rr7r?rr r;rrr!test_folding_of_sets_of_constantss$     z/TestTranforms.test_folding_of_sets_of_constantscCs,dD]N\}}t|dd}||d|t|D]}||jdq0||qtddd}||dd||dd||td dd}||dd |d d |j ||td dd}||dd |dd >|j ||tddd}||dd |dd |j ||dS)N))z a = 2+3+4rW)z"@"*4z@@@@)za="abc" + "def"Zabcdef)za = 3**4Q)za = 3*4 )z a = 13//4rH)za = 14%4r()za = 2+3rS)za = 13-4rW)za = (12,13)[1] )z a = 13 << 24)z a = 13 >> 2rH)z a = 13 & 7rS)z a = 13 ^ 7 )z a = 13 | 7r4r5r=BINARY_za=2+"b"r(rKz a="x"*10000rPr*z a=1<<1000irz a=2**1000) r6r0rr assertFalser startswithr'Z assertNotInr )rr7r?rrrrr#test_folding_of_binops_on_constantss,          z1TestTranforms.test_folding_of_binops_on_constantscCstddd}||dd||d||tddd}||dd||d||td dd}||dd ||d||td dd}||d||dS) Nz"foo"[0]r4r5r=r Z BINARY_SUBSCRu "a￿"[1]u￿u "𒍅"[0]u𒍅z "fuu"[10]rD)rrrrrtest_binary_subscr_on_unicodes           z+TestTranforms.test_binary_subscr_on_unicodecCsdD]N\}}t|dd}||d|t|D]}||jdq0||qdd}t|D]}||jdqf||dD]:\}}}t|dd}||d||||||qdS) N))z-0.5g)z-0.0)z -(1.0-1.0)rq)z-0r)z~-2r)z+1rr4r5r=UNARY_cSsdS)Nrqrrrrrnegzero&szDTestTranforms.test_folding_of_unaryops_on_constants..negzero))z-"abc"abcZUNARY_NEGATIVE)z~"abc"rtZ UNARY_INVERT)r6r0rrrmrrnr')rr7r?rrrsrrrr%test_folding_of_unaryops_on_constantss      z3TestTranforms.test_folding_of_unaryops_on_constantscCsHdd}||ddddt|D}|t|d||dS)NcSs|Sr:rr)rrrr 9sz/TestTranforms.test_elim_extra_return..fr=cSsg|]}|jdkr|qSrrQrrrrrR<s z8TestTranforms.test_elim_extra_return..r)r/rrr\r&r'rr returnsrrrtest_elim_extra_return7s z$TestTranforms.test_elim_extra_returncCs\dd}||||d||dddt|D}|t|d||dS)NcSs |r|S|Sr:r)condZ true_valueZ false_valuerrrr Csz1TestTranforms.test_elim_jump_to_return..frrcSsg|]}|jdkr|qSrvrQrrrrrRJs z:TestTranforms.test_elim_jump_to_return..r()rr/rrr\r&r'rwrrrtest_elim_jump_to_returnAs   z&TestTranforms.test_elim_jump_to_returncCs dd}||||dS)NcSstrts trtntdSr:)rJrdfooZbazrrrrr Qs z6TestTranforms.test_elim_jump_to_uncond_jump..frr'rr rrrtest_elim_jump_to_uncond_jumpOs z+TestTranforms.test_elim_jump_to_uncond_jumpcCs dd}||||dS)NcSs|rts trt}qdSr:)rr|r}rbrrrr ^s z7TestTranforms.test_elim_jump_to_uncond_jump2..fr~rrrrtest_elim_jump_to_uncond_jump2\s z,TestTranforms.test_elim_jump_to_uncond_jump2cCsdd}|||||t|dddd}|||||t|dddd}||||||d||d||dd d}||||||d||d||d dS) NcSs |o |o |Sr:rrIrrrr jsz7TestTranforms.test_elim_jump_to_uncond_jump3..fZJUMP_IF_FALSE_OR_POPr(cSs |p |p |Sr:rrIrrrr qsZJUMP_IF_TRUE_OR_POPcSs |r|p |Sr:rrIrrrr xsr,cSs |s|o |Sr:rrIrrrr sr-)rr'r\r r/r0rrrrtest_elim_jump_to_uncond_jump3gs(             z,TestTranforms.test_elim_jump_to_uncond_jump3cCsRdd}||d||dddt|D}|t|d||dS)NcSs |rdS|rdSdS|rdSdS)Nrr(rHrOrSrTrZcond1Zcond2rrrr s z5TestTranforms.test_elim_jump_after_return1..frrcSsg|]}|jdkr|qSrvrQrrrrrRs z>TestTranforms.test_elim_jump_after_return1..rT)r/rrassertLessEqualr&r'rwrrrtest_elim_jump_after_return1s   z*TestTranforms.test_elim_jump_after_return1cCsjdd}||dddt|D}|t|dddt|D}|t|d||dS) NcSs|rdSqdS)NrOrrrrrr sz5TestTranforms.test_elim_jump_after_return2..frcSsg|]}|jdkr|qSrArQrrrrrRs z>TestTranforms.test_elim_jump_after_return2..rcSsg|]}|jdkr|qSrvrQrrrrrRs r()r/rrr\r&rr'rwrrrtest_elim_jump_after_return2s z*TestTranforms.test_elim_jump_after_return2cCs"dd}||d||dS)NcSsdddd}|S)Nr()returncSsdSr:rrrrrr;szBTestTranforms.test_make_function_doesnt_bail..f..gr)r;rrrr sz7TestTranforms.test_make_function_doesnt_bail..fZ BINARY_ADD)r/r'rrrrtest_make_function_doesnt_bails z,TestTranforms.test_make_function_doesnt_bailcCsrgd}|D]`}t|dd}t|D]:}||jd||jd||jdq&||q dS)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}r4r5rrrlZBUILD_)r6rrrmrrnr')rZexprserrrrrtest_constant_foldings  z#TestTranforms.test_constant_foldingcCs(dd}|t|dd||dS)NcSs tttfvSr:)r*rJrKrrrr containtestsz7TestTranforms.test_in_literal_list..containtestr^rr\r r')rrrrrtest_in_literal_listsz"TestTranforms.test_in_literal_listcCs(dd}|t|dd||dS)NcSsttfD]}qdSr:)rJrKr)rrrforloops z8TestTranforms.test_iterate_literal_list..forloopr^rr)rrrrrtest_iterate_literal_listsz'TestTranforms.test_iterate_literal_listcCs$dd}||d||dS)NcSsdsdr dSdS)NTFrrrrrrrr sz=TestTranforms.test_condition_with_binop_with_bools..fr)r\r'rrrr$test_condition_with_binop_with_boolssz2TestTranforms.test_condition_with_binop_with_boolscCs$dd}||d||dS)NcSs|rndrdSdS)NTFrr)rrrr s z3TestTranforms.test_if_with_if_expression..fT)rdr'rrrrtest_if_with_if_expressionsz(TestTranforms.test_if_with_if_expressioncCsdd}||dS)NcSsdSdS)NrHrSrTrr)rrrr sz+TestTranforms.test_trailing_nops..f)r'rrrrtest_trailing_nopssz TestTranforms.test_trailing_nopscCsldd}|t|dddd}|t|dddd}|t|ddd d }|t|dddS) NcSsddtDS)NcSsg|]}t|}|qSrr rr*yrrrrRrz[TestTranforms.test_assignment_idiom_in_comprehensions..listcomp..rbrrrrlistcompszGTestTranforms.test_assignment_idiom_in_comprehensions..listcompZFOR_ITERrcSsddtDS)NcSsh|]}t|}|qSrrrrrr rzYTestTranforms.test_assignment_idiom_in_comprehensions..setcomp..rbrrrrsetcompszFTestTranforms.test_assignment_idiom_in_comprehensions..setcompcSsddtDS)NcSsi|]}t|}||qSrrrrrrrrz[TestTranforms.test_assignment_idiom_in_comprehensions..dictcomp..rbrrrrdictcompszGTestTranforms.test_assignment_idiom_in_comprehensions..dictcompcSsddtDS)Ncss|]}t|}|VqdSr:rrrrrr#rzYTestTranforms.test_assignment_idiom_in_comprehensions..genexpr..rbrrrrgenexprszFTestTranforms.test_assignment_idiom_in_comprehensions..genexpr)r\r )rrrrrrrr'test_assignment_idiom_in_comprehensionssz5TestTranforms.test_assignment_idiom_in_comprehensionsN)__name__ __module__ __qualname__rr'r1r9r@rBrFr]r_rerorpruryr{rrrrrrrrrrrrrrrrrrs8    ( ,!   #  rc@seZdZddZddZdS) TestBugletscCs<dd}|t|Wdn1s.0YdS)NcSsddh\}}||fS)Nrr)r*rrrrr s z%TestBuglets.test_bug_11510..f)Z assertRaises ValueErrorrrrrtest_bug_11510s zTestBuglets.test_bug_11510c Cs2tdD]$}ztWqtpty*Yq0qdS)Nrj)r[ Exception)rirrrtest_bpo_42057 s  zTestBuglets.test_bpo_42057N)rrrrrrrrrrs r__main__) rZunittestZtest.support.bytecode_helperrr rZTestCaserrmainrrrrs  p