Your IP : 216.73.216.85
��Yf�% � @ s� d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z d d � Z d d � Z Gd d � d e j � Z
e d k r� e j � d S)
� N)�supportc C s� t j | � } xb | D]Z } t j j | | � } t j j | � rc t j j | � rc t | � q t j | � q Wt j | � d S)N) �os�listdir�path�join�isdir�islink�cleanout�remove�rmdir)�root�names�name�fullname� r �2/opt/alt/python35/lib64/python3.5/test/test_pkg.pyr
s
%
r c C s6 d | k r | j d � d | k r2 | j d � | S)N� __builtins__�__initializing__)r
)Zlstr r r �fixdir s
r c @ s� e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z
e j e j j d k d � d d � � Z d S) �TestPkgc C s7 d | _ d | _ t t j � | _ t j � | _ d S)N) r �pkgname�list�sysr �syspathr Z
modules_setup�modules_before)�selfr r r �setUp1 s z
TestPkg.setUpc s� � j t j d d � <t j � j � � j r< t � j � � j r| � f d d � t j D� } x | D] } t j | =qh Wd S)Nc s. g | ]$ } � j | j d � k r | � q S)�.)r �split)�.0r )r r r �
<listcomp>? s z$TestPkg.tearDown.<locals>.<listcomp>)
r r r r Zmodules_cleanupr r r r �modules)r r! r r )r r �tearDown7 s
zTestPkg.tearDownc C s& t t j | � t � d | i � d S)Nr )�exec�textwrap�dedent�globals)r �coder r r �run_codeD s zTestPkg.run_codec C s t j � } t j j d | � t j j | � s> t j | � x� | D]� \ } } | j � } | } x# | D] } t j j | | � } qj W| d k r� t j | � qE t
| d � } | j | � | r� | d d k r� | j d � | j � qE W| | _
| d d | _ d S)Nr �w� �
���)�tempfileZmkdtempr r �insertr r �mkdirr r �open�write�closer r ) r Zdescrr r �contents�compsr �c�fr r r �mkhierG s$
zTestPkg.mkhierc C s) d d g } | j | � d d l } d S)N�t1�t1 __init__.py� r )r8 N)r9 r: )r7 r8 )r �hierr8 r r r �test_1] s
zTestPkg.test_1c C s� d d d d d d g } | j | � d
d l } d
d l } | j | j d � | j | j j d � | j | j j j d � d
} | j | � d
d l m } d
d l m } d
d l m } | j | j d � | j | j d � | j | j j d � x5 d d d d g D]! } | j
t � d d | � qWd
d l } d
d l } | j | j d � | j | j j d � | j | j j j d � d } | j | � d S) N�t2�t2 __init__.py� 'doc for t2'�t2 sub�t2 sub __init__.pyr: �
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)�spamrG rE rF r zFailed to import %sz^
from t2 import *
self.assertTrue(dir(), ['self', 'sub'])
)r= N)r> r? )r@ N)rA r: )rB N)rC rD ) r7 Zt2.subZ
t2.sub.subsub� assertEqual�__name__rE rF r( r= rG Z
assertTrue�locals)r r; r= �srE rF rG r r r r �test_2b s:
zTestPkg.test_2c C sw d d
d d d d g } | j | � d d l } | j | j d � | j | j j d
� | j | j j j d � d S)N�t3�t3 __init__.pyr: �t3 sub�t3 sub __init__.py�
t3 sub subsub�t3 sub subsub __init__.py�spam = 1r zt3.subz
t3.sub.subsub)rM N)rN r: )rO N)rP r: )rQ N)rR rS )r7 Z
t3.sub.subsubrH rI rE rF )r r; rM r r r �test_3� s
zTestPkg.test_3c
C sE d d d d d d d d d g } | j | � d } | j | � d S)N�t4.py�)raise RuntimeError('Shouldnt load t4.py')�t4�t4 __init__.pyr: � 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)
)rU rV )rW N)rX r: )rY rZ )r[ N)r\ r: )r] r^ )r_ N)r` ra )r7 r( )r r; rK r r r �test_4� s
zTestPkg.test_4c
C s� d d d d g } | j | � d d l } d } | j | � d d l } | j t t | � � d
d d d
d d d d d d d g � | j t t | j � � d
d d d
d d d d g � | j t t | j � � d
d d d
d d d d g � d S)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__rI � __package__�__path__�__spec__�foo�stringrG )rc N)rd re )rf rg )rh ri )r7 rc r( rH r �dirrq rr )r r; rc rK r r r �test_5� s(
zTestPkg.test_5c C s| d d d d d g } | j | � d d l } | j t t | � � d d
d d d
d d d d g � d } | j | � d S)N�t6�t6 __init__.py�!__all__ = ['spam', 'ham', 'eggs']�
t6 spam.pyr: � t6 ham.py�
t6 eggs.pyr �__all__rj rk rl rm rI rn ro rp 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'])
)ru N)rv rw )rx r: )ry r: )rz r: )r7 ru rH r rs r( )r r; ru rK r r r �test_6� s
zTestPkg.test_6c C s� d d d d d d! d" d# d$ g } | j | � d% \ } } } d d l } | j t t | � � d d d d d d d d g � | j | � d d l m } | j t t | � � d d d d d d d d g � | j | � | j | � d d l m } | j t t | � � d d d d d d d d d g � | j | � | j | � | j | � d d l m
} | j | d � | j | � | j | � | j | � d S)&N�t7.pyr: �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 rj rk rl rm rI rn ro rp )rE )rF rG )rG r* )r} r: )r~ N)r r: )r� r� )r� N)r� r: )r� r� )r� N)r� r� )NNN) r7 r~ rH r rs Z assertFalserE Zt7.subrF Z
t7.sub.subsubrG ) r r; r~ rE rF ZtasZsubparZ subsubsubZhamr r r �test_7� sL
zTestPkg.test_7� z)Docstrings are omitted with -O2 and abovec C sM d d t j d d f g } | j | � d d l } | j | j d � d S)N�t8z t8 __init__�pyz 'doc for t8'r z
doc for t8)r� N)r �extsepr7 r� rH rk )r r; r� r r r �test_8 s
zTestPkg.test_8N)rI �
__module__� __qualname__r r" r( r7 r< rL rT rb rt r| r� �unittestZskipIfr �flags�optimizer� r r r r r / s
/ ,r �__main__)
r r r- r$ r� Ztestr r r ZTestCaser rI �mainr r r r �<module> s
�