ó
¦–Õ\c           @   s+  d  d l  Z  d  d l Z d  d l Z d  d l j Z d  d l m Z e j	 d ƒ Z
 e
 rn e j	 d d d ƒn  d „  d „  d	 „  d
 „  d „  d „  d „  d „  d „  d „  d „  d „  d „  d „  d „  e j d „  d e j j d d ƒ ƒe j d „  d e j j d d ƒ ƒd „  d „  d „  d „  d „  d  „  d! „  d" „  d# „  d$ „  d% „  d& „  d' „  d( „  g Z e j j d) e ƒ d* „  ƒ Z d+ „  Z e j j d d, d- e ƒ e j j d) e ƒ d. „  ƒ ƒ Z e j j d d, d- e ƒ e j j d) e ƒ d/ „  ƒ ƒ Z e j j d d, d- e ƒ d0 „  ƒ Z d S(1   iÿÿÿÿN(   t	   assert_eqt   sparset   numbat
   minversions   0.40.0c         C   s   |  S(   N(    (   t   x(    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyt   <lambda>   t    c         C   s   t  j |  ƒ S(   N(   t   dat   expm1(   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR      R   c         C   s   d |  S(   Ni   (    (   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR      R   c         C   s   |  d S(   Ni   (    (   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR      R   c         C   s   |  d S(   Ni   (    (   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR      R   c         C   s   |  |  S(   N(    (   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR      R   c         C   s   |  |  S(   N(    (   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR      R   c         C   s   |  d S(   Ni    (    (   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR      R   c         C   s   |  d  d  … d f S(   Ni   (    (   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR      R   c         C   s   |  d  d … d  d d … f S(   Ni   i   (   t   None(   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR      R   c         C   s   |  j  S(   N(   t   T(   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR      R   c         C   s   t  j |  d ƒ S(   Ni   i   i    (   i   i   i    (   R   t	   transpose(   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR      R   c         C   s
   |  j  ƒ  S(   N(   t   sum(   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR      R   c         C   s
   |  j  ƒ  S(   N(   t   mean(   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR      R   c         C   s   |  j  d d ƒ S(   Nt   orderi    (   t   moment(   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR       R   c         C   s
   |  j  ƒ  S(   N(   t   std(   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR   !   R   t   markst   reasons2   fixed in https://github.com/pydata/sparse/pull/243c         C   s
   |  j  ƒ  S(   N(   t   var(   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR   #   R   c         C   s   |  j  t j |  j d ƒ ƒ S(   Niÿÿÿÿ(   t   dott   npt   aranget   shape(   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR   %   R   c         C   s   |  j  t j |  j d ƒ ƒ S(   Niÿÿÿÿ(   R   R   t   eyeR   (   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR   &   R   c         C   s,   t  j |  t j |  j d  ƒ d d d g ƒS(   Ni   t   axesi    i   (   i    i   (   i    i   (   R   t	   tensordotR   t   onesR   (   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR   '   R   c         C   s   |  j  d d ƒ S(   Nt   axisi    (   R   (   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR   (   R   c         C   s   |  j  d d ƒ S(   NR   i    (   t   max(   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR   )   R   c         C   s   |  j  d d ƒ S(   NR   i   i   (   i   i   (   R   (   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR   *   R   c         C   s   |  j  t j ƒ S(   N(   t   astypeR   t
   complex128(   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR   +   R   c         C   s   |  j  d „  ƒ S(   Nc         S   s   |  d S(   Ni   (    (   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR   ,   R   (   t
   map_blocks(   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR   ,   R   c         C   s   |  j  d ƒ S(   Ni   (   t   round(   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR   -   R   c         C   s,   |  j  |  j d |  j d |  j d f ƒ S(   Ni    i   i   (   t   reshapeR   (   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR   .   R   c         C   s
   t  |  ƒ S(   N(   t   abs(   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR   /   R   c         C   s
   |  d k S(   Ng      à?(    (   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR   0   R   c         C   s   |  j  d ƒ S(   Ni   (   i   i   i   (   t   rechunk(   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR   1   R   c         C   s   |  j  d ƒ S(   Ni   i   (   i   i   i   (   R$   (   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR   2   R   t   funcc         C   sÁ   t  j j d d d	 ƒ} d | | d k  <| j t j j ƒ } |  | ƒ } |  | ƒ } t | | ƒ | j r½ | j ƒ  } t	 | t j ƒ s½ | d k j
 ƒ  t j | j ƒ d k sº t ‚ q½ n  d  S(
   Ni   i   i   t   chunksi   i    gš™™™™™é?(   i   i   i   (   i   i   i   (   R   t   randomR    R   t   COOt
   from_numpyR    R   t   computet
   isinstanceR   R   t   prodt   AssertionError(   R%   R   t   yt   xxt   yyt   zz(    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyt
   test_basic6   s    	c       	   C   s  t  j j d	 d d
 ƒ}  d |  |  d k  <t  j j d d d ƒ} d | | d k  <|  j t j j ƒ } | j t j j ƒ } t t  j |  | d d ƒt  j | | d d ƒƒ t t  j |  | d d ƒt  j | | d d ƒƒ t t  j |  | d d d f ƒt  j | | d d d f ƒƒ d  S(   Ni   i   i   R&   i   i    gš™™™™™é?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    (   R   R'   R    R   R(   R)   R    R   (   R   R.   R/   R0   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyt   test_tensordotH   s    s   upstream changet   strictc         C   s´   t  j j d	 d d
 ƒ} t  j j d d d ƒ} d | | d k  <| j t j j ƒ } t  j | | g d d ƒ} t  j | | g d d ƒ} |  | ƒ } |  | ƒ } t | | ƒ d  S(   Ni   i   i   R&   i   i    gš™™™™™é?R   (   i   i   i   (   i   i   i   (   i   i   i   (   i   i   i   (   R   R'   R    R   R(   R)   t   concatenateR    (   R%   R   R.   R0   t   dt   st   ddt   ss(    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyt   test_mixed_concatenateY   s    c         C   si   t  j j d	 d d
 ƒ} d | | d k  <d „  } | j | ƒ } |  | ƒ } |  | ƒ } t | | ƒ d  S(   Ni   i   R&   i   i   i    gffffffæ?c         S   s&   t  j  ƒ  d k  r" t j j |  ƒ S|  S(   Ng      à?(   R'   R   R(   R)   (   R   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyR   q   R   (   i   i   i   (   i   i   i   (   R   R'   R    R    (   R%   R6   t   fnR7   R8   R9   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyt   test_mixed_randomk   s    	c          C   sÈ   t  j j d	 d d
 ƒ}  d |  |  d k  <|  j t j j ƒ }  t  j d d d ƒ} t  j | |  g d d ƒ} | j d k s‚ t	 ‚ | j
 ƒ  } t | t j ƒ s¦ t	 ‚ | j |  j
 ƒ  j k sÄ t	 ‚ d  S(   Ni
   R&   i   i    gš™™™™™é?i   R   i   (   i
   i
   (   i   i   (   i
   i   (   i   i   (   i
   i   (   R   R'   R    R   R(   R)   t   zerosR5   R   R-   R*   R+   t   nnz(   R.   R   t   zR1   (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyt   test_mixed_output_typez   s    (   R'   t   numpyR   t   pytestt
   dask.arrayt   arrayR   t   dask.array.utilsR    t   importorskipR   t   paramt   markt   xfailt	   functionst   parametrizeR2   R3   t   FalseR:   R<   R@   (    (    (    s;   lib/python2.7/site-packages/dask/array/tests/test_sparse.pyt   <module>   s^   	!!