3


 \`&�@s^dd
lZdd
l
Z
dd
lZdd
lZdd
lZdd�Zdd�ZGdd�dej�Ze	dkrZej
�
d
S)	�Nc
Csbtj
|�
}
xH|
D]@}tjj||�}tjj|�
rFtjj|�
rFt|�

qtj|�

qWtj|�

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








rc

Cs(d
|kr|jd
�

d|kr$|jd�

|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
ejejjdkd�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|
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
j|
�
t�d
|i
�
dS)Nr)�exec�textwrap�dedent�globals)r�coderrr�run_codeEs
zTestPkg.run_codec	
Cs�tj
�}tjjd
|�
tjj|�
s,tj|�

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

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

q2t|d��$}|j
|�

|jd�
s�|j
d�

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

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



















zTestPkg.mkhierc
Csddg}
|j|
�

ddl
}dS)N�t1�t1 __init__.py�r
)r7N)r8r9)r6r7)r�hierr7rrr�test_1^s



zTestPkg.test_1c
Cs.
ddddddg}
|j|
�

d
dl
}d
dl}|j|jd
�
|j|jjd�
|j|jjjd�
d
}|j|�

d
dlm}
d
dl
m}
d
dlm	}
|j|jd�
|j|jd�
|j|jjd�
x"dD]}|j
t�dd|�
q�Wd
dl
}d
dl}|j|jd
�
|j|jjd�
|j|jjjd�
d}|j|�

dS)N�t2�t2 __init__.py�'doc for t2'�t2 sub�t2 sub __init__.pyr9�
t2 sub subsub�t2 sub subsub __init__.py�spam = 1r
zt2.subz
t2.sub.subsubz{
            import t2
            from t2 import *
            self.assertEqual(dir(), ['self', 'sub', 't2'])
            )
�sub)
�subsub)
�spamrFrDrEr
zFailed to import %sz_
            from t2 import *
            self.assertEqual(dir(), ['self', 'sub'])
            )r<N)r=r>)r?N)r@r9)rAN)rBrC)rFrDrEr<)r6Zt2.subZ
t2.sub.subsub�assertEqual�__name__rDrEr(r<rFZ
assertTrue�locals)rr:r<�
srDrErFr
rrr�test_2cs:
























zTestPkg.test_2c
CsVdd
ddddg}
|j|
�

d	dl
}|j|jd
�
|j|jjd
�
|j|jjjd�
dS)N�t3�t3 __init__.pyr9�t3 sub�t3 sub __init__.py�
t3 sub subsub�t3 sub subsub __init__.py�spam = 1r
zt3.subz
t3.sub.subsub)rLN)rMr9)rNN)rOr9)rPN)rQrR)r6Z
t3.sub.subsubrGrHrDrE)rr:rLrrr�test_3�s








zTestPkg.test_3c

Cs2dddddddddg	}
|j|
�

d}|j
|�

dS)N�t4.py�)raise RuntimeError('Shouldnt load t4.py')�t4�t4 __init__.pyr9�	t4 sub.py�*raise RuntimeError('Shouldnt load sub.py')�t4 sub�t4 sub __init__.py�t4 sub subsub.py�-raise RuntimeError('Shouldnt load subsub.py')�
t4 sub subsub�t4 sub subsub __init__.py�spam = 1z[
            from t4.sub.subsub import *
            self.assertEqual(spam, 1)
            )rTrU)rVN)rWr9)rXrY)rZN)r[r9)r\r])r^N)r_r`)r6r()rr:rJrrr�test_4�s










zTestPkg.test_4c

Cs�ddddg}
|j|
�

ddl
}d	}|j|�

ddl
}|jtt|�
�
d
ddd
ddddddd
g�
|jtt|j�
�
d
ddd
ddddg�
|jtt|j�
�
d
ddd
ddddg�
dS)N�t5�t5 __init__.py�
import t5.foo�t5 string.py�spam = 1�	t5 foo.py�-from . import string; assert string.spam == 1r
zo
            from t5 import *
            self.assertEqual(dir(), ['foo', 'self', 'string', 't5'])
            �
__cached__�__doc__�__file__�
__loader__rH�__package__�__path__�__spec__�foo�stringrF)rbN)rcrd)rerf)rgrh)r6rbr(rGr�dirrprq)rr:rbrJrrr�test_5�s(
















zTestPkg.test_5c
CsXdddddg}
|j|
�

ddl
}|jtt|�
�
d	d
ddd
ddddg	�
d}|j|�

dS)N�t6�t6 __init__.py�!__all__ = ['spam', 'ham', 'eggs']�
t6 spam.pyr9�	t6 ham.py�
t6 eggs.pyr
�__all__rirjrkrlrHrmrnroa�

            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'])
            )rtN)rurv)rwr9)rxr9)ryr9)r6rtrGrrrr()rr:rtrJrrr�test_6�s









	
zTestPkg.test_6c
	Cs.
ddddd d!d"d#d$g	}
|j|
�

d%\}}}ddl
}|jtt|�
�
ddddddddg�
|j|�

ddl
m}
|jtt|�
�
ddddddddg�
|j|�

|j|�

ddlm}
|jtt|�
�
dddddddddg	�
|j|�

|j|�

|j|�

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

|j|�

|j|�

dS)&N�t7.pyr9�t7�t7 __init__.py�	t7 sub.py�*raise RuntimeError('Shouldnt load sub.py')�t7 sub�t7 sub __init__.py�
t7 sub .py�-raise RuntimeError('Shouldnt load subsub.py')�
t7 sub subsub�t7 sub subsub __init__.py�spam = 1r
rirjrkrlrHrmrnro)
rD)
rErF)
rF�
)r|r9)r}N)r~r9)rr�)r�N)r�r9)r�r�)r�N)r�r�)NNN)r6r}rGrrrZassertFalserDZt7.subrEZ
t7.sub.subsubrF)	rr:r}rDrEZtasZsubparZ	subsubsubZhamrrr�test_7�sL












































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

ddl}|j|jd�
dS)N�t8zt8 __init__�pyz'doc for t8'r
z
doc for t8)r�N)r�extsepr6r�rGrj)rr:r�rrr�test_8
s



zTestPkg.test_8N)rH�
__module__�__qualname__rr"r(r6r;rKrSrarsr{r��unittestZskipIfr�flags�optimizer�rrrrr.s/,
r�__main__)rrr+r$r�rrZTestCaserrH�mainrrrr�<module>s




z