File: //usr/local/lib/python3.10/test/__pycache__/test_trace.cpython-310.pyc
o
�i�Q � @ s� d dl Z d dlmZ d dlZd dlmZ d dlmZmZm Z d dl
m Z m Z d dl
Z
d dlZd dlZd dlmZ d dlmZ dd � Zd
d � Zd d
� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Z d d � Z d d � Z d d!� Z d"d#� Z G d$d%� d%e!�Z"G d&d'� d'ej#�Z$G d(d)� d)ej#�Z%G d*d+� d+ej#�Z&G d,d-� d-ej#�Z'G d.d/� d/ej#�Z(G d0d1� d1ej#�Z)G d2d3� d3ej#�Z*G d4d5� d5ej#�Z+e,d6kr�e�-� dS dS )7� N)�dump)�captured_stdout)�TESTFN�rmtree�unlink��assert_python_ok�assert_python_failure)�Trace)�testmodc C s | � d�r | dd� } | S )z8Given a .pyc filename converts it to the appropriate .pyz.pycN���)�endswith)�filename� r �,/usr/local/lib/python3.10/test/test_trace.py�
fix_ext_py s
r c C s$ t j�t j�t��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 � r c C s | }|}|| }|S r r )�x�y�a�b�cr r r �traced_func_linear& s r% c C s | }t d�D ]}||7 }q|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_importing2 s r- c C s t | | �}|| S r �r% )r r$ r r r �traced_func_simple_caller5 s
r/ c C s t | �}|t|| �7 }|S r )r/ r- )r �kr r r � traced_func_importing_caller9 s r1 c c s" � d}t | �D ]}|| V qd S r& r( )�numr$ r* r r r �traced_func_generator> s
� �r3 c C s d} t d�D ]}| |7 } qd S )Nr �
)r3 )r0 r* r r r � traced_func_calling_generatorC s
�r5 c C � | d S �N� r )r2 r r r �traced_doublerH r r9 c O s | |fS r r )�args�kwargsr r r �traced_capturerK r r< c C s d} dd� t | �D �}|S )Nr4 c S s g | ]}t |��qS r )r9 )�.0r* r r r �
<listcomp>P s z4traced_caller_list_comprehension.<locals>.<listcomp>r( )r0 Zmylistr r r � traced_caller_list_comprehensionN s r? c C s, dd� } dd� }| |� dd� ��}|� d S )Nc S � | S r r ��fr r r �
decorator1T � z-traced_decorated_function.<locals>.decorator1c S s dd� } | S )Nc S r@ r r rA r r r �
decorator2W rD zGtraced_decorated_function.<locals>.decorator_fabric.<locals>.decorator2r �rE r r r �decorator_fabricV s z3traced_decorated_function.<locals>.decorator_fabricc S s d S r r r r r r r Z s z'traced_decorated_function.<locals>.funcr )rC rG r r r r �traced_decorated_functionS s
rH c @ s< e Zd Zdd� Zdd� Zdd� Zedd� �Zed d
� �Z d S ) � TracedClassc C s
|| _ d S r �r" )�selfr r r r �__init__b �
zTracedClass.__init__c C s
| j | S r rJ )rK r! r r r �inst_method_lineare rM � TracedClass.inst_method_linearc C s | � |�}|t||� S r )rN r% )rK r r$ r r r �inst_method_callingh �
� TracedClass.inst_method_callingc C s |d S r7 r )�clsr! r r r �class_method_linearl � z TracedClass.class_method_linearc C r6 r7 r )r! r r r �static_method_linearp rU z TracedClass.static_method_linearN)
�__name__�
__module__� __qualname__rL rN rP � classmethodrT � staticmethodrV r r r r rI a s
rI c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dS )�TestLineCountsz/White-box testing of line-counting, via runfuncc C s2 | � tjt�� � tddddd�| _tt�| _d S )Nr, r ��count�trace�
countfuncs� countcallers) �
addCleanup�sys�settrace�gettracer
�tracerr r �my_py_filename�rK r r r �setUpz s zTestLineCounts.setUpc C sb | j �tdd�}| �|d� i }tt�}tdd�D ] }d|| j|| f< q| �| j �� j|� d S )Nr8 r' � r, ) rf �runfuncr% � assertEqualr r) rg �results�counts)rK �result�expected� firstlinenor* r r r �test_traced_func_linear 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 )Nr8 � r, � r' � )rf rk r+ r rg rl rm rn �rK rq rp 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 )Nr8 r' r, rs ) rf rk r- r rg r r r rl rm rn rv 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, r8 � rs r4 ) rf rk r5 r r3 rg rl rm rn )rK �firstlineno_callingZfirstlineno_genrp 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, r8 � rs r4 ) rf rk r? r r9 rg rl rm rn )rK rz Zfirstlineno_calledrp r r r � test_trace_list_comprehension� s �z,TestLineCounts.test_trace_list_comprehensionc C s� | j �t� tt�}| j|d fd| j|d fd| j|d fd| j|d fd| j|d fd| j|d fd| j|d fd| j|d fd| j|d fd| j|d
fd| j|d fdi }| �| j �� j|� d S ) Nr, r8 rs ru r' rt rj � � r4 ry )rf rk rH r rg rl rm rn rv r r r � test_traced_decorated_function� s �
z-TestLineCounts.test_traced_decorated_functionc C sh dD ]/}t ddddd�}td�}t||�}|�|d� t|�}| j|d fdi}| �|�� j|� qd S )N)rN r, r r] � � ) r
rI �getattrrk r rg rl rm rn )rK Zmethnamerf Z
traced_obj�methodrq rp r r r �test_linear_methods� s
��z"TestLineCounts.test_linear_methodsN) rW rX rY �__doc__ri rr rw rx r{ r} r� r� r r r r r\ x s r\ c @ s e Zd ZdZdd� Zdd� ZdS )�TestRunExecCountsz8A simple sanity test of line-counting, via runctx (exec)c C s t t�| _| �tjt�� � d S r )r r rg rb rc rd re rh r r r ri � 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}|�
� D ]}| � | j� � j
| || � qAd S ) Nr, r r] ztraced_func_loop(2, 5)�execr8 rt rs r' ru )r
rf �compiler Zrunctx�globals�varsr r+ rg �keysrl rm rn )rK �coderq rp r0 r r r �test_exec_counts� s � �z"TestRunExecCounts.test_exec_countsN)rW rX rY r� ri r� r r r r r� � s r� c @ sb e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Ze �
e e d �o$e �
� d
�dd� �Zdd� ZdS )� TestFuncsz"White-box testing of funcs tracingc C s8 | � tjt�� � tdddd�| _t� | _t�� | _d S )Nr r, )r^ r_ r` ) rb rc rd re r
rf r �filemod�_saved_tracefuncrh r r r ri s zTestFuncs.setUpc C s | j d ur
t�| j � d S d S r )r� rc rd rh r r r �tearDown s
�zTestFuncs.tearDownc C s> | j �td� | jd d| jd di}| �| j �� j|� d S )Nr, �r/ r. )rf rk r/ r� rl rm � calledfuncs�rK rp r r r �test_simple_caller
s
�z TestFuncs.test_simple_callerc C s� | j jtddddd�}| �|dddd�f� | �t�� | j jtdd� W d � n1 s.w Y | �t�� | j �� W d � d S 1 sIw Y d S )Nr, r8 rs ru )rK r )r, r8 )r �arg)rf rk r< rl Z assertRaises� TypeError)rK �resr 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. �r1 �r- r r )
rf rk r1 r� r r r rl rm r� r� r r r �test_loop_caller_importing s
�z$TestFuncs.test_loop_caller_importingre �3pre-existing trace function throws off measurementsc C sR t d�}| j�|jd� | jd d| jd d| jd di}| �| j�� j|� d S )Nr� r, )rR )rO r. )rI rf rk rP r� rl rm r� )rK �objrp r r r �test_inst_method_calling* s
�z"TestFuncs.test_inst_method_callingc
C sZ | j �t� | jd d| jd d| jd d| jd d| jd di}| �| j �� j|� d S )N)rH r, )rG rF )rC r )rf rk rH r� rl rm r� r� r r r r� 7 s
�z(TestFuncs.test_traced_decorated_functionN)rW rX rY r� ri r� r� r� r� �unittest�skipIf�hasattrrc re r� r� r r r r r� s �
r� c @ s: e Zd ZdZdd� Ze�eed�oe� � d�dd� �Z
dS ) � TestCallersz$White-box testing of callers tracingc C s. | � tjt�� � tdddd�| _t� | _d S )Nr r, )r^ r_ ra )rb rc rd re r
rf r r� rh r r r ri F s zTestCallers.setUpre r� c 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 �.pyr_ z
Trace.runfuncr� r� r. r� r r )rf rk r1 r r r r_ r r� r r rl rm �callersr� r r r r� K s. ������ z&TestCallers.test_loop_caller_importingN) rW rX rY r� ri r� r� r� rc re r� r r r r r� D s �r� c @ sH e Zd Zdd� Zdd� Z ddd�Zdd � Zd
d � Zd d
� Zdd� Z dS )� TestCoveragec C s | � tjt�� � d S r )rb rc rd re rh r r r ri a s zTestCoverage.setUpc C s t t� tt� d S r )r r r rh r r r r� d 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)�runrm Z
write_resultsr )rK rf �cmd�rr r r � _coverageh s
zTestCoverage._coveragec C s� t jddd�}t� �
}| �|� W d � n1 sw Y |�� }| �d|� | �d|� t�t�}| �d|� | �d|� d S )Nr r, �r_ r^ z pprint.pyzcase.pyz pprint.coverzunittest.case.cover) r_ r
r r� �getvalue�assertInr �listdirr )rK rf �stdout�filesr r r �
test_coverageo s �
zTestCoverage.test_coveragec C s� t j�t j�t j��}tjtjtj |gddd�}t
� �
}| � |� W d � n1 s+w Y t j� t
�rDt �t
�}| �|dg� d S d S )Nr r, )Z
ignoredirsr_ r^ z_importlib.cover)r r �normpath�dirnamer r_ r
rc � base_prefix�base_exec_prefixr r� �existsr r� rl )rK Zlibpathrf r� r� r r r �test_coverage_ignorez s �� �
�z!TestCoverage.test_coverage_ignorec
C s� t jddd�}d}|tjv rtj|= d}t� �}| �||� W d � n1 s(w Y |�d� |�� i }|D ]}|�� d d� \}}} t |�t |d d� �f|| < q:t �
tj| j �}| � ||� | �
|| d� d S ) Nr r, r� ztest.tracedmodules.testmodz<import test.tracedmodules.testmod as t;t.func(0); t.func2();rs r )r' �d )r_ r
rc �modulesr r� �seek�readline�split�intZ _fullmodnamer r� rl )
rK rf r r� r� Zcoverage�line�linesZcov�moduler r r �test_issue9936� s"
�
zTestCoverage.test_issue9936c C s| t d }t|d��}ti i ddif|dd� W d � n1 s w Y | �t|� t�i i |i �}| �|jddi� d S )Nz-infile�wbZcallerr, )�protocol) r �openr rb r r_ ZCoverageResultsrl r� )rK ZinfilerB rm r r r �test_coverageresults_update� s � z(TestCoverage.test_coverageresults_updateN)r� )
rW rX rY ri r� r� r� r� r� r� r r r r r� ` 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 �joinr_ Z_Ignore�
assertTrue�names� assertFalse)rK Zjn�ignorer r r � test_ignored� s zTest_Ignore.test_ignoredN)rW rX rY r� r r r r r� � s r� c @ s4 e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd S ) � TestCoverageCommandLineOutputztmp.pyz tmp.coverc C sF t | jddd��}|�t�d�� W d � d S 1 s w Y d S )N�w� iso-8859-15��encodingu� # coding: iso-8859-15
x = 'spœm'
if []:
print('unreachable')
)r� �codefile�write�textwrap�dedent)rK rB r r r ri � s "�z#TestCoverageCommandLineOutput.setUpc C s t | j� t | j� d S r )r r� � coverfilerh r r r r� � rQ 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| jdd��}| � |�� d� W d � d S 1 sYw Y d S )Nz trace.coverz-m trace --count� r� r� u^ # coding: iso-8859-15
1: x = 'spœm'
1: if []:
print('unreachable')
)r r r� �abspathr_ r r� r r� r� r rl r� r� r� r� r� �read)rK ZtracedirZtracecoverpath�argv�statusr� �stderrrB r r r �%test_cover_files_written_no_highlight� s
�"�zCTestCoverageCommandLineOutput.test_cover_files_written_no_highlightc C s| d� � | jg }t|� \}}}| �tj�| j�� t| jdd��}| � |�
� t � d�� W d � d S 1 s7w Y d S )Nz-m trace --count --missingr� r� u� # coding: iso-8859-15
1: x = 'spœm'
1: if []:
>>>>>> print('unreachable')
)
r� r� r r� r r r� r� r� rl r� r� r� )rK r� r� r� r� rB r r r �'test_cover_files_written_with_highlight� s "�zETestCoverageCommandLineOutput.test_cover_files_written_with_highlightN) rW rX rY r� r� ri r� r� r� r r r r r� � s r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �TestCommandLinec C s: d}|D ]^}}t ddg|�R � �^ }}| �||� qd S )N))s3 progname is missing: required with the main options�-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�-mr_ )r r� )rK �_errors�messager: �_r� r r r �
test_failures� s
�z TestCommandLine.test_failuresc C s� t d }tj�t �}t|ddd��7}| �t|� |�d� tddd|dd �\}}}| � d
|� d |� d |� d
�}| � |�
� |� W d � d S 1 sKw Y d S )Nr� r� �utf-8r� za = 1
r� r_ r� �ZPYTHONIOENCODINGs functions called:z
filename: z, modulename: z, funcname: <module>) r r r r r� rb r r� r r� �encode)rK r �
modulename�fdr� r� r� rp r r r �test_listfuncs_flag_success� s
� "�z+TestCommandLine.test_listfuncs_flag_successc C s� t tddd��}| �tt� |�d� |�d� W d � n1 s"w Y tt�\}}}tdddtdd �\}}}| �|�� |� d S )
Nr� r� r� z import sys
zprint(type(sys.argv))
r� r_ r� r� )r� r rb r r� r r� �strip)rK 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�}tj�t �}t|ddd�� }| �t|� | �t|� |�t� d�� W d � n1 s6w Y t
ddd |dd
�\}}}|� � }| � |d � | �
d |� | �
d
|� d|� d�|� d S )Nr� z.coverr� r� r� z� x = 1
y = 2
def f():
return x + y
for i in range(10):
f()
r� r_ z-csr� r z lines cov% module (path)z 6 100% z (�))r r r r r� rb r r� r� r� r �decoderl r� )rK r Z
coverfilenamer� r� r� r� r� r r r �test_count_and_summary s
�
� z&TestCommandLine.test_count_and_summaryc C s( t ddddddd� tddddd� d S ) Nr� r_ r� z--moduleZtimeitz-n�1Znot_a_module_zzzr rh r r r �test_run_as_module$ s z"TestCommandLine.test_run_as_moduleN)rW rX rY r� r� r� r� r� r r r r r� � s
r� �__main__).r �pickler rc Z test.supportr Ztest.support.os_helperr r r Ztest.support.script_helperr r r� r� r_ r
Ztest.tracedmodulesr r r r r% r+ r- r/ r1 r3 r5 r9 r< r? rH �objectrI ZTestCaser\ r� r� r� r� r� r� r� rW �mainr r r r �<module> sJ
m C G
2C �