a DOgx @s2dZgdZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z ddlmZeddZiZdd Zed Zed Zed Zed ZedZedZeeBeBeBeBeBZedZedZedZedZedZeeBeBeBeBZdZ dZ!ddZ"dqddZ#ddZ$ddZ%drd!d"Z&d#d$Z'Gd%d&d&e Z(d'd(Z)d)d*Z*d+d,Z+Gd-d.d.ej,Z-d/d0Z.Gd1d2d2Z/Gd3d4d4Z0Gd5d6d6Z1Gd7d8d8Z2Gd9d:d:Z3Gd;d<d<Z4Gd=d>d>e5Z6Gd?d@d@e5Z7GdAdBdBe3Z8da9dsdEdFZ:dCdddddCdddDe1df dGdHZ;dtdJdKZGdNdOdOe j?Z@GdPdQdQe@ZAGdRdSdSe jBZCdudTdUZDGdVdWdWe@ZEdCdde1dfdXdYZFdZd[ZGd\d]ZHd^d_ZIdvd`daZJdwdbdcZKdxdddeZLGdfdgdgZMeMdhdidjdkdldmZNdndoZOePdpkr.e QeOdS)yzreStructuredText en)!register_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 attemptedcCst|dtt>S)N)OPTIONFLAGS_BY_NAME setdefaultlennamer,/usr/lib64/python3.9/doctest.pyrsrrrrrrrr r r r rz z...cCs8d}tjD](}||d}|tt|ur ||jO}q |S)Nr") __future__Zall_feature_namesgetgetattrZ compiler_flag)globsflagsfnameZfeaturer,r,r-_extract_future_flagss    r4cCsVt|r|St|tr,t|ttdgS|durJtjt |j dSt ddS)N*__name__z"Expected a module, string, or None) inspectismodule isinstancestr __import__globalslocalssysmodules _getframe f_globals TypeError)moduledepthr,r,r-_normalize_modules  rFcCsdD]}||d}q|S)N)z   )replace)datanewliner,r,r-_newline_convertsrLcCs|rVt|d}t||}t|dddurVt|jdrV|j|}||}t||fSt||d}| |fWdS1s0YdS)N __loader__get_data)encoding) rF_module_relative_pathr0hasattrrNrOdecoderLopenread)filenamepackagemodule_relativerPZ file_contentsfr,r,r-_load_testfiles      rZcCstd|d|S)Nz (?m)^(?!$) )resub)sindentr,r,r-_indentsracCs*t}|\}}}tj||||d|S)N)file)r# tracebackprint_exceptiongetvalue)exc_infoZexcoutexc_typeexc_valexc_tbr,r,r-_exception_tracebacks rjc@seZdZddZdddZdS) _SpoofOutcCs$t|}|r |ds |d7}|SNrH)r#reendswith)selfresultr,r,r-res z_SpoofOut.getvalueNcCs||t|dSN)seekr#truncate)rnsizer,r,r-rr s z_SpoofOut.truncate)N)r7 __module__ __qualname__rerrr,r,r,r-rks rkcCst|vr||kS|t}dt|}}|d}|rR||rNt|}|d=ndS|d}|r||r||t|8}|d=ndS||krdS|D],}||||}|dkrdS|t|7}qdS)Nr"FT)ELLIPSIS_MARKERsplitr) startswithrmfind)wantgotwsstartposendposwr,r,r-_ellipsis_matchs0    rcCs|}|rd|SdSdS)Nz# #)rstrip)liner,r,r- _comment_line?srcCshdt|}}|d}|dkr$|}|dd|}|dkr>|}|dd|}|dkr\|d}|||S)Nr"rH:.r&)r)rzrfind)msgstartendir,r,r-_strip_exception_detailsGs  rc@s.eZdZddZd ddZddZdd ZdS) _OutputRedirectingPdbcCs(||_d|_tjj||ddd|_dS)NFT)stdoutnosigintr&)_OutputRedirectingPdb__out$_OutputRedirectingPdb__debugger_usedpdbPdb__init__Z use_rawinput)rnoutr,r,r-rgsz_OutputRedirectingPdb.__init__NcCs*d|_|durtj}tj||dS)NT)rr?rAf_backrr set_trace)rnframer,r,r-ros z_OutputRedirectingPdb.set_tracecCs|jrtj|dSrp)rrr set_continuernr,r,r-rusz"_OutputRedirectingPdb.set_continuecGs8tj}|jt_ztjj|g|RW|t_S|t_0dSrp)r?rrrrtrace_dispatch)rnargs save_stdoutr,r,r-r{sz$_OutputRedirectingPdb.trace_dispatch)N)r7rtrurrrrr,r,r,r-ras rcCst|std||dr(tdtjj|d}t |drXtj|j d}n|j dkrt t jdkrt jddkrtjt jdd}qtj}nFt |dr|jD]&}tj||}tj|r|Sqtd |j tj||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__))r8r9rCry ValueErrorospathjoinrxrRrr7r)r?argvcurdirrexists)rDZ test_pathZbasedirZ directoryfullpathr,r,r-rQs(         rQc@s&eZdZd ddZddZddZdS) rNr"cCsv|ds|d7}|r(|ds(|d7}|durB|dsB|d7}||_||_||_||_|durfi}||_||_dSrl)rmsourcer{linenor`optionsexc_msg)rnrr{rrr`rr,r,r-rs  zExample.__init__cCs\t|t|urtS|j|jkoZ|j|jkoZ|j|jkoZ|j|jkoZ|j|jkoZ|j|jkSrp)typeNotImplementedrr{rr`rrrnotherr,r,r-__eq__s      zExample.__eq__cCst|j|j|j|j|jfSrp)hashrr{rr`rrr,r,r-__hash__szExample.__hash__)Nr"r"N)r7rtrurrrr,r,r,r-rs #  rc@s4eZdZddZddZddZddZd d Zd S) rcCs,||_||_||_||_||_||_dSrp)examples docstringcopyr1r+rVr)rnrr1r+rVrrr,r,r-rs  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__r7r+rVr)rnrr,r,r-__repr__szDocTest.__repr__cCs\t|t|urtS|j|jkoZ|j|jkoZ|j|jkoZ|j|jkoZ|j|jkoZ|j|jkSrp)rrrrr1r+rVrrr,r,r-r)s      zDocTest.__eq__cCst|j|j|j|jfSrp)rrr+rVrrr,r,r-r4szDocTest.__hash__cCs:t|tstS|j|j|jt|f|j|j|jt|fkSrp)r:rrr+rVridrr,r,r-__lt__8s  zDocTest.__lt__N)r7rtrurrrrrr,r,r,r-rs   rc@seZdZedejejBZedejejBejBZ edj Z dddZ ddZ dd d Zd d Zed ejZddZedejZddZddZddZdS)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|}||dkr8dfdd|dD}g}d\}}|j|D]}||||||d||7}| |||\}}} } | |s|t || | |t | d|d||d||7}|}qP|||d|S)Nr"rHcsg|]}|dqSrpr,.0lZ min_indentr,r- z'DocTestParser.parse..r"r"r`)rr`r) expandtabs _min_indentrrx _EXAMPLE_REfinditerappendrcount_parse_example_IS_BLANK_OR_COMMENTrr)groupr) rnstringr+outputZcharnormrrr{rr,rr-parsess*     zDocTestParser.parsecCst||||||||Srp)r get_examples)rnrr1r+rVrr,r,r- get_doctests zDocTestParser.get_doctestcCsdd|||DS)NcSsg|]}t|tr|qSr,)r:r)rxr,r,r-rs z.DocTestParser.get_examples..)r)rnrr+r,r,r-rs zDocTestParser.get_examplesc s t|d|dd}||||||dddd||dfdd|D}|d }|d}t|dkrtd |d r|d =||d||t|dfd d|D}|j|}|r|d }nd}| |||} || ||fS)Nr`rrHr&r\rcsg|]}|ddqS)r[Nr,)rslr`r,r-rrz0DocTestParser._parse_example..r{z *$rvcsg|]}|dqSrpr,)rZwlrr,r-rrr) r)rrx_check_prompt_blank _check_prefixrr]match _EXCEPTION_RE _find_options) rnrr+r source_linesrr{ want_linesrrr,rr-rs&       zDocTestParser._parse_examplez#\s*doctest:\s*([^\n\'"]*)$c Csi}|j|D]v}|ddd}|D]V}|ddvsN|ddtvrdtd|d||ft|dd}|ddk||<q.q|r||rtd|||f|S) Nr&,r\r"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_RErrrIrxr'rr) rnrr+rrroption_stringsoptionflagr,r,r-rs"  zDocTestParser._find_optionsz ^([ ]*)(?=\S)cCs2dd|j|D}t|dkr*t|SdSdS)NcSsg|] }t|qSr,)r))rr`r,r,r-r rz-DocTestParser._min_indent..r") _INDENT_REfindallr)min)rnr_indentsr,r,r-rs zDocTestParser._min_indentc Cs^t|D]P\}}t||dkr||ddkrtd||d||||d|fqdS)Nr[rMr\z8line %r of the docstring for %s lacks blank after %s: %rr&) enumerater)r)rnlinesr`r+rrrr,r,r-rs  z!DocTestParser._check_prompt_blankcCs>t|D]0\}}|r||std||d||fqdS)NzGline %r of the docstring for %s has inconsistent leading whitespace: %rr&)rryr)rnrprefixr+rrrr,r,r-rs zDocTestParser._check_prefixN)r)r)r7rtrur]compile MULTILINEVERBOSErDOTALLrrrrrrrrrrrrrr,r,r,r-rCs&     ' 3rc@sJeZdZdeddfddZdddZdd Zd d Zd d ZddZ dS)rFTcCs||_||_||_||_dSrp)_parser_verbose_recurse_exclude_empty)rnverboseparserrecurse exclude_emptyr,r,r-r7szDocTestFinder.__init__Nc CsL|dur.t|dd}|dur.tdt|f|durr)r0rrr8 getmoduleZ getsourcefilerCZgetfile linecachegetlines__dict__rupdate_findsort) rnobjr+rDr1 extraglobsrbrtestsr,r,r-rzMsJ$          zDocTestFinder.findcCs|dur dSt|dur(|t|uSt|r>|j|juSt|r|t|dr\|jj}nt|drn|j}ndS|j |kSt |r|j |jkSt|dr|j |jkSt |t rdSt ddS)NT __objclass__rtz"object must be a class or function)r8r isfunctionr __globals__ZismethoddescriptorrRrrtr7isclassr:propertyr)rnrDobjectZobj_modr,r,r- _from_modules(            zDocTestFinder._from_modulec Cs|jrtd|t||vr"dSd|t|<||||||}|durR||t|r|jr|j D]P\} } d|| f} t t | st | rl| || rl||| | ||||qlt|rn|jrnt|di D]\} } t| tstdt| ft | sJt | sJt| sJt| tsJtdt| fd|| f} ||| | ||||qt |r|jr|j D]t\} } t| ttfr| j} t | st | st| tr| || rd|| f} ||| | ||||qdS)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_testrr8r9rritemsZ isroutineunwraprr rr0r:r;rr staticmethod classmethod__func__r) rnrrr+rDrr1seentestZvalnamevalr,r,r-rsd          zDocTestFinder._findc Cst|tr|}nHz,|jdur"d}n|j}t|ts:t|}WnttfyVd}Yn0|||}|jrr|srdS|durd}n.t|ddp|j}|dddkr|dd}|j |||||S)Nrr.pycrv) r:r;__doc__rCAttributeError _find_linenorr0r7rr) rnrr+rDr1rrrrVr,r,r-r  s*         zDocTestFinder._get_testcCsd}t|rd}t|rb|dur(dStdt|dd}t|D]\}}||rF|}qbqFt|rr|j }t |r|j }t |r|j }t|r|j}t|rt|ddd}|dur |dur|dStd}t|t|D]}|||r|SqdS)Nr"z^\s*class\s*%s\br7-co_firstlinenor&z(^|.*:)\s*\w*("|\'))r8r9rr]rr0rrZismethodrr__code__Z istracebacktb_frameZisframef_codeZiscoderanger))rnrrrpatrrr,r,r-r2s6        zDocTestFinder._find_lineno)NNNN) r7rtrurrrzr rr rr,r,r,r-r.s   f=&rc@seZdZdZdddZddZdd Zd d Zd d ZddZ ddZ ddZ e dZdddZd ddZd!ddZddZdS)"rzF**********************************************************************Nr"cCsN|pt|_|durdtjv}||_||_||_d|_d|_i|_ t |_ dS)N-vr") r_checkerr?rr optionflagsoriginal_optionflagstriesfailures_name2ftrk_fakeout)rncheckerrr#r,r,r-rs  zDocTestRunner.__init__cCsH|jrD|jr.|dt|jdt|jn|dt|jddS)NzTrying: z Expecting: zExpecting nothing )rr{rar)rnrrexampler,r,r- report_startszDocTestRunner.report_startcCs|jr|ddS)Nzok )rrnrrr*r|r,r,r-report_successszDocTestRunner.report_successcCs&|||||j|||jdSrp)_failure_headerr"output_differencer#r,r,r,r-report_failures zDocTestRunner.report_failurecCs$||||dtt|dS)NzException raised: )r.rarjrnrrr*rfr,r,r-report_unexpected_exceptions   z)DocTestRunner.report_unexpected_exceptioncCs|jg}|jrR|jdur4|jdur4|j|jd}nd}|d|j||jfn|d|jd|jf|d|j}|t|d|S)Nr&?zFile "%s", line %s, in %szLine %s, in %szFailed example:rH)DIVIDERrVrrr+rrar)rnrr*rrrr,r,r-r.s  zDocTestRunner._failure_headerc Csd}}|j}td\}}} |jj} t|jD]4\} } |jt@oH|dk} ||_| jr| jD],\}}|r||j|O_q`|j|M_q`|jt @rq.|d7}| s| ||| d|j | f}z,t t | j|d|d|j|jd}Wn2tyYnt}|jYn0|j}|jd|}|dur^| | j||jr|}n|tj|ddd}| s|t|7}| jdur| }nB| | j||jr|}n*|jt@r| t| jt||jr|}||ur| sP|||| |nT||ur&| s| ||| ||d7}n*|| urP| sF|!||| ||d7}n|r.|jt"@r.qfq.||_|#|||t$||S) Nr"rMr&zZsingleTr5rv)%r#rr" check_outputrrr rr rr+r+execrrr1debuggerrKeyboardInterruptr?rfr(rerrr{rcformat_exception_onlyrjrrrr-r0r2r_DocTestRunner__record_outcomer%)rnr compileflagsrr&r%r$SUCCESSFAILUREZBOOMcheck examplenumr*quietZ optionflagrrV exceptionr|outcomerr,r,r-Z__runs                  zDocTestRunner.__runcCsL|j|jd\}}||||f|j|j<|j|7_|j|7_dS)Nr)r'r/r+r&r%)rnrrYtf2t2r,r,r-Z__record_outcomezszDocTestRunner.__record_outcomez..+)\[(?P\d+)\]>$cCsV|j|}|rF|d|jjkrF|jjt|d}|jjddS| ||SdS)Nr+r?Tkeepends) %_DocTestRunner__LINECACHE_FILENAME_RErrrr+rintr splitlinessave_linecache_getlines)rnrVmodule_globalsrr*r,r,r-Z__patched_linecache_getliness  z*DocTestRunner.__patched_linecache_getlinesTc sD||_|durt|j}tj|durVjdus@dkrHj}nfdd}|jt_t }t j }t |_ |j |j j t _ tj|_|jt_tj}tjt_zR||||Wt_|t _ t||jt_|t_|r|jddl}d|_SnDt_|t _ t||jt_|t_|r>|jddl}d|_0dS)Nutf-8cs t|d}|dS)Nbackslashreplace)r;encodewrite)r_rPrr,r-rszDocTestRunner.run..outr")rr4r1r?rrPlowerrPr(gettracerrrr7resetrrrK*_DocTestRunner__patched_linecache_getlines displayhook__displayhook___DocTestRunner__runsettraceclearbuiltins_) rnrr;r clear_globsZ save_traceZsave_set_traceZsave_displayhookr[r,rQr-runsb        zDocTestRunner.runc Cs|dur|j}g}g}g}d}}|jD]V}|\}\} } || 7}|| 7}| dkr`||q,| dkrx||| fq,||q,|r|rtt|d||D]} td| q|rtt|d||D]\} } td| | fq|r:t|jtt|d||D] \} \} } td| | | fq|rft|dt|jd t||d |d |rztd |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.) rr'r rr r)rr4r%) rnrZnotestsZpassedZfailedZtotaltZtotalfrr+rYrCthingrr,r,r- summarizesP       zDocTestRunner.summarizecCsR|j}|jD]<\}\}}||vr@||\}}||}||}||f||<qdSrp)r'r )rnrdr+rYrCrDrEr,r,r-merges zDocTestRunner.merge)NNr")N)NNT)N)r7rtrur4rr+r-r0r2r.rXr:r]rrHrUr^r`rbr,r,r,r-rfs: $ }   I 9rc@s,eZdZddZddZddZddZd S) rcCst|dddS)NASCIIrN)r;rO)rnr_r,r,r-_toAscii&szOutputChecker._toAsciicCs||}||}||kr dS|t@sH||fdkr8dS||fdkrHdS|t@stdttd|}tdd|}||krdS|t@rd| }d| }||krdS|t @rt ||rdSdS) NT)zTrue z1 )zFalse z0 z (?m)^%s\s*?$rz(?m)^[^\S\n]+$r\F) rdrrr]r^escapeBLANKLINE_MARKERrrrxrrrnr{r|r#r,r,r-r5,s2     zOutputChecker.check_outputcCs<|ttBtB@sdS|t@r dS|ddko:|ddkS)NFTrHr5)r r r rrgr,r,r-_do_a_fancy_diffks zOutputChecker._do_a_fancy_diffc Cs(|j}|t@stdt|}||||r|jdd}|jdd}|t@rptj ||dd}t |dd}d}nZ|t @rtj ||dd}t |dd}d}n,|t @rtjtjd} t | ||}d }nd |td |S|r|rd t|t|fS|rd t|S|r dt|SdSdS)Nz(?m)^[ ]*(?= )TrFr5)nz#unified diff with -expected +actualz-context diff with expected followed by actual)Zcharjunkzndiff with -expected +actualzDifferences (%s): rzExpected: %sGot: %szExpected: %sGot nothing zExpected nothing Got: %szExpected nothing Got nothing )r{rr]r^rfrhrJr difflibZ unified_difflistr Z context_diffr ZDifferZIS_CHARACTER_JUNKZcomparerar) rnr*r|r#r{rZ got_linesdiffkindZenginer,r,r-r/s4    zOutputChecker.output_differenceN)r7rtrurdr5rhr/r,r,r,r-rs?rc@seZdZddZddZdS)rcCs||_||_||_dSrp)rr*r|)rnrr*r|r,r,r-rszDocTestFailure.__init__cCs t|jSrpr;rrr,r,r-__str__szDocTestFailure.__str__Nr7rtrurror,r,r,r-rs rc@seZdZddZddZdS)rcCs||_||_||_dSrp)rr*rf)rnrr*rfr,r,r-rszUnexpectedException.__init__cCs t|jSrprnrr,r,r-roszUnexpectedException.__str__Nrpr,r,r,r-rs rc@s&eZdZd ddZddZddZdS) rNTcCs$t||||d}|r |j|S)NF)rr^r1rZ)rnrr;rr]rr,r,r-r^1s zDebugRunner.runcCst|||dSrp)rr1r,r,r-r27sz'DebugRunner.report_unexpected_exceptioncCst|||dSrp)rr,r,r,r-r0:szDebugRunner.report_failure)NNT)r7rtrur^r2r0r,r,r,r-rs[ rTFc Cs|durtjd}t|s,td|f|dur:|j}t|d} |rVt||d} n t ||d} | j ||||dD]} | | qt|r| t dur| a n t | t| j| jS)Nrztestmod: module required; %r)rrr#r1r)r?r@r/r8r9rCr7rrrrzr^r`masterrbr%r&r%) rr+r1rreportr#rraise_on_errorrfinderrunnerrr,r,r-rFs$E      rc Cs|r|stdt|||| pd\} }|dur:tj|}|durHi}n|}|durb||d|vrrd|d<| rt||d} n t||d} | | |||d}| ||r| t dur| a n t | t| j| jS)N8Package may only be specified for module-relative paths.rMr7rrrr")rrZrrbasenamerrrrrr^r`rtrbr%r&r%)rVrXr+rWr1rrur#rrvrrPtextrxrr,r,r-rs2R     rNoNamec Cs@t|dd}t||d}|j|||dD]}|j||dq(dS)NF)rrrr)r1)r;)rrrzr^) rYr1rr+r;r#rwrxrr,r,r-r)s  rcCs"|t@|krtd|t}|a|S)NzOnly reporting flags allowed)r r_unittest_reportflags)r2oldr,r,r-rFs   rc@sleZdZdddZddZddZd d Zd d Zd dZddZ ddZ ddZ ddZ e jZddZdS) DocTestCaser"NcCs:tj|||_||_|j|_||_||_ ||_ dSrp) unittestTestCaser_dt_optionflags _dt_checkerr1r _dt_globs_dt_test _dt_setUp _dt_tearDown)rnrr#setUptearDownr)r,r,r-rls  zDocTestCase.__init__cCs|j}|jdur||dSrp)rrrnrr,r,r-rws zDocTestCase.setUpcCs6|j}|jdur|||j|j|jdSrp)rrr1rZrrrr,r,r-r}s    zDocTestCase.tearDowncCs|j}tj}t}|j}|t@s(|tO}t||jdd}z&d|_ |j ||j dd\}}W|t_n|t_0|r| | |dS)NFr#r)rzF----------------------------------------------------------------------)rr])rr?rr#rr r}rrr4r^rPZfailureExceptionformat_failurere)rnrr~newr#rxr&r%r,r,r-runTests" zDocTestCase.runTestcCsP|j}|jdurd}n d|j}d|jddd}d|j|j|||fS)Nzunknown line numberz%srrvz:Failed doctest test for %s File "%s", line %s, in %s %s)rrrr+rxrV)rnerrrrZlnamer,r,r-rs  zDocTestCase.format_failurecCs6|t|j|jdd}|j|jdd|dS)NFr)r])rrrrr^rr)rnrxr,r,r-r!s BzDocTestCase.debugcCs|jjSrprr+rr,r,r-rszDocTestCase.idcCsPt|t|urtS|j|jkoN|j|jkoN|j|jkoN|j|jkoN|j|jkSrp)rrrrrrrrr,r,r-rs     zDocTestCase.__eq__cCst|j|j|j|jfSrp)rrrrrrr,r,r-r szDocTestCase.__hash__cCs,|jjd}d|dd|ddfS)Nrz%s (%s)rv)rr+rxr)rnr+r,r,r-r szDocTestCase.__repr__cCs d|jjS)Nz Doctest: rrr,r,r-shortDescription szDocTestCase.shortDescription)r"NNN)r7rtrurrrrrr!rrrrrrorr,r,r,r-rjs   H rc@s0eZdZddZddZddZddZeZd S) SkipDocTestCasecCs||_t|ddSrp)rDrr)rnrDr,r,r-r szSkipDocTestCase.__init__cCs|ddS)Nz-DocTestSuite will not work with -O2 and above)ZskipTestrr,r,r-r szSkipDocTestCase.setUpcCsdSrpr,rr,r,r- test_skip szSkipDocTestCase.test_skipcCs d|jjS)NzSkipping tests from %s)rDr7rr,r,r-r sz SkipDocTestCase.shortDescriptionN)r7rtrurrrrror,r,r,r-r s rc@seZdZddZdS) _DocTestSuitecCsdSrpr,)rnindexr,r,r-_removeTestAtIndex sz _DocTestSuite._removeTestAtIndexN)r7rtrurr,r,r,r-r src Ks|durt}t|}|j|||d}|sNtjjdkrNt}|t||S| t}|D]X}t |j dkrtq`|j s|j }|dddkr|dd}||_ |t|fi|q`|S)Nrsr5r"rrrv)rrFrzr?r2optimizeraddTestrrr)rrVrr) rDr1rZ test_finderrrsuiterrVr,r,r-r$ s(% rc@s$eZdZddZddZddZdS) DocFileCasecCsd|jjdS)Nr\r)rrr+rxrr,r,r-rf szDocFileCase.idcCs|jjSrp)rrVrr,r,r-ri szDocFileCase.__repr__cCsd|jj|jj|fS)Nz2Failed doctest test for %s File "%s", line 0 %s)rr+rV)rnrr,r,r-rl szDocFileCase.format_failureN)r7rtrurrrr,r,r,r-rd src Ksz|duri}n|}|r&|s&tdt||||p4d\}}d|vrL||d<tj|}|||||d} t| fi|S)NryrMrr")rrrZrrrzrr) rrXrWr1rrPrdocr+rr,r,r- DocFileTestq s rcOsHt}|ddr$t|d|d<|D]}|t|fi|q(|S)NrXTrW)rr/rFrr)pathskwrrr,r,r-r s 8 rcCsg}t|D]x}t|trh||jdd|j}|r|d|dd|dddD7}q|dd|dddD7}q|r|ddkr|q|r|ddkr|dqd |dS) Nrvz # Expected:cSsg|] }d|qS)z## r,rr,r,r-r rz(script_from_examples..rHcSsg|] }t|qSr,)rrr,r,r-r srr") rrr:rrrr{rxpopr)r_rZpiecer{r,r,r-r s :  "   rcsJt|}t|}fdd|D}|s4td|d}t|j}|S)Ncsg|]}|jkr|qSr,r*)rrCr*r,r-r/ rztestsource..znot found in testsr")rFrrzrrr)rDr+rrtestsrcr,r*r-r& s   rcCst|}t|||dSrp)r debug_script)srcpmr1rr,r,r-r 6 sr cCsddl}|r|}ni}|rvzt|||Wqttd|jdd}||dtdYq0n|jdd d|||dS)Nr"r&T)rr5zexec(%r)) rrr6r r?rfrrTZ interactionr^)rrr1rpr,r,r-r; s  rcCs$t|}t||}t|||jdSrp)rFrrr)rDr+rrr,r,r-r!O s r!c@s$eZdZddZddZddZdS) _TestClasscCs ||_dSrpr)rnrr,r,r-rk sz_TestClass.__init__cCs|jd|_|S)Nr5rrr,r,r-squareu s z_TestClass.squarecCs|jSrprrr,r,r-r/ sz_TestClass.getN)r7rtrurrr/r,r,r,r-r] s  rz 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] )rrzbool-int equivalencez blank linesZellipsiszwhitespace normalizationc Cs"ddl}|jdd}|jdddddd |jd d d tgd d|jddddd|jdddd|}|j}|j}d}|jD]}|t|O}q|j r|t O}|D]v}| drt j |\}}tj d|t|dd} tj d=t| ||d\} } nt|d||d\} } | rdSqdS)Nr"zdoctest runner) descriptionr!z --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))rrrbrz file containing the tests to run)nargsrz.pyrr)rXrr#r&)argparseArgumentParser add_argumentr'keys parse_argsrbrrZ fail_fastrrmrrrxr?insertr<rr) rrrZ testfilesrrrrVdirnamerr&r\r,r,r-_test sH       rr)r5)r[) NNNNTr"NFF)Fr|Nr")NNNN)FN)FN)F)RZ __docformat____all__r.rjr8rrrr]r?rcrior# collectionsr$r%r'rrrrrrrrr r r r rr rfrwr4rFrLrZrarjrkrrrrrrQrrrrrr Exceptionrrrrtrrrr}rrrrZ TestSuiterrrrrrrr rr!rr rr7exitr,r,r,r-0s-      1%.DKl:;n h { $$ @  IR   , 3-