File: //usr/local/lib/python3.10/test/__pycache__/test_peepholer.cpython-310.opt-2.pyc
o
�iO � @ s^ d dl Z d dlZd dlmZ dd� ZG dd� de�ZG dd� dej�Zed kr-e� � dS dS )
� N)�BytecodeTestCasec C s^ d}t �| �D ] }|j|kr|d7 }qt| d�r| j} | jD ]}t|d�r,|t||�7 }q |S )Nr � �__code__�co_code)�dis�get_instructions�opname�hasattrr � co_consts�count_instr_recursively)�fr �count�instr�c� r �0/usr/local/lib/python3.10/test/test_peepholer.pyr s
�
�r c @ 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%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7S )8�
TestTranformsc
C s� t t�|��}dd� |D �}|D ]b}d|jvrq||j }|jdv r6| �|j� d|j� d|j� d|j� �� |jdv rT|jdkrT| �|j� d|j� d|j� d|j� �� d|jv rrd |jv rr| �|j� d|j� d|j� d|j� �� qd S )
Nc S s i | ]}|j |�qS r )�offset��.0r r r r �
<dictcomp> s z4TestTranforms.check_jump_targets.<locals>.<dictcomp>ZJUMP_)�
JUMP_ABSOLUTE� JUMP_FORWARDz at z
jumps to � RETURN_VALUEZ_OR_POPZJUMP_IF_)�listr r r ZargvalZfailr )�self�codeZ instructions�targetsr Ztgtr r r �check_jump_targets s6
��
������z TestTranforms.check_jump_targetsc C s` t �|�}tt �|��}tdd� |D ��}tdd� |D ��}| �|d� | �|t|j �� d S )Nc s � � | ]}|d V qdS �r Nr �r �tr r r � <genexpr>1 � � z-TestTranforms.check_lnotab.<locals>.<genexpr>c s r r r r! r r r r# 2 r$ r )
r Z_get_code_objectr Zfindlinestarts�min�maxZassertGreaterEqualZ
assertLess�lenr )r r ZlnotabZ min_bytecodeZ max_bytecoder r r � check_lnotab+ s
zTestTranforms.check_lnotabc C s: dd� }| � |d� | � |d� | �|d� | �|� d S )Nc S s | dks~ d S d S )N� r ��xr r r �unot; s �z%TestTranforms.test_unot.<locals>.unotZ UNARY_NOT�POP_JUMP_IF_FALSE�POP_JUMP_IF_TRUE��assertNotInBytecode�assertInBytecoder( )r r, r r r � test_unot9 s
zTestTranforms.test_unotc C s8 dD ]\}}}t |dd�}| �|||� | �|� qd S )N))z
not a is b�IS_OPr )znot a is not br3 r )z
not a in b� CONTAINS_OPr )znot a not in br4 r � �single)�compiler1 r( )r �lineZcmp_op�invertr r r r � test_elim_inversion_of_is_or_inC s
�z-TestTranforms.test_elim_inversion_of_is_or_inc C s� dd� }dd� }dd� }|d f|df|dffD ]\}}| � |d � | �|d
|� | �|� qd d� }| � |d � | �|d
d � | �|� d S ) Nc S s d } d } | S �Nr r* r r r r P s z0TestTranforms.test_global_as_constant.<locals>.fc S � d} | S )NTr r* r r r �gT � z0TestTranforms.test_global_as_constant.<locals>.gc S r< )NFr r* r r r �hW r> z0TestTranforms.test_global_as_constant.<locals>.hTFZ LOAD_GLOBAL�
LOAD_CONSTc S s d S r; r r r r r r ` � r/ )r r r= r? �func�elemr r r �test_global_as_constantN s z%TestTranforms.test_global_as_constantc C sB dd� }dD ]}| � ||� qdD ]}| �||� q| �|� d S )Nc S s qr; )r r r r r r j s �z'TestTranforms.test_while_one.<locals>.f)r@ r- )r r/ )r r rC r r r �test_while_oneh s z TestTranforms.test_while_onec C sL dD ]!\}}t |dd�}| �||� | �|d� | �|d� | �|� qd S )N))za, = a,r@ )z a, b = a, bZROT_TWO)za, b, c = a, b, cZ ROT_THREEr5 r6 � BUILD_TUPLEZ UNPACK_TUPLE�r7 r1 r0 r( �r r8 rC r r r r �test_pack_unpackt s �z TestTranforms.test_pack_unpackc C s� dD ] \}}t |dd�}| �|d|� | �|d� | �|� qt tttd���dd�}| �|d� dd� t�|�D �}| � t
|�d � | �|� d
d � }| �|� d S ) N))z a = 1,2,3�r r) � )z
("a","b","c")��a�br )z
a,b,c = 1,2,3rJ )z(None, 1, None)�Nr N)z((1, 2), 3, 4)��r r) rK � r5 r6 r@ rF �' c S � g | ] }|j d kr|�qS )r@ �r r r r r �
<listcomp>� �
�zETestTranforms.test_folding_of_tuples_of_constants.<locals>.<listcomp>r) c S s g d� f d S )N)dr r r) rK rR � � � � � r r r) rK rR rX rY rZ r[ r\ r r r) rK rR rX rY rZ r[ r\ r r r) rK rR rX rY rZ r[ r\ r r r) rK rR rX rY rZ r[ r\ r r r) rK rR rX rY rZ r[ r\ r r r) rK rR rX rY rZ r[ r\ r r r) rK rR rX rY rZ r[ r\ r r r) rK rR rX rY rZ r[ r\ r r r) rK rR rX rY rZ r[ r\ r r r r r �crater� � zATestTranforms.test_folding_of_tuples_of_constants.<locals>.crater) r7 r1 r0 r( �repr�tuple�ranger r � assertEqualr' )r r8 rC r Z load_constsr] r r r �#test_folding_of_tuples_of_constants� s
z1TestTranforms.test_folding_of_tuples_of_constantsc C sB dD ] \}}t |dd�}| �|d|� | �|d� | �|� qd S )N))z a in [1,2,3]rJ )za not in ["a","b","c"]rL )za in [None, 1, None]rO )za not in [(1, 2), 3, 4]rP r5 r6 r@ �
BUILD_LISTrG rH r r r �"test_folding_of_lists_of_constants� s �z0TestTranforms.test_folding_of_lists_of_constantsc C s� dt h d��fdt h d��fdt dd h�fdt h d��fd t h d��ffD ] \}}t|d
d �}| �|d � | �|d
|� | �|� q%dd� }dd� }| �|d�� | �|d� � | �|� | �|d� � | �|d�� | �|� d S )Nz a in {1,2,3}� r r) rK za not in {"a","b","c"}> rM r rN za in {None, 1, None}r za not in {(1, 2), 3, 4}> rQ rK rR za in {1, 2, 3, 3, 2, 1}r5 r6 Z BUILD_SETr@ c S s | dv S �Nrf r �rM r r r r � � z:TestTranforms.test_folding_of_sets_of_constants.<locals>.fc S s | dvS rg r rh r r r r= � ri z:TestTranforms.test_folding_of_sets_of_constants.<locals>.grK rR )� frozensetr7 r0 r1 r( �
assertTrue)r r8 rC r r r= r r r �!test_folding_of_sets_of_constants� s$ �
z/TestTranforms.test_folding_of_sets_of_constantsc C s, dD ]'\}}t |dd�}| �|d|� t�|�D ] }| �|j�d�� q| �|� qt 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+4r\ )z"@"*4z@@@@)za="abc" + "def"Zabcdef)za = 3**4�Q )za = 3*4� )z a = 13//4rK )za = 14%4r) )za = 2+3rX )za = 13-4r\ )za = (12,13)[1]�
)z a = 13 << 2�4 )z a = 13 >> 2rK )z
a = 13 & 7rX )z
a = 13 ^ 7�
)z
a = 13 | 7� r5 r6 r@ �BINARY_za=2+"b"r) rN z a="x"*10000rS r+ z a=1<<1000i� r z a=2**1000)
r7 r1 r r � assertFalser �
startswithr( Z assertNotInr
)r r8 rC 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]r5 r6 r@ r Z
BINARY_SUBSCRu "a"[1]u u "𒍅"[0]u 𒍅z "fuu"[10]rG )r r r r r � test_binary_subscr_on_unicode� s
z+TestTranforms.test_binary_subscr_on_unicodec C s� dD ]'\}}t |dd�}| �|d|� t�|�D ] }| �|j�d�� q| �|� qdd� }t�|�D ] }| �|j�d�� q3| �|� dD ] \}}}t |dd�}| �|d|� | �||� | �|� qFd S ) N))z-0.5g �)z-0.0� �)z
-(1.0-1.0)rx )z-0r )z~-2r )z+1r r5 r6 r@ �UNARY_c S s dS )Nrx r r r r r �negzero& � zDTestTranforms.test_folding_of_unaryops_on_constants.<locals>.negzero))z-"abc"�abcZUNARY_NEGATIVE)z~"abc"r| Z UNARY_INVERT)r7 r1 r r rt r ru r( )r r8 rC r r rz r r r r �%test_folding_of_unaryops_on_constants s
�z3TestTranforms.test_folding_of_unaryops_on_constantsc C sH dd� }| � |dd � dd� t�|�D �}| �t|�d� | �|� d S )Nc S s | S r; r r* r r r r 9 r{ z/TestTranforms.test_elim_extra_return.<locals>.fr@ c S rT �r rU r r r r rV < rW z8TestTranforms.test_elim_extra_return.<locals>.<listcomp>r )r0 r r rb r' r( �r r �returnsr r r �test_elim_extra_return7 s
z$TestTranforms.test_elim_extra_returnc C s\ dd� }| � |� | �|d� | �|d� dd� t�|�D �}| �t|�d� | �|� d S )Nc S s | r|S |S r; r )�condZ
true_valueZ false_valuer r r r C s �z1TestTranforms.test_elim_jump_to_return.<locals>.fr r c S rT r~ rU r r r r rV J rW z:TestTranforms.test_elim_jump_to_return.<locals>.<listcomp>r) )r r0 r r rb r' r( r r r r �test_elim_jump_to_returnA s
z&TestTranforms.test_elim_jump_to_returnc C � dd� }| � |� | �|� d S )Nc S s$ t r
tstr t� d S d S t� d S r; )rM r �d�fooZbazr r r r r Q s �
�
z6TestTranforms.test_elim_jump_to_uncond_jump.<locals>.f�r r( �r r r r r � test_elim_jump_to_uncond_jumpO s
z+TestTranforms.test_elim_jump_to_uncond_jumpc C r� )Nc S s | r
t str t� } | sd S d S r; )r r� r� rh r r r r ^ s � �z7TestTranforms.test_elim_jump_to_uncond_jump2.<locals>.fr� r� r r r � test_elim_jump_to_uncond_jump2\ s
z,TestTranforms.test_elim_jump_to_uncond_jump2c C s� dd� }| � |� | �|� | �t|d�d� dd� }| � |� | �|� | �t|d�d� dd� }| � |� | �|� | �|d� | �|d� | �|d� d d� }| � |� | �|� | �|d� | �|d� | �|d
� d S ) Nc S s | o|o|S r; r rL r r r r j � �z7TestTranforms.test_elim_jump_to_uncond_jump3.<locals>.fZJUMP_IF_FALSE_OR_POPr) c S s | p|p|S r; r rL r r r r q r� ZJUMP_IF_TRUE_OR_POPc S s | r|p|S r; r rL r r r r x r� r- c S s | s|o|S r; r rL r r r r � r� r. )r r( rb r r0 r1 r� r r r � test_elim_jump_to_uncond_jump3g s(
z,TestTranforms.test_elim_jump_to_uncond_jump3c C sR dd� }| � |d� | � |d� dd� t�|�D �}| �t|�d� | �|� d S )Nc S s | rdS |rdS dS )Nr r) rK r )Zcond1Zcond2r r r r � s z5TestTranforms.test_elim_jump_after_return1.<locals>.fr r c S rT r~ rU r r r r rV � rW z>TestTranforms.test_elim_jump_after_return1.<locals>.<listcomp>rY )r0 r r ZassertLessEqualr' r( r r r r � test_elim_jump_after_return1� s z*TestTranforms.test_elim_jump_after_return1c C s" dd� }| � |d� | �|� d S )Nc S s ddd�} | S )N�returnr) c S s d S r; r r r r r r= � r{ zBTestTranforms.test_make_function_doesnt_bail.<locals>.f.<locals>.g)r� r) r )r= r r r r � s
z7TestTranforms.test_make_function_doesnt_bail.<locals>.fZ
BINARY_ADD)r0 r( r� r r r � test_make_function_doesnt_bail� s z,TestTranforms.test_make_function_doesnt_bailc C sr g d�}|D ]0}t |dd�}t�|�D ] }| �|j�d�� | �|j�d�� | �|j�d�� q| �|� qd S )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}r5 r6 ry rs ZBUILD_)r7 r r rt r ru r( )r Zexprs�er r r r r �test_constant_folding� s �z#TestTranforms.test_constant_foldingc C �( dd� }| � t|d�d� | �|� d S )Nc S s t ttfv S r; )r+ rM rN r r r r � containtest� s z7TestTranforms.test_in_literal_list.<locals>.containtestrd r �rb r r( )r r� r r r �test_in_literal_list� s z"TestTranforms.test_in_literal_listc C r� )Nc S s t tfD ]} qd S r; )rM rN r* r r r �forloop� s �z8TestTranforms.test_iterate_literal_list.<locals>.forlooprd r r� )r r� r r r �test_iterate_literal_list� s z'TestTranforms.test_iterate_literal_listc C s$ dd� }| � |� d� | �|� d S )Nc S s dS )NTFr r r r r r r � rA z=TestTranforms.test_condition_with_binop_with_bools.<locals>.fr )rb r( r� r r r �$test_condition_with_binop_with_bools� s z2TestTranforms.test_condition_with_binop_with_boolsc C s$ dd� }| � |d�� | �|� d S )Nc S s | rdS dS )NTFr r* r r r r � s �z3TestTranforms.test_if_with_if_expression.<locals>.fT)rk r( r� r r r �test_if_with_if_expression� s z(TestTranforms.test_if_with_if_expressionc C s dd� }| � |� d S )Nc S s dS )Nr rK r r* r r r r � rA z+TestTranforms.test_trailing_nops.<locals>.f)r( r� r r r �test_trailing_nops� s z TestTranforms.test_trailing_nopsc C sl dd� }| � t|d�d� dd� }| � t|d�d� dd� }| � t|d�d� d d
� }| � t|d�d� d S ) Nc S � dd� t D �S )Nc S s g | ]}t |�}|�qS r �r �r r+ �yr r r rV � � z[TestTranforms.test_assignment_idiom_in_comprehensions.<locals>.listcomp.<locals>.<listcomp>rh r r r r �listcomp� � zGTestTranforms.test_assignment_idiom_in_comprehensions.<locals>.listcompZFOR_ITERr c S r� )Nc S s h | ]}t |�}|�qS r r� r� r r r � <setcomp>� r� zYTestTranforms.test_assignment_idiom_in_comprehensions.<locals>.setcomp.<locals>.<setcomp>rh r r r r �setcomp� r� zFTestTranforms.test_assignment_idiom_in_comprehensions.<locals>.setcompc S r� )Nc S s i | ] }t |�}||�qS r r� r� r r r r � s z[TestTranforms.test_assignment_idiom_in_comprehensions.<locals>.dictcomp.<locals>.<dictcomp>rh r r r r �dictcomp� r� zGTestTranforms.test_assignment_idiom_in_comprehensions.<locals>.dictcompc S r� )Nc s s � | ] }t |�}|V qd S r; r� r� r r r r# � s � zYTestTranforms.test_assignment_idiom_in_comprehensions.<locals>.genexpr.<locals>.<genexpr>rh r r r r �genexpr� r� zFTestTranforms.test_assignment_idiom_in_comprehensions.<locals>.genexpr)rb r )r r� r� r� r� r r r �'test_assignment_idiom_in_comprehensions� s z5TestTranforms.test_assignment_idiom_in_comprehensionsN) �__name__�
__module__� __qualname__r r( r2 r: rD rE rI rc re rl rv rw r} r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r s8
(
,!
# r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� TestBugletsc C s> dd� }| � t�� |� W d � d S 1 sw Y d S )Nc S s ddh\} }| |fS )Nr r )r+ r� r r r r � s z%TestBuglets.test_bug_11510.<locals>.f)Z assertRaises�
ValueErrorr� r r r �test_bug_11510� s "�zTestBuglets.test_bug_11510c C s, t d�D ]}zt� tp
ty Y qw d S )Nrq )ra � Exception)r �ir r r �test_bpo_42057� s ��zTestBuglets.test_bpo_42057c C � t ddd� d S )Nzwhile True or spam: pass�<test>�exec�r7 �r r r r � test_bpo_45773_pop_jump_if_true r^ z+TestBuglets.test_bpo_45773_pop_jump_if_truec C r� )Nz while True or not spam: passr� r� r� r� r r r � test_bpo_45773_pop_jump_if_false r^ z,TestBuglets.test_bpo_45773_pop_jump_if_falseN)r� r� r� r� r� r� r� r r r r r� � s
r� �__main__)
r ZunittestZ test.support.bytecode_helperr r r ZTestCaser� r� �mainr r r r �<module> s
a �