Your IP : 216.73.216.213
3
� f�? � @ 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 � ZG 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 d.d/� d/e j �Z&e'd0k�rze j(� dS )1� N)�TESTFN�rmtree�unlink�captured_stdout)�assert_python_ok�assert_python_failure)�Trace)�testmodc C s | j d�r| dd� } | S )z8Given a .pyc filename converts it to the appropriate .pyz.pycN� ���)�endswith)�filename� r �4/opt/alt/python36/lib64/python3.6/test/test_trace.py�
fix_ext_py s
r c C s$ t jjt jjt��d } tt�| fS )z4The .py file and module name of this file (__file__)r )�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 jd� S )Nr
)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 C s d} dd� t | �D �}|S )Nr, c S s g | ] }t |��qS r )r/ )�.0r$ r r r �
<listcomp>K s z4traced_caller_list_comprehension.<locals>.<listcomp>)r# )r( Zmylistr r r � traced_caller_list_comprehensionI s r2 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__P s zTracedClass.__init__c C s
| j | S )N)r )r4 r r r r �inst_method_linearS s z TracedClass.inst_method_linearc C s | j |�}|t||� S )N)r6 r! )r4 r r r r r �inst_method_callingV s
z TracedClass.inst_method_callingc C s |d S )Nr. r )�clsr r r r �class_method_linearZ s z TracedClass.class_method_linearc C s | d S )Nr. r )r r r r �static_method_linear^ s z TracedClass.static_method_linearN)
�__name__�
__module__� __qualname__r5 r6 r7 � classmethodr9 � staticmethodr: r r r r r3 O s
r3 c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )�TestLineCountsz/White-box testing of line-counting, via runfuncc C s2 | j tjtj� � tddddd�| _tt�| _d S )Nr
r )�count�trace�
countfuncs� countcallers) �
addCleanup�sys�settrace�gettracer �tracerr r �my_py_filename)r4 r r r �setUph s zTestLineCounts.setUpc C sf | j jtdd�}| j|d� i }tt�}x$tdd�D ]}d|| j|| f< q4W | j| j j� j|� d S )Nr. r" � r
) rI �runfuncr! � assertEqualr r# rJ �results�counts)r4 �result�expected� firstlinenor$ r r r �test_traced_func_linearm s z&TestLineCounts.test_traced_func_linearc C sl | j jtdd� tt�}| j|d fd| j|d fd| j|d fd| j|d fdi}| j| j j� j|� d S )Nr. � r
� r" � )rI rM r% r rJ rN rO rP )r4 rS rR r r r �test_traced_func_loopy s z$TestLineCounts.test_traced_func_loopc C s^ | j jtdd� tt�}| j|d fdttj�dfdttj�dfdi}| j| j j � j
|� d S )Nr. r" r
rU ) rI rM r& r rJ r r r rN rO rP )r4 rS rR r r r �test_traced_func_importing� s z)TestLineCounts.test_traced_func_importingc
C s� | j jt� tt�}tt�}| j|d fd| j|d fd| j|d fd| j|d fd| j|d fd| j|d fdi}| j| j j� j|� d S )Nr
r. � rU r, ) rI rM r- r r+ rJ rN rO rP )r4 �firstlineno_callingZfirstlineno_genrR r r r �test_trace_func_generator� s z(TestLineCounts.test_trace_func_generatorc C sp | j jt� tt�}tt�}| j|d fd| j|d fd| j|d fd| j|d fdi}| j| j j� j|� d S )Nr
r. � rU r, ) rI rM r2 r r/ rJ rN rO rP )r4 r[ Zfirstlineno_calledrR r r r � test_trace_list_comprehension� s z,TestLineCounts.test_trace_list_comprehensionc C sl xfdD ]^}t ddddd�}td�}t||�}|j|d� t|�}| j|d fdi}| j|j� j|� qW d S )Nr6 r
r )rA rB rC rD � � )r6 ) r r3 �getattrrM r rJ rN rO rP )r4 ZmethnamerI Z
traced_obj�methodrS rR r r r �test_linear_methods� s
z"TestLineCounts.test_linear_methodsN) r; r< r= �__doc__rK rT rX rY r\ r^ rc r r r r r@ f s r@ c @ s e Zd ZdZdd� Zdd� ZdS )�TestRunExecCountsz8A simple sanity test of line-counting, via runctx (exec)c C s t t�| _| jtjtj� � d S )N)r r rJ rE rF rG rH )r4 r r r rK � s
zTestRunExecCounts.setUpc C s� t ddddd�| _d}t|td�}| jj|t� t� � tt�}| j |d fd| j |d fd| j |d fd | j |d
fdi}x,|j
� D ] }| j | jj � j
| || � q�W d S ) Nr
r )rA rB rC rD ztraced_func_loop(2, 5)�execr. rV rU r" rW )r rI �compiler Zrunctx�globals�varsr r% rJ �keysrN rO rP )r4 �coderS rR r( r r r �test_exec_counts� s z"TestRunExecCounts.test_exec_countsN)r; r< r= rd rK rl r r r r re � s re c @ sR e Zd ZdZdd� Zdd� Zdd� Zdd � Zej e
e d
�o@e j � d �d d
� �Z
dS )� TestFuncsz"White-box testing of funcs tracingc C s8 | j tjtj� � tdddd�| _t� | _tj� | _d S )Nr r
)rA rB rC ) rE rF rG rH r rI r �filemod�_saved_tracefunc)r4 r r r rK � s zTestFuncs.setUpc C s | j d k rtj| j � d S )N)ro rF rG )r4 r r r �tearDown� s
zTestFuncs.tearDownc C s> | j jtd� | jd d| jd di}| j| j j� j|� d S )Nr
r' r! )r' )r! )rI rM r' rn rN rO � calledfuncs)r4 rR r r r �test_simple_caller� s
z TestFuncs.test_simple_callerc C sb | j jtd� | jd d| jd d| jd
d| jd dttj�ddfdi}| j| j j� j |� d S ) Nr
r' r! r) r& r r )r' )r! )r) )r& )
rI rM r) rn r r r rN rO rq )r4 rR r r r �test_loop_caller_importing� s
z$TestFuncs.test_loop_caller_importingrH z3pre-existing trace function throws off measurementsc C sR t d�}| jj|jd� | jd d| jd d| jd di}| j| jj� j|� d S ) Nr` r
� TracedClass.inst_method_calling� TracedClass.inst_method_linearr! )rt )ru )r! )r3 rI rM r7 rn rN rO rq )r4 �objrR r r r �test_inst_method_calling� s
z"TestFuncs.test_inst_method_callingN)r; r< r= rd rK rp rr rs �unittest�skipIf�hasattrrF rH rw r r r r rm � s rm c @ s: e Zd ZdZdd� Zejeed�o(ej � d�dd� �Z
dS ) � TestCallersz$White-box testing of callers tracingc C s. | j tjtj� � tdddd�| _t� | _d S )Nr r
)rA rB rD )rE rF rG rH r rI r rn )r4 r r r rK
s zTestCallers.setUprH z3pre-existing trace function throws off measurementsc C s� | j jtd� tjjtj�d d ddf| jd fd| jd
| jd fd| jd | jd fd| jd | jd fd| jd t t
j�d
d ffdi}| j | j j � j
|� d S )Nr
r z.pyrB z
Trace.runfuncr) r' r! r& r r )r) )r' )r! )r) )r' )r) )r& )r& )rI rM r) r r r rB r rn r r rN rO Zcallers)r4 rR r r r rs s z&TestCallers.test_loop_caller_importingN) r; r< r= rd rK rx ry rz rF rH rs r r r r r{ s r{ c @ s> e Zd Zdd� Zdd� Zddd�Zdd � Zd
d � Zd d
� ZdS )� TestCoveragec C s | j tjtj� � d S )N)rE rF rG rH )r4 r r r rK ( s zTestCoverage.setUpc C s t t� tt� d S )N)r r r )r4 r r r rp + s zTestCoverage.tearDown�_import test.support, test.test_pprint;test.support.run_unittest(test.test_pprint.QueryTestCase)c C s& |j |� |j� }|jddtd� d S )NT)Z show_missingZsummaryZcoverdir)ZrunrO Z
write_resultsr )r4 rI �cmd�rr r r � _coverage/ s
zTestCoverage._coveragec
C sp t jddd�}t� �}| j|� W d Q R X |j� }| jd|� | jd|� tjt�}| jd|� | jd|� d S )Nr r
)rB rA z pprint.pyzcase.pyz pprint.coverzunittest.case.cover) rB r r r� �getvalue�assertInr �listdirr )r4 rI �stdout�filesr r r �
test_coverage6 s
zTestCoverage.test_coveragec
C st t jjt jjt j��}tjtjtj |gddd�}t
� �}| j |� W d Q R X t jj t
�rpt jt
�}| j|dg� d S )Nr r
)Z
ignoredirsrB rA z_importlib.cover)r r �normpath�dirnamer rB r rF � base_prefix�base_exec_prefixr r� �existsr r� rN )r4 ZlibpathrI r� r� r r r �test_coverage_ignoreA s
z!TestCoverage.test_coverage_ignorec
C s� t jddd�}d}|tjkr$tj|= d}t� �}| j||� W d Q R X |jd� |j� i }x>|D ]6}|j� d d� \}}} t |�t |d d � �f|| < qbW t j
tj| j �}| j ||� | j
|| d
� d S ) Nr r
)rB rA ztest.tracedmodules.testmodz<import test.tracedmodules.testmod as t;t.func(0); t.func2();rU r" �d r )r" r� )rB r rF �modulesr r� �seek�readline�split�intZ _fullmodnamer r� rN )
r4 rI r r~ r� Zcoverage�line�linesZcov�moduler r r �test_issue9936M s
zTestCoverage.test_issue9936N)r} ) r; r< r= rK rp r� r� r� r� r r r r r| ' s
r| c @ s e Zd Zdd� ZdS )� Test_Ignorec C s� t jj}tjddg|dd�g�}| j|jdd�� | j|jdd�� | j|jdd �� | j|j|ddd
�d �� | j|j|dd �d
�� | j|j|dd
�d �� d S )Nr zy.zZfooZbarzx.pyzxy.pyZxyzy.pyr zbaz.pyZbazzz.py�z)r r �joinrB Z_Ignore�
assertTrue�namesZ assertFalse)r4 Zjn�ignorer r r � test_ignoredf s zTest_Ignore.test_ignoredN)r; r< r= r� r r r r r� e s r� c @ s4 e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd S ) � TestCoverageCommandLineOutputztmp.pyz tmp.coverc
C s, t | jd��}|jtjd�� W d Q R X d S )N�wzc x = 42
if []:
print('unreachable')
)�open�codefile�write�textwrap�dedent)r4 �fr r r rK w s z#TestCoverageCommandLineOutput.setUpc C s t | j� t | j� d S )N)r r� � coverfile)r4 r r r rp s
z&TestCoverageCommandLineOutput.tearDownc
C s\ dj � | jg }t|� \}}}| jtjj| j�� t| j��}| j |j
� d� W d Q R X d S )Nz-m trace --countz< 1: x = 42
1: if []:
print('unreachable')
) r� r� r r� r r r� r� r� rN �read)r4 �argv�statusr� �stderrr� r r r �%test_cover_files_written_no_highlight� s
zCTestCoverageCommandLineOutput.test_cover_files_written_no_highlightc C sb dj � | jg }t|� \}}}| jtjj| j�� t| j�� }| j |j
� t j 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� rN r� r� r� )r4 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� rK rp r� r� r r r r r� r s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestCommandLinec C s8 d}x.|D ]&^}}t d|�� �^ }}| j||� q
W d S )N�3 filename is missing: required with the main options�-l�-T�8 cannot specify both --listfuncs and (--trace or --count)�-lc�> argument -R/--no-report: not allowed with argument -r/--report�-rR�L must specify one of --trace, --count, --report, --listfuncs, or --trackcalls�-g� -r/--report requires -f/--file�-r�3 --summary can only be used with --count or --report�-sT� unrecognized arguments: -y�-y�-mrB �r� r� r� �r� r� �r� r� �r� r� �r� r� �r� r� �r� r� )r� r� r� r� r� r� r� )r� rB )r r� )r4 Z_errors�message�args�_r� r r r �
test_failures� s z TestCommandLine.test_failuresc C sP t td��<}| jtt� |jd� tdddt�\}}}| jd|� W d Q R X d S )Nr� za = 1
z-mrB z-ls functions called:)r� r rE r r� r r� )r4 �fdr� r� r� r r r �test_listfuncs_flag_success� s
z+TestCommandLine.test_listfuncs_flag_successc C sl t td��&}| jtt� |jd� |jd� W d Q R X tt�\}}}tdddt�\}}}| j|j� |� d S )Nr� z import sys
zprint(type(sys.argv))
z-mrB z-l)r� r rE r r� r r� �strip)r4 r� r� Z
direct_stdoutr� Z trace_stdoutr r r �test_sys_argv_list� s
z"TestCommandLine.test_sys_argv_listN)r; r< r= r� r� r� r r r r r� � s
r� �__main__))r rF Z test.supportr r r r Ztest.support.script_helperr r r� rx rB r Ztest.tracedmodulesr r r r r! r% r& r' r) r+ r- r/ r2 �objectr3 ZTestCaser@ re rm r{ r| r� r� r� r; �mainr r r r �<module> s>
Z / >
'