Your IP : 216.73.216.213
Z��Z+ � @ s
d d l m Z d d l Z d d l Z d d l Z d d l m Z d d l m Z d d l m Z m
Z
d d l m Z m
Z
d d l m Z Gd d
� d
e � Z d d � Z Gd
d � d e j � Z Gd d � d e � Z Gd d � d e
� Z d d d � Z d S)� )�absolute_importN� )�Errors)�
CodeWriter)� TreeFragment�strip_common_indent)� TreeVisitor�VisitorTransform)�TreePathc s. e Z d Z � f d d � Z d d � Z � S)�NodeTypeWriterc s) t t | � j � d | _ g | _ d S)Nr )�superr �__init__�_indents�result)�self)� __class__� �C/opt/alt/python35/lib64/python3.5/site-packages/Cython/TestUtils.pyr
s zNodeTypeWriter.__init__c C s� | j s d } n> | j d } | d d k rF d | d d � } n
| d } | j j d | j d | | j j f � | j d 7_ | j | � | j d 8_ d S) Nz(root)r � z%s[%d]� z z%s: %s���)Z access_pathr �appendr r �__name__�
visitchildren)r �node�nameZtipr r r �
visit_Node s
zNodeTypeWriter.visit_Node)r �
__module__� __qualname__r
r r r )r r r s r c C s4 t � } | j | � d j d g | j d g � S)z�Returns a string representing the tree by class names.
There's a leading and trailing whitespace so that it can be
compared by simple string comparison while still making test
cases look ok.�
� )r Zvisit�joinr )�root�wr r r � treetypes% s
r$ c @ s� 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 d d � Z
d d � Z e d d � Z
d d � Z d S)�
CythonTestc C s, t j | _ t j | _ d t _ t _ d S)N)r � listing_file� echo_file)r r r r �setUp1 s zCythonTest.setUpc C s | j t _ | j t _ d S)N)r& r r' )r r r r �tearDown6 s zCythonTest.tearDownc C s� t | t � s | j d � } t | t � s< | j d � } xI t t | | � � D]2 \ } \ } } | j | | d | | | f � qR W| j t | � t | � d d j | � d j | � f � d S)zHChecks that the given strings or lists of strings are equal line by liner zLine %d:
Exp: %s
Got: %sz%Unmatched lines. Got:
%s
Expected:
%sN)�
isinstance�list�split� enumerate�zip� assertEqual�lenr! )r �expectedr �idx�
expected_lineZ result_liner r r � assertLines: s ( zCythonTest.assertLinesc C s t � } | j | � | j j S)N)r �writer �lines)r �tree�writerr r r � codeToLinesF s
zCythonTest.codeToLinesc C s d j | j | � � S)Nr )r! r9 )r r7 r r r � codeToStringK s zCythonTest.codeToStringc C s� | j | � } t | j d � � } xI t t | | � � D]2 \ } \ } } | j | | d | | | f � q: W| j t | � t | � d d j | � | f � d S)Nr zLine %d:
Got: %s
Exp: %sz%Unmatched lines. Got:
%s
Expected:
%s)r9 r r, r- r. r/ r0 r! )r r1 � result_treeZ result_linesZexpected_linesr2 �liner3 r r r �
assertCodeN s ( zCythonTest.assertCodec C s' | j t j | | � d d | � d S)Nz"Path '%s' not found in result tree)ZassertNotEqualr
�
find_first)r �pathr; r r r �assertNodeExistsY s zCythonTest.assertNodeExistsNc C s} | d k r i } | d k r$ g } | j � } | j d � rU | t d � d � } | j d d � } t | | | d | �S)zNSimply create a tree fragment using the name of the test-case in parse errors.Nz __main__.�.�_�pipeline)�id�
startswithr0 �replacer )r �code�pxdsrC r r r r �fragment] s zCythonTest.fragmentc C s
t | � S)N)r$ )r r" r r r r$ i s zCythonTest.treetypesc C sb y | � | j d | � Wn? | k
r] } z | j t | | � � | SWYd d } ~ Xn Xd S)z�Calls "func" and fails if it doesn't raise the right exception
(any exception by default). Also returns the exception in question.
z Expected an exception of type %rN)�failZ
assertTruer* )r �func�exc_type�er r r � should_faill s zCythonTest.should_failc C sJ y | � SWn8 t k
rE } z | j t | � � WYd d } ~ Xn Xd S)z�Calls func and succeeds if and only if no exception is raised
(i.e. converts exception raising into a failed testcase). Returns
the return value of func.N)� ExceptionrJ �str)r rK �excr r r �should_not_failw s zCythonTest.should_not_fail)r r r r( r) r4 r9 r: r= r@ rI r$ rO rN rR r r r r r% / s r% c @ s% e Z d Z d Z d d d � Z d S)�
TransformTesta.
Utility base class for transform unit tests. It is based around constructing
test trees (either explicitly or by parsing a Cython code string); running
the transform, serialize it using a customized Cython serializer (with
special markup for nodes that cannot be represented in Cython),
and do a string-comparison line-by-line of the result.
To create a test case:
- Call run_pipeline. The pipeline should at least contain the transform you
are testing; pyx should be either a string (passed to the parser to
create a post-parse tree) or a node representing input to pipeline.
The result will be a transformed result.
- Check that the tree is correct. If wanted, assertCode can be used, which
takes a code string as expected, and a ModuleNode in result_tree
(it serializes the ModuleNode to a string and compares line-by-line).
All code strings are first stripped for whitespace lines and then common
indentation.
Plans: One could have a pxd dictionary parameter to run_pipeline.
Nc C sH | d k r i } | j | | � j } x | D] } | | � } q. W| S)N)rI r" )r rC ZpyxrH r7 �Tr r r � run_pipeline� s
zTransformTest.run_pipeline)r r r �__doc__rU r r r r rS � s rS c @ s% e Z d Z d d � Z e j Z d S)�TreeAssertVisitorc C s� | j } d | k rY xA | d D]5 } t j | | � d k r t j | j d | � q Wd | k r� xA | d D]5 } t j | | � d k rp t j | j d | � qp W| j | � | S)NZtest_assert_path_existsz+Expected path '%s' not found in result treeZtest_fail_if_path_existsz)Unexpected path '%s' found in result tree)�
directivesr
r> r �error�posr )r r rX r? r r r � visit_CompilerDirectivesNode� s
z.TreeAssertVisitor.visit_CompilerDirectivesNodeN)r r r r[ r Zrecurse_to_childrenr r r r r rW � s rW c C s� | d k r t j � } g } d } t | � } z | j � } Wd | j � X~ z6x/| D]'} | d d � d k r%| j � j d � j � j d t j j � } t j j
| | � } t j j t j j | � � s� t j
t j j | � � | d k r| d } } | j � t | d � } q[ | d k rA| j | � q[ | j � r[ | j � j d � r[ | j � d k r[ | j | � q[ WWd | d k r�| j � X| d j
| � f S)
N� z#####�#�/r# �"""�'''r )r_ r` )�tempfileZmkdtemp�open� readlines�close�striprF �osr? �sepr! �exists�dirname�makedirsr5 �lstriprE r ) Z tree_file�dir�headerZcur_file�fr6 r< �filenamer? r r r �unpack_source_tree� s8
-
" rp )Z
__future__r rf Zunittestra ZCompilerr r ZCompiler.TreeFragmentr r ZCompiler.Visitorr r r
r r$ ZTestCaser% rS rW rp r r r r �<module> s
R"