ó
ĤĠ\c           @   s   d  d l  m Z d  d l Z d  d l Z e j d  d  d l m Z d  d l Z	 d  d l
 m Z d  d l m Z e d    Z d   Z d	   Z d
   Z d S(   i˙˙˙˙(   t   contextmanagerNt   skimage(   t   imread(   t   imsave(   t   tmpdirc      
   c   s   t    | } x^ t |   D]P } t j j | d |  } t j j d d d | j d  } t	 | |  q Wt j j | d  VWd  QXd  S(   Ns   image.%d.pngi    i˙   t   sizet   u1s   *.png(
   R   t   ranget   ost   patht   joint   npt   randomt   randintt   astypeR   (   t   nt   shapet   dirnamet   it   fnt   x(    (    s:   lib/python2.7/site-packages/dask/array/tests/test_image.pyt   random_images   s    $c       	   C   sİ   t  d d   }  t |   } | j d k s3 t  | j d	 d
 d d f k sT t  | j d k si t  | j   j d k s t  | j   j d k s t  Wd  QXd  S(   Ni   i   i   i   i   t   uint8(   i   i   i   (   i   i   i   i   (   i   i   i   i   (   i   (   i   (   i   (   i   i   i   i   (   R   t	   da_imreadR   t   AssertionErrort   chunkst   dtypet   compute(   t
   globstringt   im(    (    s:   lib/python2.7/site-packages/dask/array/tests/test_image.pyt   test_imread   s    !c       
   C   sd   d   }  t  d d
  F } t | d |  } | j   t j d d d	 k j   sZ t  Wd  QXd  S(   Nc         S   s   t  j d d d S(   Ni   i   i   R   t   i1(   i   i   i   (   R   t   ones(   R   (    (    s:   lib/python2.7/site-packages/dask/array/tests/test_image.pyt   imread2$   s    i   i   i   i   R   i   R   R   (   i   i   i   (   i   i   i   i   (   R   R   R   R   R    t   allR   (   R!   R   R   (    (    s:   lib/python2.7/site-packages/dask/array/tests/test_image.pyt    test_imread_with_custom_function#   s    	c       	   C   sd   d   }  t  d d  F } t | d |  } | j   t j d	 d d k j   sZ t  Wd  QXd  S(
   Nc         S   s$   d |  (|  d  d   d  d   d f S(   Ni   i    (    (   R   (    (    s:   lib/python2.7/site-packages/dask/array/tests/test_image.pyt
   preprocess,   s    i   i   i   R$   R   R   (   i   i   i   (   i   i   i   (   R   R   R   R   R    R"   R   (   R$   R   R   (    (    s:   lib/python2.7/site-packages/dask/array/tests/test_image.pyt   test_preprocess+   s    	(   t
   contextlibR    R   t   pytestt   importorskipt   dask.array.imageR   R   t   numpyR   t
   skimage.ioR   t
   dask.utilsR   R   R   R#   R%   (    (    (    s:   lib/python2.7/site-packages/dask/array/tests/test_image.pyt   <module>   s   		