o
e5 @ sB 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 e e dddj dZ e e e dddj dZ G dd deZ dd 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| d| _g | _d S )Nr )superr
__init___indentsresultself __class__ @D:\Projects\ConvertPro\env\Lib\site-packages\Cython/TestUtils.pyr s
zNodeTypeWriter.__init__c C s | j sd}n| j d }|d d urd|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 t d S Nr Z init_threadr r r r setUp7 zCythonTest.setUpc C r, r- r. r r r r tearDown: r0 zCythonTest.tearDownc C s t |ts
|d}t |ts|d}tt||D ]\}\}}| ||d|||f q| 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 codeToLinesI s
zCythonTest.codeToLinesc C s d | |S )Nr$ )r' rB )r r@ r r r codeToStringN s zCythonTest.codeToStringc C st | |}t|d}tt||D ]\}\}}| ||d|||f q| t|t|dd||f d S )Nr$ zLine %d:
Got: %s
Exp: %sr2 )rB r r5 r6 r7 r8 r9 r' )r r: result_treeZ result_linesZexpected_linesr; liner< r r r
assertCodeQ s
zCythonTest.assertCodec C s | t||d d| d S )Nz"Path '%s' not found in result tree)ZassertNotEqualr
find_first)r pathrD r r r assertNodeExists\ s zCythonTest.assertNodeExistsNc C sV |du ri }|du r g }| }|dr |tdd }|dd}t||||dS )zNSimply create a tree fragment using the name of the test-case in parse errors.Nz __main__.._)pipeline)id
startswithr9 replacer )r codepxdsrL r r r r fragment` s
zCythonTest.fragmentc C s t |S r- )r* )r r( r r r r* l s zCythonTest.treetypesc
C sT z
| | d| W dS |y) } z| t|| |W Y d}~S d}~ww )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
assertTruer3 )r funcexc_typeer r r should_failo s zCythonTest.should_failc
C s> z| W S t y } z| t| W Y d}~dS d}~ww )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) ExceptionrS str)r rU excr r r should_not_failz s zCythonTest.should_not_fail)NN)r r! r" r/ r1 r= rB rC rF rI rR r* rY rX r\ 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 s0 |d u ri }| ||j}|D ]}||}q|S r- )rR r( )r rL pyxrQ r@ Tr r r run_pipeline s
zTransformTest.run_pipeliner- )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 sB e Zd Z fddZdd Zdd Zdd Zd d
Zej Z
Z S ) TreeAssertVisitorc s$ t t| d | _g | _g | _d S r- )r rb r _module_pos _c_patterns_c_antipatternsr r r r r s
zTreeAssertVisitor.__init__c s< j j dd fdd fdd}|S )Nc S s" t | d||r dnd|f d S )NzPattern '%s' %s found in %swaszwas not)r error)pospatternfound file_pathr r r rS s
z7TreeAssertVisitor.create_c_file_validator..failc sT D ]}t ||sj|d| d q D ]}t ||r'j|d| d qd S )NF)rj rk T)researchrc )rk contentri Z antipattern) antipatternsrS patternsr r r validate_file_content s zHTreeAssertVisitor.create_c_file_validator..validate_file_contentc s | j }s s | S t|dd }| }W d n1 s w Y t|}|| tj|d d }tj|rmtj|tj|krot|dd }| }W d n1 s]w Y t |}|| d S d S d S )Nutf8)encodingr z.html)
c_filer read_strip_c_commentsosrH splitextexistsgetmtime _strip_cython_code_from_html)r rt frn Z html_file)ro rp rq r r validate_c_file s
$
zBTreeAssertVisitor.create_c_file_validator..validate_c_file)rd re )r r} r )ro rS rp r rq r create_c_file_validator s
z)TreeAssertVisitor.create_c_file_validatorc C s |j }d|v r |d D ]}t||d u r t|jd| q d|v r=|d D ]}t||}|d ur lstriprN shlexr5 r KeyError)Z tree_fileZworkdirZ cython_rootZprogramsheaderZcur_filer| rE filenamerH to_closecommandprogargsr r r unpack_source_tree sX
$
r Fc C s~ t |tr d}d}d}nd}d}d}|du r|}|r t|}t| |||d}|| W d dS 1 s8w Y 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)moders newline)r3 bytestextwrapdedentr r> )rk rn r rs r r Zdefault_encodingr| r r r
write_file@ s
"r c C s t | |||d ztj|}W n ty d}Y nw |du s(|tj| kr>t | |||d |du s(|tj| ks(dS 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 rs N)r rw rH rz OSError)rk Z
newer_thanrn r rs Z
other_timer r r write_newer_file^ s r )FN)$
__future__r rw rl Zunittestr r r r ior functoolsr ZCompilerr r ZCompiler.TreeFragmentr r ZCompiler.Visitorr
r r r
r* ZTestCaser+ r] compilesubrv r{ rb r r r r r r r sB
O&
U
(