}eJ> d dl Z d dlZd dlZd dlmZ d dlZd dlmZ d dlm Z d dl
m Z d dl m
Z
d dlmZ d dlmZ d dlmc mZ d Zd
Zd Zd Zd
Zd Zd Z d Z d Z ej> d d Z ddddddZ! ej> d d Z" ej> d d Z#d Z$d Z%d Z& ejN d d Z(d Z)d! Z*d" Z+d# Z,d$ Z-d% Z.d& Z/d' Z0d( Z1d) Z2y)* N)mockdisplay) get_ipython)capture_output) NamedFileInTemporaryDirectory)paths)AssertNotPrintsc d} t j | dd }d| z |j k( sJ t j | ddd }d| z |j k( sJ t j | d }d| z |j k( sJ t j | }d
| z |j k( sJ t j | d }d
| z |j k( sJ y)z5Simple test for display.Image(args, width=x,height=y)z+http://www.google.fr/images/srpr/logo3w.png )urlwidthheightz()r r r
metadata)r
r zr
zT)r
unconfinedz"N)r Image _repr_html_)thisurlimgs A/usr/lib/python3/dist-packages/IPython/core/tests/test_display.pytest_image_sizer s ;G
--G3s
;C 5 AS__EV VV V
--GsS.I
JC 5 AS__EV VV V
--G3
/C (G 48I II I
--G
$C (COO,= == =
--G
5C /7 ;s?P PP P c 2 t j j } t j j
D ]\ }t j j | }t j d| } | | \ }}t | t |dg k( r\J y )Ns garbage)format
text/plain)r display_formatterr r r _ACCEPTABLE_EMBEDDINGS
_MIMETYPESsorted)fmtr mimer datar s r test_image_mimesr% $ s{
-
)
)
0
0C--66 <}}''/mmJv6 Shd|vt\&: ;;;; )r GeoJSONstr)gjs r test_geojsonr8 - s_ ' 1
n;.
B$ 3c"g == =r c F t j j t } t j
t j j
| d d }|j dk( sJ |j dk( sJ |j \ }}|d dk( sJ |d dk( sJ y )N2x2.pngTretina r r )
ospathdirname__file__r r joinr r
_repr_png_herer r$ mds r test_retina_pngrG D s
77??8 $D
-- T95d
CC ::?? 99>>~~ HD"
g;!
h<1 r c dd l } ddlm} dd | | j d 5 }|j d d d j fd}t
j d| 5 t j }|j j d
d u sJ t j d z }|j j d
d u sJ d d d y # 1 sw Y xY w# 1 sw Y y xY w)
Nr )BytesIOs& zhttp://test.com/circle.svgwbc G d d }| d k( r | S | d dz k( r | }d|j d< |S |d S )Nc e Zd Zd Zd Zy)@test_embed_svg_url..mocked_urlopen..MockResponsec $ || _ ddi| _ y )Nz content-type
image/svg+xml) _svg_dataheaders)selfsvgs r __init__zItest_embed_svg_url..mocked_urlopen..MockResponse.__init__[ s !$ . @ r c | j S )N)rP )rR s r readzEtest_embed_svg_url..mocked_urlopen..MockResponse.read_ s ~~%r N)__name__
__module__ __qualname__rT rV r r MockResponserM Z s
A
&r r[ r zgzipzcontent-encoding)rQ )argskwargsr[ retgzip_svgsvg_datar
s r mocked_urlopenz*test_embed_svg_url..mocked_urlopenY s^ & & 7c> ) )
!Wc
! x(C.4CKK* +JD!!r zurllib.request.urlopen) side_effectr z r? r@ rA r r rB r r _repr_jpeg_rD s r test_retina_jpegrq q s
77??8 $D
-- T95d
CC ::?? 99>> HD"
g;!
h<1 r c . t j d y )NڴiVBORw0KGgoAAAANSUhEUgAAAAEAAAABAQMAAAAl21bKAAAAA1BMVEUAAACnej3aAAAAAWJLR0QAiAUdSAAAAAlwSFlzAAALEwAACxMBAJqcGAAAAAd0SU1FB94BCRQnOqNu0b4AAAAKSURBVAjXY2AAAAACAAHiIbwzAAAAAElFTkSuQmCC)r r rZ r r test_base64imagert { s MM I Jr c t j } t j t t
j t j j | d d t j t t
j t j t t
j ddd t j j | d }t j | }d |j k( sJ |j J t j t j j | d d }d
|j k( sJ |j J y
)z4test format constraint, and validity of jpeg and pngztesting/tests/badformat.zipT)filenameembedzthis is not an image badformat)r$ r rw zcore/tests/2x2.png)rv pngNztesting/tests/logo.jpgFjpeg)
ipathget_ipython_package_dirpytestraises
ValueErrorr r r> r? rB r rC rp )tpathimgfiler s r test_image_filename_defaultsr ~ s ) ) +E
MM
e%BC MM*gmm,
MM
# ggll5"67G
--
)C CJJ >> '' '
-- e%=>e C SZZ ?? $$ $r c . ddl m} | j S )Nr
InlineBackend)matplotlib_inline.configr instancer s r _get_inline_configr s 6 ! ! ##r
matplotlibc T t } d| _ t j 5 t j
d d d | j sJ t j 5 t j
d d d d | j rJ y # 1 sw Y XxY w# 1 sw Y $xY w)NF)r
close_figuresr} deprecated_callr set_matplotlib_close)cfgs r test_set_matplotlib_closer s
C C ! '$$&' ! ,$$U+, ' ', ,s B%B B B' image/pngz
image/jpegzapplication/pdfrO )ry rz pdfr< rS c X ddl m} t j j }dD ]p }|D ch c] }t
|
}}t
j 5 t j | d d d |j D ] \ }}||v r| |v rJ | |vrJ r y c c}w # 1 sw Y :xY w)Nr Figure)ry )r rS )rz r< ry rZ ) matplotlib.figurer r r
formatters
_fmt_mime_mapr} r r set_matplotlib_formatsitems)r r formatsr" active_mimesr# fs r test_set_matplotlib_formatsr s ( 00;;J
' 7>>s
c*> >
#
#
% 5 * *G 4 5!'') 'GD!|# {"{ Q& '
' ? 5 5s BB B) c ddl m} t }t }|j j t
d t
d }t j 5 t j d i | d d d |j j d }|j | }|j }|}d |d
< d|d < |j |j ||k( sJ y # 1 sw Y nxY w)
Nr r bar)foo )dpiry r Tbase64r" r )r r r r print_figure_kwargsupdatedictr} r r r r r lookup_by_typekeywords)r ipr r_ formatterr formatter_kwargsexpecteds r "test_set_matplotlib_formats_kwargsr s ( B
C""4E?3
c]F ! 8&&778$$// T)r Prettyreprr$ _show_mem_addrobject__repr__ps r test_textdisplayobj_pretty_reprr sZ ./A 7< << < 66, ,, ,A 7fooa( (( (r c t j d } t | dk( sJ d| _ t | t j | k( sJ d| _ t | dk( sJ t j d }t | dk( sJ d|_ t | t j | k( sJ d|_ t | dk( sJ y )N z"TF z()r HTMLr r r r
Javascript)hjs r test_displayobject_reprr s X A 7: :: :A 7fooa( (( ( A 7: :: :2 A 7@ @@ @A 7fooa( (( ( A 7@ @@ @r z
warnings.warnc t j | j t j d | j t j d | j t j d | j d | j t j d | j d y )Nr zE
Lots of content here
z$z-Consider using IPython.display.IFrame insteadz$)r r assert_not_calledassert_called_with
reset_mock)m_warns r test_encourage_iframe_over_htmlr s LLN
LL
LLXY
LL78
MN
LL78
MNr c t j d } dt | v sJ d| _ d| _ | j dk( sJ y )N
z0/10z100% z;)r ProgressBarr
html_widthprogressr r s r
test_progressr sI B A T!W ALAJ
XXXr c t d 5 } t j d D ]# }| j }dj | |v r#J d d d j }d|v sJ y # 1 sw Y xY w)NFr r z{0}/5z5/5)r r r stdoutr )capturediouts r test_progress_iterr sr & ,($$Q' ,A//C>>!$+ ++ ,, //C C<< , ,s 8A%A%%A.c ddi} | g}ddddddddddddg}t j | t j | d t j | d t j | dd g}t || D ] \ }}|j | |fk( r J t j
d
5 }t j d t j t j | }t | d k( sJ |j | |d
fk( sJ d d d t j | t j |d t j |d t j |dd g}t || D ] \ }}|j ||fk( r J t j
d
5 }t j d t j t j | }t | d k( sJ |j ||d
fk( sJ d d d y # 1 sw Y xY w# 1 sw Y y xY w)Nar Froot)expandedr Tcustom)r )r )recordalwaysr= r )
r JSONzip _repr_json_warningscatch_warnings simplefilterjsondumpslen)dlisr json_objsr rF ws r test_jsonr ' s aA #CF+F+H-H- H Q Q& QX& QH5 I Y) *2}}1b')))*
- 3 h' LLA '1v{{}}1hqk"2222 3 S S4( Sx( S4h7 I Y) ,2}}3)+++,
- 5 h' LLC )1v{{}}3 "4444 5 5 3 3 5 5s 4A(H1?A(H>1H;>Ic t j d } | j rJ | j }d|vsJ d|v sJ t j
t 5 t j d } ddd t d 5 }|j d |j t j |j } | j rJ | j }d|vsJ t j |j d } | j }d |v sJ t j |j dd
} | j }d |v sJ t j ddd
} | j }d |v sJ t j ddd } | j }d|v sJ ddd y# 1 sw Y OxY w# 1 sw Y yxY w)zMuse a tempfile, with dummy-data, to ensure that video embedding doesn't crashzhttp://ignoredz
src="data:zsrc="http://ignored"s abcNztest.mp4T)rw z src="data:video/mp4;base64,YWJj"z video/other)rw mimetypez"src="data:video/other;base64,YWJj"z video/mp4YWJjz video/xyzz src="data:video/xyz;base64,YWJj") r Videorw r r} r~ r r rg closer+ )vhtmlr s r test_video_embeddingr O s
&'Aww; ==?D t ## # !T )) ) z " " MM& !"
'z 2 :a MM!&& !77{}}4''' MM!&& -}}1T999 MM!&&} E}}3t;;; MM&{ C}}1T999 MM' D}}1T999/: :" ": :s F<D0G <G Gc l d} t j | ddi }|j | ddifk( sJ y )Nz
Test
isolatedT)r )r r r )sr s r test_html_metadatar s s: A Q*d!34A ==?q:t"4 5 55 5r c 6 t } t j j | j d 5 }t j
d }|J t j
dd }t
|t
j sJ t j
dd }t
|t
j sJ d d d j j k7 sJ j dk( sJ |j d \ }}|d
k( sJ |d t d ii d k( sJ |j d
\ }}|d
k( sJ |d t d ii d|j idk( sJ |j d \ }}|d
k( sJ |d t d ii d|j idk( sJ y # 1 sw Y xY w)Npublishxysecret
display_idr\ T r rZ r )r$ r r= r r$ r transient ) r r ri r display_pubr
isinstance
DisplayHandler
call_countcall_args_listr )r pubhandlehandle2r^ r_ s r test_display_idr y s B
2>>9 5 : %~ ~ :&'"7"7888//#$7'7#8#8999
: 2 2 22 2 >>Q %%a(LD& 2:: $s)
%%a(LD& 2:: $s)
&++
%%a(LD& 2:: $s)
',,
?: :s B FFc @ t } t j j | j d 5 }t j t 5 t j d d d d t j dd t j dd d d d j d \ }}|dk( sJ |d t d ii d
did d k( sJ |j d
\ }}|dk( sJ |d t d ii d
did d k( sJ y # 1 sw Y xY w# 1 sw Y zxY w)Nr r 1r r 2r rZ r r Tr$ r r r r= ) r r ri r r r} r~ TypeErrorr update_displayr r )r r r^ r_ s r test_update_displayr s= B
2>>9 5 4
]]9
% ( " "3 ' ( ss3 ss3 4
%%a(LD& 2:: $s)
#
%%a(LD& 2:: $s)
#
% ( (4 4s# DD%6DD
DD c t t } t j }t |j t
sJ t j d }|j dk( sJ t j j | j d 5 }|j d |j d d d d j d \ }}|dk( sJ |dt d ii d|j id k( sJ |j d
\ }}|dk( sJ |dt d ii d|j id d k( sJ y # 1 sw Y xY w)
Nzmy-idr r r r rZ r r r r= Tr )
r r r r r r6 r ri r r r r r )r r r r^ r_ s r test_display_handler sU B
"
"
$F f'' -- -
"
"7
+F '' '
2>>9 5 s
c %%a(LD& 2:: $s)
&++
%%a(LD& 2:: $s)
&++
# s #D..D7c d} t j | d }d| z |j k( sJ t j | dd }d| z |j k( sJ t j | d }d | z |j k( sJ t j | d
di }|j dk( sJ t j
j
t }t j t j
j |d d
}|j dk( sJ |j \ }}|d
dk( sJ y)z+Simple test for display.Image(args, alt=x,)z http://example.com/image.pngzan image)r
altz T)r
r r z1z>"& r r r: )r N)
r r r r r> r? r@ rA rB rC )r r rE _rF s r test_image_alt_tagr s ,G
--G
4C +w 73??;L LL L
--G*
EC;wG ?? --G
1C 77 CsGX XX X
--Guj.A
BC 77j
77??8 $D
-- T95:
FC 77j NN EAr
e9
"" "r c t j t 5 t j d j d d d y # 1 sw Y y xY w)Nz/this/file/does/not/exist/)r} r~ FileNotFoundErrorr r rC rZ r r /test_image_bad_filename_raises_proper_exceptionr s> ( ) A
23>>@A A As $AA)3r r> r unittestr r} IPythonr IPython.core.getipythonr IPython.utils.ior IPython.utils.tempdirr r r{ IPython.testing.toolsr
IPython.testing.decoratorstesting
decoratorsdecr r% r8 rG rn rq rt r r skip_withoutr r r r r r r ri r r r r r r r r r r r rZ r r r' sB
/ + ? " 1 ( ( Q <>. ;F J%8$
, ! !
, ' '& , ( (" $)
A O O O"%5P":H6 )X B F#,Ar