
    &e                        d Z ddlZddlZddlmZ  edddddd	
          ZdddZddddiZdddddddZ	ddiddiddiddiddiddidZ
dZdZej                            de                                          d             Zd  Zej                            de                                          d!             Zej                            de                                          d"             Zd# Zd$ Zd% Zd& Zd' ZdS )(zTests for IPython.utils.capture    N)capture	image/png
image/jpegimage/svg+xml	text/htmlapplication/jsonapplication/javascript)
_repr_png__repr_jpeg_
_repr_svg__repr_html__repr_json__repr_javascript_s
   binarydataz<b>bold</b>)r   r   
      )widthheightz<svg>zalert();z{})r   r   r   r   r	   r   pngexistsjpegsvghtmljsjsonzhello, stdoutzhello, stderrmethod_mimec                 j    t          j                    }| \  }} t          ||                      J dS )zRichOutput with no argsN)r   
RichOutputgetattr)r   richmethodmimes       @lib/python3.11/site-packages/IPython/utils/tests/test_capture.pytest_rich_output_emptyr#   E   s?     DLFD 74  ""*****    c                  b   t           } t          }t          j        | |          }|                                | d         k    sJ |                                | d         |d         fk    sJ |                                J |                                J |                                J dS )ztest RichOutput basicsdatametadatar   r   N)	
basic_databasic_metadatar   r   r   r
   _repr_latex_r   r   )r'   r(   r   s      r"   test_rich_outputr,   L   s    DH4(;;;Dk!22222??k!2H[4I JJJJJ&&&!!##+++??$$$$$r$   c                     t           }t          j        |          }| \  }} t          ||                      ||         k    sJ dS )z test RichOutput with no metadatar'   N)	full_datar   r   r   )r   r'   r   r    r!   s        r"   test_rich_output_no_metadatar0   X   sR     D4(((DLFD 74  ""d4j000000r$   c                     t           }t          }t          j        ||          }| \  }} t	          ||                      ||         ||         fk    sJ dS )ztest RichOutput with metadatar&   N)r/   full_metadatar   r   r   )r   r'   r(   r   r    r!   s         r"   test_rich_output_metadatar3   a   sb     DH4(;;;DLFD 74  ""tDz8D>&BBBBBBBr$   c                  R   t           } t          j        |           }t          j                    5 }|                                 ddd           n# 1 swxY w Y   t          |j                  dk    sJ |j        d         }|j        |j        k    sJ |j        |j        k    sJ dS )ztest RichOutput.display

    This is a bit circular, because we are actually using the capture code we are testing
    to test itself.
    r.   N   r   )	r/   r   r   capture_outputdisplaylenoutputsr'   r(   )r'   r   caprich2s       r"   test_rich_output_displayr<   j   s     D4(((D			!	! S              s{q    KNE:"""">T]******s   AAAc                  n   t          j        t                    } t          j                    5 }t	          t
          d           t	          t          dt          j                   | 	                                 ddd           n# 1 swxY w Y   t
          |j
        k    sJ t          |j        k    sJ dS )zcapture_output worksr.    endr@   fileN)r   r   r/   r6   printhello_stdouthello_stderrsysstderrr7   stdoutr   r:   s     r"   test_capture_outputrJ   y   s    9---D			!	! Sl####l4444               3:%%%%3:%%%%%%s   ABB
B
c                     t          j        t                    } t          j        d          5 }t	          t
          d           t	          t          dt          j                   | 	                                 ddd           n# 1 swxY w Y   d|j
        k    sJ t          |j        k    sJ t          |j                  dk    sJ dS )	z!test capture_output(stdout=False)r.   F)rH   r>   r?   rA   Nr5   r   r   r/   r6   rC   rD   rE   rF   rG   r7   rH   r8   r9   rI   s     r"   test_capture_output_no_stdoutrM      s    9---D		u	-	-	- l####l4444               3:%%%%s{q         ABBBc                     t          j        t                    } t          j                    5  t          j        d          5 }t	          t
          d           t	          t          dt          j                   | 	                                 ddd           n# 1 swxY w Y   ddd           n# 1 swxY w Y   t
          |j
        k    sJ d|j        k    sJ t          |j                  dk    sJ dS )	z!test capture_output(stderr=False)r.   F)rG   r>   r?   rA   Nr5   rL   rI   s     r"   test_capture_output_no_stderrrP      s^   9---D			!	!  7#9#G#G#G 3l####l4444                              3:%%%%s{q      s6   B3ABB3B 	 B3#B 	$B33B7:B7c                     t          j        t                    } t          j        d          5 }t	          t
          d           t	          t          dt          j                   | 	                                 ddd           n# 1 swxY w Y   t
          |j
        k    sJ t          |j        k    sJ |j        g k    sJ dS )z"test capture_output(display=False)r.   F)r7   r>   r?   rA   N)r   r   r/   r6   rC   rD   rE   rF   rG   r7   rH   r9   rI   s     r"   test_capture_output_no_displayrR      s    9---D			.	.	. #l####l4444               3:%%%%3:%%%%;"rN   )__doc__rF   pytestIPython.utilsr   dict	_mime_mapr)   r*   r/   r2   rD   rE   markparametrizeitemsr#   r,   r0   r3   r<   rJ   rM   rP   rR    r$   r"   <module>r\      s   % % 


  ! ! ! ! ! ! D".  	   

     ) 	 8$X&x(H% $x0 8,  
 	(9(9::+ + ;:+	% 	% 	% 	(9(9::1 1 ;:1 	(9(9::C C ;:C+ + +& & &	! 	! 	!
! 
! 
!	 	 	 	 	r$   