o
0i @ sF d Z g dZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl m
Z
mZ ddlmZ eddZi Zdd Zed
Zed Zed Zed
ZedZedZeeB eB eB eB eB ZedZedZ edZ edZ edZ ee B e B e B e B Z dZ!dZ"dd Z#dqddZ$d d Z%d d Z&drd!d"Z'd#d$ Z(G d%d& d&e
Z)d'd( Z*d)d* Z+d+d, Z,G d-d. d.ej-Z.d/d0 Z/G d1d2 d2Z0G d3d4 d4Z1G d5d6 d6Z2G d7d8 d8Z3G d9d: d:Z4G d;d< d<Z5G d=d> d>e6Z7G d?d@ d@e6Z8G dAdB dBe4Z9da: DdsdEdFZ;dCdddddCdddDe2 df dGdHZ< I dtdJdKZ=da>dLdM Z?G dNdO dOe j@ZAG dPdQ dQeAZBG dRdS dSe jCZDdudTdUZEG dVdW dWeAZFdCdde2 dfdXdYZGdZd[ ZHd\d] ZId^d_ ZJdvd`daZKdvdbdcZLdwdddeZMG dfdg dgZNeNdhdidjdkdldmZOdndo ZPeQdpkre ReP dS dS )xzreStructuredText 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_srcdebug N)StringIOIncrementalNewlineDecoder)
namedtuple TestResultszfailed attemptedc C s t | dtt > S )N )OPTIONFLAGS_BY_NAME
setdefaultlenname r- ./opt/alt/python310/lib64/python3.10/doctest.pyr s r r r r r r r r r
r r r z z...c C s: d}t jD ]}| |d }|tt |u r||jO }q|S )Nr" )
__future__Zall_feature_namesgetgetattrZ
compiler_flag)globsflagsZfnameZfeaturer- r- r. _extract_future_flags s
r4 c C sT t | r| S t| trt| t t dgS | d u r&tjt |j
d S t d)N*__name__z"Expected a module, string, or None) inspectismodule
isinstancestr
__import__globalslocalssysmodules _getframe f_globals TypeError)moduleZdepthr- r- r. _normalize_module s
rE c C s t d d| dS NT)r$ decode)datar- r- r. _newline_convert s rI c C s |r;t |d}t|| } t|dd }d u r&z|jj}W n ty% Y nw t|dr;|| }||}t || fS t
| |d}| | fW d S 1 sRw Y d S )N
__loader__get_data)encoding) rE _module_relative_pathr1 __spec__loaderAttributeErrorhasattrrL rG rI openread)filenamepackagemodule_relativerM rP Z
file_contentsfr- r- r. _load_testfile s
$rY 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
rg c @ s e Zd Zdd ZdddZdS ) _SpoofOutc C s$ t | }|r|ds|d7 }|S N
)r# rd endswith)selfresultr- r- r. rd s
z_SpoofOut.getvalueNc C s | | t| d S N)seekr# truncate)rl sizer- r- r. rp
s
z_SpoofOut.truncatern )r7
__module__ __qualname__rd rp r- r- r- r. rh s rh c C s t | vr | |kS | t }dt|}}|d }|r*||r(t|}|d= ndS |d }|rA||r?|t|8 }|d= ndS ||krGdS |D ]}||||}|dk rY dS |t|7 }qIdS )Nr" FT)ELLIPSIS_MARKERsplitr*
startswithrk find)wantgotZwsZstartposendposwr- r- r. _ellipsis_match s2
r} c C s | } | r d| S dS )Nz# #)rstrip)liner- r- r.
_comment_line@ s
r c C sh dt | }}| d}|dkr|}| dd|}|dkr |}| dd|}|dkr.|d }| || S )Nr" rj :.r' )r* rx rfind)msgstartendir- r- r. _strip_exception_detailsH s
r c @ s0 e Zd Z dd Zd
ddZdd Zdd ZdS ) _OutputRedirectingPdbc C s( || _ d| _tjj| |dd d| _d S )NFT)stdoutnosigintr' )_OutputRedirectingPdb__out$_OutputRedirectingPdb__debugger_usedpdbPdb__init__Z use_rawinput)rl outr- r- r. r h s
z _OutputRedirectingPdb.__init__Nc C s* d| _ |d u r t j}tj| | d S rF )r r? rA f_backr r set_trace)rl framer- r- r. r p s
z _OutputRedirectingPdb.set_tracec C s | j r tj| d S d S rn )r r r set_continuerl r- r- r. r v s z"_OutputRedirectingPdb.set_continuec G s4 t j}| jt _ztjj| g|R W |t _S |t _w rn )r? r r r r trace_dispatch)rl args save_stdoutr- r- r. r | s
z$_OutputRedirectingPdb.trace_dispatchrn )r7 rr rs r r r r r- r- r- r. r b s
r c C s t | s td| |drtdtjj|d }t | dr,tj| j
d }nF| j dkrOt t
jdkrKt
jd dkrKtjt
jd d }n'tj}n#t | drk| jD ]}tj||}tj|rj| S qWtd | 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__))r8 r9 rC rw
ValueErrorospathjoinrv rR r r7 r* r? argvcurdirr exists)rD Z test_pathZbasedirZ directoryfullpathr- r- r. rN s*
rN c @ s, e Zd Z d ddZdd Zdd ZdS )
r Nr" c C sv | ds |d7 }|r| ds|d7 }|d ur!| ds!|d7 }|| _|| _|| _|| _|d u r3i }|| _|| _d S ri )rk sourcery linenor_ optionsexc_msg)rl r ry r r r_ r r- r- r. r s
zExample.__init__c C \ t | t |ur
tS | j|jko-| j|jko-| j|jko-| j|jko-| j|jko-| j|jkS rn )typeNotImplementedr ry r r_ r r rl otherr- r- r. __eq__
zExample.__eq__c C s t | j| j| j| j| jfS rn )hashr ry r r_ r r r- r- r. __hash__ s zExample.__hash__)Nr" r" N)r7 rr rs r r r r- r- r- r. r s "
r c @ s6 e Zd Z dd Zdd Zdd Zdd Zd d
Zd S ) r c C s. || _ || _| | _|| _|| _|| _d S rn )examples docstringcopyr2 r, rU r )rl r r2 r, rU r r r- r- r. r s
zDocTest.__init__c C sR t | jdkr
d}nt | jdkrd}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__r7 r, rU r )rl r r- r- r. __repr__ s zDocTest.__repr__c C r rn )r r r r r2 r, rU r r r- r- r. r * r zDocTest.__eq__c C t | j| j| j| jfS rn )r r r, rU r r r- r- r. r 5 s zDocTest.__hash__c C s: t |tstS | j| j| jt| f|j|j|jt|fk S rn )r: r r r, rU r idr r- r- r. __lt__9 s
zDocTest.__lt__N)r7 rr rs r r r r r r- r- r- r. r s r c @ s e Zd Z edejejB 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 | }| | dkr d fdd|dD }g }d\}}| j|D ]M}|||| ||d|| 7 }| |||\}}} }
|
|se|t || |
| t |
d |d ||d| | 7 }| }q)|||d |S )Nr" rj c g | ]}| d qS rn r- .0lZ
min_indentr- r.
z'DocTestParser.parse..r" r" r_ )r r_ r )
expandtabs _min_indentr rv _EXAMPLE_REfinditerappendr count_parse_example_IS_BLANK_OR_COMMENTr r* groupr ) rl stringr, outputZcharnor mr r ry r r- r r. parset s,
zDocTestParser.parsec C s t | |||||||S rn )r get_examples)rl r r2 r, rU r r- r- r. get_doctest s zDocTestParser.get_doctestc C s dd | ||D S )Nc S s g | ] }t |tr|qS r- )r: r )r xr- r- r. r s
z.DocTestParser.get_examples..)r )rl r r, r- r- r. r s
zDocTestParser.get_examplesc
s t |d |dd}| | || | |dd d d || d fdd|D }|d }|d}t |dkrOtd
|d rO|d = | |d ||t | d fd d|D }| j|}|rw|d
}nd }| |||} || ||fS )Nr_ r rj r' r[ r c s g | ]
}| d d qS )rZ Nr- )r Zslr_ r- r. r s z0DocTestParser._parse_example..ry z *$rt c r rn r- )r Zwlr r- r. r r r )
r* r rv _check_prompt_blank
_check_prefixr r\ match
_EXCEPTION_RE
_find_options)
rl r r, r source_linesr ry
want_linesr r r- r r. r s(
z DocTestParser._parse_examplez#\s*doctest:\s*([^\n\'"]*)$c C s i }| j |D ];}|ddd }|D ]+}|d dvs(|dd tvr3td|d ||f t|dd }|d dk||< qq |rU| |rUtd|||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_REr r replacerv r( r r ) rl r r, r r r Zoption_stringsoptionflagr- r- r. r s$ zDocTestParser._find_optionsz
^([ ]*)(?=\S)c C s0 dd | j |D }t|dkrt|S dS )Nc S g | ]}t |qS r- )r* )r r_ r- r- r. r z-DocTestParser._min_indent..r" )
_INDENT_REfindallr* min)rl r^ indentsr- r- r. r s
zDocTestParser._min_indentc C s` t |D ](\}}t||d kr-||d dkr-td|| d ||||d |f qd S )NrZ rJ r[ z8line %r of the docstring for %s lacks blank after %s: %rr' ) enumerater* r )rl linesr_ r, r r r r- r- r. r s z!DocTestParser._check_prompt_blankc C s@ t |D ]\}}|r ||s td|| d ||f qd S )NzGline %r of the docstring for %s has inconsistent leading whitespace: %rr' )r rw r )rl r prefixr, r r r r- r- r. r s zDocTestParser._check_prefixN)r )r7 rr rs r\ compile MULTILINEVERBOSEr DOTALLr r r r r r r r r r r r r r- r- r- r. r D s*
'
3 r c @ sT e Zd Z de ddfddZdddZdd Zd
d Zd d
Zdd Z dd Z
dS )r FTc C s || _ || _|| _|| _d S rn )_parser_verbose_recurse_exclude_empty)rl verboseparserrecurse
exclude_emptyr- r- r. r 8 s
zDocTestFinder.__init__Nc C sJ |d u rt |dd }|d u rtdt|f |du r d }n |d u r(t|}zt|}W n ty: d }Y n2w |sPt|}|d |dd dksPd }|d u rWd }n|d urct ||j
}nt |}|sld }|d u r}|d u rwi }n
|j
}n| }|d ur| | d|vrd|d< g }|
||||||i | |S )Nr7 zJDocTestFinder.find: name must be given when obj.__name__ doesn't exist: %rFr" z<]>r )r1 r r r8 getmoduleZ
getsourcefilerC Zgetfile linecachegetlines__dict__r update_findsort) rl objr, rD r2
extraglobsra r testsr- r- r. rx N sN #
zDocTestFinder.findc C s |d u rdS t |d ur|t |u S t |r |j|ju S t |r?t|dr/|jj}n t|dr8|j}ndS |j |kS t
|rJ|j |jkS t|drU|j |jkS t |t r\dS t
d)NT __objclass__rr z"object must be a class or function)r8 r
isfunctionr __globals__ZismethoddescriptorrR r rr r7 isclassr: propertyr )rl rD objectZobj_modr- r- r. _from_module s*
zDocTestFinder._from_modulec C s2 |}zt |}W n ty Y nw t |S rn )r8 Zunwrapr isroutine)rl r Z
maybe_routiner- r- r. _is_routine s
zDocTestFinder._is_routinec
C s | j r
td| t||v rd S d|t|< | |||||}|d ur*|| t|r]| jr]|j D ]%\} }
d|| f } |
|
sKt |
r\| ||
r\|
||
| |||| q7t|r| jrt|di D ]@\} }
t| tstdt| f t|
st |
st|
st|
tstdt|
f d|| f } |
||
| |||| qmt |r| jr|j D ]8\} }
t|
ttfr|
j}
t|
st |
st|
tr| ||
rd|| f } |
||
| |||| qd S d S 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 r8 r9 r r itemsr r r
r r1 r: r; r r r staticmethod classmethod__func__r ) rl r r r, rD r r2 seentestZvalnamevalr- r- r. r sj
zDocTestFinder._findc C s t |tr |}n$z|jd u rd}n |j}t |ts t|}W n
ttfy, d}Y nw | ||}| jr:|s:d S |d u rAd }nt|dd pI|j}|dd dkrX|d d }| j
|||||S )Nr r .pycrt ) r: r; __doc__rC rQ _find_linenor r1 r7 r r ) rl r r, rD r2 r r r rU r- r- r. r s0
zDocTestFinder._get_testc C s6 d }t |dd }t|r|d urd}t|r@|d ur@|d u r#d S tdt |dd }t|D ]
\}}||r?|} nq2t|rH|j }t
|rVt |dd rV|j }t |r^|j
}t|rf|j}t|rst |dd d }|d ur|d u r|d S td}t|t|D ]
}||| r| S qd S ) Nr r" z^\s*class\s*%s\br7 -co_firstlinenor' z(^|.*:)\s*\w*("|\'))r1 r8 r9 r r\ r r r Zismethodr r __code__Z istracebacktb_frameZisframef_codeZiscoderanger* )rl r r r r Zpatr r r- r- r. r > s@
zDocTestFinder._find_linenoNNNN) r7 rr rs r r rx r
r r r r r- r- r- r. r / s
f = &r c @ s e Zd Z dZd ddZdd Zdd Zd
d Zd d
Zdd Z dd Z
dd Z e
dZd ddZd ddZd ddZd d ZdS )!r zF**********************************************************************Nr" c C sP |pt | _|d u rdtjv }|| _|| _|| _d| _d| _i | _ t
| _ d S )N-vr" ) r _checkerr? r r optionflagsoriginal_optionflagstriesfailures_name2ftrh _fakeout)rl checkerr r$ r- r- r. r s
zDocTestRunner.__init__c C sP | j r&|jr|dt|j d t|j d S |dt|j d d S d S )NzTrying:
z Expecting:
zExpecting nothing
)r ry r` r )rl r r exampler- r- r. report_start s
zDocTestRunner.report_startc C s | j r
|d d S d S )Nzok
)r rl r r r+ rz r- r- r. report_success s z DocTestRunner.report_successc C s( || ||| j||| j d S rn )_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` rg rl r r r+ re r- r- r. report_unexpected_exception s
z)DocTestRunner.report_unexpected_exceptionc C s | j g}|jr)|jd ur|jd ur|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:rj )DIVIDERrU r r r, r r` r )rl r r+ r r r r- r- r. r/ s
z DocTestRunner._failure_headerc C sj d }}| j }td\}}} | jj}
t|jD ] \} } | j t@ o%|dk}
|| _ | jrH| j D ]\}}|r?| j |O _ q1| j | M _ q1| j t @ rNq|d7 }|
s[|
||| d|j | f }zt t
| j|d|d|j | j d }W n ty t }| j Y | j }| jd |}|d u r|
| j|| j r|}n9tj|d d d }|
s|t|7 }| j d u r| }n |
| j || j r|}n| j t @ r|
t | j t || j r|}||u r|
s| ||| | n)||u r|
s| ||| | |d7 }n|| u r|
s| !||| | |d7 }|r%| j t"@ r% nq|| _ | #||| t$||S ) Nr" rJ r' zZsingleTr5 rt )%r$ r r# check_outputr r r r r r r, r, execr r r2 debuggerr KeyboardInterruptr? re r) rd rp ry rb format_exception_onlyrg r r r r. r1 r3 r _DocTestRunner__record_outcomer& )rl r compileflagsr r' r& r% SUCCESSFAILUREZBOOMZcheck
examplenumr+ quietZ
optionflagr rU Z exceptionrz Zoutcomer r- r- r. Z__run
s
zDocTestRunner.__runc C sN | j |jd\}}|| || f| j |j< | j|7 _| j|7 _d S )Nr )r( r0 r, r' r& )rl r rX tf2t2r- r- r. Z__record_outcome s
z DocTestRunner.__record_outcomez..+)\[(?P\d+)\]>$c C sR | j |}|r#|d| jjkr#| jjt|d }|jjddS | ||S )Nr, r? Tkeepends)
%_DocTestRunner__LINECACHE_FILENAME_REr r r r, r intr
splitlinessave_linecache_getlines)rl rU module_globalsr r+ r- r- r. Z __patched_linecache_getlines s
z*DocTestRunner.__patched_linecache_getlinesTc s@ || _ |d u r
t|j}tj|d u r,j d u s! dkr%j}n fdd}| jt_t }t
j }t | _
| j
| j
j t
_ tj| _| jt_tj}tjt_z)| |||W t_|t
_ t| | jt_|t_|r~|j dd l}d |_S S t_|t
_ t| | jt_|t_|r|j dd l}d |_w )Nutf-8c s t | d } | d S )Nbackslashreplace)r; encodewrite)r^ rM r r- r. r s z DocTestRunner.run..outr" )r r4 r2 r? r rM lowerrN r) gettracer r r r8 resetr r rI *_DocTestRunner__patched_linecache_getlines displayhook__displayhook___DocTestRunner__runsettraceclearbuiltins_) rl r r< r clear_globsZ
save_traceZsave_set_traceZsave_displayhookrY r- rO r. run sR
zDocTestRunner.runc
C s |d u r| j }g }g }g }d }}| j D ]+}|\}\} }
||
7 }|| 7 }|
dkr1|| q| dkr=|||
f q|| q|rx|r\tt|d | |D ]} td| qT|rxtt|d | |D ] \} } td| | f qk|rt| j tt|d | |D ]\} \} }
td| |
| f q|rt|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 r5 r& )
rl r ZnotestsZpassedZfailedZtotaltZtotalfr r, rX rA thingr r- r- r. summarize sR
zDocTestRunner.summarizec C sR | j }|j D ] \}\}}||v r || \}}|| }|| }||f||< qd S rn )r( r )rl r dr, rX rA rB rC r- r- r. merge! s zDocTestRunner.merge)NNr" rn NNT)r7 rr rs r5 r r, r. r1 r3 r/ rV r; r\ r rF rS r\ r^ r` r- r- r- r. r v s 9
$
}
I 9r c @ s. e Zd Z dd Zdd Zdd Zdd Zd S )
r c C s t |dddS )NASCIIrL )r; rM )rl r^ r- r- r. _toAscii6 s zOutputChecker._toAsciic C s | |}| |}||krdS |t@ s%||fdkr dS ||fdkr%dS |t@ sBtdtt d|}tdd|}||krBdS |t@ rZd| }d| }||krZdS |t
@ ret ||redS dS ) NT)zTrue
z1
)zFalse
z0
z (?m)^%s\s*?$r z(?m)^[^\S\n]+$r[ F) rc r r r\ r] escapeBLANKLINE_MARKERr r rv r r} rl ry rz r$ r- r- r. r6 < s4
zOutputChecker.check_outputc C s< |t tB tB @ s
dS |t@ rdS |ddko |ddkS )NFTrj r5 )r r
r r rf r- r- r. _do_a_fancy_diff{ s z OutputChecker._do_a_fancy_diffc
C s |j }|t@ stdt|}| |||rp|jdd}|jdd}|t@ r9tj ||dd}t
|dd }d}n,|t @ rPtj ||dd}t
|dd }d}n|t
@ retjtjd} t
| ||}d }d
| td | S |r~|r~d t|t|f S |rd
t| S |rdt| S dS )Nz(?m)^[ ]*(?=
)TrD r5 )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
)ry r r\ r] re rg rH r difflibZ unified_difflistr
Z context_diffr ZDifferZIS_CHARACTER_JUNKZcomparer` r )
rl r+ rz r$ ry r Z got_linesZdiffkindZenginer- r- r. r0 s6 z OutputChecker.output_differenceN)r7 rr rs rc r6 rg r0 r- r- r- r. r . s ? r c @ e Zd Z dd Zdd ZdS )r c C || _ || _|| _d S rn )r r+ rz )rl r r+ rz r- r- r. r
zDocTestFailure.__init__c C
t | jS rn r; r r r- r- r. __str__
zDocTestFailure.__str__Nr7 rr rs r rq r- r- r- r. r
r c @ rl )r c C rm rn )r r+ re )rl r r+ re r- r- r. r rn z UnexpectedException.__init__c C ro rn rp r r- r- r. rq rr zUnexpectedException.__str__Nrs r- r- r- r. r rt r c @ s( e Zd Z d ddZdd Zdd ZdS )
r NTc C s$ t | |||d}|r|j |S )NF)r r\ r2 rX )rl r r< r r[ rr- r- r. r\ A s
zDebugRunner.runc C t |||rn )r r2 r- r- r. r3 G z'DebugRunner.report_unexpected_exceptionc C rv rn )r r- r- r- r. r1 J rw zDebugRunner.report_failurera )r7 rr rs r\ r3 r1 r- r- r- r. r s
Z r TFc C s | d u r t jd} t| std| f |d u r | j}t|d} |r,t||d}
nt ||d}
| j
| |||dD ]} |
| q;|rI|
t
d u rP|
a
nt
|
t|
j|
jS )Nr z testmod: module required; %r)r r r$ r2 r )r? r@ r0 r8 r9 rC r7 r r r rx r\ r^ masterr` r& r' r& ) r r, r2 r reportr$ r raise_on_errorr finderrunnerr r- r- r. r V s& B
r c C s |r |s t dt| ||| pd\} } |d u r tj| }|d u r%i }n| }|d ur2|| d|vr:d|d< | rCt||d}
nt||d}
|
| ||| d}|
| |r]|
t d u rd|
a nt
|
t|
j|
jS )N8Package may only be specified for module-relative paths.rK r7 r rx r" )r rY r r basenamer r r r r r\ r^ rz r` r& r' r& )rU rW r, rV r2 r r{ r$ r r| r rM textr~ r r- r- r. r s4 N
r NoNamec C sB t |dd}t||d}|j| ||dD ] }|j||d qd S )NF)r r rx )r2 )r< )r r rx r\ ) rX r2 r r, r< r$ r} r~ r r- r- r. r 9 s r c C s$ | t @ | kr td| t}| a|S )Nz Only reporting flags allowed)r
r _unittest_reportflags)r3 oldr- r- r. r V s
r c @ sp e Zd Z dddZdd Zdd Zd d
Zd d Zd
d Zdd Z dd Z
dd Z dd Z e
jZdd ZdS ) DocTestCaser" Nc C s: t j| || _|| _|j | _|| _|| _ || _
d S rn ) unittestTestCaser _dt_optionflags _dt_checkerr2 r _dt_globs_dt_test _dt_setUp _dt_tearDown)rl r r$ setUptearDownr* r- r- r. r | s
zDocTestCase.__init__c C s" | j }| jd ur| | d S d S rn )r r rl r r- r- r. r s
zDocTestCase.setUpc C s6 | j }| jd ur
| | |j |j| j d S rn )r r r2 rX r r r r- r- r. r s
zDocTestCase.tearDownc C s | j }tj}t }| j}|t@ s|tO }t|| jdd}zd|_ |j
||j dd\}}W |t_n|t_w |r@| |
| d S )NFr$ r* r zF----------------------------------------------------------------------)r r[ )r r? r r# r r
r r r r5 r\ rN ZfailureExceptionformat_failurerd )rl r r newr$ r~ r' r& r- r- r. runTest s$ zDocTestCase.runTestc C sP | j }|jd u r d}nd|j }d|jddd }d|j|j|||f S )Nzunknown line numberz%sr rt z:Failed doctest test for %s
File "%s", line %s, in %s
%s)r r r r, rv rU )rl errr r Zlnamer- r- r. r s
zDocTestCase.format_failurec C s8 | t| j| jdd}|j| jdd | d S )NFr )r[ )r r r r r\ r r )rl r~ r- r- r. r! s A zDocTestCase.debugc C | j jS rn r r, r r- r- r. r zDocTestCase.idc C sP t | t |ur
tS | j|jko'| j|jko'| j|jko'| j|jko'| j|jkS rn )r r r r r r r r r- r- r. r s
zDocTestCase.__eq__c C r rn )r r r r r r r- r- r. r s zDocTestCase.__hash__c C s, | j jd}d|d d|d d f S )Nr z%s (%s)rt )r r, rv r )rl r, r- r- r. r s zDocTestCase.__repr__c C s d| j j S )Nz Doctest: r r r- r- r. shortDescription rw z DocTestCase.shortDescription)r" NNN)r7 rr rs r r r r r r! r r r r r rq r r- r- r- r. r z s
H
r c @ s0 e Zd Zdd Zdd Zdd Zdd ZeZd S )
SkipDocTestCasec C s || _ t| d d S rn )rD r r )rl rD 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. r " s zSkipDocTestCase.setUpc C d S rn r- r r- r- r. test_skip% zSkipDocTestCase.test_skipc C s d| j j S )NzSkipping tests from %s)rD r7 r r- r- r. r ( rw z SkipDocTestCase.shortDescriptionN)r7 rr rs r r r r rq r- r- r- r. r s r c @ s e Zd Zdd ZdS )
_DocTestSuitec C r rn r- )rl indexr- r- r. _removeTestAtIndex0 r z _DocTestSuite._removeTestAtIndexN)r7 rr rs r r- r- r- r. r . s r c K s |d u rt }t| } |j| ||d}|s(tjjdkr(t }|t| |S | t }|D ],}t
|j dkr;q1|j sR| j
}|dd dkrO|d d }||_ |t|fi | q1|S )Nry r5 r" r r rt )r rE rx r? r3 optimizer addTestr r r* r rU r r ) rD r2 r Z test_finderr r suiter rU r- r- r. r 4 s* # r c @ s$ e Zd Zdd Zdd Zdd ZdS ) DocFileCasec C s d | jjdS )NrZ r )r r r, rv r r- r- r. r v s zDocFileCase.idc C r rn )r rU r r- r- r. r y r zDocFileCase.__repr__c C s d| j j| j j|f S )Nz2Failed doctest test for %s
File "%s", line 0
%s)r r, rU )rl r r- r- r. r | s zDocFileCase.format_failureN)r7 rr rs r r r r- r- r- r. r t s r c
K sz |d u ri }n| }|r|stdt| |||pd\}} d|vr&| |d< tj| }||||| d} t| fi |S )Nr rK r r" )r r rY r r r r r )
r rW rV r2 r rM r docr, r r- r- r. DocFileTest s r c O sJ t }|ddrt|d|d< | D ]
}|t|fi | q|S )NrW TrV )r r0 rE r r )pathskwr r r- r- r. r s 7 r c C s g }t | D ]<}t|tr5||jd d |j}|r4|d |dd |dd d D 7 }q |dd |dd d D 7 }q |rZ|d dkrZ| |rZ|d dksN|ro|d dkro|d |ro|d dksbd |d S ) Nrt z # Expected:c S s g | ]}d | qS )z## r- r r- r- r. r '
r z(script_from_examples..rj c S r r- )r r r- r- r. r *
s r~ r" )
r r r: r r r ry rv popr )r^ r Zpiecery r- r- r. r s( 9
r c sL t | } t | } fdd|D }|st d|d }t|j}|S )Nc s g | ] }|j kr|qS r- r+ )r rA r+ r- r. r ?
s z testsource..znot found in testsr" )rE r rx r r r )rD r, r r testsrcr- r+ r. r 6
s
r c C s t | }t||| d S rn )r debug_script)srcpmr2 r r- r- r. r F
s r c C s dd l }|r | }ni }|r>> 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 }qA|j rQ|t
O }|D ];}| drt j
|\}}tj
d| t|d d } tj
d= t| ||d\}
} n
t|d||d\}
} |
r dS qSdS ) Nr" zdoctest runner) 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 ra r z file containing the tests to run)nargsr z.pyrx )rW r r$ r' )argparseArgumentParser add_argumentr( keys
parse_argsra r r Z fail_fastr rk r r rv r? insertr< r r ) r r r Z testfilesr r r rU dirnamer r' rZ r- r- r. _test
sJ
r r )r5 )rZ ) NNNNTr" NFF)Fr Nr" r! )FN)F)SZ
__docformat____all__r/ ri r8 r r r r\ r? rb r ior# r$ collectionsr% r& r( r r r r r r r r r r
r r r r
re ru r4 rE rI rY r` rg rh r} r r r r rN r r r r r r Exceptionr r r rz r r r r r r r r Z TestSuiter r r r r r r r r r! r r
r r7 exitr- r- r- r. s '-
1%.DK l I ; n
h
{
$ $
@
IR
, 3
-