ó
ŚŐ\c           @` s   d  d l  m Z m Z m Z d  d l Z e j d  d  d l Z d  d l m	 Z	 m
 Z
 d  d l j Z d   Z d   Z d   Z d   Z d	   Z d S(
   i    (   t   absolute_importt   divisiont   print_functionNt   numpy(   t   coarsent   keepdims_wrapperc          C` s?  d  d  }  t |   } | |  k s* t  t j d  j d d d d  } |  |  } | | d t } | | d t } | j d k s t  | j	 t
   k s¨ t  | d	 k sş t  | j d k sĎ t  | j	 d
 k sä t  | d	 k j   sü t  | j d k st  | j	 t
   k s)t  | d	 k s;t  d  S(   Nc         S` s   |  j  d |  S(   Nt   axis(   t   sum(   t   aR   (    (    s:   lib/python2.7/site-packages/dask/array/tests/test_chunk.pyt   summer   s    i   i   i   i   i   t   keepdimsi    i  (   i   i   i   i   (   t   NoneR   t   AssertionErrort   npt   aranget   reshapet   Truet   Falset   ndimt   shapet   tuplet   all(   R	   t   summer_wrappedR   t   rt   rwt   rwf(    (    s:   lib/python2.7/site-packages/dask/array/tests/test_chunk.pyt   test_keepdims_wrapper_no_axis   s     !c          C` sŰ  d  d  }  t |   } | |  k s* t  t j d  j d d d d  } |  | d d } | | d d d t } | | d d d t } | j d k s˘ t  | j	 d k sˇ t  | t j
 d	 d
 d d g d d d d g g g  k j   sů t  | j d k st  | j	 d k s#t  | t j
 d	 d
 d d g g d d d d g g g g  k j   skt  | j d k st  | j	 d k st  | t j
 d	 d
 d d g d d d d g g g  k j   s×t  d  S(   Nc         S` s   |  j  d |  S(   NR   (   R   (   R   R   (    (    s:   lib/python2.7/site-packages/dask/array/tests/test_chunk.pyR	   (   s    i   i   i   i   i   R   R
   i   i   i   i   i0   i3   i6   i9   (   i   i   i   (   i   i   i   i   (   i   i   i   (   R   R   R   R   R   R   R   R   R   R   t   arrayR   (   R	   R   R   R   R   R   (    (    s:   lib/python2.7/site-packages/dask/array/tests/test_chunk.pyt   test_keepdims_wrapper_one_axis'   s     !BHc          C` s˘  d  d  }  t |   } | |  k s* t  t j d  j d d d d  } |  | d d } | | d d d t } | | d d d t } | j d k s˘ t  | j	 d k sˇ t  | t j
 d	 d
 d g g  k j   sä t  | j d k sů t  | j	 d k st  | t j
 d	 g d
 g d g g g g  k j   sGt  | j d k s\t  | j	 d k sqt  | t j
 d	 d
 d g g  k j   st  d  S(   Nc         S` s   |  j  d |  S(   NR   (   R   (   R   R   (    (    s:   lib/python2.7/site-packages/dask/array/tests/test_chunk.pyR	   C   s    i   i   i   i   i   R   R
   i<   i\   i|   (   i   i   (   i   i   (   i   i   (   i   i   (   i   i   i   i   (   i   i   (   R   R   R   R   R   R   R   R   R   R   R   R   (   R	   R   R   R   R   R   (    (    s:   lib/python2.7/site-packages/dask/array/tests/test_chunk.pyt   test_keepdims_wrapper_two_axesB   s     !-9c          C` s   t  j j d d d
 }  t t  j |  i d d 6d d 6 } | j d k sP t  | d t  j |  d  d  d  d  f  k s t  d  S(   Ni
   t   sizei   i   i    i   i   i   i   (   i   i   (   i   i   (   i    i    (   R   t   randomt   randintR   R   R   R   (   t   xt   y(    (    s:   lib/python2.7/site-packages/dask/array/tests/test_chunk.pyt   test_coarsen]   s    #c           C` s7   t  j d d d j d  j d d	 f k s3 t  d  S(
   Ni   i   t   chunksi   i   i   (   i   i   (   i   i   (   i   i   (   t   dat   zerost   rechunkR$   R   (    (    (    s:   lib/python2.7/site-packages/dask/array/tests/test_chunk.pyt   test_integer_inputn   s    (   t
   __future__R    R   R   t   pytestt   importorskipR   R   t   dask.array.chunkR   R   t
   dask.arrayR   R%   R   R   R   R#   R(   (    (    (    s:   lib/python2.7/site-packages/dask/array/tests/test_chunk.pyt   <module>   s   				