Edit File: test.cpython-36.pyc
3 U('d�# � @ 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 , )