B
    18™\W  ã               @   sn   d Z ddlZddlZddlZddlmZ ddlmZ dZG dd„ de	ƒZ
G d	d
„ d
ejƒZG dd„ de	ƒZdS )zRunning testsé    Né   )Úresult)ÚregisterResultTc               @   s*   e Zd ZdZdd„ Zdd„ Zd	dd„ZdS )
Ú_WritelnDecoratorz@Used to decorate file-like objects with a handy 'writeln' methodc             C   s
   || _ d S )N)Ústream)Úselfr   © r   úS/oak/stanford/groups/akundaje/marinovg/programs/Python-3.7.3/Lib/unittest/runner.pyÚ__init__   s    z_WritelnDecorator.__init__c             C   s   |dkrt |ƒ‚t| j|ƒS )N)r   Ú__getstate__)ÚAttributeErrorÚgetattrr   )r   Úattrr   r   r	   Ú__getattr__   s    z_WritelnDecorator.__getattr__Nc             C   s   |r|   |¡ |   d¡ d S )NÚ
)Úwrite)r   Úargr   r   r	   Úwriteln   s    
z_WritelnDecorator.writeln)N)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r
   r   r   r   r   r   r	   r      s   r   c                   s”   e Zd ZdZdZdZ‡ fdd„Zdd„ Z‡ fdd	„Z‡ fd
d„Z	‡ fdd„Z
‡ fdd„Z‡ fdd„Z‡ fdd„Z‡ fdd„Zdd„ Zdd„ Z‡  ZS )ÚTextTestResultzhA test result class that can print formatted text results to a stream.

    Used by TextTestRunner.
    zF======================================================================zF----------------------------------------------------------------------c                s8   t t| ƒ |||¡ || _|dk| _|dk| _|| _d S )Nr   )Úsuperr   r
   r   ÚshowAllÚdotsÚdescriptions)r   r   r   Ú	verbosity)Ú	__class__r   r	   r
   %   s
    

zTextTestResult.__init__c             C   s0   |  ¡ }| jr$|r$d t|ƒ|f¡S t|ƒS d S )Nr   )ÚshortDescriptionr   ÚjoinÚstr)r   ÚtestZdoc_first_liner   r   r	   ÚgetDescription,   s    
zTextTestResult.getDescriptionc                sB   t t| ƒ |¡ | jr>| j |  |¡¡ | j d¡ | j ¡  d S )Nz ... )r   r   Ú	startTestr   r   r   r#   Úflush)r   r"   )r   r   r	   r$   3   s
    zTextTestResult.startTestc                sD   t t| ƒ |¡ | jr$| j d¡ n| jr@| j d¡ | j ¡  d S )NÚokÚ.)	r   r   Ú
addSuccessr   r   r   r   r   r%   )r   r"   )r   r   r	   r(   :   s    zTextTestResult.addSuccessc                sF   t t| ƒ ||¡ | jr&| j d¡ n| jrB| j d¡ | j ¡  d S )NÚERRORÚE)	r   r   ÚaddErrorr   r   r   r   r   r%   )r   r"   Úerr)r   r   r	   r+   B   s    zTextTestResult.addErrorc                sF   t t| ƒ ||¡ | jr&| j d¡ n| jrB| j d¡ | j ¡  d S )NÚFAILÚF)	r   r   Ú
addFailurer   r   r   r   r   r%   )r   r"   r,   )r   r   r	   r/   J   s    zTextTestResult.addFailurec                sL   t t| ƒ ||¡ | jr,| j d |¡¡ n| jrH| j d¡ | j 	¡  d S )Nzskipped {0!r}Ús)
r   r   ÚaddSkipr   r   r   Úformatr   r   r%   )r   r"   Úreason)r   r   r	   r1   R   s    zTextTestResult.addSkipc                sF   t t| ƒ ||¡ | jr&| j d¡ n| jrB| j d¡ | j ¡  d S )Nzexpected failureÚx)	r   r   ÚaddExpectedFailurer   r   r   r   r   r%   )r   r"   r,   )r   r   r	   r5   Z   s    z!TextTestResult.addExpectedFailurec                sD   t t| ƒ |¡ | jr$| j d¡ n| jr@| j d¡ | j ¡  d S )Nzunexpected successÚu)	r   r   ÚaddUnexpectedSuccessr   r   r   r   r   r%   )r   r"   )r   r   r	   r7   b   s    z#TextTestResult.addUnexpectedSuccessc             C   s6   | j s| jr| j ¡  |  d| j¡ |  d| j¡ d S )Nr)   r-   )r   r   r   r   ÚprintErrorListÚerrorsÚfailures)r   r   r   r	   ÚprintErrorsj   s    
zTextTestResult.printErrorsc             C   s\   xV|D ]N\}}| j  | j¡ | j  d||  |¡f ¡ | j  | j¡ | j  d| ¡ qW d S )Nz%s: %sz%s)r   r   Ú
separator1r#   Ú
separator2)r   Zflavourr9   r"   r,   r   r   r	   r8   p   s
    zTextTestResult.printErrorList)r   r   r   r   r<   r=   r
   r#   r$   r(   r+   r/   r1   r5   r7   r;   r8   Ú__classcell__r   r   )r   r	   r      s   r   c               @   s4   e Zd ZdZeZdddœdd„Zd	d
„ Zdd„ ZdS )ÚTextTestRunnerzÉA test runner class that displays results in textual form.

    It prints out the names of tests as they are run, errors as they
    occur, and a summary of the results at the end of the test run.
    NTr   F)Ú	tb_localsc      	      C   sN   |dkrt j}t|ƒ| _|| _|| _|| _|| _|| _|| _	|dk	rJ|| _
dS )z‰Construct a TextTestRunner.

        Subclasses should accept **kwargs to ensure compatibility as the
        interface changes.
        N)ÚsysÚstderrr   r   r   r   ÚfailfastÚbufferr@   ÚwarningsÚresultclass)	r   r   r   r   rC   rD   rF   rE   r@   r   r   r	   r
   €   s    
zTextTestRunner.__init__c             C   s   |   | j| j| j¡S )N)rF   r   r   r   )r   r   r   r	   Ú_makeResult”   s    zTextTestRunner._makeResultc             C   s2  |   ¡ }t|ƒ | j|_| j|_| j|_t ¡ † | jr^t | j¡ | jdkr^tjdt	dd t
 ¡ }t|ddƒ}|dk	r€|ƒ  z||ƒ W dt|ddƒ}|dk	r¨|ƒ  X t
 ¡ }W dQ R X || }| ¡  t|dƒrä| j |j¡ |j}| j d	||d
koþdpd|f ¡ | j ¡  d }	 }
}ytt|j|j|jfƒ}W n tk
rT   Y nX |\}	}
}g }| ¡ sº| j d¡ t|jƒt|jƒ }}|r¤| d| ¡ |rÆ| d| ¡ n| j d¡ |rÚ| d| ¡ |	rî| d|	 ¡ |
r| d|
 ¡ |r"| j dd |¡f ¡ n| j d¡ |S )z&Run the given test case or test suite.)ÚdefaultÚalwaysÚmodulezPlease use assert\w+ instead.)ÚcategoryÚmessageÚstartTestRunNÚstopTestRunr=   zRan %d test%s in %.3fsr   r0   Ú r   ZFAILEDzfailures=%dz	errors=%dZOKz
skipped=%dzexpected failures=%dzunexpected successes=%dz (%s)z, r   )rG   r   rC   rD   r@   rE   Úcatch_warningsÚsimplefilterÚfilterwarningsÚDeprecationWarningÚtimeÚperf_counterr   r;   Úhasattrr   r   r=   ÚtestsRunÚmapÚlenÚexpectedFailuresÚunexpectedSuccessesÚskippedr   ÚwasSuccessfulr   r:   r9   Úappendr    )r   r"   r   Z	startTimerM   rN   ZstopTimeZ	timeTakenÚrunZexpectedFailsr[   r\   ÚresultsZinfosZfailedZerroredr   r   r	   r_   —   sr    





zTextTestRunner.run)NTr   FFNN)	r   r   r   r   r   rF   r
   rG   r_   r   r   r   r	   r?   x   s    r?   )r   rA   rT   rE   rO   r   Úsignalsr   Ú
__unittestÚobjectr   Ú
TestResultr   r?   r   r   r   r	   Ú<module>   s   [