Your IP : 216.73.216.213
�
h fhg � @ s� d Z d Z d Z d d l Z d d l Z d d l m Z d d l m Z m Z d d l
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 d& d' � Z e d( k r�e � n d S))z7Unit tests for the with statement specified in PEP 343.z
Mike Blandzmbland at acm dot org� N)�deque)�_GeneratorContextManager�contextmanager)� run_unittestc 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__� �3/opt/alt/python34/lib64/python3.4/test/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 8 s zNested.__init__c C s� | j d k r t d � � n t � | _ g } y; x4 | j D]) } | j | j � � | j j | � q= WWn# | j t j � � s� � n 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� | } xD | j D]9 } y | j | � r1 d } n Wq t j � } Yq Xq Wd | _ | | k r� | d | d � j | d � � n d S)Nr � � )NNN)r$ r r( r) �with_traceback)r r) Zexr+ r r r r J s
zNested.__exit__N)r r r r r r r r r r r" 6 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 [ s zMockNested.__init__c C s d | _ t j | � S)NT)r r" r )r r r r r a s zMockNested.__enter__c G s" d | _ | | _ t j | | � S)NT)r
r r" r )r r) r r r r e s zMockNested.__exit__N)r 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 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 QXd S)N)�foor r r r �fooNotDeclaredm s z5FailureTestCase.testNameError.<locals>.fooNotDeclared)� assertRaises� NameError)r r2 r r r �
testNameErrorl 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 s s zDFailureTestCase.testEnterAttributeError.<locals>.LacksEnter.__exit__N)r r r r r r r r �
LacksEnterr s r6 c s � � } | � Wd QXd S)Nr )r1 )r6 r r �
fooLacksEnterv s z>FailureTestCase.testEnterAttributeError.<locals>.fooLacksEnter)�objectr3 �AttributeError)r r7 r )r6 r �testEnterAttributeErrorq 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 QXd S)Nr )r1 )r; r r � fooLacksExit� s z<FailureTestCase.testExitAttributeError.<locals>.fooLacksExit)r8 r3 r9 )r r< r )r; r �testExitAttributeError{ 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 QXd S)N)r1 )Zct)rH r r r � shouldThrow� s z4FailureTestCase.testEnterThrows.<locals>.shouldThrow)r8 r3 r% � assertEqualr1 )r rI 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 rM c s � � � Wd QXd S)Nr r )rM r r rI � s
z3FailureTestCase.testExitThrows.<locals>.shouldThrow)r8 r3 r% )r rI r )rM r �testExitThrows� s z FailureTestCase.testExitThrowsN)
r r r r5 r: r= rC rD rE rF rG rK rN r r r r r0 k 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
rJ r )r � mock_managerr r r � assertInWithManagerInvariants� s z:ContextmanagerAssertionMixin.assertInWithManagerInvariantsc C s7 | j | j � | j | j � | j | j | � d S)N)rR r r
rJ r )r rT r r r r � assertAfterWithManagerInvariants� s z=ContextmanagerAssertionMixin.assertAfterWithManagerInvariantsc C s | j | d � d S)N)NNN)rV )r rT r r r �'assertAfterWithManagerInvariantsNoError� s zDContextmanagerAssertionMixin.assertAfterWithManagerInvariantsNoErrorc C s$ | j | j � | j | j � d S)N)rR r rS r )r �mock_generatorr r r � assertInWithGeneratorInvariants� s z<ContextmanagerAssertionMixin.assertInWithGeneratorInvariantsc C s$ | j | j � | j | j � d S)N)rR r r )r rX r r r �)assertAfterWithGeneratorInvariantsNoError� s zFContextmanagerAssertionMixin.assertAfterWithGeneratorInvariantsNoErrorc C s
| j � d S)N)rP )r r r r �raiseTestException� s z/ContextmanagerAssertionMixin.raiseTestExceptionNc C s� | j | j � | j | j � | d k rX | j | j d | j � t | j � } n | j | j d | � | j | j d | � | j | j d d � d S)Nr, r r- ) rR r r
rJ r rP r ZassertIsInstanceZ assertIsNot)r rT �exc_typer r r �)assertAfterWithManagerInvariantsWithError� s zFContextmanagerAssertionMixin.assertAfterWithManagerInvariantsWithErrorc C s$ | j | j � | j | j � d S)N)rR r r )r rX r r r �+assertAfterWithGeneratorInvariantsWithError� s zHContextmanagerAssertionMixin.assertAfterWithGeneratorInvariantsWithError) r r r rQ rU rV rW rY rZ r[ r] r^ r r r r rO � s rO 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 QXd S)N)r! )r r r r �testInlineGeneratorSyntax� s
z0NonexceptionalTestCase.testInlineGeneratorSyntaxc C s' t � } | � Wd QX| j | � d S)N)r! rW )r r r r r �testUnboundGenerator� s z+NonexceptionalTestCase.testUnboundGeneratorc C s0 t � � } | j | � Wd QX| j | � d S)N)r! rY rZ )r r1 r r r � testInlineGeneratorBoundSyntax� s z5NonexceptionalTestCase.testInlineGeneratorBoundSyntaxc C s6 d } t � � } | j | � Wd QX| j | � d S)N)r! rY rZ )r r1 r r r �*testInlineGeneratorBoundToExistingVariable� s zANonexceptionalTestCase.testInlineGeneratorBoundToExistingVariablec C s9 t � � | _ | j | j � Wd QX| j | j � d S)N)r! r1 rY rZ )r r r r �(testInlineGeneratorBoundToDottedVariable s z?NonexceptionalTestCase.testInlineGeneratorBoundToDottedVariablec C sP t � } | �! } | j | � | j | � Wd QX| j | � | j | � d S)N)r! rY rU rZ rW )r r r1 r r r �testBoundGenerator s
z)NonexceptionalTestCase.testBoundGeneratorc C s� t � } | �� } t � } | �; } | j | � | j | � | j | � | j | � Wd QX| j | � | j | � | j | � | j | � Wd QX| j | � | j | � d S)N)r! rU rY rW rZ )r �mock_ar1 �mock_b�barr r r �testNestedSingleStatements
s
z1NonexceptionalTestCase.testNestedSingleStatementsN)
r r r r` ra rb rc rd re ri 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 QXd S)N)r" r! )r r r r �"testSingleArgInlineGeneratorSyntax s z?NestedNonexceptionalTestCase.testSingleArgInlineGeneratorSyntaxc C s< t � } t | � � } | j | � Wd QX| j | � d S)N)r! r" rU rW )r �mr1 r r r � testSingleArgBoundToNonTuple$ s z9NestedNonexceptionalTestCase.testSingleArgBoundToNonTuplec C s< t � } t | � � } | j | � Wd QX| j | � d S)N)r! r" rU rW )r rl r1 r r r �2testSingleArgBoundToSingleElementParenthesizedList, s zONestedNonexceptionalTestCase.testSingleArgBoundToSingleElementParenthesizedListc C s d d � } | j t | � d S)Nc S s" t t � � �
\ } } Wd QXd S)N)r" r! )r1 rh r r r �shouldThrowValueError5 s ziNestedNonexceptionalTestCase.testSingleArgBoundToMultipleElementTupleError.<locals>.shouldThrowValueError)r3 �
ValueError)r ro r r r �-testSingleArgBoundToMultipleElementTupleError4 s zJNestedNonexceptionalTestCase.testSingleArgBoundToMultipleElementTupleErrorc C sZ t � } t | � } | � | j | � | j | � Wd QX| j | � | j | � d S)N)r! r/ rU rW )r r � mock_nestedr r r �testSingleArgUnbound: s
z1NestedNonexceptionalTestCase.testSingleArgUnboundc C s� t � } t � } t � } t | | | � } | �9 | j | � | j | � | j | � | j | � Wd QX| j | � | j | � | j | � | j | � d S)N)r! r/ rU rW )r rl �n�orr r r r �testMultipleArgUnboundC s
z3NestedNonexceptionalTestCase.testMultipleArgUnboundc
C s� t t � t � t � � } | �D \ } } } | j | � | j | � | j | � | j | � Wd QX| j | � | j | � | j | � | j | � d S)N)r/ r! rY rU rZ rW )r rr rl rt ru r r r �testMultipleArgBoundR s
z1NestedNonexceptionalTestCase.testMultipleArgBoundN)
r r r rk rm rn rq rs rv rw r r r r rj s rj 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 QXd S)N)�resourcerU rY r[ r )�cmr r r rI c s
z;ExceptionalTestCase.testSingleResource.<locals>.shouldThrow)r! r3 r% r] r^ ry )r rI r )rz r r �testSingleResourcea s
z&ExceptionalTestCase.testSingleResourcec sB t � � � � f d d � } � j t | � � j � t � d S)Nc s � � � _ d d Wd QXd S)Nr, r )ry r )rz r r r rI n s z@ExceptionalTestCase.testExceptionNormalized.<locals>.shouldThrow)r! r3 �ZeroDivisionErrorr] )r rI r )rz r r �testExceptionNormalizedl s z+ExceptionalTestCase.testExceptionNormalizedc sx t � � t � � � � � f d d � } � j t | � � j � � � j � � � j � j � � j � j � d S)Nc sl � �` � _ � �N � _ � j � � � j � � � j � j � � j � j � � j � Wd QXWd QXd S)N)r1 rh rU rY r[ r )rf rg r r r rI y s
zCExceptionalTestCase.testNestedSingleStatements.<locals>.shouldThrow)r! r3 r% r] r^ r1 rh )r rI r )rf rg r r ri v s
z.ExceptionalTestCase.testNestedSingleStatementsc s� t � � t � � t � � � � � � � � f d d � } � j t | � � j � � � j � � � j � � � j � j � � j � j � d S)Nc
sp � �d \ � _ � _ � j � � � j � � � j � � � j � j � � j � j � � j � Wd QXd S)N)�
resource_a�
resource_brU rY r[ r )�cm_a�cm_brr r r r rI � s
zOExceptionalTestCase.testMultipleResourcesInSingleStatement.<locals>.shouldThrow)r! r/ r3 r% r] r^ r~ r )r rI r )r� r� rr 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 sO � �C � _ � j � � � j � j � � j � � �
� _ Wd QXWd QXd S)N)r1 rU rY r[ rh r )rf rg r r r rI � s
zPExceptionalTestCase.testNestedExceptionBeforeInnerStatement.<locals>.shouldThrow) r! rh r3 r% r] r^ r1 rJ rS r r
r )r rI r )rf rg r r �'testNestedExceptionBeforeInnerStatement� s
z;ExceptionalTestCase.testNestedExceptionBeforeInnerStatementc sx t � � t � � � � � f d d � } � j t | � � j � � � j � � � j � j � � j � j � d S)Nc sl � �` � _ � �D � _ � j � � � j � � � j � j � � j � j � Wd QX� j � Wd QXd S)N)r1 rh rU rY r[ r )rf rg r r r rI � s
zOExceptionalTestCase.testNestedExceptionAfterInnerStatement.<locals>.shouldThrow) r! r3 r% r] rW r^ r1 rZ rh )r rI r )rf rg r r �&testNestedExceptionAfterInnerStatement� s
z:ExceptionalTestCase.testNestedExceptionAfterInnerStatementc s8 t d d � � � � f d d � } | j t | � d S)Nc s s d Vd S)Nr r r r r rz � s z8ExceptionalTestCase.testRaisedStopIteration1.<locals>.cmc s � � � t d � � Wd QXd S)Nz from with)�
StopIterationr )rz r r rI � s
zAExceptionalTestCase.testRaisedStopIteration1.<locals>.shouldThrow)r r3 r� )r rI r )rz 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 rz � s rz c s � � � t d � � Wd QXd S)Nz from with)r� r )rz r r rI � s
zAExceptionalTestCase.testRaisedStopIteration2.<locals>.shouldThrow)r8 r3 r� )r rI r )rz r �testRaisedStopIteration2� s z,ExceptionalTestCase.testRaisedStopIteration2c s8 t d d � � � � f d d � } | j t | � d S)Nc s s d Vd S)Nr r r r r rz � s z8ExceptionalTestCase.testRaisedStopIteration3.<locals>.cmc
s& � � � t t g � � � Wd QXd S)N)�next�iterr )rz r r rI � s
zAExceptionalTestCase.testRaisedStopIteration3.<locals>.shouldThrow)r r3 r� )r rI r )rz 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 rz � s z8ExceptionalTestCase.testRaisedGeneratorExit1.<locals>.cmc s � � � t d � � Wd QXd S)Nz from with)�
GeneratorExitr )rz r r rI � s
zAExceptionalTestCase.testRaisedGeneratorExit1.<locals>.shouldThrow)r r3 r� )r rI r )rz 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 rz � s rz c s � � � t d � � Wd QXd S)Nz from with)r� r )rz r r rI � s
zAExceptionalTestCase.testRaisedGeneratorExit2.<locals>.shouldThrow)r8 r3 r� )r rI r )rz 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 rz s rz c s* � d d � � � � j d � Wd QXd S)Nc S s d S)NTr r r r r �<lambda> s zJExceptionalTestCase.testErrorsInBool.<locals>.trueAsBool.<locals>.<lambda>zShould NOT see this)�failr )rz r r r �
trueAsBool s z8ExceptionalTestCase.testErrorsInBool.<locals>.trueAsBoolc s* � d d � � � � j d � Wd QXd S)Nc S s d S)NFr r r r r r� s zKExceptionalTestCase.testErrorsInBool.<locals>.falseAsBool.<locals>.<lambda>z Should raise)r� r )rz r r r � falseAsBool s z9ExceptionalTestCase.testErrorsInBool.<locals>.falseAsBoolc s* � d d � � � � j d � Wd QXd 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 )rz r r r �
failAsBool s z8ExceptionalTestCase.testErrorsInBool.<locals>.failAsBool)r8 r3 �AssertionErrorr| )r r� r� r� r )rz r r �testErrorsInBool s z$ExceptionalTestCase.testErrorsInBoolN)r r r r{ r} ri r� r� r� r� r� r� r� r� r� r r r r rx ` s
rx 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 sP d } x3 | d 7} t � � | d 7} PWd QX| d 7} q W| j | d � d S)Nr r, �
�d � )r! rJ )r �counterr r r �
testWithBreak" s
z)NonLocalFlowControlTestCase.testWithBreakc C sb d } xE | d 7} | d k r# Pn t � � | d 7} w Wd QX| d 7} q W| j | d � d S)Nr r, r- r� r� � )r! rJ )r r� r r r �testWithContinue, s
z,NonLocalFlowControlTestCase.testWithContinuec C s# d d � } | j | � d � d S)Nc S sC d } x6 | d 7} t � � | d 7} | SWd QX| d 7} q Wd S)Nr r, r� r� )r! )r� r r r r1 9 s
z7NonLocalFlowControlTestCase.testWithReturn.<locals>.foor� )rJ )r r1 r r r �testWithReturn8 s z*NonLocalFlowControlTestCase.testWithReturnc C s5 d d � } t | � � } | j | d d g � d S)Nc s s t � � d Vd VWd QXd S)Nr� �
)r! r r r r �genD s
z6NonLocalFlowControlTestCase.testWithYield.<locals>.genr� r� )�listrJ )r r� �xr r r �
testWithYieldC s z)NonLocalFlowControlTestCase.testWithYieldc C st d } y8 | d 7} t � � | d 7} t � Wd QX| d 7} Wn" t k
rb | j | d � Yn X| j d � d S)Nr r, r� r� r� zDidn't raise RuntimeError)r! r% rJ r� )r r� r r r �
testWithRaiseK 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 s` i d d d g d 6} t � �I | d d <| j t | j � � d g � | j | d d j t � Wd QXt � �U t | j � � d d <| j t | j � � d g � | j | d d j t � Wd QXt � �= | d <t | j � � } | j � | j | d d g � Wd QXGd d � d � } | � } t � �# | _ | j t | d � d � Wd QXd S)Nr r, r- c @ s e Z d Z d S)z;AssignmentTargetTestCase.testSingleComplexTarget.<locals>.CN)r r r r r r r �Cg s r� r1 T)
r! rJ r� �keysr r �values�sortr1 �hasattr)r �targetsr� r� �blahr r r �testSingleComplexTarget[ s ! !
z0AssignmentTargetTestCase.testSingleComplexTargetc C s� Gd d � d � } i d d d g d 6} | � �H \ | d d <| d d <| d d <| j | i d d d g d 6� Wd QX| � �l \ t | j � � d d <t | j � � d d <t | j � � d d <| j | i d d d g d 6� Wd QX| � �A \ | d <| d <| d <| j | i d d 6d d 6d d 6� Wd QXGd d � d � } | � } | � �R \ | _ | _ | _ | j | j d � | j | j d � | j | j d � Wd QXd 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 n s zHAssignmentTargetTestCase.testMultipleComplexTargets.<locals>.C.__enter__c S s d S)Nr )r �t�v�tbr r r r o s zGAssignmentTargetTestCase.testMultipleComplexTargets.<locals>.C.__exit__N)r r r r r r r r r r� m 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 �Bw s r� )rJ r� r� ZoneZtwoZthree)r r� r� r� r� r r r �testMultipleComplexTargetsl s -&Q&!+ z3AssignmentTargetTestCase.testMultipleComplexTargetsN)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 S)� ExitSwallowsExceptionTestCasec C sU Gd d � d � } y | � �
d d Wd QXWn t k
rP | 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 sU Gd d � d � } y | � �
d d Wd QXWn t k
rC 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 | } n | | _ | | _ 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 sn | j � �< } | j � �' } | j | j � | j | j � Wd QXWd QX| j | j � | j | j � d S)N)r� rR r r
)r r� r� r r r �testNoExceptions� s
zNestedWith.testNoExceptionsc C s[ y, | j � � } | j � � Wd QXWd QXWn Yn X| j | j � | j | j � d S)N)r� r� rR r r
)r r� r r r �testExceptionInExprList� s z"NestedWith.testExceptionInExprListc C s y9 | j � �' } | j � � | j d � Wd QXWd QXWn t k
rM Yn X| j d � | j | j � | j | j � d S)Nzbody of bad with executedzRuntimeError not reraised)r� r� r� r% rR r r
)r r� r r r �testExceptionInEnter� s
z NestedWith.testExceptionInEnterc C s� d } | j d d � � } | j � � d } Wd QXWd QX| j | j � | j | j � | j | � | j | j d d � d S)NFr� Tr )r� r� rR r r
ZassertNotEqualr) )r Z
body_executedr� r r r �testExceptionInExit� s "
z NestedWith.testExceptionInExitc C s� | j d d � �n \ } } | j d d � �M \ } } | j d | � | j d | � | j d | � | j d | � Wd QXWd QXd S)Nr r, r- r� � )r, r- )r� r� )r� rJ )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� c C s# t t t t t t t t t � d S)N) r r0 r_ rj rx r� r� r� r� r r r r � test_main� s r� �__main__) �__doc__�
__author__Z __email__r( Zunittest� collectionsr �
contextlibr r Z test.supportr r r r8 r r! r" r/ ZTestCaser0 rO r_ rj rx r� r� r� r� r� r r r r r �<module> s4
$O/ 5 A �9&L