Your IP : 216.73.216.85
���]�% � @ 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 �-/opt/alt/python35/lib64/python3.5/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� �
���)�tempfile�mkdtempr 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 )r9 N)r: r; )r8 r9 )r �hierr9 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)�spamrH rF rG r zFailed to import %sz^
from t2 import *
self.assertTrue(dir(), ['self', 'sub'])
)r> N)r? r@ )rA N)rB r; )rC N)rD rE ) r8 Zt2.subZ
t2.sub.subsub� assertEqual�__name__rF rG r( r> rH Z
assertTrue�locals)r r< r> �srF rG rH 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)rN N)rO r; )rP N)rQ r; )rR N)rS rT )r8 Z
t3.sub.subsubrI rJ rF rG )r r< rN 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)
)rV rW )rX N)rY r; )rZ r[ )r\ N)r] r; )r^ r_ )r` N)ra rb )r8 r( )r r< rL 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__rJ � __package__�__path__�__spec__�foo�stringrH )rd N)re rf )rg rh )ri rj )r8 rd r( rI r �dirrr rs )r r< rd rL 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__rk rl rm rn rJ ro rp rq 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'])
)rv N)rw rx )ry r; )rz r; )r{ r; )r8 rv rI r rt r( )r r< rv rL 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 rk rl rm rn rJ ro rp rq )rF )rG rH )rH r* )r~ r; )r N)r� r; )r� r� )r� N)r� r; )r� r� )r� N)r� r� )NNN) r8 r rI r rt Z assertFalserF Zt7.subrG Z
t7.sub.subsubrH ) r r< r rF rG 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 �extsepr8 r� rI rl )r r< r� r r r �test_8 s
zTestPkg.test_8N)rJ �
__module__� __qualname__r r" r( r8 r= rM rU rc ru r} r� �unittestZskipIfr �flags�optimizer� r r r r r / s
/ ,r �__main__)
r r r- r$ r� Ztestr r r ZTestCaser rJ �mainr r r r �<module> s
�