
\c           @   s<  d  d l  Z  d  d l Z d  d l m Z m Z m Z e  j j d d d d d d f d d d d d  d! f f d" d# d% d' d( d) f f d* d+ d- d/ d0 d1 f f d2 d3 d4 d5 f d6 d7 f d9 f d: d; d< d= f d> d? f dA f dB dC dD dE dF f dG dH dI f dJ dK f f dL dM dN dO f dP dQ f dR dS dT f f dU dV dW dX f dY dZ f d\ f d] d^ d_ d` f da db f dd f de df dg dh f di dj f dl f dm dn dp dr ds dt du f f dv dw dy d{ d| d} d~ f f d d d d d d d f f d d d d d d d f f d d d d f d d f d d d f f g  d    Z	 d   Z
 d   Z d S(   iN(   t   reshape_rechunkt   expand_tuplet   contract_tuples-   inshape,outshape,prechunks,inchunks,outchunksi   i   i   i@   i   i   i    i0   i   i   i,   c         C   s   t  |  | |  \ } } | | k s* t  | | k s< t  t j t t t |    t j t t t |    k s~ t  d  S(   N(   R    t   AssertionErrort   npt   prodt   listt   mapt   len(   t   inshapet   outshapet	   prechunkst   inchunkst	   outchunkst	   result_int
   result_out(    (    s<   lib/python2.7/site-packages/dask/array/tests/test_reshape.pyt   test_reshape_rechunk   s    c           C   sp   t  d d  d k s t  t  d d  d	 k s6 t  t  d
 d  d k sQ t  t  d d  d k sl t  d  S(   Ni   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   (   R   R   (    (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_reshape.pyt   test_expand_tuple"   s    c           C   sp   t  d d  d k s t  t  d	 d  d
 k s6 t  t  d d  d k sQ t  t  d d  d k sl t  d  S(   Ni   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   (   R   R   (    (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_reshape.pyt   test_contract_tuple)   s    (   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   (   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   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   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   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   (   (   i   i0   (   i   i0   (   (   i   i0   (   i   i   (   i   (   i   (   i@   (   i   i   i   (   i   i,   (   (   i   i,   (   i   i0   (   (   i   i0   (   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   (   t   pytestt   numpyR   t   dask.array.reshapeR    R   R   t   markt   parametrizeR   R   R   (    (    (    s<   lib/python2.7/site-packages/dask/array/tests/test_reshape.pyt   <module>   s*   *'9		