a
    
d\                     @   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   e/var/www/html/stable-diffusion-webui/venv/lib/python3.9/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J | jtjks.J t tddd} t| jtjd v sXJ d S )Nzdata/color.pngT)Zas_gray   zdata/camera.pngZ
AllInteger)r   r   ndimdtypenpfloat64Zsctype2char	typecodesimgr   r   r   test_imread_as_gray   s
    r   c                  C   s   t td} | jdksJ d S )Nzdata/palette_color.png   )r   r   r   r   r   r   r   test_imread_palette    s    r   c                   C   s<   t t ttd W d    n1 s.0    Y  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   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   s\   t dd}|j}W d    n1 s&0    Y  t|| t|}t|| tj| d S )Nz.png)suffix)r   namer   r   r   r$   r   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floatingr0   r$   )r+   r   shaper,   r   r   r   test_imsave_roundtrip<   s    zTestSave.test_imsave_roundtripN)r!   )__name__
__module____qualname__r0   r6   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   	