Your IP : 3.15.15.91
3
9��f�# � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZmZ d dl m
Z
d dlm Z d dl m
Z
d dlmZmZ d dlmZmZmZmZmZmZmZmZmZ d dlm Z G d d
� d
e �Z G d d � d e �Z G d
d� de �Z dS )� N)�DistutilsError�DistutilsOptionError)�log)�
TestLoader)�six)�map�filter) �resource_listdir�resource_exists�normalize_path� working_set�_namespace_packages�evaluate_marker�add_activation_listener�require�
EntryPoint)�Commandc @ s e Zd Zdd� Zddd�ZdS )�ScanningLoaderc C s t j| � t� | _d S )N)r �__init__�set�_visited)�self� r �/usr/lib/python3.6/test.pyr s
zScanningLoader.__init__Nc C s� || j krdS | j j|� g }|jtj| |�� t|d�rH|j|j� � t|d�r�xpt|jd�D ]`}|j d�r�|dkr�|jd |dd � }n"t
|j|d �r`|jd | }nq`|j| j |�� q`W t |�d
kr�| j
|�S |d S dS )
a Return a suite of all tests cases contained in the given module
If the module is a package, load tests from all the modules in it.
If the module has an ``additional_tests`` function, call it and add
the return value to the tests.
N�additional_tests�__path__� z.pyz __init__.py�.� z /__init__.py� r ���)r �add�appendr �loadTestsFromModule�hasattrr r �__name__�endswithr
ZloadTestsFromName�lenZ
suiteClass)r �module�patternZtests�fileZ submoduler r r r# s$
z"ScanningLoader.loadTestsFromModule)N)r% �
__module__� __qualname__r r# r r r r r s r c @ s e Zd Zdd� Zddd�ZdS )�NonDataPropertyc C s
|| _ d S )N)�fget)r r. r r r r > s zNonDataProperty.__init__Nc C s |d kr | S | j |�S )N)r. )r �objZobjtyper r r �__get__A s zNonDataProperty.__get__)N)r% r+ r, r r0 r r r r r- = s r- c @ s� e Zd ZdZdZd%d&d'gZd d
� Zdd� Zedd� �Z dd� Z
dd� Z e j
g fdd��Zee j
dd� ��Zedd� �Zd d � Zd d � Zed d!� �Zed"d#� �Zd$S )(�testz.Command to run unit tests after in-place buildz#run unit tests after in-place build� test-module=�m�$Run 'test_suite' in specified module� test-suite=�s�9Run single test, case or suite (e.g. 'module.test_suite')� test-runner=�r�Test runner to usec C s d | _ d | _d | _d | _d S )N)�
test_suite� test_module� test_loader� test_runner)r r r r �initialize_optionsS s ztest.initialize_optionsc C s� | j r| jrd}t|��| j d krD| jd kr8| jj | _ n | jd | _ | jd kr^t| jdd �| _| jd krnd| _| jd kr�t| jdd �| _d S )Nz1You may specify a module or a suite, but not bothz .test_suiter= z&setuptools.command.test:ScanningLoaderr> )r; r< r � distributionr= �getattrr> )r �msgr r r �finalize_optionsY s
ztest.finalize_optionsc C s t | j� �S )N)�list�
_test_args)r r r r � test_argsl s ztest.test_argsc c s6 | j rtjdkrdV | jr$dV | j r2| j V d S )N� � Zdiscoverz --verbose)rG rH )r; �sys� version_info�verbose)r r r r rE p s ztest._test_argsc C s | j � � |� W dQ R X dS )zI
Backward compatibility for project_on_sys_path context.
N)�project_on_sys_path)r �funcr r r �with_project_on_sys_pathx s
z test.with_project_on_sys_pathc c sP t jot| jdd�}|rv| jddd� | jd� | jd�}t|j�}| jd|d� | jd� | jddd� | jd� n"| jd� | jdd d� | jd� | jd�}t j
d d � }t j j � }zbt|j
�}t j
jd|� tj� td
d � � td |j|jf � | j|g�� d V W d Q R X W d |t j
d d �< t j j� t j j|� tj� X d S )
N�use_2to3FZbuild_pyr )ZinplaceZegg_info)�egg_baseZ build_extr c S s | j � S )N)Zactivate)�distr r r �<lambda>� s z*test.project_on_sys_path.<locals>.<lambda>z%s==%s)r �PY3rA r@ Zreinitialize_commandZ run_commandZget_finalized_commandr Z build_librI �path�modules�copyrP �insertr r r r Zegg_nameZ egg_version�paths_on_pythonpath�clear�update) r Z
include_distsZ with_2to3Zbpy_cmdZ
build_pathZei_cmdZold_pathZ old_modulesZ project_pathr r r rL s8
ztest.project_on_sys_pathc c s� t � }tjjd|�}tjjdd�}z>tjj| �}td||g�}tjj|�}|rX|tjd< dV W d||krztjjdd� n
|tjd< X dS )z�
Add the indicated paths to the head of the PYTHONPATH environment
variable so that subprocesses will also see the packages at
these paths.
Do this in a context that restores the value on exit.
�
PYTHONPATHr N)�object�os�environ�get�pathsep�joinr �pop)�pathsZnothingZorig_pythonpathZcurrent_pythonpath�prefixZto_join�new_pathr r r rX � s
ztest.paths_on_pythonpathc C sD | j | j�}| j | jpg �}| j dd� | jj� D ��}tj|||�S )z�
Install the requirements indicated by self.distribution and
return an iterable of the dists that were built.
c s s0 | ](\}}|j d �rt|dd� �r|V qdS )�:r N)�
startswithr )�.0�k�vr r r � <genexpr>� s z%test.install_dists.<locals>.<genexpr>)Zfetch_build_eggsZinstall_requiresZ
tests_requireZextras_require�items� itertools�chain)rQ Zir_dZtr_dZer_dr r r �
install_dists� s
ztest.install_distsc C s� | j | j�}dj| j�}| jr0| jd| � d S | jd| � ttjd�|�}| j |��" | j
� � | j � W d Q R X W d Q R X d S )N� zskipping "%s" (dry run)z running "%s"�location) ro r@ ra �_argvZdry_run�announcer �operator�
attrgetterrX rL � run_tests)r Zinstalled_dists�cmdrc r r r �run� s
ztest.runc C s� t jr�t| jdd�r�| jjd�d }|tkr�g }|tjkrD|j |� |d7 }x"tjD ]}|j
|�rT|j |� qTW t t tjj
|�� tjd d | j| j| j�| j| j�dd�}|jj� s�d|j }| j|tj� t|��d S )NrO Fr r )Z
testLoaderZ
testRunner�exitzTest failed: %s)r rS rA r@ r; �splitr
rI rU r" rg rD r � __delitem__�unittest�mainrr �_resolve_as_epr= r> �resultZ
wasSuccessfulrs r ZERRORr )r r( Z del_modules�namer1 rB r r r rv � s(
ztest.run_testsc C s dg| j S )Nr| )rF )r r r r rr � s z
test._argvc C s$ | dkr dS t jd| �}|j� � S )zu
Load the indicated attribute value, called, as a as if it were
specified as an entry point.
Nzx=)r �parseZresolve)�valZparsedr r r r~ s ztest._resolve_as_epN)r2 r3 r4 )r5 r6 r7 )r8 r9 r: )r% r+ r, �__doc__� descriptionZ user_optionsr? rC r- rF rE rN �
contextlib�contextmanagerrL � staticmethodrX ro rx rv �propertyrr r~ r r r r r1 G s( - r1 )!r] rt rI r� rm r| Zdistutils.errorsr r Z distutilsr r Zsetuptools.externr Zsetuptools.extern.six.movesr r Z
pkg_resourcesr r
r r r
r r r r Z
setuptoolsr r r\ r- r1 r r r r �<module> s , )