
&]\c           @` s   d  d l  m Z m Z m Z d  d l Z d  d l j Z d  d l j	 j
 Z d  d l m Z m Z m Z m Z d   Z d   Z d   Z d   Z d   Z d	   Z d
 e f d     YZ d S(   i    (   t   divisiont   print_functiont   absolute_importN(   t   assert_equalt   assert_almost_equalt   assert_t   assert_allclosec          C` sg   t  j d  }  |  d } t t j |  |  d  t  j d d d d } t t j |  |  d  d  S(   Ni   i
   i    i   t   numgX?(   t   npt   arangeR   t   mst   compare_medians_mst   linspace(   t   xt   yt   y2(    (    sC   lib/python2.7/site-packages/scipy/stats/tests/test_mstats_extras.pyt   test_compare_medians_ms   s
    
c          C` s  t  j d  }  t t j |   d d d t  j |   |  _ t |  j d *t t j |   d d d t t j |  d t	 j
 d k  t  j d	  j d  } t t j | d
 d d d g  t  j |  | _ t | j d  d  d  d   f <t t j | d
 d d d g  d  S(   Ni   i   t   rtolg+=i   i   t   vari   i   t   axisi    i
   i   (   i   i   (   t   maR	   R   R
   t   hdmediant	   make_maskt   maskt   FalseR   t   Truet   sizet   reshape(   R   t   x2(    (    sC   lib/python2.7/site-packages/scipy/stats/tests/test_mstats_extras.pyt   test_hdmedian   s    ""c          C` sy   t  j j d  t  j j d  }  t j |   } t | j |  j k  t j |  d d d g } t | j d k  d  S(   Ni id   t   pointsi    g      ?i   (	   R   t   randomt   seedt   randnR
   t   rshR   t   shapeR   (   R   t   res(    (    sC   lib/python2.7/site-packages/scipy/stats/tests/test_mstats_extras.pyt   test_rsh(   s    c          C` sn   t  j d d d d d d d d d	 d
 d d d d d d d d d g  }  t t j |   d d d g d  d  S(   NiM   iW   iX   ir   i   i   i   i   i   i  i(  i+  i2  ix  i  i  i  i  i3
  gTK@gϽKF@gQh@i   (   R   t   arrayR   R
   t   mjci(   t   data(    (    sC   lib/python2.7/site-packages/scipy/stats/tests/test_mstats_extras.pyt	   test_mjci5   s    $$c          C` s   t  j d d d d d d d d d	 d
 d d d d d g  }  t t j |  d  d d  t t j t j |  d  d  d d g  d  S(   Ni!  i+  i.  i<  i?  i@  iB  iD  iR  i]  i{  i  i  i  i  g?g@i   gfffff@g̴@(   g?g?(	   R   R&   R   R
   t   trimmed_meanR   R   t   roundt   trimmed_mean_ci(   R(   (    (    sC   lib/python2.7/site-packages/scipy/stats/tests/test_mstats_extras.pyt   test_trimmed_mean_ci<   s
    c          C` s   t  j d  }  t t  j t j |    d d g d  |  j d  j d d  } t t j | d d d d d g d d d g g d  t t j | d d	 |  j d
  j d d
   d d g }  t j |   } t t  j	 |  j
    d  S(   Nid   gRBj8@g]kSUR@i   i   iR   i    i   i   (   R   R	   R   t   asarrayR
   t   idealfourthst   repeatR   R   t   isnant   all(   t   testt   test_2Dt   _result(    (    sC   lib/python2.7/site-packages/scipy/stats/tests/test_mstats_extras.pyt   test_idealfourthsE   s    t   TestQuantilesc        d   B` sU  e  Z d  d d d d d d d d d	 d
 d d d d d d d d d d d d d d d d d d d d d d  d! d" d# d$ d% d& d' d( d) d* d+ d, d- d. d/ d0 d1 d2 d3 d4 d5 d6 d7 d8 d9 d: d; d< d= d> d? d@ dA dB dC dD dE dF dG dH dI dJ dK dL dM dN dO dP dQ dR dS dT dU dV dW dX dY dZ d[ d\ d] d^ d_ d` da db dc gd Z dd   Z de   Z df   Z RS(g   gxc%?gF4vE?g݌Y?gn?gJ]X?g'h?g|f2?g!}_z`?g3!?g[Q,?gW1y?g+?gZy7В?g?g|GA?gPw2?g0?g3;??g,7tZ?gxދ?gޠt$O?giL?g|rmq?gsl?g݋Y?g)چ?g
zS?g'	A?g9io?gOB?g5G~?g9 ?g*sD?g\?gN?g6??gƟ	'?g$	?gLw5?g2J(?g2^?g?]?gͬ?g8~n7?gxY?gE!?ga@'?gʇ3?gF~\?gB誽?gX/u?g/:L@{?gb9?gvY?g5XEs?gd"2?g-=X.?giE"?g/E?g4J3H?gǒ?g+[{?go?gbm?g\-?g'W?@?g0?gtz?g=z5?g}v?gXPT?g<*R?gX?gO{F?g::2?gǺ?g/i?g6?gRfU?gu9?g!G[?gT=~?g?gAM07?g,jKV?g"ЄI?g΅?g\?gRSI?g	o?g2QJ)d~?g4n/?g+--C?gX?g|wiƘ?gttњc,?g{˻;Ib?gّ?gv+:?gR <e?c         C` s  |  j  } t t j | d d g  d d g  t j | d d d g  } t | d d	 d
 g  t j | d d d g  } t | d d d g d  t j |  j d d  } t j | d d d g d d } t | d  d   d f t j | d  d   d f d d d g   t | d  d   d f t j | d  d   d f d d d g   t j | d d d g d d d t } t | d t j | d  d   d f d d d g d t  t | d t j | d  d   d f d d d g d t  d  S(   Ng        g      ?gtz?gǒ?g      ?g      ?g      ?g]Ar4?g5$?i?gJJ5d?gc?g7{?gS2kt?i   i
   R   i    iR   .(   .i    (   .i(	   R(   R   R
   t   hdquantilest   hdquantiles_sdR   R&   R   R   (   t   selfR(   t   hdq(    (    sC   lib/python2.7/site-packages/scipy/stats/tests/test_mstats_extras.pyt   test_hdquantilesk   s     	!BB'
2
c         C` s)   t  j |  j  } t | j d k  d  S(   Ni   (   R
   R9   R(   R   R   (   R:   R$   (    (    sC   lib/python2.7/site-packages/scipy/stats/tests/test_mstats_extras.pyt   test_hdquantiles_sd~   s    c         C` sB   t  j |  j  \ } } t | j | j k o8 d k n  d  S(   Ni   (   R
   t   mquantiles_cimjR(   R   R   (   R:   t   ci_lowert   ci_upper(    (    sC   lib/python2.7/site-packages/scipy/stats/tests/test_mstats_extras.pyt   test_mquantiles_cimj   s    (   t   __name__t
   __module__R(   R<   R=   RA   (    (    (    sC   lib/python2.7/site-packages/scipy/stats/tests/test_mstats_extras.pyR7   U   s.   		(   t
   __future__R    R   R   t   numpyR   t   numpy.maR   t   scipy.stats.mstatst   statst   mstatsR
   t   numpy.testingR   R   R   R   R   R   R%   R)   R-   R6   t   objectR7   (    (    (    sC   lib/python2.7/site-packages/scipy/stats/tests/test_mstats_extras.pyt   <module>   s   "								