
    ^h                         S r SSKrSSKJr  SSKrSSKJrJr  S r " S S\R                  5      r
 " S S	\R                  5      rg)
zTests for io.py    N)StringIO)Teecapture_outputc                  t    [        5       n Sn[        U SS9n[        XS9  U R                  5       US-   :X  d   eg)z"Very simple check with stdout onlyHellostdoutchannelfile
N)r   r   printgetvalue)chantexttees      U/home/james-whalen/.local/lib/python3.13/site-packages/IPython/utils/tests/test_io.pytest_tee_simpler      s:    :DD
dH
%C	$==?dTk)))    c                        \ rS rSrS rS rSrg)TeeTestCase   c                 d   [        5       n[        5       nSn[        [        U5      n[        [        X5        [	        X1S9n[        USUS9  UR                  5       nU R                  UR                  5       U5        UR                  5         [        [        X5        [        [        U5      U:X  d   eg )Nr   r	    )endr   )	r   getattrsyssetattrr   r   r   assertEqualclose)selfr
   trapr   r   std_orir   trap_vals           r   tchanTeeTestCase.tchan   s    zz#w'W#$(d&==?$/		W&sG$///r   c                 8    S H  nU R                  U5        M     g )N)r   stderr)r%   )r!   r   s     r   testTeeTestCase.test.   s    (DJJt )r    N)__name__
__module____qualname____firstlineno__r%   r)   __static_attributes__r+   r   r   r   r      s    0&r   r   c                       \ rS rSrS rSrg)TestIOStream2   c                    [        5        n[        S5        [        S[        R                  S9  SSS5        U R	                  WR
                  S5        U R	                  UR                  S5        g! , (       d  f       NG= f)zcapture_output() context worksz
hi, stdoutz
hi, stderrr   Nzhi, stdout
zhi, stderr
)r   r   r   r(   r   r   )r!   ios     r   test_capture_output TestIOStream.test_capture_output4   sZ     ,,SZZ0  	N3N3 s   $A00
A>r+   N)r,   r-   r.   r/   r6   r0   r+   r   r   r2   r2   2   s    4r   r2   )__doc__r   r5   r   unittestIPython.utils.ior   r   r   TestCaser   r2   r+   r   r   <module>r<      s?        0*(## 2
48$$ 
4r   