Your IP : 216.73.216.213
U
i�f`& � @ s^ d dl Z d dlZd dlZd dlZd dlZdd� Zdd� ZG dd� dej�Ze dkrZe�
� dS ) � Nc C s\ t �| �}|D ]>}t j�| |�}t j�|�rBt j�|�sBt|� qt �|� qt �| � d S �N) �os�listdir�path�join�isdir�islink�cleanout�remove�rmdir)�root�names�name�fullname� r �2/opt/alt/python38/lib64/python3.8/test/test_pkg.pyr s
r c C s( d| kr| � d� d| kr$| � d� | S )N� __builtins__�__initializing__)r
)Zlstr r r �fixdir s
r c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
e�ejjdkd�dd� �ZdS ) �TestPkgc C s$ d | _ d | _ttj�| _t� | _d S r )r �pkgname�list�sysr �syspath�set�modules_to_cleanup��selfr r r �setUp0 s z
TestPkg.setUpc sn � j tjd d �< � jD ]}|tjkrtj|= q� jr>t� j� � jrj� fdd�tjD �}|D ] }tj|= q\d S )Nc s g | ]}� j |�d �kr|�qS )�.)r �split)�.0r r r r �
<listcomp>@ s �z$TestPkg.tearDown.<locals>.<listcomp>)r r r r �modulesr r r )r Z
modulenamer# r r r r �tearDown6 s
zTestPkg.tearDownc C s t t�|�t� d| i� d S )Nr )�exec�textwrap�dedent�globals)r �coder r r �run_codeE s zTestPkg.run_codec
C s� t �� }tj�d|� tj�|�s,t�|� |D ]�\}}|�� }| j �
d� |�� |}|D ]}tj� ||�}qZ|d kr�t�|� q0t |d��$}|�
|� |�d�s�|�
d� W 5 Q R X q0|| _|d d | _d S )Nr r �w�
)�tempfileZmkdtempr r �insertr r �mkdirr r �addr �open�write�endswithr r ) r Zdescrr r �contents�compsr �c�fr r r �mkhierH s$
zTestPkg.mkhierc C s ddg}| � |� dd l}d S )N)�t1N)zt1 __init__.py� r )r8 r9 )r �hierr9 r r r �test_1^ s
zTestPkg.test_1c C s* ddddddg}| � |� dd l}dd l}| �|jd� | �|jjd � | �|jjjd
� d }| �|� dd lm} dd
lm} ddlm } | �|jd � | �|jd
� | �|jjd
� dD ]}| �
t � d d| � q�dd l}dd l}| �|jd� | �|jjd � | �|jjjd
� d}| �|� d S )N)�t2N)zt2 __init__.pyz 'doc for t2')zt2 subN)zt2 sub __init__.pyr: )z
t2 sub subsubN)zt2 sub subsub __init__.py�spam = 1r r= zt2.subz
t2.sub.subsubz{
import t2
from t2 import *
self.assertEqual(dir(), ['self', 'sub', 't2'])
��sub��subsub��spam)rD r@ rB r= r zFailed to import %sz_
from t2 import *
self.assertEqual(dir(), ['self', 'sub'])
) r8 Zt2.subZ
t2.sub.subsub� assertEqual�__name__r@ rB r* r= rD Z
assertTrue�locals)r r; r= �sr@ rB rD r r r r �test_2c s< �
zTestPkg.test_2c C sV ddddddg}| � |� dd l}| �|jd� | �|jjd � | �|jjjd
� d S ) N)�t3N)zt3 __init__.pyr: )zt3 subN)zt3 sub __init__.pyr: )z
t3 sub subsubN)zt3 sub subsub __init__.pyr> r rJ zt3.subz
t3.sub.subsub)r8 Z
t3.sub.subsubrE rF r@ rB )r r; rJ r r r �test_3� s �
zTestPkg.test_3c C s2 ddddddddd g }| � |� d
}| �|� d S ) N)zt4.pyz)raise RuntimeError('Shouldnt load t4.py'))Zt4N)zt4 __init__.pyr: )z t4 sub.py�*raise RuntimeError('Shouldnt load sub.py'))zt4 subN)zt4 sub __init__.pyr: )zt4 sub subsub.py�-raise RuntimeError('Shouldnt load subsub.py'))z
t4 sub subsubN)zt4 sub subsub __init__.pyr> z[
from t4.sub.subsub import *
self.assertEqual(spam, 1)
)r8 r* )r r; rH r r r �test_4� s �
zTestPkg.test_4c C s� ddddg}| � |� dd l}d}| �|� dd l}| �tt|��ddd d
d d d
ddddg � | �tt|j��ddd d
d d ddg� | �tt|j��ddd d
d d ddg� d S )N)�t5N)zt5 __init__.pyz
import t5.foo)z t5 string.pyr> )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__rF � __package__�__path__�__spec__�foo�stringrO rD )r8 rO r* rE r �dirrW rX )r r; rO rH r r r �test_5� sJ �
�� �� ��zTestPkg.test_5c C sX dddddg}| � |� dd l}| �tt|��ddd d
d d d
ddg � d}| �|� d S )N)�t6N)zt6 __init__.pyz!__all__ = ['spam', 'ham', 'eggs'])z
t6 spam.pyr: )z t6 ham.pyr: )z
t6 eggs.pyr: r �__all__rP rQ rR rS rF rT rU rV a�
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'])
)r8 r[ rE r rY r* )r r; r[ rH r r r �test_6� s& �
�� zTestPkg.test_6c C s. ddddddddd g }| � |� d
\}}}d d l}| �tt|��d d
ddddddg� | �|� d dlm} | �tt|��d d
ddddddg� | �|� | �|� d dlm} | �tt|��d d
dddddddg � | �|� | �|� | �|� d dl m
} | �|d� | �|� | �|� | �|� d S )N)zt7.pyr: )�t7N)zt7 __init__.pyr: )z t7 sub.pyrL )zt7 subN)zt7 sub __init__.pyr: )z
t7 sub .pyrM )z
t7 sub subsubN)zt7 sub subsub __init__.pyr> )NNNr rP rQ rR rS rF rT rU rV r? rA rD rC � ) r8 r^ rE r rY Z assertFalser@ Zt7.subrB Z
t7.sub.subsubrD ) r r; r^ r@ rB ZtasZsubparZ subsubsubZhamr r r �test_7� sf �
��
��
��
zTestPkg.test_7� z)Docstrings are omitted with -O2 and abovec C s: ddt j d dfg}| �|� dd l}| �|jd� d S )N)�t8Nz t8 __init__�pyz 'doc for t8'r z
doc for t8)r �extsepr8 rb rE rQ )r r; rb r r r �test_8 s �
zTestPkg.test_8N)rF �
__module__� __qualname__r r$ r* r8 r< rI rK rN rZ r] r` �unittestZskipIfr �flags�optimizere r r r r r . s / ,�r �__main__) r r r- r& rh r r ZTestCaser rF �mainr r r r �<module> s
z