o
wi: @ 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__ M/home/ubuntu/sommelier/.venv/lib/python3.10/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) access_pathr appendr r __name__
visitchildren)r nodenametipr 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 init_threadr r r r setUp5 zCythonTest.setUpc C r. r/ r0 r r r r tearDown8 r3 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_line 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) rF )r rD 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: %sr5 )rF r r8 r9 r: r; r< r) )r r= result_tree result_linesexpected_linesr> liner? r r r
assertCodeO s
zCythonTest.assertCodec C s | t||d d| d S )Nz"Path '%s' not found in result tree)assertNotEqualr
find_first)r pathrH r r r assertNodeExistsZ 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
startswithr< replacer )r codepxdsrT 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 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
assertTruer6 )r funcexc_typeer r r should_failm 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) Exceptionr[ str)r r] excr r r should_not_failx s zCythonTest.should_not_fail)NN)r r# r$ r2 r4 rA rF rG rL rP rZ r, ra r` rd 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 u ri }| ||j}|D ]}||}q|S r/ )rZ r* )r rT pyxrY rD Tr r r run_pipeline s
zTransformTest.run_pipeliner/ )r r# r$ __doc__rh r r r r re s re z\s+r' zz
/[*] (
(?: [^*\n] | [*][^/] )*
[\n]
(?: [^*] | [*][^/] )*
) [*]/
z\s\s+z
(?:
(?:[^<]|<(?!/pre))+
)|(?:
]*>
(?:[^<]|<(?!/style))+
)
c C s d }}| drtjd| dd dd\}} | } | dr?| dd } | dr?tjd| dd dd\}} | } ||| fS )N/z(?.failc sv |r t ||}|r|| d }n j|d| d |r9t ||}|r0|d | }|S j|d| d |S )NFrz r{ )rm searchrq rt rp )r{ contentrp rq r8 )r[ 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||rCj|d| d q$d S )NFr| T)rr rm r} rt )r{ r~ ro rp rq section antipattern) antipatternsr r[ 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|}|| d S )Nutf8)encoding)c_fileopenread_strip_c_comments)r r fr~ )r r r r r validate_c_file s
zBTreeAssertVisitor.create_c_file_validator..validate_c_file)ru rv )r r r )r r r[ r r r 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 urt | |||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 r N)r r rO getmtimeOSError)r{
newer_thanr~ r r
other_timer r r write_newer_fileu s r c
C s t d}z*t }t| t|gd}| }W d n1 s!w Y |r,|d |W S tjy@ } zt|jd}~ww )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.
testrS Nr ) r r local_errorsr r
substitute CompileError SyntaxError message_only)rX contexterrorsr r_ r r r
py_parse_code s
r )FN)%r rm unittestr r r r functoolsr Compilerr r Compiler.TreeFragmentr r r Compiler.Visitorr r
r Compiler.ParseTreeTransformsr r
r, TestCaser- re compilesubr _strip_cython_code_from_htmlrr rs r r r r r r r r sD
O&
a
(