­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ 3 \Ǘ!@spdZddddddddd d d d d ddddddddddddddddddd d!g!Zd"d#lZd"d#lZd"d#lZd"d#lZd"d#lZd"d#lZd"d#lZd"d#l Z d"d#l Z d"d#l Z d"d#l Z d"d$l mZd"d%lmZed&d'ZiZd(dZedZedZedZedZedZedZeeBeBeBeBeBZed Zed Zed Zed ZedZeeBeBeBeBZ d)Z!d*Z"d+d,Z#dpd.d/Z$d0d1Z%dqd3d4Z&d5d6Z'Gd7d8d8eZ(d9d:Z)d;d<Z*d=d>Z+Gd?d@d@ej,Z-dAdBZ.GdCddZ/GdDddZ0GdEddZ1GdFddZ2GdGddZ3GdHddZ4GdIdde5Z6GdJdde5Z7GdKdde3Z8d#a9drdNdZ:dLd#d#d#d#dLd"d#dMe1d#f dOdZ;dsdQdZGdSdTdTe j?Z@GdUdVdVe@ZAGdWdXdXe jBZCdtdYdZDGdZd[d[e@ZEdLd#d#e1d#fd\d]ZFd^dZGd_dZHd`dZIdudad ZJdvdbdcZKdwddd!ZLGdedfdfZMeMdgdhdidjdkdlZNdmdnZOePdokrle jQeOd#S)xzreStructuredText enregister_optionflagDONT_ACCEPT_TRUE_FOR_1DONT_ACCEPT_BLANKLINENORMALIZE_WHITESPACEELLIPSISSKIPIGNORE_EXCEPTION_DETAILCOMPARISON_FLAGS REPORT_UDIFF REPORT_CDIFF REPORT_NDIFFREPORT_ONLY_FIRST_FAILUREREPORTING_FLAGS FAIL_FASTExampleDocTest DocTestParser DocTestFinder DocTestRunner OutputCheckerDocTestFailureUnexpectedException DebugRunnertestmodtestfilerun_docstring_examples DocTestSuite DocFileSuiteset_unittest_reportflagsscript_from_examples testsource debug_srcdebugN)StringIO) namedtuple TestResultszfailed attemptedcCstj|dtt>S)N)OPTIONFLAGS_BY_NAME setdefaultlen)namer+/usr/lib64/python3.6/doctest.pyrsz z...cCs<d}x2tjD](}|j|d}|tt|kr ||jO}q W|S)Nr") __future__Zall_feature_namesgetgetattrZ compiler_flag)globsflagsZfnameZfeaturer+r+r,_extract_future_flagss   r2cCsVtj|r|St|tr,t|ttdgS|dkrJtjtj |j dSt ddS)N*__name__z"Expected a module, string, or None) inspectismodule isinstancestr __import__globalslocalssysmodules _getframe f_globals TypeError)moduleZdepthr+r+r,_normalize_modules  rCc Cs|r\t|d}t||}t|dddk r\t|jdr\|jj|}|j|}|jtj d|fSt ||d}|j |fSQRXdS)N __loader__get_data )encoding) rC_module_relative_pathr/hasattrrErFdecodereplaceoslinesepopenread)filenamepackagemodule_relativerHZ file_contentsfr+r+r,_load_testfiles     rUcCstjd|d|S)Nz (?m)^(?!$) )resub)sindentr+r+r,_indentsr\cCs*t}|\}}}tj||||d|jS)N)file)r# tracebackprint_exceptiongetvalue)exc_infoZexcoutexc_typeZexc_valZexc_tbr+r+r,_exception_tracebacks rcc@seZdZddZdddZdS) _SpoofOutcCs&tj|}|r"|jd r"|d7}|S)NrG)r#r`endswith)selfresultr+r+r,r`s z_SpoofOut.getvalueNcCs|j|tj|dS)N)seekr#truncate)rfsizer+r+r,ris z_SpoofOut.truncate)N)r5 __module__ __qualname__r`rir+r+r+r,rds rdcCst|kr||kS|jt}dt|}}|d}|rR|j|rNt|}|d=ndS|d}|r|j|r||t|8}|d=ndS||krdSx2|D]*}|j|||}|dkrdS|t|7}qWdS)Nr"Fr&Trm)ELLIPSIS_MARKERsplitr) startswithrefind)wantgotZwsstartposendposwr+r+r,_ellipsis_matchs0     rwcCs|j}|rd|SdSdS)Nz# #)rstrip)liner+r+r, _comment_line9sr{cCshdt|}}|jd}|dkr$|}|jdd|}|dkr>|}|jdd|}|dkr\|d}|||S)Nr"rG:.r&)r)rqrfind)msgstartendir+r+r,_strip_exception_detailsAs  rc@s.eZdZddZd ddZddZdd ZdS) _OutputRedirectingPdbcCs(||_d|_tjj||ddd|_dS)NFT)stdoutnosigintr&)_OutputRedirectingPdb__out$_OutputRedirectingPdb__debugger_usedpdbPdb__init__Z use_rawinput)rfoutr+r+r,rasz_OutputRedirectingPdb.__init__NcCs*d|_|dkrtjj}tjj||dS)NT)rr=r?f_backrr set_trace)rfframer+r+r,ris z_OutputRedirectingPdb.set_tracecCs|jrtjj|dS)N)rrr set_continue)rfr+r+r,rosz"_OutputRedirectingPdb.set_continuec Gs.tj}|jt_ztjj|f|S|t_XdS)N)r=rrrrtrace_dispatch)rfargs save_stdoutr+r+r,rus z$_OutputRedirectingPdb.trace_dispatch)N)r5rkrlrrrrr+r+r+r,r[s rcCstj|std||jdr(tdtjj|jd}t |drXtjj|j d}n|j dkrt t jdkrt jddkrtjjt jdd}qtj}nFt |drx,|jD]"}tjj||}tjj|r|SqWtd |j tjj||S) NzExpected a module: %r/z1Module-relative files may not have absolute paths__file__r"__main____path__zBCan't resolve paths relative to the module %r (it has no __file__))r6r7rArp ValueErrorrMpathjoinrorJrr5r)r=argvcurdirrexists)rBZ test_pathZbasedirZ directoryZfullpathr+r+r,rIs&         rIc@s&eZdZd ddZddZddZdS) rNr"cCsz|jds|d7}|r*|jd r*|d7}|dk rF|jd rF|d7}||_||_||_||_|dkrji}||_||_dS)NrG)resourcerrlinenor[optionsexc_msg)rfrrrrrr[rr+r+r,rs zExample.__init__cCs\t|t|k rtS|j|jkoZ|j|jkoZ|j|jkoZ|j|jkoZ|j|jkoZ|j|jkS)N)typeNotImplementedrrrrr[rr)rfotherr+r+r,__eq__s     zExample.__eq__cCst|j|j|j|j|jfS)N)hashrrrrr[r)rfr+r+r,__hash__szExample.__hash__)Nr"r"N)r5rkrlrrrr+r+r+r,rs#  c@s4eZdZddZddZddZddZd d Zd S) rcCs,||_||_|j|_||_||_||_dS)N)examples docstringcopyr0r*rQr)rfrr0r*rQrrr+r+r,r s  zDocTest.__init__cCsRt|jdkrd}n"t|jdkr(d}ndt|j}d|jj|j|j|j|fS)Nr"z no examplesr&z 1 examplez %d examplesz<%s %s from %s:%s (%s)>)r)r __class__r5r*rQr)rfrr+r+r,__repr__szDocTest.__repr__cCs\t|t|k rtS|j|jkoZ|j|jkoZ|j|jkoZ|j|jkoZ|j|jkoZ|j|jkS)N)rrrrr0r*rQr)rfrr+r+r,r#s     zDocTest.__eq__cCst|j|j|j|jfS)N)rrr*rQr)rfr+r+r,r.szDocTest.__hash__cCs:t|tstS|j|j|jt|f|j|j|jt|fkS)N)r8rrr*rQrid)rfrr+r+r,__lt__2s zDocTest.__lt__N)r5rkrlrrrrrr+r+r+r,rs   c@seZdZejdejejBZejdejejBejBZ ejdj Z dddZ ddZ dd d Zd d Zejd ejZddZejdejZddZddZddZdS)ra # Source consists of a PS1 line followed by zero or more PS2 lines. (?P (?:^(?P [ ]*) >>> .*) # PS1 line (?:\n [ ]* \.\.\. .*)*) # PS2 lines \n? # Want consists of any non-blank lines that do not start with PS1. (?P (?:(?![ ]*$) # Not a blank line (?![ ]*>>>) # Not a line starting with PS1 .+$\n? # But any other line )*) a # Grab the traceback header. Different versions of Python have # said different things on the first traceback line. ^(?P Traceback\ \( (?: most\ recent\ call\ last | innermost\ last ) \) : ) \s* $ # toss trailing whitespace on the header. (?P .*?) # don't blink: absorb stuff until... ^ (?P \w+ .*) # a line *starts* with alphanum. z ^[ ]*(#.*)?$c s|j}|j|dkr8djfdd|jdD}g}d\}}x|jj|D]}|j|||j||jd||j7}|j |||\}}} } |j |s|jt || | |t |j d|d||jd|j|j7}|j}qRW|j||d|S)Nr"rGcsg|]}|dqS)Nr+).0l) min_indentr+r, ysz'DocTestParser.parse..r[)rr[r)r"r") expandtabs _min_indentrro _EXAMPLE_REfinditerappendrcount_parse_example_IS_BLANK_OR_COMMENTrr)groupr) rfstringr*outputZcharnormrrrrrr+)rr,parsems&     zDocTestParser.parsecCst|j|||||||S)N)r get_examples)rfrr0r*rQrr+r+r, get_doctests zDocTestParser.get_doctestcCsdd|j||DS)NcSsg|]}t|tr|qSr+)r8r)rxr+r+r,rsz.DocTestParser.get_examples..)r)rfrr*r+r+r,rs zDocTestParser.get_examplesc s t|jd|jdjd}|j||||j|dddd||djfdd|D}|jd }|jd}t|dkrtjd |d r|d=|j|d||t|djfd d|D}|jj|}|r|jd }nd}|j |||} || ||fS)Nr[rrGr&rWr}csg|]}|ddqS)rVNr+)rZsl)r[r+r,rsz0DocTestParser._parse_example..rrz *$csg|]}|dqS)Nr+)rZwl)r[r+r,rsrrmrm) r)rro_check_prompt_blank _check_prefixrrXmatch _EXCEPTION_RE _find_options) rfrr*r source_linesrrr want_linesrrr+)r[r,rs$      zDocTestParser._parse_examplez#\s*doctest:\s*([^\n\'"]*)$c Csi}x|jj|D]z}|jdjddj}x^|D]V}|ddksR|ddtkrhtd|d||ft|dd}|ddk||<q2WqW|r|j|rtd|||f|S) Nr&,rWr"z+-z7line %r of the doctest for %s has an invalid option: %r+zSline %r of the doctest for %s has an option directive on a line with no example: %r)_OPTION_DIRECTIVE_RErrrLror'rr) rfrr*rrrZoption_stringsoptionflagr+r+r,rs  zDocTestParser._find_optionsz ^([ ]*)(?=\S)cCs2dd|jj|D}t|dkr*t|SdSdS)NcSsg|] }t|qSr+)r))rr[r+r+r,rsz-DocTestParser._min_indent..r") _INDENT_REfindallr)min)rfrZindentsr+r+r,rs zDocTestParser._min_indentc Csbx\t|D]P\}}t||dkr ||ddkr td||d||||d|fq WdS)NrVrDrWz8line %r of the docstring for %s lacks blank after %s: %rr&) enumerater)r)rflinesr[r*rrrzr+r+r,r s   z!DocTestParser._check_prompt_blankcCsDx>t|D]2\}}|r |j| r td||d||fq WdS)NzGline %r of the docstring for %s has inconsistent leading whitespace: %rr&)rrpr)rfrprefixr*rrrzr+r+r,rszDocTestParser._check_prefixN)r)r)r5rkrlrXcompile MULTILINEVERBOSErDOTALLrrrrrrrrrrrrrr+r+r+r,r=s       ' 3c@sJeZdZdeddfddZdddZdd Zd d Zd d ZddZ dS)rFTcCs||_||_||_||_dS)N)_parser_verbose_recurse_exclude_empty)rfverboseparserrecurse exclude_emptyr+r+r,r1szDocTestFinder.__init__Nc CsP|dkr.t|dd}|dkr.tdt|f|dkrr)r/rrr6 getmoduleZ getsourcefilerAZgetfile linecachegetlines__dict__rupdate_findsort) rfobjr*rBr0 extraglobsr]rtestsr+r+r,rqGsJ$          zDocTestFinder.findcCs|dkr dStj|dk r(|tj|kStj|r>|j|jkStj|r|t|dr\|jj}nt|drn|j}ndS|j |kStj |r|j |jkSt|dr|j |jkSt |t rdSt ddS)NT __objclass__rkz"object must be a class or function)r6r isfunctionr __globals__ZismethoddescriptorrJrrkr5isclassr8propertyr)rfrBobjectZobj_modr+r+r, _from_modules(            zDocTestFinder._from_modulec Cs(|jrtd|t||kr"dSd|t|<|j|||||}|dk rR|j|tj|r|jrx^|jj D]P\} } d|| f} tj tj | stj | rn|j || rn|j|| | ||||qnWtj|o|jrtxt|dij D]\} } t| ts tdt| ftj | p8tj | p8tj| p8t| tsNtdt| fd|| f} |j|| | ||||qWtj |r$|jr$x|jj D]\} } t| trt|| } t| trt|| j} tj | stj | st| tr|j || rd|| f} |j|| | ||||qWdS)NzFinding tests in %sr&z%s.%s__test__z5DocTestFinder.find: __test__ keys must be strings: %rz`DocTestFinder.find: __test__ values must be strings, functions, methods, classes, or modules: %rz%s.__test__.%s)rprintr _get_testrr6r7rritemsZ isroutineZunwraprrrr/r8r9rr staticmethod classmethod__func__r) rfrrr*rBrr0seentestZvalnamevalr+r+r,rsP                 zDocTestFinder._findc Cst|tr|}nJy,|jdkr"d}n|j}t|ts:t|}Wnttfk rXd}YnX|j||}|jrv| rvdS|dkrd}n*t|d|j}|dddkr|dd}|j j |||||S)NrrrVz.pycr&rm) r8r9__doc__rAAttributeError _find_linenorr/r5rr) rfrr*rBr0rrrrQr+r+r,rs(         zDocTestFinder._get_testcCsd}tj|rd}tj|rd|dkr(dStjdt|dd}x$t|D]\}}|j|rH|}PqHWtj|rt|j }tj |r|j }tj |r|j }tj|r|j}tj|rt|ddd}|dk r |dkr|dStjd}x(t|t|D]}|j||r|SqWdS)Nr"z^\s*class\s*%s\br5-co_firstlinenor&z(^|.*:)\s*\w*("|\'))r6r7rrXrr/rrZismethodrr__code__Z istracebacktb_frameZisframef_codeZiscoderanger))rfrrrZpatrrzr+r+r,r-s<          zDocTestFinder._find_lineno)NNNN) r5rkrlrrrqrrrrr+r+r+r,r(s  f?%c@seZdZddZdddZddZd d Zd d Zd dZddZ ddZ ddZ e j dZd ddZd!ddZd"ddZddZdS)#rr4FNr"cCsN|pt|_|dkrdtjk}||_||_||_d|_d|_i|_ t |_ dS)Nz-vr") r_checkerr=rr optionflagsoriginal_optionflagstriesfailures_name2ftrd_fakeout)rfcheckerrr r+r+r,rs  zDocTestRunner.__init__cCsH|jrD|jr.|dt|jdt|jn|dt|jddS)NzTrying: z Expecting: zExpecting nothing )rrrr\r)rfrrexampler+r+r, report_starts  zDocTestRunner.report_startcCs|jr|ddS)Nzok )r)rfrrrrsr+r+r,report_successszDocTestRunner.report_successcCs&||j|||jj|||jdS)N)_failure_headerr output_differencer )rfrrrrsr+r+r,report_failures zDocTestRunner.report_failurecCs$||j||dtt|dS)NzException raised: )rr\rc)rfrrrrar+r+r,report_unexpected_exceptionsz)DocTestRunner.report_unexpected_exceptioncCs|jg}|jrR|jdk r4|jdk r4|j|jd}nd}|jd|j||jfn|jd|jd|jf|jd|j}|jt|dj|S)Nr&?zFile "%s", line %s, in %szLine %s, in %szFailed example:rG)DIVIDERrQrrr*rr\r)rfrrrrrr+r+r,rs zDocTestRunner._failure_headercCsd}}|j}td\}}} |jj} xFt|jD]6\} } |jt@oL|dk} ||_| jrx:| jjD],\}}|r|j|O_qf|j|M_qfW|jt @rq2|d7}| s|j ||| d|j | f}y,t t | j|d|d|j|jjd}Wn4tk rYntj}|jjYnX|jj}|jjd|}|dkrh| | j||jr|}n|tj|ddd}| s|t|7}| jdkr| }nB| | j||jr|}n*|jt@r| t| jt||jr|}||kr| sZ|j||| |nT||kr0| s&|j ||| ||d7}n*|| krZ| sP|j!||| ||d7}n|r2|jt"@r2Pq2W||_|j#|||t$||S)Nr"rDr&zZsingler3rm)%r r r  check_outputrrr rrrrr*execrrr0debuggerrKeyboardInterruptr=rarr`rirrr^format_exception_onlyrcrrrrrrr_DocTestRunner__record_outcomer%)rfr compileflagsrrrr SUCCESSFAILUREZBOOMZcheck examplenumrquietZ optionflagrrQZ exceptionrsZoutcomerr+r+r,Z__runs                   zDocTestRunner.__runcCsL|jj|jd\}}||||f|j|j<|j|7_|j|7_dS)Nr")r"r")rr.r*rr)rfrrTtf2t2r+r+r,Z__record_outcomeuszDocTestRunner.__record_outcomez..+)\[(?P\d+)\]>$cCsV|jj|}|rF|jd|jjkrF|jjt|jd}|jjddS|j ||SdS)Nr*r%T)keepends) %_DocTestRunner__LINECACHE_FILENAME_RErrrr*rintr splitlinessave_linecache_getlines)rfrQmodule_globalsrrr+r+r,Z__patched_linecache_getliness  z*DocTestRunner.__patched_linecache_getlinesTc s||_|dkrt|j}tj|dkrVjdks@jdkrHj}nfdd}|jt_tj }t j }t |_ |j j|j j t _ tj|_|jt_tj}tjt_z|j|||St_|t _ tj||jt_|t_|r|jjddl}d|_XdS)Nzutf-8cs t|jd}j|dS)Nbackslashreplace)r9encodewrite)rZ)rHrr+r,rszDocTestRunner.run..outr")rr2r0r=rrHlowerr2rgettracerrrrresetrrr.*_DocTestRunner__patched_linecache_getlines displayhook__displayhook___DocTestRunner__runsettraceclearbuiltins_) rfrr"r clear_globsZ save_traceZsave_set_traceZsave_displayhookr<r+)rHrr,runs<      zDocTestRunner.runc Cs|dkr|j}g}g}g}d}}xd|jjD]V}|\}\} } || 7}|| 7}| dkrb|j|q.| dkrz|j|| fq.|j|q.W|r|rtt|d|jx|D]} td| qW|rtt|d|jx |D]\} } td| | fqW|rJt|jtt|d|jx(|D] \} \} } td| | | fq&W|rvt|dt|jd t||d |d |rtd |d n|rtdt||S)Nr"zitems had no tests:z zitems passed all tests:z %3d tests in %szitems had failures:z %3d of %3d in %sztests inzitems.z passed andzfailed.z***Test Failed***z failures.z Test passed.) rrrrrr)rrr%) rfrZnotestsZpassedZfailedZtotaltZtotalfrr*rTr'thingrr+r+r, summarizesP     zDocTestRunner.summarizecCsV|j}xJ|jjD]<\}\}}||krB||\}}||}||}||f||<qWdS)N)rr)rfrdr*rTr'r(r)r+r+r,merge s zDocTestRunner.merge)NNr")N)NNT)N)r5rkrlrrrrrrrr9r!rXrr+r6r?rArCr+r+r+r,ras: $ }   I 9c@s,eZdZddZddZddZddZd S) rcCst|jdddS)NASCIIr0)r9r1)rfrZr+r+r,_toAscii!szOutputChecker._toAsciicCs|j|}|j|}||kr dS|t@sH||fd kr8dS||fd krHdS|t@stjdtjtd|}tjdd|}||krdS|t@rd j|j }d j|j }||krdS|t @rt ||rdSd S) NTTrue 1 False 0 z (?m)^%s\s*?$rz (?m)^\s*?$rWF)rFrG)rHrI) rErrrXrYescapeBLANKLINE_MARKERrrrorrw)rfrrrsr r+r+r,r's0     zOutputChecker.check_outputcCs<|ttBtB@sdS|t@r dS|jddko:|jddkS)NFTrGr3)r r r r)rfrrrsr r+r+r,_do_a_fancy_difffs zOutputChecker._do_a_fancy_diffc Cs8|j}|t@stjdt|}|j|||r|jdd}|jdd}|t@rptj ||dd}t |dd}d}nZ|t @rtj ||dd}t |dd}d}n,|t @rtjtjd} t | j||}d }nd d |D}d |td j|S|o|r dt|t|fS|rdt|S|r0dt|SdSdS)Nz(?m)^[ ]*(?= )T)r*r3)nz#unified diff with -expected +actualz-context diff with expected followed by actual)Zcharjunkzndiff with -expected +actualcSsg|]}|jdqS)rG)ry)rrzr+r+r,rsz3OutputChecker.output_difference..zDifferences (%s): rzExpected: %sGot: %szExpected: %sGot nothing zExpected nothing Got: %szExpected nothing Got nothing )rrrrXrYrKrLr-r difflibZ unified_difflistr Z context_diffr ZDifferZIS_CHARACTER_JUNKZcomparer\r) rfrrsr rrrZ got_linesZdiffZkindZenginer+r+r,r}s6     zOutputChecker.output_differenceN)r5rkrlrErrLrr+r+r+r,rs?c@seZdZddZddZdS)rcCs||_||_||_dS)N)rrrs)rfrrrsr+r+r,rszDocTestFailure.__init__cCs t|jS)N)r9r)rfr+r+r,__str__szDocTestFailure.__str__N)r5rkrlrrPr+r+r+r,rs c@seZdZddZddZdS)rcCs||_||_||_dS)N)rrra)rfrrrar+r+r,rszUnexpectedException.__init__cCs t|jS)N)r9r)rfr+r+r,rPszUnexpectedException.__str__N)r5rkrlrrPr+r+r+r,rs c@s&eZdZd ddZddZddZdS) rNTcCs$tj||||d}|r |jj|S)NF)rr?r0r;)rfrr"rr>rr+r+r,r?.s zDebugRunner.runcCst|||dS)N)r)rfrrrrar+r+r,r4sz'DebugRunner.report_unexpected_exceptioncCst|||dS)N)r)rfrrrrsr+r+r,r7szDebugRunner.report_failure)NNT)r5rkrlr?rrr+r+r+r,rs[ TFc Cs|dkrtjjd}tj|s,td|f|dkr:|j}t|d} |rVt||d} n t ||d} x$| j ||||dD]} | j | qvW|r| j t dkr| a n t j| t| j| jS)Nrztestmod: module required; %r)r)rr )r0r)r=r>r.r6r7rAr5rrrrqr?rAmasterrCr%rr) rr*r0rreportr rraise_on_errorrfinderrunnerrr+r+r,rCs$E     c Cs|r| rtdt|||| p d\} }|dkr)rar=rr#r_r rZrr`rr?r2ZfailureExceptionformat_failurer`)rfrr[newr rVrrr+r+r,runTests  zDocTestCase.runTestcCsP|j}|jdkrd}n d|j}dj|jjddd}d|j|j|||fS)Nzunknown line numberz%sr}r&z:Failed doctest test for %s File "%s", line %s, in %s %srm)rarrr*rorQ)rferrrrZlnamer+r+r,rfs  zDocTestCase.format_failurecCs6|jt|j|jdd}|j|jdd|jdS)NF)r rr)r>)rdrr_r`r?rare)rfrVr+r+r,r!s B zDocTestCase.debugcCs|jjS)N)rar*)rfr+r+r,rszDocTestCase.idcCsPt|t|k rtS|j|jkoN|j|jkoN|j|jkoN|j|jkoN|j|jkS)N)rrrar_rbrcr`)rfrr+r+r,rs    zDocTestCase.__eq__cCst|j|j|j|jfS)N)rr_rbrcr`)rfr+r+r,rszDocTestCase.__hash__cCs,|jjjd}d|ddj|ddfS)Nr}z%s (%s)r&rmrm)rar*ror)rfr*r+r+r,rszDocTestCase.__repr__cCs d|jjS)Nz Doctest: )rar*)rfr+r+r,shortDescription szDocTestCase.shortDescription)r"NNN)r5rkrlrrdrerhrfr!rrrrrPrjr+r+r+r,r\gs  H r\c@s0eZdZddZddZddZddZeZd S) SkipDocTestCasecCs||_tj|ddS)N)rBr\r)rfrBr+r+r,r szSkipDocTestCase.__init__cCs|jddS)Nz-DocTestSuite will not work with -O2 and above)ZskipTest)rfr+r+r,rd szSkipDocTestCase.setUpcCsdS)Nr+)rfr+r+r, test_skip szSkipDocTestCase.test_skipcCs d|jjS)NzSkipping tests from %s)rBr5)rfr+r+r,rj sz SkipDocTestCase.shortDescriptionN)r5rkrlrrdrlrjrPr+r+r+r,rk s rkc@seZdZddZdS) _DocTestSuitecCsdS)Nr+)rfindexr+r+r,_removeTestAtIndex sz _DocTestSuite._removeTestAtIndexN)r5rkrlror+r+r+r,rm srmc Ks|dkrt}t|}|j|||d}| rPtjjdkrPt}|jt||S|j t}x\|D]T}t |j dkrxqd|j s|j }|dddkr|dd}||_ |jt|f|qdW|S) N)r0rr3r"rVz.pycr&rrm)rrCrqr=r1optimizermaddTestrkrr)rrQrr\) rBr0rZ test_finderrrsuiterrQr+r+r,r s(%  c@s(eZdZddZddZeZddZdS) DocFileCasecCsdj|jjjdS)Nr=r})rrar*ro)rfr+r+r,r` szDocFileCase.idcCs|jjS)N)rarQ)rfr+r+r,rc szDocFileCase.__repr__cCsd|jj|jj|fS)Nz2Failed doctest test for %s File "%s", line 0 %s)rar*rQ)rfrir+r+r,rfg szDocFileCase.format_failureN)r5rkrlrrrPrfr+r+r+r,rs^ srsc Ksx|dkri}n|j}|r(| r(tdt||||p6d\}}d|krN||d<tjj|}|j||||d} t| f|S)Nz8Package may only be specified for module-relative paths.zutf-8rr")rrrUrMrrWrrs) rrSrRr0rrHrdocr*rr+r+r, DocFileTestl s  rucOsHt}|jddr$t|jd|d<x|D]}|jt|f|q*W|S)NrSTrR)rmr.rCrqru)pathskwrrrr+r+r,r s 8  cCsg}xtj|D]x}t|trj|j|jdd |j}|r|jd|dd|jddd D7}q|dd|jddd D7}qWx|r|d dkr|jqWx|r|ddkr|jdqWdj |dS) Nr&z # Expected:cSsg|] }d|qS)z## r+)rrr+r+r,r sz(script_from_examples..rGcSsg|] }t|qSr+)r{)rrr+r+r,r srxr"rmrmrmrm) rrr8rrrrrropopr)rZrZpiecerrr+r+r,r s:  " csJt|}tj|}fdd|D}|s4td|d}t|j}|S)Ncsg|]}|jkr|qSr+)r*)rr')r*r+r,r* sztestsource..znot found in testsr")rCrrqrrr)rBr*rrtestsrcr+)r*r,r! s   cCst|}t|||dS)N)r debug_script)srcpmr0ryr+r+r,r 1 sc Csddl}|r|j}ni}|rvyt|||Wqttjd|jdd}|j|jdtjdYqXn|jddj d|||dS)Nr"r&T)rr3zexec(%r)) rrrrr=rarr5Z interactionr?)r{r|r0rpr+r+r,rz6 s  rzcCs$t|}t||}t|||jdS)N)rCrrzr)rBr*r|ryr+r+r,r!J s c@s$eZdZddZddZddZdS) _TestClasscCs ||_dS)N)r)rfrr+r+r,rf sz_TestClass.__init__cCs|jd|_|S)Nr3)r)rfr+r+r,squarep s z_TestClass.squarecCs|jS)N)r)rfr+r+r,r.z sz_TestClass.getN)r5rkrlrrr.r+r+r+r,r~X s  r~z Example of a string object, searched as-is. >>> x = 1; y = 2 >>> x + y, x * y (3, 2) a In 2.2, boolean expressions displayed 0 or 1. By default, we still accept them. This can be disabled by passing DONT_ACCEPT_TRUE_FOR_1 to the new optionflags argument. >>> 4 == 4 1 >>> 4 == 4 True >>> 4 > 4 0 >>> 4 > 4 False z Blank lines can be marked with : >>> print('foo\n\nbar\n') foo bar z If the ellipsis flag is used, then '...' can be used to elide substrings in the desired output: >>> print(list(range(1000))) #doctest: +ELLIPSIS [0, 1, 2, ..., 999] a If the whitespace normalization flag is used, then differences in whitespace are ignored. >>> print(list(range(30))) #doctest: +NORMALIZE_WHITESPACE [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29] )r~rzbool-int equivalencez blank linesZellipsiszwhitespace normalizationc Cs tjdd}|jdddddd|jd d d tjgd d |jddddd|jdddd|j}|j}|j}d}x|jD]}|t|O}q|W|j r|t O}x||D]t}|j drt j j|\}}tj jd|t|dd}tj d=t|||d\} } nt|d||d\} } | rdSqWdS)Nzdoctest runner)Z descriptionz-vz --verbose store_trueFz'print very verbose output for all tests)actiondefaulthelpz-oz--optionrzqspecify a doctest option flag to apply to the test run; may be specified more than once to apply multiple options)rchoicesrrz-fz --fail-fastzystop running tests after first failure (this is a shorthand for -o FAIL_FAST, and is in addition to any other -o options))rrr]rz file containing the tests to run)nargsrr"z.pyrD)rr )rSrr r&)argparseArgumentParser add_argumentr'keys parse_argsr]rrZ fail_fastrrerMrror=insertr:rr) rrZ testfilesrrrrQdirnamerrr=r+r+r,_test s<       rr)r3)rV) NNNNTr"NFF)FrYNr")NNNN)FN)FN)F)RZ __docformat____all__r-rrNr6rrMrrXr=r^r]ior# collectionsr$r%r'rrrrrrrrr r r r rr rKrnr2rCrUr\rcrdrwr{rrrrIrrrrrr ExceptionrrrrRrrrrZrr^r\rkZ TestSuitermrrsrurrrr rzr!r~rrr5exitr+r+r+r,0s      1%.DKl;;n f x $! @ IR   ,  +