a
    åzes  ã                   @   sj   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 )é    Né   )Úresult)ÚregisterResultTc                   @   s&   e Zd Zdd„ Zdd„ Zddd„ZdS )	Ú_WritelnDecoratorc                 C   s
   || _ d S ©N)Ústream)Úselfr   © r	   ú%/usr/lib/python3.9/unittest/runner.pyÚ__init__   s    z_WritelnDecorator.__init__c                 C   s   |dv 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__r   r   r   r	   r	   r	   r
   r      s   r   c                       s   e 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 )ÚTextTestResultzF======================================================================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 r   )Z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                    sN   t t| ƒ |¡ | jr.| j d¡ | j ¡  n| jrJ| j d¡ | j ¡  d S )NÚokÚ.)	r   r   Ú
addSuccessr   r   r   r&   r   r   r'   r   r	   r
   r*   :   s    zTextTestResult.addSuccessc                    sP   t t| ƒ ||¡ | jr0| j d¡ | j ¡  n| jrL| j d¡ | j ¡  d S )NÚERRORÚE)	r   r   ÚaddErrorr   r   r   r&   r   r   ©r   r#   Úerrr   r	   r
   r-   C   s    zTextTestResult.addErrorc                    sP   t t| ƒ ||¡ | jr0| j d¡ | j ¡  n| jrL| j d¡ | j ¡  d S )NÚFAILÚF)	r   r   Ú
addFailurer   r   r   r&   r   r   r.   r   r	   r
   r2   L   s    zTextTestResult.addFailurec                    sV   t t| ƒ ||¡ | jr6| j d |¡¡ | j ¡  n| jrR| j 	d¡ | j ¡  d S )Nzskipped {0!r}Ús)
r   r   ÚaddSkipr   r   r   Úformatr&   r   r   )r   r#   Úreasonr   r	   r
   r4   U   s    zTextTestResult.addSkipc                    sP   t t| ƒ ||¡ | jr0| j d¡ | j ¡  n| jrL| j d¡ | j ¡  d S )Nzexpected failureÚx)	r   r   ÚaddExpectedFailurer   r   r   r&   r   r   r.   r   r	   r
   r8   ^   s    z!TextTestResult.addExpectedFailurec                    sN   t t| ƒ |¡ | jr.| j d¡ | j ¡  n| jrJ| j d¡ | j ¡  d S )Nzunexpected successÚu)	r   r   ÚaddUnexpectedSuccessr   r   r   r&   r   r   r'   r   r	   r
   r:   g   s    z#TextTestResult.addUnexpectedSuccessc                 C   s@   | j s| jr | j ¡  | j ¡  |  d| j¡ |  d| j¡ d S )Nr+   r0   )r   r   r   r   r&   ÚprintErrorListÚerrorsÚfailures©r   r	   r	   r
   ÚprintErrorsp   s
    

zTextTestResult.printErrorsc                 C   sb   |D ]X\}}| j  | j¡ | j  d||  |¡f ¡ | j  | j¡ | j  d| ¡ | j  ¡  qd S )Nz%s: %sz%s)r   r   Ú
separator1r$   Ú
separator2r&   )r   Úflavourr<   r#   r/   r	   r	   r
   r;   w   s    zTextTestResult.printErrorList)r   r   r   r@   rA   r   r$   r%   r*   r-   r2   r4   r8   r:   r?   r;   Ú__classcell__r	   r	   r   r
   r      s   						r   c                   @   s0   e Zd ZeZdddœdd„Zdd	„ Zd
d„ ZdS )ÚTextTestRunnerNTr   F)Ú	tb_localsc          	      C   sN   |d u rt j}t|ƒ| _|| _|| _|| _|| _|| _|| _	|d urJ|| _
d S r   )ÚsysÚstderrr   r   r   r   ÚfailfastÚbufferrE   ÚwarningsÚresultclass)	r   r   r   r   rH   rI   rK   rJ   rE   r	   r	   r
   r   ˆ   s    
zTextTestRunner.__init__c                 C   s   |   | j| j| j¡S r   )rK   r   r   r   r>   r	   r	   r
   Ú_makeResultœ   s    zTextTestRunner._makeResultc                 C   sl  |   ¡ }t|ƒ | j|_| j|_| j|_t ¡ ª | jr^t | j¡ | jdv r^tjdt	dd t
 ¡ }t|dd ƒ}|d ur€|ƒ  z&||ƒ W t|dd ƒ}|d urÄ|ƒ  nt|dd ƒ}|d urÂ|ƒ  0 t
 ¡ }W d   ƒ n1 sà0    Y  || }| ¡  t|dƒr| j |j¡ |j}| j d||d	kr4d
p6d|f ¡ | j ¡  d }	 }
}ztt|j|j|jfƒ}W n ty„   Y n0 |\}	}
}g }| ¡ sê| j d¡ t|jƒt|jƒ }}|rÔ| d| ¡ |rö| d| ¡ n| j d¡ |r
| d| ¡ |	r| d|	 ¡ |
r2| d|
 ¡ |rR| j dd |¡f ¡ n| j d¡ | j ¡  |S )N)ÚdefaultÚalwaysÚmodulezPlease use assert\w+ instead.)ÚcategoryÚmessageÚstartTestRunÚstopTestRunrA   zRan %d test%s in %.3fsr   r3   Ú r   ZFAILEDzfailures=%dz	errors=%dZOKz
skipped=%dzexpected failures=%dzunexpected successes=%dz (%s)z, r   ) rL   r   rH   rI   rE   rJ   Úcatch_warningsÚsimplefilterÚfilterwarningsÚDeprecationWarningÚtimeÚperf_counterr   r?   Úhasattrr   r   rA   ZtestsRunÚmapÚlenZexpectedFailuresÚunexpectedSuccessesÚskippedr   ZwasSuccessfulr   r=   r<   Úappendr!   r&   )r   r#   r   Z	startTimerR   rS   ZstopTimeZ	timeTakenÚrunZexpectedFailsr^   r_   ÚresultsZinfosZfailedZerroredr	   r	   r
   ra   Ÿ   s€    

þ
þ&ÿ
þ



zTextTestRunner.run)NTr   FFNN)r   r   r   r   rK   r   rL   ra   r	   r	   r	   r
   rD   €   s     ÿþrD   )rF   rY   rJ   rT   r   Zsignalsr   Z
__unittestÚobjectr   Z
TestResultr   rD   r	   r	   r	   r
   Ú<module>   s   c