Your IP : 216.73.216.213
���]�f � @ s� d Z d Z d d l Z d d l Z d d l m Z d d l m Z m Z Gd d � d e � Z d d � Z
Gd
d � d e � Z e
d d
� � Z
Gd d � d e � Z Gd d � d e � Z Gd d � d e j � Z Gd d � d e � Z Gd d � d e j e � Z Gd d � d e j e � Z Gd d � d e e j � Z Gd d � d e j � Z Gd d � d e j � Z Gd d! � d! e j � Z Gd" d# � d# e j � Z e d$ k r�e j � d S)%z
Mike Blandzmbland at acm dot org� N)�deque)�_GeneratorContextManager�contextmanagerc s: e Z d Z � f d d � Z d d � Z d d � Z � S)�MockContextManagerc s/ t � j | � d | _ d | _ d | _ d S)NF)�super�__init__� enter_called� exit_called� exit_args)�self�args)� __class__� �./opt/alt/python35/lib64/python3.5/test_with.pyr s zMockContextManager.__init__c C s d | _ t j | � S)NT)r r � __enter__)r r r r r s z MockContextManager.__enter__c C s1 d | _ | | | f | _ t j | | | | � S)NT)r r
r �__exit__)r �type�value� tracebackr r r r s zMockContextManager.__exit__)�__name__�
__module__� __qualname__r r r r r )r
r r
s r c s � f d d � } | S)Nc s t � | | � S)N)r )r �kwds)�funcr r �helper s z#mock_contextmanager.<locals>.helperr )r r r )r r �mock_contextmanager s r c @ s e Z d Z d d � Z d S)� MockResourcec C s d | _ d | _ d S)NF)�yielded�stopped)r r r r r & s zMockResource.__init__N)r r r r r r r r r % s r c c s, t � } z d | _ | VWd d | _ Xd S)NT)r r r )�mockr r r � mock_contextmanager_generator+ s
r c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�Nestedc G s | | _ d | _ d S)N)�managers�entered)r r" r r r r 7 s zNested.__init__c C s� | j d k r t d � � t � | _ g } y; x4 | j D]) } | j | j � � | j j | � q: WWn | j t j � � s� � Yn X| S)NzContext is not reentrant)
r# � RuntimeErrorr r" �appendr �
appendleftr �sys�exc_info)r �vars�mgrr r r r ; s zNested.__enter__c
G s� | } xA | j D]6 } y | j | � r. d } Wq t j � } Yq Xq Wd | _ | | k r� | d | d � j | d � � d S)Nr � � )NNN)r# r r' r( �with_traceback)r r( �exr* r r r r I s
zNested.__exit__N)r r r r r r r r r r r! 5 s r! c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�
MockNestedc G s/ t j | | � d | _ d | _ d | _ d S)NF)r! r r r r
)r r" r r r r Z s zMockNested.__init__c C s d | _ t j | � S)NT)r r! r )r r r r r ` s zMockNested.__enter__c G s" d | _ | | _ t j | | � S)NT)r r
r! r )r r( r r r r d s zMockNested.__exit__N)r r r r r r r r r r r/ Y 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 S)�FailureTestCasec C s d d � } | j t | � d S)Nc S s t � Wd QRXd S)N)�foor r r r �fooNotDeclaredl s z5FailureTestCase.testNameError.<locals>.fooNotDeclared)� assertRaises� NameError)r r2 r r r �
testNameErrork s z FailureTestCase.testNameErrorc s< Gd d � d t � � � f d d � } | j t | � d S)Nc @ s e Z d Z d d � Z d S)z;FailureTestCase.testEnterAttributeError.<locals>.LacksEnterc S s d S)Nr )r r r r r r r r r s zDFailureTestCase.testEnterAttributeError.<locals>.LacksEnter.__exit__N)r r r r r r r r �
LacksEnterq s r6 c s � � } | � Wd QRXd S)Nr )r1 )r6 r r �
fooLacksEnteru s z>FailureTestCase.testEnterAttributeError.<locals>.fooLacksEnter)�objectr3 �AttributeError)r r7 r )r6 r �testEnterAttributeErrorp s z'FailureTestCase.testEnterAttributeErrorc s< Gd d � d t � � � f d d � } | j t | � d S)Nc @ s e Z d Z d d � Z d S)z9FailureTestCase.testExitAttributeError.<locals>.LacksExitc S s d S)Nr )r r r r r | s zCFailureTestCase.testExitAttributeError.<locals>.LacksExit.__enter__N)r r r r r r r r � LacksExit{ s r; c s � � } | � Wd QRXd S)Nr )r1 )r; r r � fooLacksExit s z<FailureTestCase.testExitAttributeError.<locals>.fooLacksExit)r8 r3 r9 )r r< r )r; r �testExitAttributeErrorz s z&FailureTestCase.testExitAttributeErrorc C s# d d � } | j t | | � d S)Nc S s t | d d � d S)N� Zsingle)�compile)�sr r r �shouldRaiseSyntaxError� s zGFailureTestCase.assertRaisesSyntaxError.<locals>.shouldRaiseSyntaxError)r3 � SyntaxError)r ZcodestrrA r r r �assertRaisesSyntaxError� s z'FailureTestCase.assertRaisesSyntaxErrorc C s | j d � | j d � d S)Nzwith mock as None:
passzwith mock as (None):
pass)rC )r r r r �testAssignmentToNoneError� s
z)FailureTestCase.testAssignmentToNoneErrorc C s | j d � d S)Nzwith mock as ():
pass)rC )r r r r � testAssignmentToEmptyTupleError� s z/FailureTestCase.testAssignmentToEmptyTupleErrorc C s | j d � | j d � d S)Nzwith mock as None,:
passz with mock as (None,):
pass)rC )r r r r �,testAssignmentToTupleOnlyContainingNoneError� s
z<FailureTestCase.testAssignmentToTupleOnlyContainingNoneErrorc C s | j d � d S)Nz%with mock as (foo, None, bar):
pass)rC )r r r r �(testAssignmentToTupleContainingNoneError� s z8FailureTestCase.testAssignmentToTupleContainingNoneErrorc sR Gd d � d t � � � � f d d � } � j t | � � j � j d � d S)Nc @ s( e Z d Z d d � Z d d � Z d S)z4FailureTestCase.testEnterThrows.<locals>.EnterThrowsc S s t d � � d S)Nz Enter threw)r$ )r r r r r � s z>FailureTestCase.testEnterThrows.<locals>.EnterThrows.__enter__c W s d S)Nr )r r r r r r � s z=FailureTestCase.testEnterThrows.<locals>.EnterThrows.__exit__N)r r r r r r r r r � EnterThrows� s rH c s) � � } d � _ | �
� _ Wd QRXd S)N)r1 )�ct)rH r r r � shouldThrow� s z4FailureTestCase.testEnterThrows.<locals>.shouldThrow)r8 r3 r$ � assertEqualr1 )r rJ r )rH r r �testEnterThrows� s z FailureTestCase.testEnterThrowsc s< Gd d � d t � � � f d d � } | j t | � d S)Nc @ s( e Z d Z d d � Z d d � Z d S)z2FailureTestCase.testExitThrows.<locals>.ExitThrowsc S s d S)Nr )r r r r r � s z<FailureTestCase.testExitThrows.<locals>.ExitThrows.__enter__c W s t d � � d S)N�* )r$ )r r r r r r � s z;FailureTestCase.testExitThrows.<locals>.ExitThrows.__exit__N)r r r r r r r r r �
ExitThrows� s rN c s � � � Wd QRXd S)Nr r )rN r r rJ � s
z3FailureTestCase.testExitThrows.<locals>.shouldThrow)r8 r3 r$ )r rJ r )rN r �testExitThrows� s z FailureTestCase.testExitThrowsN)
r r r r5 r: r= rC rD rE rF rG rL rO r r r r r0 j s
r0 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 d � Z
d d � Z d S)� ContextmanagerAssertionMixinc C s t d � | _ d S)Nztest exception)r$ �TEST_EXCEPTION)r r r r �setUp� s z"ContextmanagerAssertionMixin.setUpc C s7 | j | j � | j | j � | j | j d � d S)N)�
assertTruer � assertFalser rK r
)r � mock_managerr r r � assertInWithManagerInvariants� s z:ContextmanagerAssertionMixin.assertInWithManagerInvariantsc C s7 | j | j � | j | j � | j | j | � d S)N)rS r r rK r
)r rU r
r r r � assertAfterWithManagerInvariants� s z=ContextmanagerAssertionMixin.assertAfterWithManagerInvariantsc C s | j | d � d S)N)NNN)rW )r rU r r r �'assertAfterWithManagerInvariantsNoError� s zDContextmanagerAssertionMixin.assertAfterWithManagerInvariantsNoErrorc C s$ | j | j � | j | j � d S)N)rS r rT r )r �mock_generatorr r r � assertInWithGeneratorInvariants� s z<ContextmanagerAssertionMixin.assertInWithGeneratorInvariantsc C s$ | j | j � | j | j � d S)N)rS r r )r rY r r r �)assertAfterWithGeneratorInvariantsNoError� s zFContextmanagerAssertionMixin.assertAfterWithGeneratorInvariantsNoErrorc C s
| j � d S)N)rQ )r r r r �raiseTestException� s z/ContextmanagerAssertionMixin.raiseTestExceptionNc C s� | j | j � | j | j � | d k rU | j | j d | j � t | j � } | j | j d | � | j | j d | � | j | j d d � d S)Nr+ r r, ) rS r r rK r
rQ r ZassertIsInstanceZ assertIsNot)r rU �exc_typer r r �)assertAfterWithManagerInvariantsWithError� s zFContextmanagerAssertionMixin.assertAfterWithManagerInvariantsWithErrorc C s$ | j | j � | j | j � d S)N)rS r r )r rY r r r �+assertAfterWithGeneratorInvariantsWithError� s zHContextmanagerAssertionMixin.assertAfterWithGeneratorInvariantsWithError) r r r rR rV rW rX rZ r[ r\ r^ r_ r r r r rP � s rP c @ sd 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 S)�NonexceptionalTestCasec C s t � � Wd QRXd S)N)r )r r r r �testInlineGeneratorSyntax� s
z0NonexceptionalTestCase.testInlineGeneratorSyntaxc C s( t � } | � Wd QRX| j | � d S)N)r rX )r r r r r �testUnboundGenerator� s z+NonexceptionalTestCase.testUnboundGeneratorc C s1 t � � } | j | � Wd QRX| j | � d S)N)r rZ r[ )r r1 r r r � testInlineGeneratorBoundSyntax� s z5NonexceptionalTestCase.testInlineGeneratorBoundSyntaxc C s7 d } t � � } | j | � Wd QRX| j | � d S)N)r rZ r[ )r r1 r r r �*testInlineGeneratorBoundToExistingVariable� s zANonexceptionalTestCase.testInlineGeneratorBoundToExistingVariablec C s: t � � | _ | j | j � Wd QRX| j | j � d S)N)r r1 rZ r[ )r r r r �(testInlineGeneratorBoundToDottedVariable� s z?NonexceptionalTestCase.testInlineGeneratorBoundToDottedVariablec C sQ t � } | �! } | j | � | j | � Wd QRX| j | � | j | � d S)N)r rZ rV r[ rX )r r r1 r r r �testBoundGenerator s
z)NonexceptionalTestCase.testBoundGeneratorc C s� t � } | �� } t � } | �; } | j | � | j | � | j | � | j | � Wd QRX| j | � | j | � | j | � | j | � Wd QRX| j | � | j | � d S)N)r rV rZ rX r[ )r �mock_ar1 �mock_b�barr r r �testNestedSingleStatements s
z1NonexceptionalTestCase.testNestedSingleStatementsN)
r r r ra rb rc rd re rf rj r r r r r` � s r` c @ sd 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 S)� NestedNonexceptionalTestCasec C s t t � � � Wd QRXd S)N)r! r )r r r r �"testSingleArgInlineGeneratorSyntax s z?NestedNonexceptionalTestCase.testSingleArgInlineGeneratorSyntaxc C s= t � } t | � � } | j | � Wd QRX| j | � d S)N)r r! rV rX )r �mr1 r r r � testSingleArgBoundToNonTuple# s z9NestedNonexceptionalTestCase.testSingleArgBoundToNonTuplec C s= t � } t | � � } | j | � Wd QRX| j | � d S)N)r r! rV rX )r rm r1 r r r �2testSingleArgBoundToSingleElementParenthesizedList+ s zONestedNonexceptionalTestCase.testSingleArgBoundToSingleElementParenthesizedListc C s d d � } | j t | � d S)Nc S s# t t � � �
\ } } Wd QRXd S)N)r! r )r1 ri r r r �shouldThrowValueError4 s ziNestedNonexceptionalTestCase.testSingleArgBoundToMultipleElementTupleError.<locals>.shouldThrowValueError)r3 �
ValueError)r rp r r r �-testSingleArgBoundToMultipleElementTupleError3 s zJNestedNonexceptionalTestCase.testSingleArgBoundToMultipleElementTupleErrorc C s[ t � } t | � } | � | j | � | j | � Wd QRX| j | � | j | � d S)N)r r/ rV rX )r r � mock_nestedr r r �testSingleArgUnbound9 s
z1NestedNonexceptionalTestCase.testSingleArgUnboundc C s� t � } t � } t � } t | | | � } | �9 | j | � | j | � | j | � | j | � Wd QRX| j | � | j | � | j | � | j | � d S)N)r r/ rV rX )r rm �n�ors r r r �testMultipleArgUnboundB s
z3NestedNonexceptionalTestCase.testMultipleArgUnboundc
C s� t t � t � t � � } | �D \ } } } | j | � | j | � | j | � | j | � Wd QRX| j | � | j | � | j | � | j | � d S)N)r/ r rZ rV r[ rX )r rs rm ru rv r r r �testMultipleArgBoundQ s
z1NestedNonexceptionalTestCase.testMultipleArgBoundN)
r r r rl rn ro rr rt rw rx r r r r rk s rk 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
d d � Z d S)�ExceptionalTestCasec sO t � � � � f d d � } � j t | � � j � � � j � j � d S)Nc s> � �1 � _ � j � � � j � j � � j � Wd QRXd S)N)�resourcerV rZ r\ r )�cmr r r rJ b s
z;ExceptionalTestCase.testSingleResource.<locals>.shouldThrow)r r3 r$ r^ r_ rz )r rJ r )r{ r r �testSingleResource` s
z&ExceptionalTestCase.testSingleResourcec sB t � � � � f d d � } � j t | � � j � t � d S)Nc s � � � _ d d Wd QRXd S)Nr+ r )rz r )r{ r r r rJ m s z@ExceptionalTestCase.testExceptionNormalized.<locals>.shouldThrow)r r3 �ZeroDivisionErrorr^ )r rJ r )r{ r r �testExceptionNormalizedk s z+ExceptionalTestCase.testExceptionNormalizedc sx t � � t � � � � � f d d � } � j t | � � j � � � j � � � j � j � � j � j � d S)Nc sn � �a � _ � �N � _ � j � � � j � � � j � j � � j � j � � j � Wd QRXWd QRXd S)N)r1 ri rV rZ r\ r )rg rh r r r rJ x s
zCExceptionalTestCase.testNestedSingleStatements.<locals>.shouldThrow)r r3 r$ r^ r_ r1 ri )r rJ r )rg rh r r rj u s
z.ExceptionalTestCase.testNestedSingleStatementsc s� t � � t � � t � � � � � � � � f d d � } � j t | � � j � � � j � � � j � � � j � j � � j � j � d S)Nc
sq � �d \ � _ � _ � j � � � j � � � j � � � j � j � � j � j � � j � Wd QRXd S)N)�
resource_a�
resource_brV rZ r\ r )�cm_a�cm_brs r r r rJ � s
zOExceptionalTestCase.testMultipleResourcesInSingleStatement.<locals>.shouldThrow)r r/ r3 r$ r^ r_ r r� )r rJ r )r� r� rs r r �&testMultipleResourcesInSingleStatement� s
z:ExceptionalTestCase.testMultipleResourcesInSingleStatementc s� t � � t � � d � _ � � � f d d � } � j t | � � j � � � j � j � � j � j d � � j � j � � j � j
� � j � j d � d S)Nc sQ � �D � _ � j � � � j � j � � j � � �
� _ Wd QRXWd QRXd S)N)r1 rV rZ r\ ri r )rg rh r r r rJ � s
zPExceptionalTestCase.testNestedExceptionBeforeInnerStatement.<locals>.shouldThrow) r ri r3 r$ r^ r_ r1 rK rT r r r
)r rJ r )rg rh r r �'testNestedExceptionBeforeInnerStatement� s
z;ExceptionalTestCase.testNestedExceptionBeforeInnerStatementc sx t � � t � � � � � f d d � } � j t | � � j � � � j � � � j � j � � j � j � d S)Nc sn � �a � _ � �D � _ � j � � � j � � � j � j � � j � j � Wd QRX� j � Wd QRXd S)N)r1 ri rV rZ r\ r )rg rh r r r rJ � s
zOExceptionalTestCase.testNestedExceptionAfterInnerStatement.<locals>.shouldThrow) r r3 r$ r^ rX r_ r1 r[ ri )r rJ r )rg rh r r �&testNestedExceptionAfterInnerStatement� s
z:ExceptionalTestCase.testNestedExceptionAfterInnerStatementc
sR t d d � � � � f d d � } | j t d � � | j t | � Wd QRXd S)Nc s s d Vd S)Nr r r r r r{ � s z8ExceptionalTestCase.testRaisedStopIteration1.<locals>.cmc s! � � � t d � � Wd QRXd S)Nz from with)�
StopIterationr )r{ r r rJ � s
zAExceptionalTestCase.testRaisedStopIteration1.<locals>.shouldThrowr� )r �assertWarnsRegex�PendingDeprecationWarningr3 r� )r rJ r )r{ r �testRaisedStopIteration1� s z,ExceptionalTestCase.testRaisedStopIteration1c s< Gd d � d t � � � f d d � } | j t | � d S)Nc @ s( e Z d Z d d � Z d d � Z d S)z8ExceptionalTestCase.testRaisedStopIteration2.<locals>.cmc S s d S)Nr )r r r r r � s zBExceptionalTestCase.testRaisedStopIteration2.<locals>.cm.__enter__c S s d S)Nr )r r r r r r r r � s zAExceptionalTestCase.testRaisedStopIteration2.<locals>.cm.__exit__N)r r r r r r r r r r{ � s r{ c s! � � � t d � � Wd QRXd S)Nz from with)r� r )r{ r r rJ � s
zAExceptionalTestCase.testRaisedStopIteration2.<locals>.shouldThrow)r8 r3 r� )r rJ r )r{ r �testRaisedStopIteration2� s z,ExceptionalTestCase.testRaisedStopIteration2c
sR t d d � � � � f d d � } | j t d � � | j t | � Wd QRXd S)Nc s s d Vd S)Nr r r r r r{ � s z8ExceptionalTestCase.testRaisedStopIteration3.<locals>.cmc
s' � � � t t g � � � Wd QRXd S)N)�next�iterr )r{ r r rJ � s
zAExceptionalTestCase.testRaisedStopIteration3.<locals>.shouldThrowr� )r r� r� r3 r� )r rJ r )r{ r �testRaisedStopIteration3� s z,ExceptionalTestCase.testRaisedStopIteration3c s8 t d d � � � � f d d � } | j t | � d S)Nc s s d Vd S)Nr r r r r r{ � s z8ExceptionalTestCase.testRaisedGeneratorExit1.<locals>.cmc s! � � � t d � � Wd QRXd S)Nz from with)�
GeneratorExitr )r{ r r rJ � s
zAExceptionalTestCase.testRaisedGeneratorExit1.<locals>.shouldThrow)r r3 r� )r rJ r )r{ r �testRaisedGeneratorExit1� s z,ExceptionalTestCase.testRaisedGeneratorExit1c s< Gd d � d t � � � f d d � } | j t | � d S)Nc @ s( e Z d Z d d � Z d d � Z d S)z8ExceptionalTestCase.testRaisedGeneratorExit2.<locals>.cmc S s d S)Nr )r r r r r � s zBExceptionalTestCase.testRaisedGeneratorExit2.<locals>.cm.__enter__c S s d S)Nr )r r r r r r r r � s zAExceptionalTestCase.testRaisedGeneratorExit2.<locals>.cm.__exit__N)r r r r r r r r r r{ � s r{ c s! � � � t d � � Wd QRXd S)Nz from with)r� r )r{ r r rJ � s
zAExceptionalTestCase.testRaisedGeneratorExit2.<locals>.shouldThrow)r8 r3 r� )r rJ r )r{ r �testRaisedGeneratorExit2� s z,ExceptionalTestCase.testRaisedGeneratorExit2c s� Gd d � d t � � � � f d d � } | � � � f d d � } � j t | � � � f d d � } � j t | � d S) Nc @ s4 e Z d Z d d � Z d d � Z d d � Z d S)z0ExceptionalTestCase.testErrorsInBool.<locals>.cmc s) G� f d d � d � } | � | _ d S)Nc s" e Z d Z � f d d � Z d S)zGExceptionalTestCase.testErrorsInBool.<locals>.cm.__init__.<locals>.Boolc s � � S)Nr )r )�bool_conversionr r �__bool__ s zPExceptionalTestCase.testErrorsInBool.<locals>.cm.__init__.<locals>.Bool.__bool__N)r r r r� r )r� r r �Bool s r� )� exit_result)r r� r� r )r� r r s z9ExceptionalTestCase.testErrorsInBool.<locals>.cm.__init__c S s d S)N� r )r r r r r s z:ExceptionalTestCase.testErrorsInBool.<locals>.cm.__enter__c S s | j S)N)r� )r �a�b�cr r r r s z9ExceptionalTestCase.testErrorsInBool.<locals>.cm.__exit__N)r r r r r r r r r r r{ s r{ c s+ � d d � � � � j d � Wd QRXd S)Nc S s d S)NTr r r r r �<lambda> s zJExceptionalTestCase.testErrorsInBool.<locals>.trueAsBool.<locals>.<lambda>zShould NOT see this)�failr )r{ r r r �
trueAsBool s z8ExceptionalTestCase.testErrorsInBool.<locals>.trueAsBoolc s+ � d d � � � � j d � Wd QRXd S)Nc S s d S)NFr r r r r r� s zKExceptionalTestCase.testErrorsInBool.<locals>.falseAsBool.<locals>.<lambda>z Should raise)r� r )r{ r r r � falseAsBool s z9ExceptionalTestCase.testErrorsInBool.<locals>.falseAsBoolc s+ � d d � � � � j d � Wd QRXd S)Nc S s d d S)Nr+ r r r r r r r� s zJExceptionalTestCase.testErrorsInBool.<locals>.failAsBool.<locals>.<lambda>zShould NOT see this)r� r )r{ r r r �
failAsBool s z8ExceptionalTestCase.testErrorsInBool.<locals>.failAsBool)r8 r3 �AssertionErrorr} )r r� r� r� r )r{ r r �testErrorsInBool s z$ExceptionalTestCase.testErrorsInBoolN)r r r r| r~ rj r� r� r� r� r� r� r� r� r� r r r r ry _ s
ry c @ sL e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d S) �NonLocalFlowControlTestCasec C sQ d } x4 | d 7} t � � | d 7} PWd QRX| d 7} q W| j | d � d S)Nr r+ �
�d � )r rK )r �counterr r r �
testWithBreak# s
z)NonLocalFlowControlTestCase.testWithBreakc C s` d } xC | d 7} | d k r Pt � � | d 7} w Wd QRX| d 7} q W| j | d � d S)Nr r+ r, r� r� � )r rK )r r� r r r �testWithContinue- s
z,NonLocalFlowControlTestCase.testWithContinuec C s# d d � } | j | � d � d S)Nc S sD d } x7 | d 7} t � � | d 7} | SWd QRX| d 7} q Wd S)Nr r+ r� r� )r )r� r r r r1 : s
z7NonLocalFlowControlTestCase.testWithReturn.<locals>.foor� )rK )r r1 r r r �testWithReturn9 s z*NonLocalFlowControlTestCase.testWithReturnc C s5 d d � } t | � � } | j | d d g � d S)Nc s s t � � d Vd VWd QRXd S)Nr� �
)r r r r r �genE s
z6NonLocalFlowControlTestCase.testWithYield.<locals>.genr� r� )�listrK )r r� �xr r r �
testWithYieldD s z)NonLocalFlowControlTestCase.testWithYieldc C su d } y9 | d 7} t � � | d 7} t � Wd QRX| d 7} Wn" t k
rc | j | d � Yn X| j d � d S)Nr r+ r� r� r� zDidn't raise RuntimeError)r r$ rK r� )r r� r r r �
testWithRaiseL s
z)NonLocalFlowControlTestCase.testWithRaiseN)r r r r� r� r� r� r� r r r r r� ! s
r� c @ s( e Z d Z d d � Z d d � Z d S)�AssignmentTargetTestCasec C sc d d d d g i } t � �I | d d <| j t | j � � d g � | j | d d j t � Wd QRXt � �U t | j � � d d <| j t | j � � d g � | j | d d j t � Wd QRXt � �= | d <t | j � � } | j � | j | d d g � Wd QRXGd d � d � } | � } t � �# | _ | j t | d � d � Wd QRXd S)Nr+ r r, c @ s e Z d Z d S)z;AssignmentTargetTestCase.testSingleComplexTarget.<locals>.CN)r r r r r r r �Ch s r� r1 T)
r rK r� �keysr
r �values�sortr1 �hasattr)r �targetsr� r� �blahr r r �testSingleComplexTarget\ s " "
z0AssignmentTargetTestCase.testSingleComplexTargetc C s� Gd d � d � } d d d d g i } | � �G \ | d d <| d d <| d d <| j | d d d d g i � Wd QRX| � �k \ t | j � � d d <t | j � � d d <t | j � � d d <| j | d d d d g i � Wd QRX| � �> \ | d <| d <| d <| j | d d d d d d i � Wd QRXGd d � d � } | � } | � �R \ | _ | _ | _ | j | j d � | j | j d � | j | j d � Wd QRXd S) Nc @ s( e Z d Z d d � Z d d � Z d S)z>AssignmentTargetTestCase.testMultipleComplexTargets.<locals>.Cc S s d S)Nr+ r, r� )r+ r, r� r )r r r r r o s zHAssignmentTargetTestCase.testMultipleComplexTargets.<locals>.C.__enter__c S s d S)Nr )r �t�v�tbr r r r p s zGAssignmentTargetTestCase.testMultipleComplexTargets.<locals>.C.__exit__N)r r r r r r r r r r� n s r� r+ r r, r� c @ s e Z d Z d S)z>AssignmentTargetTestCase.testMultipleComplexTargets.<locals>.BN)r r r r r r r �Bx s r� )rK r� r� ZoneZtwoZthree)r r� r� r� r� r r r �testMultipleComplexTargetsm s -&Q&!) z3AssignmentTargetTestCase.testMultipleComplexTargetsN)r r r r� r� r r r r r� Z s r� c @ s( e Z d Z d d � Z d d � Z d S)� ExitSwallowsExceptionTestCasec C sV Gd d � d � } y | � �
d d Wd QRXWn t k
rQ | j d � Yn Xd S)Nc @ s( e Z d Z d d � Z d d � Z d S)zSExitSwallowsExceptionTestCase.testExitTrueSwallowsException.<locals>.AfricanSwallowc S s d S)Nr )r r r r r � s z]ExitSwallowsExceptionTestCase.testExitTrueSwallowsException.<locals>.AfricanSwallow.__enter__c S s d S)NTr )r r� r� r� r r r r � s z\ExitSwallowsExceptionTestCase.testExitTrueSwallowsException.<locals>.AfricanSwallow.__exit__N)r r r r r r r r r �AfricanSwallow� s r� r+ r z,ZeroDivisionError should have been swallowed)r} r� )r r� r r r � testExitTrueSwallowsException� s
z;ExitSwallowsExceptionTestCase.testExitTrueSwallowsExceptionc C sV Gd d � d � } y | � �
d d Wd QRXWn t k
rD Yn X| j d � d S)Nc @ s( e Z d Z d d � Z d d � Z d S)zZExitSwallowsExceptionTestCase.testExitFalseDoesntSwallowException.<locals>.EuropeanSwallowc S s d S)Nr )r r r r r � s zdExitSwallowsExceptionTestCase.testExitFalseDoesntSwallowException.<locals>.EuropeanSwallow.__enter__c S s d S)NFr )r r� r� r� r r r r � s zcExitSwallowsExceptionTestCase.testExitFalseDoesntSwallowException.<locals>.EuropeanSwallow.__exit__N)r r r r r r r r r �EuropeanSwallow� s r� r+ r z)ZeroDivisionError should have been raised)r} r� )r r� r r r �#testExitFalseDoesntSwallowException� s
zAExitSwallowsExceptionTestCase.testExitFalseDoesntSwallowExceptionN)r r r r� r� r r r r r� � s
r� c @ s� e Z d Z Gd d � d e � Z Gd d � d e � Z Gd d � d e � Z Gd d � d e � Z d d
� Z d d � Z d
d � Z
d d � Z d d � Z d S)�
NestedWithc @ s: e Z d Z d d d d � Z d d � Z d d � Z d S) zNestedWith.DummyNFc C s: | d k r | } | | _ | | _ d | _ d | _ d S)NF)r �gobbler r )r r r� r r r r � s zNestedWith.Dummy.__init__c C s d | _ | j S)NT)r r )r r r r r � s zNestedWith.Dummy.__enter__c G s# d | _ | | _ | j r d Sd S)NT)r r( r� )r r( r r r r � s zNestedWith.Dummy.__exit__)r r r r r r r r r r �Dummy� s r� c @ s e Z d Z d d � Z d S)zNestedWith.InitRaisesc C s
t � � d S)N)r$ )r r r r r � s z NestedWith.InitRaises.__init__N)r r r r r r r r �
InitRaises� s r� c @ s( e Z d Z d d � Z d d � Z d S)zNestedWith.EnterRaisesc C s
t � � d S)N)r$ )r r r r r � s z NestedWith.EnterRaises.__enter__c G s d S)Nr )r r( r r r r � s z NestedWith.EnterRaises.__exit__N)r r r r r r r r r � EnterRaises� s r� c @ s( e Z d Z d d � Z d d � Z d S)zNestedWith.ExitRaisesc C s d S)Nr )r r r r r � s z NestedWith.ExitRaises.__enter__c G s
t � � d S)N)r$ )r r( r r r r � s z NestedWith.ExitRaises.__exit__N)r r r r r r r r r �
ExitRaises� s r� c C sp | j � �= } | j � �' } | j | j � | j | j � Wd QRXWd QRX| j | j � | j | j � d S)N)r� rS r r )r r� r� r r r �testNoExceptions� s
zNestedWith.testNoExceptionsc C s] y. | j � � } | j � � Wd QRXWd QRXWn Yn X| j | j � | j | j � d S)N)r� r� rS r r )r r� r r r �testExceptionInExprList� s z"NestedWith.testExceptionInExprListc C s� y; | j � �( } | j � � | j d � Wd QRXWd QRXWn t k
rO Yn X| j d � | j | j � | j | j � d S)Nzbody of bad with executedzRuntimeError not reraised)r� r� r� r$ rS r r )r r� r r r �testExceptionInEnter� s
z NestedWith.testExceptionInEnterc C s� d } | j d d � �! } | j � � d } Wd QRXWd QRX| j | j � | j | j � | j | � | j | j d d � d S)NFr� Tr )r� r� rS r r ZassertNotEqualr( )r Z
body_executedr� r r r �testExceptionInExit� s "
z NestedWith.testExceptionInExitc C s� | j d d � �o \ } } | j d d � �M \ } } | j d | � | j d | � | j d | � | j d | � Wd QRXWd QRXd S)Nr r+ r, r� � )r+ r, )r� r� )r� rK )r Za1Za2Zb1Zb2r r r �testEnterReturnsTuple� s z NestedWith.testEnterReturnsTupleN)
r r r r8 r� r� r� r� r� r� r� r� r� r r r r r� � s r� �__main__) �
__author__Z __email__r' Zunittest� collectionsr �
contextlibr r r r r8 r r r! r/ ZTestCaser0 rP r` rk ry r� r� r� r� r �mainr r r r �<module> s.
$O/ 5 A �9&L