B

��g`&�@s^dd
lZdd
l
Z
dd
lZdd
lZdd
lZdd�Zdd�ZGdd�dej�Ze	dkrZe�
�
d
S)	�Nc
Cs`t�
|�
}
xF|
D]>}tj�||�}tj�|�
rDtj�|�
sDt|�

qt�|�

qWt�|�

dS)
N)	�os�listdir�path�join�isdir�islink�cleanout�remove�rmdir)�root�names�name�fullname�r�)/usr/local/lib/python3.7/test/test_pkg.pyrs








rc

Cs(d
|kr|�d
�

d|kr$|�d�

|S)N�__builtins__�__initializing__)
r	)
Zlstrrr�fixdirs







rc@s�eZ
dZd
d�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�Zdd�Zdd�Z
e�ejjdkd�dd��
ZdS)�TestPkgc

Cs$d|_d|_
ttj�
|_t�|_dS)
N)r�pkgname�list�sysr�syspath�set�modules_to_cleanup)
�selfrrr�setUp0s



z
TestPkg.setUpc
sv�jt
jdd�<x �jD]}
|
t
jkrt
j|
=qW�jrBt�j�

�jrr�f
d
d�t
jD�
}x|D]}t
j|=qbWdS)Nc
s g|]}
�j|
�
d�
kr|
�qS)
�
.)r�split)�.0r
)
rrr�
<listcomp>@s
z$TestPkg.tearDown.<locals>.<listcomp>)rrrr�modulesrrr)rZ
modulenamer!r
r)
rr�tearDown6s










zTestPkg.tearDowncCstt
�|
�
t�d
|i
�
dS)Nr)�exec�textwrap�dedent�globals)r�coderrr�run_codeEs
zTestPkg.run_codec	
Cs�t�
�}tj�d
|�
tj�|�
s,t�|�

x�|
D]�\}}|��}|j	�
d�|�
�

|}x|D]}tj�||�}q^W|dkr�t�|�

q2t|d��$}|�
|�

|�d�
s�|�
d�

WdQRXq2W||_|
d
d
|_dS)Nr
r�
w�

)�tempfileZmkdtemprr�insertrr�mkdirrr�addr�open�write�endswithrr)	rZdescrrr
�contents�compsr�
c�
frrr�mkhierHs$



















zTestPkg.mkhierc
Csd
dg}
|�|
�

ddl
}dS)N)�t1N)zt1 __init__.py�r
)r6r7)r�hierr7rrr�test_1^s



zTestPkg.test_1c
Cs.
d
dddddg}
|�|
�

ddl
}ddl}|�|jd�
|�|jjd	�
|�|jjjd
�
d}|�|�

ddlm}
dd
l
m}
ddlm	}
|�|jd	�
|�|jd
�
|�|jjd
�
x"dD]}|�
t�dd|�
q�Wddl
}ddl}|�|jd�
|�|jjd	�
|�|jjjd
�
d}|�|�

dS)N)�t2N)zt2 __init__.pyz'doc for t2')zt2 subN)zt2 sub __init__.pyr8)z
t2 sub subsubN)zt2 sub subsub __init__.pyzspam = 1r
r;zt2.subz
t2.sub.subsubz{
            import t2
            from t2 import *
            self.assertEqual(dir(), ['self', 'sub', 't2'])
            )
�sub)
�subsub)
�spam)r>r<r=r;r
zFailed to import %sz_
            from t2 import *
            self.assertEqual(dir(), ['self', 'sub'])
            )r6Zt2.subZ
t2.sub.subsub�assertEqual�__name__r<r=r(r;r>Z
assertTrue�locals)rr9r;�
sr<r=r>r
rrr�test_2cs:
























zTestPkg.test_2c
CsVd
dddddg}
|�|
�

ddl
}|�|jd�
|�|jjd	�
|�|jjjd
�
dS)N)�t3N)zt3 __init__.pyr8)zt3 subN)zt3 sub __init__.pyr8)z
t3 sub subsubN)zt3 sub subsub __init__.pyzspam = 1r
rDzt3.subz
t3.sub.subsub)r6Z
t3.sub.subsubr?r@r<r=)rr9rDrrr�test_3�s








zTestPkg.test_3c
	Cs2d
dddddddd	g	}
|�|
�

d
}|�
|�

dS)N)zt4.pyz)raise RuntimeError('Shouldnt load t4.py'))Zt4N)zt4 __init__.pyr8)z	t4 sub.pyz*raise RuntimeError('Shouldnt load sub.py'))zt4 subN)zt4 sub __init__.pyr8)zt4 sub subsub.pyz-raise RuntimeError('Shouldnt load subsub.py'))z
t4 sub subsubN)zt4 sub subsub __init__.pyzspam = 1z[
            from t4.sub.subsub import *
            self.assertEqual(spam, 1)
            )r6r()rr9rBrrr�test_4�s








zTestPkg.test_4c
Cs�d
dddg}
|�|
�

ddl
}d}|�|�

ddl
}|�tt|�
�
ddd	d
ddd
ddddg�
|�tt|j�
�
ddd	d
ddddg�
|�tt|j�
�
ddd	d
ddddg�
dS)N)�t5N)zt5 __init__.pyz
import t5.foo)zt5 string.pyzspam = 1)z	t5 foo.pyz-from . import string; assert string.spam == 1r
zo
            from t5 import *
            self.assertEqual(dir(), ['foo', 'self', 'string', 't5'])
            �
__cached__�__doc__�__file__�
__loader__r@�__package__�__path__�__spec__�foo�stringrGr>)r6rGr(r?r�dirrOrP)rr9rGrBrrr�test_5�s&















zTestPkg.test_5c
CsXd
ddddg}
|�|
�

ddl
}|�tt|�
�
ddd	d
ddd
ddg	�
d}|�|�

dS)N)�t6N)zt6 __init__.pyz!__all__ = ['spam', 'ham', 'eggs'])z
t6 spam.pyr8)z	t6 ham.pyr8)z
t6 eggs.pyr8r
�__all__rHrIrJrKr@rLrMrNa�

            import t6
            from t6 import *
            self.assertEqual(fixdir(dir(t6)),
                             ['__all__', '__cached__', '__doc__', '__file__',
                              '__loader__', '__name__', '__package__',
                              '__path__', '__spec__', 'eggs', 'ham', 'spam'])
            self.assertEqual(dir(), ['eggs', 'ham', 'self', 'spam', 't6'])
            )r6rSr?rrQr()rr9rSrBrrr�test_6�s







	
zTestPkg.test_6c
	Cs.
d
dddddddd	g	}
|�|
�

d
\}}}ddl
}|�tt|�
�
dd
ddddddg�
|�|�

ddl
m}
|�tt|�
�
dd
ddddddg�
|�|�

|�|�

ddlm}
|�tt|�
�
dd
dddddddg	�
|�|�

|�|�

|�|�

ddl	m
}
|�|d�
|�|�

|�|�

|�|�

dS)N)zt7.pyr8)�t7N)zt7 __init__.pyr8)z	t7 sub.pyz*raise RuntimeError('Shouldnt load sub.py'))zt7 subN)zt7 sub __init__.pyr8)z
t7 sub .pyz-raise RuntimeError('Shouldnt load subsub.py'))z
t7 sub subsubN)zt7 sub subsub __init__.pyzspam = 1)NNNr
rHrIrJrKr@rLrMrN)
r<)
r=r>)
r>�
)r6rVr?rrQZassertFalser<Zt7.subr=Z
t7.sub.subsubr>)	rr9rVr<r=ZtasZsubparZ	subsubsubZhamrrr�test_7�sF







































zTestPkg.test_7�z)Docstrings are omitted with -O2 and abovec
Cs:d
dtj
ddfg}
|�|
�

ddl}|�|jd�
dS)N)�t8Nzt8 __init__�pyz'doc for t8'r
z
doc for t8)r�extsepr6rZr?rI)rr9rZrrr�test_8
s



zTestPkg.test_8N)r@�
__module__�__qualname__rr"r(r6r:rCrErFrRrUrX�unittestZskipIfr�flags�optimizer]rrrrr.s/,
r�__main__)rrr+r$r`rrZTestCaserr@�mainrrrr�<module>s




z