Format: application/x-python-code
1 o
2
3 ��h� � @ s d Z ddlZddlZddlmZ edddddd d
4 �Zd d d
5 �Zdddd�iZd d dd ddd�Z ddiddiddiddiddiddid�Z
6 d Z d Z ej
7 �d e�� �d d � �Zd!d"� Zej
8 �d e�� �d#d$� �Zej
9 �d e�� �d%d&� �Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� ZdS )1z Tests for IPython.utils.capture� N)�capture� image/png�
10 image/jpeg�
11 image/svg+xml� text/html�application/json�application/javascript)�
12 _repr_png_� _repr_jpeg_�
13 _repr_svg_� _repr_html_� _repr_json_�_repr_javascript_s
14 binarydataz <b>bold</b>)r r �
15 � )�width�heightz<svg>zalert();z{})r r r r r r Zpng�existsZjpeg�svg�html�js�jsonz
16 hello, stdoutz
17 hello, stderr� method_mimec C s( t �� }| \}}t||�� du sJ �dS )zRichOutput with no argsN)r �
18 RichOutput�getattr)r �rich�method�mime� r �Vc:\Users\shubh\Videos\agent\nova\Lib\site-packages\IPython/utils/tests/test_capture.py�test_rich_output_emptyE s r c C sz t } t}tj| |d�}|�� | d ksJ �|�� | d |d fks#J �|�� du s+J �|�� du s3J �|�� du s;J �dS )ztest RichOutput basics��data�metadatar r N) �
19 basic_data�basic_metadatar r r r � _repr_latex_r r )r" r# r r r r �test_rich_outputL s r' c C s4 t }tj|d�}| \}}t||�� || ksJ �dS )z test RichOutput with no metadata�r" N)� full_datar r r )r r" r r r r r r � test_rich_output_no_metadataX s r* c C sB t }t}tj||d�}| \}}t||�� || || fks J �dS )z test RichOutput with metadatar! N)r) �
20 full_metadatar r r )r r" r# r r r r r r �test_rich_output_metadataa s
21 $r, c C s� t } tj| d�}t�� � }|�� W d � n1 sw Y t|j�dks)J �|jd }|j|jks6J �|j|jks>J �dS )z�test RichOutput.display
22
23 This is a bit circular, because we are actually using the capture code we are testing
24 to test itself.
25 r( N� r ) r) r r �capture_output�display�len�outputsr" r# )r" r �capZrich2r r r �test_rich_output_displayj s
26
27 �
28 r3 c C sx t jtd�} t �� �}ttdd� ttdtjd� | � � W d � n1 s'w Y t|j
29 ks3J �t|jks:J �dS )zcapture_output worksr( � ��end�r6 �fileN) r r r) r. �print� hello_stdout� hello_stderr�sys�stderrr/ �stdout�r r2 r r r �test_capture_outputy s
30
31 �r@ c C s� t jtd�} t jdd��}ttdd� ttdtjd� | � � W d � n1 s)w Y d|j
32 ks5J �t|jks<J �t |j �dksEJ �dS ) z!test capture_output(stdout=False)r( F)r> r4 r5 r7 Nr- �
33 r r r) r. r9 r: r; r< r= r/ r> r0 r1 r? r r r � test_capture_output_no_stdout� s
34 �rB c C s� t jtd�} t �� �0 t jdd��}ttdd� ttdtjd� | � � W d � n1 s.w Y W d � n1 s=w Y t|j
35 ksIJ �d|jksPJ �t |j �dksYJ �dS ) z!test capture_output(stderr=False)r( F)r= r4 r5 r7 Nr- rA r? r r r � test_capture_output_no_stderr� s
36 �� rC c C s� t jtd�} t jdd��}ttdd� ttdtjd� | � � W d � n1 s)w Y t|j
37 ks5J �t|jks<J �|j g ksCJ �dS )z"test capture_output(display=False)r( F)r/ r4 r5 r7 N) r r r) r. r9 r: r; r< r= r/ r> r1 r? r r r � test_capture_output_no_display� s
38 �rD )�__doc__r< ZpytestZ
39 IPython.utilsr �dictZ _mime_mapr$ r% r) r+ r: r; �markZ parametrize�itemsr r' r* r, r3 r@ rB rC rD r r r r �<module> s` �
40 ���� �
41
42
43
44