a
    ëÝEb  ã                   @   sn   d dl mZ d dlmZ d dlZd dlmZ d dlmZ	 d dl
mZ edgdd�d	d
„ ƒZdd„ Zdd„ ZdS )é    )ÚBytesIO)ÚPathN)Úimage_comparison)Úpyplotzpngsuite.pngg¸…ëQ¸ž?)Ztolc                  C   s    t ttƒjd  d¡ƒ} tjt| ƒdfd� t| ƒD ]B\}}t 	|¡}d }|j
dkrZtj}tj|||d ddg|d� q4t ¡ j d¡ t ¡  dt| ƒ¡ d S )	Nzbaseline_images/pngsuitez	basn*.pngé   )Zfigsizeé   r   )ZextentÚcmapz#ddffff)Úsortedr   Ú__file__ÚparentÚglobÚpltÚfigureÚlenÚ	enumerateÚimreadÚndimÚcmÚgrayZimshowZgcaÚpatchZset_facecolorZset_xlim)ÚfilesÚiÚfnameÚdatar   © r   úY/home/ja/django-apps/lartica_env/lib/python3.9/site-packages/matplotlib/tests/test_png.pyÚtest_pngsuite   s    ÿ

r   c                 C   sÖ   |   d¡}t| d¡ƒ}t| d¡ƒ}t |¡ t|dƒ�}| ¡ }W d   ƒ n1 sX0    Y  t|dƒ�"}| |d d… ¡ W d   ƒ n1 s”0    Y  t 	t
¡� t |¡ W d   ƒ n1 sÈ0    Y  d S )NÚtestztest.pngztest_truncated.pngÚrbÚwbé   )ÚmkdirÚstrÚjoinr   ÚsavefigÚopenÚreadÚwriteÚpytestÚraisesÚ	Exceptionr   )ZtmpdirÚdr   Zfname_tZfinÚbufZfoutr   r   r   Útest_truncated_file   s    

&0r-   c                  C   sj   t ƒ } t | ¡ |  d¡ t |  d¡ƒ}| d¡ t t¡� t |¡ W d   ƒ n1 s\0    Y  d S )Nr   r    )	r   r   r$   Úseekr&   r(   r)   r*   r   )ÚbÚb2r   r   r   Útest_truncated_buffer,   s    


r1   )Úior   Úpathlibr   r(   Zmatplotlib.testing.decoratorsr   Z
matplotlibr   r   Zmatplotlib.cmr   r   r-   r1   r   r   r   r   Ú<module>   s   
