File: //usr/local/lib/python3.7/test/__pycache__/test_pstats.cpython-37.opt-2.pyc
B
��g� � @ sl d dl Z d dlmZ d dlmZ d dlZd dlmZ G dd� de j�ZG dd� de j�Z e
d krhe � � 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 �,/usr/local/lib/python3.7/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 @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� 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 sP ddddd�}x<|� � D ]0\}}| j�|� | �| jj| jj| d � qW 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 s: x4dD ],}| j �|� | �| j j| j j| d � qW d S )N)
r# ZncallsZcumtimer% �filename�line�module�nameZnflZpcallsr" r$ Ztottimer& )r r( r r) r* )r � sort_namer r r �test_sort_stats_string/ s
z$StatsTestCase.test_sort_stats_stringc C s> d}x4dD ],}| j �|� | �| j j| j j| d � q
W d S )Nr, )�f�fiZfil�fileZfilenZfilenaZfilenamr, r& )r r( r r) r* )r Zsortkeyr0 r r r �test_sort_stats_partial7 s
z%StatsTestCase.test_sort_stats_partialc C s< x6t D ].}| j�|� | �| jj| jj|j d � qW d 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_mixF s z"StatsTestCase.test_sort_starts_mixN)
r r r r r! r+ r1 r5 r8 r: r r r r r s
r �__main__) ZunittestZtestr �ior r r ZTestCaser r r �mainr r r r �<module> s 4