o
    û›h’  ã                   @   s`   d Z ddlZddlmZ ddlZddlmZmZ dd„ ZG dd„ dej	ƒZ
G d	d
„ d
ej	ƒZdS )zTests for io.pyé    N)ÚStringIO)ÚTeeÚcapture_outputc                  C   sB   t ƒ } d}t| dd�}t|| d� |  ¡ |d ksJ ‚| ¡  dS )z"Very simple check with stdout onlyÚHelloÚstdout©Úchannel©ÚfileÚ
N)r   r   ÚprintÚgetvalueÚclose)ÚchanÚtextÚtee© r   úQc:\Users\shubh\Videos\agent\nova\Lib\site-packages\IPython/utils/tests/test_io.pyÚtest_tee_simple   s   r   c                   @   s   e Zd Zdd„ Zdd„ ZdS )ÚTeeTestCasec                 C   s‚   t ƒ }t ƒ }d}tt|ƒ}tt||ƒ t||d�}t|d|d� | ¡ }|  | ¡ |¡ | ¡  tt||ƒ tt|ƒ|ks?J ‚d S )Nr   r   Ú )Úendr
   )	r   ÚgetattrÚsysÚsetattrr   r   r   ÚassertEqualr   )Úselfr   Ztrapr   r   Zstd_orir   Ztrap_valr   r   r   Útchan   s   
zTeeTestCase.tchanc                 C   s   dD ]}|   |¡ qd S )N)r   Ústderr)r   )r   r   r   r   r   Útest/   s   ÿzTeeTestCase.testN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r   r      s    r   c                   @   s   e Zd Zdd„ ZdS )ÚTestIOStreamc                 C   s\   t ƒ �}tdƒ tdtjd� W d  ƒ n1 sw   Y  |  |jd¡ |  |jd¡ dS )zcapture_output() context worksz
hi, stdoutz
hi, stderrr	   Nzhi, stdout
zhi, stderr
)r   r   r   r   r   r   )r   Úior   r   r   Útest_capture_output5   s   þz TestIOStream.test_capture_outputN)r    r!   r"   r%   r   r   r   r   r#   3   s    r#   )Ú__doc__r   r$   r   ZunittestZIPython.utils.ior   r   r   ZTestCaser   r#   r   r   r   r   Ú<module>   s   
