Sindbad~EG File Manager
a
��g�P � @ sZ d dl Z d dlZd dlmZ dd� ZG dd� de�ZG dd� dej�Zed krVe� � dS )
� N)�BytecodeTestCasec C s^ d}t �| �D ]}|j|kr|d7 }qt| d�r6| 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 �//usr/local/lib/python3.9/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 d7d8� Z d9S ):�
TestTranformsc
C s� t t�|��}dd� |D �}|D ]�}d|jvr0q ||j }|jdv rl| �|j� d|j� d|j� d|j� �� |jdv r�|jdkr�| �|j� d|j� d|j� d|j� �� d|jv r d |jv r | �|j� d|j� d|j� d|j� �� q d S )
Nc S s i | ] }|j |�qS r )�offset��.0r r r r �
<dictcomp> � 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 s4
��
�����z TestTranforms.check_jump_targetsc C s^ t �|�}tt �|��}tdd� |D ��}tdd� |D ��}| �|d� | �|t|j �� d S )Nc s s | ]}|d V qdS �r Nr �r �tr r r � <genexpr>1 r z-TestTranforms.check_lnotab.<locals>.<genexpr>c s s | ]}|d V qdS 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 )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 br2 r )z
not a in b� CONTAINS_OPr )znot a not in br3 r � �single)�compiler0 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
|� | �|� q.d 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 s d} | S )NTr r) r r r �gT s z0TestTranforms.test_global_as_constant.<locals>.gc S s d} | S )NFr r) r r r �hW s z0TestTranforms.test_global_as_constant.<locals>.hTFZ LOAD_GLOBAL�
LOAD_CONSTc S s d S r: r r r r r r ` s r. )r r r; r< �func�elemr r r �test_global_as_constantN s z%TestTranforms.test_global_as_constantc C sB dd� }dD ]}| � ||� q dD ]}| �||� q"| �|� d S )Nc S s q t S r: )r r r r r r j s z'TestTranforms.test_while_one.<locals>.f)r= r, �r r. )r r r? r r r �test_while_oneh s z TestTranforms.test_while_onec C sL dD ]B\}}t |dd�}| �||� | �|d� | �|d� | �|� qd S )N))za, = a,r= )z a, b = a, bZROT_TWO)za, b, c = a, b, cZ ROT_THREEr4 r5 � BUILD_TUPLEZ UNPACK_TUPLE�r6 r0 r/ r' �r r7 r? r r r r �test_pack_unpackt s z TestTranforms.test_pack_unpackc C s� dD ]8\}}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,3rG )z(None, 1, None)�Nr N)z((1, 2), 3, 4)��r r( rH � r4 r5 r= rC �' c S s g | ]}|j d kr|�qS )r= �r r r r r �
<listcomp>� s
�zETestTranforms.test_folding_of_tuples_of_constants.<locals>.<listcomp>r( c S s g d� f d S )N)dr r r( rH rO � � � � � r r r( rH rO rS rT rU rV rW r r r( rH rO rS rT rU rV rW r r r( rH rO rS rT rU rV rW r r r( rH rO rS rT rU rV rW r r r( rH rO rS rT rU rV rW r r r( rH rO rS rT rU rV rW r r r( rH rO rS rT rU rV rW r r r( rH rO rS rT rU rV rW r r r( rH rO rS rT rU rV rW r r r r r �crater� s zATestTranforms.test_folding_of_tuples_of_constants.<locals>.crater) r6 r0 r/ r' �repr�tuple�ranger r � assertEqualr& )r r7 r? r Z load_constsrX r r r �#test_folding_of_tuples_of_constants� s
z1TestTranforms.test_folding_of_tuples_of_constantsc C sB dD ]8\}}t |dd�}| �|d|� | �|d� | �|� qd S )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]rM r4 r5 r= �
BUILD_LISTrD rE 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 ]8\}}t|d
d �}| �|d � | �|d
|� | �|� qJdd� }dd� }| �|d�� | �|d� � | �|� | �|d� � | �|d�� | �|� d S )Nz a in {1,2,3}� r r( rH za not in {"a","b","c"}> rJ r rK za in {None, 1, None}r za not in {(1, 2), 3, 4}> rN rH rO za in {1, 2, 3, 3, 2, 1}r4 r5 Z BUILD_SETr= c S s | dv S �Nr` r �rJ r r r r � s z:TestTranforms.test_folding_of_sets_of_constants.<locals>.fc S s | dvS ra r rb r r r r; � s z:TestTranforms.test_folding_of_sets_of_constants.<locals>.grH rO )� frozensetr6 r/ r0 r' �
assertTrue)r r7 r? r r r; r r r �!test_folding_of_sets_of_constants� s$ �
z/TestTranforms.test_folding_of_sets_of_constantsc C s, dD ]N\}}t |dd�}| �|d|� t�|�D ]}| �|j�d�� q0| �|� 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+4rW )z"@"*4z@@@@)za="abc" + "def"Zabcdef)za = 3**4�Q )za = 3*4� )z a = 13//4rH )za = 14%4r( )za = 2+3rS )za = 13-4rW )za = (12,13)[1]�
)z a = 13 << 2�4 )z a = 13 >> 2rH )z
a = 13 & 7rS )z
a = 13 ^ 7�
)z
a = 13 | 7� r4 r5 r= �BINARY_za=2+"b"r( rK z a="x"*10000rP r* z a=1<<1000i� r z a=2**1000)
r6 r0 r r � assertFalser �
startswithr' Z assertNotInr
)r r7 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]r4 r5 r= r Z
BINARY_SUBSCRu "a"[1]u u "𒍅"[0]u 𒍅z "fuu"[10]rD )r r r r r � test_binary_subscr_on_unicode� s
z+TestTranforms.test_binary_subscr_on_unicodec C s� dD ]N\}}t |dd�}| �|d|� t�|�D ]}| �|j�d�� q0| �|� qdd� }t�|�D ]}| �|j�d�� qf| �|� dD ]:\}}}t |dd�}| �|d|� | �||� | �|� q�d S ) N))z-0.5g �)z-0.0� �)z
-(1.0-1.0)rq )z-0r )z~-2r )z+1r r4 r5 r= �UNARY_c S s dS )Nrq r r r r r �negzero& s zDTestTranforms.test_folding_of_unaryops_on_constants.<locals>.negzero))z-"abc"�abcZUNARY_NEGATIVE)z~"abc"rt Z UNARY_INVERT)r6 r0 r r rm r rn r' )r r7 r? r r rs 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 s z/TestTranforms.test_elim_extra_return.<locals>.fr= c S s g | ]}|j d kr|�qS �r rQ r r r r rR < s
�z8TestTranforms.test_elim_extra_return.<locals>.<listcomp>r )r/ r r r\ 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 s g | ]}|j d kr|�qS rv rQ r r r r rR J s
�z:TestTranforms.test_elim_jump_to_return.<locals>.<listcomp>r( )r r/ r r r\ r& r' rw r r r �test_elim_jump_to_returnA s
z&TestTranforms.test_elim_jump_to_returnc C s dd� }| � |� | �|� d S )Nc S s t rts trt� nt� d S r: )rJ 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 s dd� }| � |� | �|� d S )Nc S s | rt s tr t� } q d S r: )r r| r} rb 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 rI r r r r j s �z7TestTranforms.test_elim_jump_to_uncond_jump3.<locals>.fZJUMP_IF_FALSE_OR_POPr( c S s | p
|p
|S r: r rI r r r r q s �ZJUMP_IF_TRUE_OR_POPc S s | r|p
|S r: r rI r r r r x s �r, c S s | s|o
|S r: r rI r r r r � s �r- )r r' r\ r r/ r0 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 | r dS dS )Nr r( rH rO rS rT r �Zcond1Zcond2r r r r � s z5TestTranforms.test_elim_jump_after_return1.<locals>.fr r c S s g | ]}|j d kr|�qS rv rQ r r r r rR � s
�z>TestTranforms.test_elim_jump_after_return1.<locals>.<listcomp>rT )r/ r r �assertLessEqualr& r' rw r r r � test_elim_jump_after_return1� s z*TestTranforms.test_elim_jump_after_return1c C sj dd� }| � |d� dd� t�|�D �}| �t|�d� dd� t�|�D �}| �t|�d� | �|� d S ) Nc S s | r dS q d S )NrO r r� r r r r � s z5TestTranforms.test_elim_jump_after_return2.<locals>.fr c S s g | ]}|j d kr|�qS rA rQ r r r r rR � s
�z>TestTranforms.test_elim_jump_after_return2.<locals>.<listcomp>r c S s g | ]}|j d kr|�qS rv rQ r r r r rR � s
�r( )r/ r r r\ r& r� r' rw 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 r: r 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 sr g d�}|D ]`}t |dd�}t�|�D ]:}| �|j�d�� | �|j�d�� | �|j�d�� q&| �|� q d 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}r4 r5 rr rl ZBUILD_)r6 r r rm r rn r' )r Zexprs�er r r r r �test_constant_folding� s z#TestTranforms.test_constant_foldingc C s( dd� }| � t|d�d� | �|� d S )Nc S s t ttfv S r: )r* rJ rK r r r r � containtest� s z7TestTranforms.test_in_literal_list.<locals>.containtestr^ r �r\ r r' )r r� r r r �test_in_literal_list� s z"TestTranforms.test_in_literal_listc C s( dd� }| � t|d�d� | �|� d S )Nc S s t tfD ]} qd S r: )rJ rK r) r r r �forloop� s z8TestTranforms.test_iterate_literal_list.<locals>.forloopr^ 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 dsdr dS dS )NTFr r r r r r r r � s z=TestTranforms.test_condition_with_binop_with_bools.<locals>.fr )r\ 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 | rndrdS dS )NTFr r) r r r r � s
z3TestTranforms.test_if_with_if_expression.<locals>.fT)rd 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 dS )NrH rS rT r r) r r r r � s 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 s dd� t D �S )Nc S s g | ]}t |�}|�qS r �r �r r* �yr r r rR � r z[TestTranforms.test_assignment_idiom_in_comprehensions.<locals>.listcomp.<locals>.<listcomp>rb r r r r �listcomp� s zGTestTranforms.test_assignment_idiom_in_comprehensions.<locals>.listcompZFOR_ITERr c S s dd� t D �S )Nc S s h | ]}t |�}|�qS r r� r� r r r � <setcomp>� r zYTestTranforms.test_assignment_idiom_in_comprehensions.<locals>.setcomp.<locals>.<setcomp>rb r r r r �setcomp� s zFTestTranforms.test_assignment_idiom_in_comprehensions.<locals>.setcompc S s dd� t D �S )Nc S s i | ]}t |�}||�qS r r� r� r r r r � r z[TestTranforms.test_assignment_idiom_in_comprehensions.<locals>.dictcomp.<locals>.<dictcomp>rb r r r r �dictcomp� s zGTestTranforms.test_assignment_idiom_in_comprehensions.<locals>.dictcompc S s dd� t D �S )Nc s s | ]}t |�}|V qd S r: r� r� r r r r# � r zYTestTranforms.test_assignment_idiom_in_comprehensions.<locals>.genexpr.<locals>.<genexpr>rb r r r r �genexpr� s zFTestTranforms.test_assignment_idiom_in_comprehensions.<locals>.genexpr)r\ 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' r1 r9 r@ rB rF r] r_ re ro rp ru ry 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� ZdS )� TestBugletsc C s< dd� }| � t�� |� W d � n1 s.0 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 s2 t d�D ]$}zt�W q tp ty* Y q0 qd S )Nrj )r[ � Exception)r �ir r r �test_bpo_42057
s
zTestBuglets.test_bpo_42057N)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
p
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists