3
\Ǘ � ! @ 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 Z d"d$l
mZ d"d%lmZ ed&d'�Zi Zd(d� Zed�Zed�Zed�Zed�Zed�Zed�ZeeB eB eB eB eB Zed �Zed
�Z ed �Z ed �Z ed�Z ee B e B e B e B Z d)Z!d*Z"d+d,� Z#dpd.d/�Z$d0d1� Z%dqd3d4�Z&d5d6� Z'G d7d8� d8e�Z(d9d:� Z)d;d<� Z*d=d>� Z+G d?d@� d@ej,�Z-dAdB� Z.G dCd� d�Z/G dDd� d�Z0G dEd� d�Z1G dFd� d�Z2G dGd� d�Z3G dHd� d�Z4G dId� de5�Z6G dJd� de5�Z7G dKd� de3�Z8d#a9drdNd�Z:dLd#d#d#d#dLd"d#dMe1� d#f dOd�Z;dsdQd�ZG dSdT� dTe j?�Z@G dUdV� dVe@�ZAG dWdX� dXe jB�ZCdtdYd�ZDG dZd[� d[e@�ZEdLd#d#e1� d#fd\d]�ZFd^d � ZGd_d � ZHd`d � ZIdudad �ZJdvdbdc�ZKdwddd!�ZLG dedf� df�ZMeMdgdhdidjdkdl�ZNdmdn� ZOePdok�rle
jQeO� � d#S )xzreStructuredText en�register_optionflag�DONT_ACCEPT_TRUE_FOR_1�DONT_ACCEPT_BLANKLINE�NORMALIZE_WHITESPACE�ELLIPSIS�SKIP�IGNORE_EXCEPTION_DETAIL�COMPARISON_FLAGS� REPORT_UDIFF� REPORT_CDIFF� REPORT_NDIFF�REPORT_ONLY_FIRST_FAILURE�REPORTING_FLAGS� FAIL_FAST�Example�DocTest�
DocTestParser�
DocTestFinder�
DocTestRunner�
OutputChecker�DocTestFailure�UnexpectedException� DebugRunner�testmod�testfile�run_docstring_examples� DocTestSuite� DocFileSuite�set_unittest_reportflags�script_from_examples�
testsource� debug_src�debug� N)�StringIO)�
namedtuple� TestResultszfailed attemptedc C s t j| dtt �> �S )N� )�OPTIONFLAGS_BY_NAME�
setdefault�len)�name� r+ � /usr/lib64/python3.6/doctest.pyr � s z z...c C s< d}x2t jD ](}| j|d �}|tt |�kr ||jO }q W |S )Nr" )�
__future__Zall_feature_names�get�getattrZ
compiler_flag)�globs�flagsZfnameZfeaturer+ r+ r, �_extract_future_flags� s r2 � c C sV t j| �r| S t| t�r,t| t� t� dg�S | d krJtjtj |�j
d S t d��d S )N�*�__name__z"Expected a module, string, or None) �inspect�ismodule�
isinstance�str�
__import__�globals�locals�sys�modules� _getframe� f_globals� TypeError)�moduleZdepthr+ r+ r, �_normalize_module� s
rC c C s� |r\t |d�}t|| �} t|dd �d k r\t|jd�r\|jj| �}|j|�}|jtj d�| fS t
| |d��}|j � | fS Q R X d S )N� �
__loader__�get_data�
)�encoding) rC �_module_relative_pathr/ �hasattrrE rF �decode�replace�os�linesep�open�read)�filename�package�module_relativerH Z
file_contents�fr+ r+ r, �_load_testfile� s
rU � c C s t jd|d | �S )Nz
(?m)^(?!$)� )�re�sub)�s�indentr+ r+ r, �_indent� s r\ c C s* t � }| \}}}tj||||d� |j� S )N)�file)r# � traceback�print_exception�getvalue)�exc_infoZexcout�exc_typeZexc_valZexc_tbr+ r+ r, �_exception_traceback� s
rc c @ s e Zd Zdd� Zddd�ZdS )� _SpoofOutc C s& t j| �}|r"|jd� r"|d7 }|S )NrG )r# r` �endswith)�self�resultr+ r+ r, r` � s
z_SpoofOut.getvalueNc C s | j |� tj| � d S )N)�seekr# �truncate)rf �sizer+ r+ r, ri s
z_SpoofOut.truncate)N)r5 �
__module__� __qualname__r` ri r+ r+ r+ r, rd � s rd c C s� t | kr| |kS | jt �}dt|� }}|d }|rR|j|�rNt|�}|d= ndS |d }|r�|j|�r||t|�8 }|d= ndS ||kr�dS x2|D ]*}|j|||�}|dk r�dS |t|�7 }q�W dS )Nr" Fr& T�����rm )�ELLIPSIS_MARKER�splitr) �
startswithre �find)�want�gotZws�startpos�endpos�wr+ r+ r, �_ellipsis_match s0
rw c C s | j � } | rd| S dS d S )Nz# �#)�rstrip)�liner+ r+ r, �
_comment_line9 s r{ c C sh dt | � }}| jd�}|dkr$|}| jdd|�}|dkr>|}| jdd|�}|dkr\|d }| ||� S )Nr" rG �:�.r& )r) rq �rfind)�msg�start�end�ir+ r+ r, �_strip_exception_detailsA s
r� c @ s. e Zd Zdd� Zd
dd�Zdd� Zdd � ZdS ) �_OutputRedirectingPdbc C s( || _ d| _tjj| |dd� d| _d S )NFT)�stdout�nosigintr& )�_OutputRedirectingPdb__out�$_OutputRedirectingPdb__debugger_used�pdb�Pdb�__init__Z use_rawinput)rf �outr+ r+ r, r� a s z _OutputRedirectingPdb.__init__Nc C s* d| _ |d krtj� j}tjj| |� d S )NT)r� r= r? �f_backr� r� � set_trace)rf �framer+ r+ r, r� i s
z _OutputRedirectingPdb.set_tracec C s | j rtjj| � d S )N)r� r� r� � set_continue)rf r+ r+ r, r� o 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)rf �args� save_stdoutr+ r+ r, r� u s
z$_OutputRedirectingPdb.trace_dispatch)N)r5 rk rl r� r� r� r� r+ r+ r+ r, r� [ s
r� c C s� t j| �std| ��|jd�r(td��tjj|jd�� }t | d�rXtjj| j
�d }n�| j dkr�t t
j�dkr�t
jd dkr�tjjt
jd �d }q�tj}nFt | d�r�x,| jD ]"}tjj||�}tjj|�r�|S q�W td | 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__))r6 r7 rA rp �
ValueErrorrM �path�joinro rJ r� r5 r) r= �argv�curdirr� �exists)rB Z test_pathZbasedirZ directoryZfullpathr+ r+ r, rI � s&
rI c @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
r Nr" c C sz |j d�s|d7 }|r*|j d� r*|d7 }|d k rF|j d� rF|d7 }|| _|| _|| _|| _|d krji }|| _|| _d S )NrG )re �sourcerr �linenor[ �options�exc_msg)rf r� rr 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)�type�NotImplementedr� rr r� r[ r� r� )rf �otherr+ r+ r, �__eq__� s zExample.__eq__c C s t | j| j| j| j| jf�S )N)�hashr� rr r� r[ r� )rf r+ r+ r, �__hash__� s zExample.__hash__)Nr" r" N)r5 rk rl 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, || _ || _|j� | _|| _|| _|| _d S )N)�examples� docstring�copyr0 r* rQ r� )rf r� r0 r* rQ r� r� r+ r+ r, r�
s
zDocTest.__init__c C sR t | j�dkrd}n"t | j�dkr(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* rQ r� )rf 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* rQ r� )rf r� r+ r+ r, r� # s zDocTest.__eq__c C s t | j| j| j| jf�S )N)r� r� r* rQ r� )rf r+ r+ r, r� . s zDocTest.__hash__c C s: t |t�stS | j| j| jt| �f|j|j|jt|�fk S )N)r8 r r� r* rQ r� �id)rf r� r+ r+ r, �__lt__2 s
zDocTest.__lt__N)r5 rk rl r� r� r� r� r� r+ r+ r+ r, r � s
c @ s� e Zd ZejdejejB �ZejdejejB ejB �Z ejd�j
Z ddd�Z dd� Z
dd d
�Zd d � Zejd
ej�Zdd� Zejdej�Zdd� 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 |j � }| j|�� � dkr8dj� fdd�|jd�D ��}g }d\}}x�| jj|�D ]�}|j|||j� � � ||jd||j� �7 }| j |||�\}}} }
| j
|�s�|jt || |
|� t |j
d�� |d�� ||jd|j� |j� �7 }|j� }qRW |j||d � � |S )Nr" rG c s g | ]}|� d � �qS )Nr+ )�.0�l)�
min_indentr+ r, �
y s z'DocTestParser.parse..r[ )r� r[ r� )r" r" )�
expandtabs� _min_indentr� ro � _EXAMPLE_RE�finditer�appendr� �count�_parse_example�_IS_BLANK_OR_COMMENTr r) �groupr� ) rf �stringr* �outputZcharnor� �mr� r� rr r� r+ )r� r, �parsem s&
zDocTestParser.parsec C s t | j||�|||||�S )N)r � get_examples)rf r� r0 r* rQ r� r+ r+ r, � get_doctest� s zDocTestParser.get_doctestc C s dd� | j ||�D �S )Nc S s g | ]}t |t�r|�qS r+ )r8 r )r� �xr+ r+ r, r� � s z.DocTestParser.get_examples..)r� )rf r� r* r+ r+ r, r� � s zDocTestParser.get_examplesc
s
t |jd��� |jd�jd�}| j|� ||� | j|dd � d� d ||� dj� fdd�|D ��}|jd �}|jd�}t |�dkr�tjd
|d
�r�|d= | j|d� ||t |� � dj� fd d�|D ��}| jj|�}|r�|jd �}nd }| j |||�} || ||fS )Nr[ r� rG r& rW r} c s g | ]}|� d d� �qS )rV Nr+ )r� Zsl)r[ r+ r, r� � s z0DocTestParser._parse_example..rr z *$c s g | ]}|� d � �qS )Nr+ )r� Zwl)r[ r+ r, r� � s r rm rm )
r) r� ro �_check_prompt_blank�
_check_prefixr� rX �match�
_EXCEPTION_RE�
_find_options)
rf r� r* r� � source_linesr� rr �
want_linesr� r� r+ )r[ r, r� � s$
z DocTestParser._parse_examplez#\s*doctest:\s*([^\n\'"]*)$c C s� i }x�| j j|�D ]z}|jd�jdd�j� }x^|D ]V}|d dksR|dd � tkrhtd|d ||f ��t|dd � }|d dk||< q2W qW |r�| j|�r�td|||f ��|S ) Nr& �,rW 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� rL ro r' r� r� ) rf r� r* r� r� r� Zoption_strings�option�flagr+ r+ r, r� � s
zDocTestParser._find_optionsz
^([ ]*)(?=\S)c C s2 dd� | j 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_RE�findallr) �min)rf rZ �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 )NrV rD rW z8line %r of the docstring for %s lacks blank after %s: %rr& )� enumerater) r� )rf �linesr[ r* r� r� rz r+ r+ r, r�
s
z!DocTestParser._check_prompt_blankc C sD x>t |�D ]2\}}|r
|j|� r
td|| d ||f ��q
W d S )NzGline %r of the docstring for %s has inconsistent leading whitespace: %rr& )r� rp r� )rf r� �prefixr* r� r� rz r+ r+ r, r� s zDocTestParser._check_prefixN)r� )r� )r5 rk rl rX �compile� MULTILINE�VERBOSEr� �DOTALLr� r� r� r� r� r� r� r� r� r� r� r� r� r+ r+ r+ r, r = s
'
3 c @ sJ e Zd Zde� ddfdd�Zddd�Zdd � Zd
d � Zd d
� Zdd� Z dS )r FTc C s || _ || _|| _|| _d S )N)�_parser�_verbose�_recurse�_exclude_empty)rf �verbose�parser�recurse�
exclude_emptyr+ r+ r, r� 1 s zDocTestFinder.__init__Nc C sP |d kr.t |dd �}|d kr.tdt|�f ��|dkrr� �����)r/ r� r� r6 � getmoduleZ
getsourcefilerA Zgetfile� linecache�getlines�__dict__r� �update�_find�sort) rf �objr* rB r0 �
extraglobsr] r� �testsr+ r+ r, rq G sJ $
zDocTestFinder.findc C s� |d kr dS t j|�d k r(|t j|�kS t j|�r>|j|jkS t j|�r|t|d�r\|jj}nt|d�rn|j}ndS |j |kS t j
|�r�|j |jkS t|d�r�|j |jkS t |t �r�dS t
d��d S )NT� __objclass__rk z"object must be a class or function)r6 r� �
isfunctionr� � __globals__ZismethoddescriptorrJ r� rk r5 �isclassr8 �propertyr� )rf rB �objectZobj_modr+ r+ r, � _from_module� s(
zDocTestFinder._from_modulec C s( | j rtd| � t|�|kr"d S d|t|�< | j|||||�}|d k rR|j|� tj|�r�| jr�x^|jj � D ]P\} }
d|| f } tj
tj |
��s�tj |
�rn| j
||
�rn| j||
| ||||� qnW tj|�o�| j�rtx�t|di �j � D ]�\} }
t| t��s tdt| �f ��tj
|
��p8tj |
��p8tj|
��p8t|
t��sNtdt|
�f ��d|| f } | j||
| ||||� q�W tj |��r$| j�r$x�|jj � D ]�\} }
t|
t��r�t|| �}
t|
t��r�t|| �j}
tj
|
��s�tj |
��s�t|
t��r�| j
||
��r�d|| f } | j||
| ||||� �q�W 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� ) rf r� r� r* rB r� r0 �seen�testZvalname�valr+ r+ r, r� � sP
zDocTestFinder._findc C s� t |t�r|}nJy,|jd kr"d}n|j}t |t�s:t|�}W n ttfk
rX d}Y nX | j||�}| jrv| rvd S |d kr�d }n*t|d|j�}|dd � dkr�|d d� }| j j
|||||�S )Nr� r� rV z.pycr& �����rm ) r8 r9 �__doc__rA �AttributeError� _find_linenor� r/ r5 r� r� ) rf r� r* rB r0 r� r� r� rQ r+ r+ r, r� s(
zDocTestFinder._get_testc C s d }t j|�rd}t j|�rd|d kr(d S tjdt|dd� �}x$t|�D ]\}}|j|�rH|}P qHW t j|�rt|j }t j
|�r�|j }t j |�r�|j
}t j|�r�|j}t j|�r�t|dd �d }|d k �r |d kr�|d S tjd�}x(t|t|��D ]}|j|| �r�|S q�W d S )Nr" z^\s*class\s*%s\br5 �-�co_firstlinenor& z(^|.*:)\s*\w*("|\'))r6 r7 r� rX r� r/ r� r� Zismethodr� r� �__code__Z istraceback�tb_frameZisframe�f_codeZiscode�ranger) )rf r� r� r� Zpatr� rz r+ r+ r, r - s<
zDocTestFinder._find_lineno)NNNN)
r5 rk rl r r� rq r� r� r� r r+ r+ r+ r, r ( s
f ?%c @ s� e Zd Zdd Zd dd�Zdd� Zd d
� Zd d � Zd
d� Zdd� Z dd� Z
dd� Z e j
d�Zd dd�Zd!dd�Zd"dd �Zd d � ZdS )#r r4 �F Nr" c C sN |pt � | _|d kr dtjk}|| _|| _|| _d| _d| _i | _ t
� | _ d S )Nz-vr" ) r �_checkerr= r� r� � optionflags�original_optionflags�tries�failures�_name2ftrd �_fakeout)rf �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� rr r\ r� )rf r� r� �exampler+ r+ r, � report_start� s zDocTestRunner.report_startc C s | j r|d� d S )Nzok
)r� )rf r� r� r rs r+ r+ r, �report_success� s z DocTestRunner.report_successc C s& || j ||�| jj||| j� � d S )N)�_failure_headerr �output_differencer )rf r� r� r rs r+ r+ r, �report_failure� s z DocTestRunner.report_failurec C s$ || j ||�d tt|�� � d S )NzException raised:
)r r\ rc )rf r� r� r ra 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}|jd|j||jf � n|jd|jd |jf � |jd� |j}|jt|�� dj|�S )Nr& �?zFile "%s", line %s, in %szLine %s, in %szFailed example:rG )�DIVIDERrQ r� r� r* r� r\ r� )rf r� r r� r� r� r+ r+ r, r � s
z DocTestRunner._failure_headerc C s� d }}| j }td�\}}} | jj}
�xFt|j�D �]6\} } | j t@ oL|dk}
|| _ | jr�x:| jj� D ],\}}|r�| j |O _ qf| j | M _ qfW | j t @ r�q2|d7 }|
s�| j
||| � d|j | f }y,t t
| j|d|d�|j� | jj� d }W n4 tk
�r � Y n tj� }| jj� Y nX | jj� }| jjd� |}|d k�rh|
| j|| j ��r�|}n|tj|d d� � d }|
�s�|t|�7 }| j d k�r�| }nB|
| j || j ��r�|}n*| j t @ �r�|
t | j �t |�| j ��r�|}||k�r|
�sZ| j ||| |� nT||k�r0|
�s&| j ||| |� |d7 }n*|| k�rZ|
�sP| j!||| |� |d7 }n |r2| j t"@ r2P q2W || _ | j#|||� t$||�S )Nr" rD r& zZsingler3 rm )%r r r � check_outputr� r� r r� r� r r r* �execr� r� r0 �debuggerr� �KeyboardInterruptr= ra r r` ri rr r^ �format_exception_onlyrc r� r r� r r r r � _DocTestRunner__record_outcomer% )rf r� � compileflagsr� r r r
�SUCCESS�FAILUREZBOOMZcheck�
examplenumr �quietZ
optionflagr� rQ Z exceptionrs Zoutcomer� r+ r+ r, Z__run� s�
zDocTestRunner.__runc C sL | j j|jd�\}}|| || f| j |j< | j|7 _| j|7 _d S )Nr" )r" r" )r r. r* r r )rf r� rT �t�f2�t2r+ r+ r, Z__record_outcomeu s z DocTestRunner.__record_outcomez..+)\[(?P\d+)\]>$c C sV | j j|�}|rF|jd�| jjkrF| jjt|jd�� }|jjdd�S | j ||�S d S )Nr* r% T)�keepends)
�%_DocTestRunner__LINECACHE_FILENAME_REr� r� r� r* r� �intr� �
splitlines�save_linecache_getlines)rf rQ �module_globalsr� r r+ r+ r, Z __patched_linecache_getlines� s
z*DocTestRunner.__patched_linecache_getlinesTc s� || _ |d krt|j�}tj�|d krV�j� � d ks@� j� dkrH�j}n� �fdd�}| jt_tj � }t
j }t ��| _
| j
j� | j
j t
_ tj| _| jt_tj}tjt_z| j|||�S �t_|t
_ tj|� | jt_|t_|r�|jj� dd l}d |_X d S )Nzutf-8c s t | j� d�� �} �j| � d S )N�backslashreplace)r9 �encode�write)rZ )rH r� r+ r, r� � s z DocTestRunner.run..outr" )r� r2 r0 r= r� rH �lowerr2 r �gettracer� r� r� r �resetr� r� r. �*_DocTestRunner__patched_linecache_getlines� displayhook�__displayhook__�_DocTestRunner__run�settrace�clear�builtins�_) rf r� r" r� � clear_globsZ
save_traceZsave_set_traceZsave_displayhookr< r+ )rH r� r, �run� s<
zDocTestRunner.runc
C s� |d kr| j }g }g }g }d }}xd| jj� D ]V}|\}\} }
||
7 }|| 7 }|
dkrb|j|� q.| dkrz|j||
f� q.|j|� q.W |r�|r�tt|�d� |j� x|D ]} td| � q�W |r�tt|�d� |j� x |D ]\} } td| | f � q�W |�rJt| j� tt|�d� |j� x(|D ] \} \} }
td| |
| f � �q&W |�rvt|dt| j�d � t|| d
|d � |�r�td |d
� n|�r�td� 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% )
rf r� ZnotestsZpassedZfailedZtotaltZtotalfr� r* rT r' �thingr� r+ r+ r, � summarize� sP
zDocTestRunner.summarizec C sV | j }xJ|j j� D ]<\}\}}||krB|| \}}|| }|| }||f||< qW d S )N)r r� )rf r� �dr* rT r' r( r) r+ r+ r, �merge s zDocTestRunner.merge)NNr" )N)NNT)N)r5 rk rl r r� r r r r r r9 r! rX r� r+ r6 r? rA rC r+ r+ r+ r, r a s :
$
}
I
9c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
r c C s t |jdd�d�S )N�ASCIIr0 )r9 r1 )rf rZ r+ r+ r, �_toAscii! s zOutputChecker._toAsciic C s� | j |�}| j |�}||kr dS |t@ sH||fd kr8dS ||fd krHdS |t@ s�tjdtjt� d|�}tjdd|�}||kr�dS |t@ r�d j|j � �}d j|j � �}||kr�dS |t
@ r�t ||�r�dS d
S )
NT�True
�1
�False
�0
z (?m)^%s\s*?$r� z
(?m)^\s*?$rW F)rF rG )rH rI ) rE r r rX rY �escape�BLANKLINE_MARKERr r� ro r rw )rf rr rs r r+ r+ r, r ' s0
zOutputChecker.check_outputc C s< |t tB tB @ sdS |t@ r dS |jd�dko:|jd�dkS )NFTrG r3 )r r
r r� )rf rr rs r r+ r+ r, �_do_a_fancy_difff s z OutputChecker._do_a_fancy_diffc
C s8 |j }|t@ s tjdt|�}| j|||�r�|jdd�}|jdd�}|t@ rptj ||dd�}t
|�dd � }d}nZ|t @ r�tj ||dd�}t
|�dd � }d}n,|t
@ r�tjtjd�} t
| j||��}d }n d
d � |D �}d | td
j|�� S |o�|�r dt|�t|�f S |�r dt|� S |�r0dt|� 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 +actualc S s g | ]}|j � d �qS )rG )ry )r� rz r+ r+ r, r� � s z3OutputChecker.output_difference..zDifferences (%s):
r� zExpected:
%sGot:
%szExpected:
%sGot nothing
zExpected nothing
Got:
%sz Expected nothing
Got nothing
)rr r rX rY rK rL r- r �difflibZ unified_diff�listr
Z context_diffr ZDifferZIS_CHARACTER_JUNKZcomparer\ r� )
rf r rs r rr r� Z got_linesZdiffZkindZenginer+ r+ r, r } s6
z OutputChecker.output_differenceN)r5 rk rl rE r rL r r+ r+ r+ r, r s ?c @ s e Zd Zdd� Zdd� ZdS )r c C s || _ || _|| _d S )N)r� r rs )rf r� r rs r+ r+ r, r� � s zDocTestFailure.__init__c C s
t | j�S )N)r9 r� )rf r+ r+ r, �__str__� s zDocTestFailure.__str__N)r5 rk rl r� rP r+ r+ r+ r, r � s c @ s e Zd Zdd� Zdd� ZdS )r c C s || _ || _|| _d S )N)r� r ra )rf r� r ra r+ r+ r, r� � s z UnexpectedException.__init__c C s
t | j�S )N)r9 r� )rf r+ r+ r, rP � s zUnexpectedException.__str__N)r5 rk rl r� rP r+ r+ r+ r, r � s c @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
r NTc C s$ t j| |||d�}|r |jj� |S )NF)r r? r0 r; )rf r� r" r� r> �rr+ r+ r, r? . s
zDebugRunner.runc C s t |||��d S )N)r )rf r� r� r ra r+ r+ r, r 4 s z'DebugRunner.report_unexpected_exceptionc C s t |||��d S )N)r )rf r� r� r rs r+ r+ r, r 7 s zDebugRunner.report_failure)NNT)r5 rk rl r? r r r+ r+ r+ r, r � s [
TFc C s� | d krt jjd�} tj| �s,td| f ��|d kr:| j}t|d�} |rVt||d�}
n t ||d�}
x$| j
| |||d�D ]} |
j | � qvW |r�|
j � t
d kr�|
a
n
t
j|
� t|
j|
j�S )Nr� z testmod: module required; %r)r� )r� r )r0 r� )r= r> r. r6 r7 rA r5 r r r rq r? rA �masterrC r% r r ) r� r* r0 r� �reportr r� �raise_on_errorr� �finder�runnerr� r+ r+ r, r C s$ E
c C s� |r| rt d��t| ||| p d�\} } |d kr )ra r= r� r# r_ r
rZ r r` r r? r2 ZfailureException�format_failurer` )rf r� r[ �newr rV r r r+ r+ r, �runTest� s
zDocTestCase.runTestc C sP | j }|jd krd}n
d|j }dj|jjd�dd � �}d|j|j|||f S )Nzunknown line numberz%sr} r& z:Failed doctest test for %s
File "%s", line %s, in %s
%srm )ra r� r� r* ro rQ )rf �errr� r� Zlnamer+ r+ r, rf � s
zDocTestCase.format_failurec C s6 | j � t| j| jdd�}|j| jdd� | j� d S )NF)r r r� )r> )rd r r_ r` r? ra re )rf rV r+ r+ r, r! � s
B zDocTestCase.debugc C s | j jS )N)ra r* )rf 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� ra r_ rb rc r` )rf r� r+ r+ r, r� � s zDocTestCase.__eq__c C s t | j| j| j| jf�S )N)r� r_ rb rc r` )rf r+ r+ r, r� � s zDocTestCase.__hash__c C s, | j jjd�}d|d dj|d d� �f S )Nr} z%s (%s)r& rm rm )ra r* ro r� )rf r* r+ r+ r, r� � s zDocTestCase.__repr__c C s d| j j S )Nz Doctest: )ra r* )rf r+ r+ r, �shortDescription s z DocTestCase.shortDescription)r" NNN)r5 rk rl r� rd re rh rf r! r� r� r� r� rP rj r+ r+ r+ r, r\ g s
H
r\ c @ s0 e Zd Zdd� Zdd� Zdd� Zdd� ZeZd S )
�SkipDocTestCasec C s || _ tj| d � d S )N)rB r\ r� )rf rB r+ r+ r, r� s zSkipDocTestCase.__init__c C s | j d� d S )Nz-DocTestSuite will not work with -O2 and above)ZskipTest)rf r+ r+ r, rd s zSkipDocTestCase.setUpc C s d S )Nr+ )rf r+ r+ r, � test_skip s zSkipDocTestCase.test_skipc C s d| j j S )NzSkipping tests from %s)rB r5 )rf r+ r+ r, rj s z SkipDocTestCase.shortDescriptionN)r5 rk rl r� rd rl rj rP r+ r+ r+ r, rk s
rk c @ s e Zd Zdd� ZdS )�
_DocTestSuitec C s d S )Nr+ )rf �indexr+ r+ r, �_removeTestAtIndex s z _DocTestSuite._removeTestAtIndexN)r5 rk rl ro r+ r+ r+ r, rm s rm c K s� |d krt � }t| �} |j| ||d�}| rPtjjdkrPt� }|jt| �� |S |j � t� }x\|D ]T}t
|j �dkrxqd|j s�| j
}|dd � dkr�|d d� }||_ |jt|f|�� qdW |S ) N)r0 r� r3 r" rV z.pycr& r rm )r rC rq r= r1 �optimizerm �addTestrk r� r) r� rQ r� r\ ) rB r0 r� Z test_finderr� r� �suiter� rQ r+ r+ r, r s( %
c @ s( e Zd Zdd� Zdd� ZeZdd� ZdS )� DocFileCasec C s dj | jjjd��S )Nr= r} )r� ra r* ro )rf r+ r+ r, r� ` s zDocFileCase.idc C s | j jS )N)ra rQ )rf r+ r+ r, r� c s zDocFileCase.__repr__c C s d| j j| j j|f S )Nz2Failed doctest test for %s
File "%s", line 0
%s)ra r* rQ )rf ri r+ r+ r, rf g s zDocFileCase.format_failureN)r5 rk rl r� r� rP rf r+ r+ r+ r, rs ^ s rs c
K sx |d kri }n|j � }|r(| r(td��t| |||p6d�\}} d|krN| |d< tjj| �}|j|||| d�} t| f|�S )Nz8Package may only be specified for module-relative paths.zutf-8r� r" )r� r� rU rM r� rW r� rs )
r� rS rR r0 r� rH r� �docr* r� r+ r+ r, � DocFileTestl s
ru c O sH t � }|jdd�r$t|jd��|d< x | D ]}|jt|f|�� q*W |S )NrS TrR )rm r. rC rq ru )�paths�kwrr r� r+ r+ r, r � s 8
c C s� g }x�t � j| �D ]x}t|t�rj|j|jd d � � |j}|r�|jd� |dd� |jd�d d
� D �7 }q|dd� |jd�d d � D �7 }qW x |r�|d dkr�|j� q�W x |r�|d dkr�|jd� q�W dj |�d S )
Nr& z # Expected:c S s g | ] }d | �qS )z## r+ )r� r� r+ r+ r, r�
s z(script_from_examples..rG c S s g | ] }t |��qS r+ )r{ )r� r� r+ r+ r, r�
s rx r" rm rm rm rm )
r r� r8 r r� r� rr ro �popr� )rZ r� Zpiecerr r+ r+ r, r � s :
" c sJ t | �} t� j| �}� 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" )rC r rq r� r r� )rB r* r� r� �testsrcr+ )r* r, r !
s
c C s t | �}t|||� d S )N)r � debug_script)�src�pmr0 ry r+ r+ r, r 1
s c
C s� dd l }|r|j� }ni }|rvyt| ||� W q� ttj� d � |jdd�}|j� |jd tj� d � Y q�X n|jdd�j d| ||� d S )Nr" r& T)r� r3 zexec(%r))
r� r� r r� r= ra r� r5 Z interactionr? )r{ r| r0 r� �pr+ r+ r, rz 6
s
rz c C s$ t | �} t| |�}t||| j� d S )N)rC r rz r� )rB r* r| ry r+ r+ r, r! J
s
c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�
_TestClassc C s
|| _ d S )N)r� )rf r� r+ r+ r, r� f
s z_TestClass.__init__c C s | j d | _ | S )Nr3 )r� )rf r+ r+ r, �squarep
s z_TestClass.squarec C s | j S )N)r� )rf r+ r+ r, r. z
s z_TestClass.getN)r5 rk rl r� r r. 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~ r� zbool-int equivalencez blank linesZellipsiszwhitespace normalizationc C s t jdd�} | jdddddd� | jd d
d tj� g d d
� | jddddd� | jdddd� | j� }|j}|j}d}x|jD ]}|t| O }q|W |j r�|t
O }x||D ]t}|j d�r�t j
j|�\}}tj
jd|� t|d d � �}tj
d= t|||d�\} }
nt|d||d�\} }
| r�dS q�W dS ) Nzdoctest runner)Z descriptionz-vz --verbose�
store_trueFz'print very verbose output for all tests)�action�default�helpz-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� r" z.pyrD )r� r )rS r� r r& �����)�argparse�ArgumentParser� add_argumentr' �keys�
parse_argsr] r� r� Z fail_fastr re rM r� ro r= �insertr: r r ) r� r� Z testfilesr� r� r� rQ �dirnamer� r r= r+ r+ r, �_test�
s<
r� r� )r3 )rV ) NNNNTr" NFF)FrY Nr" )NNNN)FN)FN)F)RZ
__docformat__�__all__r- r� rN r6 r� rM r� rX r= r^ r] �ior# � collectionsr$ r% r' r r r r r r r r r r
r r r r
rK rn r2 rC rU r\ rc rd rw r{ r� r� r� rI r r r r r r � Exceptionr r r rR r r r rZ r r^ r\ rk Z TestSuiterm r rs ru r r r r rz r! r~ r� r� r5 �exitr+ r+ r+ r, �0 s
1%.DK l ; ; n
f x
$ !
@ IR
, +