B
    `                 @   s   d dl Zd dlmZ d dlm  mZ d dlm	Z	m
Z
mZmZ dd Zdd Zdd Zd	d
 Zdd Zdd ZG dd deZdS )    N)assert_equalassert_almost_equalassert_assert_allclosec              C   sJ   t d} | d }tt| |d t jdddd}tt| |d d S )N   
   r      )numgX?)nparanger   msZcompare_medians_msZlinspace)xyy2 r   S/home/dcms/DCMS/lib/python3.7/site-packages/scipy/stats/tests/test_mstats_extras.pytest_compare_medians_ms	   s
    
r   c              C   s   t d} tt| ddd t | | _d| jd d< tt| ddd ttj| dd	jd
k t d	d}ttj|ddddg t ||_d|jd dd d f< ttj|ddddg d S )N      g+=)ZrtolFr      T)var      )r   r   r   )axisr      )
mar   r   r   ZhdmedianZ	make_maskmaskr   sizereshape)r   Zx2r   r   r   test_hdmedian   s    
r   c              C   sV   t jd t jd} t| }t|j| jk tj| ddgd}t|jdk d S )Ni d   r   g      ?)Zpointsr   )	r
   randomseedZrandnr   Zrshr   shaper   )r   resr   r   r   test_rsh&   s    
r%   c              C   sL   t ddddddddd	d
dddddddddg} tt| dddgd d S )NM   W   X   r                  i  i(  i+  i2  ix  i  i  i  i  i3
  gTK@gϽKF@gQh@r   )r   arrayr   r   Zmjci)datar   r   r   	test_mjci3   s    r1   c              C   s^   t ddddddddd	d
dddddg} tt| ddd ttt| ddddg d S )Ni!  i+  i.  i<  i?  i@  iB  iD  iR  i]  i{  i  i  i  i  g?g@r   )g?g?gfffff@g̴@)	r   r/   r   r   Ztrimmed_meanr   r
   roundZtrimmed_mean_ci)r0   r   r   r   test_trimmed_mean_ci:   s
    r3   c              C   s   t d} tt t| ddgd | ddd}ttj|dddddgdddggd ttj|d	d| d
dd
 ddg} t| }tt 	|
  d S )Nr    gRBj8@g]kSUR@r   r   r   )r   r   r   )r
   r   r   Zasarrayr   Zidealfourthsrepeatr   r   isnanall)testZtest_2DZ_resultr   r   r   test_idealfourthsC   s    

r9   c            d   @   s   e Zd 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/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcddgdZdedf Zdgdh Zdidj ZdkS )lTestQuantilesg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   sb  | j }tt|ddgddg t|dddg}t|dd	d
g t|dddg}t|dddgd t|dd}tj|dddgdd}t|d d df t|d d df dddg t|d d df t|d d df dddg tj|dddgddd}t|d tj|d d df dddgdd t|d tj|d d df dddgdd d S )Ng        g      ?gtz?gǒ?g      ?g      ?g      ?g]Ar4?g5$?i?gJJ5d?gc?g7{?gS2kt?   r   r   )r   r4   T)r   r   ).r   )r   ).r4   )r0   r   r   Zhdquantileshdquantiles_sdr
   r/   r   )selfr0   Zhdqr   r   r   test_hdquantilesi   s     
00$zTestQuantiles.test_hdquantilesc             C   s   t | j}t|jdk d S )Nr   )r   r<   r0   r   r   )r=   r$   r   r   r   test_hdquantiles_sd|   s    z!TestQuantiles.test_hdquantiles_sdc             C   s4   t | j\}}t|j|j  ko(dkn   d S )Nr   )r   Zmquantiles_cimjr0   r   r   )r=   Zci_lowerZci_upperr   r   r   test_mquantiles_cimj   s    z"TestQuantiles.test_mquantiles_cimjN)__name__
__module____qualname__r0   r>   r?   r@   r   r   r   r   r:   S   s.   


















r:   )numpyr
   Znumpy.mar   Zscipy.stats.mstatsstatsZmstatsr   Znumpy.testingr   r   r   r   r   r   r%   r1   r3   r9   objectr:   r   r   r   r   <module>   s   		