Format: application/x-python-code
1 o
2
3 ��hI> � @ s� d dl Z d dlZd dlZd dlmZ d dlZd dlmZ d dlm Z d dl
4 m Z d dl m
5 Z
6 d dlmZ d dlmZ d dlm mZ d d
7 � Zd d � Zd
8 d� Zdd� Zdd� Zdd� Zdd� Z dd� Z dd� Z e� d�d d � �Z d d d d d!d"�Z!e� d�d#d$� �Z"e� d�d%d&� �Z#d'd(� Z$d)d*� Z%d+d,� Z&e�'d-�d.d/� �Z(d0d1� Z)d2d3� Z*d4d5� Z+d6d7� Z,d8d9� Z-d:d;� Z.d<d=� Z/d>d?� Z0d@dA� Z1dBdC� Z2dS )D� N)�mock��display)� get_ipython)�capture_output)� NamedFileInTemporaryDirectory)�paths)�AssertNotPrintsc C s� d} t j| ddd�}d| |�� ksJ �t j| ddd�d�}d| |�� ks(J �t j| dd�}d| |�� ks9J �t j| d �}d
9 | |�� ksIJ �t j| d d �}d
10 | |�� ksZJ �dS )z5Simple test for display.Image(args, width=x,height=y)z+http://www.google.fr/images/srpr/logo3w.png�� )�url�width�heightz(<img src="%s" width="200" height="200"/>)r r
11 �r �metadata)r r z<img src="%s" width="200"/>�r z<img src="%s"/>T)r �
12 unconfinedz"<img src="%s" class="unconfined"/>N)r �Image� _repr_html_)�thisurl�img� r �Uc:\Users\shubh\Videos\agent\nova\Lib\site-packages\IPython/core/tests/test_display.py�test_image_size s r c C sZ t � jj} tjjD ]!}tjj| }tjd|d�}| |�\}}t|�t|dg�ks*J �q d S )Ns garbage)�format�
13 text/plain)r �display_formatterr r r Z_ACCEPTABLE_EMBEDDINGSZ
14 _MIMETYPES�sorted)�fmtr �mimer �datar r r r �test_image_mimes$ s
15 �r c C sD t jddddgd�ddid�d d
16 d d d
17 d�d�} dt| �ks J �d S )NZFeatureZPointg�rh��TT�g�S㥛�r@)�typeZ coordinates�namez Inca City)r! Zgeometry�
18 propertieszVhttp://s3-eu-west-1.amazonaws.com/whereonmars.cartodb.net/{basemap_id}/{z}/{x}/{y}.pngz celestia_mars-shaded-16k_globalzCelestia/praeseper � )Z
19 basemap_idZ attributionZminZoomZmaxZoom)r Z url_templateZ
20 layer_optionsz%<IPython.core.display.GeoJSON object>)r ZGeoJSON�str)Zgjr r r � test_geojson- s ���
21 ��r& c C �p t j�t�} tjt j�| d�dd�}|jdksJ �|jdks J �|� � \}}|d dks.J �|d dks6J �d S )N�2x2.pngT��retina� r r
22 )
23 �os�path�dirname�__file__r r �joinr
24 r �
25 _repr_png_��herer r �mdr r r �test_retina_pngD � r5 c s� dd l } ddlm} d�d�|� � | �� d��
26 }|��� W d � n1 s'w Y � �� � � ��fdd�}tjd|d ��- tj �d
27 �}|�
28 � � d �d u sQJ �tj �d
29 d
30 �}|�
31 � � d �d u sdJ �W d � d S 1 sow Y d S )Nr )�BytesIOs& <svg><circle x="0" y="0" r="1"/></svg>zhttp://test.com/circle.svg�wbc sP G dd� d�}| d �kr|��S | d �d kr$|� �}d|j d< |S |d �S )Nc @ s e Zd Zdd� Zdd� ZdS )z@test_embed_svg_url.<locals>.mocked_urlopen.<locals>.MockResponsec S s || _ ddi| _d S )Nz content-type�
32 image/svg+xml)� _svg_data�headers)�self�svgr r r �__init__[ s zItest_embed_svg_url.<locals>.mocked_urlopen.<locals>.MockResponse.__init__c S s | j S )N)r: )r< r r r �read_ s zEtest_embed_svg_url.<locals>.mocked_urlopen.<locals>.MockResponse.readN)�__name__�
33 __module__� __qualname__r> r? r r r r � MockResponseZ s rC r �z�gzipzcontent-encoding)r; )�args�kwargsrC �ret�Zgzip_svgZsvg_datar r r �mocked_urlopenY s
34 z*test_embed_svg_url.<locals>.mocked_urlopenzurllib.request.urlopen)Z side_effectr z<svgTrD ) rE �ior7 �open�write�getvaluer �patchr ZSVG�
35 _repr_svg_�
36 startswith)rE r7 �fprJ r= r rI r �test_embed_svg_urlN s � "�rS c C r' )Nz2x2.jpgTr) r+ r r
37 )
38 r, r- r. r/ r r r0 r
39 r � _repr_jpeg_r2 r r r �test_retina_jpegq r6 rU c C s t �d� d S )NZ�iVBORw0KGgoAAAANSUhEUgAAAAEAAAABAQMAAAAl21bKAAAAA1BMVEUAAACnej3aAAAAAWJLR0QAiAUdSAAAAAlwSFlzAAALEwAACxMBAJqcGAAAAAd0SU1FB94BCRQnOqNu0b4AAAAKSURBVAjXY2AAAAACAAHiIbwzAAAAAElFTkSuQmCC)r r r r r r �test_base64image{ s rV c C s� t �� } tjttjtj� | d�dd� t�ttj� tjttjdddd� tj� | d�}tj|d�}d |j
40 ks9J �|� � d
41 usAJ �tjtj� | d �d d�}d
42 |j
43 ksTJ �|� � d
44 u s\J �d
45 S )z4test format constraint, and validity of jpeg and pngztesting/tests/badformat.zipT)�filename�embedzthis is not an imageZ badformat)r r rX zcore/tests/2x2.png)rW �pngNztesting/tests/logo.jpgF�jpeg)
46 �ipathZget_ipython_package_dir�pytest�raises�
47 ValueErrorr r r, r- r0 r r1 rT )ZtpathZimgfiler r r r � test_image_filename_defaults~ s0 �� �r_ c C s ddl m} | �� S )Nr ��
48 InlineBackend)Zmatplotlib_inline.configra �instancer` r r r �_get_inline_config� s rc Z
49 matplotlibc C s� t � } d| _t�� � t�� W d � n1 sw Y | js#J �t�� �
50 t�d� W d � n1 s7w Y | jrAJ �d S )NF)rc Z
51 close_figuresr\ �deprecated_callr Zset_matplotlib_close)�cfgr r r �test_set_matplotlib_close� s
52
53 �
54
55 �rf � image/pngz
56 image/jpegzapplication/pdfr9 )rY rZ �pdfr* r= c C s� ddl m} t� jj}dD ]<}dd� |D �}t�� �
57 tj|� W d � n1 s*w Y |� � D ]\}}||v rB| |v sAJ �q3| |vsHJ �q3q
58 d S )Nr ��Figure)�rY )rh r= )rZ r* rY r c S s h | ]}t | �qS r )�
59 _fmt_mime_map)�.0r r r r � <setcomp>� s z.test_set_matplotlib_formats.<locals>.<setcomp>)
60 �matplotlib.figurerj r r �
61 formattersr\ rd r �set_matplotlib_formats�items)rj rp �formatsZ active_mimesr �fr r r �test_set_matplotlib_formats� s
62
63 ���ru c C s� ddl m} t� }t� }|j�tdd�� tdd�}t�� � t j
64 d i |�� W d � n1 s1w Y |j j d }|�
65 | �}|j}|}d |d
66 < d|d < |�|j� ||ksZJ �d S )
67 Nr ri �bar)Zfoo� )ZdpirY rg T�base64r rk )ro rj r rc Zprint_figure_kwargs�update�dictr\ rd r rq r rp Zlookup_by_type�keywords)rj �ipre rG � formatterrt Zformatter_kwargs�expectedr r r �"test_set_matplotlib_formats_kwargs� s
68
69 �
70 r c C s� t � } td��
71 | �d� W d � n1 sw Y z| �d� W n ty, Y nw td�� | �d� W d � dS 1 sBw Y dS )z�
72 Test that display is available without import
73
74 We don't really care if it's in builtin or anything else, but it should
75 always be available.
76 � NameErrorr Nz del display)r r Zrun_cellr� )r| r r r �test_display_available� s
77 � �
78 "�r� c C sH t �d�} t| �dks
79 J �| jdksJ �d| _t| �t�| �ks"J �d S )NzThis is a simple testz$<IPython.core.display.Pretty object>T)r �Pretty�reprr �_show_mem_addr�object�__repr__��pr r r � test_textdisplayobj_pretty_repr� s
80
81 r� c C s� t �d�} t| �dks
82 J �d| _t| �t�| �ksJ �d| _t| �dks&J �t �d�}t|�dks3J �d|_t|�t�|�ksAJ �d|_t|�dksLJ �d S )N�<br />z"<IPython.core.display.HTML object>TF� z(<IPython.core.display.Javascript object>)r �HTMLr� r� r� r� Z
83 Javascript)�h�jr r r �test_displayobject_repr� s
84
85 r� z
86 warnings.warnc C sh t �� | �� t �d� | �� t �d� | �� t �d� | �d� | �� t �d� | �d� d S )Nr� zE<html><p>Lots of content here</p><iframe src="http://a.com"></iframe>z$<iframe src="http://a.com"></iframe>z-Consider using IPython.display.IFrame insteadz$<IFRAME SRC="http://a.com"></IFRAME>)r r� Zassert_not_calledZassert_called_withZ
87 reset_mock)Zm_warnr r r � test_encourage_iframe_over_html s
88
89
90
91
92 r� c C s: t �d�} dt| �v s
93 J �d| _d| _| �� dksJ �d S )N�
94 z0/10z100%� z;<progress style='width:100%' max='10' value='5'></progress>)r � ProgressBarr� Z
95 html_width�progressr r� r r r �
96 test_progress s
97
98 r� c C sh t dd�� } t�d�D ]}| j}d�|�|v sJ �q W d � n1 s$w Y | j}d|v s2J �d S )NFr r� z{0}/5z5/5)r r r� �stdoutr )Zcaptured�i�outr r r �test_progress_iter s � �r� c C s� ddi} | g}ddd�ddd�ddd�ddd�g}t �| �t j| dd�t j| dd �t j| ddd�g}t||�D ]\}}|�� | |fksEJ �q7tjdd
99 ��)}t�d � t �t�| ��}t |�d ksbJ �|�� | |d
100 fksnJ �W d � n1 sxw Y t �|�t j|dd�t j|dd �t j|ddd�g}t||�D ]\}}|�� ||fks�J �q�tjdd
101 ��*}t�d � t �t�|��}t |�d ks�J �|�� ||d
102 fks�J �W d � d S 1 s�w Y d S )N�ar� F�root)�expandedr� TZcustom)r� )r� )�record�alwaysr+ r )
103 r �JSON�zip� _repr_json_�warnings�catch_warnings� simplefilter�json�dumps�len)�dZlisr Z json_objsr� r4 �wr r r � test_json' sB � �
104 � �
105 "�r� c C sd t �d�} | jr
106 J �| �� }d|vsJ �d|v sJ �t�t��
107 t �d�} W d � n1 s/w Y td��p}|�d� |� � t �|j
108 �} | jrMJ �| �� }d|vsWJ �t j|j
109 dd�} | �� }d |v siJ �t j|j
110 dd
111 d �} | �� }d |v s|J �t jddd
112 d �} | �� }d |v s�J �t jdddd �} | �� }d|v s�J �W d � dS 1 s�w Y dS )zMuse a tempfile, with dummy-data, to ensure that video embedding doesn't crashzhttp://ignoredz
113 src="data:zsrc="http://ignored"s abcNztest.mp4T)rX z src="data:video/mp4;base64,YWJj"z video/other)rX �mimetypez"src="data:video/other;base64,YWJj"z video/mp4ZYWJjz video/xyzz src="data:video/xyz;base64,YWJj") r ZVideorX r r\ r] r^ r rM �closer" )�v�htmlrt r r r �test_video_embeddingO s8
114
115 �
116
117
118 "�r� c C s2 d} t j| ddid�}|�� | ddifksJ �d S )Nz
119 <h1>Test</h1>�isolatedT)r )r r� r )�sr� r r r �test_html_metadatas s r� c C sV t � } tj�| jd��1}t�d�}|d u sJ �tjddd�}t|tj�s&J �tjddd�}t|tj�s5J �W d � n1 s?w Y |j|jksLJ �|j dksSJ �|j
120 d \}}|d
121 ks`J �|d t d�ii d �ksmJ �|j
122 d
123 \}}|d
124 kszJ �|d t d�ii d|jid�ks�J �|j
125 d \}}|d
126 ks�J �|d t d�ii d|jid�ks�J �d S )N�publish�x�y�secret��
127 display_idrD T� r r r )r r r+ r� �r r � transient� ) r r rO r� � display_pubr �
128 isinstance�
129 DisplayHandler� Z
130 call_count�call_args_listr� )r| �pub�handleZhandle2rF rG r r r �test_display_idy sH
131 � � � �� � ���r� c C s t � } tj�| jd��0}t�t��
132 t� d� W d � n1 s!w Y tj ddd� tj ddd� W d � n1 s>w Y |j
133 d \}}|dksPJ �|d t d�ii d
134 did d �ksaJ �|j
135 d
136 \}}|dksnJ �|d t d�ii d
137 did d �ksJ �d S )Nr� r� �1r� r� �2r r r r� T�r r r� ry r+ ) r r rO r� r� r\ r] � TypeErrorr Zupdate_displayr� r� )r| r� rF rG r r r �test_update_display� s8 � � �� �
138 ���r� c C s� t � } t�� }t|jt�sJ �t�d�}|jdksJ �tj�| j d��}|�d� |�
139 d� W d � n1 s8w Y |j d \}}|dksJJ �|dt d�ii d|jid �ks[J �|j d
140 \}}|dkshJ �|dt d�ii d|jid d �kszJ �d S )
141 Nzmy-idr� r� r� r r r r� r� r+ Tr� )
142 r r r� r� r� r% r rO r� r� ry r� r� )r| r� r� rF rG r r r �test_display_handle� s8
143
144 � �� � ���r� c C s� d} t j| dd�}d| |�� ksJ �t j| ddd�}d| |�� ks%J �t j| dd�}d | |�� ks6J �t j| d
145 did �}|jdksFJ �tj�t�}t jtj�|d �dd
146 �}|jdks_J �|� � \}}|d
147 dksmJ �dS )z+Simple test for display.Image(args, alt=x,)z http://example.com/image.pngzan image)r �altz <img src="%s" alt="an image"/>T)r r r� z1<img src="%s" class="unconfined" alt="an image"/>z>"& <z*<img src="%s" alt=">"& <"/>r� r r( )r� N)
148 r r r r� r, r- r. r/ r0 r1 )r r r3 �_r4 r r r �test_image_alt_tag� s � r� c C s> t �t�� t�d��� W d � d S 1 sw Y d S )Nz/this/file/does/not/exist/)r\ r] �FileNotFoundErrorr r r1 r r r r �/test_image_bad_filename_raises_proper_exception� s "�r� )3r� r, r� Zunittestr r\ ZIPythonr ZIPython.core.getipythonr ZIPython.utils.ior ZIPython.utils.tempdirr r r[ ZIPython.testing.toolsr ZIPython.testing.decorators�testingZ
149 decorators�decr r r& r5 rS rU rV r_ rc Z skip_withoutrf rl ru r r� r� r� rO r� r� r� r� r� r� r� r� r� r� r� r r r r �<module> s`
150 #
151
152 �
153
154
155
156 ($,!#