File: //usr/local/lib/python3.7/test/__pycache__/test_with.cpython-37.pyc
B
��g g � @ sH d Z dZdZddlZddlZddlmZ ddlmZm Z G dd� de�Z
d d
� Z G d d � d e �Z
e d
d� �ZG dd� de �ZG dd� de�ZG dd� dej�ZG dd� de �ZG dd� deje�ZG dd� deje�ZG dd � d eej�ZG d d � d ej�ZG d d � d ej�ZG d!d"� d"ej�ZG d#d$� d$ej�Zed%k�rDe� � dS )&z7Unit tests for the with statement specified in PEP 343.z
Mike Blandzmbland at acm dot org� N)�deque)�_GeneratorContextManager�contextmanagerc s, e Zd Z� fdd�Zdd� Zdd� Z� ZS )�MockContextManagerc s" t � j|� d| _d| _d | _d S )NF)�super�__init__� enter_called� exit_called� exit_args)�self�args)� __class__� �*/usr/local/lib/python3.7/test/test_with.pyr s zMockContextManager.__init__c C s d| _ t�| �S )NT)r r � __enter__)r r r r r s z MockContextManager.__enter__c C s" d| _ |||f| _t�| |||�S )NT)r r
r �__exit__)r �type�value� tracebackr r r r s zMockContextManager.__exit__)�__name__�
__module__� __qualname__r r r �
__classcell__r r )r
r r
s r c s � fdd�}|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 Zd Zdd� ZdS )� 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 � } zd| _| V W d d| _X d S )NT)r r r )�mockr r r � mock_contextmanager_generator+ s
r! c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�Nestedc G s || _ d | _d S )N)�managers�entered)r r# r r r r 7 s zNested.__init__c C sn | j d k rtd��t� | _ g }y.x(| jD ] }|�|�� � | j �|� q(W W n | jt� � � sd� Y nX |S )NzContext is not reentrant)
r$ � RuntimeErrorr r# �appendr �
appendleftr �sys�exc_info)r �vars�mgrr r r r ; s
zNested.__enter__c G sh |}x6| j D ],}y|j|� r d}W q t�� }Y q X q W d | _ ||k rd|d |d ��|d ��d S )N)NNNr � � )r$ r r( r) �with_traceback)r r) Zexr+ r r r r I s
zNested.__exit__N)r r r r r r r r r r r" 5 s r" c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�
MockNestedc G s& t j| f|�� 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�| �S )NT)r r" r )r r r r r ` s zMockNested.__enter__c G s d| _ || _tj| f|�� 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 Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dS )�FailureTestCasec C s dd� }| � t|� d S )Nc S s t � W d Q R X d 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. G dd� dt �� � fdd�}| �td|� d S )Nc @ s e Zd Zdd� ZdS )z<FailureTestCase.testEnterAttributeError1.<locals>.LacksEnterc S s d S )Nr )r r r r r r r r r s zEFailureTestCase.testEnterAttributeError1.<locals>.LacksEnter.__exit__N)r r r r r r r r �
LacksEnterq s r6 c s � � } | � W d Q R X d S )Nr )r1 )r6 r r �
fooLacksEnteru s z?FailureTestCase.testEnterAttributeError1.<locals>.fooLacksEnterr )�object�assertRaisesRegex�AttributeError)r r7 r )r6 r �testEnterAttributeError1p s z(FailureTestCase.testEnterAttributeError1c s. G dd� dt �� � fdd�}| �td|� d S )Nc @ s e Zd ZdS )zCFailureTestCase.testEnterAttributeError2.<locals>.LacksEnterAndExitN)r r r r r r r �LacksEnterAndExit{ s r< c s � � } | � W d Q R X d S )Nr )r1 )r< r r �fooLacksEnterAndExit~ s zFFailureTestCase.testEnterAttributeError2.<locals>.fooLacksEnterAndExitr )r8 r9 r: )r r= r )r< r �testEnterAttributeError2z s z(FailureTestCase.testEnterAttributeError2c s. G dd� dt �� � fdd�}| �td|� d S )Nc @ s e Zd Zdd� ZdS )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 � � } | � W d Q R X d S )Nr )r1 )r? r r � fooLacksExit� s z<FailureTestCase.testExitAttributeError.<locals>.fooLacksExitr )r8 r9 r: )r r@ r )r? r �testExitAttributeError� s z&FailureTestCase.testExitAttributeErrorc C s dd� }| � t||� d S )Nc S s t | dd� d S )N� Zsingle)�compile)�sr r r �shouldRaiseSyntaxError� s zGFailureTestCase.assertRaisesSyntaxError.<locals>.shouldRaiseSyntaxError)r3 � SyntaxError)r ZcodestrrE r r r �assertRaisesSyntaxError� s z'FailureTestCase.assertRaisesSyntaxErrorc C s | � d� | � d� d S )Nzwith mock as None:
passzwith mock as (None):
pass)rG )r r r r �testAssignmentToNoneError� s
z)FailureTestCase.testAssignmentToNoneErrorc C s | � d� | � d� d S )Nzwith mock as None,:
passz with mock as (None,):
pass)rG )r r r r �,testAssignmentToTupleOnlyContainingNoneError� s
z<FailureTestCase.testAssignmentToTupleOnlyContainingNoneErrorc C s | � d� d S )Nz%with mock as (foo, None, bar):
pass)rG )r r r r �(testAssignmentToTupleContainingNoneError� s z8FailureTestCase.testAssignmentToTupleContainingNoneErrorc s< G dd� dt �� � �fdd�}��t|� ���jd � d S )Nc @ s e Zd Zdd� Zdd� ZdS )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 rK c s" � � } d �_ | ��_ W d Q R X d S )N)r1 )Zct)rK r r r � shouldThrow� s z4FailureTestCase.testEnterThrows.<locals>.shouldThrow)r8 r3 r% � assertEqualr1 )r rL r )rK r r �testEnterThrows� s z FailureTestCase.testEnterThrowsc s, G dd� dt �� � fdd�}| �t|� d S )Nc @ s e Zd Zdd� Zdd� ZdS )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 rP c s � � � W d Q R X d S )Nr r )rP r r rL � s z3FailureTestCase.testExitThrows.<locals>.shouldThrow)r8 r3 r% )r rL r )rP r �testExitThrows� s z FailureTestCase.testExitThrowsN)
r r r r5 r; r> rA rG rH rI rJ rN rQ r r r r r0 j s
r0 c @ sV e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z ddd�Z
dd� Z dS )� ContextmanagerAssertionMixinc C s t d�| _d S )Nztest exception)r% �TEST_EXCEPTION)r r r r �setUp� s z"ContextmanagerAssertionMixin.setUpc C s* | � |j� | �|j� | �|jd � d S )N)�
assertTruer � assertFalser rM r
)r � mock_managerr r r � assertInWithManagerInvariants� s z:ContextmanagerAssertionMixin.assertInWithManagerInvariantsc C s* | � |j� | � |j� | �|j|� d S )N)rU r r rM r
)r rW r
r r r � assertAfterWithManagerInvariants� s z=ContextmanagerAssertionMixin.assertAfterWithManagerInvariantsc C s | � |d� d S )N)NNN)rY )r rW r r r �'assertAfterWithManagerInvariantsNoError� s zDContextmanagerAssertionMixin.assertAfterWithManagerInvariantsNoErrorc C s | � |j� | �|j� d S )N)rU r rV r )r �mock_generatorr r r � assertInWithGeneratorInvariants� s z<ContextmanagerAssertionMixin.assertInWithGeneratorInvariantsc C s | � |j� | � |j� d S )N)rU r r )r r[ r r r �)assertAfterWithGeneratorInvariantsNoError� s zFContextmanagerAssertionMixin.assertAfterWithGeneratorInvariantsNoErrorc C s
| j �d S )N)rS )r r r r �raiseTestException� s z/ContextmanagerAssertionMixin.raiseTestExceptionNc C sx | � |j� | � |j� |d kr>| �|jd | j� t| j�}| �|jd |� | �|jd |� | �|jd d � d S )Nr, r r- ) rU r r rM r
rS r ZassertIsInstanceZ assertIsNot)r rW �exc_typer r r �)assertAfterWithManagerInvariantsWithError� s
zFContextmanagerAssertionMixin.assertAfterWithManagerInvariantsWithErrorc C s | � |j� | � |j� d S )N)rU r r )r r[ r r r �+assertAfterWithGeneratorInvariantsWithError� s zHContextmanagerAssertionMixin.assertAfterWithGeneratorInvariantsWithError)N) r r r rT rX rY rZ r\ r] r^ r` ra r r r r rR � s
rR c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�NonexceptionalTestCasec C s t � � W d Q R X d S )N)r! )r r r r �testInlineGeneratorSyntax� s z0NonexceptionalTestCase.testInlineGeneratorSyntaxc C s$ t � }|� W d Q R X | �|� d S )N)r! rZ )r r r r r �testUnboundGenerator� s
z+NonexceptionalTestCase.testUnboundGeneratorc C s* t � �}| �|� W d Q R X | �|� d S )N)r! r\ r] )r r1 r r r � testInlineGeneratorBoundSyntax� s z5NonexceptionalTestCase.testInlineGeneratorBoundSyntaxc C s. d }t � �}| �|� W d Q R X | �|� d S )N)r! r\ r] )r r1 r r r �*testInlineGeneratorBoundToExistingVariable� s zANonexceptionalTestCase.testInlineGeneratorBoundToExistingVariablec C s0 t � �| _| �| j� W d Q R X | �| j� d S )N)r! r1 r\ r] )r r r r �(testInlineGeneratorBoundToDottedVariable s
z?NonexceptionalTestCase.testInlineGeneratorBoundToDottedVariablec C sB t � }|�}| �|� | �|� W d Q R X | �|� | �|� d S )N)r! r\ rX r] rZ )r r r1 r r r �testBoundGenerator s
z)NonexceptionalTestCase.testBoundGeneratorc
C s� t � }|�l}t � }|�.}| �|� | �|� | �|� | �|� W d Q R X | �|� | �|� | �|� | �|� W d Q R X | �|� | �|� d S )N)r! rX r\ rZ r] )r �mock_ar1 �mock_b�barr r r �testNestedSingleStatements s
z1NonexceptionalTestCase.testNestedSingleStatementsN)
r r r rc rd re rf rg rh rl r r r r rb � s rb c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )� NestedNonexceptionalTestCasec C s t t� �� W d Q R X d S )N)r" r! )r r r r �"testSingleArgInlineGeneratorSyntax# s z?NestedNonexceptionalTestCase.testSingleArgInlineGeneratorSyntaxc C s2 t � }t|��}| �|� W d Q R X | �|� d S )N)r! r" rX rZ )r �mr1 r r r � testSingleArgBoundToNonTuple' s
z9NestedNonexceptionalTestCase.testSingleArgBoundToNonTuplec C s2 t � }t|��}| �|� W d Q R X | �|� d S )N)r! r" rX rZ )r ro r1 r r r �2testSingleArgBoundToSingleElementParenthesizedList/ s
zONestedNonexceptionalTestCase.testSingleArgBoundToSingleElementParenthesizedListc C s dd� }| � t|� d S )Nc S s t t� ��
\} }W d Q R X d S )N)r" r! )r1 rk r r r �shouldThrowValueError8 s ziNestedNonexceptionalTestCase.testSingleArgBoundToMultipleElementTupleError.<locals>.shouldThrowValueError)r3 �
ValueError)r rr r r r �-testSingleArgBoundToMultipleElementTupleError7 s zJNestedNonexceptionalTestCase.testSingleArgBoundToMultipleElementTupleErrorc C sJ t � }t|�}|� | �|� | �|� W d Q R X | �|� | �|� d S )N)r! r/ rX rZ )r r � mock_nestedr r r �testSingleArgUnbound= s
z1NestedNonexceptionalTestCase.testSingleArgUnboundc C s� t � }t � }t � }t|||�}|�. | �|� | �|� | �|� | �|� W d Q R X | �|� | �|� | �|� | �|� d S )N)r! r/ rX rZ )r ro �n�oru r r r �testMultipleArgUnboundF s
z3NestedNonexceptionalTestCase.testMultipleArgUnboundc C s| t t� t� t� �}|�4\}}}| �|� | �|� | �|� | �|� W d Q R X | �|� | �|� | �|� | �|� d S )N)r/ r! r\ rX r] rZ )r ru ro rw rx r r r �testMultipleArgBoundU s
z1NestedNonexceptionalTestCase.testMultipleArgBoundN)
r r r rn rp rq rt rv ry rz r r r r rm ! s rm c @ sl e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )�ExceptionalTestCasec s: t � � � �fdd�}��t|� ��� � ���j� d S )Nc s4 � �&�_ ��� � ���j � ��� W d Q R X d S )N)�resourcerX r\ r^ r )�cmr r r rL f s
z;ExceptionalTestCase.testSingleResource.<locals>.shouldThrow)r! r3 r% r` ra r| )r rL r )r} r r �testSingleResourced s
z&ExceptionalTestCase.testSingleResourcec s0 t � � � �fdd�}��t|� ��� t� d S )Nc s � ��_ dd W d Q R X d S )Nr, r )r| r )r} r r r rL q s z@ExceptionalTestCase.testExceptionNormalized.<locals>.shouldThrow)r! r3 �ZeroDivisionErrorr` )r rL r )r} r r �testExceptionNormalizedo s z+ExceptionalTestCase.testExceptionNormalizedc sX t � � t � �� ��fdd�}��t|� ��� � ���� ���j� ���j� d S )Nc
s\ � �N�_ ��<�_��� � ���� ���j � ���j� ��� W d Q R X W d Q R X d S )N)r1 rk rX r\ r^ r )ri rj r r r rL | s
zCExceptionalTestCase.testNestedSingleStatements.<locals>.shouldThrow)r! r3 r% r` ra r1 rk )r rL r )ri rj r r rl y s
z.ExceptionalTestCase.testNestedSingleStatementsc sn t � � t � �t� ���� ���fdd�}��t|� ��� � ���� ���� ���j� ���j� d S )Nc sZ ��L\�_ �_��� � ���� ���� ���j � ���j� ��� W d Q R X d S )N)�
resource_a�
resource_brX r\ r^ r )�cm_a�cm_bru r r r rL � s
zOExceptionalTestCase.testMultipleResourcesInSingleStatement.<locals>.shouldThrow)r! r/ r3 r% r` ra r� r� )r rL r )r� r� ru r r �&testMultipleResourcesInSingleStatement� s
z:ExceptionalTestCase.testMultipleResourcesInSingleStatementc s| t � � t � �d �_� ��fdd�}��t|� ��� � ���j� ���jd � ���j � ���j
� ���j d � d S )Nc
sF � �8�_ ��� � ���j � ��� ���_W d Q R X W d Q R X d S )N)r1 rX r\ r^ rk r )ri rj r r r rL � s
zPExceptionalTestCase.testNestedExceptionBeforeInnerStatement.<locals>.shouldThrow) r! rk r3 r% r` ra r1 rM rV r r r
)r rL r )ri rj r r �'testNestedExceptionBeforeInnerStatement� s
z;ExceptionalTestCase.testNestedExceptionBeforeInnerStatementc sX t � � t � �� ��fdd�}��t|� ��� � ���� ���j� ���j� d S )Nc
s\ � �N�_ ��4�_��� � ���� ���j � ���j� W d Q R X ��� W d Q R X d S )N)r1 rk rX r\ r^ r )ri rj r r r rL � s
zOExceptionalTestCase.testNestedExceptionAfterInnerStatement.<locals>.shouldThrow) r! r3 r% r` rZ ra r1 r] rk )r rL r )ri rj r r �&testNestedExceptionAfterInnerStatement� s
z:ExceptionalTestCase.testNestedExceptionAfterInnerStatementc s: t dd� �� � fdd�}| �td�� |� W d Q R X d S )Nc s s
d V d S )Nr r r r r r} � s z8ExceptionalTestCase.testRaisedStopIteration1.<locals>.cmc s � � � t d��W d Q R X d S )Nz from with)�
StopIterationr )r} r r rL � s zAExceptionalTestCase.testRaisedStopIteration1.<locals>.shouldThrowz from with)r r9 r� )r rL r )r} r �testRaisedStopIteration1� s z,ExceptionalTestCase.testRaisedStopIteration1c s> G dd� dt �� � fdd�}| �td�� |� W d Q R X d S )Nc @ s e Zd Zdd� Zdd� ZdS )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��W d Q R X d S )Nz from with)r� r )r} r r rL � s zAExceptionalTestCase.testRaisedStopIteration2.<locals>.shouldThrowz from with)r8 r9 r� )r rL r )r} r �testRaisedStopIteration2� s z,ExceptionalTestCase.testRaisedStopIteration2c s8 t dd� �� � fdd�}| �t�� |� W d Q R X d S )Nc s s
d V d S )Nr r r r r r} � s z8ExceptionalTestCase.testRaisedStopIteration3.<locals>.cmc s" � � � t tg ���W d Q R X d S )N)�next�iterr )r} r r rL � s zAExceptionalTestCase.testRaisedStopIteration3.<locals>.shouldThrow)r r3 r� )r rL r )r} r �testRaisedStopIteration3� s z,ExceptionalTestCase.testRaisedStopIteration3c s( t dd� �� � fdd�}| �t|� d S )Nc s s
d V d S )Nr r r r r r} � s z8ExceptionalTestCase.testRaisedGeneratorExit1.<locals>.cmc s � � � t d��W d Q R X d S )Nz from with)�
GeneratorExitr )r} r r rL � s zAExceptionalTestCase.testRaisedGeneratorExit1.<locals>.shouldThrow)r r3 r� )r rL r )r} r �testRaisedGeneratorExit1� s z,ExceptionalTestCase.testRaisedGeneratorExit1c s, G dd� dt �� � fdd�}| �t|� d S )Nc @ s e Zd Zdd� Zdd� ZdS )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��W d Q R X d S )Nz from with)r� r )r} r r rL s zAExceptionalTestCase.testRaisedGeneratorExit2.<locals>.shouldThrow)r8 r3 r� )r rL r )r} r �testRaisedGeneratorExit2� s z,ExceptionalTestCase.testRaisedGeneratorExit2c s\ G dd� dt �� � �fdd�}|� � �fdd�}��t|� � �fdd�}��t|� d S ) Nc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )z0ExceptionalTestCase.testErrorsInBool.<locals>.cmc s G � fdd�d�}|� | _ d S )Nc s e Zd Z� fdd�ZdS )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 dS )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& � dd� �� �� d� W d Q R X d S )Nc S s dS )NTr r r r r �<lambda> � zJExceptionalTestCase.testErrorsInBool.<locals>.trueAsBool.<locals>.<lambda>zShould NOT see this)�failr )r} r r r �
trueAsBool s z8ExceptionalTestCase.testErrorsInBool.<locals>.trueAsBoolc s& � dd� �� �� d� W d Q R X d S )Nc S s dS )NFr r r r r r� r� zKExceptionalTestCase.testErrorsInBool.<locals>.falseAsBool.<locals>.<lambda>z Should raise)r� r )r} r r r � falseAsBool s z9ExceptionalTestCase.testErrorsInBool.<locals>.falseAsBoolc s& � dd� �� �� d� W d Q R X d S )Nc S s dd S )Nr, r r r r r r r� ! r� 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� rl r� r� r� r� r� r� r� r� r� r r r r r{ c s
r{ c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �NonLocalFlowControlTestCasec C sF d}x0|d7 }t � � |d7 }P W d Q R X |d7 }qW | �|d� d S )Nr r, �
�d � )r! rM )r �counterr r r �
testWithBreak( s z)NonLocalFlowControlTestCase.testWithBreakc C sP d}x:|d7 }|dkrP t � � |d7 }wW d Q R X |d7 }qW | �|d� d S )Nr r, r- r� r� � )r! rM )r r� r r r �testWithContinue2 s z,NonLocalFlowControlTestCase.testWithContinuec C s dd� }| � |� d� d S )Nc S s8 d} x.| d7 } t � � | d7 } | S Q R X | d7 } qW d S )Nr r, r� r� )r! )r� r r r r1 ? s
z7NonLocalFlowControlTestCase.testWithReturn.<locals>.foor� )rM )r r1 r r r �testWithReturn> s z*NonLocalFlowControlTestCase.testWithReturnc C s& dd� }t |� �}| �|ddg� d S )Nc s s" t � � dV dV W d Q R X d S )Nr� �
)r! r r r r �genJ s z6NonLocalFlowControlTestCase.testWithYield.<locals>.genr� r� )�listrM )r r� �xr r r �
testWithYieldI s
z)NonLocalFlowControlTestCase.testWithYieldc C sf d}y2|d7 }t � � |d7 }t�W d Q R X |d7 }W n tk
rV | �|d� Y n X | �d� d S )Nr r, r� r� r� zDidn't raise RuntimeError)r! r% rM r� )r r� r r r �
testWithRaiseQ s z)NonLocalFlowControlTestCase.testWithRaiseN)r r r r� r� r� r� r� r r r r r� & s
r� c @ s e Zd Zdd� Zdd� ZdS )�AssignmentTargetTestCasec C s ddddgi}t � �:|d d< | �t|�� �dg� | �|d d jt� W d Q R X t � �Bt|�� �d d< | �t|�� �dg� | �|d d jt� W d Q R X t � �.|d< t|�� �}|�� | �|ddg� W d Q R X G dd� d�}|� }t � �|_| �t |d�d� W d Q R X d S )Nr, r r- c @ s e Zd ZdS )z;AssignmentTargetTestCase.testSingleComplexTarget.<locals>.CN)r r r r r r r �Cm s r� r1 T)
r! rM r� �keysr
r �values�sortr1 �hasattr)r �targetsr� r� �blahr r r �testSingleComplexTargeta s
z0AssignmentTargetTestCase.testSingleComplexTargetc C sX G dd� d�}ddddgi}|� �:\|d d< |d d< |d d< | � |ddddgi� W d Q R X |� �R\t|�� �d d< t|�� �d d< t|�� �d d< | � |ddddgi� W d Q R X |� �,\|d< |d< |d< | � |dddd�� W d Q R X G dd � d �}|� }|� �<\|_|_|_| � |jd� | � |jd� | � |jd� W d Q R X d S )
Nc @ s e Zd Zdd� Zdd� ZdS )z>AssignmentTargetTestCase.testMultipleComplexTargets.<locals>.Cc S s dS )N)r, r- r� r )r r r r r t r� zHAssignmentTargetTestCase.testMultipleComplexTargets.<locals>.C.__enter__c S s d S )Nr )r �t�v�tbr r r r u r� zGAssignmentTargetTestCase.testMultipleComplexTargets.<locals>.C.__exit__N)r r r r r r r r r r� s s r� r, r r- r� )r, r- r� c @ s e Zd ZdS )z>AssignmentTargetTestCase.testMultipleComplexTargets.<locals>.BN)r r r r r r r �B} s r� )rM r� r� ZoneZtwoZthree)r r� r� r� r� r r r �testMultipleComplexTargetsr s & > z3AssignmentTargetTestCase.testMultipleComplexTargetsN)r r r r� r� r r r r r� _ s r� c @ s e Zd Zdd� Zdd� ZdS )� ExitSwallowsExceptionTestCasec C sP G dd� d�}y |� � dd W d Q R X W n t k
rJ | �d� Y nX d S )Nc @ s e Zd Zdd� Zdd� ZdS )zSExitSwallowsExceptionTestCase.testExitTrueSwallowsException.<locals>.AfricanSwallowc S s d S )Nr )r r r r r � r� z]ExitSwallowsExceptionTestCase.testExitTrueSwallowsException.<locals>.AfricanSwallow.__enter__c S s dS )NTr )r r� r� r� r r r r � r� 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 sP G dd� d�}y |� � dd W d Q R X W n t k
r@ Y n X | �d� d S )Nc @ s e Zd Zdd� Zdd� ZdS )zZExitSwallowsExceptionTestCase.testExitFalseDoesntSwallowException.<locals>.EuropeanSwallowc S s d S )Nr )r r r r r � r� zdExitSwallowsExceptionTestCase.testExitFalseDoesntSwallowException.<locals>.EuropeanSwallow.__enter__c S s dS )NFr )r r� r� r� r r r r � r� 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 @ st e Zd ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dS )�
NestedWithc @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
zNestedWith.DummyNFc C s( |d kr | }|| _ || _d| _d| _d S )NF)r �gobbler r )r r r� r r r r � s zNestedWith.Dummy.__init__c C s d| _ | jS )NT)r r )r r r r r � s zNestedWith.Dummy.__enter__c G s d| _ || _| jrdS d S )NT)r r) r� )r r) r r r r � s zNestedWith.Dummy.__exit__)NF)r r r r r r r r r r �Dummy� s
r� c @ s e Zd Zdd� ZdS )zNestedWith.InitRaisesc C s
t � �d S )N)r% )r r r r r � r� z NestedWith.InitRaises.__init__N)r r r r r r r r �
InitRaises� s r� c @ s e Zd Zdd� Zdd� ZdS )zNestedWith.EnterRaisesc C s
t � �d S )N)r% )r r r r r � r� z NestedWith.EnterRaises.__enter__c G s d S )Nr )r r) r r r r � r� z NestedWith.EnterRaises.__exit__N)r r r r r r r r r � EnterRaises� s r� c @ s e Zd Zdd� Zdd� ZdS )zNestedWith.ExitRaisesc C s d S )Nr )r r r r r � r� z NestedWith.ExitRaises.__enter__c G s
t � �d S )N)r% )r r) r r r r � r� z NestedWith.ExitRaises.__exit__N)r r r r r r r r r �
ExitRaises� s r� c
C s\ | � � �2}| � � � }| �|j� | �|j� W d Q R X W d Q R X | �|j� | �|j� d S )N)r� rU r r )r r� r� r r r �testNoExceptions� s
zNestedWith.testNoExceptionsc
C sV y,| � � �}| �� � W d Q R X W d Q R X W n Y nX | �|j� | �|j� d S )N)r� r� rU r r )r r� r r r �testExceptionInExprList� s z"NestedWith.testExceptionInExprListc
C sr y6| � � �$}| �� � | �d� W d Q R X W d Q R X W n tk
rJ Y n X | �d� | �|j� | �|j� d S )Nzbody of bad with executedzRuntimeError not reraised)r� r� r� r% rU r r )r r� r r r �testExceptionInEnter� s "
z NestedWith.testExceptionInEnterc
C sl d}| j dd�� }| �� �
d}W d Q R X W d Q R X | �|j� | �|j� | �|� | �|jd d � d S )NFT)r� r )r� r� rU r r ZassertNotEqualr) )r Z
body_executedr� r r r �testExceptionInExit� s
z NestedWith.testExceptionInExitc
C sl | j dd��V\}}| j dd��:\}}| �d|� | �d|� | �d|� | �d|� W d Q R X W d Q R X d S )N)r, r- )r )r� � r, r- r� r� )r� rM )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__) �__doc__�
__author__Z __email__r( Zunittest� collectionsr �
contextlibr r r r r8 r r! r" r/ ZTestCaser0 rR rb rm r{ r� r� r� r� r �mainr r r r �<module> s2
$S/5A D9&L