U
e5dW � @ sj d dl Z d dlZd dlZddlmZ ddlmZ dZG dd� de�Z G dd � d ej
�Z G d
d � d e�Z dS ) � N� )�result)�registerResultTc @ s& e Zd Zdd� Zdd� Zddd�ZdS ) �_WritelnDecoratorc C s
|| _ d S �N)�stream)�selfr � r �'/usr/lib64/python3.8/unittest/runner.py�__init__ s z_WritelnDecorator.__init__c C s |dkrt |��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 Zd ZdZdZ� fdd�Zdd� Z� fdd�Z� fd d
�Z� fd d �Z � fd
d�Z
� fdd�Z � fdd�Z � fdd�Z
dd� Zdd� Z� ZS )�TextTestResultzF======================================================================zF----------------------------------------------------------------------c s8 t t| ��|||� || _|dk| _|dk| _|| _d S )Nr )�superr r r �showAll�dots� descriptions)r r r � verbosity�� __class__r r
r % s
zTextTestResult.__init__c C s0 |� � }| jr$|r$d�t|�|f�S t|�S d S r )ZshortDescriptionr �join�str)r �testZdoc_first_liner r r
�getDescription, s
z TextTestResult.getDescriptionc sB t t| ��|� | jr>| j�| �|�� | j�d� | j�� d S )Nz ... )r r � startTestr r r r$ �flush�r r# r r r
r% 3 s
zTextTestResult.startTestc sD t t| ��|� | jr$| j�d� n | jr@| j�d� | j�� d S )N�ok�.) r r �
addSuccessr r r r r r& r' r r r
r* : s zTextTestResult.addSuccessc sF t t| ��||� | jr&| j�d� n | jrB| j�d� | j�� d S )N�ERROR�E) r r �addErrorr r r r r r& �r r# �errr r r
r- B s zTextTestResult.addErrorc sF t t| ��||� | jr&| j�d� n | jrB| j�d� | j�� d S )N�FAIL�F) r r �
addFailurer r r r r r& r. r r r
r2 J s zTextTestResult.addFailurec sL t t| ��||� | jr,| j�d�|�� n | jrH| j�d� | j� � d S )Nz
skipped {0!r}�s)
r r �addSkipr r r �formatr r r& )r r# �reasonr r r
r4 R s zTextTestResult.addSkipc sF t t| ��||� | jr&| j�d� n | jrB| j�d� | j�� d S )Nzexpected failure�x) r r �addExpectedFailurer r r r r r& r. r r r
r8 Z s z!TextTestResult.addExpectedFailurec sD t t| ��|� | jr$| j�d� n | jr@| j�d� | j�� d S )Nzunexpected success�u) r r �addUnexpectedSuccessr r r r r r& r' r r r
r: b s z#TextTestResult.addUnexpectedSuccessc C s6 | j s | jr| j�� | �d| j� | �d| j� d S )Nr+ r0 )r r r r �printErrorList�errors�failures�r r r r
� printErrorsj s
zTextTestResult.printErrorsc C sX |D ]N\}}| j �| j� | j �d|| �|�f � | j �| j� | j �d| � qd S )Nz%s: %sz%s)r r �
separator1r$ �
separator2)r Zflavourr< r# r/ r r r
r; p 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 Zd ZeZd dd�dd�Zdd � Zd
d � ZdS )
�TextTestRunnerNTr F)� tb_localsc C sN |d krt j}t|�| _|| _|| _|| _|| _|| _|| _ |d k rJ|| _
d S r ) �sys�stderrr r r r �failfast�bufferrD �warnings� resultclass) r r r r rG rH rJ rI rD r r r
r � s
zTextTestRunner.__init__c C s | � | j| j| j�S r )rJ r r r r> r r r
� _makeResult� s zTextTestRunner._makeResultc C s2 | � � }t|� | j|_| j|_| j|_t�� �� | jr^t�| j� | jdkr^tjdt dd� t
� � }t |dd �}|d k r�|� z ||� W 5 t |dd �}|d k r�|� X t
� � }W 5 Q R X || }|�
� t|d�r�| j�|j� |j}| j�d||d ko�d
�pd |f � | j�� d } }
} ztt|j|j|jf�} W n tk
�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�dd� |
�f � n | j�d� |S )N)�default�always�modulez Please use assert\w+ instead.)�category�message� startTestRun� stopTestRunrA zRan %d test%s in %.3fsr r3 � r ZFAILEDz failures=%dz errors=%dZOKz
skipped=%dzexpected failures=%dzunexpected successes=%dz (%s)z, r ) rK r rG rH rD rI �catch_warnings� simplefilter�filterwarnings�DeprecationWarning�time� perf_counterr r? �hasattrr r rA ZtestsRun�map�lenZexpectedFailures�unexpectedSuccesses�skippedr
Z
wasSuccessfulr r= r< �appendr! )r r# r Z startTimerQ rR ZstopTimeZ timeTaken�runZ
expectedFailsr] r^ ZresultsZinfosZfailedZerroredr r r
r` � sx
�
�
�
zTextTestRunner.run)NTr FFNN)r r r r rJ r rK r` r r r r
rC x s �� rC )
rE rX rI rS r Zsignalsr Z
__unittest�objectr Z
TestResultr rC r r r r
�<module> s [ |