o
    i                     @   sB   d Z ddlmZ ddlmZ dddZdd	 Zd
d Zdd ZdS )z\Test display of images

The effect of shape metadata is validated, using Image(retina=True)
    )By   )wait_for_tags]   iVBORw0KGgoAAAANSUhEUgAAAAIAAAACCAIAAAD91JpzAAAAC0lEQVR4nGNgQAYAAA4AAamRc7EA\nAAAASUVORK5CYIIsW  /9j/4AAQSkZJRgABAQAAAQABAAD/2wBDAAgGBgcGBQgHBwcJCQgKDBQNDAsLDBkSEw8UHRofHh0a
HBwgJC4nICIsIxwcKDcpLDAxNDQ0Hyc5PTgyPC4zNDL/2wBDAQkJCQwLDBgNDRgyIRwhMjIyMjIy
MjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjL/wAARCAACAAIDASIA
AhEBAxEB/8QAHwAAAQUBAQEBAQEAAAAAAAAAAAECAwQFBgcICQoL/8QAtRAAAgEDAwIEAwUFBAQA
AAF9AQIDAAQRBRIhMUEGE1FhByJxFDKBkaEII0KxwRVS0fAkM2JyggkKFhcYGRolJicoKSo0NTY3
ODk6Q0RFRkdISUpTVFVWV1hZWmNkZWZnaGlqc3R1dnd4eXqDhIWGh4iJipKTlJWWl5iZmqKjpKWm
p6ipqrKztLW2t7i5usLDxMXGx8jJytLT1NXW19jZ2uHi4+Tl5ufo6erx8vP09fb3+Pn6/8QAHwEA
AwEBAQEBAQEBAQAAAAAAAAECAwQFBgcICQoL/8QAtREAAgECBAQDBAcFBAQAAQJ3AAECAxEEBSEx
BhJBUQdhcRMiMoEIFEKRobHBCSMzUvAVYnLRChYkNOEl8RcYGRomJygpKjU2Nzg5OkNERUZHSElK
U1RVVldYWVpjZGVmZ2hpanN0dXZ3eHl6goOEhYaHiImKkpOUlZaXmJmaoqOkpaanqKmqsrO0tba3
uLm6wsPExcbHyMnK0tPU1dbX2Nna4uPk5ebn6Onq8vP09fb3+Pn6/9oADAMBAAIRAxEAPwD5/ooo
oAoo2Qoo)	image/png
image/jpegc                 C   s*   ddg}| j dd|d | d d S )Nzimport base64z*from IPython.display import display, Imager   
)indexcontent)	edit_celljoinexecute_cell)notebookcommands r   e/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/notebook/tests/selenium/test_display_image.pyimports   s
   r   c                 C   s   t | }d| dd| dg}| d| | | t| j| ddd | j| tjd}|	d	}|
d
d }d| d}	||	ksIJ |rMdnd}
|jd |
ksXJ |jd |
ksaJ |	dt|
kslJ |	dt|
kswJ dS )z(Validate that image renders as expected.z
b64data = z"data = base64.decodebytes(b64data)zdisplay(Image(data, retina=z))r   imgT)singlesrc,r   zdata:z;base64r      widthheightN)b64_image_dataappendr   r   r   cellsfind_elementr   TAG_NAMEget_attributesplitsizestr)r   
cell_index	image_fmtretinab64datar   img_elementr   prefixexpected_prefixexpected_sizer   r   r   validate_img   s$   


r*   c                 C   sD   t |  t| ddd t| ddd t| ddd t| ddd d S )	Nr   r   Fr   T   r      )r   r*   )r   r   r   r   test_display_image7   s
   r-   N)	__doc__selenium.webdriver.common.byr   utilsr   r   r   r*   r-   r   r   r   r   <module>   s    	