U
c±³eo9 ã @ 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 eƒZd
d„ ZG dd„ dejƒZG dd„ deƒZe e e ddd¡¡j dƒZ e e e ddd¡¡j dƒZ dd„ Z G dd„ deƒZ!d d „ Z"d#d d „Z#d$d!d"„Z$dS )%é )Úabsolute_importN)Úopen)Úpartialé )ÚErrors)Ú
CodeWriter)Ú TreeFragmentÚstrip_common_indent)Ú TreeVisitorÚVisitorTransform)ÚTreePathc s$ e Zd Z‡ fdd„Zdd„ Z‡ ZS )ÚNodeTypeWriterc s t t| ƒ ¡ d| _g | _d S )Nr )Úsuperr
Ú__init__Ú_indentsÚresult©Úself©Ú __class__© ú/lib/python3.8/site-packages/Cython/TestUtils.pyr s zNodeTypeWriter.__init__c C sˆ | j s d}n0| j d }|d d k r4d|dd… }n|d }| j d| 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 ÚnodeÚnameZtipr 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 ƒ }| | ¡ d dg|j dg ¡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 @ sj e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd d
„ Zd d „ Zd
d„ Z ddd„Z
dd„ Z e fdd„Z
dd„ ZdS )Ú
CythonTestc C s t ¡ d S ©N©r Z init_threadr r r r ÚsetUp7 s zCythonTest.setUpc C s t ¡ d S r+ r, r r r r ÚtearDown: s zCythonTest.tearDownc C sŠ t |tƒs| d¡}t |tƒs(| 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)Ú
isinstanceÚlistÚsplitÚ enumerateÚzipÚ assertEqualÚlenr& )r Úexpectedr ÚidxÚ
expected_lineZ result_liner r r Ú assertLines= s
ÿÿzCythonTest.assertLinesc C s t ƒ }| |¡ |jjS r+ )r Úwriter Úlines)r ÚtreeÚwriterr r r Ú codeToLinesI s
zCythonTest.codeToLinesc C s d | |¡¡S )Nr$ )r& r? )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: %sr/ )r? r r2 r3 r4 r5 r6 r& )r r7 Ú result_treeZ result_linesZexpected_linesr8 Úliner9 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 ÚpathrA r r r ÚassertNodeExists\ s ÿzCythonTest.assertNodeExistsNc C sV |dkr i }|dkrg }| ¡ }| d¡r:|tdƒd… }| dd¡}t||||dS )zNSimply create a tree fragment using the name of the test-case in parse errors.Nz __main__.Ú.Ú_)Úpipeline)ÚidÚ
startswithr6 Úreplacer )r ÚcodeÚpxdsrI 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 sX z|ƒ | d| ¡ W n: |k
rR } z | t||ƒ¡ | W Y ¢S d}~X Y nX dS )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
assertTruer0 )r ÚfuncÚexc_typeÚer r r Ú should_failo s zCythonTest.should_failc
C s> z|ƒ W S t k
r8 } z| t|ƒ¡ W 5 d}~X Y nX dS )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)Ú ExceptionrP Ústr)r rQ Úexcr r r Úshould_not_failz s zCythonTest.should_not_fail)NN)r r! r" r- r. r: r? r@ rC rF rO r) rU rT rX r r r r r* 5 s
r* c @ s e Zd ZdZddd„ZdS )Ú
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+ )rO r' )r rI ZpyxrN r= ÚTr r r Ú run_pipelineœ s
zTransformTest.run_pipeline)N)r r! r" Ú__doc__r[ r r r r rY „ s rY z\s+r% zz
/[*] (
(?: [^*\n] | [*][^/] )*
[\n]
(?: [^*] | [*][^/] )*
) [*]/
z\s\s+zÅ
(?:
(?:[^<]|<(?!/pre))+
)|(?:
]*>
(?:[^<]|<(?!/style))+
)
c C sˆ d }}| d¡r6tjd| dd … dd\}} | ¡ } | d¡r~| dd … ¡ } | d¡r~tjd| dd … dd\}} | ¡ } ||| fS )Nú/z(?.failc st |r8t ||¡}|r&|| ¡ d … }nˆ ˆj|d| d |rpt ||¡}|r^|d | ¡ … }nˆ ˆj|d| d |S )NF©rl rm )r` Úsearchrd rg rc )rm Úcontentrc rd r2 )rP r r r Úextract_sectionä s zBTreeAssertVisitor.create_c_file_validator..extract_sectionc sŒ ˆD ]>}t |ƒ\}}}ˆ| |||ƒ}t ||¡sˆˆj|d| d qˆ D ]>}t |ƒ\}}}ˆ| |||ƒ}t ||¡rHˆˆj|d| d qHd S )NFrn T)re r` ro rg )rm rp rb rc rd ZsectionZ antipattern)Ú antipatternsrq rP Úpatternsr r r Úvalidate_file_contentó s zHTreeAssertVisitor.create_c_file_validator..validate_file_contentc s² | j }ˆsˆ s| S t|dd}| ¡ }W 5 Q R X t|ƒ}ˆ||ƒ tj |¡d d }tj |¡r®tj |¡tj |¡kr®t|dd}| ¡ }W 5 Q R X t |ƒ}ˆ||ƒ d S )NÚutf8)Úencodingr z.html)
Úc_filer ÚreadÚ_strip_c_commentsÚosrE ÚsplitextÚexistsÚgetmtimeÚ _strip_cython_code_from_html)r rw Úfrp Z html_file)rr rs rt r r Úvalidate_c_file s
$zBTreeAssertVisitor.create_c_file_validator..validate_c_file)rh ri )r r€ r )rr rq rP rs r rt 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 rD r rj rk rh Úextendri )r r r‚ rE Z
first_noder r r Ú_check_directives s( þ þz#TreeAssertVisitor._check_directivesc C s |j | _| |¡ | |¡ |S r+ )rk rg 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_CompilerDirectivesNode0 s
z.TreeAssertVisitor.visit_CompilerDirectivesNode) r r! r" r r r„ r† r‡ r Zrecurse_to_childrenr r# r r r r rf Ð s <rf c
C s¾ t jgt jtj |d¡gt jtj |d¡gdœ}|d kr@t ¡ }g d }}t| dƒZ}z<|D ]0}|d d… dkrô| ¡ d¡ ¡
d¡ d tjj ¡}tj ||¡} tj
tj | ¡¡sÎt tj | ¡¡ |d k rè|d }
}|
¡ t| d
ƒ}q`|d k r
| |¡ q`| ¡ r`| ¡ d¡s`| ¡ 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)ZPYTHONZCYTHONZ CYTHONIZEÚrbé s #####ó #ru r] Úwb)s """s '''r r )ÚsysÚ
executablerz rE r& ÚtempfileZmkdtempr Úclosera ÚdecoderL Úsepr| ÚdirnameÚmakedirsr; ÚlstriprK Úshlexr2 r ÚKeyError)Z tree_fileZworkdirZ cython_rootZprogramsÚheaderZcur_filer rB ÚfilenamerE Zto_closeZcommandÚprogÚargsr r r Úunpack_source_tree8 sD ý
$
r› Fc C sh t |tƒrd}d}d}n d}d}d}|dkr0|}|r>t |¡}t| |||d}| |¡ W 5 Q R X dS )zœWrite 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)Úmoderv Únewline)r0 ÚbytesÚtextwrapÚdedentr r; )rm rp r rv rœ r Zdefault_encodingr r r r Ú
write_file` 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 rv N)r¡ rz rE r} ÚOSError)rm Z
newer_thanrp r rv Z
other_timer r r Úwrite_newer_file~ s
r£ )FN)FN)%Z
__future__r rz r` Zunittestr• rŒ rŽ rŸ Úior Ú functoolsr ZCompilerr r ZCompiler.TreeFragmentr r ZCompiler.Visitorr
r r r
r) ZTestCaser* rY ÚcompileÚsubry r~ re rf r› r¡ r£ r r r r Ú sB
O& ÿø
ÿ ô
h(