3
ho9 @ sJ d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl m
Z
d dl m Z ddl
mZ ddlmZ ddlmZmZ dd lmZmZ dd
l
mZ G d d d eZd
d ZG dd dejZG dd deZe ej ej dddj dZ e ej ej dddj dZ dd Z G dd deZ!d d Z"d#d d Z#d$d!d"Z$dS )% )absolute_importN)open)partial )Errors)
CodeWriter) TreeFragmentstrip_common_indent) TreeVisitorVisitorTransform)TreePathc s$ e Zd Z fddZdd Z ZS )NodeTypeWriterc s t t| j d| _g | _d S )Nr )superr
__init___indentsresult)self) __class__ b/var/www/html/CrowdFlow/Picklecon/ble_analysis_env/lib/python3.6/site-packages/Cython/TestUtils.pyr s zNodeTypeWriter.__init__c C s | j s d}n0| j d }|d d k r4d|dd }n|d }| jjd| j d||jjf | jd7 _| j| | jd8 _d S ) Nz(root)r z%s[%d] 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 }|j| djdg|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 j d S )N)r init_thread)r r r r setUp7 s zCythonTest.setUpc C s t j d S )N)r r* )r r r r tearDown: s zCythonTest.tearDownc C s t |ts|jd}t |ts(|jd}x6tt||D ]$\}\}}| j||d|||f q8W | jt|t|ddj|dj|f dS )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)
isinstancelistsplit enumeratezip assertEquallenr% )r expectedr idx
expected_lineZ result_liner r r assertLines= s
zCythonTest.assertLinesc C s t }|j| |jjS )N)r writer lines)r treewriterr r r codeToLinesI s
zCythonTest.codeToLinesc C s dj | j|S )Nr" )r% r< )r r: r r r codeToStringN s zCythonTest.codeToStringc C sx | j |}t|jd}x6tt||D ]$\}\}}| j||d|||f q(W | jt|t|ddj||f d S )Nr" zLine %d:
Got: %s
Exp: %sz%Unmatched lines. Got:
%s
Expected:
%s)r< r r/ r0 r1 r2 r3 r% )r r4 result_treeZ result_linesZexpected_linesr5 liner6 r r r
assertCodeQ s
zCythonTest.assertCodec C s | j tj||d d| d S )Nz"Path '%s' not found in result tree)ZassertNotEqualr
find_first)r pathr> r r r assertNodeExists\ s zCythonTest.assertNodeExistsNc C sV |dkr i }|dkrg }| j }|jdr:|tdd }|jdd}t||||dS )zNSimply create a tree fragment using the name of the test-case in parse errors.Nz __main__.._)pipeline)id
startswithr3 replacer )r codepxdsrF r r r r fragment` s
zCythonTest.fragmentc C s t |S )N)r( )r r& r r r r( l s zCythonTest.treetypesc C sN y| | j d| W n0 |k
rH } z| jt|| |S d}~X 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
assertTruer- )r funcexc_typeer r r should_failo s zCythonTest.should_failc C s< y| S t k
r6 } z| jt| W Y dd}~X 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) ExceptionrM str)r rO excr r r should_not_failz s zCythonTest.should_not_fail)NN)r r r r+ r, r7 r< r= r@ rC rL r( rS rR rV r r r r r) 5 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 s4 |d kr i }| j ||j}x|D ] }||}q W |S )N)rL r& )r rF ZpyxrK r: Tr r r run_pipeline s
zTransformTest.run_pipeline)N)r r r __doc__rY r r r r rW s rW z\s+r# zz
/[*] (
(?: [^*\n] | [*][^/] )*
[\n]
(?: [^*] | [*][^/] )*
) [*]/
z\s\s+z
(?:
(?:[^<]|<(?!/pre))+
)|(?:
]*>
(?:[^<]|<(?!/style))+
)
c C s d }}| j dr6tjd| dd dd\}} | j } | j dr~| dd j } | j dr~tjd| dd dd\}} | j } ||| fS )N/z(?.failc st |r8t j||}|r&||j d }n j|d| d |rpt j||}|r^|d |j }n j|d| d |S )NF)rk rl )r^ searchrb re ra )rl contentra rb r/ )rM r r r extract_section s zBTreeAssertVisitor.create_c_file_validator..extract_sectionc s xFD ]>}t |\}}}| |||}tj||sj|d| d qW xF D ]>}t |\}}}| |||}tj||rNj|d| d qNW d S )NF)rk rl T)rc r^ rm re )rl rn r` ra rb sectionZ antipattern) antipatternsro rM patternsr r r validate_file_content s
zHTreeAssertVisitor.create_c_file_validator..validate_file_contentc s | j }p s| S t|dd}|j }W d Q R X t|}|| tjj|d d }tjj|rtjj|tjj|krt|dd}|j }W d Q R X t |}|| d S )Nutf8)encodingr z.html)
c_filer read_strip_c_commentsosrB splitextexistsgetmtime _strip_cython_code_from_html)r rv frn Z html_file)rq rr rs r r validate_c_file s
$zBTreeAssertVisitor.create_c_file_validator..validate_c_file)rf rg )r r r )rq ro rM rr r rs r create_c_file_validator s z)TreeAssertVisitor.create_c_file_validatorc C s |j }d|krBx2|d D ]&}tj||d krtj|jd| qW d|krx6|d D ]*}tj||}|d k rTtj|jd| qTW d|kr| jj|d d|kr| jj|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 rA r ri rj rf extendrg )r r r rB Z
first_noder r r _check_directives s$ z#TreeAssertVisitor._check_directivesc C s |j | _| j| | j| |S )N)rj re r r )r r r r r visit_ModuleNode* s
z"TreeAssertVisitor.visit_ModuleNodec C s | j | | j| |S )N)r r )r r r r r visit_CompilerDirectivesNode0 s
z.TreeAssertVisitor.visit_CompilerDirectivesNode) r r r r r r r r r Zrecurse_to_childrenr r! r r )r r rd s <rd c C s t jgt jtjj|dgt jtjj|dgd}|d kr@tj }g d }}t| db}zDx<|D ]2}|d d dkr|j jdj j dj
d tjj }tjj||} tjj tjj
| stjtjj
| |d k r|d }
}|
j t| d
}qd|d k r|j| qd|j rd|j jd rd|j dkrdtj|j d} | sJqd| d
| dd } }
y|j|| |
W qd tk
r |j| Y qdX qdW W d |d k r|j X W d Q R X ||fS )Nz cython.pyz cythonize.py)PYTHONZCYTHONZ CYTHONIZErb s ##### #rt r[ wb """ '''r r )r r )sys
executablery rB r% tempfilemkdtempr r_ decoderI sepr{ dirnamemakedirscloser8 lstriprH shlexr/ r KeyError)Z tree_fileZworkdirZ cython_rootZprogramsheaderZcur_filer~ r? filenamerB to_closecommandprogargsr r r unpack_source_tree8 sB
$
r Fc C sh t |trd}d}d}n d}d}d}|dkr0|}|r>tj|}t| |||d}|j| W dQ 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)moderu newline)r- bytestextwrapdedentr r8 )rl rn r ru r r default_encodingr~ r r r
write_file` s
r c C sl t | |||d ytjj|}W n tk
r8 d}Y nX x,|dksT|tjj| krft | |||d q s>
O&
h(