U
{h: @ sJ d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZ ddl m
Z
ddl m Z ddl m
Z
mZmZ ddlmZmZ ddl mZ dd lmZ G d
d d eZd d
ZG dd dejZG dd deZeee dddj dZ eee dddj dZ dd Z G dd deZ dd Z!d$d d Z"d%d d!Z#d"d# Z$dS )& N)partial )Errors)
CodeWriter) TreeFragmentstrip_common_indentStringParseContext) TreeVisitorVisitorTransform)TreePath) PostParsec s$ e Zd Z fddZdd Z ZS )NodeTypeWriterc s t d| _g | _d S )Nr )super__init___indentsresultself __class__ 4/tmp/pip-unpacked-wheel-fhl22ezh/Cython/TestUtils.pyr s
zNodeTypeWriter.__init__c C s | j s d}n0| j d }|d d k r4d|dd }n|d }| jd| j d||jjf | jd7 _| | | jd8 _d S ) Nz(root) z%s[%d]r z z%s: %s)Z access_pathr appendr r __name__
visitchildren)r nodenameZtipr r r
visit_Node s
zNodeTypeWriter.visit_Node)r
__module__ __qualname__r r
__classcell__r r r r r
s r
c C s( t }|| ddg|j dg S )zReturns 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
visitjoinr )rootwr r r treetypes) s
r* c @ sj e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
d Z dddZ
dd Z e fddZ
dd ZdS )
CythonTestc C s t d S Nr Z init_threadr r r r setUp5 s zCythonTest.setUpc C s t d S r, r- r r r r tearDown8 s zCythonTest.tearDownc C s t |ts|d}t |ts(|d}tt||D ]$\}\}}| ||d|||f q6| t|t|dd|d|f dS )zHChecks that the given strings or lists of strings are equal line by liner$ zLine %d:
Exp: %s
Got: %s%Unmatched lines. Got:
%s
Expected:
%sN)
isinstancelistsplit enumeratezip assertEquallenr' )r expectedr idx
expected_lineZ result_liner r r assertLines; s
zCythonTest.assertLinesc C s t }|| |jjS r, )r writer lines)r treewriterr r r codeToLinesG s
zCythonTest.codeToLinesc C s d | |S )Nr$ )r' r@ )r r> r r r codeToStringL s zCythonTest.codeToStringc C st | |}t|d}tt||D ]$\}\}}| ||d|||f q&| t|t|dd||f d S )Nr$ zLine %d:
Got: %s
Exp: %sr0 )r@ r r3 r4 r5 r6 r7 r' )r r8 result_treeZ result_linesZexpected_linesr9 liner: r r r
assertCodeO s
zCythonTest.assertCodec C s | t||d d| d S )Nz"Path '%s' not found in result tree)ZassertNotEqualr
find_first)r pathrB r r r assertNodeExistsZ s zCythonTest.assertNodeExistsNc C sV |dkr i }|dkrg }| }|dr:|tdd }|dd}t||||dS )zNSimply create a tree fragment using the name of the test-case in parse errors.Nz __main__.._pipeline)id
startswithr7 replacer )r codepxdsrK r r r r fragment^ s
zCythonTest.fragmentc C s t |S r, )r* )r r( r r r r* j s zCythonTest.treetypesc
C sX z| | d| W n: |k
rR } z | t|| | W Y S d}~X Y nX dS )zCalls "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)fail
assertTruer1 )r funcexc_typeer r r should_failm s zCythonTest.should_failc
C s> z| W S t k
r8 } z| t| W 5 d}~X Y nX dS )zCalls 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) ExceptionrR str)r rT excr r r should_not_failx s zCythonTest.should_not_fail)NN)r r! r" r. r/ r; r@ rA rD rG rQ r* rX rW r[ r r r r r+ 3 s
r+ c @ s e Zd ZdZdddZdS )
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 s0 |d kr i }| ||j}|D ] }||}q |S r, )rQ r( )r rK ZpyxrP r> Tr r r run_pipeline s
zTransformTest.run_pipeline)N)r r! r" __doc__r^ r r r r r\ s r\ z\s+r% zz
/[*] (
(?: [^*\n] | [*][^/] )*
[\n]
(?: [^*] | [*][^/] )*
) [*]/
z\s\s+z
(?:
(?:[^<]|<(?!/pre))+
)|(?:
]*>
(?:[^<]|<(?!/style))+
)
c C s d }}| dr6tjd| dd dd\}} | } | dr~| dd } | dr~tjd| dd dd\}} | } ||| fS )N/z(?.failc st |r8t ||}|r&|| d }n j|d| d |rpt ||}|r^|d | }n j|d| d |S )NFrp rq )rc searchrg rj rf )rq contentrf rg r3 )rR r r r extract_section s zBTreeAssertVisitor.create_c_file_validator..extract_sectionc s D ]>}t |\}}}| |||}t||sj|d| d q D ]>}t |\}}}| |||}t||rHj|d| d qHd S )NFrr T)rh rc rs rj )rq rt re rf rg sectionZ antipattern) antipatternsru rR patternsr r r validate_file_content s zHTreeAssertVisitor.create_c_file_validator..validate_file_contentc sH | j }s s| S t|dd}| }W 5 Q R X t|}|| d S )Nutf8)encoding)c_fileopenread_strip_c_comments)r r| frt )rw rx ry r r validate_c_file s zBTreeAssertVisitor.create_c_file_validator..validate_c_file)rk rl )r r r )rw ru rR rx r ry r create_c_file_validator s z)TreeAssertVisitor.create_c_file_validatorc C s |j }d|kr>|d D ]&}t||d krt|jd| qd|krz|d D ]*}t||}|d k rNt|jd| qNd|kr| j|d d|kr| j|d d S )NZtest_assert_path_existsz+Expected path '%s' not found in result treeZtest_fail_if_path_existsz)Unexpected path '%s' found in result treeZtest_assert_c_code_hasZtest_fail_if_c_code_has)
directivesr rE r rn ro rk extendrl )r r r rF Z
first_noder r r _check_directives
s( z#TreeAssertVisitor._check_directivesc C s |j | _| | | | |S r, )ro rj r r r r r r r visit_ModuleNode! s
z"TreeAssertVisitor.visit_ModuleNodec C s | | | | |S r, )r r r r r r visit_CompilerDirectivesNode' s
z.TreeAssertVisitor.visit_CompilerDirectivesNode) r r! r" r r r r r r
Zrecurse_to_childrenr r# r r r r ri s 5ri c
C s t jgt jtj|dgt jtj|dgd}|d kr@t }g d }}t| dZ}z<|D ]0}|d d dkr| d
d d tjj }tj||} tj
tj| sttj| |d k r|d }
}|
t| d
}q`|d k r
|| q`| r`| ds`| d kr`t|
d} | sDq`| d | d
d } }
z||| |
W q` tk
r || Y q`X q`W 5 |d k r| X W 5 Q R X ||fS )Nz cython.pyz cythonize.py)PYTHONZCYTHONZ CYTHONIZErb s ##### #rz r` wb)s """s '''r r )sys
executableosrF r' tempfilemkdtempr} closerd decoderN sepexistsdirnamemakedirsr< lstriprM shlexr3 r KeyError)Z tree_fileZworkdirZ cython_rootZprogramsheaderZcur_filer rC filenamerF to_closecommandprogargsr r r unpack_source_tree/ sD
$
r Fc C sh t |trd}d}d}n d}d}d}|dkr0|}|r>t|}t| |||d}|| W 5 Q R X dS )zWrite some content (text or bytes) to the file
at `file_path` without translating `'\n'` into `os.linesep`.
The default encoding is `'utf-8'`.
r Nr) r$ zutf-8)moder{ newline)r1 bytestextwrapdedentr} r< )rq rt r r{ r r default_encodingr r r r
write_fileW s
r c C sh t | |||d ztj|}W n tk
r8 d}Y nX |dksR|tj| krdt | |||d q:dS )z
Write `content` to the file `file_path` without translating `'\n'`
into `os.linesep` and make sure it is newer than the file `newer_than`.
The default encoding is `'utf-8'` (same as for `write_file`).
)r r{ N)r r rF getmtimeOSError)rq Z
newer_thanrt r r{ Z
other_timer r r write_newer_fileu s
r c
C s t d}zFt }t| t|gd}| }W 5 Q R X |rF|d n|W S W n. tjk
r| } zt|jW 5 d}~X Y nX dS )z
Compiles code far enough to get errors from the parser and post-parse stage.
Is useful for checking for syntax errors, however it doesn't generate runable
code.
testrJ r N) r r Z local_errorsr r
substitute CompileError SyntaxErrorZ message_only)rO contexterrorsr rV r r r
py_parse_code s
r )FN)FN)%r rc Zunittestr r r r functoolsr ZCompilerr r ZCompiler.TreeFragmentr r r ZCompiler.Visitorr r
r Z Compiler.ParseTreeTransformsr r
r* ZTestCaser+ r\ compilesubr Z _strip_cython_code_from_htmlrh ri r r r r r r r r sB
O&
a(