U
e5d_ ! @ 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
setdefaultlenname r, /usr/lib64/python3.8/doctest.pyr s z z...c C s8 d}t jD ](}| |d }|tt |kr
||jO }q
|S )Nr" )
__future__Zall_feature_namesgetgetattrZ
compiler_flag)globsflagsZfnameZfeaturer, r, r- _extract_future_flags s
r3 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)moduleZdepthr, r, r- _normalize_module s
rD c C s dD ]}| |d} q| S )N)z
)replace)datanewliner, r, r- _newline_convert s rJ c
C s |rVt |d}t|| } t|dd d k rVt|jdrV|j| }||}t|| fS t| |d}| | fW 5 Q R S Q R X d S )N
__loader__get_data)encoding)
rD _module_relative_pathr0 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)rj sizer, r, r- rn s
z_SpoofOut.truncate)N)r6
__module__ __qualname__rc rn 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 |D ],}||||}|dk r dS |t|7 }qdS )Nr" FT)ELLIPSIS_MARKERsplitr)
startswithri find)wantgotZwsstartposendposwr, r, r- _ellipsis_match s0
r| 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) rv 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)rj 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)rj framer, r, r- r o s
z _OutputRedirectingPdb.set_tracec C s | j rtj| d S rl )r r r set_continuerj r, r, r- r u s z"_OutputRedirectingPdb.set_continuec G s2 t j}| jt _ztjj| f| W S |t _X d S rl )r> r r r r trace_dispatch)rj args save_stdoutr, r, r- r { s
z$_OutputRedirectingPdb.trace_dispatch)N)r6 rp rq 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 | dr| jD ]&}tj||}tj|r| S qtd | 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__))r7 r8 rB ru
ValueErrorospathjoinrt rP r r6 r) r> argvcurdirr exists)rC Z test_pathZbasedirZ directoryfullpathr, 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 rh )ri sourcerw linenor^ optionsexc_msg)rj r rw 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 rl )typeNotImplementedr rw r r^ r r rj otherr, r, r- __eq__ s
zExample.__eq__c C s t | j| j| j| j| jfS rl )hashr rw r r^ r r r, r, r- __hash__ s zExample.__hash__)Nr" r" N)r6 rp rq 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 rl )examples docstringcopyr1 r+ rT r )rj r r1 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__r6 r+ rT r )rj 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 rl )r r r r r1 r+ rT r r r, r, r- r ) s
zDocTest.__eq__c C s t | j| j| j| jfS rl )r r r+ rT r r r, r, r- r 4 s zDocTest.__hash__c C s: t |tstS | j| j| jt| f|j|j|jt|fk S rl )r9 r r r+ rT r idr r, r, r- __lt__8 s
zDocTest.__lt__N)r6 rp rq 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\}}| j|D ]}|||| ||d|| 7 }| |||\}}} }
|
|s|t || |
| t |
d |d ||d| | 7 }| }qP|||d |S )Nr" rF c s g | ]}| d qS rl r, .0lZ
min_indentr, r-
s z'DocTestParser.parse..r" r" r^ )r r^ r )
expandtabs _min_indentr rt _EXAMPLE_REfinditerappendr count_parse_example_IS_BLANK_OR_COMMENTr r) groupr ) rj stringr+ outputZcharnor mr r rw r r, r r- parses s*
zDocTestParser.parsec C s t | |||||||S rl )r get_examples)rj r r1 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, )r9 r )r xr, r, r- r s
z.DocTestParser.get_examples..)r )rj 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 Zslr^ r, r- r s z0DocTestParser._parse_example..rw z *$rr c s g | ]}| d qS rl r, )r Zwlr r, r- r s r )
r) r rt _check_prompt_blank
_check_prefixr r[ match
_EXCEPTION_RE
_find_options)
rj r r+ r source_linesr rw
want_linesr r r, r r- r s&
z DocTestParser._parse_examplez#\s*doctest:\s*([^\n\'"]*)$c C s i }| j |D ]v}|ddd }|D ]V}|d dksN|dd tkrdtd|d ||f t|dd }|d dk||< q.q|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 rt r' r r ) rj r r+ r r r Zoption_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)rj r] indentsr, r, r- r s zDocTestParser._min_indentc C s^ t |D ]P\}}t||d kr||d dkrtd|| d ||||d |f qd S )NrY rK rZ z8line %r of the docstring for %s lacks blank after %s: %rr& ) enumerater) r )rj linesr^ r+ r r r r, r, r- r s z!DocTestParser._check_prompt_blankc C s> t |D ]0\}}|r||std|| d ||f qd S )NzGline %r of the docstring for %s has inconsistent leading whitespace: %rr& )r ru r )rj r prefixr+ r r r r, r, r- r s
zDocTestParser._check_prefixN)r )r )r6 rp rq r[ compile MULTILINEVERBOSEr DOTALLr r r r r r r r r r r r r r, r, r, r- r C s&
'
3 c @ sJ e Zd Zde ddfddZdddZdd Zd
d Zd d
Zdd Z dS )r FTc C s || _ || _|| _|| _d S rl )_parser_verbose_recurse_exclude_empty)rj verboseparserrecurse
exclude_emptyr, r, r- r 7 s zDocTestFinder.__init__Nc C sN |d kr.t |dd }|d kr.tdt|f |dkrr )r0 r r r7 getmoduleZ
getsourcefilerB Zgetfile linecachegetlines__dict__r update_findsort) rj objr+ rC r1
extraglobsr` r testsr, r, r- rv M sL $
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__rp z"object must be a class or function)r7 r
isfunctionr __globals__ZismethoddescriptorrP r rp r6 isclassr9 propertyr )rj rC 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| jr|j D ]P\} }
d|| f } t
t |
st |
rl|
||
rl| ||
| |||| qlt|rn| jrnt|di D ]\} }
t| tstdt| f t
|
sJt |
sJt|
sJt|
tsJtdt|
f d|| f } | ||
| |||| qt |r| jr|j D ]\} }
t|
trt|| }
t|
trt|| j}
t
|
st |
st|
tr|
||
rd|| f } | ||
| |||| qd 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 r7 r8 r r itemsZ isroutineZunwrapr r r r0 r9 r: r r staticmethod classmethod__func__r ) rj r r r+ rC r r1 seentestZvalnamevalr, r, r- r sn
zDocTestFinder._findc C s t |tr|}nJz,|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 .pycrr ) r9 r: __doc__rB AttributeError _find_linenor r0 r6 r r ) rj r r+ rC r1 r r r rT r, r, r- r s,
zDocTestFinder._get_testc C s d }t |rd}t |rb|d kr(d S tdt|dd }t|D ]\}}||rF|} qbqFt |rr|j }t
|r|j }t |r|j
}t |r|j}t |rt|dd d }|d k r
|d kr|d S td}t|t|D ]}||| r| S qd S )Nr" z^\s*class\s*%s\br6 -co_firstlinenor& z(^|.*:)\s*\w*("|\'))r7 r8 r r[ r r0 r r Zismethodr r __code__Z istracebacktb_frameZisframef_codeZiscoderanger) )rj r r r Zpatr r r, r, r- r 4 s>
zDocTestFinder._find_lineno)NNNN)
r6 rp rq r r rv 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 )N-vr" ) r _checkerr> r r optionflagsoriginal_optionflagstriesfailures_name2ftrg _fakeout)rj 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 rw r_ r )rj r r
exampler, r, r- report_start s zDocTestRunner.report_startc C s | j r|d d S )Nzok
)r rj r r
r rx r, r, r- report_success s z DocTestRunner.report_successc C s& || ||| j||| j d S rl )_failure_headerr output_differencer r" r, r, r- report_failure s z DocTestRunner.report_failurec C s$ || ||d tt| d S )NzException raised:
)r$ r_ rf rj 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 )rj r
r r r r r, r, r- r$ s
z DocTestRunner._failure_headerc C s d }}| j }td\}}} | jj}
t|jD ]6\} } | j t@ oH|dk}
|| _ | jr| j D ],\}}|r|| j |O _ q`| j | M _ q`| j t @ rq.|d7 }|
s|
||| d|j | f }z,t t
| j|d|d|j | j d }W n4 tk
r Y n t }| j Y nX | j }| jd |}|d kr`|
| 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|
sR| ||| | nT||kr(|
s | ||| | |d7 }n*|| krR|
sH| !||| | |d7 }n |r.| j t"@ r. qhq.|| _ | #||| t$||S )Nr" rK r& zZsingler4 rr )%r r r check_outputr r r r r r r! r+ execr r r1 debuggerr KeyboardInterruptr> rd r rc rn rw ra format_exception_onlyrf r r r r# r& r( r _DocTestRunner__record_outcomer% )rj r
compileflagsr r r r SUCCESSFAILUREZBOOMZcheck
examplenumr quietZ
optionflagr rT Z exceptionrx Zoutcomer r, r, r- Z__run s
zDocTestRunner.__runc C sL | j |jd\}}|| || f| j |j< | j|7 _| j|7 _d S )Nr )r r/ r+ r r )rj 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+ r4 Tkeepends)
%_DocTestRunner__LINECACHE_FILENAME_REr r r
r+ r intr
splitlinessave_linecache_getlines)rj 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| |||W S t_|t
_ t| | jt_|t_| r|j dd l}d |_X d S )Nutf-8c s t | d } | d S )Nbackslashreplace)r: encodewrite)r] rN r r, r- r s z DocTestRunner.run..outr" )r
r3 r1 r> r rN lowerrC r gettracer r r r- resetr r r> *_DocTestRunner__patched_linecache_getlines displayhook__displayhook__settraceclearbuiltins__DocTestRunner__run) rj r
r1 r clear_globsZ
save_traceZsave_set_traceZsave_displayhookrM r, rD r- run s<
zDocTestRunner.runc
C s |d kr| j }g }g }g }d }}| j D ]V}|\}\} }
||
7 }|| 7 }|
dkr`|| q,| dkrx|||
f q,|| q,|r|rtt|d | |D ]} td| q|rtt|d | |D ]\} } td| | f q|r:t| j tt|d | |D ] \} \} }
td| |
| f q|rft|dt| jd t|| d
|d |rztd |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% )
rj r ZnotestsZpassedZfailedZtotaltZtotalfr r+ rW r6 thingr r, r, r- summarize sP
zDocTestRunner.summarizec C sR | j }|j D ]<\}\}}||kr@|| \}}|| }|| }||f||< qd S rl )r r )rj r dr+ rW r6 r7 r8 r, r, r- merge s zDocTestRunner.merge)NNr" )N)NNT)N)r6 rp rq r* r r! r# r& r( r$ rO r0 r[ r r; rH rQ rS rU r, r, r, r- r h s :
$
}
I
9c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
r c C s t |dddS )NASCIIrA )r: rB )rj 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) rW r r r[ r\ escapeBLANKLINE_MARKERr r rt r r| rj rw rx r r, r, r- r+ . s4
zOutputChecker.check_outputc C s< |t tB tB @ sdS |t@ r dS |ddko:|ddkS )NFTrF r4 )r r
r r rZ 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)^[ ]*(?=
)Tr9 r4 )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
)rw r r[ r\ rY r[ r= r difflibZ unified_difflistr
Z context_diffr ZDifferZIS_CHARACTER_JUNKZcomparer_ r )
rj r rx r rw r Z got_linesZdiffZkindZenginer, r, r- r% s4 z OutputChecker.output_differenceN)r6 rp rq rW r+ r[ r% r, r, r, r- r s ?c @ s e Zd Zdd Zdd ZdS )r c C s || _ || _|| _d S rl )r
r rx )rj r
r rx r, r, r- r s zDocTestFailure.__init__c C s
t | jS rl r: r
r r, r, r- __str__ s zDocTestFailure.__str__Nr6 rp rq r r` r, r, r, r- r s c @ s e Zd Zdd Zdd ZdS )r c C s || _ || _|| _d S rl )r
r rd )rj r
r rd r, r, r- r s z UnexpectedException.__init__c C s
t | jS rl r_ r r, r, r- r` s zUnexpectedException.__str__Nra 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 rQ r1 rL )rj r
r1 r rP rr, r, r- rQ 3 s
zDebugRunner.runc C s t |||d S rl )r r' r, r, r- r( 9 s z'DebugRunner.report_unexpected_exceptionc C s t |||d S rl )r r" r, r, r- r& < s zDebugRunner.report_failure)NNT)r6 rp rq rQ 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}
| j
| |||dD ]} |
| qt|r|
t
d kr|
a
n
t
|
t|
j|
jS )Nr z testmod: module required; %r)r r r r1 r )r> r? r/ r7 r8 rB r6 r r r rv rQ rS masterrU r% r r ) r r+ r1 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 )N8Package may only be specified for module-relative paths.r@ r6 r rc r" )r rX r r basenamer r r r r rQ rS re rU r% r r )rT rV r+ rU r1 r rf r r rg r rN textri r
r, r, r- r s2 R
NoNamec C s@ t |dd}t||d}|j| ||dD ]}|j||d q(d S )NF)r r rc )r1 )r1 )r r rv rQ ) rW r1 r r+ r1 r rh ri r
r, r, r- r + s c C s" | t @ | krtd| t}| a|S )Nz Only reporting flags allowed)r
r _unittest_reportflags)r2 oldr, r, r- r H s
c @ sl e Zd ZdddZdd Zdd Zd d
Zd d Zd
d Zdd Z dd Z
dd Z dd Z e
jZdd ZdS ) DocTestCaser" Nc C s. t j| || _|| _|| _|| _|| _d S rl )unittestTestCaser _dt_optionflags _dt_checker_dt_test _dt_setUp _dt_tearDown)rj r
r setUptearDownr r, r, r- r n s zDocTestCase.__init__c C s | j }| jd k r| | d S rl )ru rv rj r
r, r, r- rx x s
zDocTestCase.setUpc C s( | j }| jd k r| | |j d S rl )ru rw r1 rL rz r, r, r- ry ~ s
zDocTestCase.tearDownc C s~ | j }tj}t }| j}|t@ s(|tO }t|| jdd}z d|_ |j
||j dd\}}W 5 |t_X |rz| |
| d S )NFr r r zF----------------------------------------------------------------------)r rP )ru r> r r# rs r
rn r rt r* rQ rC ZfailureExceptionformat_failurerc )rj r
ro newr ri 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 rr z:Failed doctest test for %s
File "%s", line %s, in %s
%s)ru r r r+ rt rT )rj errr
r Zlnamer, r, r- r| s
zDocTestCase.format_failurec C s6 | t| j| jdd}|j| jdd | d S )NFr{ )rP )rx r rs rt rQ ru ry )rj ri r, r, r- r! s B zDocTestCase.debugc C s | j jS rl ru r+ r r, r, r- r s zDocTestCase.idc C sP t | t |k rtS | j|jkoN| j|jkoN| j|jkoN| j|jkoN| j|jkS rl )r r ru rs rv rw rt r r, r, r- r s
zDocTestCase.__eq__c C s t | j| j| j| jfS rl )r rs rv rw rt r r, r, r- r s zDocTestCase.__hash__c C s, | j jd}d|d d|d d f S )Nr z%s (%s)rr )ru r+ rt r )rj r+ r, r, r- r s zDocTestCase.__repr__c C s d| j j S )Nz Doctest: r r r, r, r- shortDescription s z DocTestCase.shortDescription)r" NNN)r6 rp rq r rx ry r~ r| r! r r r r r r` r r, r, r, r- rp l s
H
rp c @ s0 e Zd Zdd Zdd Zdd Zdd ZeZd S )
SkipDocTestCasec C s || _ t| d d S rl )rC rp r )rj rC r, r, r- r
s zSkipDocTestCase.__init__c C s | d d S )Nz-DocTestSuite will not work with -O2 and above)ZskipTestr r, r, r- rx s zSkipDocTestCase.setUpc C s d S rl r, r r, r, r- test_skip s zSkipDocTestCase.test_skipc C s d| j j S )NzSkipping tests from %s)rC r6 r r, r, r- r s z SkipDocTestCase.shortDescriptionN)r6 rp rq r rx r r r` r, r, r, r- r s
r c @ s e Zd Zdd ZdS )
_DocTestSuitec C s d S rl r, )rj indexr, r, r- _removeTestAtIndex s z _DocTestSuite._removeTestAtIndexN)r6 rp rq r r, r, r, r- r s r c K s |d krt }t| } |j| ||d}|sNtjjdkrNt }|t| |S | t }|D ]T}t
|j dkrtq`|j s| j
}|dd dkr|d d }||_ |t|f| q`|S )Nrd r4 r" r r
rr )r rD rv r> r2 optimizer addTestr r r) r rT r rp ) rC r1 r Z test_finderr r suiter
rT r, r, r- r # s( % c @ s$ e Zd Zdd Zdd Zdd ZdS ) DocFileCasec C s d | jjdS )NrN r )r ru r+ rt r r, r, r- r e s zDocFileCase.idc C s | j jS rl )ru rT r 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)ru r+ rT )rj r r, r, r- r| k s zDocFileCase.format_failureN)r6 rp rq r r r| r, r, r, r- r c s r c
K sv |d kri }n| }|r&|s&tdt| |||p4d\}} d|krL| |d< tj| }||||| d} t| f|S )Nrj r@ r r" )r r rX r r rk r r )
r rV rU r1 r rN r docr+ r
r, r, r- DocFileTestp s r c O sD t }|ddr$t|d|d< | D ]}|t|f| q(|S )NrV TrU )r r/ rD r r )pathskwr r r, r, r- r s 8 c C s g }t | D ]x}t|trh||jd d |j}|r|d |dd |dd d D 7 }q|dd |dd d D 7 }q|r|d dkr| q|r|d dkr|d qd |d S ) Nrr z # Expected:c S s g | ] }d | qS )z## 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
s r} r" )
r r r9 r r r rw rt popr )r] r Zpiecerw 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 r6 r* r, r- r .
s
z testsource..znot found in testsr" )rD r rv r r r )rC r+ r r
testsrcr, r* r- r %
s
c C s t | }t||| d S rl )r debug_script)srcpmr1 r r, r, r- r 5
s c C s dd l }|r| }ni }|rvzt| || W q tt d |jdd}| |d t d Y qX n|jdd d| || d S )Nr" r& T)r r4 zexec(%r))
r r r, r r> rd r rG Z interactionrQ )r r r1 r pr, r, r- r :
s
r c C s$ t | } t| |}t||| j d S rl )rD r r r )rC r+ r r r, r, r- r! N
s
c @ s$ e Zd Zdd Zdd Zdd ZdS )
_TestClassc C s
|| _ d S rl r )rj r r, r, r- r j
s z_TestClass.__init__c C s | j d | _ | S )Nr4 r r r, r, r- squaret
s z_TestClass.squarec C s | j S rl r r r, r, r- r/ ~
s z_TestClass.getN)r6 rp rq 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}|jD ]}|t| O }q|j r|t
O }|D ]v}| drt j
|\}}tj
d| t|d d } tj
d= t| ||d\}
} nt|d||d\}
} |
r dS qdS ) Nr" zdoctest runner)Z descriptionr z --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.pyrc )rV r r r& )argparseArgumentParser add_argumentr' keys
parse_argsr` r r Z fail_fastr ri r r rt r> insertr; r r ) r r r Z testfilesr r r rT dirnamer r rN r, r, r- _test
sL
r r )r4 )rY ) NNNNTr" NFF)Frm Nr" )NNNN)FN)FN)F)RZ
__docformat____all__r. r] r7 r r r r[ r> ra rq ior# collectionsr$ r% r' r r r r r r r r r r
r r r r
rY rs r3 rD rJ rX r_ rf rg r| r r r r rO r r r r r r Exceptionr r r re r r r rn r rr rp r Z TestSuiter r r r r r r r r r! r r r r6 exitr, r, r, r- 0 sD -
1%.DK l < ; n
h
{
$ !
@
IR
, 3-