
    _Mh                         d dl Zd dlmZ d dlmc mZ d dlm	Z	m
Z
mZmZ d Zd Zd Zd Zd Zd Z G d	 d
          Zd ZdS )    N)assert_equalassert_almost_equalassert_assert_allclosec                      t          j        d          } | dz   }t          t          j        | |          d           t          j        ddd          }t          t          j        | |          d           d S )N   
   r      )numgX?)nparanger   mscompare_medians_mslinspace)xyy2s      d/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/scipy/stats/tests/test_mstats_extras.pytest_compare_medians_msr   	   sn    
	!A	BA-a33Q777	Qr	"	"	"B-a44nEEEEE    c                     t          j        d          } t          t          j        |           dd           t          j        |           | _        d| j        d d<   t          t          j        |           dd           t          t          j        | d	          j        d
k               t          j        d          	                    d          }t          t          j        |d          ddg           t          j        |          |_        d|j        d dd d f<   t          t          j        |d          ddg           d S )N      g+=)rtolFr      Tvar      )r   r   r   axisr	      )
mar   r   r   hdmedian	make_maskmaskr   sizereshape)r   x2s     r   test_hdmedianr*      s   
	"ABKNNAE2222\!__AFAF2A2JBKNNAE2222 BKt$$$)Q./// 
2		w	'	'BBK+++b"X666l2BGBGBQBENBK+++aV44444r   c                  D   t           j                            d           t           j                            d          } t	          j        |           }t          |j        | j        k               t	          j        | ddg          }t          |j        dk               d S )Ni d   r         ?)pointsr   )	r   randomseedrandnr   rshr   shaper'   )r   ress     r   test_rshr5   &   s    INN6
	A
&))C CI !!! &Ar7
#
#
#CCHMr   c                  |    t          j        g d          } t          t          j        |           g dd           d S )N)M   W   X   r                  i  i(  i+  i2  ix  i  i    i  i3
  )gTK@gϽKF@gQh@r   )r#   arrayr   r   mjcidatas    r   	test_mjcirE   3   sI    8 < < < = =D&C&C&CAFFFFFr   c                      t          j        g d          } t          t          j        | d          dd           t          t          j        t          j        | d          d          ddg           d S )N)i!  i+  i.  i<  i?  i@  iB  iD  iR  i]  i{  i  i  i  r@   皙?g@r
   )rG   rG   gfffff@g̴@)	r#   rA   r   r   trimmed_meanr   r   roundtrimmed_mean_cirC   s    r   test_trimmed_mean_cirK   :   s~    8 2 2 2 3 3DS115!<<<",T)<<Q??! ! ! ! !r   c                  p   t          j        d          } t          t          j        t	          j        |                     ddgd           |                     d                              dd          }t          t	          j        |d          g d	g d
gd           t          t	          j        |d          |                     d                              dd                     ddg} t	          j        |           }t          t          j	        |          
                                           d S )Nr,   RBj8@]kSUR@r"   r   r   r    )rM   rM   rM   )rN   rN   rN   r
   r   )r   r   r   asarrayr   idealfourthsrepeatr(   r   isnanall)testtest_2D_results      r   test_idealfourthsrX   C   s!   9S>>D
2?4#8#899"9-a1 1 1kk!nn$$R**Ga888888888::;= = = a888A..r!446 6 6q6Dod##GBHW!!##$$$$$r   c                   (    e Zd Zg dZd Zd Zd ZdS )TestQuantiles)dg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o?gbm?g\-?g'W?@?g0?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                 *   | j         }t          t          j        |ddg          ddg           t          j        |g d          }t          |g d           t	          j        |                              dd          }t          j        |g dd	          }t          |d d df         t          j        |d d df         g d                     t          |d d d
f         t          j        |d d d
f         g d                     t          j        |g ddd          }t          |d         t          j        |d d df         g dd                     t          |d         t          j        |d d d
f         g dd                     d S )Ng        r-   r\   r[   )g      ?      ?g      ?)g]Ar4?g5$?i?gJJ5d?r	   r   r    rO   T)r!   r   ).r   r   ).rO   )rD   r   r   hdquantilesr   rA   r(   )selfrD   hdqs      r   test_hdquantileszTestQuantiles.test_hdquantilesi   s   yBN4R99(+6	8 	8 	8nT"3"3"344C!I!I!IJJJx~~%%b,,nT///q999C!HbnT!!!A#Y&O&OPPPC"Ir~d111R4j'Q'QRRRnT///qTBBBCJN4!9___NNN	P 	P 	PCKN4":ooo4PPP	R 	R 	R 	R 	Rr   c                 ^   t          j        | j                  }t          | j                  }t	          j        | j        ||f          }t	          j        t	          j        |                    }||                             ||dz
            }t          j	        |d          }t	          j
        |d          }|dz
  |z  t	          j        ||z
  dz  d          z  dz  }t          ||           t          |g d           t          j        ddg          }t          |g d           d S )Nr
   r    r   r   r^   )giYk?gK4U}?gN5i	t?)r^   r^   r^   )r   hdquantiles_sdrD   lenr   broadcast_tological_noteyer(   r_   meansumr   )	r`   hd_std_errsnjdata	jselectorjdist
jdist_meanjstdtwo_data_pointss	            r   test_hdquantiles_sdz!TestQuantiles.test_hdquantiles_sdz   s   
 '	22 	NN	Aq622N26!99--	i ((AaC00u1---WU+++
1a"&%*"4q!8qAAAABFK...K)J)J)JKKK+QF33O___=====r   c                     t          j        | j                  \  }}t          |j        |j        cxk    odk    nc            d S )Nr   )r   mquantiles_cimjrD   r   r'   )r`   ci_lowerci_uppers      r   test_mquantiles_cimjz"TestQuantiles.test_mquantiles_cimj   sM    /	::(3333!333344444r   N)__name__
__module____qualname__rD   rb   rs   rx    r   r   rZ   rZ   S   sX        I I ID*R R R"> > >.5 5 5 5 5r   rZ   c                      t           j                            d          } |                     d          }t          t	          j        |          d           t          t	          j        |d          d           d S )Nl   WE~-S    )r'   )gl?gz0JL?g?)g/?gL0?)r   r/   default_rngr   r   median_cihs)rngr   s     r   test_median_cihsr      sm     )

 3
4
4C


ABN1%%'?@@@ BN1c**,JKKKKKr   )numpyr   numpy.mar#   scipy.stats.mstatsstatsmstatsr   numpy.testingr   r   r   r   r   r*   r5   rE   rK   rX   rZ   r   r|   r   r   <module>r      s)                     , , , , , , , , , , , ,F F F5 5 5(
 
 
G G G! ! !% % % A5 A5 A5 A5 A5 A5 A5 A5HL L L L Lr   