Your IP : 216.73.216.52
���] v �
@ s� d d l Z d d l Z d d l Z d d l Z d d l Td d l m Z y d d l Z Wn e k
rt d Z Yn XGd d � d e j � Z
Gd d � d e j � Z Gd d � d e j � Z e j
e d
� Gd d � d e j � � Z Gd
d � d e � Z Gd d � d e j � Z Gd d � d e j � Z Gd d � d � Z Gd d � d e e j � Z Gd d � d e e j � Z Gd d � d e j � Z e d k r�e j � d S) � N)�*)�supportc @ 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 e
j d d � � Z e
j d d � � Z d d � Z d S) �ContextManagerTestCasec s~ g � t � f d d � � } | � �7 } | j � d g � | j | d � � j | � Wd QRX| j � d d d g � d S)Nc 3 s# � j d � d V� j d � d S)N� �* i� )�append� )�stater �4/opt/alt/python35/lib64/python3.5/test_contextlib.py�woohoo s
z@ContextManagerTestCase.test_contextmanager_plain.<locals>.woohoor r i� )�contextmanager� assertEqualr )�selfr �xr )r r
�test_contextmanager_plain s z0ContextManagerTestCase.test_contextmanager_plainc s� g � t � f d d � � } | j t � �Q | � �@ } | j � d g � | j | d � � j | � t � � Wd QRXWd QRX| j � d d d g � d S)Nc 3 s+ � j d � z d VWd � j d � Xd S)Nr r i� )r r )r r r
r s
zBContextManagerTestCase.test_contextmanager_finally.<locals>.woohoor r i� )r � assertRaises�ZeroDivisionErrorr
r )r r r r )r r
�test_contextmanager_finally s
z2ContextManagerTestCase.test_contextmanager_finallyc C sK t d d � � } | � } | j � | j | j t t d � d � � d S)Nc s s d Vd S)Nr r r r r
�whee0 s zCContextManagerTestCase.test_contextmanager_no_reraise.<locals>.whee�foo)r � __enter__� assertFalse�__exit__� TypeError)r r �ctxr r r
� test_contextmanager_no_reraise/ s
z5ContextManagerTestCase.test_contextmanager_no_reraisec C sK t d d � � } | � } | j � | j t | j t t d � d � d S)Nc s s y d VWn
d VYn Xd S)Nr r r r r
�whoo9 s zOContextManagerTestCase.test_contextmanager_trap_yield_after_throw.<locals>.whoor )r r r � RuntimeErrorr r )r r r r r r
�*test_contextmanager_trap_yield_after_throw8 s
zAContextManagerTestCase.test_contextmanager_trap_yield_after_throwc s� g � t � � f d d � � } | � �C } � j � d g � � j | d � � j | � t d � � Wd QRX� j � d d d g � d S)Nc 3 so � j d � y d VWnR t k
rj } z2 � j | j d � � j � d d d g � WYd d } ~ Xn Xd S)Nr r r i� )r r �argsr
)�e)r r r r
r G s
zAContextManagerTestCase.test_contextmanager_except.<locals>.woohoor r i� )r r
r r )r r r r )r r r
�test_contextmanager_exceptE s
z1ContextManagerTestCase.test_contextmanager_exceptc C s� t d � } t d d � � } y5 | j t d � � | � � | � Wd QRXWd QRXWn5 t k
r� } z | j | | � WYd d } ~ Xn X| j d � d S)N�spamc s s d Vd S)Nr r r r r
r X s zJContextManagerTestCase.test_contextmanager_except_stopiter.<locals>.woohoo�
StopIterationz StopIteration was suppressed)r# r ZassertWarnsRegex�PendingDeprecationWarning� Exception�assertIs�fail)r �stop_excr �exr r r
�#test_contextmanager_except_stopiterV s
#z:ContextManagerTestCase.test_contextmanager_except_stopiterc C s� d } i } t | | | � | d } t d � } y | � � | � Wd QRXWn5 t k
r� } z | j | | � WYd d } ~ Xn X| j d � d S)Nztfrom __future__ import generator_stop
from contextlib import contextmanager
@contextmanager
def woohoo():
yield
r r" z StopIteration was suppressed)�execr# r% r& r' )r �code�localsr r( r) r r r
�!test_contextmanager_except_pep479e s
#z8ContextManagerTestCase.test_contextmanager_except_pep479c C s t d d � � } y | � � t � Wd QRXWne t k
r� } zE | j t | � t � | j | j d d � | j | j t � WYd d } ~ Xn Xy! | � � t
d � � Wd QRXWnb t k
r} zB | j t | � t
� | j | j d d � | j | j � WYd d } ~ Xn Xd S)Nc s sD y d VWn4 t k
r? } z t d � | � WYd d } ~ Xn Xd S)Nzissue29692:Chained)r% r )�excr r r
�test_issue29692{ s zoContextManagerTestCase.test_contextmanager_do_not_unchain_non_stopiteration_exceptions.<locals>.test_issue29692r zissue29692:Chainedzissue29692:Unchained) r r r% r& �typer r
r �assertIsInstance� __cause__r# � assertIsNone)r r0 r) r r r
�?test_contextmanager_do_not_unchain_non_stopiteration_exceptionsz s
&
zVContextManagerTestCase.test_contextmanager_do_not_unchain_non_stopiteration_exceptionsc C s1 d d � } t | d d � d d � � � } | S)Nc s � f d d � } | S)Nc s1 x* � j � D] \ } } t | | | � q
W| S)N)�items�setattr)�func�k�v)�kwr r
�decorate� s zXContextManagerTestCase._create_contextmanager_attribs.<locals>.attribs.<locals>.decorater )r; r< r )r; r
�attribs� s zFContextManagerTestCase._create_contextmanager_attribs.<locals>.attribsr �barc S s d S)Nr )r" r r r
�baz� s zBContextManagerTestCase._create_contextmanager_attribs.<locals>.baz)r )r r= r? r r r
� _create_contextmanager_attribs� s z5ContextManagerTestCase._create_contextmanager_attribsc C s6 | j � } | j | j d � | j | j d � d S)Nr? r> )r@ r
�__name__r )r r? r r r
�test_contextmanager_attribs� s z2ContextManagerTestCase.test_contextmanager_attribsc C s# | j � } | j | j d � d S)NzWhee!)r@ r
�__doc__)r r? r r r
� test_contextmanager_doc_attrib� s z5ContextManagerTestCase.test_contextmanager_doc_attribc C s) | j � d � } | j | j d � d S)NzWhee!)r@ r
rC )r r? r r r
�*test_instance_docstring_given_cm_docstring� s zAContextManagerTestCase.test_instance_docstring_given_cm_docstringc
C sQ t d d � � } | d d d d d d d d
� � } | j | d � Wd QRXd S) Nc s s | | | | f Vd S)Nr )r r8 r �kwdsr r r
r � s z4ContextManagerTestCase.test_keywords.<locals>.woohoor � r8 � r �! rF �, )rG rH rI rJ )r r
)r r �targetr r r
�
test_keywords� s $z$ContextManagerTestCase.test_keywordsN)rA �
__module__� __qualname__r r r r r! r* r. r5 r@ rB r �requires_docstringsrD rE rL r r r r
r s
r c @ s= e Z d Z e j d d � � Z d d � Z d d � Z d S)�ClosingTestCasec C s, t j } t d � } | j | j | � d S)N)�closingrC r
)r � cm_docstring�objr r r
�test_instance_docs� s z"ClosingTestCase.test_instance_docsc
su g � G� f d d � d � } | � } | j � g � t | � � } | j | | � Wd QRX| j � d g � d S)Nc s" e Z d Z � f d d � Z d S)z'ClosingTestCase.test_closing.<locals>.Cc s � j d � d S)Nr )r )r )r r r
�close� s z-ClosingTestCase.test_closing.<locals>.C.closeN)rA rM rN rU r )r r r
�C� s rV r )r
rQ )r rV r �yr )r r
� test_closing� s z ClosingTestCase.test_closingc s� g � G� f d d � d � } | � } | j � g � | j t � �3 t | � � } | j | | � d d Wd QRXWd QRX| j � d g � d S)Nc s" e Z d Z � f d d � Z d S)z-ClosingTestCase.test_closing_error.<locals>.Cc s � j d � d S)Nr )r )r )r r r
rU � s z3ClosingTestCase.test_closing_error.<locals>.C.closeN)rA rM rN rU r )r r r
rV � s rV r r )r
r r rQ )r rV r rW r )r r
�test_closing_error� s z"ClosingTestCase.test_closing_errorN)rA rM rN r rO rT rX rY r r r r
rP � s rP c @ s e Z d Z d d � Z d S)�FileContextTestCasec C s� t j � } z� d } t | d � �$ } | j | j � | j d � Wd QRX| j | j � d } | j t � �L t | d � �5 } | j | j � | j | j
� d � d d Wd QRXWd QRX| j | j � Wd t j | � Xd S)N�wzBooh
�rr r )
�tempfile�mktemp�openr �closed�write�
assertTruer r r
�readr �unlink)r Ztfn�fr r r
� testWithOpen� s z FileContextTestCase.testWithOpenN)rA rM rN rf r r r r
rZ � s rZ z!Threading required for this test.c @ sX e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
S)�LockContextTestCasec C s� | j | � � | � | j | � � Wd QRX| j | � � | j t � �+ | � | j | � � d d Wd QRXWd QRX| j | � � d S)Nr r )r rb r r )r �lock�lockedr r r
� boilerPlate� s z LockContextTestCase.boilerPlatec C s# t j � } | j | | j � d S)N)� threading�Lockrj ri )r rh r r r
� testWithLock� s z LockContextTestCase.testWithLockc C s# t j � } | j | | j � d S)N)rk �RLockrj � _is_owned)r rh r r r
�
testWithRLock� s z!LockContextTestCase.testWithRLockc s2 t j � � � f d d � } | j � | � d S)Nc s
� j � S)N)ro r )rh r r
ri s z5LockContextTestCase.testWithCondition.<locals>.locked)rk � Conditionrj )r ri r )rh r
�testWithCondition s z%LockContextTestCase.testWithConditionc s2 t j � � � f d d � } | j � | � d S)Nc s% � j d � r � j � d Sd Sd S)NFT)�acquire�releaser )rh r r
ri s
z5LockContextTestCase.testWithSemaphore.<locals>.locked)rk � Semaphorerj )r ri r )rh r
�testWithSemaphore s z%LockContextTestCase.testWithSemaphorec s2 t j � � � f d d � } | j � | � d S)Nc s% � j d � r � j � d Sd Sd S)NFT)rs rt r )rh r r
ri s
z<LockContextTestCase.testWithBoundedSemaphore.<locals>.locked)rk �BoundedSemaphorerj )r ri r )rh r
�testWithBoundedSemaphore s z,LockContextTestCase.testWithBoundedSemaphoreN) rA rM rN rj rm rp rr rv rx r r r r
rg � s
rg c @ s: e Z d Z d Z d Z d Z d d � Z d d � Z d S)� mycontextFNc C s
d | _ | S)NT)�started)r r r r
r " s zmycontext.__enter__c G s | | _ | j S)N)r/ �catch)r r/ r r r
r & s zmycontext.__exit__)rA rM rN rz r/ r{ r r r r r r
ry s
ry c @ s� e Z d Z e j 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 S)�TestContextDecoratorc C s) t j } t � } | j | j | � d S)N)ry rC r
)r rR rS r r r
rT - s z'TestContextDecorator.test_instance_docsc C sP t � } | �' } | j | | � | j | j � Wd QRX| j | j d � d S)N)NNN)ry r& rb rz r
r/ )r �context�resultr r r
�test_contextdecorator4 s
z*TestContextDecorator.test_contextdecoratorc C s� t � } | j t d � � | � t d � � Wd QRXWd QRX| j | j � | j | j d t � t � } d | _ | � t d � � Wd QRX| j | j � | j | j d t � d S)Nr r T)ry �assertRaisesRegex� NameError�assertIsNotNoner/ r& r{ )r r} r r r
�$test_contextdecorator_with_exception= s z9TestContextDecorator.test_contextdecorator_with_exceptionc sB t � � � � � f d d � � } | � � j � j d � d S)Nc s$ � j � j � � j � j � d S)N)r4 r/ rb rz r )r} r r r
�testQ s z1TestContextDecorator.test_decorator.<locals>.test)NNN)ry r
r/ )r r� r )r} r r
�test_decoratorN s z#TestContextDecorator.test_decoratorc
sp t � � � � � f d d � � } � j t d � � | � Wd QRX� j � j � � j � j d t � d S)Nc s0 � j � j � � j � j � t d � � d S)Nr )r4 r/ rb rz r� r )r} r r r
r� \ s z@TestContextDecorator.test_decorator_with_exception.<locals>.testr r )ry r� r� r� r/ r& )r r� r )r} r r
� test_decorator_with_exceptionY s z2TestContextDecorator.test_decorator_with_exceptionc s t � � G� f d d � d t � } | � } | j d d � | j | j d � | j | j d � | j | j d � | � } | j d d d � | j | j d � | j | j d � | j | j d � | � } | j d d d d � | j | j d � | j | j d � d S)Nc s% e Z d Z � d d d � � Z d S)z9TestContextDecorator.test_decorating_method.<locals>.TestNc S s | | _ | | _ | | _ d S)N)�a�b�c)r r� r� r� r r r
�methodm s z@TestContextDecorator.test_decorating_method.<locals>.Test.method)rA rM rN r� r )r} r r
�Testk s r� r � r� r� r� )ry �objectr� r
r� r� r� )r r� r� r )r} r
�test_decorating_methodh s z+TestContextDecorator.test_decorating_methodc C sB Gd d � d t � } | j t � � | � � Wd QRXWd QRXd S)Nc @ s( e Z d Z d d � Z d d � Z d S)z7TestContextDecorator.test_typo_enter.<locals>.mycontextc S s d S)Nr )r r r r
� __unter__� s zATestContextDecorator.test_typo_enter.<locals>.mycontext.__unter__c W s d S)Nr )r r/ r r r
r � s z@TestContextDecorator.test_typo_enter.<locals>.mycontext.__exit__N)rA rM rN r� r r r r r
ry � s ry )�ContextDecoratorr �AttributeError)r ry r r r
�test_typo_enter� s
z$TestContextDecorator.test_typo_enterc C sB Gd d � d t � } | j t � � | � � Wd QRXWd QRXd S)Nc @ s( e Z d Z d d � Z d d � Z d S)z6TestContextDecorator.test_typo_exit.<locals>.mycontextc S s d S)Nr )r r r r
r � s z@TestContextDecorator.test_typo_exit.<locals>.mycontext.__enter__c W s d S)Nr )r r/ r r r
�__uxit__� s z?TestContextDecorator.test_typo_exit.<locals>.mycontext.__uxit__N)rA rM rN r r� r r r r
ry � s ry )r� r r� )r ry r r r
�test_typo_exit� s
z#TestContextDecorator.test_typo_exitc sq Gd d � d t � } Gd d � d | t � } | � � � � � f d d � � } | � � j � j d � d S)Nc @ s4 e Z d Z d Z d Z d d � Z d d � Z d S)zHTestContextDecorator.test_contextdecorator_as_mixin.<locals>.somecontextFNc S s
d | _ | S)NT)rz )r r r r
r � s zRTestContextDecorator.test_contextdecorator_as_mixin.<locals>.somecontext.__enter__c W s
| | _ d S)N)r/ )r r/ r r r
r � s zQTestContextDecorator.test_contextdecorator_as_mixin.<locals>.somecontext.__exit__)rA rM rN rz r/ r r r r r r
� somecontext� s r� c @ s e Z d Z d S)zFTestContextDecorator.test_contextdecorator_as_mixin.<locals>.mycontextN)rA rM rN r r r r
ry � s ry c s$ � j � j � � j � j � d S)N)r4 r/ rb rz r )r} r r r
r� � s zATestContextDecorator.test_contextdecorator_as_mixin.<locals>.test)NNN)r� r� r
r/ )r r� ry r� r )r} r r
� test_contextdecorator_as_mixin� s z3TestContextDecorator.test_contextdecorator_as_mixinc s� t � f d d � � } g � | d � � � f d d � � } | d � � j � d d d g � g � | d � � j � d d d g � d S) Nc 3 s# � j | � d V� j d � d S)Ni� )r )rW )r r r
r � s
zETestContextDecorator.test_contextmanager_as_decorator.<locals>.woohoor c s$ � j � d g � � j | � d S)Nr )r
r )r )r r r r
r� � s zCTestContextDecorator.test_contextmanager_as_decorator.<locals>.testZ somethingi� zsomething else)r r
)r r r� r )r r r
� test_contextmanager_as_decorator� s !
z5TestContextDecorator.test_contextmanager_as_decoratorN)rA rM rN r rO rT r r� r� r� r� r� r� r� r� r r r r
r| + s r| c @ s� e Z d Z e j 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 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' S)(�
TestExitStackc C s) t j } t � } | j | j | � d S)N)� ExitStackrC r
)r rR rS r r r
rT � s z TestExitStack.test_instance_docsc C s t � � Wd QRXd S)N)r� )r r r r
�test_no_resources� s
z TestExitStack.test_no_resourcesc s f i f d i f d i f f t d d � f d t d d � f d t d d � f g } g � � f d d � } t � �� } x� t | � D]� \ } } | r� | r� | j | | | � } nE | r� | j | | � } n* | r� | j | | � } n | j | � } | j | | � q� WxP | j D]E } | j | j | � | j | j | j � | j | j
| j
� qWWd QRX| j � | � d S)
Nr r� Zexamplec s � j | | f � d S)N)r )r rF )r~ r r
�_exit� s z*TestExitStack.test_callback.<locals>._exit)r )r r� )r )r r� ) �dictr� �reversed�callbackr& �_exit_callbacks� __wrapped__ZassertNotEqualrA r4 rC r
)r Zexpectedr� �stackr rF re �wrapperr )r~ r
�
test_callback� s. !zTestExitStack.test_callbackc
sd t � � � f d d � } d d � } � f d d � } Gd d � d t � } t � �} | j | � � j | j d | � | | � } | j | � � j | j d j | � | j | � � j | j d
| � | | � } | j | � � j | j d j | � | j | � � j | j d | � | j | � � j | j d | � d d
Wd QRXd S)Nc s � j | � � d S)N)r& )�exc_typer/ �exc_tb)�
exc_raisedr r r
� _expect_exc� s z,TestExitStack.test_push.<locals>._expect_excc W s d S)NTr )� exc_detailsr r r
�
_suppress_exc� s z.TestExitStack.test_push.<locals>._suppress_excc s+ � j | � � j | � � j | � d S)N)r4 )r� r/ r� )r r r
�
_expect_ok� s
z+TestExitStack.test_push.<locals>._expect_okc @ s4 e Z d Z d d � Z d d � Z d d � Z d S)z'TestExitStack.test_push.<locals>.ExitCMc S s
| | _ d S)N)� check_exc)r r� r r r
�__init__ s z0TestExitStack.test_push.<locals>.ExitCM.__init__c S s | j d � d S)NzShould not be called!)r' )r r r r
r s z1TestExitStack.test_push.<locals>.ExitCM.__enter__c W s | j | � d S)N)r� )r r� r r r
r s z0TestExitStack.test_push.<locals>.ExitCM.__exit__N)rA rM rN r� r r r r r r
�ExitCM s r� r r ���r� r� r� r� r� )r r� r� �pushr& r� �__self__)r r� r� r� r� r� �cmr )r� r r
� test_push� s*
zTestExitStack.test_pushc
s� G� f d d � d t � } g � | � } t � �c } | j � f d d � � } | j | � | j | � | j | j d j | � � j d � Wd QRX| j � d d d d g � d S)
Nc s4 e Z d Z � f d d � Z � f d d � Z d S)z0TestExitStack.test_enter_context.<locals>.TestCMc s � j d � d S)Nr )r )r )r~ r r
r s z:TestExitStack.test_enter_context.<locals>.TestCM.__enter__c s � j d � d S)N� )r )r r� )r~ r r
r s z9TestExitStack.test_enter_context.<locals>.TestCM.__exit__N)rA rM rN r r r )r~ r r
�TestCM s r� c s � j d � d S)N� )r r )r~ r r
r� " s z/TestExitStack.test_enter_context.<locals>._exitr r� r� r� r� )
r� r� r� r� �
enter_contextr& r� r� r r
)r r� r� r� r� r )r~ r
�test_enter_context s
z TestExitStack.test_enter_contextc sr g � t � �F } | j � f d d � � } | j | � | j � � j d � Wd QRX| j � d d g � d S)Nc s � j d � d S)Nr )r r )r~ r r
r� . s z'TestExitStack.test_close.<locals>._exitr� r )r� r� r� rU r r
)r r� r� r )r~ r
�
test_close+ s
zTestExitStack.test_closec s� g � t � �H } | j � f d d � � } | j | � | j � } � j d � Wd QRX� j d � | j � | j � d d d g � d S)Nc s � j d � d S)Nr� )r r )r~ r r
r� 9 s z)TestExitStack.test_pop_all.<locals>._exitr r� r� )r� r� r� �pop_allr rU r
)r r� r� � new_stackr )r~ r
� test_pop_all6 s
zTestExitStack.test_pop_allc C sI | j t � �3 t � �" } | j d d � � d d Wd QRXWd QRXd S)Nc W s d S)NFr )r/ r r r
�<lambda>F s z/TestExitStack.test_exit_raise.<locals>.<lambda>r r )r r r� r� )r r� r r r
�test_exit_raiseC s z TestExitStack.test_exit_raisec
C s2 t � �" } | j d d � � d d Wd QRXd S)Nc W s d S)NTr )r/ r r r
r� K s z2TestExitStack.test_exit_suppress.<locals>.<lambda>r r )r� r� )r r� r r r
�test_exit_suppressI s z TestExitStack.test_exit_suppressc % C s= Gd d � d � } Gd d � d � } Gd d � d � } y\ | t � �I | t t � �2 | � �! | t � �
d d Wd QRXWd QRXWd QRXWd QRXWnd t k
r� } zD | j | j t � | j | j j t � | j | j j j � WYd d } ~ Xn X| j d � | j d } | j | t � | j | j t � d S)
Nc @ s4 e Z d Z d d � Z d d � Z d d � Z d S)zFTestExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExcc S s
| | _ d S)N)r/ )r r/ r r r
r� R s zOTestExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExc.__init__c S s | S)Nr )r r r r
r T s zPTestExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExc.__enter__c W s
| j � d S)N)r/ )r r� r r r
r V s zOTestExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExc.__exit__N)rA rM rN r� r r r r r r
�RaiseExcQ s r� c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)zQTestExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExcWithContextc S s | | _ | | _ d S)N)�outer�inner)r r� r� r r r
r� Z s zZTestExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExcWithContext.__init__c S s | S)Nr )r r r r
r ] s z[TestExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExcWithContext.__enter__c W s% y
| j � Wn | j � Yn Xd S)N)r� r� )r r� r r r
r _ s
zZTestExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExcWithContext.__exit__N)rA rM rN r� r r r r r r
�RaiseExcWithContextY s r� c @ s( e Z d Z d d � Z d d � Z d S)zITestExitStack.test_exit_exception_chaining_reference.<locals>.SuppressExcc S s | S)Nr )r r r r
r f s zSTestExitStack.test_exit_exception_chaining_reference.<locals>.SuppressExc.__enter__c W s | t | � _ d S)NT)r1 �
saved_details)r r� r r r
r h s zRTestExitStack.test_exit_exception_chaining_reference.<locals>.SuppressExc.__exit__N)rA rM rN r r r r r r
� SuppressExce s r� r r z0Expected IndexError, but no exception was raised)
�
IndexError�KeyErrorr� �
ValueErrorr2 � __context__r4 r' r� r )r r� r� r� r/ � inner_excr r r
�&test_exit_exception_chaining_referenceN s"
()
z4TestExitStack.test_exit_exception_chaining_referencec s5 d d � } d � � f d d � } yl t � �\ } | j | t � | j | t � | j | t � | j | � | j | t � d d Wd QRXWnd t k
r� } zD | j | j t � | j | j j t � | j | j j j � WYd d } ~ Xn X| j
d � � d } | j | t � | j | j t � d S)Nc S s
| � d S)Nr )r/ r r r
� raise_exc� s z=TestExitStack.test_exit_exception_chaining.<locals>.raise_excc s
| � d S)NTr )r� )r� r r
� suppress_exc� s z@TestExitStack.test_exit_exception_chaining.<locals>.suppress_excr r z0Expected IndexError, but no exception was raised) r� r� r� r� r� r� r� r2 r� r4 r' r )r r� r� r� r/ r� r )r� r
� test_exit_exception_chaining~ s&
)
z*TestExitStack.test_exit_exception_chainingc C s$ d d � } d d � } y: t � �* } | j d d � � | j | t � Wd QRXWn5 t k
r� } z | j | t � WYd d } ~ Xn X| j d � yD t � �4 } | j | t � | j | � | j | t � Wd QRXWn5 t k
r} z | j | t � WYd d } ~ Xn X| j d � d S) Nc S s
| � d S)Nr )r/ r r r
r� � s zDTestExitStack.test_exit_exception_non_suppressing.<locals>.raise_excc W s d S)NTr )r� r r r
r� � s zGTestExitStack.test_exit_exception_non_suppressing.<locals>.suppress_excc S s d S)Nr r r r r
r� � s zCTestExitStack.test_exit_exception_non_suppressing.<locals>.<lambda>z0Expected IndexError, but no exception was raisedz.Expected KeyError, but no exception was raised)r� r� r� r% r2 r' r� r� )r r� r� r� r/ r r r
�#test_exit_exception_non_suppressing� s" #
#z1TestExitStack.test_exit_exception_non_suppressingc C s/ t d d � � } t d � } t d � } t d � } t d � } yV t � �F } | j | | � � | j | | � � | j | | � � | � Wd QRXWn� t k
r*} zp | j | | � | j | j | � | j | j j | � | j | j j j | � | j | j j j j � WYd d } ~ Xn Xd S)Nc
s s z d VWd | � Xd S)Nr )r/ r r r
�gets_the_context_right� s zVTestExitStack.test_exit_exception_with_correct_context.<locals>.gets_the_context_rightr r� r� r� )r r% r� r� r& r� r4 )r r� �exc1�exc2�exc3�exc4r� r/ r r r
�(test_exit_exception_with_correct_context� s$ z6TestExitStack.test_exit_exception_with_correct_contextc C sA d d � } t d � } t d � } t d � } t d � } t d � } yC t � �3 } | j | | | � | j | | | � | � Wd QRXWn� t k
r<} z� | j | | � | j | j | � | j | j j | � | j | j j j | � | j | j j j j | � | j | j j j j j � WYd d } ~ Xn Xd S)Nc
S s z
| � Wd | � Xd S)Nr )r� Z outer_excr r r
� raise_nested� s
zMTestExitStack.test_exit_exception_with_existing_context.<locals>.raise_nestedr r� r� r� � )r% r� r� r& r� r4 ) r r� r� r� r� r� Zexc5r� r/ r r r
�)test_exit_exception_with_existing_context� s( z7TestExitStack.test_exit_exception_with_existing_contextc C sq d d � } y, t � � } | j | � d d Wd QRXWn2 t k
rl } z | j d � WYd d } ~ Xn Xd S)Nc W s d S)NTr )r� r r r
r� � s z@TestExitStack.test_body_exception_suppress.<locals>.suppress_excr r z%Expected no exception, got IndexError)r� r� r� r' )r r� r� r/ r r r
� test_body_exception_suppress� s
z*TestExitStack.test_body_exception_suppressc
C sP t � �@ } | j d d � � | j d d � � | j d d � � Wd QRXd S)Nc W s d S)NTr )r/ r r r
r� s zETestExitStack.test_exit_exception_chaining_suppress.<locals>.<lambda>c W s d d S)Nr r r )r/ r r r
r� s c W s i d S)Nr r )r/ r r r
r� s )r� r� )r r� r r r
�%test_exit_exception_chaining_suppress� s z3TestExitStack.test_exit_exception_chaining_suppressc
C s; t � �+ } x! t d � D] } | j t � q WWd QRXd S)Ni' )r� �ranger� �int)r r� �ir r r
�test_excessive_nesting s z$TestExitStack.test_excessive_nestingc C sr Gd d � d t � } | � } t � | _ t � } | j t | j | � | j | � | j | j d | � d S)Nc @ s e Z d Z d S)z3TestExitStack.test_instance_bypass.<locals>.ExampleN)rA rM rN r r r r
�Example s r� r r� ) r� r r� r r� r� r� r& r� )r r� r� r� r r r
�test_instance_bypass s
z"TestExitStack.test_instance_bypassc s Gd d � d t � � Gd d � d t � } t � f d d � � } t d d � � } | j � � �F } t � �3 } | j | � � | j | � � | d � � Wd QRXWd QRX| j } | j | � � | j | j | � | j | j j � | j | j j
� | j | j
| j � d S)
Nc @ s e Z d Z d S)zETestExitStack.test_dont_reraise_RuntimeError.<locals>.UniqueExceptionN)rA rM rN r r r r
�UniqueException s r� c @ s e Z d Z d S)zHTestExitStack.test_dont_reraise_RuntimeError.<locals>.UniqueRuntimeErrorN)rA rM rN r r r r
�UniqueRuntimeError s r� c 3 sD y d VWn4 t k
r? } z � d � | � WYd d } ~ Xn Xd S)Nr z
new exception)r% )r/ )r� r r
�second s z<TestExitStack.test_dont_reraise_RuntimeError.<locals>.secondc s s; y d VWn+ t k
r6 } z | � WYd d } ~ Xn Xd S)Nr )r% )r/ r r r
�first s z;TestExitStack.test_dont_reraise_RuntimeError.<locals>.firstzplease no infinite loop.) r% r r r r� r� � exceptionr2 r� r4 r3 r& )r r� r� r� Zerr_ctxZes_ctxr/ r )r� r
� test_dont_reraise_RuntimeError s z,TestExitStack.test_dont_reraise_RuntimeErrorN)rA rM rN r rO rT r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r
r� � s& "
0
r� c @ sm e Z d Z d Z d Z e j d d � � Z d d � Z d d � Z d d � Z
d
d � Z d d
� Z d S)�TestRedirectStreamNc C s2 | j j } | j d � } | j | j | � d S)N)�redirect_streamrC r
)r rR rS r r r
rT < s z%TestRedirectStream.test_instance_docsc C s? t t | j � } | j d � | j t t | j � | � d S)N)�getattr�sys� orig_streamr� r& )r � orig_stdoutr r r
�test_no_redirect_in_initC s
z+TestRedirectStream.test_no_redirect_in_initc
C s� t j � } d } t t | j � } | j | � �! t | d t t | j � �Wd QRX| j t t | j � | � | j � j � } | j
| | � d S)Nz<Consider an API like help(), which prints directly to stdout�file) �io�StringIOr� r� r� r� �printr& �getvalue�stripr
)r re �msgr� �sr r r
�test_redirect_to_string_ioH s # z-TestRedirectStream.test_redirect_to_string_ioc
C s9 t j � } | j | � � } | j | | � Wd QRXd S)N)r� r� r� r& )r re � enter_resultr r r
�test_enter_result_is_targetR s z.TestRedirectStream.test_enter_result_is_targetc C s� t j � } | j | � } t t | j � } | �' t d d d d t t | j � �Wd QRX| �! t d d t t | j � �Wd QRX| j t t | j � | � | j � } | j | d � d S)N�Hello�end� r� zWorld!z
Hello World!
)
r� r� r� r� r� r� r� r& r� r
)r re �
write_to_fr� r� r r r
�test_cm_is_reusableW s )# z&TestRedirectStream.test_cm_is_reusablec C s� t j � } | j | � } t t | j � } | �Q t d d d d t t | j � �| �! t d d t t | j � �Wd QRXWd QRX| j t t | j � | � | j � } | j | d � d S)Nr� r� r� r� zWorld!z
Hello World!
)
r� r� r� r� r� r� r� r& r� r
)r re r� r� r� r r r
�test_cm_is_reentrantc s "* z'TestRedirectStream.test_cm_is_reentrant)
rA rM rN r� r� r rO rT r� r� r� r� r� r r r r
r� 7 s
r� c @ s e Z d Z e Z d Z d S)�TestRedirectStdout�stdoutN)rA rM rN �redirect_stdoutr� r� r r r r
r� p s r� c @ s e Z d Z e Z d Z d S)�TestRedirectStderr�stderrN)rA rM rN �redirect_stderrr� r� r r r r
r v s r c @ s� e Z d Z e j 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 S)� TestSuppressc C s) t j } t � } | j | j | � d S)N)�suppressrC r
)r rR rS r r r
rT ~ s z TestSuppress.test_instance_docsc C s' t t � � } | j | � Wd QRXd S)N)r r� r4 )r r� r r r
�test_no_result_from_enter� s z&TestSuppress.test_no_result_from_enterc C s1 t t � � | j t d d � d � Wd QRXd S)Nr� r� � )r r� r
�pow)r r r r
�test_no_exception� s
z TestSuppress.test_no_exceptionc C s" t t � � t d � Wd QRXd S)Nr� )r r �len)r r r r
�test_exact_exception� s
z!TestSuppress.test_exact_exceptionc C s t t � �
d d Wd QRXd S)Nr� �2 )r � LookupError)r r r r
�test_exception_hierarchy� s
z%TestSuppress.test_exception_hierarchyc C s7 | j t � �! t t � �
d d Wd QRXWd QRXd S)Nr r )r r r r )r r r r
�test_other_exception� s
z!TestSuppress.test_other_exceptionc C s4 | j t � � t � �
d d Wd QRXWd QRXd S)Nr r )r r r )r r r r
� test_no_args� s
zTestSuppress.test_no_argsc C sD t t t � �
d d Wd QRXt t t � � t d � Wd QRXd S)Nr r r� )r r r r
)r r r r
� test_multiple_exception_args� s z)TestSuppress.test_multiple_exception_argsc C sw t t � } | � Wd QRX| � t d � Wd QRX| �+ | � t d � Wd QRXd } d d Wd QRX| j | � d S)Nr� Tr r )r r% r
rb )r Zignore_exceptionsZouter_continuedr r r
r� � s z!TestSuppress.test_cm_is_reentrantN)rA rM rN r rO rT r r r r r r r r� r r r r
r | s r �__main__)r� r� r] Zunittest�
contextlibr� r rk � ImportErrorZTestCaser rP rZ Z
skipUnlessrg r� ry r| r� r� r� r r rA �mainr r r r
�<module> s0
�! 0�� m9 6