a
XC?hx� � @ s2 d Z g d�ZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl m
Z
ddlmZ edd�Zi Zdd � Zed
�Zed �Zed �Zed
�Zed�Zed�ZeeB eB eB eB eB Zed�Zed�Zed�Z ed�Z ed�Z eeB e B e B e B Z dZ dZ!dd� Z"dqdd�Z#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� d2�Z/G d3d4� d4�Z0G d5d6� d6�Z1G d7d8� d8�Z2G d9d:� d:�Z3G d;d<� d<�Z4G d=d>� d>e5�Z6G d?d@� d@e5�Z7G dAdB� dBe3�Z8da9dsdEdF�Z:dCdddddCdddDe1� df dGdH�Z;dtdJdK�ZG dNdO� dOe j?�Z@G dPdQ� dQe@�ZAG dRdS� dSe jB�ZCdudTdU�ZDG dVdW� dWe@�ZEdCdde1� dfdXdY�ZFdZd[� ZGd\d]� ZHd^d_� ZIdvd`da�ZJdwdbdc�ZKdxddde�ZLG dfdg� dg�ZMeMdhdidjdkdldm�ZNdndo� ZOePdpk�r.e �QeO� � dS )yzreStructuredText 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 �| dtt �> �S )N� )�OPTIONFLAGS_BY_NAME�
setdefault�len��name� r, �,/opt/alt/python39/lib64/python3.9/doctest.pyr � s r r r r r r r r r
r r r z z...c C s8 d}t jD ](}| �|d �}|tt |�u r
||jO }q
|S )Nr" )�
__future__Zall_feature_names�get�getattrZ
compiler_flag)�globs�flagsZfnameZfeaturer, r, r- �_extract_future_flags� s
r3 � c C sV t �| �r| S t| t�r,t| t� t� dg�S | d u rJtjt� |�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
rD c C s dD ]}| � |d�} q| S )N)z
�
�
)�replace)�data�newliner, r, r- �_newline_convert� s rJ c C s� |rVt |d�}t|| �} t|dd �d urVt|jd�rV|j�| �}|�|�}t|�| fS t| |d�� }|� � | fW d � S 1 s�0 Y d S )N� �
__loader__�get_data)�encoding)
rD �_module_relative_pathr0 �hasattrrL rM �decoderJ �open�read)�filename�package�module_relativerN Z
file_contents�fr, r, r- �_load_testfile� s
rX � c C s t �d|d | �S )Nz
(?m)^(?!$)� )�re�sub)�s�indentr, r, r- �_indent� s r_ c C s* t � }| \}}}tj||||d� |�� S )N)�file)r# � traceback�print_exception�getvalue)�exc_infoZexcout�exc_typeZexc_valZexc_tbr, r, r- �_exception_traceback� s
rf c @ s e Zd Zdd� Zddd�ZdS )� _SpoofOutc C s$ t �| �}|r |�d�s |d7 }|S �NrF )r# rc �endswith)�self�resultr, 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 | vr| |kS | �t �}dt|� }}|d }|rR|�|�rNt|�}|d= ndS |d }|r�|�|�r||t|�8 }|d= ndS ||kr�dS |D ],}|�|||�}|dk r� dS |t|�7 }q�dS )Nr" F�����T)�ELLIPSIS_MARKER�splitr) �
startswithri �find)�want�gotZws�startpos�endpos�wr, 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)�msg�start�end�ir, r, r- �_strip_exception_detailsG 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)rj �outr, r, r- r� g s z _OutputRedirectingPdb.__init__Nc C s* d| _ |d u rt�� 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_continue�rj r, r, r- r� u s z"_OutputRedirectingPdb.set_continuec G s8 t j}| jt _z tjj| g|�R � W |t _S |t _0 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| ��|�d�r(td��tjj|�d�� }t | d�rXtj�| j
�d }n�| j dkr�t t
j�dkr�t
jd dkr�tj�t
jd �d }q�tj}nFt | d�r�| jD ]&}tj�||�}tj�|�r�| S q�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__))r7 r8 rB ru �
ValueError�os�path�joinrt rP r� r6 r) r> �argv�curdirr� �exists)rC Z test_pathZbasedirZ directory�fullpathr, r, r- rO � s(
�rO c @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
r Nr" c C sv |� d�s|d7 }|r(|� d�s(|d7 }|d urB|� d�sB|d7 }|| _|| _|| _|| _|d u rfi }|| _|| _d S rh )ri �sourcerw �linenor^ �options�exc_msg)rj r� rw r� r� r^ r� r, r, r- r� � s
zExample.__init__c C s\ t | �t |�urtS | j|jkoZ| j|jkoZ| j|jkoZ| j|jkoZ| j|jkoZ| j|jkS rl )�type�NotImplementedr� rw r� r^ r� r� �rj �otherr, r, r- �__eq__� s
�
�
�
�
�zExample.__eq__c C s t | j| j| j| j| jf�S 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
# �
r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) r c C s, || _ || _|�� | _|| _|| _|| _d S rl )�examples� docstring�copyr1 r+ rT r� )rj r� r1 r+ rT 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__r6 r+ rT r� )rj r� r, r, r- �__repr__ s ��zDocTest.__repr__c C s\ t | �t |�urtS | 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| jf�S rl )r� r� r+ rT r� r� r, r, r- r� 4 s zDocTest.__hash__c C s: t |t�stS | 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
r c @ s� e Zd Ze�dejejB �Ze�dejejB ejB �Z e�d�j
Z ddd�Z dd� Z
dd d
�Zd d � Ze�d
ej�Zdd� Ze�dej�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 |� � }| �|�� � dkr8d�� fdd�|�d�D ��}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, ��.0�l�Z
min_indentr, r- �
� z'DocTestParser.parse..�r" r" r^ )r� r^ r� )�
expandtabs� _min_indentr� rt � _EXAMPLE_RE�finditer�appendr� �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 |t�r|�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��� |�d��d�}| �|� ||� | �|dd � d� d ||� d�� fdd�|D ��}|�d �}|�d�}t |�dkr�t�d
|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� Zsl�r^ r, r- r� � r� z0DocTestParser._parse_example..rw z *$rr c s g | ]}|� d � �qS rl r, )r� Zwlr� r, r- r� � r� 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}|�d��dd��� }|D ]V}|d dvsN|dd � tvrdtd|d ||f ��t|dd � }|d dk||< q.q|r�| �|�r�td|||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_strings�option�flagr, 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�
r� z-DocTestParser._min_indent..r" )�
_INDENT_RE�findallr) �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� MULTILINE�VERBOSEr� �DOTALLr� r� r� r� r� r� r� r� r� r� r� r� r� r, r, r, r- r C s&
� �
'
3� r c @ sJ e Zd Zde� ddfdd�Zddd�Zdd � Zd
d � Zd d
� Zdd� Z dS )r FTc C s || _ || _|| _|| _d S rl )�_parser�_verbose�_recurse�_exclude_empty)rj �verbose�parser�recurse�
exclude_emptyr, r, r- r� 7 s zDocTestFinder.__init__Nc C sL |d u r.t |dd �}|d u r.tdt|�f ��|du rr� )r0 r� r� r7 � getmoduleZ
getsourcefilerB Zgetfile� linecache�getlines�__dict__r� �update�_find�sort) rj �objr+ rC r1 �
extraglobsr` r� �testsr, r, r- rv M sJ $ �
zDocTestFinder.findc C s� |d u r dS t �|�d ur(|t �|�u S t �|�r>|j|ju S t �|�r|t|d�r\|jj}nt|d�rn|j}ndS |j |kS t �
|�r�|j |jkS t|d�r�|j |jkS t |t �r�dS t
d��d 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|�|v r"d S d|t|�< | �|||||�}|d urR|�|� t�|�r�| jr�|j� � D ]P\} }
d|| f } t�
t� |
��s�t� |
�rl| �
||
�rl| �||
| ||||� qlt�|��rn| j�rnt|di �� � D ]�\} }
t| t��stdt| �f ��t�
|
��sJt� |
��sJt�|
��sJt|
t��sJtdt|
�f ��d|| f } | �||
| ||||� q�t� |��r| j�r|j� � D ]t\} }
t|
ttf��r�|
j}
t�
|
��s�t� |
��s�t|
t��r�| �
||
��r�d|| f } | �||
| ||||� �q�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� r7 r8 r� r� �itemsZ isroutineZunwrapr� r r� r0 r9 r: r� r� � staticmethod� classmethod�__func__r ) rj r� r� r+ rC r� r1 �seen�testZvalname�valr, r, r- r� � sd
�
�� ���� ��
� �zDocTestFinder._findc C s� t |t�r|}nHz,|jd u r"d}n|j}t |t�s:t|�}W n ttfyV d}Y n0 | �||�}| jrr|srd S |d u r�d }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 u r(d S t�dt|dd� �}t|�D ]\}}|�|�rF|} qbqFt �|�rr|j }t �
|�r�|j }t � |�r�|j
}t �|�r�|j}t �|�r�t|dd �d }|d u�r
|d u r�|d S t�d�}t|t|��D ]}|�|| �r�| S q�d 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 istraceback�tb_frameZisframe�f_codeZiscode�ranger) )rj r� r� r� Zpatr� r r, r, r- r 2 s6
�
zDocTestFinder._find_lineno)NNNN)
r6 rp rq r r� rv r r� r r r, r, r, r- r . s �
f =&r c @ s� e Zd ZdZd dd�Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z e �
d�Zd dd�Zd dd�Zd!dd�Zd d � ZdS )"r zF**********************************************************************Nr" c C sN |pt � | _|d u r dtjv }|| _|| _|| _d| _d| _i | _ t
� | _ d S )N�-vr" ) r �_checkerr> r� r� � optionflags�original_optionflags�tries�failures�_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 ur4|jd ur4|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|j�D �]4\} } | j t@ oH|dk}
|| _ | jr�| j�� D ],\}}|r|| j |O _ q`| j | M _ q`| j t @ r�q.|d7 }|
s�| �
||| � d|j | f }z,t t
| j|d|d�|j� | j�� d }W n2 t�y � Y n t�� }| j�� Y n0 | j�� }| j�d� |}|d u �r^|
| j|| j ��r�|}n|tj|d d� � d }|
�s�|t|�7 }| j d u �r�| }nB|
| j || j ��r�|}n*| j t @ �r�|
t | j �t |�| j ��r�|}||u �r�|
�sP| � ||| |� nT||u �r&|
�s | � ||| |� |d7 }n*|| u �rP|
�sF| �!||| |� |d7 }n |r.| j t"@ r. �qfq.|| _ | �#|||� t$||�S ) Nr" rK r& zZsingleTr4 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 �SUCCESS�FAILUREZBOOMZcheck�
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 �t�f2�t2r, r, r- Z__record_outcomez s z DocTestRunner.__record_outcomez..+)\[(?P\d+)\]>$c C sV | j �|�}|rF|�d�| jjkrF| jjt|�d�� }|jjdd�S | � ||�S d S )Nr+ r5 T��keepends)
�%_DocTestRunner__LINECACHE_FILENAME_REr� r� r r+ r� �intr� �
splitlines�save_linecache_getlines)rj rT �module_globalsr� r! r, r, r- Z __patched_linecache_getlines� s
z*DocTestRunner.__patched_linecache_getlinesTc sD || _ |d u rt|j�}tj�|d u rV�j� � d u s@� �� dkrH�j}n� �fdd�}| jt_t� � }t
j }t ��| _
| j
�� | j
j t
_ tj| _| jt_tj}tjt_zR| �|||�W �t_|t
_ t�|� | jt_|t_|r�|j�� dd l}d |_S nD�t_|t
_ t�|� | jt_|t_|�r>|j�� dd l}d |_0 d S )N�utf-8c s t | �� d�� �} ��| � d S )N�backslashreplace)r: �encode�write)r] �rN r� r, r- r� � s z DocTestRunner.run..outr" )r r3 r1 r> r� rN �lowerrD r �gettracer� r� r� r. �resetr� r� r? �*_DocTestRunner__patched_linecache_getlines� displayhook�__displayhook__�_DocTestRunner__run�settrace�clear�builtins�_) rj r r2 r� � clear_globsZ
save_traceZsave_set_traceZsave_displayhookrO r, rE r- �run� sb
� �
zDocTestRunner.runc
C s� |d u r| j }g }g }g }d }}| j�� D ]V}|\}\} }
||
7 }|| 7 }|
dkr`|�|� q,| dkrx|�||
f� q,|�|� q,|r�|r�tt|�d� |�� |D ]} td| � q�|r�tt|�d� |�� |D ]\} } td| | f � q�|�r:t| j� tt|�d� |�� |D ] \} \} }
td| |
| f � �q|�rft|dt| j�d � t|| d
|d � |�rztd |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% )
rj r� ZnotestsZpassedZfailedZtotaltZtotalfr� r+ rW r7 �thingr� r, r, r- � summarize� sP
zDocTestRunner.summarizec C sR | j }|j �� D ]<\}\}}||v r@|| \}}|| }|| }||f||< qd S rl )r r )rj r� �dr+ rW r7 r8 r9 r, r, r- �merge s zDocTestRunner.merge)NNr" )N)NNT)N)r6 rp rq r+ r� r" r$ r' r) r% rL r1 r[ r� r< rI rR rT rV r, r, r, r- r f s :
$
}
I
9r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
r c C s t |�dd�d�S )N�ASCIIrB )r: rC )rj r] r, r, r- �_toAscii& s zOutputChecker._toAsciic C s� | � |�}| � |�}||kr dS |t@ sH||fdkr8dS ||fdkrHdS |t@ s�t�dt�t� d|�}t�dd|�}||kr�dS |t@ r�d�|� � �}d�|� � �}||kr�dS |t
@ r�t ||�r�dS dS ) NT)zTrue
z1
)zFalse
z0
z (?m)^%s\s*?$r� z(?m)^[^\S\n]+$rZ F) rX r r r[ r\ �escape�BLANKLINE_MARKERr r� rt r r| �rj rw rx r r, r, r- r, , s2
�
zOutputChecker.check_outputc C s< |t tB tB @ sdS |t@ r dS |�d�dko:|�d�dkS )NFTrF r4 )r r
r r� r[ r, r, r- �_do_a_fancy_diffk s �� z OutputChecker._do_a_fancy_diffc
C s( |j }|t@ s t�dt|�}| �|||�r�|jdd�}|jdd�}|t@ rptj ||dd�}t
|�dd � }d}nZ|t @ r�tj ||dd�}t
|�dd � }d}n,|t
@ r�tjtjd�} t
| �||��}d }n d
| td �|�� S |r�|r�d t|�t|�f S |�rd
t|� S |�r dt|� S dS d S )Nz(?m)^[ ]*(?=
)Tr: 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\ rZ r\ r> r �difflibZ unified_diff�listr
Z context_diffr ZDifferZIS_CHARACTER_JUNKZcomparer_ r� )
rj r! rx r rw r� Z got_linesZdiff�kindZenginer, r, r- r& � s4 z OutputChecker.output_differenceN)r6 rp rq rX r, r\ r& r, r, r, r- r s ?r 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 | j�S rl �r: r r� r, r, r- �__str__� s zDocTestFailure.__str__N�r6 rp rq r� rb r, r, r, r- r � s r 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 | j�S rl ra r� r, r, r- rb � s zUnexpectedException.__str__Nrc r, r, r, r- r � s r c @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
r NTc C s$ t �| |||d�}|r |j�� |S )NF)r rR r1 rN )rj r r2 r� rQ �rr, r, r- rR 1 s
zDebugRunner.runc C s t |||��d S rl )r r( r, r, r- r) 7 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 rR r) r' r, r, r, r- r � s [
r TFc C s� | d u rt j�d�} t�| �s,td| f ��|d u r:| j}t|d�} |rVt||d�}
n t ||d�}
| j
| |||d�D ]} |
� | � qt|r�|
� � t
d u r�|
a
n
t
�|
� t|
j|
j�S )Nr� z testmod: module required; %r)r� �r� r �r1 r� )r> r? r/ r7 r8 rB r6 r r r rv rR rT �masterrV r% r r ) r� r+ r1 r� �reportr r� �raise_on_errorr� �finder�runnerr r, r, r- r F s$ E
r c C s� |r|st d��t| ||| p d�\} } |d u r:tj�| �}|d u rHi }n|�� }|d urb|�|� d|vrrd|d< | r�t||d�}
n t||d�}
|
� | ||| d�}|
�
|� |r�|
� � t d u r�|
a n
t �
|
� t|
j|
j�S )N�8Package may only be specified for module-relative paths.rA r6 r� re r" )r� rX r� r� �basenamer� r� r r r� rR rT rg rV r% r r )rT rV r+ rU r1 r� rh r r� ri r� rN �textrk r r, r, r- r � s2 R�
r �NoNamec C s@ t |dd�}t||d�}|j| ||d�D ]}|j||d� q(d S )NF)r� r� re )r1 )r2 )r r rv rR ) rW r1 r� r+ r2 r rj rk r r, r, r- r ) s r c C s" | t @ | krtd| ��t}| a|S )Nz Only reporting flags allowed)r
r� �_unittest_reportflags)r2 �oldr, r, r- r F s
r c @ sl e Zd Zddd�Zdd� 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 rl ) �unittest�TestCaser� �_dt_optionflags� _dt_checkerr1 r� � _dt_globs�_dt_test� _dt_setUp� _dt_tearDown)rj r r �setUp�tearDownr r, r, r- r� l s zDocTestCase.__init__c C s | j }| jd ur| �|� d S rl )rx ry �rj r r, r, r- r{ w s
zDocTestCase.setUpc C s6 | j }| jd ur| �|� |j�� |j�| j� d S rl )rx rz r1 rN r� rw r} r, r, r- r| } s
zDocTestCase.tearDownc C s� | j }tj}t� }| j}|t@ s(|tO }t|| jdd�}z&d|_ |j
||j dd�\}}W |t_n|t_0 |r�| � | �
|�� ���d S )NF�r r r� zF----------------------------------------------------------------------)r� rQ )rx r> r� r# ru r
rp r rv r+ rR rD ZfailureException�format_failurerc )rj r rq �newr rk r r r, r, r- �runTest� s" �� zDocTestCase.runTestc C sP | j }|jd u rd}n
d|j }d�|j�d�dd � �}d|j|j|||f S )Nzunknown line numberz%sr� rr z:Failed doctest test for %s
File "%s", line %s, in %s
%s)rx 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~ )rQ )r{ r ru rv rR rx r| )rj rk r, r, r- r! � s B�zDocTestCase.debugc C s | j jS rl �rx r+ r� r, r, r- r� � s zDocTestCase.idc C sP t | �t |�urtS | j|jkoN| j|jkoN| j|jkoN| j|jkoN| j|jkS rl )r� r� rx ru ry rz rv r� r, r, r- r� � s
�
�
�
�zDocTestCase.__eq__c C s t | j| j| j| jf�S rl )r� ru ry rz rv r� r, r, r- r� s �zDocTestCase.__hash__c C s, | j j�d�}d|d d�|d d� �f S )Nr� z%s (%s)rr )rx 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� r{ r| r� r r! r� r� r� r� r rb r� r, r, r, r- rr j s �
H
rr c @ s0 e Zd Zdd� Zdd� Zdd� Zdd� ZeZd S )
�SkipDocTestCasec C s || _ t�| d � d S rl )rC rr 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- r{ 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� r{ r� r� rb 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 u rt � }t| �} |j| ||d�}|sNtjjdkrNt� }|�t| �� |S |� � t� }|D ]X}t
|j �dkrtq`|j s�| j
}|dd � dkr�|d d� }||_ |�t|fi |��� q`|S )Nrf r4 r" r
r rr )r rD rv r> r2 �optimizer� �addTestr� r� r) r� rT r� rr ) rC r1 r� Z test_finderr� r� �suiter rT r, r, r- r $ s( % r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� DocFileCasec C s d� | jj�d��S )NrP r� )r� rx r+ rt r� r, r, r- r� f s zDocFileCase.idc C s | j jS rl )rx rT r� r, r, r- r� i s zDocFileCase.__repr__c C s d| j j| j j|f S )Nz2Failed doctest test for %s
File "%s", line 0
%s)rx r+ rT )rj r� r, r, r- r l s �zDocFileCase.format_failureN)r6 rp rq r� r� r r, r, r, r- r� d s r� c
K sz |d u ri }n|� � }|r&|s&td��t| |||p4d�\}} d|vrL| |d< tj�| �}|�|||| d�} t| fi |��S )Nrl rA r� r" )r� r� rX r� r� rm r� r� )
r� rV rU r1 r� rN r� �docr+ r r, r, r- � DocFileTestq s � r� c O sH t � }|�dd�r$t|�d��|d< | D ]}|�t|fi |��� q(|S )NrV TrU )r� r/ rD r� r� )�paths�kwr� r� r, r, r- r � s 8 r c C s� g }t � �| �D ]x}t|t�rh|�|jd d� � |j}|r�|�d� |dd� |�d�d d� D �7 }q|dd� |�d�d d� D �7 }q|r�|d dkr�|�� q�|r�|d dkr�|�d� q�d� |�d S ) Nrr z # Expected:c S s g | ] }d | �qS )z## r, r� r, r, r- r�
r� 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 :
"�
r c sJ t | �} t� �| �}� fdd�|D �}|s4t� d��|d }t|j�}|S )Nc s g | ]}|j � kr|�qS r, r* )r� r7 r* r, r- r� /
r� z testsource..znot found in testsr" )rD r rv r� r r� )rC r+ r� r �testsrcr, r* r- r &
s
r c C s t | �}t|||� d S rl )r � debug_script)�src�pmr1 r� r, r, r- r 6
s r c C s� dd l }|r|�� }ni }|rvzt| ||� W q� tt�� d � |jdd�}|�� |�d t�� d � Y q�0 n|jdd�� d| ||� d S )Nr" r& T)r� r4 zexec(%r))
r� r� r- r r> rd r� rH Z interactionrR )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! O
s
r! c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�
_TestClassc C s
|| _ d S rl �r )rj r r, r, r- r� k
s z_TestClass.__init__c C s | j d | _ | S )Nr4 r� r� r, r, r- �squareu
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}|� d�r�t j
�|�\}}tj
�d|� t|d d� �} tj
d= t| ||d�\}
} nt|d||d�\}
} |
r� dS q�dS ) Nr" zdoctest runner)Z descriptionr z --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� z.py�����re )rV r� r r& )�argparse�ArgumentParser� 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 rP r, r, r- �_test�
sH �
�
��
�
r� r� )r4 )rY ) NNNNTr" NFF)Fro Nr" )NNNN)FN)FN)F)RZ
__docformat__�__all__r. r^ r7 r� r� r� r[ r> ra rs �ior# � collectionsr$ r% r' r r r r r r r r r r
r r r r
rZ rs r3 rD rJ rX r_ rf rg r| r� r� r� r� rO r r r r r r � Exceptionr r r rg r r r rp r rt rr r� Z TestSuiter� r r� r� r r r r r� r! r� r r� r6 �exitr, r, r, r- �0 s� -
���������
1%.DK l : ; n �
h�
{ �
$ $
@
�
IR
, �3-