Your IP : 216.73.216.52
3
� f�� � @ sh d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZm Z m
Z
d dl m Z y d dl
Z
W n ek
rx dZ
Y nX y d dlZW n ek
r� dZY nX ejd�Zdd� Zdd � Zd
d � Zd d
� Zdd� Zdd� Zdd� ZG dd� dej�ZG dd� dej�ZG dd� dej�Z G dd� dej�Z ej edkd �G d d � d ej��Z d d � Z e!d!k�rde � dS )"� N)�patch)�assert_python_ok�assert_python_failure� interpreter_requires_environment)�support� c C s\ g }t jd�}xDt| �D ]8}|j}|j| }|j|j|f� d}|j}|d krP qW t|�S )N� r ) �sys� _getframe�range�f_code�f_lineno�append� co_filename�f_back�tuple)�nframeZ lineno_delta�frames�frame�index�code�lineno� r �:/opt/alt/python36/lib64/python3.6/test/test_tracemalloc.py�
get_frames s
r c C s0 t j� }| t }t|d�}d| }|t j|�fS )Nr � x)� tracemalloc�get_traceback_limit�EMPTY_STRING_SIZEr � Traceback)�sizer Z bytes_lenr �datar r r �allocate_bytes$ s
r" c C sD d} ddd d d#d&g}t j|| �}d*d.d2d6d:d=g}t j|| �}||fS )>N� r �
�a.py�b.py� r � �B � � � <unknown>� � �c.py�B �r% r# �r& r' �r1 r2 )r r$ r3 �r% r# �r& r' �r4 r5 )r r$ r6 �r% r# �r& r' �r7 r8 )r r$ r9 �r% r( �r& r' �r: r; )r r# r< �r& r �r= )r# r) r> �r, r �r? )r* r+ r@ �r% r# �r& r' �rA rB )r r$ rC �r% r# �r& r' �rD rE )r r$ rF �r% r# �r& r' �rG rH )r r$ rI �r% r( �r& r' �rJ rK )r# r# rL �r% r( �r& r' �rM rN )r# r- rO �r/ r0 �rP )r' r. rQ )r �Snapshot)�traceback_limit�
raw_traces�snapshotZ raw_traces2� snapshot2r r r �create_snapshots+ s rW c C s t j| |f�S )N)r Z_Frame)�filenamer r r r r L s r c G s
t j| �S )N)r r )r r r r � tracebackO s rY c C s t | |f�S )N)rY )rX r r r r �traceback_linenoR s rZ c C s
t | d�S )Nr )rZ )rX r r r �traceback_filenameU s r[ 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 � Zejeed �d �d d � �Zd!S )"�TestTracemallocEnabledc C s t j� r| jd� t jd� d S )Nz+tracemalloc must be stopped before the testr )r �
is_tracing�skipTest�start)�selfr r r �setUpZ s
z TestTracemallocEnabled.setUpc C s t j� d S )N)r �stop)r` r r r �tearDown` s z TestTracemallocEnabled.tearDownc C sR dd� t d�D �}tj� }| j|d� tj� tj� }| j|d� | j||� d S )Nc S s g | ] }t d ��qS )�{ )r" )�.0�countr r r �
<listcomp>d s zFTestTracemallocEnabled.test_get_tracemalloc_memory.<locals>.<listcomp>i� r )r r Zget_tracemalloc_memory�assertGreaterEqual� clear_traces�assertLessEqual)r` r! r �size2r r r �test_get_tracemalloc_memoryc s z2TestTracemallocEnabled.test_get_tracemalloc_memoryc C s2 t j� d}t|�\}}t j|�}| j||� d S )Ni90 )r ri r" �get_object_traceback� assertEqual)r` �obj_size�obj�
obj_tracebackrY r r r �test_get_object_tracebackm s
z0TestTracemallocEnabled.test_get_object_tracebackc C s� d}t j� | jtt jd� t j� t jd� t|�\}}t j|�}| jt|�d� | j||� t j� t jd� t|�\}}t j|�}| jt|�d� | j||� d S )Nr$ r ���) r rb � assertRaises�
ValueErrorr_ r" rm rn �len)r` ro �obj2�obj2_tracebackrY rp rq r r r �test_set_traceback_limitt s
z/TestTracemallocEnabled.test_set_traceback_limitc C s. x |D ]}|d |j kr|S qW | jd� d S )Nr# ztrace not found)�_framesZfail)r` �tracesrY �tracer r r �
find_trace� s
z!TestTracemallocEnabled.find_tracec C sx t j� d}t|�\}}t j� }| j||�}| j|t� |\}}}| j||� | j||j� t j � | jt j� g � d S )Ni90 )
r ri r" � _get_tracesr} ZassertIsInstancer rn rz rb ) r` ro rp rq r{ r| �domainr rY r r r �test_get_traces� s
z&TestTracemallocEnabled.test_get_tracesc s� dd� � � fdd���fdd�}t j� t jd� d}||�\}}||�\}}t j� }| j||�}| j||�} |\}
} } | \}
}}| j|| � d S ) Nc S s t | �S )N)r" )r r r r �allocate_bytes2� s zPTestTracemallocEnabled.test_get_traces_intern_traceback.<locals>.allocate_bytes2c s � | �S )Nr )r )r� r r �allocate_bytes3� s zPTestTracemallocEnabled.test_get_traces_intern_traceback.<locals>.allocate_bytes3c s � | �S )Nr )r )r� r r �allocate_bytes4� s zPTestTracemallocEnabled.test_get_traces_intern_traceback.<locals>.allocate_bytes4r' rd )r rb r_ r~ r} ZassertIs)r` r� ro Zobj1Zobj1_tracebackrw rx r{ Ztrace1Ztrace2Zdomain1Zsize1Z
traceback1Zdomain2rk �
traceback2r )r� r� r � test_get_traces_intern_traceback� s
z7TestTracemallocEnabled.test_get_traces_intern_tracebackc C s� d}d}t j� t|�\}}t j� \}}| j||� | j||� | j|| |� | j|| |� d }t j� \}}| j||� | j|| || � | j||� t j� | jt j� d� t|�\}}t j� \}}| j||� t j� | jt j� d� d S )Ni i r i )r r )r r ) r ri r" �get_traced_memoryrh rj Z
assertLessrn rb ) r` Z max_errorro rp rq r Z peak_sizerk Z
peak_size2r r r �test_get_traced_memory� s* z-TestTracemallocEnabled.test_get_traced_memoryc C s@ t d�\}}tj|�}| j|� tj� tj|�}| j|� d S )Nrd )r" r rm �assertIsNotNoneri � assertIsNone)r` rp rq rY r� r r r �test_clear_traces� s
z(TestTracemallocEnabled.test_clear_tracesc C s0 t j� | jt j� � t j� | jt j� � d S )N)r rb � assertFalser] r_ �
assertTrue)r` r r r �test_is_tracing� s z&TestTracemallocEnabled.test_is_tracingc
C s� t d�\}}tj� }|jtj� | jtjtj� tjj tj�}| j
|j |j � tj � | j
t��}tj� W d Q R X | j
t|j�d� d S )Nrd zLthe tracemalloc module must be tracing memory allocations to take a snapshot)r" r �
take_snapshot�dumpr �TESTFN�
addCleanup�unlinkrR �loadrn r{ rb rt � RuntimeError�strZ exception)r` rp �sourcerU rV �cmr r r �
test_snapshot� s z$TestTracemallocEnabled.test_snapshotc C sJ t j� }d|_|jtj� | jtjtj� t jj tj�}| j
|jd� d S )N�new) r r� Z test_attrr� r r� r� r� rR r� rn )r` rU rV r r r �test_snapshot_save_attr s z.TestTracemallocEnabled.test_snapshot_save_attrc C s6 t j� s dS d}t|�\}}t j|�}|d kr2dS dS )Nr# i90 r* r )r r] r" rm )r` ro rp rq rY r r r �
fork_child s
z!TestTracemallocEnabled.fork_child�forkzneed os.fork()c C sf t j� }|s,d}z | j� }W d t j|� X n6t j|d�\}}| jt j|�� t j|�}| j|d� d S )Nr r ) �osr� r� �_exit�waitpidr� � WIFEXITED� WEXITSTATUSrn )r` �pidZexitcodeZpid2Zstatusr r r � test_fork s
z TestTracemallocEnabled.test_forkN)�__name__�
__module__� __qualname__ra rc rl rr ry r} r� r� r� r� r� r� r� r� �unittestZ
skipUnless�hasattrr� r� r r r r r\ Y s
$
r\ c @ sx e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd S ) � TestSnapshoti� c
C s� d
g}t j� ��}|jtjtddd�� |jtjtddd�� |jtjtd |d�� tj� }| j|jd� | jt |j
�d
� |j
d }| j|j d� | jt |j �d
� | j|j d j
d� | j|j d jd� W d Q R X d S )Nr r( �a.pyr# r] T)Z return_valuer r~ r �r� r# �r� )r r( r� )�
contextlib� ExitStack�
enter_contextr �objectr r� rn rS rv r{ r rY rX r )r` rT �stackrU r| r r r �test_create_snapshot. s
z!TestSnapshot.test_create_snapshotc
C s� t � \}}tjdd�}tjddd�}tjddd�}t|jj�}|j|f�}| j|jjdddd d g� | j|jj|� |j||f�}| j|jjd$d(d,d0g� |jf �} | j| |� | j| j|j� | j| j|j� | j t
|j|� d S )1NF�b.pyT�a.pyr# r( r r$ r' r r* r+ � <unknown>�r� r# �r� r' �r� r� )r r$ r� �r� r# �r� r' �r� r� )r r$ r� �r� r# �r� r' �r� r� )r r$ r� �r� r( �r� r' �r� r� )r r# r� �r� r �r� )r* r+ r� �r� r# �r� r' �r� r� )r r$ r� �r� r# �r� r' �r� r� )r r$ r� �r� r# �r� r' �r� r� )r r$ r� �r� r( �r� r' �r� r� )r r# r� ) rW r �Filter�listr{ �_traces�
filter_tracesrn Z assertIsNotrt � TypeError)
r` rU rV �filter1�filter2Zfilter3�original_traces� snapshot3Z snapshot4Z snapshot5r r r �test_filter_tracesB s0
z TestSnapshot.test_filter_tracesc
C s� t � \}}tjdddd�}tjdddd�}t|jj�}|j|f�}| j|jjdddd d g� |j|f�}| j|jjd$d(d,d/d2g� d S )3NF�a.pyr )r Tr r$ r# �b.pyr' r) r* r+ � <unknown>�r� r# �r� r' �r� r� )r r$ r� �r� r# �r� r' �r� r� )r r$ r� �r� r# �r� r' �r� r� )r r$ r� �r� r �r� )r# r) r� �r� r �r� )r* r+ r� �r� r# �r� r' �r� r� )r r$ r� �r� r# �r� r' �r� r� )r r$ r� �r� r# �r� r' �r� r� )r r$ r� �r� r �r� )r# r) r� �r� r �r� )r* r+ r� )rW r r� r� r{ r� r� rn )r` rU rV r� r� r� r� r r r �test_filter_traces_domainh s$
z&TestSnapshot.test_filter_traces_domainc
C sn t � \}}tjddd�}tjddd�}|j|f�}| j|jjdddd d"g� |j|f�}| j|jjd%g� d S )&NFr* )r Tr r$ �a.pyr# �b.pyr' r r( r) r+ � <unknown>�r� r# �r� r' �r� r� )r r$ r� �r� r# �r� r' �r� r� )r r$ r �r� r# �r� r' �r r )r r$ r �r� r( �r� r' �r r )r r# r �r� r �r )r# r) r �r� r �r )r* r+ r
)rW r � DomainFilterr� rn r{ r� )r` rU rV r� r� r� r r r � test_filter_traces_domain_filter� s
z-TestSnapshot.test_filter_traces_domain_filterc C s( t � \}}tdd�}tdd�}tdd�}tdd�}tdd �}|jd
�}| j|tj|d d�tj|d d
�tj|dd�tj|dd�g� |jd
�} | j| tj|dd�tj|dd�tj|d d
�g� |j|d
�}
| j|
tj|dddd�tj|dddd�tj|dddd�tj|dddd�tj|d dd
d�g� d S )Nz <unknown>r za.pyr# r( zb.pyr zc.pyiB r r) � r* r+ i� i� i� i����rs i����rs )rW rZ �
statisticsrn r � Statistic�
compare_to�
StatisticDiff) r` rU rV �tb_0�tb_a_2�tb_a_5�tb_b_1Ztb_c_578�stats1�stats2r r r r �test_snapshot_group_by_line� s0
z(TestSnapshot.test_snapshot_group_by_linec
C s� t � \}}td�}td�}td�}td�}|jd�}| j|tj|dd�tj|dd �tj|d
d�g� |jd�}| j|tj|d d �tj|d
d�g� |j|d�} | j| tj|d dd d�tj|d
d
dd�tj|dddd�tj|dddd�g� d S )Nz <unknown>za.pyzb.pyzc.pyrX r) r � r' r+ i� r( i� i� r i����rs i����rs )rW r[ r rn r r r r )
r` rU rV r �tb_a�tb_bZtb_cr r �diffr r r �test_snapshot_group_by_file� s(
z(TestSnapshot.test_snapshot_group_by_filec C s6 t � \}}tdd�}tdd�}td�}td�}|jd �}| j|tj|d
d�tj|d d �tj|d
d�tj|dd�g� td�}|jd �} | j| tj|dd�tj|dd�tj|d d �g� |j|d �}
| j|
tj|dddd�tj|dddd�tj|dd dd �tj|dd dd �tj|d dd d�g� | jt |jd dd� d S ) N�a.pyr# �b.pyr' r( r � <unknown>r rY r) r
r* r+ �c.py�B i� i� i� T)Z
cumulative)r r# )r r' )r r( )r r' )r r )r r )r! r" i����rs i����rs )
rW rY r rn r r r r rt ru ) r` rU rV Ztb1Ztb2Ztb3Ztb4r Ztb5r r r r r � test_snapshot_group_by_traceback� s4
z-TestSnapshot.test_snapshot_group_by_tracebackc
C s� t � \}}td�}td�}td�}tdd�}tdd�}tdd�}tdd�} |jdd �}
| j|
tj|d
d�tj|d d�tj|d d�g� |jd
d �}
| j|
tj|dd�tj| d d�tj|dd�tj|d d�tj|dd�g� d S )Nz <unknown>za.pyzb.pyr# r( r r' rX T�b r r+ r r) r
r* )rW r[ rZ r rn r r ) r` rU rV r r r r r r Ztb_b_4�statsr r r �!test_snapshot_group_by_cumulative s(
z.TestSnapshot.test_snapshot_group_by_cumulativec C sV t � \}}|jd }| jt|�d� |j}| jt|�d� |d }| jt|�d� d S )Nr z a.py:2: 10 Bza.py:2)rW r{ rn r� rY )r` rU rV r| rY r r r r �test_trace_format s
z TestSnapshot.test_trace_formatc C s0 t � \}}|jd�}|d }| jt|�d� d S )Nr r z(b.py:1: size=66 B, count=1, average=66 B)rW r rn r� )r` rU rV r% �statr r r �test_statistic_format% s
z"TestSnapshot.test_statistic_formatc C s2 t � \}}|j|d�}|d }| jt|�d� d S )Nr r z;a.py:5: size=5002 B (+5000 B), count=2 (+1), average=2501 B)rW r rn r� )r` rU rV r% r( r r r �test_statistic_diff_format, s
z'TestSnapshot.test_statistic_diff_formatc C s` t � \}}| j|jd d� |jd |jd f� |jd j}| j|d d� |d |d f� d S )Nr# r r )rW rn r{ rY )r` rU rV rY r r r � test_slices3 s
zTestSnapshot.test_slicesc
C s� t � \}}dd� }tjjd|d��V |jd j}| j|j� dddd g� | j|jd
d �ddg� | j|jd d �g � W d Q R X d S )
Nc S s d| |f S )Nz
<%s, %s>r )rX r r r r �getline> s z3TestSnapshot.test_format_traceback.<locals>.getlinez tracemalloc.linecache.getline)Z side_effectr z File "a.py", line 2z
<a.py, 2>z File "b.py", line 4z
<b.py, 4>r )�limitrs )rW r� Zmockr r{ rY rn �format)r` rU rV r, �tbr r r �test_format_traceback< s
z"TestSnapshot.test_format_tracebackN)r� r� r� �maxDiffr� r� r� r r r r# r& r' r) r* r+ r0 r r r r r� + s &# & r� c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
� TestFiltersi c C s� t jdd�}| j|jd� | j|jd� | j|j� | j|jd� t jdddd�}| j|jd� | j|jd� | j|jd� | j|jd� t jddddd�}| j|jd� | j|jd� | j|jd� | j|jd� | jt t
|dd� d S )NT�abcFztest.pyrd )� inclusive�filename_patternr �
all_framesr5 ) r r� rn r4 r5 r� r r6 rt �AttributeError�setattr)r` �fr r r �test_filter_attributesT s z"TestFilters.test_filter_attributesc C s� t jdd�}| j|jdd�� | j|jdd�� | j|jdd�� | j|jdd�� | j|jdd�� | j|jdd�� t jdd�}| j|jdd�� | j|jdd�� | j|jdd�� | j|jdd�� | j|jdd�� | j|jdd�� t jddd�}| j|jdd�� | j|jdd�� | j|jdd�� | j|jdd�� | j|jdd�� | j|jdd�� t jddd�}| j|jdd�� | j|jdd�� | j|jdd�� | j|jdd�� | j|jdd�� | j|jdd�� t jddd�}| j|jdd�� | j|jdd�� | j|jdd�� | j|jdd�� | j|jdd�� | j|jdd�� t jddd�}| j|jdd�� | j|jdd�� | j|jdd�� | j|jdd�� | j|jdd�� | j|jdd�� d S )NTr3 r r( r$ �12356F)r r� r� � _match_framer� )r` r9 r r r �test_filter_matchm sT z TestFilters.test_filter_matchc C sx dd� }| j |ddd�� | j|ddd�� | j|ddd�� | j|ddd�� | j |ddd�� | j |ddd�� d S )Nc S s t j| |�}|j|d�S )Nr )r r� r< )r4 rX �patternr9 r r r �fnmatch� s z7TestFilters.test_filter_match_filename.<locals>.fnmatchTr3 r; z <unknown>F)r� r� )r` r? r r r �test_filter_match_filename� s z&TestFilters.test_filter_match_filenamec C s� dd� }| j |dd�� | j |dd�� | j|dd�� | j|dd�� | j|dd�� | j |dd�� | j |dd�� | j|dd�� | j|dd �� | j |dd
�� | j |dd �� | j|dd �� | j|d
d�� | j |dd �� | j |dd�� | j|dd�� | j|dd�� | j |dd�� | j |dd�� | j|dd�� | j|dd�� tjdk�r�| j|dd�� | j|d d �� | j|dd �� | j|dd �� n@| j |dd�� | j |d d �� | j |dd �� | j |dd �� tjdk�r4| j|d d!�� | j|d!d �� | j|d"d#�� | j|d$d%�� n@| j |d d!�� | j |d!d �� | j |d"d#�� | j |d$d%�� | j |d&d�� d S )'Nc S s t jd|�}|j| d�S )NTr )r r� r< )rX r> �filterr r r r? � s z=TestFilters.test_filter_match_filename_joker.<locals>.fnmatchr3 � �*Zabcd�defza*zabc*zb*zabcd*za*cZabcdcxza*cxZabbZabcdceZabcdeza*c*eZ abcbdefegza*bd*egZabcddZ abcbdefefza.pycza.py�ntZaBCZABcZaBcDezAb*dEza.PYza.PYCza/bza\bza/b\cza\b/cza/b/cza\b\cza.pyo)r� r� r� �name)r` r? r r r � test_filter_match_filename_joker� sR z,TestFilters.test_filter_match_filename_jokerc C sD d}d}d}d}t jd
dd
d �}| j|j|�� | j|j|�� | j|j|�� | j|j|�� t jd
dd d �}| j|j|�� | j|j|�� | j|j|�� | j|j|�� t jd dd
d �}| j|j|�� | j|j|�� | j|j|�� | j|j|�� t jd dd d �}| j|j|�� | j|j|�� | j|j|�� | j|j|�� t jd dd d �}| j|j|�� | j|j|�� | j|j|�� | j|j|�� t jd
dd
d �}| j|j|�� | j|j|�� | j|j|�� | j|j|�� t jd dd
d �}| j|j|�� | j|j|�� | j|j|�� | j|j|�� d S )N�a.pyr# �b.pyr* r' r( �c.py� <unknown>r T)r6 F�rH r# �rI r* )rL rM �rI r' �rI r( )rN rO �rJ r( �rK r )rP rQ �rK r )rR )r r� r� Z_match_tracebackr� )r` Zt1Zt2Zt3�unknownr9 r r r �test_filter_match_trace� sN z#TestFilters.test_filter_match_traceN) r� r� r� r1 r: r= r@ rG rT r r r r r2 Q s 4
Er2 c @ sj e Zd Zdd� Zeje� d�dd� �Zdd� Zdd � Z d
d � Z
d d
� Z dd� Z eje
dkd�dd� �ZdS )�TestCommandLinec C s, d}t d|�\}}}|j� }| j|d� d S )Nz3import tracemalloc; print(tracemalloc.is_tracing())z-cs False)r �rstriprn )r` r �ok�stdout�stderrr r r � test_env_var_disabled_by_default% s z0TestCommandLine.test_env_var_disabled_by_defaultz6Cannot run -E tests when PYTHON env vars are required.c C s2 d}t dd|dd�\}}}|j� }| j|d� dS )zAPYTHON* environment variables must be ignored when -E is present.z3import tracemalloc; print(tracemalloc.is_tracing())z-Ez-c�1)�PYTHONTRACEMALLOCs FalseN)r rV rn )r` r rW rX rY r r r �test_env_var_ignored_with_E, s z+TestCommandLine.test_env_var_ignored_with_Ec C s0 d}t d|dd�\}}}|j� }| j|d� d S )Nz3import tracemalloc; print(tracemalloc.is_tracing())z-cr[ )r\ s True)r rV rn )r` r rW rX rY r r r � test_env_var_enabled_at_startup5 s z/TestCommandLine.test_env_var_enabled_at_startupc C s0 d}t d|dd�\}}}|j� }| j|d� d S )Nz<import tracemalloc; print(tracemalloc.get_traceback_limit())z-cZ10)r\ s 10)r rV rn )r` r rW rX rY r r r �test_env_limit< s z TestCommandLine.test_env_limitc C sb x\d D ]T}| j |d��> tj� �* tddt|�d�\}}}| jd |� W d Q R X W d Q R X qW d S )
Nr r r# r
)r z-c�pass)r\ s+ PYTHONTRACEMALLOC: invalid number of framesrs � @)rs r ra )�subTestr �SuppressCrashReportr r� �assertIn)r` r rW rX rY r r r �test_env_var_invalidC s
z$TestCommandLine.test_env_var_invalidc
C sf x`dD ]X\}}| j ||d��< d}td|d |�\}}}|j� }| j|t|�jd
�� W d Q R X qW d S )Nr r �
tracemalloc=1�tracemalloc=15� )�xoptionsr z<import tracemalloc; print(tracemalloc.get_traceback_limit())z-Xz-c�ascii�r r �rf r �rg rh )rk rl rm )rb r rV rn r� �encode)r` ri r r rW rX rY r r r �test_sys_xoptionsN s z!TestCommandLine.test_sys_xoptionsc C sh xbd
D ]Z}| j |d��D tj� �0 dd| dd f}t|� \}}}| jd
|� W d Q R X W d Q R X qW d S )Nr r r# r
)r z-Xztracemalloc=%sz-cr` s/ -X tracemalloc=NFRAME: invalid number of framesrs � @)rs r rp )rb r rc r rd )r` r �argsrW rX rY r r r �test_sys_xoptions_invalidZ s
z)TestCommandLine.test_sys_xoptions_invalidNzneed _testcapic C s d}t ddd|� d S )Nz2import _testcapi; _testcapi.test_pymem_alloc0(); 1z-Xr z-c)r )r` r r r r �test_pymem_alloc0d s z!TestCommandLine.test_pymem_alloc0)r� r� r� rZ r� �skipIfr r] r^ r_ re ro rr � _testcapirs r r r r rU $ s
rU zneed _testcapic @ sz e Zd Zd Zdd� Zdd� Zdd� Zd!d d �Zd
d� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd � Zd d � Zd S )"�TestCAPI�P � c C s> t j� r| jd� d| _d| _t| j�d | _t| j�| _d S )Nz+tracemalloc must be stopped before the testr( rd r ) r r] r^ r r r" rp �id�ptr)r` r r r ra p s
zTestCAPI.setUpc C s t j� d S )N)r rb )r` r r r rc | s zTestCAPI.tearDownc C s* t j| j| j�}|d k r"tj|�S d S d S )N)ru Ztracemalloc_get_tracebackr rz r r )r` r r r r �
get_traceback s
zTestCAPI.get_tracebackFr c C s$ t |d�}tj| j| j| j|� |S )Nr# )r ru Ztracemalloc_trackr rz r )r` � release_gilr r r r r �track� s
zTestCAPI.trackc C s t j| j| j� d S )N)ru Ztracemalloc_untrackr rz )r` r r r �untrack� s zTestCAPI.untrackc C s6 t j� }t jd| j�}|j|g�}tdd� |jD ��S )NTc s s | ] }|j V qd S )N)r )re r| r r r � <genexpr>� s z-TestCAPI.get_traced_memory.<locals>.<genexpr>)r r� r r r� �sumr{ )r` rU Z
domain_filterr r r r� � s zTestCAPI.get_traced_memoryc C sR d}t j|� t j� d }| j||�}| j| j� t j|�� | j| j� | j� d S )Nr( r )r r_ r� r} rn r{ r r )r` r| r r r r r r � check_track� s
zTestCAPI.check_trackc C s | j d� d S )NF)r� )r` r r r �
test_track� s zTestCAPI.test_trackc C s | j d� d S )NT)r� )r` r r r �test_track_without_gil� s z TestCAPI.test_track_without_gilc C s< d}t j|� | j� | j|d�}| j| j� t j|�� d S )Nr( )r )r r_ r} rn r{ r )r` r r r r r �test_track_already_tracked� s
z#TestCAPI.test_track_already_trackedc C sj t j� | j� | j| j� � | j| j� | j� | j� | j | j� � | j| j� d� | j� | j� d S )Nr )
r r_ r} r� r{ rn r� r r~ r� )r` r r r � test_untrack� s zTestCAPI.test_untrackc C s@ t j� t j� | jt�� | j� W d Q R X | j| j� � d S )N)r r_ rb rt r� r} r� r{ )r` r r r �test_stop_track� s
zTestCAPI.test_stop_trackc C s: t j� | j� t j� | jt�� | j� W d Q R X d S )N)r r_ r} rb rt r� r~ )r` r r r �test_stop_untrack� s
zTestCAPI.test_stop_untrackNi@ )Fr )r� r� r� r1 ra rc r{ r} r~ r� r� r� r� r� r� r� r� r r r r rv l s
rv c C s t jttttt� d S )N)r Z run_unittestr\ r� r2 rU rv r r r r � test_main� s r� �__main__)"r� r� r r r� Z
unittest.mockr Ztest.support.script_helperr r r Ztestr Z threading� ImportErrorru � getsizeofr r r" rW r rY rZ r[ ZTestCaser\ r� r2 rU rt rv r� r� r r r r �<module> sH
! S ( TH
k