B
    18\                 @   s8  d dl Z d dlZd dlZd dlZd dlZd dlZd dlm  mZ	 d dl
m
Z
 G dd dejZG dd dZddd	Zdd
dZedkr4G dd dejZe Zeee e Zeedd ejD ZeejZeeZe de!  e ddd x(e	"e# D ]Z$e e$% dd qW e   dS )    N)datetimec                   s   e Zd ZdZdZ fddZedd Z fddZd"d
dZ	dd Z
e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d d! Z  ZS )#RegressionTestResultzG======================================================================
zG----------------------------------------------------------------------
c                s\   t  j||dd d| _td| _| jdt 	d d | _
d | _g | _t|| _d S )Nr   )streamdescriptions	verbosityTZ	testsuitestart )super__init__bufferETZElement_RegressionTestResult__suitesetr   utcnow	isoformat_RegressionTestResult__e!_RegressionTestResult__start_timeZ_RegressionTestResult__resultsbool_RegressionTestResult__verbose)selfr   r   r   )	__class__ [/oak/stanford/groups/akundaje/marinovg/programs/Python-3.7.3/Lib/test/support/testresult.pyr
      s    zRegressionTestResult.__init__c             C   sL   y
|j }W n tk
r"   t|S X y| S  tk
rB   t|S X t|S )N)idAttributeErrorstr	TypeErrorrepr)clstesttest_idr   r   r   Z__getId   s    


zRegressionTestResult.__getIdc                sV   t  | t| jd | _}t | _| j	rR| j
| | d | j
  d S )Ntestcasez ... )r	   	startTestr   
SubElementr   r   timeperf_counterr   r   r   writegetDescriptionflush)r   r   e)r   r   r   r"   +   s    
zRegressionTestResult.startTestFc          	   K   sL  | j }d | _ |d krd S |d|d| | |d|dd |d|dd | jrz|dt | j d |r| jd k	r| j 	 }|t
|d_| jd k	r| j 	 }|t
|d	_x|| D ]p\}}|r|sqt
||}	t|d
r:xD| D ],\}
}|
r(|	|
t| n
t||	_qW qt||	_qW d S )NnamestatusrunresultZ	completedr$   z0.6fz
system-outz
system-erritems)r   r   pop_RegressionTestResult__getIdr   r$   r%   _stdout_buffergetvaluerstripr   r#   text_stderr_bufferr.   hasattrr   )r   r   captureargsr)   stdoutstderrkvZe2k2Zv2r   r   r   _add_result3   s4    

z RegressionTestResult._add_resultc             C   s   | j r| j| d d S )N
)r   r   r&   )r   cwordr   r   r   Z__writeS   s    zRegressionTestResult.__writec             C   sl   t |tr0|jdkr|j}q8|j d|j }nt|}t||d }t|||}|d|d|dS )Nbuiltins. )typemessagerD   )
isinstancerE   
__module____name__r   	tracebackformat_exceptionjoin)r   Zerr_typeZ	err_valueZerr_tbtypenamemsgtbr   r   r   Z__makeErrorDictW   s    

z$RegressionTestResult.__makeErrorDictc                s4   | j |d| j| d t || | dd d S )NT)errorEERROR)r>   $_RegressionTestResult__makeErrorDictr	   addError_RegressionTestResult__write)r   r   err)r   r   r   rT   j   s    zRegressionTestResult.addErrorc                s4   | j |d| j| d t || | dd d S )NT)outputxzexpected failure)r>   rS   r	   addExpectedFailurerU   )r   r   rV   )r   r   r   rY   o   s    z'RegressionTestResult.addExpectedFailurec                s4   | j |d| j| d t || | dd d S )NT)ZfailureFFAIL)r>   rS   r	   
addFailurerU   )r   r   rV   )r   r   r   r\   t   s    zRegressionTestResult.addFailurec                s2   | j ||d t || | dd| d S )N)skippedSzskipped )r>   r	   addSkiprU   )r   r   reason)r   r   r   r_   y   s    zRegressionTestResult.addSkipc                s&   |  | t | | dd d S )NrC   ok)r>   r	   
addSuccessrU   )r   r   )r   r   r   rb   ~   s    
zRegressionTestResult.addSuccessc                s*   | j |dd t | | dd d S )NZUNEXPECTED_SUCCESS)outcomeuzunexpected success)r>   r	   addUnexpectedSuccessrU   )r   r   )r   r   r   re      s    z)RegressionTestResult.addUnexpectedSuccessc             C   s2   | j r| jd | d| j | d| j d S )Nr?   rR   r[   )r   r   r&   printErrorListerrorsfailures)r   r   r   r   printErrors   s    z RegressionTestResult.printErrorsc             C   s`   xZ|D ]R\}}| j | j | j | d| | d | j | j | j d|  qW d S )Nz: r?   z%s
)r   r&   
separator1r'   
separator2)r   Zflavorrg   r   rV   r   r   r   rf      s
    z#RegressionTestResult.printErrorListc             C   sH   | j }|dt| j |dtt| j |dtt| j |S )Ntestsrg   rh   )r   r   r   testsRunlenrg   rh   )r   r)   r   r   r   get_xml_element   s
    z$RegressionTestResult.get_xml_element)F)rI   rH   __qualname__rj   rk   r
   classmethodr0   r"   r>   rU   rS   rT   rY   r\   r_   rb   re   ri   rf   ro   __classcell__r   r   )r   r   r      s"   
 r   c               @   s   e Zd ZdddZdd ZdS )QuietRegressionTestRunnerFc             C   s   t |d d| _|| j_d S )Nr   )r   r-   r   )r   r   r   r   r   r   r
      s    z"QuietRegressionTestRunner.__init__c             C   s   || j  | j S )N)r-   )r   r   r   r   r   r,      s    
zQuietRegressionTestRunner.runN)F)rI   rH   rp   r
   r,   r   r   r   r   rs      s   
rs   Fc             C   s&   | rt jtjt|| dS t jt|dS )N)resultclassr   r   )r   )	functoolspartialunittestTextTestRunnerr   rs   )r   r   r   r   r   get_test_runner_class   s    ry   c             C   s   t ||| S )N)ry   )r   r   capture_outputr   r   r   get_test_runner   s    r{   __main__c               @   s,   e Zd Zdd Zdd Zdd Zdd Zd	S )
	TestTestsc             C   s   d S )Nr   )r   r   r   r   	test_pass   s    zTestTests.test_passc             C   s   t d d S )Ng      ?)r$   sleep)r   r   r   r   test_pass_slow   s    zTestTests.test_pass_slowc             C   s*   t dtjd t dtjd | d d S )Nr9   )filer:   zfailure message)printsysr9   r:   fail)r   r   r   r   	test_fail   s    zTestTests.test_failc             C   s(   t dtjd t dtjd tdd S )Nr9   )r   r:   zerror message)r   r   r9   r:   RuntimeError)r   r   r   r   
test_error   s    zTestTests.test_errorN)rI   rH   rp   r~   r   r   r   r   r   r   r   r}      s   r}   c             c   s   | ]}|d kV  qdS )z-vNr   ).0ar   r   r   	<genexpr>   s    r   zOutput:zXML: rD   )end)F)F)&ru   ior   r$   rJ   rw   Zxml.etree.ElementTreeZetreeZElementTreer   r   TextTestResultr   rs   ry   r{   rI   TestCaser}   	TestSuitesuiteaddTest	makeSuiteStringIOr   sumargvZ
runner_clsr9   runnerr,   r-   r   r2   Ztostringlistro   sdecoder   r   r   r   <module>   s2    	




