Your IP : 216.73.216.213
���]R1 � @ s� d d l Z d d l Z d d l Z d d l m Z d d l Z d d l m Z d d l m Z Gd d � d e � Z
Gd d � d e j � Z e
d k r� e j � d S)
� N)�StringIO)�copysign)�BytecodeTestCasec @ s� e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d � Z d! d" � Z d# S)$�
TestTranformsc C s@ d d � } | j | d � | j | d � | j | d � d S)Nc S s | d k s ~ d S)N� � )�xr r �3/opt/alt/python35/lib64/python3.5/test_peepholer.py�unot s z%TestTranforms.test_unot.<locals>.unotZ UNARY_NOT�POP_JUMP_IF_FALSEZPOP_JUMP_IF_TRUE)�assertNotInBytecode�assertInBytecode)�selfr
r r r � test_unot s zTestTranforms.test_unotc C s@ x9 d D]1 \ } } t | d d
� } | j | d | � q Wd S)N�
not a is b�is not�
not a in b�not in�not a is not b�is�not a not in b�in� �singleZ
COMPARE_OP�r r �r r �r r �r r )r r r r )�compiler
)r �lineZcmp_op�coder r r � test_elim_inversion_of_is_or_in s z-TestTranforms.test_elim_inversion_of_is_or_inc C s� d d � } d d � } d d � } xR | d f | d f | d f f D]/ \ } } | j | d � | j | d
| � qF Wd d � } | j | d � | j | d
d � d S) Nc S s d d | S)Nr )r r r r �f! s z0TestTranforms.test_global_as_constant.<locals>.fc S s d | S)NTr )r r r r �g% s z0TestTranforms.test_global_as_constant.<locals>.gc S s d | S)NFr )r r r r �h( s z0TestTranforms.test_global_as_constant.<locals>.hTFZ LOAD_GLOBAL�
LOAD_CONSTc S s d S)Nr r r r r r" . s )r r
)r r" r# r$ �func�elemr r r �test_global_as_constant s . z%TestTranforms.test_global_as_constantc C sR d d � } x d D] } | j | | � q Wx d D] } | j | | � q4 Wd S)Nc S s x q Wt S)N)�listr r r r r" 6 s z'TestTranforms.test_while_one.<locals>.fr% r �
JUMP_ABSOLUTE)r% r )r* )r r
)r r" r' r r r �test_while_one4 s
z TestTranforms.test_while_onec C s] xV d D]N \ } } t | d d � } | j | | � | j | d � | j | d
� q Wd S)N�a, = a,r% � a, b = a, b�ROT_TWO�a, b, c = a, b, c� ROT_THREEr r � BUILD_TUPLEZ UNPACK_TUPLE�r, r% �r- r. �r/ r0 )r2 r3 r4 )r r
r )r r r' r r r r �test_pack_unpack? s z TestTranforms.test_pack_unpackc C s� xI d! D]A \ } } t | d
d � } | j | d | � | j | d � q Wt t t t d � � � d
d � } | j | d � d d � t j | � D� } | j t | � d � d d � } d S)"N� a = 1,2,3� r � �
("a","b","c")�a�b�c�
a,b,c = 1,2,3�(None, 1, None)�((1, 2), 3, 4)� r r r% r1 i' c S s% g | ] } | j d k r | � q S)r% )�opname)�.0�instrr r r �
<listcomp>Z s zETestTranforms.test_folding_of_tuples_of_constants.<locals>.<listcomp>c d S s8 d d d d d d d d d d
d d d d d d d d d d
d d d d d d d d d d
d d d d d d d d d d
d d d d d d d d d d
d d d d d d d d d d
d d d d d d d d d d
d d d d d d d d d d
d d d d d d d d d d
d d d d d d d d d d
gd f d S) Nr r7 r r8 r@ � � � � � r r r r r �cratera s zATestTranforms.test_folding_of_tuples_of_constants.<locals>.crater�r7 r r8 �r6 rK �r: r; r< �r9 rM �r7 r r8 �r= rO �Nr7 N�r> rQ �r7 r �rS r8 r@ �r? rT )rL rN rP rR rU )
r r
r �repr�tuple�range�dis�get_instructions� assertEqual�len)r r r' r Z load_constsrJ r r r �#test_folding_of_tuples_of_constantsJ s $ z1TestTranforms.test_folding_of_tuples_of_constantsc C sP xI d D]A \ } } t | d d
� } | j | d | � | j | d � q Wd S)N� a in [1,2,3]r7 r r8 �a not in ["a","b","c"]r: r; r< �a in [None, 1, None]�a not in [(1, 2), 3, 4]r@ r r r% Z
BUILD_LIST�r7 r r8 �r^ rb �r: r; r< �r_ rd �Nr7 N�r` rf �r7 r �rh r8 r@ �ra ri )rc re rg rj )r r
r )r r r' r r r r �"test_folding_of_lists_of_constantso s z0TestTranforms.test_folding_of_lists_of_constantsc C s+ x� d t d d d h � f d t d d d h � f d t d d h � f d
t d d d h � f d t d d d h � f f D]A \ } } t | d
d � } | j | d � | j | d | � q| Wd d � } d d � } | j | d � � | j | d � � | j | d � � | j | d � � d S)Nz a in {1,2,3}r7 r r8 za not in {"a","b","c"}r: r< r; za in {None, 1, None}za not in {(1, 2), 3, 4}r@ za in {1, 2, 3, 3, 2, 1}r r Z BUILD_SETr% c S s
| d k S)Nr7 r r8 > r7 r r8 r )r: r r r r" � s z:TestTranforms.test_folding_of_sets_of_constants.<locals>.fc S s
| d k S)Nr7 r r8 > r7 r r8 r )r: r r r r# � s z:TestTranforms.test_folding_of_sets_of_constants.<locals>.g)r7 r )� frozensetr r r
Z
assertTrue)r r r' r r" r# r r r �!test_folding_of_sets_of_constants{ s ( z/TestTranforms.test_folding_of_sets_of_constantsc C s� xl d3 D]d \ } } t | d d � } | j | d | � x0 t j | � D] } | j | j j d � � qH Wq Wt d d d � } | j | d d � | j | d d! � t d" d d � } | j | d d# � d S)4N� a = 2+3+4rI �"@"*4�@@@@�a="abc" + "def"�abcdef�a = 3**4�Q �a = 3*4� � a = 13//4r8 �a = 14%4r �a = 2+3rE �a = 13-4�a = (12,13)[1]�
� a = 13 << 2�4 � a = 13 >> 2�
a = 13 & 7�
a = 13 ^ 7�
�
a = 13 | 7� r r r% �BINARY_za=2+"b"r; z
a="x"*1000i� �rn rI �ro rp �rq rr �rs rt �ru rv �rw r8 �rx r �ry rE �rz rI �r{ r| �r} r~ �r r8 �r� rE �r� r� �r� r� )r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� )r r
rY rZ � assertFalserA �
startswith)r r r' r rC r r r �#test_folding_of_binops_on_constants� s2 !z1TestTranforms.test_folding_of_binops_on_constantsc C s� t d d d � } | j | d d � | j | d � t d d d � } | j | d d � | j | d � t d d d � } | j | d d
� | j | d � t d d d � } | j | d � d S) Nz"foo"[0]r r r% r" Z
BINARY_SUBSCRu "a"[1]u u "𒍅"[0]u 𒍅z "fuu"[10])r r
r )r r r r r � test_binary_subscr_on_unicode� s z+TestTranforms.test_binary_subscr_on_unicodec C s xl d D]d \ } } t | d d � } | j | d
| � x0 t j | � D] } | j | j j d � � qH Wq Wd d � } x0 t j | � D] } | j | j j d � � q� WxL d" D]D \ } } } t | d d � } | j | d
| � | j | | � q� Wd S)#N�-0.5g �?�-0.0g �
-(1.0-1.0)�-0r �~-2r7 �+1r r r% �UNARY_c S s d S)Ng �?g g �r r r r r �negzero� s zDTestTranforms.test_folding_of_unaryops_on_constants.<locals>.negzero�-"abc"�abc�UNARY_NEGATIVE�~"abc"� UNARY_INVERT� r� r� � ��r� r� � ��r� r� �r� r �r� r7 �r� r7 )r� r� r� r� r� r� �r� r� r� �r� r� r� )r� r� )r r
rY rZ r� rA r� )r r r' r rC r� rA r r r �%test_folding_of_unaryops_on_constants� s( ! z3TestTranforms.test_folding_of_unaryops_on_constantsc C sU d d � } | j | d d � d d � t j | � D� } | j t | � d � d S)Nc S s | S)Nr )r r r r r" � s z/TestTranforms.test_elim_extra_return.<locals>.fr% c S s% g | ] } | j d k r | � q S)� RETURN_VALUE)rA )rB rC r r r rD � s z8TestTranforms.test_elim_extra_return.<locals>.<listcomp>r7 )r rY rZ r[ r\ )r r" �returnsr r r �test_elim_extra_return� s z$TestTranforms.test_elim_extra_returnc C sb d d � } | j | d � | j | d � d d � t j | � D� } | j t | � d � d S)Nc S s | r
| S| S)Nr )ZcondZ
true_valueZ false_valuer r r r" � s z1TestTranforms.test_elim_jump_to_return.<locals>.f� JUMP_FORWARDr* c S s% g | ] } | j d k r | � q S)r� )rA )rB rC r r r rD � s z:TestTranforms.test_elim_jump_to_return.<locals>.<listcomp>r )r rY rZ r[ r\ )r r" r� r r r �test_elim_jump_to_return� s
z&TestTranforms.test_elim_jump_to_returnc C sb d d � } | j | d � | j | d � d d � t j | � D� } | j t | � d � d S)Nc S s2 | r
d S| r d Sx d SWx | r) d Sd SWd S)Nr7 r r8 r@ rE rF r )�cond1�cond2r r r r" � s z5TestTranforms.test_elim_jump_after_return1.<locals>.fr� r* c S s% g | ] } | j d k r | � q S)r� )rA )rB rC r r r rD s z>TestTranforms.test_elim_jump_after_return1.<locals>.<listcomp>rF )r rY rZ r[ r\ )r r" r� r r r � test_elim_jump_after_return1� s
z*TestTranforms.test_elim_jump_after_return1c C s� d d � } | j | d � d d � t j | � D� } | j t | � d � d d � t j | � D� } | j t | � d � d S) Nc S s x | r d Sq Wd S)Nr@ r )r� r� r r r r"
s z5TestTranforms.test_elim_jump_after_return2.<locals>.fr� c S s% g | ] } | j d k r | � q S)r* )rA )rB rC r r r rD s z>TestTranforms.test_elim_jump_after_return2.<locals>.<listcomp>r7 c S s% g | ] } | j d k r | � q S)r� )rA )rB rC r r r rD s r )r rY rZ r[ r\ )r r" r� r r r � test_elim_jump_after_return2 s z*TestTranforms.test_elim_jump_after_return2c C s d d � } | j | d � d S)Nc S s d d d d � � } | S)Nr7 )�returnc S s d S)Nr r r r r r# s zBTestTranforms.test_make_function_doesnt_bail.<locals>.f.<locals>.gr r )r# r r r r" s z7TestTranforms.test_make_function_doesnt_bail.<locals>.fZ
BINARY_ADD)r )r r" r r r � test_make_function_doesnt_bail s z,TestTranforms.test_make_function_doesnt_bailc C s� d d d d d d d d d g } x� | D]} } t | d
d � } xb t j | � D]Q } | j | j j d � � | j | j j d
� � | j | j j d � � qP Wq( Wd S)Nz3 * -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}r r r� r� ZBUILD_)r rY rZ r� rA r� )r Zexprs�er rC r r r �test_constant_folding s
z#TestTranforms.test_constant_foldingN)�__name__�
__module__� __qualname__r r! r( r+ r5 r] rk rm r� r� r� r� r� r� r� r� r� r r r r r
s"
%
r c @ s e Z d Z d d � Z d S)� TestBugletsc C s. d d � } | j t � � | � Wd QRXd S)Nc S s d d h \ } } | | f S)Nr7 r )r �yr r r r" ; s z%TestBuglets.test_bug_11510.<locals>.f)Z assertRaises�
ValueError)r r" r r r �test_bug_115107 s zTestBuglets.test_bug_11510N)r� r� r� r� r r r r r� 5 s r� �__main__)rY �re�sys�ior Zunittest�mathr Ztest.bytecode_helperr r ZTestCaser� r� �mainr r r r �<module> s � ,