Your IP : 216.73.216.213
�
g f�9 � @ s� d d l Z d d l Z d d l Z d d l m Z m Z m Z m Z d d l Z d d l Z d d l m
Z
m Z d d l m
Z
d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z Gd d � d e � Z Gd d � d e j � Z Gd! d" � d" e j � Z Gd# d$ � d$ e j � Z Gd% d& � d& e j � Z Gd' d( � d( e j � Z! Gd) d* � d* e j � Z" Gd+ d, � d, e j � Z# e$ d- k r�e j% � n d S).� N)�TESTFN�rmtree�unlink�captured_stdout)�CoverageResults�Trace)�testmodc C s& | j d � r" | d d � } n | S)z=Given a .pyc/.pyo filename converts it to the appropriate .py�.pyc�.pyoN� )r r
���)�endswith)�filename� r �4/opt/alt/python34/lib64/python3.4/test/test_trace.py�
fix_ext_py s r c C s2 t j j t j j t � � d } t t � | f S)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 j S)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+ | } x t d � D] } | | 7} q W| S)N� )�range)r r r! �ir r r �traced_func_loop* s r& c C s | | t j d � 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 } x t | � D] } | | Vq Wd S)Nr# )r$ )�numr! r% r r r �traced_func_generator< s r, c C s+ d } x t d � D] } | | 7} q Wd 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 r0 c C s# d } d d � t | � D� } | S)Nr- c S s g | ] } t | � � q Sr )r0 )�.0r% r r r �
<listcomp>K s z4traced_caller_list_comprehension.<locals>.<listcomp>)r$ )r) Zmylistr r r � traced_caller_list_comprehensionI s r3 c @ sX e Z d Z d d � Z d d � Z d d � Z e d d � � Z e d 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 )r5 r r r r �inst_method_linearS s z TracedClass.inst_method_linearc C s | j | � } | t | | � S)N)r7 r" )r5 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__r6 r7 r8 � classmethodr: � staticmethodr; r r r r r4 O s
r4 c @ sj e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z
d S)�TestLineCountsz/White-box testing of line-counting, via runfuncc C sP | j t j t j � � t d d d d d d d d � | _ t t � | _ d S)N�countr �tracer �
countfuncs� countcallers) �
addCleanup�sys�settrace�gettracer �tracerr r �my_py_filename)r5 r r r �setUph s $zTestLineCounts.setUpc C s� | j j t d d � } | j | d � i } t t � } x. t d d � D] } d | | j | | f <qJ W| j | j j � j | � d S)Nr/ r# � r ) rJ �runfuncr" � assertEqualr r$ rK �results�counts)r5 �result�expected� firstlinenor% r r r �test_traced_func_linearm s z&TestLineCounts.test_traced_func_linearc C s� | j j t d d � t t � } i d | j | d f 6d | j | d f 6d | j | d f 6d | j | d f 6} | j | j j � j | � d S)Nr/ � r � r# � )rJ rN r& r rK rO rP rQ )r5 rT rS r r r �test_traced_func_loopy s z$TestLineCounts.test_traced_func_loopc C s� | j j t d d � t t � } i d | j | d f 6d t t j � d f 6d t t j � d f 6} | j | j j � j
| � d S)Nr/ r# r rV ) rJ rN r' r rK r r r rO rP rQ )r5 rT rS r r r �test_traced_func_importing� s z)TestLineCounts.test_traced_func_importingc C s� | j j t � t t � } t t � } i d | j | d f 6d | j | d f 6d | j | d f 6d | j | d f 6d | j | d f 6d | j | d f 6} | j | j j � j | � d S)Nr � r/ r- rV ) rJ rN r. r r, rK rO rP rQ )r5 �firstlineno_callingZfirstlineno_genrS r r r �test_trace_func_generator� s z(TestLineCounts.test_trace_func_generatorc C s� | j j t � t t � } t t � } i d | j | d f 6d | j | d f 6d | j | d f 6d | j | d f 6} | j | j j � j | � d S)Nr � r/ rV r- ) rJ rN r3 r r0 rK rO rP rQ )r5 r\ Zfirstlineno_calledrS r r r � test_trace_list_comprehension� s z,TestLineCounts.test_trace_list_comprehensionc
C s� x� d g D]� } t d d d d d d d d � } t d � } t | | � } | j | d � t | � } i d | j | d f 6} | j | j � j | � q
Wd S)
Nr7 rB r rC r rD rE � � ) r r4 �getattrrN r rK rO rP rQ )r5 ZmethnamerJ Z
traced_obj�methodrT rS r r r �test_linear_methods� s ! z"TestLineCounts.test_linear_methodsN) r< r= r> �__doc__rL rU rY rZ r] r_ rd r r r r rA f s rA c @ s. e Z d Z d Z d d � Z d d � Z d S)�TestRunExecCountsz8A simple sanity test of line-counting, via runctx (exec)c C s, t t � | _ | j t j t j � � d S)N)r r rK rF rG rH rI )r5 r r r rL � s zTestRunExecCounts.setUpc C s� t d d d d d d d d � | _ d } t | t d � } | j j | t � t � � t t � } i d | j | d f 6d | j | d
f 6d | j | d f 6d | j | d
f 6} x8 | j
� D]* } | j | j j � j
| | | � q� Wd S)NrB r rC r rD rE ztraced_func_loop(2, 5)�execrW r/ r# rV rX )r rJ �compiler Zrunctx�globals�varsr r& rK �keysrO rP rQ )r5 �coderT rS r) r r r �test_exec_counts� s $ z"TestRunExecCounts.test_exec_countsN)r< r= r> re rL rm r r r r rf � s rf c @ sm e Z d Z d Z d d � Z d d � Z d d � Z e j e e
d � oT e
j � d � d
d � � Z d S)
� TestFuncsz"White-box testing of funcs tracingc C sG | j t j t j � � t d d d d d d � | _ t � | _ d S)NrB r rC rD r )rF rG rH rI r rJ r �filemod)r5 r r r rL � s zTestFuncs.setUpc C sU | j j t d � i d | j d 6d | j d 6} | j | j j � j | � d S)Nr r( r" )ztraced_func_simple_caller)ztraced_func_linear)rJ rN r( ro rO rP � calledfuncs)r5 rS r r r �test_simple_caller� s
z TestFuncs.test_simple_callerc C s� | j j t d � i d | j d 6d | j d 6d | j d
6d | j d 6d t t j � d d f 6} | j | j j � j | � d S) Nr r( r" r* r' r r )ztraced_func_simple_caller)ztraced_func_linear)z traced_func_importing_caller)ztraced_func_importing)
rJ rN r* ro r r r rO rP rp )r5 rS r r r �test_loop_caller_importing� s z$TestFuncs.test_loop_caller_importingrI z3pre-existing trace function throws off measurementsc C sr t d � } | j j | j d � i d | j d 6d | j d 6d | j d 6} | j | j j � j | � d S) Nra r � TracedClass.inst_method_calling� TracedClass.inst_method_linearr" )rs )rt )ztraced_func_linear)r4 rJ rN r8 ro rO rP rp )r5 �objrS r r r �test_inst_method_calling� s z"TestFuncs.test_inst_method_callingN)
r< r= r> re rL rq rr �unittest�skipIf�hasattrrG rI rv r r r r rn � s rn c @ sU e Z d Z d Z d d � Z e j e e d � o< e j � d � d d � � Z
d S) � TestCallersz$White-box testing of callers tracingc C sG | j t j t j � � t d d d d d d � | _ t � | _ d S)NrB r rC rE r )rF rG rH rI r rJ r ro )r5 r r r rL s zTestCallers.setUprI z3pre-existing trace function throws off measurementsc C s� | j j t d � i d t j j t j � d d d d f | j d f 6d | j d
| j d f 6d | j d | j d f 6d | j d | j d f 6d | j d t t
j � d
d f f 6} | j | j j � j
| � d S)Nr r z.pyrC z
Trace.runfuncr* r( r" r' r r )� traced_func_importing_caller)�traced_func_simple_caller)ztraced_func_linear)r{ )r| )r{ )�traced_func_importing)r} )rJ rN r* r r r rC r ro r r rO rP Zcallers)r5 rS r r r rr
s 4)z&TestCallers.test_loop_caller_importingN) r< r= r> re rL rw rx ry rG rI rr r r r r rz s rz c @ s[ e Z d Z d d � Z d d � Z d d d � Z d d � Z d
d � Z d d
� Z d S)� TestCoveragec C s | j t j t j � � d S)N)rF rG rH rI )r5 r r r rL # s zTestCoverage.setUpc C s t t � t t � d S)N)r r r )r5 r r r �tearDown& s
zTestCoverage.tearDownz_import test.support, test.test_pprint;test.support.run_unittest(test.test_pprint.QueryTestCase)c C s9 | j | � | j � } | j d d d d d t � d S)NZ show_missingTZsummaryZcoverdir)ZrunrP Z
write_resultsr )r5 rJ �cmd�rr r r � _coverage* s
zTestCoverage._coveragec C s� t j d d d d � } t � � } | j | � Wd QX| j � } | j d | k � | j d | k � t j t � } | j d | k � | j d | k � d S) NrC r rB r z pprint.pyzcase.pyz pprint.coverzunittest.case.cover) rC r r r� �getvalue�
assertTruer �listdirr )r5 rJ �stdout�filesr r r �
test_coverage1 s zTestCoverage.test_coveragec C s� t j j t j j t j � � } t j d t j t j | g d d d d � } t
� � } | j | � Wd QXt j j t
� r� t j t
� } | j | d g � n d S)NZ
ignoredirsrC r rB r z_importlib.cover)r r �normpath�dirnamer rC r rG � base_prefix�base_exec_prefixr r� �existsr r� rO )r5 ZlibpathrJ r� r� r r r �test_coverage_ignore< s ! z!TestCoverage.test_coverage_ignorec
C s t j d d d d � } d } | t j k r: t j | =n d } t � � } | j | | � Wd QX| j d � | j � i } xS | D]K } | j � d d � \ } } } t | � t | d d
� � f | | <q� Wt j
t j | j � } | j | | � | j
| | d � d S) NrC r rB r ztest.tracedmodules.testmodz<import test.tracedmodules.testmod as t;t.func(0); t.func2();rV r# �d r )r# r� )rC r rG �modulesr r� �seek�readline�split�intZ _fullmodnamer �assertInrO )
r5 rJ r r� r� Zcoverage�line�linesZcov�moduler r r �test_issue9936H s
*zTestCoverage.test_issue9936N) r< r= r> rL r r� r� r� r� r r r r r~ " s r~ c @ s e Z d Z d d � Z d S)� Test_Ignorec C s� t j j } t j d d g | d d � g � } | j | j d d � � | j | j d d � � | j | j d d � � | j | j | d d d
� d � � | j | j | d d � d
� � | j | j | d d
� d � � d S)Nr zy.z�fooZbarzx.pyzxy.pyZxyzy.pyr zbaz.pyZbazzz.py�z)r r �joinrC Z_Ignorer� �namesZ assertFalse)r5 Zjn�ignorer r r � test_ignoreda s $%"zTest_Ignore.test_ignoredN)r< r= r> r� r r r r r� ` s r� c @ sp e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d S)�TestDeprecatedMethodsc C sI t j � } | j t � � t j | � Wd QX| j d | j � � d S)NzUsage:)�io�StringIO� assertWarns�DeprecationWarningrC Zusager� r� )r5 Zsior r r �test_deprecated_usageo s z+TestDeprecatedMethods.test_deprecated_usagec C s$ | j t � � t j � Wd QXd S)N)r� r� rC �Ignore)r5 r r r �test_deprecated_Ignoreu s z,TestDeprecatedMethods.test_deprecated_Ignorec C s3 | j t � � | j d t j d � � Wd QXd S)N�spam)r� r� rO rC r )r5 r r r �test_deprecated_modnamey s z-TestDeprecatedMethods.test_deprecated_modnamec C s3 | j t � � | j d t j d � � Wd QXd S)Nr� )r� r� rO rC Z fullmodname)r5 r r r �test_deprecated_fullmodname} s z1TestDeprecatedMethods.test_deprecated_fullmodnamec
C s< | j t � �' d d � } t j | j d g � Wd QXd S)Nc S s d S)Nr r r r r r� � s zGTestDeprecatedMethods.test_deprecated_find_lines_from_code.<locals>.foo�eggs)r� r� rC Zfind_lines_from_coder )r5 r� r r r �$test_deprecated_find_lines_from_code� s z:TestDeprecatedMethods.test_deprecated_find_lines_from_codec
C s< | j t � �' d d � } t j | j d g � Wd QXd S)Nc S s d S)Nr r r r r r� � s z=TestDeprecatedMethods.test_deprecated_find_lines.<locals>.foor� )r� r� rC Z
find_linesr )r5 r� r r r �test_deprecated_find_lines� s z0TestDeprecatedMethods.test_deprecated_find_linesc C sR t t d � � } | j t t � Wd QX| j t � � t j | j � Wd QXd S)N�w) �openr rF r r� r� rC Z find_strings�name)r5 �fdr r r � test_deprecated_find_strings� s z2TestDeprecatedMethods.test_deprecated_find_stringsc C sR t t d � � } | j t t � Wd QX| j t � � t j | j � Wd QXd S)Nr� ) r� r rF r r� r� rC Zfind_executable_linenosr� )r5 r� r r r �'test_deprecated_find_executable_linenos� s z=TestDeprecatedMethods.test_deprecated_find_executable_linenosN) r< r= r> r� r� r� r� r� r� r� r� r r r r r� m s r� �__main__)&r r� rG Z test.supportr r r r rw rC r r Ztest.tracedmodulesr r r r r" r& r' r( r* r, r. r0 r3 �objectr4 ZTestCaserA rf rn rz r~ r� r� r< �mainr r r r �<module> s: "
Z * >
-