File: //usr/local/lib/python3.10/test/__pycache__/test_pstats.cpython-310.pyc
o
�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 ZdZdd� ZdS )�AddCallersTestCasez$Tests for pstats.add_callers helper.c 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 �-/usr/local/lib/python3.10/test/test_pstats.py�test_combine_results
s
z'AddCallersTestCase.test_combine_resultsN)�__name__�
__module__� __qualname__�__doc__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- )�f�fiZfil�fileZfilenZfilenaZfilenamr- r' r1 )r Zsortkeyr2 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 r; r< r r r r r �pass2J r> z3StatsTestCase.test_get_stats_profile.<locals>.pass2c S r; r< r r r r r �pass3K r> z3StatsTestCase.test_get_stats_profile.<locals>.pass3r= r? r@ ) �cProfileZProfile�enableZ 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 ZFILENAMEZassertNotEqualZCALLSr9 r r r �test_SortKey_enum[ s z StatsTestCase.test_SortKey_enumN) r r r r r" r, r3 r7 r: rE rF r r r r r s
r �__main__)
Zunittest�testr �ior r r rA ZTestCaser r r �mainr r r r �<module> s C �