File: //usr/local/lib/python3.7/test/__pycache__/test_peepholer.cpython-37.opt-1.pyc
B
��g<3 � @ sR d dl Z d dlZd dlmZ G dd� de�ZG dd� dej�ZedkrNe�� dS )� N)�BytecodeTestCasec @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%S )&�
TestTranformsc C s0 dd� }| � |d� | � |d� | �|d� d S )Nc S s | dks
~ d S )N� � )�xr r �//usr/local/lib/python3.7/test/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 s0 x*dD ]"\}}t |dd�}| �|d|� qW d S )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�coder r r � test_elim_inversion_of_is_or_in s z-TestTranforms.test_elim_inversion_of_is_or_inc C s| dd� }dd� }dd� }x<|d f|df|dffD ]"\}}| � |d � | �|d
|� q0W d d� }| � |d � | �|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 dS )z1Adding a docstring made this test fail in Py2.5.0Nr 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 s@ dd� }xdD ]}| � ||� qW xdD ]}| �||� q(W d S )Nc S s
xqW t S )N)�listr r r r r 5 s z'TestTranforms.test_while_one.<locals>.f)r r )�
JUMP_ABSOLUTE)r
r )r r r r r r �test_while_one3 s
z TestTranforms.test_while_onec C sF x@dD ]8\}}t |dd�}| �||� | �|d� | �|d� qW d S )N))za, = a,r )z a, b = a, bZROT_TWO)za, b, c = a, b, cZ ROT_THREEr r � BUILD_TUPLEZ UNPACK_TUPLE)r r r
)r r r r r r r �test_pack_unpack> s z TestTranforms.test_pack_unpackc C s� x6dD ].\}}t |dd�}| �|d|� | �|d� qW t tttd���dd�}| �|d� dd� t�|�D �}| �t |�d � d
d � }d S ) N))z a = 1,2,3)� r � )z
("a","b","c"))�a�b�c)z
a,b,c = 1,2,3)r" r r# )z(None, 1, None))Nr" N)z((1, 2), 3, 4)))r" r r# � r r r r i' c S s g | ]}|j d kr|�qS )r )�opname)�.0�instrr r r �
<listcomp>Y s zETestTranforms.test_folding_of_tuples_of_constants.<locals>.<listcomp>r c d S s� ddddddddd d
ddddddddd d
ddddddddd d
ddddddddd d
ddddddddd d
ddddddddd d
ddddddddd d
ddddddddd d
ddddddddd d
ddddddddd d
gd f d S ) Nr r" r r# r' � � � � � r r r r r �crater` s zATestTranforms.test_folding_of_tuples_of_constants.<locals>.crater)
r r r
�repr�tuple�range�dis�get_instructions� assertEqual�len)r r r r Z load_constsr1 r r r �#test_folding_of_tuples_of_constantsI s z1TestTranforms.test_folding_of_tuples_of_constantsc C s< x6dD ].\}}t |dd�}| �|d|� | �|d� qW d S )N))z a in [1,2,3])r" r r# )za not in ["a","b","c"])r$ r% r& )za in [None, 1, None])Nr" N)za not in [(1, 2), 3, 4]))r" r r# r' r r r Z
BUILD_LIST)r r r
)r r r r r r r �"test_folding_of_lists_of_constantsn s
z0TestTranforms.test_folding_of_lists_of_constantsc C s� x�dt dddh�fdt dddh�fd t dd h�fd
t d dd h�fd
t dddh�ffD ].\}}t|dd�}| �|d� | �|d|� qTW dd� }dd� }| �|d�� | �|d � � | �|d� � | �|d �� d S )Nz a in {1,2,3}r" r r# za not in {"a","b","c"}r$ r& r% za in {None, 1, None}za not in {(1, 2), 3, 4})r" r r' za in {1, 2, 3, 3, 2, 1}r r Z BUILD_SETr c S s | dkS )N> r" r r# r )r$ r r r r � s z:TestTranforms.test_folding_of_sets_of_constants.<locals>.fc S s | dkS )N> r" r r# r )r$ r r r r � s z:TestTranforms.test_folding_of_sets_of_constants.<locals>.g)� frozensetr r
r Z
assertTrue)r r r r r r r r r �!test_folding_of_sets_of_constantsz s z/TestTranforms.test_folding_of_sets_of_constantsc C s xPdD ]H\}}t |dd�}| �|d|� x$t�|�D ]}| �|j�d�� q4W qW t ddd�}| �|dd� | �|dd� t d dd�}| �|dd
� | �d d
|j� t d dd�}| �|dd
� | �dd
> |j� t ddd�}| �|dd
� | �dd
|j� d S )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]�
)z a = 13 << 2�4 )z a = 13 >> 2r# )z
a = 13 & 7r, )z
a = 13 ^ 7�
)z
a = 13 | 7� r r r �BINARY_za=2+"b"r r% z a="x"*10000i' r z a=1<<1000i� r" z a=2**1000) r r r5 r6 � assertFalser( �
startswithZ assertNotIn� co_consts)r r r r r* r r r �#test_folding_of_binops_on_constants� s$ z1TestTranforms.test_folding_of_binops_on_constantsc C s� t ddd�}| �|dd� | �|d� t ddd�}| �|dd� | �|d� t d dd�}| �|dd
� | �|d� t d dd�}| �|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� xPdD ]H\}}t |dd�}| �|d|� x$t�|�D ]}| �|j�d�� q4W qW dd� }x$t�|�D ]}| �|j�d�� qfW x8dD ]0\}}}t |dd�}| �|d|� | �||� q�W d S ) N))z-0.5g �)z-0.0g �)z
-(1.0-1.0)g �)z-0r )z~-2r" )z+1r" r r r �UNARY_c S s dS )Ng �r r r r r �negzero� s zDTestTranforms.test_folding_of_unaryops_on_constants.<locals>.negzero))z-"abc"�abcZUNARY_NEGATIVE)z~"abc"rK Z UNARY_INVERT)r r r5 r6 rD r( rE )r r r r r* rJ r( r r r �%test_folding_of_unaryops_on_constants� s z3TestTranforms.test_folding_of_unaryops_on_constantsc C s> dd� }| � |dd � dd� t�|�D �}| �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 kr|�qS )� RETURN_VALUE)r( )r) r* r r r r+ � s z8TestTranforms.test_elim_extra_return.<locals>.<listcomp>r" )r
r5 r6 r7 r8 )r r �returnsr r r �test_elim_extra_return� s z$TestTranforms.test_elim_extra_returnc C sH dd� }| � |d� | � |d� dd� t�|�D �}| �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 kr|�qS )rM )r( )r) r* r r r r+ � s z:TestTranforms.test_elim_jump_to_return.<locals>.<listcomp>r )r
r5 r6 r7 r8 )r r rN r r r �test_elim_jump_to_return� s
z&TestTranforms.test_elim_jump_to_returnc C sH dd� }| � |d� | � |d� dd� t�|�D �}| �t|�d� d S )Nc S s( | rdS |rdS xdS x | r dS dS dS )Nr" r r# r' r, r- r )�cond1�cond2r r r r s z5TestTranforms.test_elim_jump_after_return1.<locals>.frP r c S s g | ]}|j d kr|�qS )rM )r( )r) r* r r r r+
s z>TestTranforms.test_elim_jump_after_return1.<locals>.<listcomp>r- )r
r5 r6 r7 r8 )r r rN r r r � test_elim_jump_after_return1 s
z*TestTranforms.test_elim_jump_after_return1c C s` dd� }| � |d� dd� t�|�D �}| �t|�d� dd� t�|�D �}| �t|�d� d S ) Nc S s x | rdS qW d S )Nr' r )rR rS r r r r s z5TestTranforms.test_elim_jump_after_return2.<locals>.frP c S s g | ]}|j d kr|�qS )r )r( )r) r* r r r r+ s z>TestTranforms.test_elim_jump_after_return2.<locals>.<listcomp>r" c S s g | ]}|j d kr|�qS )rM )r( )r) r* r r r r+ s r )r
r5 r6 r7 r8 )r r rN r r r � test_elim_jump_after_return2 s z*TestTranforms.test_elim_jump_after_return2c C s dd� }| � |d� d S )Nc S s dd�dd�} | S )Nr )�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 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~ ddddddddd g }xb|D ]Z}t |d
d �}xHt�|�D ]:}| �|j�d �� | �|j�d
�� | �|j�d�� q8W q W d 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 rI rC ZBUILD_)r r5 r6 rD r( rE )r Zexprs�er r* r r r �test_constant_folding& s
z#TestTranforms.test_constant_foldingc C s dd� }| � |� d� d S )Nc S s dsdr dS dS )NTFr" r r r r r r r ; s z=TestTranforms.test_condition_with_binop_with_bools.<locals>.fr" )r7 )r r r r r �$test_condition_with_binop_with_bools: s z2TestTranforms.test_condition_with_binop_with_boolsN)�__name__�
__module__� __qualname__r
r r r r! r9 r: r< rG rH rL rO rQ rT rU rW rY rZ r r r r r s$
% '
r c @ s e Zd Zdd� ZdS )� TestBugletsc C s( dd� }| � t�� |� W d Q R X d S )Nc S s ddh\} }| |fS )Nr" r )r �yr r r r H s z%TestBuglets.test_bug_11510.<locals>.f)Z assertRaises�
ValueError)r r r r r �test_bug_11510D s zTestBuglets.test_bug_11510N)r[ r\ r] ra r r r r r^ B s r^ �__main__) r5 ZunittestZtest.bytecode_helperr r ZTestCaser^ r[ �mainr r r r �<module> s >