3 "LX+@sddlmZddlZddlZddlZddlmZddlmZddlm Z m Z ddl m Z m Z ddlmZGd d d e Zd d ZGd ddejZGdddeZGddde ZdddZdS))absolute_importN)Errors) CodeWriter) TreeFragmentstrip_common_indent) TreeVisitorVisitorTransform)TreePathcs$eZdZfddZddZZS)NodeTypeWritercstt|jd|_g|_dS)Nr)superr __init___indentsresult)self) __class__`/home/psgendb/BIRCHDEV/lib-linux-x86_64/python/lib64/python3.6/site-packages/Cython/TestUtils.pyr szNodeTypeWriter.__init__cCs|js d}n0|jd}|ddk r4d|dd}n|d}|jjd|jd||jjf|jd7_|j||jd8_dS) Nz(root)rz%s[%d]z z%s: %s) access_pathrappendrr__name__ visitchildren)rnodenameZtiprrr visit_Nodes   zNodeTypeWriter.visit_Node)r __module__ __qualname__r r __classcell__rr)rrr s r cCs(t}|j|djdg|jdgS)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)rootwrrr treetypes%s r'c@sjeZdZddZddZddZddZd d Zd d Zd dZ dddZ ddZ e fddZ ddZdS) CythonTestcCs tj|_tj|_dt_t_dS)N)r listing_file echo_file)rrrrsetUp1szCythonTest.setUpcCs|jt_|jt_dS)N)r)rr*)rrrrtearDown6szCythonTest.tearDowncCst|ts|jd}t|ts(|jd}x6tt||D]$\}\}}|j||d|||fq8W|jt|t|ddj|dj|fdS)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$)rexpectedridx expected_lineZ result_linerrr assertLines:s    zCythonTest.assertLinescCst}|j||jjS)N)rwriterlines)rtreewriterrrr codeToLinesFs zCythonTest.codeToLinescCsdj|j|S)Nr!)r$r<)rr:rrr codeToStringKszCythonTest.codeToStringcCsx|j|}t|jd}x6tt||D]$\}\}}|j||d|||fq(W|jt|t|ddj||fdS)Nr!zLine %d: Got: %s Exp: %sz%Unmatched lines. Got: %s Expected: %s)r<rr/r0r1r2r3r$)rr4 result_treeZ result_linesZexpected_linesr5liner6rrr assertCodeNs zCythonTest.assertCodecCs|jtj||dd|dS)Nz"Path '%s' not found in result tree)ZassertNotEqualr find_first)rpathr>rrrassertNodeExistsYszCythonTest.assertNodeExistsNcCsV|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 startswithr3replacer)rcodepxdsrFrrrrfragment]s  zCythonTest.fragmentcCst|S)N)r')rr%rrrr'iszCythonTest.treetypescCsNy||jd|Wn0|k rH}z|jt|||Sd}~XnXdS)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)failZ assertTruer-)rfuncexc_typeerrr should_faills zCythonTest.should_failcCs<y|Stk r6}z|jt|WYdd}~XnXdS)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) ExceptionrMstr)rrNexcrrrshould_not_failwszCythonTest.should_not_fail)NN)rrrr+r,r7r<r=r@rCrLr'rRrQrUrrrrr(/s    r(c@seZdZdZdddZdS) 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. NcCs4|dkr i}|j||j}x|D] }||}q W|S)N)rLr%)rrFpyxrKr:Trrr run_pipelines   zTransformTest.run_pipeline)N)rrr__doc__rYrrrrrVsrVc@seZdZddZejZdS)TreeAssertVisitorcCs|j}d|krBx2|dD]&}tj||dkrtj|jd|qWd|kr~x2|dD]&}tj||dk rTtj|jd|qTW|j||S)Ntest_assert_path_existsz+Expected path '%s' not found in result treetest_fail_if_path_existsz)Unexpected path '%s' found in result tree) directivesr rArerrorposr)rrr^rBrrrvisit_CompilerDirectivesNodes z.TreeAssertVisitor.visit_CompilerDirectivesNodeN)rrrrar recurse_to_childrenrrrrrr[sr[c Cs:|dkrtj}g}d}t|}z |j}Wd|jX~zx|D]}|dddkr|jjdjjdtjj }tjj ||}tjj tjj |stj tjj ||dk r|d}}|jt|d}qB|dk r|j|qB|jo|jjd rB|jd krB|j|qBWWd|dk r*|jX|dj |fS) Nz######/r&"""'''r")rfrg)tempfilemkdtempopen readlinesclosestriprIosrBsepr$existsdirnamemakedirsr8lstriprHr) Z tree_filedirheaderZcur_filefr9r?filenamerBrrrunpack_source_trees8         rx)N) __future__rrnZunittestrhCompilerrrZCompiler.TreeFragmentrrZCompiler.Visitorrr r r r'ZTestCaser(rVr[rxrrrrs     R"