U
pu
f @ 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_Z _repr_jpeg_
_repr_svg_ _repr_html_Z _repr_json__repr_javascript_s
binarydataz bold )r r
)widthheightzzalert();z{})r r r r r r ZpngexistsZjpegsvghtmlZjsjsonz
hello, stdoutz
hello, stderr method_mimec C s( t }| \}}t|| dks$tdS )zRichOutput with no argsN)r
RichOutputgetattrAssertionError)r richmethodmime r D/tmp/pip-unpacked-wheel-bxwx4lqe/IPython/utils/tests/test_capture.pytest_rich_output_emptyE s r c C sz t } t}tj| |d}| | d ks*t| | d |d fksFt| dksVt| dksft| dksvtdS )ztest RichOutput basicsdatametadatar r N)
basic_databasic_metadatar r r r r Z _repr_latex_r r
)r r! r r r r test_rich_outputL s r$ c C s4 t }tj|d}| \}}t|| || ks0tdS )z test RichOutput with no metadatar N) full_datar r 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>tdS )z test RichOutput with metadatar N)r&
full_metadatar r r r )r r r! r r r r r r test_rich_output_metadataa s
r) c C sl t } tj| d}t }| W 5 Q R X t|jdks>t|jd }|j|jksXt|j |j kshtdS )ztest RichOutput.display
This is a bit circular, because we are actually using the capture code we are testing
to test itself.
r% r N)
r& r r capture_outputdisplaylenoutputsr r r! )r r capZrich2r r r test_rich_output_displayj s
r0 c C sd t jtd} t *}ttdd ttdtjd | W 5 Q R X t|j
ksRt t|jks`t dS )zcapture_output worksr% endr3 fileN) r r r& r+ print hello_stdout hello_stderrsysstderrr, stdoutr r r/ r r r test_capture_outputy s
r= c C sz t jtd} t jdd*}ttdd ttdtjd | W 5 Q R X d|j
ksVt t|jksdt t |j
dksvt dS ) z!test capture_output(stdout=False)r% F)r; r1 r2 r4 r* Nr r r& r+ r6 r7 r8 r9 r: r, r; r r- r. r< r r r test_capture_output_no_stdout s r? c
C s t jtd} t B t jdd*}ttdd ttdtjd | W 5 Q R X W 5 Q R X t|j
ksjt d|jksxt t |j
dkst dS ) z!test capture_output(stderr=False)r% F)r: r1 r2 r4 r* Nr> r< r r r test_capture_output_no_stderr s r@ c C sv t jtd} t jdd*}ttdd ttdtjd | W 5 Q R X t|j
ksVt t|jksdt |j g ksrt dS )z"test capture_output(display=False)r% F)r, r1 r2 r4 N)
r r r& r+ r6 r7 r8 r9 r: r, r; r r. r< r r r test_capture_output_no_display s rA )__doc__r9 ZpytestZ
IPython.utilsr dictZ _mime_mapr" r# r& r( r7 r8 markZ parametrizeitemsr r$ r' r) r0 r= r? r@ rA r r r r s^