|
Server : LiteSpeed System : Linux server51.dnsbootclub.com 4.18.0-553.62.1.lve.el8.x86_64 #1 SMP Mon Jul 21 17:50:35 UTC 2025 x86_64 User : nandedex ( 1060) PHP Version : 8.1.33 Disable Function : NONE Directory : /opt/cppython/lib/python3.8/test/__pycache__/ |
U
>��g*� � @ sV d Z ddlZddlZddlZddlZddlZddlT ddlmZ ddl Z G dd� dej
�Z G dd� dej
�Z G d d
� d
ej
�Z
G d d � d ej
�ZG d
d� dej
�ZG dd� dej
�ZG dd� de�ZG dd� dej
�ZG dd� d�ZG dd� deej
�ZG dd� d�ZG dd � d eej
�ZG d d � d eej
�ZG d d � d ej
�Zed!k�rRe�� dS )"z9Unit tests for contextlib.py, and other context managers.� N)�*)�supportc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestAbstractContextManagerc C s* G dd� dt �}|� }| �|�� |� d S )Nc s e Zd Z� fdd�Z� ZS )z;TestAbstractContextManager.test_enter.<locals>.DefaultEnterc s t � j|� d S �N��super�__exit__��self�args�� __class__� �3/opt/cppython/lib/python3.8/test/test_contextlib.pyr s zDTestAbstractContextManager.test_enter.<locals>.DefaultEnter.__exit__��__name__�
__module__� __qualname__r �
__classcell__r r r r � DefaultEnter s r )�AbstractContextManager�assertIs� __enter__)r
r �managerr r r �
test_enter s z%TestAbstractContextManager.test_enterc C s0 G dd� dt �}| �t�� |� W 5 Q R X d S )Nc @ s e Zd ZdS )zETestAbstractContextManager.test_exit_is_abstract.<locals>.MissingExitN�r r r r r r r � MissingExit s r )r � assertRaises� TypeError)r
r r r r �test_exit_is_abstract s z0TestAbstractContextManager.test_exit_is_abstractc C s� G dd� d�}| � t|t�� G dd� dt�}| � t|t�� G dd� d|�}| �t|t�� G dd� d|�}| �t|t�� d S ) Nc @ s e Zd Zdd� Zdd� ZdS )zRTestAbstractContextManager.test_structural_subclassing.<locals>.ManagerFromScratchc S s | S r r �r
r r r r s z\TestAbstractContextManager.test_structural_subclassing.<locals>.ManagerFromScratch.__enter__c S s d S r r )r
�exc_type� exc_value� tracebackr r r r " s z[TestAbstractContextManager.test_structural_subclassing.<locals>.ManagerFromScratch.__exit__N�r r r r r r r r r �ManagerFromScratch s r% c s e Zd Z� fdd�Z� ZS )zLTestAbstractContextManager.test_structural_subclassing.<locals>.DefaultEnterc s t � j|� d S r r r r r r r ( s zUTestAbstractContextManager.test_structural_subclassing.<locals>.DefaultEnter.__exit__r r r r r r ' s r c @ s e Zd ZdZdS )zGTestAbstractContextManager.test_structural_subclassing.<locals>.NoEnterN)r r r r r r r r �NoEnter- s r& c @ s e Zd ZdZdS )zFTestAbstractContextManager.test_structural_subclassing.<locals>.NoExitN)r r r r r r r r �NoExit2 s r' )�
assertTrue�
issubclassr � assertFalse)r
r% r r& r' r r r �test_structural_subclassing s z6TestAbstractContextManager.test_structural_subclassingN)r r r r r r+ r r r r r
s r c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z e
jdd� �Ze
jdd� �Zdd� Zdd � Zd d � Zd d � Zd!S )"�ContextManagerTestCasec s` g � t � fdd��}|� �*}| �� dg� | �|d� � �|� W 5 Q R X | �� dddg� d S )Nc 3 s � � d� dV � � d� d S )N� �* �� ��appendr ��stater r �woohoo< s
z@ContextManagerTestCase.test_contextmanager_plain.<locals>.woohoor- r. r/ )�contextmanager� assertEqualr1 �r
r4 �xr r2 r �test_contextmanager_plain: s z0ContextManagerTestCase.test_contextmanager_plainc
s| g � t � fdd��}| �t��B |� �0}| �� dg� | �|d� � �|� t� �W 5 Q R X W 5 Q R X | �� dddg� d S )Nc 3 s& � � d� z
dV W 5 � � d� X d S )Nr- r/ r. r0 r r2 r r r4 I s
zBContextManagerTestCase.test_contextmanager_finally.<locals>.woohoor- r. r/ )r5 r �ZeroDivisionErrorr6 r1 r7 r r2 r �test_contextmanager_finallyG s
z2ContextManagerTestCase.test_contextmanager_finallyc C s6 t dd� �}|� }|�� | �|�ttd�d �� d S )Nc s s
d V d S r r r r r r �wheeY s zCContextManagerTestCase.test_contextmanager_no_reraise.<locals>.whee�foo)r5 r r* r r )r
r< �ctxr r r � test_contextmanager_no_reraiseX s
z5ContextManagerTestCase.test_contextmanager_no_reraisec C s6 t dd� �}|� }|�� | �t|jttd�d � d S )Nc s s" z
d V W n d V Y nX d S r r r r r r �whoob s
zOContextManagerTestCase.test_contextmanager_trap_yield_after_throw.<locals>.whoor= )r5 r r � RuntimeErrorr r )r
r@ r> r r r �*test_contextmanager_trap_yield_after_throwa s
�zAContextManagerTestCase.test_contextmanager_trap_yield_after_throwc sj g �t � �fdd��}|� �2}� ��dg� � �|d� ��|� td��W 5 Q R X � ��dddg� d S )Nc
3 s^ �� d� z
dV W nD tk
rX } z&�� | jd � � ��dddg� W 5 d } ~ X Y nX d S )Nr- r. r r/ )r1 r: r r6 )�e�r
r3 r r r4 p s
zAContextManagerTestCase.test_contextmanager_except.<locals>.woohoor- r. r/ )r5 r6 r1 r: r7 r rD r �test_contextmanager_exceptn s
z1ContextManagerTestCase.test_contextmanager_exceptc
C s� t d�}tdd� �}z2| �td�� |� �
|�W 5 Q R X W 5 Q R X W n. tk
rt } z| �||� W 5 d }~X Y n X | �d� d S )N�spamc s s
d V d S r r r r r r r4 � s zJContextManagerTestCase.test_contextmanager_except_stopiter.<locals>.woohoo�
StopIteration� StopIteration was suppressed)rG r5 ZassertWarnsRegex�DeprecationWarning� Exceptionr �fail)r
�stop_excr4 �exr r r �#test_contextmanager_except_stopiter s
� z:ContextManagerTestCase.test_contextmanager_except_stopiterc
C s| d}i }t |||� |d }td�}z|� �
|�W 5 Q R X W n. tk
rl } z| �||� W 5 d }~X Y n X | �d� d S )Nztfrom __future__ import generator_stop
from contextlib import contextmanager
@contextmanager
def woohoo():
yield
r4 rF rH )�execrG rJ r rK )r
�code�localsr4 rL rM r r r �!test_contextmanager_except_pep479� s z8ContextManagerTestCase.test_contextmanager_except_pep479c
C s� t dd� �}z|� �
t�W 5 Q R X W nR tk
rx } z4| �t|�t� | �|jd d� | �|j t� W 5 d }~X Y nX z |� � t
d��W 5 Q R X W nP tk
r� } z2| �t|�t
� | �|jd d� | � |j � W 5 d }~X Y nX d S )Nc
s s< z
d V W n, t k
r6 } ztd�| �W 5 d } ~ X Y nX d S )N�issue29692:Chained)rJ rA ��excr r r �test_issue29692� s
zoContextManagerTestCase.test_contextmanager_do_not_unchain_non_stopiteration_exceptions.<locals>.test_issue29692r rS zissue29692:Unchained) r5 r: rJ r �typerA r6 r �assertIsInstance� __cause__rG � assertIsNone)r
rV rM r r r �?test_contextmanager_do_not_unchain_non_stopiteration_exceptions� s
zVContextManagerTestCase.test_contextmanager_do_not_unchain_non_stopiteration_exceptionsc C s" dd� }t |dd�dd� ��}|S )Nc s � fdd�}|S )Nc s" � � � D ]\}}t| ||� q| S r )�items�setattr)�func�k�v��kwr r �decorate� s zXContextManagerTestCase._create_contextmanager_attribs.<locals>.attribs.<locals>.decorater )rb rc r ra r �attribs� s zFContextManagerTestCase._create_contextmanager_attribs.<locals>.attribs�bar)r= c S s dS )�Whee!Nr )rF r r r �baz� s zBContextManagerTestCase._create_contextmanager_attribs.<locals>.baz�r5 )r
rd rg r r r � _create_contextmanager_attribs� s
z5ContextManagerTestCase._create_contextmanager_attribsc C s( | � � }| �|jd� | �|jd� d S )Nrg re )ri r6 r r= �r
rg r r r �test_contextmanager_attribs� s z2ContextManagerTestCase.test_contextmanager_attribsc C s | � � }| �|jd� d S �Nrf �ri r6 �__doc__rj r r r � test_contextmanager_doc_attrib� s z5ContextManagerTestCase.test_contextmanager_doc_attribc C s | � � d �}| �|jd� d S rl rm rj r r r �*test_instance_docstring_given_cm_docstring� s zAContextManagerTestCase.test_instance_docstring_given_cm_docstringc C s8 t dd� �}|ddddd��}| �|d� W 5 Q R X d S ) Nc s s | |||fV d S r r �r
r^ r �kwdsr r r r4 � s z4ContextManagerTestCase.test_keywords.<locals>.woohoo� � �! �, rq )rs rt ru rv �r5 r6 )r
r4 �targetr r r �
test_keywords� s
z$ContextManagerTestCase.test_keywordsc s> G dd� d�}t � fdd��}||� |� d�� W 5 Q R X d S )Nc @ s e Zd ZdS )z0ContextManagerTestCase.test_nokeepref.<locals>.ANr r r r r �A� s rz c 3 s6 t �| �} t �|�}� �| � � � �|� � d V d S r )�weakref�refrZ ��a�br r r r4 � s
z5ContextManagerTestCase.test_nokeepref.<locals>.woohoo�r rh )r
rz r4 r r r �test_nokeepref� s
z%ContextManagerTestCase.test_nokeeprefc C sl t dd� �}| �t�� |� W 5 Q R X | �t�� |dd� W 5 Q R X | �t�� |dd� W 5 Q R X d S )Nc s s
d V d S r r r} r r r r4 � s z8ContextManagerTestCase.test_param_errors.<locals>.woohoo� � r� )r5 r r �r
r4 r r r �test_param_errors� s
z(ContextManagerTestCase.test_param_errorsc s@ d� t � �fdd��}|� � �fdd����� ��� d� d S )Nr c 3 s* � } � d7 � d V � d8 � �� � | � d S �Nr- )r6 )�before)�depthr
r r r4 � s
z5ContextManagerTestCase.test_recursive.<locals>.woohooc s � dk r�� d S )N�
r r )r� � recursiver r r� s z8ContextManagerTestCase.test_recursive.<locals>.recursiverw r� r )r� r� r
r �test_recursive� s z%ContextManagerTestCase.test_recursiveN)r r r r9 r; r? rB rE rN rR r[ ri rk r �requires_docstringsro rp ry r� r� r� r r r r r, 8 s$
r, c @ s* e Zd Zejdd� �Zdd� Zdd� ZdS )�ClosingTestCasec C s t j}t d �}| �|j|� d S r )�closingrn r6 �r
Z cm_docstring�objr r r �test_instance_docs s z"ClosingTestCase.test_instance_docsc sZ g � G � fdd�d�}|� }| � � g � t|��}| � ||� W 5 Q R X | � � dg� d S )Nc s e Zd Z� fdd�ZdS )z'ClosingTestCase.test_closing.<locals>.Cc s � � d� d S r� r0 r r2 r r �close s z-ClosingTestCase.test_closing.<locals>.C.closeN�r r r r� r r2 r r �C s r� r- )r6 r� �r
r� r8 �yr r2 r � test_closing s
z ClosingTestCase.test_closingc
sx g � G � fdd�d�}|� }| � � g � | �t��. t|��}| � ||� dd W 5 Q R X W 5 Q R X | � � dg� d S )Nc s e Zd Z� fdd�ZdS )z-ClosingTestCase.test_closing_error.<locals>.Cc s � � d� d S r� r0 r r2 r r r� $ s z3ClosingTestCase.test_closing_error.<locals>.C.closeNr� r r2 r r r� # s r� r- r )r6 r r: r� r� r r2 r �test_closing_error! s
z"ClosingTestCase.test_closing_errorN)r r r r r� r� r� r� r r r r r�
s
r� c @ s e Zd Zdd� ZdS )�NullcontextTestCasec C s8 G dd� d�}|� }t |��}| �||� W 5 Q R X d S )Nc @ s e Zd ZdS )z/NullcontextTestCase.test_nullcontext.<locals>.CNr r r r r r� 1 s r� )� nullcontextr )r
r� �cZc_inr r r �test_nullcontext0 s
z$NullcontextTestCase.test_nullcontextN)r r r r� r r r r r� / s r� c @ s e Zd Zdd� ZdS )�FileContextTestCasec
C s� t �� }z�d }t|d�� }| �|j� |�d� W 5 Q R X | �|j� d }| � t
��@ t|d��*}| �|j� | � |� � d� dd W 5 Q R X W 5 Q R X | �|j� W 5 t�|� X d S )N�wzBooh
�rr- r )
�tempfile�mktempr �unlink�openr* �closed�writer( r r: r6 �read)r
Ztfn�fr r r � testWithOpen: s z FileContextTestCase.testWithOpenN)r r r r� r r r r r� 8 s r� c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�LockContextTestCasec
C s~ | � |� � |� | �|� � W 5 Q R X | � |� � | �t��* |� | �|� � dd W 5 Q R X W 5 Q R X | � |� � d S �Nr- r )r* r( r r: )r
�lock�lockedr r r � boilerPlateN s z LockContextTestCase.boilerPlatec C s t �� }| �||j� d S r )� threading�Lockr� r� �r
r� r r r � testWithLockY s z LockContextTestCase.testWithLockc C s t �� }| �||j� d S r )r� �RLockr� � _is_ownedr� r r r �
testWithRLock] s z!LockContextTestCase.testWithRLockc s$ t �� � � fdd�}| �� |� d S )Nc s � � � S r )r� r �r� r r r� c s z5LockContextTestCase.testWithCondition.<locals>.locked)r� � Conditionr� �r
r� r r� r �testWithConditiona s z%LockContextTestCase.testWithConditionc s$ t �� � � fdd�}| �� |� d S )Nc s � � d�r� �� dS dS d S �NFT��acquire�releaser r� r r r� i s
z5LockContextTestCase.testWithSemaphore.<locals>.locked)r� � Semaphorer� r� r r� r �testWithSemaphoreg s z%LockContextTestCase.testWithSemaphorec s$ t �� � � fdd�}| �� |� d S )Nc s � � d�r� �� dS dS d S r� r� r r� r r r� s s
z<LockContextTestCase.testWithBoundedSemaphore.<locals>.locked)r� �BoundedSemaphorer� r� r r� r �testWithBoundedSemaphoreq s z,LockContextTestCase.testWithBoundedSemaphoreN) r r r r� r� r� r� r� r� r r r r r� L s
r� c @ s, e Zd ZdZdZdZdZdd� Zdd� ZdS )� mycontextz9Example decoration-compatible context manager for testingFNc C s
d| _ | S �NT��startedr r r r r � s zmycontext.__enter__c G s || _ | jS r )rU �catch�r
rU r r r r � s zmycontext.__exit__) r r r rn r� rU r� r r r r r r r� | s r� c @ sb e Zd Zejdd� �Zdd� Zdd� Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
dd� ZdS )�TestContextDecoratorc C s t j}t � }| �|j|� d S r )r� rn r6 r� r r r r� � s z'TestContextDecorator.test_instance_docsc C s@ t � }|� }| �||� | �|j� W 5 Q R X | �|jd� d S )N�NNN)r� r r( r� r6 rU )r
�context�resultr r r �test_contextdecorator� s
z*TestContextDecorator.test_contextdecoratorc
C s� t � }| �td�� |� td��W 5 Q R X W 5 Q R X | �|j� | �|jd t� t � }d|_|� td��W 5 Q R X | �|j� | �|jd t� d S )Nr= r T)r� �assertRaisesRegex� NameError�assertIsNotNonerU r r� )r
r� r r r �$test_contextdecorator_with_exception� s z9TestContextDecorator.test_contextdecorator_with_exceptionc s0 t � � � � �fdd��}|� ��� jd� d S )Nc s �� � j� ��� j� d S r �rZ rU r( r� r �r� r
r r �test� s z1TestContextDecorator.test_decorator.<locals>.testr� )r� r6 rU �r
r� r r� r �test_decorator� s
z#TestContextDecorator.test_decoratorc sX t � � � � �fdd��}��td�� |� W 5 Q R X ��� j� ��� jd t� d S )Nc s$ �� � j� ��� j� td��d S )Nr= )rZ rU r( r� r� r r� r r r� � s z@TestContextDecorator.test_decorator_with_exception.<locals>.testr= r )r� r� r� r� rU r r� r r� r � test_decorator_with_exception� s z2TestContextDecorator.test_decorator_with_exceptionc s� t � � G � fdd�dt�}|� }|�dd� | �|jd� | �|jd� | �|jd � |� }|�ddd� | �|jd� | �|jd� | �|jd� |� }|jddd� | �|jd� | �|jd� d S ) Nc s e Zd Z� ddd��ZdS )z9TestContextDecorator.test_decorating_method.<locals>.TestNc S s || _ || _|| _d S r )r~ r r� )r
r~ r r� r r r �method� s z@TestContextDecorator.test_decorating_method.<locals>.Test.method)N)r r r r� r �r� r r �Test� s r� r- � r~ r r� r} )r� �objectr� r6 r~ r r� )r
r� r� r r� r �test_decorating_method� s z+TestContextDecorator.test_decorating_methodc
C s< G dd� dt �}| �t�� |� � W 5 Q R X W 5 Q R X d S )Nc @ s e Zd Zdd� Zdd� ZdS )z7TestContextDecorator.test_typo_enter.<locals>.mycontextc S s d S r r r r r r � __unter__� s zATestContextDecorator.test_typo_enter.<locals>.mycontext.__unter__c W s d S r r r� r r r r � s z@TestContextDecorator.test_typo_enter.<locals>.mycontext.__exit__N)r r r r� r r r r r r� � s r� ��ContextDecoratorr �AttributeError�r
r� r r r �test_typo_enter� s z$TestContextDecorator.test_typo_enterc
C s< G dd� dt �}| �t�� |� � W 5 Q R X W 5 Q R X d S )Nc @ s e Zd Zdd� Zdd� ZdS )z6TestContextDecorator.test_typo_exit.<locals>.mycontextc S s d S r r r r r r r � s z@TestContextDecorator.test_typo_exit.<locals>.mycontext.__enter__c W s d S r r r� r r r �__uxit__� s z?TestContextDecorator.test_typo_exit.<locals>.mycontext.__uxit__N)r r r r r� r r r r r� � s r� r� r� r r r �test_typo_exit� s z#TestContextDecorator.test_typo_exitc sR G dd� dt �}G dd� d|t�}|� � � � �fdd��}|� ��� jd� d S )Nc @ s$ e Zd ZdZdZdd� Zdd� ZdS )zHTestContextDecorator.test_contextdecorator_as_mixin.<locals>.somecontextFNc S s
d| _ | S r� r� r r r r r s zRTestContextDecorator.test_contextdecorator_as_mixin.<locals>.somecontext.__enter__c W s
|| _ d S r rT r� r r r r s zQTestContextDecorator.test_contextdecorator_as_mixin.<locals>.somecontext.__exit__)r r r r� rU r r r r r r � somecontext� s r� c @ s e Zd ZdS )zFTestContextDecorator.test_contextdecorator_as_mixin.<locals>.mycontextNr r r r r r�
s r� c s �� � j� ��� j� d S r r� r r� r r r� s zATestContextDecorator.test_contextdecorator_as_mixin.<locals>.testr� )r� r� r6 rU )r
r� r� r� r r� r � test_contextdecorator_as_mixin� s z3TestContextDecorator.test_contextdecorator_as_mixinc sf t �fdd��}g �|d�� �fdd��}|d� � ��dddg� g �|d� � ��dddg� d S ) Nc 3 s � � | � d V � � d� d S )Nr/ r0 )r� r2 r r r4 s
zETestContextDecorator.test_contextmanager_as_decorator.<locals>.woohoor- c s � � �dg� ��| � d S r� )r6 r1 )r8 rD r r r� s zCTestContextDecorator.test_contextmanager_as_decorator.<locals>.testZ somethingr/ zsomething elserw )r
r4 r� r rD r � test_contextmanager_as_decorator s z5TestContextDecorator.test_contextmanager_as_decoratorN)r r r r r� r� r� r� r� r� r� r� r� r� r� r r r r r� � s
r� c @ s� e Zd ZdZejdd� �Zdd� Zdd� Zdd � Z d
d � Z
d d
� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� ZdS )(�TestBaseExitStackNc C s" | j j}| � � }| �|j|� d S r )�
exit_stackrn r6 r� r r r r� . s z$TestBaseExitStack.test_instance_docsc C s | � � � W 5 Q R X d S r )r� r r r r �test_no_resources5 s
z#TestBaseExitStack.test_no_resourcesc s� di fdi fdi fdt dd�fdt dd�fdt dd�fdt ddd�fg}g � � fd d
�}| �� ��}t|�D ]d\}}|r�|r�|j|f|�|�}n4|r�|j|f|�� }n |r�|j|f|�}n
|�|�}| �||� qj|jD ]>}| �|d j|� | �|d j|j� | � |d j
|j
� q�W 5 Q R X | � � |� g � | �� �p}| � t
�� |jdd � W 5 Q R X | � t
�� | jjd d � W 5 Q R X | �t�� |j|dd
� W 5 Q R X W 5 Q R X | � � dddifg� d S )Nr )r- )r- r� r- )Zexampler� � )r
�callbackc s � � | |f� dS )zTest metadata propagationNr0 )r rr �r� r r �_exitD s z.TestBaseExitStack.test_callback.<locals>._exit)�argr� )r� r� r� )�dictr� �reversedr� r �_exit_callbacks� __wrapped__ZassertNotEqualr rZ rn r6 r r Z assertWarnsrI )r
Zexpectedr� �stackr rr r� �wrapperr r� r �
test_callback9 sD �
"z TestBaseExitStack.test_callbackc s* t � � �fdd�}dd� }�fdd�}G dd� dt�}��� ��}|�|� ��|jd d
|� ||�}|�|� ��|jd d
j|� |�|� ��|jd d
|� ||�}|�|� ��|jd d
j|� |�|� ��|jd d
|� |�|� ��|jd d
|� d
d W 5 Q R X d S ) Nc s �� | � � d S r )r �r! rU �exc_tb�Z
exc_raisedr
r r � _expect_excd s z0TestBaseExitStack.test_push.<locals>._expect_excc W s dS r� r �� exc_detailsr r r �
_suppress_excf s z2TestBaseExitStack.test_push.<locals>._suppress_excc s" � � | � � � |� � � |� d S r )rZ r� r r r �
_expect_okh s
z/TestBaseExitStack.test_push.<locals>._expect_okc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )z+TestBaseExitStack.test_push.<locals>.ExitCMc S s
|| _ d S r �� check_exc)r
r� r r r �__init__m s z4TestBaseExitStack.test_push.<locals>.ExitCM.__init__c S s | � d� d S )NzShould not be called!)rK r r r r r o s z5TestBaseExitStack.test_push.<locals>.ExitCM.__enter__c W s | j |� d S r r� �r
r� r r r r q s z4TestBaseExitStack.test_push.<locals>.ExitCM.__exit__N�r r r r r r r r r r �ExitCMl s r ���r- r )r: r� r� �pushr r� �__self__)r
r� r� r� r r� �cmr r� r � test_pushb s*
zTestBaseExitStack.test_pushc s� G � fdd�dt �}g � |� }| �� �N}|j� fdd��}| �|� |�|� | �|jd d j|� � �d� W 5 Q R X | � � dddd g� d S )
Nc s$ e Zd Z� fdd�Z� fdd�ZdS )z4TestBaseExitStack.test_enter_context.<locals>.TestCMc s � � d� d S r� r0 r r� r r r � s z>TestBaseExitStack.test_enter_context.<locals>.TestCM.__enter__c s � � d� d S �Nr� r0 r r� r r r � s z=TestBaseExitStack.test_enter_context.<locals>.TestCM.__exit__Nr$ r r� r r �TestCM� s r
c s � � d� d S )Nr� r0 r r� r r r� � s z3TestBaseExitStack.test_enter_context.<locals>._exitr r- r� r� r� )
r� r� r� r� �
enter_contextr r� r r1 r6 )r
r
r r� r� r r� r �test_enter_context� s
z$TestBaseExitStack.test_enter_contextc sZ g � | � � �4}|j� fdd��}| �|� |�� � �d� W 5 Q R X | �� ddg� d S )Nc s � � d� d S r� r0 r r� r r r� � s z+TestBaseExitStack.test_close.<locals>._exitr� r- )r� r� r� r� r1 r6 )r
r� r� r r� r �
test_close� s
z TestBaseExitStack.test_closec sn g � | � � �4}|j� fdd��}| �|� |�� }� �d� W 5 Q R X � �d� |�� | �� dddg� d S )Nc s � � d� d S r r0 r r� r r r� � s z-TestBaseExitStack.test_pop_all.<locals>._exitr- r� r� )r� r� r� �pop_allr1 r� r6 )r
r� r� � new_stackr r� r � test_pop_all� s
z TestBaseExitStack.test_pop_allc
C sD | � t��0 | �� � }|�dd� � dd W 5 Q R X W 5 Q R X d S )Nc W s dS )NFr rT r r r �<lambda>� � z3TestBaseExitStack.test_exit_raise.<locals>.<lambda>r- r )r r: r� r �r
r� r r r �test_exit_raise� s
z!TestBaseExitStack.test_exit_raisec C s. | � � � }|�dd� � dd W 5 Q R X d S )Nc W s dS r� r rT r r r r � r z6TestBaseExitStack.test_exit_suppress.<locals>.<lambda>r- r �r� r r r r r �test_exit_suppress� s
z$TestBaseExitStack.test_exit_suppressc C s
G dd� d�}G dd� d�}G dd� d�}z\|t ��J |tt��4 |� �" |t�� dd W 5 Q R X W 5 Q R X W 5 Q R X W 5 Q R X W nP t k
r� } z2| �|jt� | �|jjt� | �|jjj� W 5 d }~X Y n X | �d � |jd }| �|t� | �|jt � d S )
Nc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )zJTestBaseExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExcc S s
|| _ d S r rT r� r r r r � s zSTestBaseExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExc.__init__c S s | S r r r r r r r � s zTTestBaseExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExc.__enter__c W s
| j �d S r rT r r r r r � s zSTestBaseExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExc.__exit__Nr r r r r �RaiseExc� s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )zUTestBaseExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExcWithContextc S s || _ || _d S r )�outer�inner)r
r r r r r r � s z^TestBaseExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExcWithContext.__init__c S s | S r r r r r r r � s z_TestBaseExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExcWithContext.__enter__c W s" z
| j �W n | j�Y nX d S r )r r r r r r r � s
z^TestBaseExitStack.test_exit_exception_chaining_reference.<locals>.RaiseExcWithContext.__exit__Nr r r r r �RaiseExcWithContext� s r c @ s e Zd Zdd� Zdd� ZdS )zMTestBaseExitStack.test_exit_exception_chaining_reference.<locals>.SuppressExcc S s | S r r r r r r r � s zWTestBaseExitStack.test_exit_exception_chaining_reference.<locals>.SuppressExc.__enter__c W s |t | �_dS r� )rW �
saved_detailsr r r r r � s
zVTestBaseExitStack.test_exit_exception_chaining_reference.<locals>.SuppressExc.__exit__Nr$ r r r r � SuppressExc� s r r- r �0Expected IndexError, but no exception was raised)
�
IndexError�KeyErrorr� �
ValueErrorrX � __context__rZ rK r r: )r
r r r rU � inner_excr r r �&test_exit_exception_chaining_reference� s"
4"
z8TestBaseExitStack.test_exit_exception_chaining_referencec
s� dd� }d � � fdd�}zZ| � � �H}|�|t� |�|t� |�|t� |�|� |�|t� dd W 5 Q R X W nP tk
r� } z2| �|jt� | �|jjt� | � |jjj� W 5 d }~X Y n X | �
d� � d }| �|t� | �|jt � d S )Nc S s | �d S r r rT r r r � raise_exc� s zATestBaseExitStack.test_exit_exception_chaining.<locals>.raise_excc s | � dS r� r r� �r r r � suppress_exc� s zDTestBaseExitStack.test_exit_exception_chaining.<locals>.suppress_excr- r r ) r� r� r r r� r r rX r! rZ rK r: )r
r$ r& r� rU r" r r% r � test_exit_exception_chaining� s&
"
z.TestBaseExitStack.test_exit_exception_chainingc
C s� dd� }dd� }z2| � � � }|�dd� � |�|t� W 5 Q R X W n. tk
rp } z| �|t� W 5 d }~X Y n X | �d� z:| � � �(}|�|t� |�|� |�|t� W 5 Q R X W n. tk
r� } z| �|t� W 5 d }~X Y n X | �d� d S ) Nc S s | �d S r r rT r r r r$ s zHTestBaseExitStack.test_exit_exception_non_suppressing.<locals>.raise_excc W s dS r� r r� r r r r& s zKTestBaseExitStack.test_exit_exception_non_suppressing.<locals>.suppress_excc S s d S r r r r r r r r zGTestBaseExitStack.test_exit_exception_non_suppressing.<locals>.<lambda>r z.Expected KeyError, but no exception was raised)r� r� r rJ rX rK r r )r
r$ r& r� rU r r r �#test_exit_exception_non_suppressing s"
z5TestBaseExitStack.test_exit_exception_non_suppressingc
C s� t dd� �}td�}td�}td�}td�}zF| �� �4}|�||�� |�||�� |�||�� |�W 5 Q R X W np tk
r� } zR| �||� | �|j|� | �|jj|� | �|jjj|� | �|jjjj� W 5 d }~X Y nX d S )Nc s s z
d V W 5 | �X d S r r rT r r r �gets_the_context_right& s
zZTestBaseExitStack.test_exit_exception_with_correct_context.<locals>.gets_the_context_rightr- r� r� r� )r5 rJ r� r r r! rZ )r
r) �exc1�exc2�exc3�exc4r� rU r r r �(test_exit_exception_with_correct_context$ s(
�z:TestBaseExitStack.test_exit_exception_with_correct_contextc
C s� dd� }t d�}t d�}t d�}t d�}t d�}z8| �� �&}|�|||� |�|||� |�W 5 Q R X W n� t k
r� } zh| �||� | �|j|� | �|jj|� | �|jjj|� | �|jjjj|� | �|jjjjj� W 5 d }~X Y nX d S )Nc S s z| �W 5 |�X d S r r )r" Z outer_excr r r � raise_nestedF s zQTestBaseExitStack.test_exit_exception_with_existing_context.<locals>.raise_nestedr- r� r� r� r� )rJ r� r� r r! rZ ) r
r/ r* r+ r, r- Zexc5r� rU r r r �)test_exit_exception_with_existing_contextC s.
� �z;TestBaseExitStack.test_exit_exception_with_existing_contextc
C sd dd� }z*| � � �}|�|� dd W 5 Q R X W n, tk
r^ } z| �d� W 5 d }~X Y nX d S )Nc W s dS r� r r� r r r r& ` s zDTestBaseExitStack.test_body_exception_suppress.<locals>.suppress_excr- r z%Expected no exception, got IndexError)r� r r rK )r
r& r� rU r r r � test_body_exception_suppress_ s
z.TestBaseExitStack.test_body_exception_suppressc C sB | � � �0}|�dd� � |�dd� � |�dd� � W 5 Q R X d S )Nc W s dS r� r rT r r r r k r zITestBaseExitStack.test_exit_exception_chaining_suppress.<locals>.<lambda>c W s dd S r� r rT r r r r l r c W s i d S r� r rT r r r r m r r r r r r �%test_exit_exception_chaining_suppressi s
z7TestBaseExitStack.test_exit_exception_chaining_suppressc C s0 | � � � }td�D ]}|�t� qW 5 Q R X d S )Ni' )r� �ranger� �int)r
r� �ir r r �test_excessive_nestingo s
z(TestBaseExitStack.test_excessive_nestingc C sZ G dd� dt �}|� }t � |_| �� }| �t|j|� |�|� | �|jd d |� d S )Nc @ s e Zd ZdS )z7TestBaseExitStack.test_instance_bypass.<locals>.ExampleNr r r r r �Examplev s r7 r r- ) r� r r� r r� r r r r� )r
r7 r r� r r r �test_instance_bypassu s
z&TestBaseExitStack.test_instance_bypassc
s� G dd� dt �� G dd� dt�}t� fdd��}tdd� �}| �� ��:}| �� �&}|�|� � |�|� � |d ��W 5 Q R X W 5 Q R X |j}| �|� � | �|j|� | � |jj� | � |jj
� | � |j
|j� d S )
Nc @ s e Zd ZdS )zITestBaseExitStack.test_dont_reraise_RuntimeError.<locals>.UniqueExceptionNr r r r r �UniqueException� s r9 c @ s e Zd ZdS )zLTestBaseExitStack.test_dont_reraise_RuntimeError.<locals>.UniqueRuntimeErrorNr r r r r �UniqueRuntimeError� s r: c
3 s< z
dV W n, t k
r6 } z� d�| �W 5 d } ~ X Y nX d S )Nr- z
new exception�rJ rT �r9 r r �second� s
z@TestBaseExitStack.test_dont_reraise_RuntimeError.<locals>.secondc
s s6 z
dV W n& t k
r0 } z| �W 5 d } ~ X Y nX d S r� r; rT r r r �first� s
z?TestBaseExitStack.test_dont_reraise_RuntimeError.<locals>.firstzplease no infinite loop.) rJ rA r5 r r� r � exceptionrX r! rZ rY r )r
r: r= r> Zerr_ctxZes_ctxrU r r< r � test_dont_reraise_RuntimeError~ s"
z0TestBaseExitStack.test_dont_reraise_RuntimeError)r r r r� r r� r� r� r� r r r
r r r r# r' r( r. r0 r1 r2 r6 r8 r@ r r r r r� + s*
)"
0
r� c @ s e Zd ZeZdS )�
TestExitStackN)r r r � ExitStackr� r r r r rA � s rA c @ sJ e Zd ZdZdZejdd� �Zdd� Zdd� Z dd � Z
d
d � Z d d
� Z dS )�TestRedirectStreamNc C s$ | j j}| � d �}| �|j|� d S r )�redirect_streamrn r6 r� r r r r� � s
z%TestRedirectStream.test_instance_docsc C s. t t| j�}| �d � | �t t| j�|� d S r )�getattr�sys� orig_streamrD r )r
� orig_stdoutr r r �test_no_redirect_in_init� s
z+TestRedirectStream.test_no_redirect_in_initc C sr t �� }d}tt| j�}| �|�� t|tt| j�d� W 5 Q R X | �tt| j�|� |�� � � }| �
||� d S )Nz<Consider an API like help(), which prints directly to stdout��file) �io�StringIOrE rF rG rD �printr �getvalue�stripr6 )r
r� �msgrH �sr r r �test_redirect_to_string_io� s z-TestRedirectStream.test_redirect_to_string_ioc C s. t �� }| �|��}| �||� W 5 Q R X d S r )rL rM rD r )r
r� � enter_resultr r r �test_enter_result_is_target� s z.TestRedirectStream.test_enter_result_is_targetc C s� t �� }| �|�}tt| j�}|� tddtt| j�d� W 5 Q R X |� tdtt| j�d� W 5 Q R X | �tt| j�|� |�� }| � |d� d S �N�Hello� )�endrK zWorld!rJ z
Hello World!
�
rL rM rD rE rF rG rN r rO r6 �r
r� Z
write_to_frH rR r r r �test_cm_is_reusable� s
z&TestRedirectStream.test_cm_is_reusablec
C s� t �� }| �|�}tt| j�}|�@ tddtt| j�d� |� tdtt| j�d� W 5 Q R X W 5 Q R X | �tt| j�|� |�� }| � |d� d S rV rZ r[ r r r �test_cm_is_reentrant� s
(z'TestRedirectStream.test_cm_is_reentrant)
r r r rD rG r r� r� rI rS rU r\ r] r r r r rC � s
rC c @ s e Zd ZeZdZdS )�TestRedirectStdout�stdoutN)r r r �redirect_stdoutrD rG r r r r r^ � s r^ c @ s e Zd ZeZdZdS )�TestRedirectStderr�stderrN)r r r �redirect_stderrrD rG r r r r ra � s ra c @ sZ e Zd Zejdd� �Zdd� Zdd� Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
dS )� TestSuppressc C s t j}t � }| �|j|� d S r )�suppressrn r6 r� r r r r� � s z TestSuppress.test_instance_docsc C s" t t��}| �|� W 5 Q R X d S r )re r rZ )r
rT r r r �test_no_result_from_enter� s
z&TestSuppress.test_no_result_from_enterc C s* t t�� | �tdd�d� W 5 Q R X d S )Nr� r� � )re r r6 �powr r r r �test_no_exception� s
z TestSuppress.test_no_exceptionc C s t t�� td� W 5 Q R X d S )Nr� )re r �lenr r r r �test_exact_exception� s
z!TestSuppress.test_exact_exceptionc C s t t�� dd W 5 Q R X d S )NrW �2 )re � LookupErrorr r r r �test_exception_hierarchy� s
z%TestSuppress.test_exception_hierarchyc
C s6 | � t��" tt�� dd W 5 Q R X W 5 Q R X d S r� )r r: re r r r r r �test_other_exception s
z!TestSuppress.test_other_exceptionc
C s4 | � t�� t� � dd W 5 Q R X W 5 Q R X d S r� )r r: re r r r r � test_no_args s zTestSuppress.test_no_argsc C s@ t tt�� dd W 5 Q R X t tt�� td� W 5 Q R X d S )Nr- r r� )re r: r rj r r r r � test_multiple_exception_args
s z)TestSuppress.test_multiple_exception_argsc
C sr t t�}|� W 5 Q R X |� td� W 5 Q R X |�* |� td� W 5 Q R X d}dd W 5 Q R X | �|� d S )Nr� Tr- r )re rJ rj r( )r
Zignore_exceptionsZouter_continuedr r r r] s
z!TestSuppress.test_cm_is_reentrantN)r r r r r� r� rf ri rk rn ro rp rq r] r r r r rd � s
rd �__main__) rn rL rF r� r� Zunittest�
contextlibr� r r{ ZTestCaser r, r� r� r� r� r� r� r� r� rA rC r^ ra rd r �mainr r r r �<module> s8 + V" 0 ! x96