File: //usr/local/lib/python3.7/test/__pycache__/test_trace.cpython-37.opt-2.pyc
B
��g�E � @ s� d dl Z d dlZd dlmZmZmZmZ d dlmZm Z d dl
Z
d dl Z d dl Z d dl m
Z
d dlmZ dd� Zdd � Zd
d � Zd d
� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Z G d d!� d!e �Z G d"d#� d#e j �Z G d$d%� d%e j �Z!G d&d'� d'e j �Z"G d(d)� d)e j �Z#G d*d+� d+e j �Z$G d,d-� d-e j �Z%G d.d/� d/e j �Z&G d0d1� d1e j �Z'e(d2k�r�e �)� dS )3� N)�TESTFN�rmtree�unlink�captured_stdout)�assert_python_ok�assert_python_failure)�Trace)�testmodc C s | � d�r| d d� } | S )Nz.pyc���)�endswith)�filename� r
�+/usr/local/lib/python3.7/test/test_trace.py�
fix_ext_py s
r c C s$ t j�t j�t��d } tt�| fS )Nr )�os�path�splitext�basename�__file__r )�modnamer
r
r �my_file_and_modname s r c C s | j jS )N)�__code__�co_firstlineno)�funcr
r
r �get_firstlineno s r c C s | }|}|| }|S )Nr
)�x�y�a�b�cr
r
r �traced_func_linear$ s r c C s" | }xt d�D ] }||7 }qW |S )N� )�range)r r r �ir
r
r �traced_func_loop* s r$ c C s | | t �d� S )N� )r r )r r r
r
r �traced_func_importing0 s r& c C s t | | �}|| S )N)r )r r r
r
r �traced_func_simple_caller3 s
r' c C s t | �}|t|| �7 }|S )N)r' r& )r �kr
r
r � traced_func_importing_caller7 s r) c c s$ d}xt | �D ]}|| V qW d S )Nr! )r" )�numr r# r
r
r �traced_func_generator< s r+ c C s" d} xt d�D ] }| |7 } qW d S )Nr �
)r+ )r( r# r
r
r � traced_func_calling_generatorA s r- c C s | d S )N� r
)r* r
r
r �traced_doublerF s r/ c O s | |fS )Nr
)�args�kwargsr
r
r �traced_capturerI s r2 c C s d} dd� t | �D �}|S )Nr, c S s g | ] }t |��qS r
)r/ )�.0r# r
r
r �
<listcomp>N s z4traced_caller_list_comprehension.<locals>.<listcomp>)r" )r( Zmylistr
r
r � traced_caller_list_comprehensionL s r5 c @ s< e Zd Zdd� Zdd� Zdd� Zedd� �Zed d
� �Z d S ) � TracedClassc C s
|| _ d S )N)r )�selfr r
r
r �__init__S s zTracedClass.__init__c C s
| j | S )N)r )r7 r r
r
r �inst_method_linearV s z TracedClass.inst_method_linearc C s | � |�}|t||� S )N)r9 r )r7 r r r
r
r �inst_method_callingY s
z TracedClass.inst_method_callingc C s |d S )Nr. r
)�clsr r
r
r �class_method_linear] s z TracedClass.class_method_linearc C s | d S )Nr. r
)r r
r
r �static_method_lineara s z TracedClass.static_method_linearN)
�__name__�
__module__� __qualname__r8 r9 r: � classmethodr< � staticmethodr= r
r
r
r r6 R s
r6 c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�TestLineCountsc C s2 | � tjt�� � tddddd�| _tt�| _d S )Nr% r )�count�trace�
countfuncs� countcallers) �
addCleanup�sys�settrace�gettracer �tracerr r �my_py_filename)r7 r
r
r �setUpk s zTestLineCounts.setUpc C sf | j �tdd�}| �|d� i }tt�}x$tdd�D ]}d|| j|| f< q4W | �| j �� j|� d S )Nr. r! � r% ) rL �runfuncr � assertEqualr r" rM �results�counts)r7 �result�expected� firstlinenor# r
r
r �test_traced_func_linearp s z&TestLineCounts.test_traced_func_linearc C sl | j �tdd� tt�}| j|d fd| j|d fd| j|d fd| j|d fdi}| �| j �� j|� d S )Nr. � r% � r! � )rL rP r$ r rM rQ rR rS )r7 rV rU r
r
r �test_traced_func_loop| s z$TestLineCounts.test_traced_func_loopc C s^ | j �tdd� tt�}| j|d fdttj�dfdttj�dfdi}| �| j � � j
|� d S )Nr. r! r% rX ) rL rP r& r rM r r r rQ rR rS )r7 rV rU r
r
r �test_traced_func_importing� s z)TestLineCounts.test_traced_func_importingc
C s� | j �t� tt�}tt�}| j|d fd| j|d fd| j|d fd| j|d fd| j|d fd| j|d fdi}| �| j �� j|� d S )Nr% r. � rX r, ) rL rP r- r r+ rM rQ rR rS )r7 �firstlineno_callingZfirstlineno_genrU r
r
r �test_trace_func_generator� s z(TestLineCounts.test_trace_func_generatorc C sp | j �t� tt�}tt�}| j|d fd| j|d fd| j|d fd| j|d fdi}| �| j �� j|� d S )Nr% r. � rX r, ) rL rP r5 r r/ rM rQ rR rS )r7 r^ Zfirstlineno_calledrU r
r
r � test_trace_list_comprehension� s z,TestLineCounts.test_trace_list_comprehensionc C sl xfdD ]^}t ddddd�}td�}t||�}|�|d� t|�}| j|d fdi}| �|�� j|� qW d S )N)r9 r% r )rD rE rF rG � � ) r r6 �getattrrP r rM rQ rR rS )r7 ZmethnamerL Z
traced_obj�methodrV rU r
r
r �test_linear_methods� s
z"TestLineCounts.test_linear_methodsN)
r> r? r@ rN rW r[ r\ r_ ra rf r
r
r
r rC i s rC c @ s e Zd Zdd� Zdd� ZdS )�TestRunExecCountsc C s t t�| _| �tjt�� � d S )N)r r rM rH rI rJ rK )r7 r
r
r rN � s
zTestRunExecCounts.setUpc C s� t ddddd�| _d}t|td�}| j�|t� t� � tt�}| j |d fd| j |d fd| j |d fd | j |d
fdi}x,|�
� D ] }| � | j� � j
| || � q�W d S ) Nr% r )rD rE rF rG ztraced_func_loop(2, 5)�execr. rY rX r! rZ )r rL �compiler Zrunctx�globals�varsr r$ rM �keysrQ rR rS )r7 �coderV rU r( r
r
r �test_exec_counts� s z"TestRunExecCounts.test_exec_countsN)r> r? r@ rN rn r
r
r
r rg � s rg c @ sV e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Ze� e
e d �oDe � � d �d
d� �Z
dS )� TestFuncsc C s8 | � tjt�� � tdddd�| _t� | _t�� | _d S )Nr r% )rD rE rF ) rH rI rJ rK r rL r �filemod�_saved_tracefunc)r7 r
r
r rN � s zTestFuncs.setUpc C s | j d k rt�| j � d S )N)rq rI rJ )r7 r
r
r �tearDown� s
zTestFuncs.tearDownc C s> | j �td� | jd d| jd di}| �| j �� j|� d S )Nr% )r' )r )rL rP r' rp rQ rR � calledfuncs)r7 rU r
r
r �test_simple_caller� s
z TestFuncs.test_simple_callerc C st | j jtddddd�}| �|dddd�f� | j jtdd�}| �|dd dif� | �t�� | j �� W d Q R X d S )
Nr% r. rX rZ )r7 r )r% r. )r �argr
ru )rL rP r2 rQ Z assertRaises� TypeError)r7 Zresr
r
r �test_arg_errors� s zTestFuncs.test_arg_errorsc C sb | j �td� | jd d| jd d| jd d| jd dttj�ddfdi}| �| j �� j |� d S )Nr% )r' )r )r) )r& r r )
rL rP r) rp r r r rQ rR rs )r7 rU r
r
r �test_loop_caller_importing� s
z$TestFuncs.test_loop_caller_importingrK z3pre-existing trace function throws off measurementsc C sR t d�}| j�|jd� | jd d| jd d| jd di}| �| j�� j|� d S )Nrc r% )z TracedClass.inst_method_calling)z TracedClass.inst_method_linear)r )r6 rL rP r: rp rQ rR rs )r7 �objrU r
r
r �test_inst_method_calling s
z"TestFuncs.test_inst_method_callingN)r> r? r@ rN rr rt rw rx �unittest�skipIf�hasattrrI rK rz r
r
r
r ro � s ro c @ s6 e Zd Zdd� Ze�eed�o$e�� d�dd� �Z dS )� TestCallersc C s. | � tjt�� � tdddd�| _t� | _d S )Nr r% )rD rE rG )rH rI rJ rK r rL r rp )r7 r
r
r rN s zTestCallers.setUprK z3pre-existing trace function throws off measurementsc C s� | j �td� tj�tj�d d ddf| jd fd| jd | jd fd| jd | jd fd| jd | jd fd| jd t t
j�d
d ffdi}| � | j � � j
|� d S ) Nr% r z.pyrE z
Trace.runfunc)r) )r' )r )r& r r )rL rP r) r r r rE r rp r r rQ rR Zcallers)r7 rU r
r
r rx s z&TestCallers.test_loop_caller_importingN)
r> r? r@ rN r{ r| r} rI rK rx r
r
r
r r~ s r~ c @ s> e Zd Zdd� Zdd� Zddd�Zdd � Zd
d � Zd d
� ZdS )� TestCoveragec C s | � tjt�� � d S )N)rH rI rJ rK )r7 r
r
r rN 3 s zTestCoverage.setUpc C s t t� tt� d S )N)r r r )r7 r
r
r rr 6 s zTestCoverage.tearDown�_import test.support, test.test_pprint;test.support.run_unittest(test.test_pprint.QueryTestCase)c C s& |� |� |�� }|jddtd� d S )NT)Z show_missingZsummaryZcoverdir)�runrR Z
write_resultsr )r7 rL �cmd�rr
r
r � _coverage: s
zTestCoverage._coveragec C sp t jddd�}t� �}| �|� W d Q R X |�� }| �d|� | �d|� t�t�}| �d|� | �d|� d S )Nr r% )rE rD z pprint.pyzcase.pyz pprint.coverzunittest.case.cover) rE r r r� �getvalue�assertInr �listdirr )r7 rL �stdout�filesr
r
r �
test_coverageA s
zTestCoverage.test_coveragec C st t j�t j�t j��}tjtjtj |gddd�}t
� �}| � |� W d Q R X t j� t
�rpt �t
�}| �|dg� d S )Nr r% )Z
ignoredirsrE rD z_importlib.cover)r r �normpath�dirnamer rE r rI � base_prefix�base_exec_prefixr r� �existsr r� rQ )r7 ZlibpathrL r� r� r
r
r �test_coverage_ignoreL s
z!TestCoverage.test_coverage_ignorec
C s� t jddd�}d}|tjkr$tj|= d}t� �}| �||� W d Q R X |�d� |�� i }x>|D ]6}|�� d d� \}}} t |�t |d d� �f|| < qbW t �
tj| j �}| � ||� | �
|| d� d S ) Nr r% )rE rD ztest.tracedmodules.testmodz<import test.tracedmodules.testmod as t;t.func(0); t.func2();rX r
)r! �d )rE r rI �modulesr r� �seek�readline�split�intZ _fullmodnamer r� rQ )
r7 rL r r� r� Zcoverage�line�linesZcov�moduler
r
r �test_issue9936X s
zTestCoverage.test_issue9936N)r� ) r> r? r@ rN rr r� r� r� r� r
r
r
r r 2 s
r c @ s e Zd Zdd� ZdS )� Test_Ignorec C s� t jj}t�ddg|dd�g�}| �|�dd�� | �|�dd�� | �|�dd �� | �|�|ddd
�d �� | �|�|dd �d
�� | �|�|dd
�d �� d S )Nr zy.zZfooZbarzx.pyzxy.pyZxyzy.pyr zbaz.pyZbazzz.py�z)r r �joinrE Z_Ignore�
assertTrue�names� assertFalse)r7 Zjn�ignorer
r
r � test_ignoredq s zTest_Ignore.test_ignoredN)r> r? r@ r� r
r
r
r r� p s r� c @ s4 e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd S ) � TestCoverageCommandLineOutputztmp.pyz tmp.coverc C s, t | jd��}|�t�d�� W d Q R X d S )N�wzc x = 42
if []:
print('unreachable')
)�open�codefile�write�textwrap�dedent)r7 �fr
r
r rN � s z#TestCoverageCommandLineOutput.setUpc C s t | j� t | j� d S )N)r r� � coverfile)r7 r
r
r rr � s
z&TestCoverageCommandLineOutput.tearDownc C s� t j�t j�tj��}t j�|d�}t|� d�� | j g }t
|� \}}}| � |d� | � t j�
|�� | �t j�
| j�� t| j��}| � |�� d� W d Q R X d S )Nz trace.coverz-m trace --count� z< 1: x = 42
1: if []:
print('unreachable')
)r r r� �abspathrE r r� r r� r� r rQ r� r� r� r� r� �read)r7 ZtracedirZtracecoverpath�argv�statusr� �stderrr� r
r
r �%test_cover_files_written_no_highlight� s
zCTestCoverageCommandLineOutput.test_cover_files_written_no_highlightc C sb d� � | jg }t|� \}}}| �tj�| j�� t| j�� }| � |�
� t � d�� W d Q R X d S )Nz-m trace --count --missingzx 1: x = 42
1: if []:
>>>>>> print('unreachable')
)
r� r� r r� r r r� r� r� rQ r� r� r� )r7 r� r� r� r� r� r
r
r �'test_cover_files_written_with_highlight� s zETestCoverageCommandLineOutput.test_cover_files_written_with_highlightN) r> r? r@ r� r� rN rr r� r� r
r
r
r r� } s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�TestCommandLinec C s8 d}x.|D ]&^}}t d|�� �^ }}| �||� q
W d S )N))s3 filename is missing: required with the main optionsz-lz-T)s8 cannot specify both --listfuncs and (--trace or --count)z-lc)s> argument -R/--no-report: not allowed with argument -r/--reportz-rR)sL must specify one of --trace, --count, --report, --listfuncs, or --trackcallsz-g)s -r/--report requires -f/--filez-r)s3 --summary can only be used with --count or --reportz-sT)s unrecognized arguments: -yz-y�-mrE )r� rE )r r� )r7 �_errors�messager0 �_r� r
r
r �
test_failures� s z TestCommandLine.test_failuresc C sP t td��<}| �tt� |�d� tdddt�\}}}| �d|� W d Q R X d S )Nr� za = 1
z-mrE z-ls functions called:)r� r rH r r� r r� )r7 �fdr� r� r� r
r
r �test_listfuncs_flag_success� s
z+TestCommandLine.test_listfuncs_flag_successc C sl t td��&}| �tt� |�d� |�d� W d Q R X tt�\}}}tdddt�\}}}| �|�� |� d S )Nr� z import sys
zprint(type(sys.argv))
z-mrE z-l)r� r rH r r� r r� �strip)r7 r� r� Z
direct_stdoutr� Z trace_stdoutr
r
r �test_sys_argv_list� s
z"TestCommandLine.test_sys_argv_listc C s� t � d�}t � d�}t|d��.}| �t|� | �t|� |�t�d�� W d Q R X tddd|�\}}}|�� }| � |d� | �
d |� | �
d
t � d |� d �|� d S )
Nz.pyz.coverr� z� x = 1
y = 2
def f():
return x + y
for i in range(10):
f()
z-mrE z-csr z lines cov% module (path)z 6 100% z (�)) r r� rH r r� r� r� r �decoderQ r� )r7 r Z
coverfilenamer� r� r� r� r
r
r �test_count_and_summary� s
z&TestCommandLine.test_count_and_summaryN)r> r? r@ r� r� r� r� r
r
r
r r� � s
r� �__main__)*r rI Z test.supportr r r r Ztest.support.script_helperr r r� r{ rE r Ztest.tracedmodulesr r r r r r$ r& r' r) r+ r- r/ r2 r5 �objectr6 ZTestCaserC rg ro r~ r r� r� r� r> �mainr
r
r
r �<module> s@
Z 7 >
/6