ó
 \c           @   s]   d  d l  Z d  d l m Z d  d l m Z m Z m Z m Z e   d    Z	 d   Z
 d S(   i˙˙˙˙N(   t
   quickshift(   t   assert_greatert   test_parallelt   assert_equalt   assert_array_equalc          C   s6  t  j j d  }  t  j d  } d | d  d  d d   f <d | d d   d  d  f <d | d d   d d   f <| d |  j d	 | j  7} t | d
 d d d d d d t d d } t t	 t  j
 |   d  xW t d  D]I } t  j | | | k d d d d d d g d } t | | d  qĺ Wd  S(   Ni    i   i   gÉ?i
   gŮ?g333333ă?gš?t   sizet   kernel_sizei   t   max_disti   t   random_seedt   convert2labt   sigmai   t   binsg333333Ó?g      ŕ?i   (   i   i   (   t   npt   randomt   RandomStatet   zerost   normalt   shapeR    t   FalseR   t   lent   uniquet   ranget	   histogramR   (   t   rndt   imgt   segt   it   hist(    (    sI   lib/python2.7/site-packages/skimage/segmentation/tests/test_quickshift.pyt	   test_grey   s    2c          C   sA  t  j j d  }  t  j d  } d | d  d  d  d  d f <d | d d   d  d  d f <d | d d   d d   d f <| d |  j d	 | j  7} d | | d k <d | | d k  <t | d
 d d d d d d d } t t t  j	 |   d  t
 | d  d  d  d  f d  t
 | d d   d  d  f d  t
 | d  d  d d   f d  t
 | d d   d d   f d  t | d d d d d
 d d t d d } t t t  j	 |   d  | d d  d   f | d d  d   f k j   st  | d  d   d f | d  d   d f k j   s=t  d  S(   Ni    i   i   i   i   i
   i   g{ŽGáz?R   R   R   i   R   R
   i   R	   i   i	   (   i   i   i   (   R   R   R   R   R   R   R    R   R   R   R   R   t   allt   AssertionError(   R   R   R   t   seg2(    (    sI   lib/python2.7/site-packages/skimage/segmentation/tests/test_quickshift.pyt
   test_color   s&    $####8(   t   numpyR   t   skimage.segmentationR    t   skimage._shared.testingR   R   R   R   R   R    (    (    (    sI   lib/python2.7/site-packages/skimage/segmentation/tests/test_quickshift.pyt   <module>   s   "