ó ¦V]c@@sddlmZddlZddlZddlZddlmZddlmZddlm Z m Z ddl m Z m Z ddlmZd e fd „ƒYZd „Zd ejfd „ƒYZdefd„ƒYZde fd„ƒYZdd„ZdS(i(tabsolute_importNi(tErrors(t CodeWriter(t TreeFragmenttstrip_common_indent(t TreeVisitortVisitorTransform(tTreePathtNodeTypeWritercB@seZd„Zd„ZRS(cC@s)tt|ƒjƒd|_g|_dS(Ni(tsuperRt__init__t_indentstresult(tself((s/lib/python2.7/site-packages/Cython/TestUtils.pyR s cC@s§|jsd}n;|jd}|ddk rCd|dd!}n |d}|jjd|jd||jjfƒ|jd7_|j|ƒ|jd8_dS( Nu(root)iÿÿÿÿiu%s[%d]iiu u%s: %s(t access_pathtNoneR tappendR t __class__t__name__t visitchildren(R tnodetnamettip((s/lib/python2.7/site-packages/Cython/TestUtils.pyt visit_Nodes     (Rt __module__R R(((s/lib/python2.7/site-packages/Cython/TestUtils.pyRs cC@s4tƒ}|j|ƒdjdg|jdgƒS(sÊ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.u u(RtvisittjoinR (troottw((s/lib/python2.7/site-packages/Cython/TestUtils.pyt treetypes%s  t CythonTestcB@steZd„Zd„Zd„Zd„Zd„Zd„Zd„Zd d d„Z d„Z e d „Z d „ZRS( cC@s,tj|_tj|_dt_t_dS(N(Rt listing_filet echo_fileR(R ((s/lib/python2.7/site-packages/Cython/TestUtils.pytsetUp1s  cC@s|jt_|jt_dS(N(RRR (R ((s/lib/python2.7/site-packages/Cython/TestUtils.pyttearDown6s cC@sÍt|tƒs!|jdƒ}nt|tƒsB|jdƒ}nxItt||ƒƒD]2\}\}}|j||d|||fƒqXW|jt|ƒt|ƒddj|ƒdj|ƒfƒdS(sHChecks that the given strings or lists of strings are equal line by lineu sLine %d: Exp: %s Got: %ss%Unmatched lines. Got: %s Expected: %ss N(t isinstancetlisttsplitt enumeratetzipt assertEqualtlenR(R texpectedR tidxt expected_linet result_line((s/lib/python2.7/site-packages/Cython/TestUtils.pyt assertLines:s( cC@s tƒ}|j|ƒ|jjS(N(RtwriteR tlines(R ttreetwriter((s/lib/python2.7/site-packages/Cython/TestUtils.pyt codeToLinesFs  cC@sdj|j|ƒƒS(Ns (RR3(R R1((s/lib/python2.7/site-packages/Cython/TestUtils.pyt codeToStringKscC@s¦|j|ƒ}t|jdƒƒ}xItt||ƒƒD]2\}\}}|j||d|||fƒq:W|jt|ƒt|ƒddj|ƒ|fƒdS(Ns sLine %d: Got: %s Exp: %ss%Unmatched lines. Got: %s Expected: %s(R3RR%R&R'R(R)R(R R*t result_treet result_linestexpected_linesR+tlineR,((s/lib/python2.7/site-packages/Cython/TestUtils.pyt assertCodeNs( cC@s'|jtj||ƒdd|ƒdS(Ns"Path '%s' not found in result tree(tassertNotEqualRt find_firstR(R tpathR5((s/lib/python2.7/site-packages/Cython/TestUtils.pytassertNodeExistsYscC@s€|dkri}n|dkr*g}n|jƒ}|jdƒrX|tdƒ}n|jddƒ}t|||d|ƒS(sNSimply create a tree fragment using the name of the test-case in parse errors.s __main__.t.t_tpipelineN(Rtidt startswithR)treplaceR(R tcodetpxdsR@R((s/lib/python2.7/site-packages/Cython/TestUtils.pytfragment]s     cC@s t|ƒS(N(R(R R((s/lib/python2.7/site-packages/Cython/TestUtils.pyRiscC@sMy|ƒ|jd|ƒWn*|k rH}|jt||ƒƒ|SXdS(s“Calls "func" and fails if it doesn't raise the right exception (any exception by default). Also returns the exception in question. s Expected an exception of type %rN(tfailt assertTrueR#(R tfunctexc_typete((s/lib/python2.7/site-packages/Cython/TestUtils.pyt should_faills cC@s8y |ƒSWn&tk r3}|jt|ƒƒnXdS(s©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(t ExceptionRGtstr(R RItexc((s/lib/python2.7/site-packages/Cython/TestUtils.pytshould_not_failws N(RRR!R"R.R3R4R9R=RRFRRMRLRP(((s/lib/python2.7/site-packages/Cython/TestUtils.pyR/s       t TransformTestcB@seZdZdd„ZRS(s. 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. cC@sK|dkri}n|j||ƒj}x|D]}||ƒ}q1W|S(N(RRFR(R R@tpyxRER1tT((s/lib/python2.7/site-packages/Cython/TestUtils.pyt run_pipeline™s    N(RRt__doc__RRT(((s/lib/python2.7/site-packages/Cython/TestUtils.pyRQstTreeAssertVisitorcB@seZd„ZejZRS(cC@sÆ|j}d|kr_xG|dD]8}tj||ƒdkr tj|jd|ƒq q Wnd|krµxG|dD]8}tj||ƒdk rvtj|jd|ƒqvqvWn|j|ƒ|S(Nttest_assert_path_existss+Expected path '%s' not found in result treettest_fail_if_path_existss)Unexpected path '%s' found in result tree(t directivesRR;RRterrortposR(R RRYR<((s/lib/python2.7/site-packages/Cython/TestUtils.pytvisit_CompilerDirectivesNode§s    (RRR\Rtrecurse_to_childrenR(((s/lib/python2.7/site-packages/Cython/TestUtils.pyRV£s c C@sÀ|dkrtjƒ}ng}d}t|ƒ}z|jƒ}Wd|jƒX~z<x5|D]-}|d dkr(|jƒjdƒjƒjdtj j ƒ}tj j ||ƒ}tj j tj j |ƒƒsðtjtj j |ƒƒn|dk r|d}}|jƒnt|dƒ}q^|dk rD|j|ƒq^|jƒr^|jƒjdƒ r^|jƒd kr‹|j|ƒq‹q^q^WWd|dk r¬|jƒnX|dj |ƒfS( Nis#####t#t/Rs"""s'''t(s"""s'''(Rttempfiletmkdtemptopent readlinestclosetstripRCtosR<tsepRtexiststdirnametmakedirsR/tlstripRBR( t tree_filetdirtheadertcur_filetfR0R8tfilenameR<((s/lib/python2.7/site-packages/Cython/TestUtils.pytunpack_source_tree»s8    -    " (t __future__RRgtunittestRatCompilerRRtCompiler.TreeFragmentRRtCompiler.VisitorRRRRRtTestCaseRRQRVRRs(((s/lib/python2.7/site-packages/Cython/TestUtils.pyts    R"