o
Zh @ s d Z ddlZddlZddlmZ edddddd d
Zd d d
ZddddiZd d dd dddZ ddiddiddiddiddiddidZ
d Z d Z ej
d e d d Zd!d" Zej
d e d#d$ Zej
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
image/jpeg
image/svg+xml text/htmlapplication/jsonapplication/javascript)
_repr_png_ _repr_jpeg_
_repr_svg_ _repr_html_ _repr_json__repr_javascript_s
binarydataz bold )r r
)widthheightzzalert();z{})r r r r r r ZpngexistsZjpegsvghtmljsjsonz
hello, stdoutz
hello, stderr method_mimec C s( t }| \}}t|| du sJ dS )zRichOutput with no argsN)r
RichOutputgetattr)r richmethodmime r W/var/www/html/lang_env/lib/python3.10/site-packages/IPython/utils/tests/test_capture.pytest_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 basicsdatametadatar r N)
basic_databasic_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 metadatar" 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)
full_metadatar r r )r r" r# r r r r r r test_rich_output_metadataa s
$r, c C s t } tj| d}t }| W d n1 sw Y t|jdks)J |jd }|j|jks6J |j|jks>J dS )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_outputdisplaylenoutputsr" r# )r" r capZrich2r r r test_rich_output_displayj s
r3 c C sx t jtd} t }ttdd ttdtjd | W d n1 s'w Y t|j
ks3J t|jks:J dS )zcapture_output worksr( endr6 fileN) r r r) r. print hello_stdout hello_stderrsysstderrr/ stdoutr r2 r r r test_capture_outputy s
r@ c C s t jtd} t jdd}ttdd ttdtjd | W d n1 s)w Y d|j
ks5J t|jks r4 r5 r7 Nr-
r r r) r. r9 r: r; r< r= r/ r> r0 r1 r? r r r test_capture_output_no_stdout s
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
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
rC c C s t jtd} t jdd}ttdd ttdtjd | W d n1 s)w Y t|j
ks5J t|jks r1 r? r r r test_capture_output_no_display s
rD )__doc__r< ZpytestZ
IPython.utilsr dictZ _mime_mapr$ r% r) r+ r: r; markZ parametrizeitemsr r' r* r, r3 r@ rB rC rD r r r r s`