File: //usr/local/lib/python3.7/test/__pycache__/test_zipimport_support.cpython-37.opt-1.pyc
B
��g�) � @ s� d dl Zd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl m
Z
mZmZmZmZ ejjZd dlmZmZmZmZ dd� ZG dd� de j�Zdd � Ze d
kr�e � � dS ) � N)� spawn_python� kill_python�assert_python_ok� make_script�make_zip_script)� test_doctest�sample_doctest�sample_doctest_no_doctests� sample_doctest_no_docstringsc C s� t jtdd�}t jtd�}yd| j| jf }W n tk
rH |j}Y nX x |�| ||�D ]}|� |� qZW |j
|j }}|r�t j
�d||f ��tr�td|j|f � ||fS )NF)�verboseZrecurse)r z%s.%sz%d of %d doctests failedz,doctest (%s) ... %d tests with zero failures)�doctestZ
DocTestFinderr Z
DocTestRunner�
__module__� __qualname__�AttributeError�__name__�find�runZfailuresZtries�test�supportZ
TestFailed�print)�obj�module�finderZrunner�nameZexample�f�t� r �7/usr/local/lib/python3.7/test/test_zipimport_support.py�_run_object_doctest# s r c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�ZipSupportTestsc C sP t �� tj�� tjd d � | _tjd d � | _tjd d � | _tj �� d S )N)
� linecache�
clearcache� zipimport�_zip_directory_cache�clear�sys�path� meta_path�
path_hooks�path_importer_cache)�selfr r r �setUp@ s
zZipSupportTests.setUpc C s> | j tj d d �< | jtjd d �< | jtjd d �< tj�� d S )N)r& r% r' r( r) r$ )r* r r r �tearDownH s zZipSupportTests.tearDownc C s� d}t j�� �~}t|d|�}tj�dtj�|��}t|d||�\}}t� |� t
j� d|� dd l }z| �
t�|j�|� W d t
jd= X W d Q R X d S )Nzdef foo(): pass
�__init__�zip_pkg�test_zipr )r r �temp_dirr �osr& �join�basenamer �remover% �insertr. Z assertEqual�inspect� getsourceZfoo�modules)r* �test_src�dZ init_nameZ name_in_zip�zip_name�run_namer. r r r � test_inspect_getsource_issue4223N s
z0ZipSupportTests.test_inspect_getsource_issue4223c C s t �t�}|�dd�}|�dd�}|�dd�}i }xLtttgD ]>}t �|�}|�dd�}|j�d�d }|�d d
�}|||< q>W t j
� � ��v}t |d|�}t
|d |�\}} t�|d �}
x$|�� D ]\}}|
�|d
|� q�W |
�� t�rt�|d�} td| � | �� | �� t�|� tj�d|� dd l} z�| j | j j | j j j | j j | j j | j j | j j!| j j"| j j#| j$| j$j | j$j"| j$j#| j%| j&| j'| j(j)| j(j*| j(j+| j(j,| j(j-| j.| j/| j0| j1| j2| j3g}
| j4| j5| j6g}x|
D ]}t7|| � �q�W W d tj8d= X W d Q R X d S )Nz from test import test_doctestz*import test_zipped_doctest as test_doctestztest.test_doctest�test_zipped_doctestztest.sample_doctestZsample_zipped_doctest�.���Zsample_Zsample_zipped_r/ �az.py�rzContents of %r:r )9r6 r7 r �replacer r r
r �splitr r r0 r r �zipfileZZipFile�itemsZwritestr�closer r Zprintdirr1 r4 r% r& r5 r> Z SampleClassZ NestedClassr- Z
a_classmethodZ
a_propertyZa_staticmethodZdouble�getZSampleNewStyleClassZ sample_funcZ test_DocTestZtest_DocTestParserZtest_DocTestRunnerZbasics�
exceptionsZoption_directivesZ optionflagsZ verbose_flagZ test_ExampleZ
test_debugZtest_testsourceZtest_trailing_space_in_testZtest_DocTestSuiteZtest_DocTestFinderZtest_DocFileSuiteZ
test_testfileZtest_unittest_reportflagsr r8 )r* r9 Zsample_sources�mod�srcZmod_namer: � script_namer; r< �zZzip_filer> Zknown_good_testsZ fail_due_to_missing_data_filesr r r r �test_doctest_issue4197^ s�
z&ZipSupportTests.test_doctest_issue4197c C s� t �d�}d}tj�� ��}t|d|�}t|�\}}}||df }trbtd|� td� tt |�� | �
|� d�|� t |d|d �\} }
t| �\}}}||
df }tr�td|� td� tt |�� | �
|� d�|� W d Q R X d S )
Nz� class Test:
">>> 'line 2'"
pass
import doctest
doctest.testmod()
zFile "%s", line 2, in %s�scriptz
__main__.Testz
Expected linez Got stdout:zutf-8r/ z __main__.py)
�textwrap�dedentr r r0 r r r r �ascii�assertIn�encoder ) r* r9 �patternr: rL Zrc�out�errZexpectedr; r< r r r �test_doctest_main_issue4197� s*
z+ZipSupportTests.test_doctest_main_issue4197c C s� t �d�}tj�� ��}t|d|�}t|�}|j�d� t |�}| �
t j �
|�d��|� t|d|d�\}}t|�}|j�d� t |�}| �
t j �
|�d��|� W d Q R X d S )Nz� def f():
pass
import pdb
pdb.Pdb(nosigint=True).runcall(f)
rO s l
zutf-8r/ z __main__.py)rP rQ r r r0 r r �stdin�writer rS r1 r& �normcaserT r )r* r9 r: rL �p�datar; r< r r r �test_pdb_issue4201� s z"ZipSupportTests.test_pdb_issue4201N) r r
r r+ r, r= rN rX r^ r r r r r 7 s ] r c C s t j�� d S )N)r r Z
reap_childrenr r r r �tearDownModule� s r_ �__main__) Z test.supportr r1 Zos.pathr% rP rE r" r r6 r ZunittestZtest.support.script_helperr r r r r r r r r r r
r ZTestCaser r_ r �mainr r r r �<module> s& :