Your IP : 216.73.216.213
a
XC?h; � @ st d dl Z d dlmZ d dlmZ d dlmZ d dlZd dlZG dd� de j�Z G dd� de j�Z
e d krpe � � dS )
� N)�support)�StringIO)�SortKeyc @ s e Zd Zdd� ZdS )�AddCallersTestCasec C sd ddi}ddd�}t �||�}| �|ddd�� ddi}ddd�}t �||�}| �|ddd�� d S ) N�a)� � � � )� � � � )r �b)r r
r r r r r )�pstatsZ add_callers� assertEqual)�self�target�sourceZ new_callers� r �5/opt/alt/python39/lib64/python3.9/test/test_pstats.py�test_combine_results
s
z'AddCallersTestCase.test_combine_resultsN)�__name__�
__module__� __qualname__r r r r r r
s r c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�
StatsTestCasec C s t �d�}t�|�| _d S )Nz
pstats.pck)r Zfindfiler �Stats�stats)r Z
stats_filer r r �setUp s
zStatsTestCase.setUpc C s& t � }tj|d�}|�| j| j� d S )N)�stream)r r r �addr )r r r r r r �test_add! s zStatsTestCase.test_addc C sL ddddd�}|� � D ]0\}}| j�|� | �| jj| jj| d � qd S )N�stdname�calls�time�
cumulative)���r r r r& )�itemsr �
sort_statsr � sort_type�sort_arg_dict_default)r Z
valid_argsZarg_intZarg_strr r r �test_sort_stats_int& s �
�z!StatsTestCase.test_sort_stats_intc C s6 dD ],}| j �|� | �| j j| j j| d � qd S )N)
r# ZncallsZcumtimer% �filename�line�module�nameZnflZpcallsr" r$ Ztottimer& �r r( r r) r* )r � sort_namer r r �test_sort_stats_string0 s
�z$StatsTestCase.test_sort_stats_stringc C s: d}dD ],}| j �|� | �| j j| j j| d � qd S )Nr, )�fZfiZfil�fileZfilenZfilenaZfilenamr, r& r0 )r Zsortkeyr1 r r r �test_sort_stats_partial8 s
�z%StatsTestCase.test_sort_stats_partialc C s8 t D ].}| j�|� | �| jj| jj|j d � qd S )Nr& )r r r( r r) r* �value)r �memberr r r �test_sort_stats_enum@ s �z"StatsTestCase.test_sort_stats_enumc C s0 | � t| jjdtj� | � t| jjtjd� d S )Nr# )Z assertRaises� TypeErrorr r( r ZTIME�r r r r �test_sort_starts_mixG s � �z"StatsTestCase.test_sort_starts_mixc C s� dd� }dd� }dd� }t �� }|�� |� |� |� |�� t�|�}|�� }t|j� � �}| �
d|� | �
d|� | �
d |� d S )
Nc S s d S �Nr r r r r �pass1P � z3StatsTestCase.test_get_stats_profile.<locals>.pass1c S s d S r< r r r r r �pass2Q r> z3StatsTestCase.test_get_stats_profile.<locals>.pass2c S s d S r< r r r r r �pass3R r> z3StatsTestCase.test_get_stats_profile.<locals>.pass3r= r? r@ ) �cProfileZProfileZenableZ create_statsr r Zget_stats_profile�setZ
func_profiles�keysZassertIn)r r= r? r@ ZprZpsZ
stats_profileZ funcs_calledr r r �test_get_stats_profileO s
z$StatsTestCase.test_get_stats_profilec C s" | � tjd� | �tjtj� d S )Nr, )r r ZFILENAMEZassertNotEqualZCALLSr: r r r �test_SortKey_enumb s z StatsTestCase.test_SortKey_enumN) r r r r r! r+ r2 r5 r8 r; rD rE r r r r r s
r �__main__)
Zunittest�testr �ior r r rA ZTestCaser r r �mainr r r r �<module> s J