ó
¦–Õ\c           @   sß  d  d l  Z  e  j d ƒ d  d l Z d  d l m Z m Z d  d l j Z	 d  d l
 m Z m Z m Z m Z m Z m Z m Z m Z m Z m Z d  d l m Z m Z d „  Z d „  Z d „  Z d	 „  Z d
 „  Z d „  Z d „  Z d „  Z d „  Z  d „  Z! e  j" j# d d d d d d d g ƒ d „  ƒ Z% d „  Z& d „  Z' d „  Z( d „  Z) d „  Z* d „  Z+ e  j" j, d „  ƒ Z- d „  Z. d „  Z/ d  „  Z0 d! „  Z1 d" „  Z2 d# „  Z3 e  j" j# d d d d d g ƒ d$ „  ƒ Z4 d S(%   iÿÿÿÿNt   numpy(   t   assert_array_almost_equalt   assert_array_equal(
   t   fractional_slicet   getitemt   trim_internalt   overlap_internalt   nearestt   constantt
   boundariest   reflectt   periodict   overlap(   t	   assert_eqt	   same_keysc          C   s  t  d i d d 6ƒ t d t d d ƒ f f k s7 t ‚ t  d i d d 6d d 6ƒ t d t d  d  d  ƒ t d	 d  ƒ f f k s„ t ‚ t  d i d d 6d d 6ƒ t d t d d ƒ t d	 d  ƒ f f k sÎ t ‚ t  d i d d 6ƒ }  t |  d d t ƒ st ‚ d  S(   Nt   xgš™™™™™@i   i    i   i   gffffff@i   iýÿÿÿg333333@(   R   gš™™™™™@(   R   i   (   R   i   gffffff@(   R   i   i   (   R   g333333@gffffff@(   R   i   i   (   R   gš™™™™™@(   R   R   t   slicet   AssertionErrort   Nonet
   isinstancet   int(   t   fs(    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyt   test_fractional_slice   s    $30c          C   sO  t  j d ƒ j dE ƒ }  t j |  d dF ƒ} t | i d d 6d d 6ƒ } | j d d	 ƒ } | j dG dH f k sw t ‚ t  j	 d d d d d d d d d
 d g
 d d d d d d d d d d g
 d d d d d d d d d d g
 d d d d  d! d  d! d" d# d$ g
 d% d& d' d( d) d( d) d* d+ d, g
 d- d. d/ d0 d1 d0 d1 d2 d3 d4 g
 d d d d d d d d d d g
 d d d d  d! d  d! d" d# d$ g
 d% d& d' d( d) d( d) d* d+ d, g
 d- d. d/ d0 d1 d0 d1 d2 d3 d4 g
 d5 d6 d7 d8 d9 d8 d9 d: d; d< g
 d= d> d? d@ dA d@ dA dB dC dD g
 g ƒ } t
 | | ƒ t t | i d d 6d d 6ƒ | ƒ sKt ‚ d  S(I   Ni@   i   t   chunksi   i   i    i   t	   schedulert   synci   i   i   i   i	   i
   i   i   i   i   i   i   i   i   i   i   i   i   i   i   i   i   i   i   i   i   i   i    i!   i"   i#   i$   i%   i&   i'   i(   i)   i*   i+   i,   i-   i.   i/   i0   i1   i2   i3   i4   i5   i6   i7   i8   i9   i:   i;   i<   i=   i>   i?   (   i   i   (   i   i   (   i   i   (   i   i   (   t   npt   aranget   reshapet   dat
   from_arrayR   t   computeR   R   t   arrayR   R   (   R   t   dt   gt   resultt   expected(    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyt   test_overlap_internal   s(    !!!!!!!!!!!*c          C   sT   t  j d d d ƒ}  t |  d i d d 6d d 6ƒ} | j d d f k sP t ‚ d  S(   Ni(   i<   R   i
   t   axesi   i    i   i   i   (   i(   i<   (   i
   i
   (   i   i   i   i   (   i   i   i   i   i   i   (   R   t   onesR   R   R   (   R!   t   e(    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyt   test_trim_internal7   s     c          C   sç   t  j d ƒ j d ƒ }  t j |  d d ƒ} t | d d d d ƒ} | j d | j d d k si t ‚ | j d	 | j d	 k s‰ t ‚ t | d	 d  d  … f | d
 d  d  … f ƒ t | d d  d  … f | d d  d  … f ƒ d  S(   Ni@   i   R   i   t   axisi    t   depthi   i   iÿÿÿÿiþÿÿÿ(   i   i   (   i   i   (	   R   R   R   R   R   R   t   shapeR   R   (   R   R!   R(   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyt   test_periodic>   s    $ -c          C   sÞ   t  j d ƒ }  t j |  d d ƒ} t | d d d d ƒ} t  j d d d d d d	 d
 d d d d d d d g ƒ } t | | ƒ t | d d d d ƒ} t  j d d d d d	 d
 d d d d d d g ƒ } t | | ƒ d  S(   Ni
   R   i   R*   i    R+   i   i   i   i   i   i   i   i	   (   i   i   (   R   R   R   R   R
   R    R   (   R   R!   R(   R$   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyt   test_reflectJ   s    93c          C   sÞ   t  j d ƒ }  t j |  d d ƒ} t | d d d d ƒ} t  j d d d d d d	 d
 d d d d d d d g ƒ } t | | ƒ t | d d d d ƒ} t  j d d d d d	 d
 d d d d d d g ƒ } t | | ƒ d  S(   Ni
   R   i   R*   i    R+   i   i   i   i   i   i   i   i	   (   i   i   (   R   R   R   R   R   R    R   (   R   R!   R(   R$   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyt   test_nearestW   s    93c          C   sù   t  j d ƒ j d ƒ }  t j |  d d ƒ} t | d d d d d	 d
 ƒ} | j d | j d d k so t ‚ | j d | j d k s t ‚ t | d d  d  … f t  j	 d d |  j
 ƒd
 ƒ t | d d  d  … f t  j	 d d |  j
 ƒd
 ƒ d  S(   Ni@   i   R   i   R*   i    R+   i   t   valuei
   i   t   dtypeiÿÿÿÿ(   i   i   (   i   i   (   R   R   R   R   R   R   R,   R   R   R'   R1   (   R   R!   R(   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyt   test_constantd   s    $ 3c          C   s  t  j d ƒ j dD ƒ }  t j |  d dE ƒ} t | i d d 6d d 6i d d 6d d 6ƒ } t  j d d d d d d d d d d g
 d d d d d d d d d d g
 d	 d d d d
 d d d d	 d g
 d d d d d d d d d d g
 d d d d d d d d d d g
 d d d d d  d! d" d# d d g
 d$ d% d& d' d( d) d* d+ d$ d% g
 d, d- d. d/ d0 d1 d2 d3 d, d- g
 d4 d5 d6 d7 d8 d9 d: d; d4 d5 g
 d< d= d> d? d@ dA dB dC d< d= g
 d d d d d d d d d d g
 d d d d d d d d d d g
 g ƒ } t | | ƒ d  S(F   Ni@   i   R   i   i   i    i   R   i   i   i   i   i   i	   i
   i   i   i   i   i   i   i   i   i   i   i   i   i   i   i   i   i   i   i   i   i'   i    i!   i"   i#   i$   i%   i&   i/   i(   i)   i*   i+   i,   i-   i.   i7   i0   i1   i2   i3   i4   i5   i6   i?   i8   i9   i:   i;   i<   i=   i>   (   i   i   (   i   i   (   R   R   R   R   R   R	   R    R   (   R   R!   R(   R$   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyt   test_boundariesp   s"    .!!!!!!!!!!!*c          C   sÊ  t  j d ƒ j dH ƒ }  t j |  d dI ƒ} t | d i d d 6d d 6d	 i d
 d 6d d 6ƒ} | j dJ dK f k s| t ‚ t  j d
 d
 d
 d
 d
 d
 d
 d
 d
 d
 d
 d
 g d
 d
 d
 d
 d
 d
 d
 d
 d
 d
 d
 d
 g d d d d d d d d d d d d g d d d d d d d d d d d d g d d d d d d d d d d d d g d d d  d! d" d# d" d# d$ d% d& d& g d' d' d( d) d* d+ d* d+ d, d- d. d. g d/ d/ d0 d1 d2 d3 d2 d3 d4 d5 d6 d6 g d d d d d d d d d d d d g d d d  d! d" d# d" d# d$ d% d& d& g d' d' d( d) d* d+ d* d+ d, d- d. d. g d/ d/ d0 d1 d2 d3 d2 d3 d4 d5 d6 d6 g d7 d7 d8 d9 d: d; d: d; d< d= d> d> g d? d? d@ dA dB dC dB dC dD dE dF dF g d
 d
 d
 d
 d
 d
 d
 d
 d
 d
 d
 d
 g d
 d
 d
 d
 d
 d
 d
 d
 d
 d
 d
 d
 g g ƒ } t	 | | ƒ t
 | t | d i d d 6d d 6d	 i d
 d 6d d 6ƒƒ sKt ‚ t | d i d d 6d d 6d	 i d
 d 6dG d 6ƒ} t  j d
 d
 d
 d
 d
 d
 d
 d
 d
 d
 g
 d
 d
 d
 d
 d
 d
 d
 d
 d
 d
 g
 d d d d d d d d d d g
 d d d d d d d d d d g
 d d d d d d d d d d g
 d d  d! d" d# d" d# d$ d% d& g
 d' d( d) d* d+ d* d+ d, d- d. g
 d/ d0 d1 d2 d3 d2 d3 d4 d5 d6 g
 d d d d d d d d d d g
 d d  d! d" d# d" d# d$ d% d& g
 d' d( d) d* d+ d* d+ d, d- d. g
 d/ d0 d1 d2 d3 d2 d3 d4 d5 d6 g
 d7 d8 d9 d: d; d: d; d< d= d> g
 d? d@ dA dB dC dB dC dD dE dF g
 d
 d
 d
 d
 d
 d
 d
 d
 d
 d
 g
 d
 d
 d
 d
 d
 d
 d
 d
 d
 d
 g
 g ƒ } t	 | | ƒ | j dL dM f k sÆt ‚ d  S(N   Ni@   i   R   i   R+   i   i    i   t   boundaryid   R
   i   i   i   i   i	   i
   i   i   i   i   i   i   i   i   i   i   i   i   i   i   i   i   i   i   i   i   i   i    i!   i"   i#   i$   i%   i&   i'   i(   i)   i*   i+   i,   i-   i.   i/   i0   i1   i2   i3   i4   i5   i6   i7   i8   i9   i:   i;   i<   i=   i>   i?   t   none(   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$   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyt   test_overlap†   sX    4'''''''''''''''0# 4!!!!!!!!!!!!!!!*c          C   sê  t  j d d d ƒ}  |  j d „  d d d |  j ƒ} t | t j d ƒ d d d ƒ t  j d d d ƒ}  |  j d „  d t j d ƒ d |  j ƒ} t g  | j D] } t	 | ƒ t
 k ^ q¤ ƒ sË t ‚ t | t j d ƒ d d d ƒ t j d	 ƒ j d ƒ }  t  j |  d d ƒ} | j d „  d d d | j ƒ} | j d „  d i d d 6d d 6d i d d 6d d 6d | j ƒ} | j d „  d i d d 6d i d d 6d | j ƒ} t | |  d	 ƒ t | |  d ƒ t | |  d ƒ d  S(   Ni
   R   i   c         S   s   |  t  |  ƒ S(   N(   t   len(   R   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyt   <lambda>¸   t    R+   i   R1   c         S   s   |  t  |  ƒ S(   N(   R7   (   R   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyR8   ¼   R9   i   i   c         S   s   |  |  j  S(   N(   t   size(   R   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyR8   Â   R9   i   c         S   s   |  |  j  S(   N(   R:   (   R   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyR8   Ã   R9   i    R4   R
   R5   c         S   s   |  |  j  S(   N(   R:   (   R   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyR8   Å   R9   i   i   (   i   i   (   i   i   (   R   R   t   map_overlapR1   R   R   t   int64t   allR,   t   typeR   R   R   R   (   R   t   yt   sR!   t   exp1t   exp2t   exp3(    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyt   test_map_overlap¶   s"    !"*4"!# R4   R
   R   R   R5   i    c         C   sM   t  j d d d ƒ} | j d „  d d d |  d | j ƒ} t | | ƒ d  S(	   Ni
   R   i   c         S   s   |  S(   N(    (   t   i(    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyR8   ×   R9   R+   i    R4   R1   (   R   R   R;   R1   R   (   R4   R   R?   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyt   test_map_overlap_no_depthÌ   s    	'c          C   s›   t  j d ƒ j d d ƒ j t ƒ }  t j |  d d ƒ} t | d i d d 6d d 6d	 i d
 d 6d
 d 6ƒ} t | i d d 6d d 6ƒ } t	 | |  ƒ d  S(   Ni   i   R   i   R+   i   i    i   R4   R   (   i   i   (
   R   R   R   t   astypet   floatR   R   R   R   R   (   t   at   darrt   garrt   tarr(    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyt   test_nearest_overlapÛ   s    $c          C   s  t  j d ƒ j d d ƒ }  t j |  d d ƒ} i d d 6d d 6} t | d | d	 d
 ƒ} t | d | d	 d ƒ} t | d | d	 d ƒ} t | d | d	 d ƒ} t | | ƒ } t | |  ƒ t | | ƒ } t | |  ƒ t | | ƒ } t | |  ƒ t | | ƒ } t | |  ƒ d  S(   Nid   i
   R   i   i   i    i   R+   R4   R
   R   R   i*   (   i   i   (   R   R   R   R   R   R   R   R   (   R$   RJ   R+   t	   reflectedR   R   R   R#   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyt   test_0_depthå   s    c          C   s  t  j d ƒ j d d ƒ }  t j |  d d ƒ} i d d 6d d 6} t | d | d	 d
 ƒ} t | d | d	 d ƒ} t | d | d	 d ƒ} t | d | d	 d ƒ} t | | ƒ } t | |  ƒ t | | ƒ } t | |  ƒ t | | ƒ } t | |  ƒ t | | ƒ } t | |  ƒ d  S(   Nid   i
   R   i   i   i    i   R+   R4   R
   R   R   i*   (   i   i   (   R   R   R   R   R   R   R   R   (   R$   RJ   R+   RN   R   R   R   R#   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyt   test_some_0_depthý   s    c          C   sj   t  j d ƒ j d d ƒ }  t j |  d d	 d
 f ƒ} t | d d d d ƒ} |  j | j k sf t ‚ d  S(   Ni   i	   R   i   R+   i    R4   i   (   i   (   i   i   i   i   (   R   R   R   R   R   R   R,   R   (   RI   RJ   R"   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyt   test_one_chunk_along_axis  s    c          C   s€   t  j d ƒ j d d ƒ }  t j |  d d d	 f ƒ} t | i d d 6d d 6i d d 6d d 6ƒ } | j | j k s| t ‚ d  S(
   Ni   i	   R   i   i   i    i	   (   i   (   i   i   i   i   (   R   R   R   R   R   R	   R   R   (   RI   RJ   t   b(    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyt   test_constant_boundaries  s    .c          C   s  t  j d ƒ j d d ƒ }  t j |  d d ƒ} i d d 6d d 6} t | d | d d	 ƒ} t | d | d d
 ƒ} t | d | d d ƒ} t | d | d d ƒ} t | | ƒ } t | |  ƒ t | | ƒ } t | |  ƒ t | | ƒ } t | |  ƒ t | | ƒ } t | |  ƒ d  S(   Nid   i
   R   i   i    i   R+   R4   R
   R   R   i*   (   i   i   (   R   R   R   R   R   R   R   R   (   R$   RJ   R+   RN   R   R   R   R#   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyt!   test_depth_equals_boundary_length#  s    c          C   s  t  j d ƒ j d d ƒ }  t j |  d d ƒ} i d d 6d d 6} t | d	 | d
 d ƒ} t | d	 | d
 d ƒ} t | d	 | d
 d ƒ} t | d	 | d
 d ƒ} t | | ƒ } t | |  ƒ t | | ƒ } t | |  ƒ t | | ƒ } t | |  ƒ t | | ƒ } t | |  ƒ d  S(   Nid   i
   R   i   i   i    i   i   R+   R4   R
   R   R   i*   (   i   i   (   R   R   R   R   R   R   R   R   (   R$   RJ   R+   RN   R   R   R   R#   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyt'   test_depth_greater_than_boundary_length;  s    c          C   sg   t  j d ƒ j d d ƒ }  t j |  d d ƒ} i d d 6d d 6} t j t t | d	 | d
 d ƒd  S(   Ni   i   R   i   i   i    i   i   R+   R4   (   i   i   (	   R   R   R   R   R   t   pytestt   raisest
   ValueErrorR   (   R$   RJ   R+   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyt   test_bad_depth_raisesT  s    c          C   sÖ   t  j t j d ƒ j d d ƒ d d ƒ}  t |  d i d d 6d d 6ƒ } t j d d d d d d	 d d g d d d d
 d d d d g d d d d d d d d g d d d d d d d d g g ƒ } t | | ƒ d  S(   Ni   i   R   i   R5   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   t   expt   res(    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyt   test_none_boundaries]  s    * $c          C   s…   t  j d d d ƒ}  |  j d „  d d ƒ} t | j ƒ d k  sH t ‚ |  j d „  d d d	 d
 ƒ} t | j ƒ d k  s t ‚ d  S(   Ni
   R   i   c         S   s   |  S(   N(    (   R   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyR8   k  R9   R+   i   iÈ   c         S   s   |  S(   N(    (   R   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyR8   n  R9   R4   R5   id   (   i
   i
   (   i   i   (   R   R'   R;   R7   t   daskR   (   R   R?   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyt   test_overlap_smallh  s
    c       
      sï   t  j j d ƒ t  j j d d d ƒ}  d „  } g  t |  j ƒ D]X ‰  |  j d „  d |  j d ‡  f d	 †  t |  j ƒ Dƒ d
 d ƒj | d |  j ƒ^ qA } t | d j	 ƒ t | d j	 ƒ @t |  j	 ƒ k sØ t
 ‚ t  j d d | Œ d  S(   Ni    i	   R   i   c         S   s   |  j  d k s t ‚ |  S(   Ni   (   i   i   (   R,   R   (   R   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyt   checkv  s    c         S   s   |  d S(   Ni   (    (   RI   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyR8   z  R9   R1   R+   c            s%   i  |  ] } t  ˆ  | k ƒ | “ q S(    (   R   (   t   .0t   j(   RE   (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pys
   <dictcomp>|  s   	 R4   R5   i   R   s   single-threaded(   i	   i	   (   i   i   (   R   t   randomt   seedt   ranget   ndimR;   R1   t
   map_blockst   setR]   R   R   (   RI   R_   t   r(    (   RE   s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyt)   test_no_shared_keys_with_different_depthsr  s    	k9c          C   s^  t  j d d d ƒ}  |  j d „  d i d d 6ƒ} t |  | ƒ t d „  | j Dƒ ƒ s` t ‚ t d	 „  | j Dƒ ƒ s t ‚ |  j d
 „  d i d d 6ƒ} t |  | ƒ t d „  | j Dƒ ƒ sÊ t ‚ t d „  | j Dƒ ƒ sé t ‚ |  j d „  d i d d 6d d 6ƒ} t |  | ƒ t d „  | j Dƒ ƒ s;t ‚ t d „  | j Dƒ ƒ sZt ‚ d  S(   Ni   R   i
   c         S   s   |  S(   N(    (   R   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyR8   ‡  R9   R+   i   i    c         s   s"   |  ] } t  | d  t ƒ Vq d S(   i   N(   R   RH   (   R`   t   k(    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pys	   <genexpr>‰  s    c         s   s"   |  ] } t  | d  t ƒ Vq d S(   i   N(   R   R   (   R`   Rj   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pys	   <genexpr>Š  s    c         S   s   |  S(   N(    (   R   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyR8   Œ  R9   c         s   s"   |  ] } t  | d  t ƒ Vq d S(   i   N(   R   R   (   R`   Rj   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pys	   <genexpr>Ž  s    c         s   s"   |  ] } t  | d  t ƒ Vq d S(   i   N(   R   RH   (   R`   Rj   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pys	   <genexpr>  s    c         S   s   |  S(   N(    (   R   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyR8   ‘  R9   c         s   s"   |  ] } t  | d  t ƒ Vq d S(   i   N(   R   RH   (   R`   Rj   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pys	   <genexpr>“  s    c         s   s"   |  ] } t  | d  t ƒ Vq d S(   i   N(   R   RH   (   R`   Rj   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pys	   <genexpr>”  s    (   i   i   (   i
   i
   (   R   R'   R;   R   t   anyR]   R   R=   (   R   RI   RR   t   c(    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyt!   test_overlap_few_dimensions_small„  s    &c          C   sí   t  j d
 d d ƒ}  |  j d „  d i d d 6ƒ} |  j d „  d i d d 6ƒ} |  j d	 „  d i d d 6d d 6ƒ} t | j ƒ t | j ƒ k s t ‚ t | j ƒ t | j ƒ k  sÁ t ‚ t | j ƒ d t | j ƒ k  sé t ‚ d  S(   Nid   R   i
   c         S   s   |  S(   N(    (   R   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyR8   š  R9   R+   i   i    c         S   s   |  S(   N(    (   R   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyR8   ›  R9   c         S   s   |  S(   N(    (   R   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyR8   œ  R9   (   id   id   (   i
   i
   (   R   R'   R;   R7   R]   R   (   R   RI   RR   Rl   (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyt   test_overlap_few_dimensions—  s    &$$c         C   sK  t  j t j d ƒ j d d ƒ d d ƒ} t  j j | d d i d d	 6|  d
 6ƒ} t  j j | d d i d d	 6|  d
 6ƒ} t j | | k ƒ s— t ‚ t  j j | d d i |  d
 6ƒ} t  j j | d d i |  d
 6ƒ} t j | | k ƒ sö t ‚ t  j j | d d |  ƒ} t  j j | d d |  ƒ} t j | | k ƒ sGt ‚ d  S(   Ni   i   i   R   i   i   R4   R
   i    i   (   i   i   (	   R   R   R   R   R   R   t   trim_overlapR=   R   (   R4   R   t   x_overlapedt	   x_trimmed(    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyt   test_trim_boundry¤  s    *(5   RV   t   importorskipR    R   t   numpy.testingR   R   t
   dask.arrayR    R   t   dask.array.overlapR   R   R   R   R   R   R	   R
   R   R   t   dask.array.utilsR   R   R   R%   R)   R-   R.   R/   R2   R3   R6   RD   t   markt   parametrizeR   RF   RM   RO   RP   RQ   RS   RT   t   xfailRU   RY   R\   R^   Ri   Rm   Rn   Rr   (    (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_overlap.pyt   <module>   sL   F									0			
									
			