U
    ?h                     @   s&   d dl Zd dlmZ G dd dZdS )    N)
histogramsc                   @   s   e Zd Zdd Zdd ZdS )TestHistogramc                 C   s:   t jdt jd}t|d}|D ]}t j| d qd S )N)2   r      Zdtype   i	  )nponesuint8r   testingZassert_equalsum)selfimgZbandsZband r   R/var/www/html/venv/lib/python3.8/site-packages/skimage/io/tests/test_histograms.py
test_basic   s    
zTestHistogram.test_basicc                 C   s   t ddd}t jddd}||d d d d df< ||d d d d df< ||d d d d d	f< t|d\}}}}t j|| t j|| t j|| t j|t d d S )
Nr   3      )r   r   r   r
   r   r         )r   ZarangeZreshapeemptyr   r   Zassert_array_equalr	   )r   Zchannelr   rgbvr   r   r   test_counts   s    zTestHistogram.test_countsN)__name__
__module____qualname__r   r   r   r   r   r   r      s   r   )numpyr   Zskimage.io._plugins._histogramsr   r   r   r   r   r   <module>   s   