NXc@sdZddlZddlZddlZddlZddlZddlZddlZddlZddlm Z ddlm Z m Z m Z m Z mZmZmZmZmZddlZddlmZmZddlmZmZmZmZmZddlmZm Z m!Z!e r5ddl"m#Z#nddl#m#Z#ej$e%Z&d Z'd e(fd YZ)d ej*fd YZ*dej+fdYZ,dej-fdYZ-dej.fdYZ/dej.fdYZ0dej1fdYZ2ej3dZ4dej e(fdYZ5dej6fdYZ6dej7fdYZ8de8fd YZ9dS(!sNose Plugin that supports IPython doctests. Limitations: - When generating examples for use as doctests, make sure that you have pretty-printing OFF. This can be done either by setting the ``PlainTextFormatter.pprint`` option in your configuration file to False, or by interactively disabling it with %Pprint. This is required so that IPython output matches that of normal Python, which is used by doctest for internal execution. - Do not rely on specific prompt numbers for results (such as using '_34==True', for example). For IPython tests run via an external process the prompt numbers may be different, and IPython tests run as normal python code won't even have these special _NN variables set at all. iN(t getmodule( tREPORTING_FLAGStREPORT_ONLY_FIRST_FAILUREt_unittest_reportflagst DocTestRunnert_extract_future_flagstpdbt_OutputRedirectingPdbt_exception_tracebackt linecache(tdocteststPlugin(tanypt getpackaget test_addresst resolve_namettolist(t builtin_modtPY3tgetcwd(tStringIOcCs tjj|djdkS(sReturn whether the given filename is an extension module. This simply checks that the extension is either .so or .pyd. is.sos.pyd(s.sos.pyd(tostpathtsplitexttlower(tfilename((sX/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/testing/plugin/ipdoctest.pytis_extension_moduleBst DocTestSkipcBs&eZdZdZdZdZRS(s*Object wrapper for doctests to be skipped.s/Doctest to skip. >>> 1 #doctest: +SKIP cCs ||_dS(N(tobj(tselfR((sX/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/testing/plugin/ipdoctest.pyt__init__QscCs0|dkrtjSttj|d|SdS(Nt__doc__R(Rtds_skiptgetattrtobjectt__getattribute__(Rtkey((sX/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/testing/plugin/ipdoctest.pyR#Ts (t__name__t __module__RR RR#(((sX/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/testing/plugin/ipdoctest.pyRJs t DocTestFindercBseZdZdZRS(cCs|dkrtStj|r/|j|jkStj|rN|j|jkStj |rm|j|jkStj |r|j|j j jkStj |dk r|tj |kSt|dr|j|jkSt|trtStj|rtStd|dS(sY Return true if the given object is defined in the given module. R&s*object must be a class or function, got %rN(tNonetTruetinspectt isfunctiont__dict__t __globals__t isbuiltinR%R&tisclasstismethodt__self__t __class__Rthasattrt isinstancetpropertytismethoddescriptortFalset ValueError(RtmoduleR"((sX/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/testing/plugin/ipdoctest.pyt _from_module^s& cCst|drt|}ntjj||||||||ddlm}m} m} tj|r|j rx||j j D]h\} } d|| f} || s| | r|j || r|j|| | ||||qqWntj|r|j rx|j j D]\} } t | trKt|| } nt | trot|| j} ntj| stj| stj| st | tr|j || rd|| f} |j|| | ||||qqWndS(sm Find tests for the given object and any contained objects, and add them to `tests`. t skip_doctesti(t isroutineR/tismodules%s.%sN(R3RtdoctestR't_findR*R<R/R=t_recurseR,titemsR:R4t staticmethodR!t classmethodt__func__R+R0R5(RttestsRtnameR9t source_linestglobstseenR<R/R=tvalnametvaltvalname1((sX/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/testing/plugin/ipdoctest.pyR?~s2 (R%R&R:R?(((sX/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/testing/plugin/ipdoctest.pyR'\s tIPDoctestOutputCheckercBs&eZdZejdZdZRS(sSecond-chance checker with support for random tests. If the default comparison doesn't pass, this checker looks in the expected output string for flags that tell us to ignore the output. s #\s*random\s+cCs<tjj||||}| r8|jj|r8tS|S(sCheck output, accepting special markers embedded in the output. If the output didn't pass the default validation but the special string '#random' is included, we accept it.(R>t OutputCheckert check_outputt random_retsearchR)(Rtwanttgott optionflagstret((sX/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/testing/plugin/ipdoctest.pyROs  (R%R&RtretcompileRPRO(((sX/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/testing/plugin/ipdoctest.pyRMst DocTestCasecBsDeZdZdddddddZdZdZdZRS(sNProxy for DocTestCase: provides an address() method that returns the correct address for the doctest case. Otherwise acts as a proxy to the test case. To provide hints for address(), an obj may also be passed -- this will be used as the test object for purposes of determining the test address, if it is provided. it_c Cs||_tjj||d|d|d|d|||_||_||_|j|_||_ ||_ t d|d|dt }||_ tjj|j|_dS(NRTtsetUpttearDowntcheckertverbose(t _result_varR RXRt_dt_optionflagst _dt_checkert_dt_testRHt_dt_test_globs_orit _dt_setUpt _dt_tearDowntIPDocTestRunnerR7t _dt_runnerRRtdirnameRt_ori_dir( RttestRTRZR[R\Rt result_vartrunner((sX/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/testing/plugin/ipdoctest.pyRs          c Cs|j}|j}tj}t}|j}|t@sD|tO}nzNt}t j |j dd|_ |j |d|jdt\}}Wd|t_t j |X|r|j|j|jndS(Nt-iFtoutt clear_globs(RaRftsyststdoutRR_RRRRtchdirRhtDIVIDERtruntwriteR7tfailureExceptiontformat_failuretgetvalue( RRiRktoldtnewRTtcurdirtfailuresttries((sX/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/testing/plugin/ipdoctest.pytrunTests"          cCst|jjdtr}i|_|jjtjtjj|jjtjj ddt tjdtExampleRtsource(RRRRtexc_msgtlinenotindenttoptions((sX/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/testing/plugin/ipdoctest.pyRMs"N(R%R&RR(R(((sX/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/testing/plugin/ipdoctest.pyRJs tIPDocTestParsercBseZdZdZdZdZdZdZej eeeeefej ej BZ ej eeeeefej ej BZ ej dZej dZdZd d Zed Zd ZRS( s A class used to parse strings containing doctest examples. Note: This is a version modified to properly recognize IPython input and convert any IPython examples into valid Python ones. s>>>s\.\.\.s In\ \[\d+\]:s\ \ \ \.\.\.+:s] # Source consists of a PS1 line followed by zero or more PS2 lines. (?P (?:^(?P [ ]*) (?P %s) .*) # PS1 line (?:\n [ ]* (?P %s) .*)*) # PS2 lines \n? # a newline # Want consists of any non-blank lines that do not start with PS1. (?P (?:(?![ ]*$) # Not a blank line (?![ ]*%s) # Not a line starting with PS1 (?![ ]*%s) # Not a line starting with PS2 .*$\n? # But any other line )*) s#\s*all-random\s+s#\s*ipdoctest:\s*EXTERNALcCs?tjj|}t|jdkr7tj|S|SdS(s/Convert input IPython source into valid Python.iN(Rtinput_transformer_managerttransform_celltlent splitlinest prefilter(RRtblock((sX/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/testing/plugin/ipdoctest.pytip2pys sc Cs/|j}|j|}|dkrYdjg|jdD]}||^q=}ng}d\}}|jj|rd}nd}t} t|jj |} | rt j } n?t|j j |} |j j|rt} n t} t} x| D]} |j||| j!||jd|| j7}|j| ||| \} }}}||7}| tkrt|t j<|d7}n|j| s|j| | ||d|d|t| jdd|n||jd| j| j7}| j}qW|j|||S( s= Divide the given string into examples and intervening text, and return them as a list of alternating Examples and strings. Line numbers for the Examples are 0-based. The optional argument `name` is a name identifying this string, and is only used for error messages. is s # randomtRRR(ii(t expandtabst _min_indenttjointsplitt _RANDOM_TESTRQR7tlistt_EXAMPLE_RE_PYtfinditerR>Rt_EXAMPLE_RE_IPt _EXTERNAL_IPRRR)tappendtstarttcountt_parse_exampletNORMALIZE_WHITESPACEt_IS_BLANK_OR_COMMENTRtgrouptend(RtstringRFt min_indenttltoutputtcharnoRt random_markerRttermsRtmRRRRR((sX/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/testing/plugin/ipdoctest.pytparsesF  2     $     %cCst|jd}|jdjd}|jd}|jd}t|} |j||||| |r|j|dd||||ndjg|D]} | || d^q} |r|j| } n|jd} | jd} t| dkr1tjd | d r1| d =n|j| d|||t|tj d d | d | d (R t configuretpluginsRFRRRt extensionR>t DocTestParserRR'tfinderRMR\R(RHt extraglobs(RRtconfigtp((sX/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/testing/plugin/ipdoctest.pyRts!    cCsxtjj|\}}tjj|d}z5tjj|t|}t|j|}Wdtjj X|S(Ni( RRRRRoRt __import__RtloadTestsFromModuleR(RRtbpathtmodtmodnameR9RE((sX/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/testing/plugin/ipdoctest.pytloadTestsFromExtensionModules ccs|j|js&tjd|dS|jj|d|jd|j}|sTdStj tj B}|j |j }|dd kr|d }nxL|D]D}|j sqn|js||_nt|d|d |jVqWdS( NsDoctest doesn't want module %sRHRis.pycs.pyoiRTR\(s.pycs.pyo(tmatchesR%tlogtdebugRtfindRHRR>RtELLIPSIStsortt__file__RRRXR\(RR9RERTt module_fileRi((sX/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/testing/plugin/ipdoctest.pyRs(         c cst|r.x|j|D] }|VqWn|jrt|j|jrtjj|}t|}z|j }Wd|j X|j j |di|d6d|d|dd}|j rt|VqtVndS(NRHRRFRRi(RRRR tendswithRRtbasenametopentreadtcloseRt get_doctestRRR7(RRttRFtdhtdocRi((sX/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/testing/plugin/ipdoctest.pytloadTestsFromFiles      ( R%R&RRFR)tenabledRtenvironRRRRR(((sX/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/testing/plugin/ipdoctest.pyRYs   tIPythonDoctestcBs;eZdZdZeZdZejdZ dZ RS(s=Nose Plugin that supports doctests in extension modules. t ipdoctestc cstjtjB}|jj|dt|}|rxL|D]A}t|jdkr\q;nt|d|d|d|j Vq;WndS(sbLook for doctests in the given object, which will be a function, method or class. R9iRRTR\N( R>RRRRRRRRXR\(RRtparentRTR Ri((sX/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/testing/plugin/ipdoctest.pytmakeTests c Cstj||||jdddddd|jdtdd |jd dd dd dd |jd}|dk r|jd t|ndS(Ns--ipdoctest-testsRRRtipdoctest_testsRtNOSE_IPDOCTEST_TESTSRsAlso look for doctests in test modules. Note that classes, methods and functions should have either doctests or non-doctest tests, not both. [NOSE_IPDOCTEST_TESTS]s--ipdoctest-extensionRtipdoctest_extensionsNAlso look for doctests in files with this extension [NOSE_IPDOCTEST_EXTENSION]tNOSE_IPDOCTEST_EXTENSION(R RRRR)R(RR(RRRR((sX/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/testing/plugin/ipdoctest.pyRs cCstj|||g|jjD]}|jdkr |^q |j_|j|_t|j|_t |_ t d|j |_ t |_d|_d|_dS(NR>R(R RRRFR RRRRRRR'RRMR\R(RHR(RRRR((sX/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/testing/plugin/ipdoctest.pyRs!    ( R%R&RRFR)RR RRRR(((sX/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/testing/plugin/ipdoctest.pyRs  (:RR>R*tloggingRRVRot tracebacktunittestRRRRRRRRRR t nose.coretnoset nose.pluginsR R t nose.utilR R RRRtIPython.utils.py3compatRRRtioRt getLoggerR%RRR"RR'RNRMRXRRRRRtregister_optionflagRReRtDoctestRR(((sX/data/av2000/mvv/env_mvv/lib/python2.7/site-packages/IPython/testing/plugin/ipdoctest.pyts>        @ ( T} m