B d+@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|d|_g|_dS)Nr)superr __init___indentsresult)self) __class__]/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/Cython/TestUtils.pyr szNodeTypeWriter.__init__cCs|js d}n0|jd}|ddk r4d|dd}n|d}|jd|jd||jjf|jd7_|||jd8_dS) Nz(root)z%s[%d]rz z%s: %s)Z access_pathrappendrr__name__ visitchildren)rnodenameZtiprrr visit_Nodes   zNodeTypeWriter.visit_Node)r __module__ __qualname__r r __classcell__rr)rrr s r cCs(t}||ddg|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.tearDownc Cst|ts|d}t|ts(|d}x6tt||D]$\}\}}|||d|||fq8W|t|t|dd|d|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}|||jjS)N)rwriterlines)rtreewriterrrr codeToLinesFs zCythonTest.codeToLinescCsd||S)Nr )r#r;)rr9rrr codeToStringKszCythonTest.codeToStringc Csx||}t|d}x6tt||D]$\}\}}|||d|||fq(W|t|t|dd||fdS)Nr zLine %d: Got: %s Exp: %sz%Unmatched lines. Got: %s Expected: %s)r;rr.r/r0r1r2r#)rr3 result_treeZ result_linesZexpected_linesr4liner5rrr assertCodeNs zCythonTest.assertCodecCs|t||dd|dS)Nz"Path '%s' not found in result tree)ZassertNotEqualr find_first)rpathr=rrrassertNodeExistsYszCythonTest.assertNodeExistsNcCsV|dkr i}|dkrg}|}|dr:|tdd}|dd}t||||dS)zNSimply create a tree fragment using the name of the test-case in parse errors.Nz __main__.._)pipeline)id startswithr2replacer)rcodepxdsrErrrrfragment]s  zCythonTest.fragmentcCst|S)N)r&)rr$rrrr&iszCythonTest.treetypesc CsPy||d|Wn2|k rJ}z|t|||Sd}~XYnXdS)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,)rfuncexc_typeerrr should_faills zCythonTest.should_failc Cs<y|Stk r6}z|t|Wdd}~XYnXdS)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) ExceptionrLstr)rrNexcrrrshould_not_failwszCythonTest.should_not_fail)NN)rrrr*r+r6r;r<r?rBrKr&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}x|D] }||}q W|S)N)rKr$)rrEpyxrJr9Trrr run_pipelines   zTransformTest.run_pipeline)N)rrr__doc__rYrrrrrVsrVc@seZdZddZejZdS)TreeAssertVisitorcCs|j}d|krBx2|dD]&}t||dkrt|jd|qWd|kr~x2|dD]&}t||dk rTt|jd|qTW|||S)NZtest_assert_path_existsz+Expected path '%s' not found in result treeZtest_fail_if_path_existsz)Unexpected path '%s' found in result tree) directivesr r@rerrorposr)rrr\rArrrvisit_CompilerDirectivesNodes z.TreeAssertVisitor.visit_CompilerDirectivesNodeN)rrrr_r Zrecurse_to_childrenrrrrrr[sr[c Cs8|dkrt}g}d}t|}z |}Wd|X~zx|D]}|dddkr|ddtjj }tj ||}tj tj |st tj ||dk r|d}}|t|d}qB|dk r||qB|rB|dsB|dkrB||qBWWd|dk r(|X|d |fS)Nz######/r%)z"""z'''r!)tempfilemkdtempopen readlinesclosestriprHosrAsepr#existsdirnamemakedirsr7lstriprGr) Z tree_filedirheaderZcur_filefr8r>filenamerArrrunpack_source_trees8         rs)N) __future__rriZunittestrcZCompilerrrZCompiler.TreeFragmentrrZCompiler.Visitorrr r r r&ZTestCaser'rVr[rsrrrrs     R"