Your IP : 216.73.216.196
o
0�i=
� @ sx d dl Z d dlmZ d dlmZ d dlmZ d dlZd dlZG dd� de j�Z G dd� de j�Z
e d kr:e � � dS dS )
� N)�support)�StringIO)�SortKeyc @ s e Zd Z dd� 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 �7/opt/alt/python310/lib64/python3.10/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 @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� 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 ]\}}| j�|� | �| jj| jj| d � q d 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 s0 | � t| jjdtj� | � t| jjtjd� d S )Nr# )Z assertRaises� TypeErrorr r( r ZTIME�r r r r �test_sort_starts_mix@ s � �z"StatsTestCase.test_sort_starts_mixc C s� dd� }dd� }dd� }t �� }|�� |� |� |� |�� t�|�}|�� }t|j� � �}| �
d|� | �
d|� | �
d |� d S )
Nc S � d S �Nr r r r r �pass1I � z3StatsTestCase.test_get_stats_profile.<locals>.pass1c S r9 r: r r r r r �pass2J r< z3StatsTestCase.test_get_stats_profile.<locals>.pass2c S r9 r: r r r r r �pass3K 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_profileH s
z$StatsTestCase.test_get_stats_profilec C s" | � tjd� | �tjtj� d S )Nr, )r r ZFILENAMEZassertNotEqualZCALLSr7 r r r �test_SortKey_enum[ s z StatsTestCase.test_SortKey_enumN) r r r r r! r+ r2 r5 r8 rB rC r r r r r s
r �__main__)
Zunittest�testr �ior r r r? ZTestCaser r r �mainr r r r �<module> s C �