B
YI~dW � @ 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 �4/opt/alt/python37/lib64/python3.7/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 )Nr )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 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 �err)r 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 r r r- 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 �reason)r r r r/ 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 r r r3 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 r5 b s z#TextTestResult.addUnexpectedSuccessc C s6 | j s | jr| j�� | �d| j� | �d| j� d S )Nr' r+ )r r r r �printErrorList�errors�failures)r r r r � printErrorsj s
zTextTestResult.printErrorsc C s\ xV|D ]N\}}| j �| j� | j �d|| �|�f � | j �| j� | j �d| � qW d S )Nz%s: %sz%s)r r �
separator1r! �
separator2)r Zflavourr7 r r* r r r r6 p s
z TextTestResult.printErrorList)r r r r: r; r
r! r" r& r) r- r/ r3 r5 r9 r6 �
__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 )N) �sys�stderrr r r r �failfast�bufferr> �warnings� resultclass) r r r r rA rB rD rC r> r r r r
� s
zTextTestRunner.__init__c C s | � | j| j| j�S )N)rD 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 d t |dd �}|d k r�|� X t
� � }W d Q R X || }|�
� t|d�r�| j�|j� |j}| j�d||d ko�d
�pd |f � | j�� d } }
} ytt|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� stopTestRunr; zRan %d test%s in %.3fsr r. � r ZFAILEDz failures=%dz errors=%dZOKz
skipped=%dzexpected failures=%dzunexpected successes=%dz (%s)z, r ) rE r rA rB r> rC �catch_warnings� simplefilter�filterwarnings�DeprecationWarning�timeZ perf_counterr
r9 �hasattrr r r; ZtestsRun�map�lenZexpectedFailures�unexpectedSuccesses�skippedr Z
wasSuccessfulr r8 r7 �appendr )r r r Z startTimerK rL ZstopTimeZ timeTaken�runZ
expectedFailsrV rW �resultsZinfosZfailedZerroredr r r rY � sr
zTextTestRunner.run)NTr FFNN)r r r r rD r
rE rY r r r r r= x s r= )
r? rR rC rM r Zsignalsr Z
__unittest�objectr Z
TestResultr r= r r r r � s [