File: //usr/local/lib/python3.7/test/__pycache__/test_sys_setprofile.cpython-37.opt-1.pyc
B
��g�- � @ s� d dl Z d dlZd dlZd dlZG dd� dej�ZG dd� d�ZG dd� de�ZG dd � d ej�ZG d
d � d e�Z G d d
� d
e�Z
dd� Z dd� Z e e �Z
ddd�Zdd� Zedkr�e�� dS )� Nc @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�TestGetProfilec C s t �d � d S )N)�sys�
setprofile)�self� r �4/usr/local/lib/python3.7/test/test_sys_setprofile.py�setUp s zTestGetProfile.setUpc C s t �d � d S )N)r r )r r r r �tearDown s zTestGetProfile.tearDownc C s | � t�� � d S )N)Z assertIsNoner �
getprofile)r r r r �
test_empty s zTestGetProfile.test_emptyc C s& dd� }t �|� | �t �� |� d S )Nc W s d S )Nr )�argsr r r �fn s z&TestGetProfile.test_setget.<locals>.fn)r r ZassertIsr
)r r
r r r � test_setget s
zTestGetProfile.test_setgetN)�__name__�
__module__� __qualname__r r r r r r r r r s r c @ s. e Zd Zdd� Zdd� Zd
dd�Zdd � ZdS ) � HookWatcherc C s g | _ g | _d S )N)�frames�events)r r r r �__init__ s zHookWatcher.__init__c C s( |dks|dks|dkr$| � ||� d S )N�call�return� exception)� add_event)r �frame�event�argr r r �callback s zHookWatcher.callbackNc C sh |dkrt �d�}y| j�|�}W n* tk
rL t| j�}| j�|� Y nX | j�||t|�f� dS )zAdd an event to the log.N� ) r � _getframer �index�
ValueError�len�appendr �ident)r r r Zframenor r r r # s
zHookWatcher.add_eventc s. t | jj�t t �g� d| _� fdd�| jD �S )z Remove calls to add_event().Nc s g | ]}|d � kr|�qS )� r )�.0�item)�
disallowedr r �
<listcomp>5 s z*HookWatcher.get_events.<locals>.<listcomp>)r$ r �__func__r r )r r )r( r �
get_events0 s zHookWatcher.get_events)N)r r r r r r r+ r r r r r s
r c @ sN e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zeeeeeed
�Z dS )�ProfileSimulatorc C s || _ g | _t�| � d S )N)�testcase�stackr r )r r- r r r r 9 s zProfileSimulator.__init__c C s | j | | |� d S )N)�dispatch)r r r r r r r r > s zProfileSimulator.callbackc C s | � d|� | j�|� d S )Nr )r r. r# )r r r r r �
trace_callB s zProfileSimulator.trace_callc C s | � d|� | j�� d S )Nr )r r. �pop)r r r r r � trace_returnF s z ProfileSimulator.trace_returnc C s | j �d� d S )Nz2the profiler should never receive exception events)r- �fail)r r r r r �trace_exceptionJ s z ProfileSimulator.trace_exceptionc C s d S )Nr )r r r r r �
trace_passN s zProfileSimulator.trace_pass)r r r Zc_callZc_returnZ c_exceptionN)
r r r r r r0 r2 r4 r5 r/ r r r r r, 8 s r, c @ s e Zd Zdd� ZdS )� TestCaseBasec C s8 t || �� �}||kr4| �dt�|�t�|�f � d S )Nz'Expected events:
%s
Received events:
%s)�capture_events� new_watcherr3 �pprintZpformat)r �callableZexpectedr r r r � check_events\ s zTestCaseBase.check_eventsN)r r r r; r r r r r6 [ s r6 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 S ) �ProfileHookTestCasec C s t � S )N)r )r r r r r8 d s z ProfileHookTestCase.new_watcherc C s0 dd� }t |�}| �|dd|fdd|fg� d S )Nc S s d S )Nr )�pr r r �fh s z*ProfileHookTestCase.test_simple.<locals>.fr r r )r$ r; )r r> �f_identr r r � test_simpleg s z ProfileHookTestCase.test_simplec C s0 dd� }t |�}| �|dd|fdd|fg� d S )Nc S s dd d S )Nr r r )r= r r r r> p s z-ProfileHookTestCase.test_exception.<locals>.fr r r )r$ r; )r r> r? r r r �test_exceptiono s z"ProfileHookTestCase.test_exceptionc C s0 dd� }t |�}| �|dd|fdd|fg� d S )Nc S s y dd W n Y nX d S )Nr r r )r= r r r r> x s z4ProfileHookTestCase.test_caught_exception.<locals>.fr r r )r$ r; )r r> r? r r r �test_caught_exceptionw s z)ProfileHookTestCase.test_caught_exceptionc C s0 dd� }t |�}| �|dd|fdd|fg� d S )Nc S s y dd W n Y nX d S )Nr r r )r= r r r r> � s z;ProfileHookTestCase.test_caught_nested_exception.<locals>.fr r r )r$ r; )r r> r? r r r � test_caught_nested_exception� s z0ProfileHookTestCase.test_caught_nested_exceptionc C s0 dd� }t |�}| �|dd|fdd|fg� d S )Nc S s dd d S )Nr r r )r= r r r r> � s z4ProfileHookTestCase.test_nested_exception.<locals>.fr r r )r$ r; )r r> r? r r r �test_nested_exception� s z)ProfileHookTestCase.test_nested_exceptionc sd dd� � � fdd�}t � �}t |�}| �|dd|fdd|fdd|fd d|fd d|fdd|fg� d S )
Nc S s dd d S )Nr r r )r= r r r r> � s z>ProfileHookTestCase.test_exception_in_except_clause.<locals>.fc s8 y � | � W n& y � | � W n Y nX Y nX d S )Nr )r= )r> r r �g� s z>ProfileHookTestCase.test_exception_in_except_clause.<locals>.gr r r% r � )r$ r; )r rE r? �g_identr )r> r � test_exception_in_except_clause� s z3ProfileHookTestCase.test_exception_in_except_clausec
s\ dd� � � fdd�}t � �}t |�}| �|dd|fdd|fdd|fdd |fdd|fg� d S )
Nc S s dd d S )Nr r r )r= r r r r> � s z9ProfileHookTestCase.test_exception_propagation.<locals>.fc s z � | � W d | � d� X d S )Nzfalling through)r )r= )r> r r rE � s z9ProfileHookTestCase.test_exception_propagation.<locals>.gr r r% r zfalling through)r$ r; )r rE r? rG r )r> r �test_exception_propagation� s z.ProfileHookTestCase.test_exception_propagationc C s0 dd� }t |�}| �|dd|fdd|fg� d S )Nc S s& y dd W n dd Y nX d S )Nr r r )r= r r r r> � s z/ProfileHookTestCase.test_raise_twice.<locals>.fr r r )r$ r; )r r> r? r r r �test_raise_twice� s z$ProfileHookTestCase.test_raise_twicec C s0 dd� }t |�}| �|dd|fdd|fg� d S )Nc S s y dd W n � Y nX d S )Nr r r )r= r r r r> � s z1ProfileHookTestCase.test_raise_reraise.<locals>.fr r r )r$ r; )r r> r? r r r �test_raise_reraise� s z&ProfileHookTestCase.test_raise_reraisec C s0 dd� }t |�}| �|dd|fdd|fg� d S )Nc S s
t � �d S )N)� Exception)r= r r r r> � s z)ProfileHookTestCase.test_raise.<locals>.fr r r )r$ r; )r r> r? r r r �
test_raise� s z ProfileHookTestCase.test_raisec s� dd� � � fdd���fdd���fdd���fd d
�}t � �}t ��}t ��}t ��}t |�}| �|d d |fd
d |fdd |fdd |fdd |fdd|fdd|fdd|fd
d|fd d|fg
� d S )Nc S s dd d S )Nr r r r r r r r> � s z5ProfileHookTestCase.test_distant_exception.<locals>.fc s
� � d S )Nr r )r> r r rE � s z5ProfileHookTestCase.test_distant_exception.<locals>.gc s
� � d S )Nr r )rE r r �h� s z5ProfileHookTestCase.test_distant_exception.<locals>.hc s
� � d S )Nr r )rN r r �i� s z5ProfileHookTestCase.test_distant_exception.<locals>.ic s
� � d S )Nr )r= )rO r r �j� s z5ProfileHookTestCase.test_distant_exception.<locals>.jr r r% rF � � r )r$ r; )r rP r? rG �h_ident�i_ident�j_identr )r> rE rN rO r �test_distant_exception� s( z*ProfileHookTestCase.test_distant_exceptionc
st dd� � � fdd�}t � �}t |�}| �|dd|fdd|fdd|fdd|fdd|fdd|fdd|fdd|fg� d S ) Nc s s xt d�D ]
} | V q
W d S )Nr% )�range)rO r r r r> � s z-ProfileHookTestCase.test_generator.<locals>.fc s x� � D ]}qW d S )Nr )r= rO )r> r r rE � s z-ProfileHookTestCase.test_generator.<locals>.gr r r% r )r$ r; )r rE r? rG r )r> r �test_generator� s z"ProfileHookTestCase.test_generatorc
st dd� � � fdd�}t � �}t |�}| �|dd|fdd|fdd|fdd|fdd|fdd|fdd|fdd|fg� d S ) Nc s s xt d�D ]
} | V q
W d S )Nr% )rW )rO r r r r> s z2ProfileHookTestCase.test_stop_iteration.<locals>.fc s x� � D ]}qW d S )Nr )r= rO )r> r r rE s z2ProfileHookTestCase.test_stop_iteration.<locals>.gr r r% r )r$ r; )r rE r? rG r )r> r �test_stop_iteration s z'ProfileHookTestCase.test_stop_iterationN)r r r r8 r@ rA rB rC rD rH rI rJ rK rM rV rX rY r r r r r< c s r< c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�ProfileSimulatorTestCasec C s t | �S )N)r, )r r r r r8 s z$ProfileSimulatorTestCase.new_watcherc C s0 dd� }t |�}| �|dd|fdd|fg� d S )Nc S s d S )Nr )r= r r r r> s z/ProfileSimulatorTestCase.test_simple.<locals>.fr r r )r$ r; )r r> r? r r r r@ s z$ProfileSimulatorTestCase.test_simplec C s0 dd� }t |�}| �|dd|fdd|fg� d S )Nc S s dd d S )Nr r r )r= r r r r> % s z8ProfileSimulatorTestCase.test_basic_exception.<locals>.fr r r )r$ r; )r r> r? r r r �test_basic_exception$ s z-ProfileSimulatorTestCase.test_basic_exceptionc C s0 dd� }t |�}| �|dd|fdd|fg� d S )Nc S s y dd W n Y nX d S )Nr r r )r= r r r r> - s z9ProfileSimulatorTestCase.test_caught_exception.<locals>.fr r r )r$ r; )r r> r? r r r rB , s z.ProfileSimulatorTestCase.test_caught_exceptionc s� dd� � � fdd���fdd���fdd���fd d
�}t � �}t ��}t ��}t ��}t |�}| �|d d |fd
d |fdd |fdd |fdd |fdd|fdd|fdd|fd
d|fd d|fg
� d S )Nc S s dd d S )Nr r r r r r r r> 6 s z:ProfileSimulatorTestCase.test_distant_exception.<locals>.fc s
� � d S )Nr r )r> r r rE 8 s z:ProfileSimulatorTestCase.test_distant_exception.<locals>.gc s
� � d S )Nr r )rE r r rN : s z:ProfileSimulatorTestCase.test_distant_exception.<locals>.hc s
� � d S )Nr r )rN r r rO < s z:ProfileSimulatorTestCase.test_distant_exception.<locals>.ic s
� � d S )Nr )r= )rO r r rP > s z:ProfileSimulatorTestCase.test_distant_exception.<locals>.jr r r% rF rQ rR r )r$ r; )r rP r? rG rS rT rU r )r> rE rN rO r rV 5 s( z/ProfileSimulatorTestCase.test_distant_exceptionc C s0 dd� }t |�}| �|dd|fdd|fg� d S )Nc S s t �� d S )N)�dict�get)r= r r r r> S s z?ProfileSimulatorTestCase.test_unbound_method_no_args.<locals>.fr r r )r$ r; )r r> r? r r r �test_unbound_method_no_argsR s z4ProfileSimulatorTestCase.test_unbound_method_no_argsc C s0 dd� }t |�}| �|dd|fdd|fg� d S )Nc S s t �td� d S )N�* )r\ r] �print)r= r r r r> [ s zDProfileSimulatorTestCase.test_unbound_method_invalid_args.<locals>.fr r r )r$ r; )r r> r? r r r � test_unbound_method_invalid_argsZ s z9ProfileSimulatorTestCase.test_unbound_method_invalid_argsN)
r r r r8 r@ r[ rB rV r^ ra r r r r rZ s rZ c C s$ t | d�r| j}n| j}|j|jfS )N�f_code)�hasattrrb �__code__�co_firstlineno�co_name)Zfunction�coder r r r$ b s
r$ c C s y | |� W n Y nX d S )Nr )r> r= r r r �protectj s rh c C sb |d krt � }t�� }t�� z$t�|j� t| |� t�d � W d |rPt�� X |� � dd� S )Nr ���)
r �gcZ isenabledZdisabler r r rh Zenabler+ )r: r= Zold_gcr r r r7 q s
r7 c C s dd l }|� t| �� d S )Nr )r9 r7 )r: r9 r r r � show_events� s rk �__main__)N)rj r9 r ZunittestZTestCaser r r, r6 r< rZ r$ rh Z
protect_identr7 rk r �mainr r r r �<module> s" # 6J