ó
¦–Õ\c           @   sà  d  d l  Z  d  d l Z d  d l j Z d  d l m Z m Z e Z	 d Z
 e  j j e	 d e
 ƒe  j j d d „  d „  d „  d	 „  d
 „  d „  d „  d „  d „  d „  d „  d „  d „  g ƒ d „  ƒ ƒ Z e  j j e	 d e
 ƒe  j j d e j j e j j g ƒ d „  ƒ ƒ Z e  j j e	 d e
 ƒe  j j d d „  d „  d „  g ƒ d „  ƒ ƒ Z e  j j e	 d e
 ƒd „  ƒ Z e  j j e	 d e
 ƒe  j j d d d e ƒ d „  ƒ ƒ Z e  j j e	 d e
 ƒd „  ƒ Z d S(   iÿÿÿÿN(   t	   assert_eqt   IS_NEP18_ACTIVEs(   NEP-18 support is not available in NumPyt   reasont   funcc         C   s   t  j |  |  |  g ƒ S(   N(   t   npt   concatenate(   t   x(    (    sC   lib/python2.7/site-packages/dask/array/tests/test_array_function.pyt   <lambda>   t    c         C   s   t  j |  |  ƒ S(   N(   R   t   cov(   R   (    (    sC   lib/python2.7/site-packages/dask/array/tests/test_array_function.pyR      R   c         C   s   t  j |  |  ƒ S(   N(   R   t   dot(   R   (    (    sC   lib/python2.7/site-packages/dask/array/tests/test_array_function.pyR      R   c         C   s   t  j |  ƒ S(   N(   R   t   dstack(   R   (    (    sC   lib/python2.7/site-packages/dask/array/tests/test_array_function.pyR      R   c         C   s   t  j |  d d ƒS(   Nt   axisi    (   R   t   flip(   R   (    (    sC   lib/python2.7/site-packages/dask/array/tests/test_array_function.pyR      R   c         C   s   t  j |  ƒ S(   N(   R   t   hstack(   R   (    (    sC   lib/python2.7/site-packages/dask/array/tests/test_array_function.pyR      R   c         C   s   t  j |  |  ƒ S(   N(   R   t   matmul(   R   (    (    sC   lib/python2.7/site-packages/dask/array/tests/test_array_function.pyR      R   c         C   s   t  j |  ƒ S(   N(   R   t   mean(   R   (    (    sC   lib/python2.7/site-packages/dask/array/tests/test_array_function.pyR      R   c         C   s   t  j |  |  g ƒ S(   N(   R   t   stack(   R   (    (    sC   lib/python2.7/site-packages/dask/array/tests/test_array_function.pyR      R   c         C   s   t  j |  ƒ S(   N(   R   t   sum(   R   (    (    sC   lib/python2.7/site-packages/dask/array/tests/test_array_function.pyR      R   c         C   s   t  j |  ƒ S(   N(   R   t   var(   R   (    (    sC   lib/python2.7/site-packages/dask/array/tests/test_array_function.pyR      R   c         C   s   t  j |  ƒ S(   N(   R   t   vstack(   R   (    (    sC   lib/python2.7/site-packages/dask/array/tests/test_array_function.pyR      R   c         C   s   t  j j |  ƒ S(   N(   R   t   linalgt   norm(   R   (    (    sC   lib/python2.7/site-packages/dask/array/tests/test_array_function.pyR      R   c         C   sh   t  j j d ƒ } t j | d d ƒ} |  | ƒ } |  | ƒ } t | t j ƒ sW t ‚ t | | ƒ d  S(   Nid   t   chunksi2   (   id   id   (   i2   i2   (   R   t   randomt   dat
   from_arrayt
   isinstancet   Arrayt   AssertionErrorR    (   R   R   t   yt   res_xt   res_y(    (    sC   lib/python2.7/site-packages/dask/array/tests/test_array_function.pyt   test_array_function_dask   s    c         C   sz   t  j j d ƒ } t j | d d ƒ} |  | ƒ } |  | ƒ } |  j d k ri t | t j ƒ si t ‚ n  t | | ƒ d  S(   Nid   R   s   mkl_fft._numpy_fft(   id   id   (   id   id   (	   R   R   R   R   t
   __module__R   R   R   R    (   R   R   R   R   R    (    (    sC   lib/python2.7/site-packages/dask/array/tests/test_array_function.pyt   test_array_function_fft$   s    c         C   s   t  j |  ƒ S(   N(   R   t   min_scalar_type(   R   (    (    sC   lib/python2.7/site-packages/dask/array/tests/test_array_function.pyR   3   R   c         C   s   t  j j |  ƒ S(   N(   R   R   t   det(   R   (    (    sC   lib/python2.7/site-packages/dask/array/tests/test_array_function.pyR   4   R   c         C   s   t  j j |  ƒ S(   N(   R   R   t   eigvals(   R   (    (    sC   lib/python2.7/site-packages/dask/array/tests/test_array_function.pyR   5   R   c         C   sK   t  j j d ƒ } t j | d d ƒ} t j t ƒ  |  | ƒ Wd  QXd  S(   Nid   R   i2   (   id   id   (   i2   i2   (   R   R   R   R   t   pytestt   raisest	   TypeError(   R   R   R   (    (    sC   lib/python2.7/site-packages/dask/array/tests/test_array_function.pyt    test_array_notimpl_function_dask1   s    c          C   sl   t  j d ƒ }  t j j d d d ƒ} d | | d k  <| j |  j ƒ } t t j | ƒ t j | ƒ ƒ d  S(	   Nt   sparseiô  R   id   i    gÍÌÌÌÌÌì?(   iô  iô  (   id   id   (	   R'   t   importorskipR   R   t
   map_blockst   COOR    R   t	   transpose(   R+   R   R   (    (    sC   lib/python2.7/site-packages/dask/array/tests/test_array_function.pyt$   test_array_function_sparse_transpose>   s
    s.   requires sparse support for __array_function__t   strictc          C   s¬   t  j d ƒ }  t j j d ƒ } d | | d k  <t j j d	 ƒ } d | | d k  <|  j | ƒ } |  j | ƒ } t t j | | d d
 ƒt j | | d d ƒj ƒ  ƒ d  S(   NR+   i   i   i   i    gÍÌÌÌÌÌì?t   axes(   i   i   i   (   i   i   i   (   i   i    (   i   i    (   R'   R,   R   R   R.   R    t	   tensordott   todense(   R+   R   R   t   xxt   yy(    (    sC   lib/python2.7/site-packages/dask/array/tests/test_array_function.pyt$   test_array_function_sparse_tensordotI   s    c    	      C   s   t  j d ƒ }  |  j j d ƒ } t j | d d d t ƒ} t j j | ƒ \ } } } t j j | ƒ \ } } } t	 | | ƒ t	 | | ƒ t	 | | ƒ d  S(   Nt   cupyiô  id   R   t   asarray(   iô  id   (   id   id   (
   R'   R,   R   R   R   t   FalseR   t   svdR   R    (	   R8   R   R   t   u_baset   s_baset   v_baset   ut   st   v(    (    sC   lib/python2.7/site-packages/dask/array/tests/test_array_function.pyt   test_array_function_cupy_svdZ   s    (   R'   t   numpyR   t
   dask.arrayt   arrayR   t   dask.array.utilsR    R   t   missing_arrfunc_condt   missing_arrfunc_reasont   markt   skipift   parametrizeR!   t   fftt   fft2R#   R*   R0   t   xfailR:   R7   RB   (    (    (    sC   lib/python2.7/site-packages/dask/array/tests/test_array_function.pyt   <module>   s@   3	!