o
    :ήch                     @   sd   d dl Zd dlmZmZ d dlmZ d dlmZm	Z	 ej
d  G dd de	ZG dd de	ZdS )	    N)prepare_for_displayWindowManager)testing)assert_array_equalTestCasec                   @   sD   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dS )TestPrepareForDisplayc                 C   s   t tjdd d S )N
   r   nprandomrandself r   H/tmp/pip-target-vg8gfxp4/lib/python/skimage/io/tests/test_plugin_util.py
test_basic   s   z TestPrepareForDisplay.test_basicc                 C   s,   t tjdd}|jttjksJ d S )Nr      )r   r
   r   r   dtypeuint8)r   xr   r   r   
test_dtype   s   z TestPrepareForDisplay.test_dtypec                 C   sV   t jdtddd }t|}t|d |d  |d dks!J |d	 d
ks)J d S )N   )r   )         ).r   ).   )r   r   r   r   )r   r   r      )r
   arangefloatreshaper   r   )r   tmpr   r   r   r   	test_gray   s
   zTestPrepareForDisplay.test_grayc                 C      t tjddd d S )Nr   r   r	   r   r   r   r   
test_color      z TestPrepareForDisplay.test_colorc                 C   r"   )Nr   r   r	   r   r   r   r   
test_alpha   r$   z TestPrepareForDisplay.test_alphac                 C   sF   t t ttjdddd W d    d S 1 sw   Y  d S )Nr      r   raises
ValueErrorr   r
   r   r   r   r   r   r   test_wrong_dimensionality   s   "z/TestPrepareForDisplay.test_wrong_dimensionalityc                 C   sD   t t ttjddd W d    d S 1 sw   Y  d S )Nr      r'   r   r   r   r   test_wrong_depth#   s   "z&TestPrepareForDisplay.test_wrong_depthN)
__name__
__module____qualname__r   r   r!   r#   r%   r*   r,   r   r   r   r   r   
   s    r   c                   @   sL   e Zd ZdZejdddd Zdd Zdd	 Zd
d Z	dd Z
dd ZdS )TestWindowManagerFT)autousec                 C   s   t  | _| jd d S Ntest)r   wmacquirer   r   r   r   setup+   s   zTestWindowManager.setupc                 C   s   | j d | j d d S )Nwindow1)r4   
add_windowremove_windowr   r   r   r   test_add_window0   s   z!TestWindowManager.test_add_windowc                 C   s
   d| _ d S )NT)callback_calledr   r   r   r   callback4   s   
zTestWindowManager.callbackc                 C   s4   | j | j | j d | j d | jsJ d S Nwindow)r4   register_callbackr<   r8   r9   r;   r   r   r   r   test_callback7   s   zTestWindowManager.test_callbackc                 C   s,   | j  rJ | j d | j  sJ d S r=   )r4   has_windowsr8   r   r   r   r   test_has_images=   s   z!TestWindowManager.test_has_imagesc                 C   s   | j d d S r2   )r4   _releaser   r   r   r   teardownB   s   zTestWindowManager.teardownN)r-   r.   r/   r;   r   fixturer6   r:   r<   r@   rB   rD   r   r   r   r   r0   (   s    

r0   )numpyr
   skimage.io._plugins.utilr   r   skimage._sharedr   skimage._shared.testingr   r   r   seedr   r0   r   r   r   r   <module>   s    