U
    ?h\                     @   s   d dl mZ d dlZd dlmZ d dlmZmZm	Z	m
Z
 d dlmZ d dlmZmZmZmZ d dlmZ ed d	d
 Zdd Zdd Zdd Zdd Zdd ZG dd deZdS )    )NamedTemporaryFileN)io)imreadimsave
use_pluginreset_plugins)testing)TestCaseassert_array_equalassert_array_almost_equalfetch)importorskipr   c                   C   s   t d d S )Nr   )r    r   r   N/var/www/html/venv/lib/python3.8/site-packages/skimage/io/tests/test_imread.pysetup   s    r   c                   C   s
   t   d S )N)r   r   r   r   r   teardown   s    r   c                  C   s\   t tddd} | jdkst| jtjks.tt tddd} t| jtjd ksXtd S )Nzdata/color.pngT)Zas_gray   zdata/camera.pngZ
AllInteger)	r   r   ndimAssertionErrordtypenpZfloat64Zsctype2char	typecodesimgr   r   r   test_imread_as_gray   s
    r   c                  C   s   t td} | jdkstd S )Nzdata/palette_color.png   )r   r   r   r   r   r   r   r   test_imread_palette    s    r   c                	   C   s(   t t ttd W 5 Q R X d S )Nzdata/truncated.jpg)r   ZraisesRuntimeErrorr   r   r   r   r   r   r   test_imread_truncated_jpg%   s    r   c                  C   s:   t dt} d| d d d< ttd}t|t|  d S )N
   r       r   zdata/checker_bilevel.png)r   Zzerosboolr   r   r
   astype)expectedr   r   r   r   test_bilevel*   s    r%   c                   @   s   e Zd ZdddZdd ZdS )TestSaver!   c              	   C   sH   t dd}|j}W 5 Q R X t|| t|}t|| tj| d S )Nz.png)suffix)r   namer   r   r   r#   r   Zint32)selfxZscalingffnameyr   r   r   	roundtrip3   s
    
zTestSave.roundtripc                 c   sr   t j}t jd dD ]V}t j||dt jj|  }t |t jrR| j|dfV  q|d 	|}| j|fV  qd S )Nr   )r   )r    r    r   )r    r       )r      )
r   Zuint8randomseedZonesZrandZ
issubdtypeZfloatingr.   r#   )r)   r   shaper*   r   r   r   test_imsave_roundtrip<   s    zTestSave.test_imsave_roundtripN)r!   )__name__
__module____qualname__r.   r4   r   r   r   r   r&   2   s   
	r&   )tempfiler   numpyr   Zskimager   Z
skimage.ior   r   r   r   Zskimage._sharedr   Zskimage._shared.testingr	   r
   r   r   Zpytestr   r   r   r   r   r   r%   r&   r   r   r   r   <module>   s   	