ó
¦–Õ\c           @   sm  d  d l  Z  e  j d ƒ Z d  d l Z d  d l j Z d  d l m	 Z	 d  d l
 m Z d  d l Z d  d l m Z e  j j d d i  f d i  f d i e d	 6f g ƒ d
 „  ƒ Z d „  Z e  j j d d d d d d g ƒ d „  ƒ Z e  j j d d i  f d i e d 6f d i  f d i  f d i  f d i  f d i d d 6f d i d  d 6f d i d d 6f g	 ƒ d „  ƒ Z e  j j d e d ƒ ƒ d „  ƒ Z d „  Z e  j j d e j j j d  f e j j j d  f e j j j d! f e j j j d! f e j j j d! f e j j j d! f e j j j  d! f g ƒ e  j j d" d# d$ g ƒ d% „  ƒ ƒ Z! d& „  Z" d' „  Z# d S((   iÿÿÿÿNt   scipy(   t	   assert_eq(   t   Delayed(   t   allcloses   kind, kwargst   skewt   kurtosist   fisherc         C   s•   t  j j d d ƒ } t j | d ƒ } t t j j |  ƒ } t t j |  ƒ } | | |  } | | |  } t	 | | ƒ t
 | t j ƒ s‘ t ‚ d  S(   Nt   sizei   i   i   (   i   i   (   t   npt   randomt   dat
   from_arrayt   getattrt   daskt   arrayt   statsR    R   t
   isinstancet   Arrayt   AssertionError(   t   kindt   kwargst   xt   yt   dfunct   sfunct   expectedt   result(    (    s:   lib/python2.7/site-packages/dask/array/tests/test_stats.pyt   test_measures   s    c       	   C   s‰   t  j j d d ƒ }  t j |  d ƒ } t j t ƒ  t j j	 j
 | d t ƒWd  QXt j t ƒ  t j j	 j | d t ƒWd  QXd  S(   NR   i   i   i   t   bias(   i   i   (   R   R	   R
   R   t   pytestt   raisest   NotImplementedErrorR   R   R   R   t   FalseR   (   R   R   (    (    s:   lib/python2.7/site-packages/dask/array/tests/test_stats.pyt   test_bias_raises   s    R   t	   chisquaret   power_divergencet
   normaltestt   skewtestt   kurtosistestc         C   sš   t  j j d d ƒ } t j | d ƒ } t t j j |  ƒ } t t j |  ƒ } | | ƒ } | | ƒ } t	 | t
 ƒ s{ t ‚ t | j ƒ  | ƒ s– t ‚ d  S(   NR   i   i   (   R   R	   R
   R   R   R   R   R   R    R   R   R   R   t   compute(   R   t   at   a_t	   dask_testt
   scipy_testR   R   (    (    s:   lib/python2.7/site-packages/dask/array/tests/test_stats.pyt   test_one(   s    t	   ttest_indt	   equal_vart   ttest_1sampt	   ttest_reli    t   lambda_t   neymanc   
      C   sã   t  j j d d ƒ } t  j j d d ƒ } t j | d ƒ } t j | d ƒ } t t j j |  ƒ } t t j |  ƒ } t	 j
 d  ƒ ) | | | |  } | | | |  }	 Wd  QXt | t ƒ sÄ t ‚ t | j ƒ  |	 ƒ sß t ‚ d  S(   NR   i   i   (   R   R	   R
   R   R   R   R   R   R    R   t   warnst   NoneR   R   R   R   R'   (
   R   R   R(   t   bR)   t   b_R*   R+   R   R   (    (    s:   lib/python2.7/site-packages/dask/array/tests/test_stats.pyt   test_two9   s    t   ki   c         C   se   t  j j d d ƒ } t j | d ƒ } t j j | |  ƒ } t j j j | |  ƒ } t	 | | ƒ d  S(   NR   i   i   i   (   i   i   (
   R   R	   R
   R   R    R   t   momentR   R   R   (   R8   R   R   R   R   (    (    s:   lib/python2.7/site-packages/dask/array/tests/test_stats.pyt   test_momentsW   s
    c          C   s    g  t  d ƒ D] }  |  t j j d d ƒ ^ q } g  | D] } t j | d d ƒ^ q9 } t j j j | Œ  } t	 j j | Œ  } t
 | j ƒ  | ƒ sœ t ‚ d  S(   Ni   R   i   t   chunksi
   (   i   (   t   rangeR   R	   R
   R   R   R   R   t   f_onewayR    R   R'   R   (   t   it   np_argsR   t   da_argsR   R   (    (    s:   lib/python2.7/site-packages/dask/array/tests/test_stats.pyt
   test_anovaa   s
    2(s   func, nargsi   i   t
   nan_policyt   omitt   raisec      	   C   s.   t  j t ƒ  |  d | d | Œ Wd  QXd  S(   NRB   (   N(   R   R   R   R4   (   t   funct   nargsRB   (    (    s:   lib/python2.7/site-packages/dask/array/tests/test_stats.pyt   test_nan_raisesk   s    c          C   sZ   t  j j d d ƒ }  t j |  d ƒ } t j t ƒ  t j j	 j
 | d d ƒWd  QXd  S(   NR   i   i   R1   t   wrong(   R   R	   R
   R   R   R   t
   ValueErrorR   R   R   R#   (   R(   R)   (    (    s:   lib/python2.7/site-packages/dask/array/tests/test_stats.pyt   test_power_divergence_invalidz   s    c          C   s\   t  j d d d ƒ}  t j t ƒ  } t j j j |  ƒ Wd  QXd t	 | ƒ k sX t
 ‚ d  S(   Ni   R;   s	   7 samples(   i   (   i   (   R
   t   onesR   R   RI   R   R   R   R%   t   strR   (   R(   t   rec(    (    s:   lib/python2.7/site-packages/dask/array/tests/test_stats.pyt   test_skew_raises‚   s    ($   R   t   importorskipR    t   numpyR   t
   dask.arrayR   R
   t   dask.array.utilsR   t   dask.delayedR   t   dask.array.statsR   R   t   markt   parametrizeR    R   R!   R,   R7   R<   R:   RA   R   R/   R0   R%   R   R&   R$   R9   RG   RJ   RN   (    (    (    s:   lib/python2.7/site-packages/dask/array/tests/test_stats.pyt   <module>   sJ   		"	!					"$
	
'	