a b+@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___indentsresultself __class__`/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/Cython/TestUtils.pyr szNodeTypeWriter.__init__cCs|js d}n0|jd}|ddur4d|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__rrrrr 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_dSN)r listing_file echo_filerrrrsetUp1szCythonTest.setUpcCs|jt_|jt_dSr*)r+rr,rrrrtearDown6szCythonTest.tearDownc Cst|ts|d}t|ts(|d}tt||D]$\}\}}|||d|||fq6|t|t|dd|d|fdS)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%)rexpectedridx expected_lineZ result_linerrr assertLines:s     zCythonTest.assertLinescCst}|||jjSr*)rwriterlines)rtreewriterrrr codeToLinesFs zCythonTest.codeToLinescCsd||S)Nr")r%r?)rr=rrr codeToStringKszCythonTest.codeToStringc Cst||}t|d}tt||D]$\}\}}|||d|||fq&|t|t|dd||fdS)Nr"zLine %d: Got: %s Exp: %sr/)r?rr2r3r4r5r6r%)rr7 result_treeZ result_linesZexpected_linesr8liner9rrr assertCodeNs  zCythonTest.assertCodecCs|t||dd|dS)Nz"Path '%s' not found in result tree)ZassertNotEqualr find_first)rpathrArrrassertNodeExistsYszCythonTest.assertNodeExistsNcCsV|dur i}|durg}|}|dr:|tdd}|dd}t||||dS)zNSimply create a tree fragment using the name of the test-case in parse errors.Nz __main__.._)pipeline)id startswithr6replacer)rcodepxdsrIrrrrfragment]s  zCythonTest.fragmentcCst|Sr*)r()rr&rrrr(iszCythonTest.treetypesc CsVz||d|Wn8|yP}z |t|||WYd}~Sd}~00dS)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 assertTruer0)rfuncexc_typeerrr should_faills zCythonTest.should_failc Cs@z|WSty:}z|t|WYd}~n d}~00dS)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) ExceptionrPstr)rrRexcrrrshould_not_failwszCythonTest.should_not_fail)NN)rrr r-r.r:r?r@rCrFrOr(rVrUrYrrrrr)/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. NcCs0|dur i}|||j}|D] }||}q|Sr*)rOr&)rrIZpyxrNr=Trrr run_pipelines  zTransformTest.run_pipeline)N)rrr __doc__r\rrrrrZsrZc@seZdZddZejZdS)TreeAssertVisitorcCs|j}d|vr>|dD]&}t||durt|jd|qd|vrv|dD]&}t||durNt|jd|qN|||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 rDrerrorposr)rrr_rErrrvisit_CompilerDirectivesNodes"   z.TreeAssertVisitor.visit_CompilerDirectivesNodeN)rrr rbr Zrecurse_to_childrenrrrrrr^sr^c CsN|durt}g}d}t|}z|}W|n |0~z|D]}|dddkr|ddtjj }tj ||}tj tj |st tj ||dur|d}}|t|d}qH|dur||qH|rH|dsH|dvrH||qHW|dur@|n|dur>|0|d |fS)Nz######/r')z"""z'''r#)tempfilemkdtempopen readlinesclosestriprLosrEsepr%existsdirnamemakedirsr;lstriprKr) Z tree_filedirheaderZcur_filefr<rBfilenamerErrrunpack_source_trees<        rv)N) __future__rrlZunittestrfZCompilerrrZCompiler.TreeFragmentrrZCompiler.Visitorrr r r r(ZTestCaser)rZr^rvrrrrs     R"