B
Z
! @ sp d Z ddddddddd d
d d d
ddddddddddddddd d d d d 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 m
Z
d"d%lmZ ed&d'Zi Zd(d ZedZedZedZedZedZedZeeB eB eB eB eB Zed Zed
Zed Z ed Z edZ eeB e B e B e B Z d)Z d*Z!d+d, Z"drd.d/Z#d0d1 Z$d2d3 Z%dsd5d6Z&d7d8 Z'G d9d: d:e
Z(d;d< Z)d=d> Z*d?d@ Z+G dAdB dBej,Z-dCdD Z.G dEd dZ/G dFd dZ0G dGd dZ1G dHd dZ2G dId dZ3G dJd dZ4G dKd de5Z6G dLd de5Z7G dMd de3Z8d#a9dtdPdZ:dNd#d#d#d#dNd"d#dOe1 d#f dQdZ;dudSdZG dUdV dVe j?Z@G dWdX dXe@ZAG dYdZ dZe jBZCdvd[dZDG d\d] d]e@ZEdNd#d#e1 d#fd^d_ZFd`d ZGdad ZHdbd ZIdwdcd ZJdxdddeZKdydfd!ZLG dgdh dhZMeMdidjdkdldmdnZNdodp ZOePdqkrle QeO d#S )zzreStructuredText 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_srcdebug N)StringIO)
namedtuple TestResultszfailed attemptedc C s t | dtt > S )N )OPTIONFLAGS_BY_NAME
setdefaultlen)name r+ ,/opt/alt/python37/lib64/python3.7/doctest.pyr s z z...c C s< d}x2t jD ](}| |d }|tt |kr ||jO }q W |S )Nr" )
__future__Zall_feature_namesgetgetattrZ
compiler_flag)globsflagsZfnameZfeaturer+ r+ r, _extract_future_flags s r2 c C sV t | r| S t| tr,t| t t dgS | d krJtjt |j
d S t dd S )N*__name__z"Expected a module, string, or None) inspectismodule
isinstancestr
__import__globalslocalssysmodules _getframe f_globals TypeError)moduledepthr+ r+ r, _normalize_module s
rD c C s xdD ]}| |d} qW | S )N)z
)replace)datanewliner+ r+ r, _newline_convert s
rJ c C sz |rVt |d}t|| } t|dd d k rVt|jdrV|j| }||}t|| fS t| |d}| | fS Q R X d S )N
__loader__get_data)encoding)
rD _module_relative_pathr/ hasattrrL rM decoderJ openread)filenamepackagemodule_relativerN Z
file_contentsfr+ r+ r, _load_testfile s
rX c C s t d|d | S )Nz
(?m)^(?!$) )resub)sindentr+ r+ r, _indent s r_ c C s* t }| \}}}tj||||d | S )N)file)r# tracebackprint_exceptiongetvalue)exc_infoZexcoutexc_typeZexc_valZexc_tbr+ r+ r, _exception_traceback s
rf c @ s e Zd Zdd ZdddZdS ) _SpoofOutc C s$ t | }|r |ds |d7 }|S )NrF )r# rc endswith)selfresultr+ r+ r, rc s
z_SpoofOut.getvalueNc C s | | t| d S )N)seekr# truncate)ri sizer+ r+ r, rl s
z_SpoofOut.truncate)N)r5
__module__ __qualname__rc rl r+ r+ r+ r, rg s rg c C s t | kr| |kS | t }dt| }}|d }|rR||rNt|}|d= ndS |d }|r||r||t|8 }|d= ndS ||krdS x2|D ]*}||||}|dk rdS |t|7 }qW dS )Nr" FT)ELLIPSIS_MARKERsplitr)
startswithrh find)wantgotZwsstartposendposwr+ r+ r, _ellipsis_match s0
rz c C s | } | rd| S dS d S )Nz# #)rstrip)liner+ r+ r,
_comment_line? s r~ c C sh dt | }}| d}|dkr$|}| dd|}|dkr>|}| dd|}|dkr\|d }| || S )Nr" rF :.r& )r) rt rfind)msgstartendir+ r+ r, _strip_exception_detailsG s
r c @ s. e Zd Zdd Zd
ddZdd Zdd ZdS ) _OutputRedirectingPdbc C s( || _ d| _tjj| |dd d| _d S )NFT)stdoutnosigintr& )_OutputRedirectingPdb__out$_OutputRedirectingPdb__debugger_usedpdbPdb__init__Z use_rawinput)ri outr+ r+ r, r g s z _OutputRedirectingPdb.__init__Nc C s* d| _ |d krt j}tj| | d S )NT)r r= r? f_backr r set_trace)ri framer+ r+ r, r o s
z _OutputRedirectingPdb.set_tracec C s | j rtj| d S )N)r r r set_continue)ri r+ r+ r, r u s z"_OutputRedirectingPdb.set_continuec G s. t j}| jt _ztjj| f| S |t _X d S )N)r= r r r r trace_dispatch)ri args save_stdoutr+ r+ r, r { s
z$_OutputRedirectingPdb.trace_dispatch)N)r5 rn ro r r r r r+ r+ r+ r, r a s
r c C s t | std| |dr(tdtjj|d }t | drXtj| j
d }n| j dkrt t
jdkrt
jd dkrtjt
jd d }qtj}nFt | drx,| jD ]"}tj||}tj|r|S qW td | 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__))r6 r7 rA rs
ValueErrorospathjoinrr rP r r5 r) r= argvcurdirr exists)rB Z test_pathZbasedirZ directoryZfullpathr+ r+ r, rO s&
rO c @ s& e Zd Zd ddZdd Zdd ZdS )
r Nr" c C sv | ds|d7 }|r(| ds(|d7 }|d k rB| dsB|d7 }|| _|| _|| _|| _|d krfi }|| _|| _d S )NrF )rh sourceru linenor^ optionsexc_msg)ri r ru r r r^ r r+ r+ r, r s
zExample.__init__c C s\ t | t |k rtS | j|jkoZ| j|jkoZ| j|jkoZ| j|jkoZ| j|jkoZ| j|jkS )N)typeNotImplementedr ru r r^ r r )ri otherr+ r+ r, __eq__ s zExample.__eq__c C s t | j| j| j| j| jfS )N)hashr ru r r^ r )ri r+ r+ r, __hash__ s zExample.__hash__)Nr" r" N)r5 rn ro r r r r+ r+ r+ r, r s #
c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
Zd S ) r c C s, || _ || _| | _|| _|| _|| _d S )N)examples docstringcopyr0 r* rT r )ri r r0 r* rT r r r+ r+ r, r s
zDocTest.__init__c C sR t | jdkrd}n"t | jdkr(d}ndt | j }d| jj| j| j| j|f S )Nr" z no examplesr& z 1 examplez %d examplesz<%s %s from %s:%s (%s)>)r) r __class__r5 r* rT r )ri r r+ r+ r, __repr__ s zDocTest.__repr__c C s\ t | t |k rtS | j|jkoZ| j|jkoZ| j|jkoZ| j|jkoZ| j|jkoZ| j|jkS )N)r r r r r0 r* rT r )ri r r+ r+ r, r ) s zDocTest.__eq__c C s t | j| j| j| jfS )N)r r r* rT r )ri r+ r+ r, r 4 s zDocTest.__hash__c C s: t |tstS | j| j| jt| f|j|j|jt|fk S )N)r8 r r r* rT r id)ri r r+ r+ r, __lt__8 s
zDocTest.__lt__N)r5 rn ro r r r r r r+ r+ r+ r, r s
c @ s e Zd ZedejejB ZedejejB ejB Z edj
Z dddZ dd Z
dd d
Zd d Zed
ejZdd ZedejZdd Zdd Zdd ZdS )r a
# 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 | }| | dkr8d fdd|dD }g }d\}}x| j|D ]}|||| ||d|| 7 }| |||\}}} }
|
|s|t || |
| t |
d |d ||d| | 7 }| }qRW |||d |S )Nr" rF c s g | ]}| d qS )Nr+ ).0l)
min_indentr+ r,
s z'DocTestParser.parse..)r" r" r^ )r r^ r )
expandtabs _min_indentr rr _EXAMPLE_REfinditerappendr count_parse_example_IS_BLANK_OR_COMMENTr r) groupr ) ri stringr* outputZcharnor mr r ru r r+ )r r, parses s&
zDocTestParser.parsec C s t | |||||||S )N)r get_examples)ri r r0 r* rT r r+ r+ r, get_doctest s zDocTestParser.get_doctestc C s dd | ||D S )Nc S s g | ]}t |tr|qS r+ )r8 r )r xr+ r+ r, r s z.DocTestParser.get_examples..)r )ri r r* r+ r+ r, r s zDocTestParser.get_examplesc
s
t |d |dd}| | || | |dd d d || d fdd|D }|d }|d}t |dkrtd
|d r|d = | |d ||t | d fd d|D }| j|}|r|d
}nd }| |||} || ||fS )Nr^ r rF r& rZ r c s g | ]}| d d qS )rY Nr+ )r sl)r^ r+ r, r s z0DocTestParser._parse_example..ru z *$rp c s g | ]}| d qS )Nr+ )r Zwl)r^ r+ r, r s r )
r) r rr _check_prompt_blank
_check_prefixr r[ match
_EXCEPTION_RE
_find_options)
ri r r* r source_linesr ru
want_linesr r r+ )r^ r, r s$
z DocTestParser._parse_examplez#\s*doctest:\s*([^\n\'"]*)$c C s i }x| j |D ]z}|ddd }x^|D ]V}|d dksR|dd tkrhtd|d ||f t|dd }|d dk||< q2W qW |r| |rtd|||f |S ) Nr& ,rZ 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_REr r rG rr r' r r ) ri r r* r r r option_stringsoptionflagr+ r+ r, r s
zDocTestParser._find_optionsz
^([ ]*)(?=\S)c C s2 dd | j |D }t|dkr*t|S dS d S )Nc S s g | ] }t |qS r+ )r) )r r^ r+ r+ r, r
s z-DocTestParser._min_indent..r" )
_INDENT_REfindallr) min)ri r] indentsr+ r+ r, r s zDocTestParser._min_indentc C sb x\t |D ]P\}}t||d kr
||d dkr
td|| d ||||d |f q
W d S )NrY rK rZ z8line %r of the docstring for %s lacks blank after %s: %rr& ) enumerater) r )ri linesr^ r* r r r} r+ r+ r, r s
z!DocTestParser._check_prompt_blankc C sB xr )r/ r r r6 getmoduleZ
getsourcefilerA Zgetfile linecachegetlines__dict__r update_findsort) ri objr* rB r0
extraglobsr` r testsr+ r+ r, rt M sJ $
zDocTestFinder.findc C s |d kr dS t |d k r(|t |kS t |r>|j|jkS t |r|t|dr\|jj}nt|drn|j}ndS |j |kS t
|r|j |jkS t|dr|j |jkS t |t rdS t
dd S )NT __objclass__rn z"object must be a class or function)r6 r
isfunctionr __globals__ZismethoddescriptorrP r rn r5 isclassr8 propertyr )ri rB objectZobj_modr+ r+ r, _from_module s(
zDocTestFinder._from_modulec
C s* | j rtd| t||kr"d S d|t|< | |||||}|d k rR|| t|r| jrx^|j D ]P\} }
d|| f } t
t |
st |
rn|
||
rn| ||
| |||| qnW t|rv| jrvxt|di D ]\} }
t| tstdt| f t
|
sPt |
sPt|
sPt|
tsPtdt|
f d|| f } | ||
| |||| qW t |r&| jr&x|j D ]\} }
t|
trt|| }
t|
trt|| j}
t
|
st |
st|
tr|
||
rd|| f } | ||
| |||| qW d S )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)r printr _get_testr r6 r7 r r itemsZ isroutineZunwrapr r r r/ r8 r9 r r staticmethod classmethod__func__r ) ri r r r* rB r r0 seentestZvalnamevalr+ r+ r, r sP
zDocTestFinder._findc C s t |tr|}nJy,|jd kr"d}n|j}t |ts:t|}W n ttfk
rX d}Y nX | ||}| jrt|std S |d krd }n.t|dd p|j}|dd dkr|d d }| j
|||||S )Nr r z.pycrp ) r8 r9 __doc__rA AttributeError _find_linenor r/ r5 r r ) ri r r* rB r0 r r r rT r+ r+ r, r s(
zDocTestFinder._get_testc C s d }t |rd}t |rd|d kr(d S tdt|dd }x$t|D ]\}}||rH|}P qHW t |rt|j }t
|r|j }t |r|j
}t |r|j}t |rt|dd d }|d k r |d kr|d S td}x(t|t|D ]}||| r|S qW d S )Nr" z^\s*class\s*%s\br5 -co_firstlinenor& z(^|.*:)\s*\w*("|\'))r6 r7 r r[ r r/ r r Zismethodr r __code__Z istracebacktb_frameZisframef_codeZiscoderanger) )ri r r r Zpatr r} r+ r+ r, r 4 s<
zDocTestFinder._find_lineno)NNNN)
r5 rn ro r r rt r r r r r+ r+ r+ r, r . s
f ?&c @ s e Zd ZdZd ddZdd Zdd Zd
d Zd d
Zdd Z dd Z
dd Z e
dZd ddZd ddZd!ddZd d ZdS )"r zF**********************************************************************Nr" c C sN |pt | _|d kr dtjk}|| _|| _|| _d| _d| _i | _ t
| _ d S )Nz-vr" ) r _checkerr= r r optionflagsoriginal_optionflagstriesfailures_name2ftrg _fakeout)ri checkerr r r+ r+ r, r s
zDocTestRunner.__init__c C sH | j rD|jr.|dt|j d t|j n|dt|j d d S )NzTrying:
z Expecting:
zExpecting nothing
)r ru r_ r )ri r r exampler+ r+ r, report_start s zDocTestRunner.report_startc C s | j r|d d S )Nzok
)r )ri r r r rv r+ r+ r, report_success s z DocTestRunner.report_successc C s& || ||| j||| j d S )N)_failure_headerr output_differencer )ri r r r rv r+ r+ r, report_failure s z DocTestRunner.report_failurec C s$ || ||d tt| d S )NzException raised:
)r r_ rf )ri r r r rd r+ r+ r, report_unexpected_exception s z)DocTestRunner.report_unexpected_exceptionc C s | j g}|jrR|jd k r4|jd k r4|j|j d }nd}|d|j||jf n|d|jd |jf |d |j}|t| d|S )Nr& ?zFile "%s", line %s, in %szLine %s, in %szFailed example:rF )DIVIDERrT r r r* r r_ r )ri r r r r r r+ r+ r, r s
z DocTestRunner._failure_headerc C s d }}| j }td\}}} | jj}
xFt|jD ]6\} } | j t@ oL|dk}
|| _ | jrx:| j D ],\}}|r| j |O _ qf| j | M _ qfW | j t @ rq2|d7 }|
s|
||| d|j | f }y,t t
| j|d|d|j | j d }W n4 tk
r Y n t }| j Y nX | j }| jd |}|d krh|
| j|| j r|}n|tj|d d d }|
s|t|7 }| j d kr| }nB|
| j || j r|}n*| j t @ r|
t | j t || j r|}||kr|
sZ| ||| | nT||kr0|
s&| ||| | |d7 }n*|| krZ|
sP| !||| | |d7 }n |r2| j t"@ r2P q2W || _ | #||| t$||S )Nr" rK r& zZsingler3 rp )%r r r check_outputr r r r r r r r* execr r r0 debuggerr KeyboardInterruptr= rd r rc rl ru ra format_exception_onlyrf r r r r r r r _DocTestRunner__record_outcomer% )ri r compileflagsr r r r SUCCESSFAILUREZBOOMZcheck
examplenumr quietZ
optionflagr rT Z exceptionrv Zoutcomer r+ r+ r, Z__run s
zDocTestRunner.__runc C sL | j |jd\}}|| || f| j |j< | j|7 _| j|7 _d S )N)r" r" )r r. r* r r )ri r rW tf2t2r+ r+ r, Z__record_outcome| s z DocTestRunner.__record_outcomez..+)\[(?P\d+)\]>$c C sV | j |}|rF|d| jjkrF| jjt|d }|jjddS | ||S d S )Nr* r* T)keepends)
%_DocTestRunner__LINECACHE_FILENAME_REr r r r* r intr
splitlinessave_linecache_getlines)ri rT module_globalsr r r+ r+ r, Z __patched_linecache_getlines s
z*DocTestRunner.__patched_linecache_getlinesTc s || _ |d krt|j}tj|d krVj d ks@ dkrHj}n fdd}| jt_t }t
j }t | _
| j
| j
j t
_ tj| _| jt_tj}tjt_z| |||S t_|t
_ t| | jt_|t_|r|j dd l}d |_X d S )Nzutf-8c s t | d } | d S )Nbackslashreplace)r9 encodewrite)r] )rN r r+ r, r s z DocTestRunner.run..outr" )r r2 r0 r= r rN lowerr7 r gettracer r r r# resetr r r3 *_DocTestRunner__patched_linecache_getlines displayhook__displayhook___DocTestRunner__runsettraceclearbuiltins_) ri r r' r clear_globsZ
save_traceZsave_set_traceZsave_displayhookrA r+ )rN r r, run s<
zDocTestRunner.runc
C s |d kr| j }g }g }g }d }}xd| j D ]V}|\}\} }
||
7 }|| 7 }|
dkrb|| q.| dkrz|||
f q.|| q.W |r|rtt|d | x|D ]} td| qW |rtt|d | x |D ]\} } td| | f qW |rJt| j tt|d | x(|D ] \} \} }
td| |
| f q&W |rvt|dt| jd t|| d
|d |rtd |d
n|rtd t||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.) r r r r r r) r r r% )
ri r ZnotestsZpassedZfailedZtotaltZtotalfr r* rW r, thingr r+ r+ r, summarize sP
zDocTestRunner.summarizec C sV | j }xJ|j D ]<\}\}}||krB|| \}}|| }|| }||f||< qW d S )N)r r )ri r dr* rW r, r- r. r+ r+ r, merge s zDocTestRunner.merge)NNr" )N)NNT)N)r5 rn ro r r r r r r r r> r& r[ r r0 r; rD rF rH r+ r+ r+ r, r h s :
$
}
I
9c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
r c C s t |dddS )NASCIIr5 )r9 r6 )ri r] r+ r+ r, _toAscii( s zOutputChecker._toAsciic C s | |}| |}||kr dS |t@ sH||fdkr8dS ||fdkrHdS |t@ stdtt d|}tdd|}||krdS |t@ rd| }d| }||krdS |t
@ rt ||rdS dS ) NT)zTrue
z1
)zFalse
z0
z (?m)^%s\s*?$r z(?m)^[^\S\n]+$rZ F) rJ r r r[ r\ escapeBLANKLINE_MARKERr r rr r rz )ri ru rv r r+ r+ r, r! . s0
zOutputChecker.check_outputc C s< |t tB tB @ sdS |t@ r dS |ddko:|ddkS )NFTrF r3 )r r
r r )ri ru rv r r+ r+ r, _do_a_fancy_diffm s z OutputChecker._do_a_fancy_diffc
C s( |j }|t@ s tdt|}| |||r|jdd}|jdd}|t@ rptj ||dd}t
|dd }d}nZ|t @ rtj ||dd}t
|dd }d}n,|t
@ rtjtjd} t
| ||}d }n d
| td | S |r|rd t|t|f S |rd
t| S |r dt| S dS d S )Nz(?m)^[ ]*(?=
)T)r/ r3 )nz#unified diff with -expected +actualz-context diff with expected followed by actual)Zcharjunkz ndiff with -expected +actualzDifferences (%s):
r zExpected:
%sGot:
%szExpected:
%sGot nothing
zExpected nothing
Got:
%sz Expected nothing
Got nothing
)ru r r[ r\ rL rM r2 r difflibZ unified_difflistr
Z context_diffr ZDifferZIS_CHARACTER_JUNKZcomparer_ r )
ri r rv r ru r Z got_linesZdiffkindZenginer+ r+ r, r s4 z OutputChecker.output_differenceN)r5 rn ro rJ r! rM r r+ r+ r+ r, r s ?c @ s e Zd Zdd Zdd ZdS )r c C s || _ || _|| _d S )N)r r rv )ri r r rv r+ r+ r, r s zDocTestFailure.__init__c C s
t | jS )N)r9 r )ri r+ r+ r, __str__ s zDocTestFailure.__str__N)r5 rn ro r rR r+ r+ r+ r, r s c @ s e Zd Zdd Zdd ZdS )r c C s || _ || _|| _d S )N)r r rd )ri r r rd r+ r+ r, r s z UnexpectedException.__init__c C s
t | jS )N)r9 r )ri r+ r+ r, rR s zUnexpectedException.__str__N)r5 rn ro r rR r+ r+ r+ r, r s c @ s& e Zd Zd ddZdd Zdd ZdS )
r NTc C s$ t | |||d}|r |j |S )NF)r rD r0 r@ )ri r r' r rC rr+ r+ r, rD 3 s
zDebugRunner.runc C s t |||d S )N)r )ri r r r rd r+ r+ r, r 9 s z'DebugRunner.report_unexpected_exceptionc C s t |||d S )N)r )ri r r r rv r+ r+ r, r < s zDebugRunner.report_failure)NNT)r5 rn ro rD r r r+ r+ r+ r, r s [
TFc C s | d krt jd} t| s,td| f |d kr:| j}t|d} |rVt||d}
n t ||d}
x$| j
| |||dD ]} |
| qvW |r|
t
d kr|
a
n
t
|
t|
j|
jS )Nr z testmod: module required; %r)r )r r )r0 r )r= r> r. r6 r7 rA r5 r r r rt rD rF masterrH r% r r ) r r* r0 r reportr r raise_on_errorr finderrunnerr r+ r+ r, r H s$ E
c C s |r|st dt| ||| p d\} } |d kr:tj| }|d krHi }n| }|d k rb|| d|krrd|d< | rt||d}
n t||d}
|
| ||| d}|
| |r|
t d kr|
a n
t
|
t|
j|
jS )Nz8Package may only be specified for module-relative paths.zutf-8r5 r )r r r" )r rX r r basenamer r r r r rD rF rT rH r% r r )rT rV r* rU r0 r rU r r rV r rN textrX r r+ r+ r, r s0 R
NoNamec C sD t |dd}t||d}x&|j| ||dD ]}|j||d q*W d S )NF)r r )r r )r0 )r' )r r rt rD ) rW r0 r r* r' r rW rX r r+ r+ r, r + s c C s" | t @ | krtd| t}| a|S )Nz Only reporting flags allowed)r
r _unittest_reportflags)r1 oldr+ r+ r, r H s
c @ sj e Zd ZdddZdd Zdd Zd d
Zd d Zd
d Zdd Z dd Z
dd Z dd Z e Z
dd ZdS ) DocTestCaser" Nc C s. t j| || _|| _|| _|| _|| _d S )N)unittestTestCaser _dt_optionflags _dt_checker_dt_test _dt_setUp _dt_tearDown)ri r r setUptearDownr r+ r+ r, r n s zDocTestCase.__init__c C s | j }| jd k r| | d S )N)rc rd )ri r r+ r+ r, rf x s
zDocTestCase.setUpc C s( | j }| jd k r| | |j d S )N)rc re r0 r@ )ri r r+ r+ r, rg ~ s
zDocTestCase.tearDownc C s~ | j }tj}t }| j}|t@ s(|tO }t|| jdd}z d|_ |j
||j dd\}}W d |t_X |rz| |
| d S )NF)r r r zF----------------------------------------------------------------------)r rC )rc r= r r# ra r
r\ r rb r rD r7 ZfailureExceptionformat_failurerc )ri r r] newr rX r r r+ r+ r, runTest s zDocTestCase.runTestc C sP | j }|jd krd}n
d|j }d|jddd }d|j|j|||f S )Nzunknown line numberz%sr rp z:Failed doctest test for %s
File "%s", line %s, in %s
%s)rc r r r* rr rT )ri errr r Zlnamer+ r+ r, rh s
zDocTestCase.format_failurec C s6 | t| j| jdd}|j| jdd | d S )NF)r r r )rC )rf r ra rb rD rc rg )ri rX r+ r+ r, r! s
B zDocTestCase.debugc C s | j jS )N)rc r* )ri r+ r+ r, r s zDocTestCase.idc C sP t | t |k rtS | j|jkoN| j|jkoN| j|jkoN| j|jkoN| j|jkS )N)r r rc ra rd re rb )ri r r+ r+ r, r s zDocTestCase.__eq__c C s t | j| j| j| jfS )N)r ra rd re rb )ri r+ r+ r, r s zDocTestCase.__hash__c C s, | j jd}d|d d|d d f S )Nr z%s (%s)rp )rc r* rr r )ri r* r+ r+ r, r s zDocTestCase.__repr__c C s d| j j S )Nz Doctest: )rc r* )ri r+ r+ r, shortDescription s z DocTestCase.shortDescription)r" NNN)r5 rn ro r rf rg rj rh r! r r r r rR rl r+ r+ r+ r, r^ l s
H
r^ c @ s0 e Zd Zdd Zdd Zdd Zdd ZeZd S )
SkipDocTestCasec C s || _ t| d d S )N)rB r^ r )ri rB r+ r+ r, r
s zSkipDocTestCase.__init__c C s | d d S )Nz-DocTestSuite will not work with -O2 and above)ZskipTest)ri r+ r+ r, rf s zSkipDocTestCase.setUpc C s d S )Nr+ )ri r+ r+ r, test_skip s zSkipDocTestCase.test_skipc C s d| j j S )NzSkipping tests from %s)rB r5 )ri r+ r+ r, rl s z SkipDocTestCase.shortDescriptionN)r5 rn ro r rf rn rl rR r+ r+ r+ r, rm s
rm c @ s e Zd Zdd ZdS )
_DocTestSuitec C s d S )Nr+ )ri indexr+ r+ r, _removeTestAtIndex s z _DocTestSuite._removeTestAtIndexN)r5 rn ro rq r+ r+ r+ r, ro s ro c K s |d krt }t| } |j| ||d}|sNtjjdkrNt }|t| |S | t }x\|D ]T}t
|j dkrvqb|j s| j
}|dd dkr|d d }||_ |t|f| qbW |S )N)r0 r r3 r" r z.pycrp )r rD rt r= r1 optimizero addTestrm r r) r rT r r^ ) rB r0 r Z test_finderr r suiter rT r+ r+ r, r # s( %
c @ s( e Zd Zdd Zdd ZeZdd ZdS ) DocFileCasec C s d | jjdS )NrB r )r rc r* rr )ri r+ r+ r, r e s zDocFileCase.idc C s | j jS )N)rc rT )ri r+ r+ r, r h s zDocFileCase.__repr__c C s d| j j| j j|f S )Nz2Failed doctest test for %s
File "%s", line 0
%s)rc r* rT )ri rk r+ r+ r, rh l s zDocFileCase.format_failureN)r5 rn ro r r rR rh r+ r+ r+ r, ru c s ru c
K sv |d kri }n| }|r&|s&tdt| |||p4d\}} d|krL| |d< tj| }||||| d} t| f|S )Nz8Package may only be specified for module-relative paths.zutf-8r r" )r r rX r r rY r ru )
r rV rU r0 r rN r docr* r r+ r+ r, DocFileTestq s rw c O sH t }|ddr$t|d|d< x | D ]}|t|f| q*W |S )NrV TrU )ro r. rD rs rw )pathskwrt r r+ r+ r, r s 8
c C s g }xt | D ]x}t|trj||jd d |j}|r|d |dd |dd d D 7 }q|dd |dd d D 7 }qW x |r|d dkr| qW x |r|d dkr|d qW d |d S ) Nrp z # Expected:c S s g | ] }d | qS )z## r+ )r r r+ r+ r, r
s z(script_from_examples..rF c S s g | ] }t |qS r+ )r~ )r r r+ r+ r, r
s r{ r" )
r r r8 r r r ru rr popr )r] r Zpieceru r+ r+ r, r s :
" c sJ t | } t | } fdd|D }|s4t d|d }t|j}|S )Nc s g | ]}|j kr|qS r+ )r* )r r, )r* r+ r, r /
s z testsource..znot found in testsr" )rD r rt r r r )rB r* r r testsrcr+ )r* r, r &
s
c C s t | }t||| d S )N)r debug_script)srcpmr0 r{ r+ r+ r, r 6
s c C s dd l }|r| }ni }|rvyt| || W q tt d |jdd}| |d t d Y qX n|jdd d| || d S )Nr" r& T)r r3 zexec(%r))
r r r" r r= rd r r: Z interactionrD )r} r~ r0 r pr+ r+ r, r| ;
s
r| c C s$ t | } t| |}t||| j d S )N)rD r r| r )rB r* r~ r{ r+ r+ r, r! O
s
c @ s$ e Zd Zdd Zdd Zdd ZdS )
_TestClassc C s
|| _ d S )N)r )ri r r+ r+ r, r k
s z_TestClass.__init__c C s | j d | _ | S )Nr3 )r )ri r+ r+ r, squareu
s z_TestClass.squarec C s | j S )N)r )ri r+ r+ r, r.
s z_TestClass.getN)r5 rn ro r r r. r+ r+ r+ r, r ]
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 r zbool-int equivalencez blank linesZellipsiszwhitespace normalizationc C s* dd l } | jdd}|jdddddd |jd
d d t g d
d |jddddd |jdddd | }|j}|j}d}x|jD ]}|t| O }qW |j r|t
O }x~|D ]v}| drt j
|\}}tj
d| t|d d } tj
d= t| ||d\}
} nt|d||d\}
} |
rdS qW dS ) Nr" zdoctest runner) descriptionz-vz --verbose
store_trueFz'print very verbose output for all tests)actiondefaulthelpz-oz--optionr zqspecify a doctest option flag to apply to the test run; may be specified more than once to apply multiple options)r choicesr r z-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))r r r` r z file containing the tests to run)nargsr z.py)r r )rV r r r& )argparseArgumentParser add_argumentr' keys
parse_argsr` r r Z fail_fastr rh r r rr r= insertr: r r ) r r r Z testfilesr r r rT dirnamer r rB r+ r+ r, _test
s>
r r )r3 )rY ) NNNNTr" NFF)Fr[ Nr" )NNNN)FN)FN)F)RZ
__docformat____all__r- rO r6 r r r r[ r= ra r_ ior# collectionsr$ r% r' r r r r r r r r r r
r r r r
rL rq r2 rD rJ rX r_ rf rg rz r~ r r r rO r r r r r r Exceptionr r r rT r r r r\ r r` r^ rm Z TestSuitero r ru rw r r r r r| r! r r r r5 exitr+ r+ r+ r, 0 s
1%.DK l < ; n
f x
$ !
@ IR
, -