Your IP : 18.226.186.109
3
\�V � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZ ddlm Z m
Z
m Z dZ ej
dej�ZG dd� de j�Zd d
� Zd d � Zd
d� Zdd� Zdd� ZG dd� de�Ze� Zd dd�Ze jfdd�Z de je
j fdd�Z de je
j fd d �Z dS ) � N)�fnmatch� )�case�suite�utilTz[_a-z]\w*\.py$c s, e Zd ZdZ� fdd�Z� fdd�Z� ZS )� _FailedTestNc s || _ tt| �j|� d S )N)�
_exception�superr �__init__)�selfZ method_name� exception)� __class__� �'/usr/lib64/python3.6/unittest/loader.pyr
s z_FailedTest.__init__c s* |� j krtt� �j|�S � fdd�}|S )Nc s
� j �d S )N)r r )r r r � testFailure! s z,_FailedTest.__getattr__.<locals>.testFailure)�_testMethodNamer r � __getattr__)r �namer )r
)r r r s
z_FailedTest.__getattr__)�__name__�
__module__� __qualname__r r
r �
__classcell__r r )r
r r s r c C s" d| t j� f }t| t|�||�S )Nz#Failed to import test module: %s
%s)� traceback�
format_exc�_make_failed_test� ImportError)r �
suiteClass�messager r r �_make_failed_import_test& s r c C s dt j� f }t| |||�S )Nz Failed to call load_tests:
%s)r r r )r r r r r r r �_make_failed_load_tests+ s r c C s t | |�}||f�|fS )N)r )�
methodnamer r r �testr r r r 0 s
r c C s< t jt|��dd� �}| |i}tdt jf|�}||| �f�S )Nc S s d S )Nr )r r r r � testSkipped5 s z'_make_skipped_test.<locals>.testSkippedZ
ModuleSkipped)r �skip�str�type�TestCase)r r r r" ZattrsZ TestClassr r r �_make_skipped_test4 s r' c C s* | j � jd�r| d d� S tjj| �d S )Nz $py.class� r i����)�lower�endswith�os�path�splitext)r, r r r �_jython_aware_splitext< s r. c s� e Zd ZdZeej�Zej Z
dZ � fdd�Z dd� Z
dd�dd �Zd d
d �Zd!d d
�Zdd� Zd"dd�Zdd� Zdd� Zdd� Zdd� Zd#d d �Zd$d d �Z� ZS )%�
TestLoaderr! Nc s t t| �j� g | _t� | _d S )N)r r/ r
�errors�set�_loading_packages)r )r
r r r
L s zTestLoader.__init__c C sH t |tj�rtd��| j|�}| r4t|d�r4dg}| jt||��}|S )NzYTest cases should not be derived from TestSuite. Maybe you meant to derive from TestCase?ZrunTest)�
issubclassr � TestSuite� TypeError�getTestCaseNames�hasattrr �map)r �
testCaseClassZ
testCaseNamesZ loaded_suiter r r �loadTestsFromTestCaseS s
z TestLoader.loadTestsFromTestCase)�patternc
O s2 t |�dksd|kr,tjdt� |jdd � t |�dkrRt |�d }tdj|���t |�dkrxt|�d }tdj|���g }x@t|�D ]4}t ||�}t
|t �r�t |t
j�r�|j| j|�� q�W t |dd �} | j|�}| d k �r.y | | ||�S tk
�r, }
z$t|j|
| j�\} } | jj| � | S d }
~
X nX |S )Nr Zuse_load_testsz(use_load_tests is deprecated and ignoredr zCloadTestsFromModule() takes 1 positional argument but {} were givenz=loadTestsFromModule() got an unexpected keyword argument '{}'�
load_tests)�len�warnings�warn�DeprecationWarning�popr5 �format�sorted�dir�getattr�
isinstancer% r3 r r& �appendr: r � Exceptionr r r0 )
r �moduler; �argsZkwsZ complaint�testsr �objr< �e�
error_case�
error_messager r r �loadTestsFromModulea s4
z TestLoader.loadTestsFromModulec C s> |j d�}d\}}|d kr�|d d � }xb|r�ydj|�}t|�}P W q( tk
r� |j� }t|| j�\}}|s�| jj|� |S Y q(X q(W |dd � }|} x�|D ]�}
y| t | |
� } } W q� t
k
�r2 } z\t | dd �d k r�|d k r�| jj|� |S t |
| | jdt j
� f �\}}| jj|� |S W Y d d } ~ X q�X q�W t| tj��rP| j| �S t| t��rtt| tj��rt| j| �S t| tj��r�t| t��r�t| tj��r�|d }| |�}
tt |
|�tj��s�| j|
g�S nt| tj��r�| S t| ��r.| � }t|tj��r|S t|tj��r | j|g�S td| |f ��n td| ��d S ) N�.r �__path__z Failed to access attribute:
%sz"calling %s returned %s, not a testz$don't know how to make test from: %s)NN���) �split�join�
__import__r rA r r r0 rG rE �AttributeErrorr r r rF �types�
ModuleTyperP r% r3 r r& r: � FunctionTyper r4 �callabler5 )r r rI �partsrN rO Z
parts_copy� module_nameZnext_attributerL �part�parentrM Zinstr! r r r �loadTestsFromName� sl
z TestLoader.loadTestsFromNamec s � �fdd�|D �}�j |�S )Nc s g | ]}�j |� ��qS r )r` )�.0r )rI r r r �
<listcomp>� s z1TestLoader.loadTestsFromNames.<locals>.<listcomp>)r )r �namesrI Zsuitesr )rI r r �loadTestsFromNames� s z TestLoader.loadTestsFromNamesc C s@ || j fdd�}tt|t|���}| jr<|jtj| j�d� |S )Nc S s | j |�ott|| ��S )N)�
startswithr[ rE )Zattrnamer9 �prefixr r r � isTestMethod� s
z1TestLoader.getTestCaseNames.<locals>.isTestMethod)�key)�testMethodPrefix�list�filterrD �sortTestMethodsUsing�sort� functools�
cmp_to_key)r r9 rg Z testFnNamesr r r r6 � s zTestLoader.getTestCaseNames�test*.pyc C sR d}|d kr | j d k r | j }n|d kr.d}|}tjj|�}|tjkrRtjjd|� || _ d}d}g }tjjtjj|��r�tjj|�}||kr�tjjtjj|d�� }�nxy t |� W n t
k
r� d}Y �nRX tj | }|j d�d } ytjjtjj
|j��}W n� tk
�r� y
|j}
W n tk
�r8 d }
Y nX |
�r�|
jd k�r�|
jd k �r�d}x�|jD ]T} | �r�| j|� �r��qd| j |jjdtjj��d | _ |j| j| |dd�� �qdW n*|jtjk�r�td�d �ntdj |��d �Y nX |�r |�s| j | �| _ tjj |� n tjj |� |�r2t
d | ��|�sHt | j||��}| j |�S )
NFTr z __init__.pyrQ )� namespacez2Can not use builtin modules as dotted module namesz$don't know how to discover from {!r}z%Start directory is not importable: %r)!�_top_level_dirr+ r, �abspath�sys�insert�isdir�isfilerU rV r �modulesrT �dirname�__file__rW �__spec__�loader�submodule_search_locationsrR re r �replace�sep�extend� _find_tests�builtin_module_namesr5 rB � _get_directory_containing_module�removerj r ) r � start_dirr; Z
top_level_dirZset_implicit_topZis_not_importable� is_namespacerK Z
the_moduleZtop_part�specr, r r r �discover� sv
zTestLoader.discoverc C sR t j| }tjj|j�}tjj|�j� jd�rBtjj tjj |��S tjj |�S d S )Nz __init__.py)
rt rx r+ r, rs rz �basenamer) re ry )r r] rI � full_pathr r r r� X s
z+TestLoader._get_directory_containing_modulec C sB || j krdS ttjj|��}tjj|| j �}|jtjjd�}|S )NrQ )rr r. r+ r, �normpath�relpathr~ r )r r, Z_relpathr r r r �_get_name_from_pathd s
z TestLoader._get_name_from_pathc C s t |� tj| S )N)rV rt rx )r r r r r �_get_module_from_namep s z TestLoader._get_module_from_namec C s
t ||�S )N)r )r r, r� r; r r r � _match_patht s zTestLoader._match_pathFc
c s� | j |�}|dkrD|| jkrD| j|||�\}}|d k r<|V |sDd S ttj|��}x||D ]t}tjj||�} | j| ||�\}}|d k r�|V |rX| j | �}| jj|� z| j | ||�E d H W d | jj
|� X qXW d S )NrQ ) r� r2 �_find_test_pathrC r+ �listdirr, rU �addr� �discard)
r r� r; rq r rK Zshould_recurse�pathsr, r� r r r r� x s*
zTestLoader._find_testsc C sP t jj|�}t jj|��rFtj|�s(dS | j|||�s:d S | j|�}y| j|�}W nX t j
k
r� } zt ||| j �dfS d }~X n� t
|| j �\}} | jj| � |dfS t jjt|d|��}
tt jj|
��} tt jj|��} | j� | j� k�r0t jj| �}
tt jj|��}t jj|�}d}t|||
|f ��| j||d�dfS �nt jj|��rH| �rzt jjt jj|d�� �rzd
S d }d }| j|�}y| j|�}W nZ t j
k
�r� } zt ||| j �dfS d }~X nz t
|| j �\}} | jj| � |dfS t|dd �}| jj|� z(| j||d�}|d k �r0|dfS |dfS | jj |� X nd S d S ) NFrz zW%r module incorrectly imported from %r. Expected %r. Is this module globally installed?)r; z __init__.pyr< T)NF)NF)NF)NF) r+ r, r� rw �VALID_MODULE_NAME�matchr� r� r� r ZSkipTestr' r r r0 rG rs rE r. �realpathr) ry r rP rv rU r2 r� r� )r r� r; rq r� r rI rM rN rO Zmod_filer� Zfullpath_noextZ
module_dirZmod_nameZ expected_dir�msgr< rK �packager r r r� � sl
zTestLoader._find_test_path)N)N)rp N)F)F)r r r ri � staticmethodr �
three_way_cmprl r r4 r rr r
r: rP r` rd r6 r� r� r� r� r� r� r� r r r )r
r r/ B s"
(
N
n
"r/ c C s t � }||_| |_|r ||_|S )N)r/ rl ri r )rf � sortUsingr r| r r r � _makeLoader� s r� c C s t ||�j| �S )N)r� r6 )r9 rf r� r r r r6 � s r6 r! c C s t |||�j| �S )N)r� r: )r9 rf r� r r r r � makeSuite� s r� c C s t |||�j| �S )N)r� rP )rI rf r� r r r r �
findTestCases� s r� )N) r+ �rert r rX rn r> r � r r r Z
__unittest�compile�
IGNORECASEr� r& r r r r r' r. �objectr/ ZdefaultTestLoaderr� r� r6 r4 r� r� r r r r �<module> s6 '