o
    g#                    @   s  d dl Z d dlZd dlZd dlmZ d dlmZmZm	Z	m
Z
mZmZmZmZ d dlZd dlmZ d dlZd dlmZ d dlmZ d dlmZ dd	lmZ d
dlmZmZ d dlmZ d dlm Z  e!e Z zd dl"Z"de"j#d< d dl$m%Z& dZ'W n e(y   dZ'Y nw g dZ)g dZ*g dZ+g dZ,g dZ-g dZ.g dZ/g dZ0g dZ1g dZ2dd Z3G dd dZ4G d d! d!Z5G d"d# d#Z6G d$d% d%Z7G d&d' d'Z8G d(d) d)Z9G d*d+ d+Z:G d,d- d-Z;G d.d/ d/Z<G d0d1 d1Z=G d2d3 d3Z>d4d5 Z?G d6d7 d7Z@G d8d9 d9ZAG d:d; d;ZBG d<d= d=ZCG d>d? d?ZDG d@dA dAZEG dBdC dCZFG dDdE dEZGg dFZHG dGdH dHZIG dIdJ dJZJG dKdL dLZKG dMdN dNZLG dOdP dPZMG dQdR dRZNG dSdT dTZOG dUdV dVZPG dWdX dXZQdS )Y    N)RandomState)assert_array_equalassert_almost_equalassert_array_lessassert_array_almost_equalassert_assert_allcloseassert_equalsuppress_warnings)raises)optimize)stats)
_abw_state   )check_named_results   )_get_wilcoxon_distr_get_wilcoxon_distr2)_binary_search_for_binom_tst)distcontAggbackendTF)
jt?Zd;?V-?      ?gX9v?-?x&1?+?+?r   )
r   r   r   r   灕Cl?r   r   r   r   V-?)
ʡE?g/$?r    r   ףp=
?r   r   r   r   r   )
gGz?r   r   r   r#   r   r   r   r   r   )
r   r   gCl?Gz?r   rh|?r   r   g\(\?r   )
羟/$?gh|?5?r&   r    r!   r   r#   r   g1Zd?r   )
r$   Mb?r   gjt?r   r   g}?5^I?g)\(?r   r   )
r   r   r   r   r   r   r   r   r   r   )
r   r   r   r#   r   r'   r'   r   r   r"   )
r"   r#   r%   r   r    r    r"   r   r'   r    c                  O   s   t tjj| i |S N)nplogr   gammarvs)argskwargs r/   ^/home/ubuntu/cloudmapper/venv/lib/python3.10/site-packages/scipy/stats/tests/test_morestats.py_old_loggamma_rvs5      r1   c                   @   $   e Zd Zdd Zdd Zdd ZdS )TestBayes_mvsc                 C   st   g d}t |\}}}t|jd t|jddd t|jd t|jddd t|jd	d
d t|jddd d S )N   	            r:            "@g4`H#j@gO[%@+=rtol      $@gai	@gӇu8@&.>_紫@   decimalgYs?g1O@)r   	bayes_mvsr   	statisticr   minmaxselfdatameanvarstdr/   r/   r0   
test_basic:   s   
zTestBayes_mvs.test_basicc                 C      t ttjg  d S r(   )assert_raises
ValueErrorr   rI   rM   r/   r/   r0   test_empty_inputL      zTestBayes_mvs.test_empty_inputc                 C   s0   t d}d}t|}|D ]}t|| qd S )N   )rJ   rK   )r)   aranger   rI   r   )rM   x
attributesresir/   r/   r0   test_result_attributesO   s   

z$TestBayes_mvs.test_result_attributesN)__name__
__module____qualname__rR   rW   r_   r/   r/   r/   r0   r4   9   s    r4   c                   @   ,   e Zd Zdd Zdd Zdd Zdd Zd	S )
TestMvsdistc                 C   s   g d}t |\}}}t| d t|dddd t| d t|ddd	d t| d
dd t|dddd d S )Nr5   r<   g?r=   r>   r?   rA   rB   rC   rD   rE   rF   rH   )r   mvsdistr   rO   r   intervalrL   r/   r/   r0   rR   Y   s   
zTestMvsdist.test_basicc                 C   rS   r(   rT   rU   r   re   rV   r/   r/   r0   rW   h   rX   zTestMvsdist.test_empty_inputc                 C      dg}t ttj| d S Nr   rg   rM   rN   r/   r/   r0   test_bad_argk      zTestMvsdist.test_bad_argc                 C   sj   t  ' t dt dd tg dD  dd tg dD  W d    d S 1 s.w   Y  d S )Nerrorc                 S      g | ]}|  qS r/   rO   .0r[   r/   r/   r0   
<listcomp>u       z*TestMvsdist.test_warns.<locals>.<listcomp>r   r      c                 S   rn   r/   ro   rp   r/   r/   r0   rr   v   rs   r   r   ru         )warningscatch_warningssimplefilterRuntimeWarningr   re   rV   r/   r/   r0   
test_warnsp   s
   
"zTestMvsdist.test_warnsN)r`   ra   rb   rR   rW   rk   r}   r/   r/   r/   r0   rd   X   s
    rd   c                   @   D   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dS )TestShapiroc           
      C   s  g d}t |\}}t |}t|ddd t|jddd t|ddd t|jddd g d}t |\}}t |}t|ddd t|jddd t|dd	d t|jdd	d t jjd
d	ddd}t |\}}t |}t|ddd t|jddd t|dd	d t|jdd	d g d}d}d}	t |\}}t |}t||dd t|j|dd t||	d
d t|j|	d
d d S )N))\(?{Gz@q=
ףp@HzG$@@Q	@q=
ףp?Q@zG?      @Q?RQ?Q	@(\?zG?zG@(\#@333333@Gz?p=
ף@   ?r6   rF      Ɍ?)(\?=
ףp=?\(\@ffffff@\(\?(\?Gz@(\ףp=
@皙?)\(?RQr   p=
ף?@
ףp=
?{Gz?\(\@{Gz@Q@IddY?Dio?ru   rx   d   iNa locscalesizerandom_stateg   E?g   Oٴ?)gn?gjt?gffffff?gMb?gjt?g;On?gjt?gX9v?gS?g(\?gI+?gʡE?gx?g|?5^?r   gCl?g1Zd@gS@g(\	@gGz@g rh@g"~j@gQ@gtV@gʡE @ghyܝ?g0M?rw   )r   shapiror   rJ   pvaluenormr,   )
rM   x1wpwshapiro_testx2x3x4
W_expected
p_expectedr/   r/   r0   rR   z   s<   



zTestShapiro.test_basicc                 C   s   g dg dg}t |\}}t |}t|ddd t|jddd t|ddd t|jddd g dg dg}t |\}}t |}t|d	dd t|jd	dd t|d
dd t|jd
dd d S )N)
r   r   r   r   r   r   r   r   r   r   )
r   r   r   r   r   r   r   r   r   r   r   r6   rF   r   )
r   r   r   r   r   r   r   r   r   r   )
r   r   r   r   r   r   r   r   r   r   r   r   ru   )r   r   r   rJ   r   )rM   r   r   r   r   r   r/   r/   r0   test_2d   s$   

zTestShapiro.test_2dc                 C   s&   t ttjg  t ttjg g g g d S r(   rT   rU   r   r   rV   r/   r/   r0   rW         zTestShapiro.test_empty_inputc                 C   s4   t ttjddg t ttjtjg dggtd d S )Nr   r   dtype)rT   rU   r   r   r)   arrayobjectrV   r/   r/   r0   test_not_enough_values   s   "z"TestShapiro.test_not_enough_valuesc                 C   rh   ri   r   rM   r[   r/   r/   r0   rk      rl   zTestShapiro.test_bad_argc                 C   s`   t d}t j|d< t|\}}t|}t|t j t|jt j t|d t|jd d S )NrA   r7   r   )	r)   rZ   nanr   r   r	   rJ   r   r   )rM   r[   r   r   r   r/   r/   r0   test_nan_input   s   



zTestShapiro.test_nan_inputc                 C   s2   t g d\}}t |}d}t||dd d S )N)i i = i@ )gPq~?gF?h㈵>r?   )r   boxcoxr   r   )rM   	trans_valmaxlogr]   refr/   r/   r0   test_gh14462   s   
zTestShapiro.test_gh14462N)
r`   ra   rb   rR   r   rW   r   rk   r   r   r/   r/   r/   r0   r   y   s    +r   c                   @   s^   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	e
jdg ddd ZdS )TestAndersonc                 C   s   t d}|jdd}|jdd}t|\}}}t|d d | t|\}}}t||dd   td}d|d< t|\}}}t|d d S )	NI2   r   
   r   g	@)	r   standard_exponentialstandard_normalr   andersonr   r)   onesr   )rM   rsr   r   Acritsigvr/   r/   r0   test_normal   s   

zTestAnderson.test_normalc                 C   s   t d}|jdd}|jdd}t|d\}}}t||dd   tjdd t|d\}}}W d    n1 s<w   Y  t||d k d S )	Nr   r   r   exponr   ignore)allr   )	r   r   r   r   r   r   r)   errstater   )rM   r   r   r   r   r   r   r/   r/   r0   
test_expon   s   zTestAnderson.test_exponc              	   C   s   t d}d|d< t|d\}}}t|}tj|\}}tj|||}tj|||}	t 	d|d }
| t 
d|
 d ||	d d d    }t|| d S )Nr           r   gumbelr   r   r   )r)   r   r   r   lengumbel_lfitlogcdflogsfrZ   rO   r   )rM   r   a2r   r   nxbarsr   r   r^   expected_a2r/   r/   r0   test_gumbel  s   
*zTestAnderson.test_gumbelc                 C      t ttjdgdd d S Nr   plate_of_shrimpdist)rT   rU   r   r   rV   r/   r/   r0   rk     r2   zTestAnderson.test_bad_argc                 C   s0   t d}|jdd}t|}d}t|| d S )Nr   r   r   rJ   critical_valuessignificance_level)r   r   r   r   r   )rM   r   r[   r]   r\   r/   r/   r0   r_     s
   
z#TestAnderson.test_result_attributesc           	      C   sF   t d}|jdd}t|d\}}}t|d\}}}t|| d S )Nr   r   r   r   r   )r   r   r   r   r   )	rM   r   r[   A1crit1sig1A2crit2sig2r/   r/   r0   test_gumbel_l   s
   zTestAnderson.test_gumbel_lc           
      C   sp   t d}|jdd}td}d|d< t|d\}}}t|d\}}}	t||dd   t||d k d S )	Nr   r   r   r   r   gumbel_rr   r   )r   r   r)   r   r   r   r   r   )
rM   r   r   r   r   r   r   r   r   r   r/   r/   r0   test_gumbel_r*  s   
zTestAnderson.test_gumbel_rdistname)r   r   r   extreme1r   r   logisticc                 C   s\   t jd}|dv rdn|}tt|}t| }|j|d|d}t||}|jj	s,J d S )Nl   'Dah$>   r   r  r     r   r   )
r)   randomdefault_rnggetattrr   r   r,   r   
fit_resultsuccess)rM   r  rngreal_distnamer   paramsr[   r]   r/   r/   r0   test_anderson_fit_params9  s   
z%TestAnderson.test_anderson_fit_paramsN)r`   ra   rb   r   r   r   rk   r_   r   r  pytestmarkparametrizer  r/   r/   r/   r0   r      s    

r   c                   @   T   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd ZdS )TestAndersonKSampc                 C   s   g d}t g d}t g d}t g d}tj||||fdd\}}}t|dd tg d	|d
d d t|ddd d S )NgYC@g     D@gfffffE@g     @F@g     F@g      G@gG@g      M@gC@gfffffC@gC@g33333D@gfffffD@g33333sE@gfffffE@gfffffF@      A@g     A@g     C@g      D@     E@r        F@g     F@r  ffffffA@r  g33333A@gB@gfffffB@gD@gffffffE@Fmidrankg@ru   gCl?gvq-?g,Ԛ?gl@gPk	@r   rx   rw   gJ4a?Mb0?atolr)   r   r   anderson_ksampr   r   r   rM   t1t2t3t4Tktmpr/   r/   r0   test_example1aI  s   z TestAndersonKSamp.test_example1ac                 C   s   t g d}t g d}t g d}t g d}tj||||fdd\}}}t|dd tg d	|d
d d t|ddd d S )Nr  r  r  r  Tr  gQ@ru   r  r   rx   rw   gMb`?r   r!  r#  r%  r/   r/   r0   test_example1bZ  s   z TestAndersonKSamp.test_example1bc                 C      g d}g d}g d}g d}g d}g d}g d}g d}g d	}	g d
}
g d}g d}g d}g d}t j|||||||||	|
||||fdd\}}}t|dd tg d|dd d t|ddd d S )N   rY   )      !      i  rE   :   %   r   A   r7      i     $      v   
"         rB  C   9   >   r9      r@  Z   r   <      =   1   rE      8      O   T   ,   ;   r3  r>        i6  L      rR     rE        F   e   r[  J   rD  0   r3  i  r8   r\     r3  i  rS     7   i@  rN  h         /            r4  rX  i  W   r9   x   rE   rE  rh     G   ri  ra  *   rO  rx   r8   rn     ru   rE   rp  rr  rE   rr     rH  r   rs  4   _   a   3   rr  rw      rB     D   M   P   r   rs  j      R   6   rA     .   o   '   ?   rB     rB     rM  r   rR  f   H   rF  r  ru   rY         rP  X   r  rx   rx   r<  rF        rw     rX  r;   rE   	g  r7   r8   i  i[  ru   re  r   i  r      rx   i  #   r8   i  rB  r   r9   b   rx   U   [   +      ru   rZ  r     rE   rD  r      rC  rS        rb  rE   r  B   rK  r@  Fr  gM
@ru   g^I+?g~:?g+e?gH.!@gcZB>h@r   rx   rw   	^)p?r   r!  r   r$  r   r   r   rM   r&  r'  r(  r)  t5t6t7t8t9t10t11t12t13t14r*  r+  r,  r/   r/   r0   test_example2aj  s0   z TestAndersonKSamp.test_example2ac                 C   r/  )Nr0  r6  r?  rG  rY  r^  rc  rl  rv  r  r  r  r  r  Tr  g'1Z
@ru   r  r   rx   rw   r  r   r!  r  r  r/   r/   r0   test_example2b  s0   z TestAndersonKSamp.test_example2bc                 C   s~  t ddd}t }|jtdd tj||d gdd\}}}W d    n1 s*w   Y  t|dd	 t|d
 t }|jtdd t||d g\}}}W d    n1 s[w   Y  t|dd	 t|d
 t }|jtdd tj||d gdd\}}}W d    n1 sw   Y  t|dd t|d t }|jtdd t||d g\}}}W d    n1 sw   Y  t|dd t|d tj||d gdd\}}}t|dd t	|dddd t||d g\}}}t|dd t	|dddd tj||d gdd\}}}t|dd t	|dddd tj||d gdd\}}}t|d d t	|d!d"dd d S )#Nr   r   zp-value flooredmessageg     @D@Fr  g=
ףpD@ru   MbP?gGzD@zp-value capped      ?g?ܵrw         ?g?ܵ      @g!u?gףp=
׳?g{Gzt?r   r"  r@   r6   gBiq?g鷯?      '@g:M@g46<Rv?gMb@?g      +@g6>W[1@g1ZGUS?-C6?)
r)   linspacer
   filterUserWarningr   r$  r   r	   r   )rM   r   supr   _r,  r/   r/   r0   test_R_kSamples  sJ   !



z!TestAndersonKSamp.test_R_kSamplesc                 C   s   t ttjtd d S Nrx   rT   rU   r   r$  r)   r   rV   r/   r/   r0   test_not_enough_samples  r2   z)TestAndersonKSamp.test_not_enough_samplesc                 C   s"   t ttjtdtdf d S r  r  rV   r/   r/   r0   test_no_distinct_observations  s   z/TestAndersonKSamp.test_no_distinct_observationsc                 C   s   t ttjtdg f d S r  r  rV   r/   r/   r0   test_empty_sample     z#TestAndersonKSamp.test_empty_samplec                 C   sH   g d}t g d}tj||fdd}d}t|| t|j|j d S )Nr  r  Fr  r   )r)   r   r   r$  r   r	   r   r   )rM   r&  r'  r]   r\   r/   r/   r0   r_     s   
z(TestAndersonKSamp.test_result_attributesN)r`   ra   rb   r-  r.  r  r  r  r  r  r  r_   r/   r/   r/   r0   r  H  s      Pr  c                   @   s   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	e
jdg dg dddfg dg dddfg dg dddfg dg dddfg dg dddfg dg dddfgdd Zdd ZdS ) 
TestAnsaric                 C   sn   g d}g d}t  }|td t||\}}W d    n1 s$w   Y  t|dd t|dd d S )Nr   r   ru   ru   rw   ru   r   r6   r   r6   r   rw   r   %Ties preclude use of exact statistic.g     7@rr  g8#oG?)r
   r  r  r   ansarir   )rM   r[   yr  Wpvalr/   r/   r0   
test_small  s   zTestAnsari.test_smallc                 C   sr   t d}t d}t }|td t||\}}W d    n1 s&w   Y  t|dd t|dd d S )Nr  k   r   c   r  r~  m   l   re  r  r]  `   rw  r  r  q   t   r  n   r  r  r  r~  r  i   g   r  r  re  r   r  r  r  re  r   r  r  r  r~  r  r  g     0g@rr  g|:?)r)   r   r
   r  r  r   r  r   )rM   ramsayparekhr  r  r  r/   r/   r0   test_approx  s   

zTestAnsari.test_approxc                 C   s4   t g dg d\}}t|dd t|dd d S )Nr   r   ru   rw   )rY   rx   rO  r:   r   r8   rA   rr  g?r9   )r   r  r   )rM   r  r  r/   r/   r0   
test_exact)  s   zTestAnsari.test_exactc                 C   s(   t ttjg dg t ttjdgg  d S ri   rT   rU   r   r  rV   r/   r/   r0   rk   .  s   zTestAnsari.test_bad_argc                 C   s`   g d}g d}t  }|td t||}W d    n1 s"w   Y  d}t|| d S )Nr  r  r  rJ   r   )r
   r  r  r   r  r   )rM   r[   r  r  r]   r\   r/   r/   r0   r_   2  s   z!TestAnsari.test_result_attributesc                 C   sV   g d}g d}d}t t|d tj||dd W d    d S 1 s$w   Y  d S )Nr  rx   r6   r9   r:   z!'alternative' must be 'two-sided'matchfooalternativer  )rM   r   r   r  r/   r/   r0   test_bad_alternative;  s   "zTestAnsari.test_bad_alternativec           
      C   s   g d}g d}t ||\}}t j||ddj}t j||ddj}|dks(J |dk s.J t|t|t|}t|| d| d	d
 t||d d	d
 t|d| |d  d	d
 t j||ddj}t j||ddj}	|dk srJ |	dksxJ d S )N)r   rx   r   rY   rO  rT  )r  g      !@      #@g      %@r  g      )@lessr  greaterffffff?皙?r   -q=r!  r   )r   r  r   r   pmfr   r   )
rM   r   r   rJ   r  pval_lpval_gprobpval_l_reversepval_g_reverser/   r/   r0   test_alternative_exactC  s   z!TestAnsari.test_alternative_exactzx, y, alternative, expectedr  r  r  g@A?r  rt   )rw   rx   r6   r9   r:   gG$I?g%I$I?rv   )r6   r9   r:   c                 C   s$   t j|||dj}t||dd d S )Nr  r  r!  )r   r  r   r   )rM   r[   r  r  expectedr  r/   r/   r0   test_alternative_exact_with_R^  s   z(TestAnsari.test_alternative_exact_with_Rc                 C   s   t jjddddd}t jjddddd}t j||ddj}t j||d	dj}t|d
dd t|ddd t jjddddd}t jjddddd}t ||j}t j||ddj}t j||d	dj}t||d dd t|d|d  dd d S )Nr   rx   r   {   r  r   r  r  r  r   r  r!  r   rI  g      ?r   )r   r   r,   r  r   r   )rM   r   r   r  r  r  r/   r/   r0   test_alternative_approx}  s   z"TestAnsari.test_alternative_approxN)r`   ra   rb   r  r  r  rk   r_   r  r  r  r  r  r  r  r/   r/   r/   r0   r    s(    		
r  c                   @   4   e Zd Zdd Zdd Zdd Zdd Zd	d
 ZdS )TestBartlettc              
   C   B   t ttttttttt	g
}t
j| \}}t|dd t|dd d S )NgzU@ /4@r9   gG?)g1g2g3g4g5g6g7g8g9g10r   bartlettr   )rM   r-   Tr  r/   r/   r0   	test_data     zTestBartlett.test_datac                 C      t ttjdg d S ri   )rT   rU   r   r  rV   r/   r/   r0   rk        zTestBartlett.test_bad_argc              
   C   4   t ttttttttt	g
}t
j| }d}t|| d S Nr  )r  r  r  r  r  r  r  r  r	  r
  r   r  r   rM   r-   r]   r\   r/   r/   r0   r_        
z#TestBartlett.test_result_attributesc                 C   s6   t ttttttttt	g f}t
tjtjftj|  d S r(   )r  r  r  r  r  r  r  r  r	  r
  r	   r)   r   r   r  )rM   r-   r/   r/   r0   test_empty_arg  s   zTestBartlett.test_empty_argc                 C   *   t ddgddgg}tttjt| d S Nr   r   ru   rw   )r)   r   rT   rU   r   r  r  r   r/   r/   r0   test_1d_input     zTestBartlett.test_1d_inputN)r`   ra   rb   r  rk   r_   r  r  r/   r/   r/   r0   r    s    r  c                   @   r  )
TestLevenec              
   C   r   )NgYpK?r9   gI{]?)r  r  r  r  r  r  r  r  r	  r
  r   levener   )rM   r-   r  r  r/   r/   r0   r    r  zTestLevene.test_datac                 C   sF   t jtttdd\}}t jtttddd\}}t|| t|| d S )NrO   centertrimmedr   r  proportiontocut)r   r  r  r  r  r   )rM   W1pval1W2pval2r/   r/   r0   test_trimmed1  s   

zTestLevene.test_trimmed1c           
      C   s   g d}g d}t jd t j|}tj||ddd\}}tj||ddd\}}tj|dd |dd d	d
\}}	t|| t|| t||	 d S )Ng333333?       @      @      @      @      @      @g      Y@r   r(        @r)  g      @r*  g      @g      i@  r        ?r  r   r   rO   r  )r)   r  seedpermutationr   r  r   )
rM   r[   r  r   W0pval0r!  r"  r#  r$  r/   r/   r0   test_trimmed2  s   



$

zTestLevene.test_trimmed2c                 C   sn   t ddd}t jd t j|}|d }tj||dd\}}tj||dd\}}t|| t|| d S )	Nr   r   ra  r/  ru   rO   r  median)r)   r  r  r1  r2  r   r  r   )rM   r[   r   r  r!  r"  r#  r$  r/   r/   r0   test_equal_mean_median  s   
z!TestLevene.test_equal_mean_medianc                 C   &   t ddd}tttj||dd d S Nr   r   ra  皙?)portiontocut)r)   r  rT   	TypeErrorr   r  r   r/   r/   r0   test_bad_keyword  r   zTestLevene.test_bad_keywordc                 C   r8  Nr   r   ra  trimr  )r)   r  rT   rU   r   r  r   r/   r/   r0   test_bad_center_value  r   z TestLevene.test_bad_center_valuec                 C   r  ri   )rT   rU   r   r  rV   r/   r/   r0   test_too_few_args  s   zTestLevene.test_too_few_argsc              
   C   r  r  )r  r  r  r  r  r  r  r  r	  r
  r   r  r   r  r/   r/   r0   r_     r  z!TestLevene.test_result_attributesc                 C   r  r  )r)   r   rT   rU   r   r  r  r   r/   r/   r0   r    r  zTestLevene.test_1d_inputN)r`   ra   rb   r  r%  r5  r7  r=  r@  rA  r_   r  r/   r/   r/   r0   r    s    	
r  c                   @   sd   e Zd ZdZedddZdd Zd	d
 Zdd Zdd Z	dd Z
ejjejdkdddd ZdS )TestBinomTestPzU
    Tests for stats.binomtest as a replacement for deprecated stats.binom_test.
    Nr  	two-sidedc                 C   s   t | t j} t| dkr| d | d  }| d } n t| dkr8| d } |d u s.|| k r2tdt |}ntdtj| |||d}|jS )Nr   r   r   zn must be >= xzIncorrect length for x.r,  r  )	r)   
atleast_1dastypeint_r   rU   r   	binomtestr   )r[   r   r,  r  resultr/   r/   r0   binom_test_func  s   
zTestBinomTestP.binom_test_funcc                 C   sR   |  dd}t|dd |  dd}t|dd | j dd	gd
d}t|dd d S )Nr      g-"^?rr  r=  i  g`w?i           ?r,  gz?rJ  r   )rM   r  r/   r/   r0   r    s   zTestBinomTestP.test_datac                 C   s   t t| jg d d S Nrt   rT   rU   rJ  rV   r/   r/   r0   test_bad_len_x     zTestBinomTestP.test_bad_len_xc                 C   s(   t t| jdg t t| jdgdd d S )Nr   r   r   rQ  rV   r/   r/   r0   
test_bad_n  s   zTestBinomTestP.test_bad_nc                 C   s   t t| jddgdd d S )Nr   r'  rN  rQ  rV   r/   r/   r0   
test_bad_p  s   
zTestBinomTestP.test_bad_pc                 C   sX   | j ddddd}t|d | j ddddd}t|d | j dddd	d}t|d
 d S )Nrx     gUUUUUU?r  rD  g-$Ǻl?r  gs0,j.?rC  gJfJ"f?rO  rM   r]   r/   r/   r0   test_alternatives"  s   

z TestBinomTestP.test_alternativesl        z32-bit does not overflowreasonc                 C   s   t t| jddtjjd d S )Nr*  r6   rN  )rT   OverflowErrorrJ  sys
float_infominrV   r/   r/   r0   test_boost_overflow_raises,  s   z)TestBinomTestP.test_boost_overflow_raises)Nr  rC  )r`   ra   rb   __doc__staticmethodrJ  r  rR  rU  rV  rY  r  r  skipifr]  maxsizer`  r/   r/   r/   r0   rB    s    
rB  c                   @   s  e Zd ZdZdd Zdd Zdd Zdd	 Zej	
d
g ddd Zej	
d
g ddd Zej	
dg ddd Zej	
dg ddd Zej	
dg dg dg dg dg dg dg d g d!g d"g d#g d$g d%g d&g d'g d(g d)g d*g d+g d,g d-g d.g d/g d0g d1gd2d3 Zd4d5 Zej	
d6d7d8gd9d: Zd;d< Zd=d> Zd?d@ ZdAdB ZdCS )DTestBinomTestzTests for stats.binomtest.c                 C   s   d}t ddd}t|jd|d t ddd}t|jd|d t d	dd}t|jd
|d t ddd}t|jddd d S )N绽|=iΙ i@o@Q?r   r?   iΙ gX7?i	ϙ g=?iϙ gybw?rC   r   rH  r   r   rM   r@   r]   r/   r/   r0   test_two_sided_pvalues19  s   z%TestBinomTest.test_two_sided_pvalues1c                 C   s   d}t jdddd}t|jd|d t ddd}t|jd	|d t d
dd}t|jd|d t ddd}t|jd|d t ddd}t|jd|d t ddd}t|jd|d t ddd}t|jd|d d S )Nrf  r7   ra  rg  r   r,  gX-h?r?   rw   g㤶y?rr  g	y}?r9   g+,?ru   r   r  g      ?r   g?g{Gz?g333333?g<,?rh  ri  r/   r/   r0   test_two_sided_pvalues2G  s   z%TestBinomTest.test_two_sided_pvalues2c                 C   s   d}t ddd}t|jd|d t ddd	}t|jd|d t d
dd}t|jd|d t ddd}t|jd|d t dd
d}t|jd|d t ddd}t|jd|d d S )Nrf  i  i  r  r   r?   ru   rh  gAL &W?r;   r  g!YB?rY   rR  gE]t?r9   r6   rr  rh  ri  r/   r/   r0   test_edge_casesX  s   zTestBinomTest.test_edge_casesc                    s,  d dd}t t    d }tj| }t jtj| |kdd}t fdd| t    } | t|tj| k }t	||d	d
 d}t t 
  d }tj| }t jtj| |kdd}t fdd|dt 
  }|d }t	||d	d
 d S )Nr   r  ru   r   r   axisc                    s   t j|   S r(   r   binomr  r   rk  r/   r0   <lambda>t  rX   z>TestBinomTest.test_binary_srch_for_binom_tst.<locals>.<lambda>rC   r?   r9   c                    s   t j|  S r(   rp  rr  rk  r/   r0   rs       )r)   rZ   ceilr   rq  r  sumr   intr   floor)rM   kr^   dy1ixy2r/   rk  r0   test_binary_srch_for_binom_tstg  s(    z,TestBinomTest.test_binary_srch_for_binom_tstz"alternative, pval, ci_low, ci_high))r  g$W?r   g?)r  g}?gGI~?r   )rC  g}G?g>?6?g4M?c                 C   X   t jddd|d}t|j|dd t|jd |jdd	}t|j|jf||fdd d S )
NrO  r   r  r   r,  r  r  r?   皙?r  confidence_level	r   rH  r   r   r	   rJ   proportion_cilowhighrM   r  r  ci_lowci_highr]   cir/   r/   r0   test_confidence_intervals1  
   z(TestBinomTest.test_confidence_intervals1))r  ga&+w?r   gny?)r  g5!Rx?g5%?r   )rC  g~th?gSuR&|?g|KnG?c                 C   r  )
Nru   r   r  r  ư>r?   gQ?r$   r  r  r  r/   r/   r0   test_confidence_intervals2  r  z(TestBinomTest.test_confidence_intervals2zalternative, pval, ci_high))r  gլ?g̓A?)r  r   r   )rC  gEow?g;Kڞj?c                 C   sN   t jddd|d}t|j|dd |jdd}t|jd	 t|j|dd d S )
Nr   r   r  rD  r  r?   r  r  r   )r   rH  r   r   r  r	   r  r  )rM   r  r  r  r]   r  r/   r/   r0   !test_confidence_interval_exact_k0  
   z/TestBinomTest.test_confidence_interval_exact_k0zalternative, pval, ci_low))r  r   r   )r  6>g@>}_?)rC  r  gcڒ ?c                 C   sN   t jddd|d}t|j|dd |jdd}t|jd t|j|dd d S )	Nr   r  rD  r  r?   r  r  r   )r   rH  r   r   r  r	   r  r  )rM   r  r  r  r]   r  r/   r/   r0   %test_confidence_interval_exact_k_is_n  r  z3TestBinomTest.test_confidence_interval_exact_k_is_nz+k, alternative, corr, conf, ci_low, ci_high)ru   rC  Tr  g`?g?)ru   rC  Tr$   grt!?g/?)ru   rC  Fr  gVK5?g5dM?)ru   rC  Fr$   gXCVu^?g*r[?)ru   r  Tr  r   gP-V?)ru   r  Tr$   r   g^'?)ru   r  Fr  r   gS5a?)ru   r  Fr$   r   g?)ru   r  Tr  g!Շ`ⰸ?r   )ru   r  Tr$   g$x"?r   )ru   r  Fr  gW~=?r   )ru   r  Fr$   g	L?r   )r   rC  Tr  r   gX?)r   rC  Fr  r   gN?)r   r  Tr  r   g/<7#9?)r   r  Fr  r   gAA?)r   r  Tr  r   r   )r   r  Fr  r   r   )r   rC  Tr  g7?r   )r   rC  Fr  g؍s?r   )r   r  Tr  r   r   )r   r  Fr  r   r   )r   r  Tr  ghadn?r   )r   r  Fr  gAB/?r   c           
      C   sL   t j|dd|d}|rd}nd}|j||d}	t|	j|	jf||fdd d S )	Nr   r:  r  wilsonccwilson)r  methodr  r?   )r   rH  r  r   r  r  )
rM   ry  r  corrconfr  r  r]   r  r  r/   r/   r0   test_ci_wilson_method  s   z#TestBinomTest.test_ci_wilson_methodc                 C   s*   t ddd}t|jd t|jd d S )Nrw   rs  r  r   )r   rH  r	   rJ   r   rX  r/   r/   r0   &test_estimate_equals_hypothesized_prop  s   z4TestBinomTest.test_estimate_equals_hypothesized_propzk, n)r   r   )r   r   c                 C   s@   t jtdd t|| W d    d S 1 sw   Y  d S )Nz must be an integer not less thanr  r  r   rU   r   rH  )rM   ry  r   r/   r/   r0   test_invalid_k_n  s
   "zTestBinomTest.test_invalid_k_nc                 C   sB   t jtdd tddd W d    d S 1 sw   Y  d S )Nzk must not be greater than nr  rr  r   r  r  rV   r/   r/   r0   test_invalid_k_too_big  s
   "z$TestBinomTest.test_invalid_k_too_bigc                 C   P   t jdddd}tjtdd |jdd W d    d S 1 s!w   Y  d S )	Nru   r   r:  rk  zmust be in the intervalr  r   r  r   rH  r  r   rU   r  rX  r/   r/   r0   test_invalid_confidence_level     "z+TestBinomTest.test_invalid_confidence_levelc                 C   r  )	Nru   r   r:  rk  zmethod must ber  zplate of shrimpr  r  rX  r/   r/   r0   test_invalid_ci_method  r  z$TestBinomTest.test_invalid_ci_methodc                 C   s"   t jdddd}t|j|j d S )Nru   r   r:  rk  )r   rH  r	   proportion_estimaterJ   rX  r/   r/   r0   
test_alias  s   zTestBinomTest.test_aliasN)r`   ra   rb   ra  rj  rl  rm  r~  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r/   r/   r/   r0   re  2  st     


	
	


re  c                   @   r~   )TestFlignerc                 C   s&   t d}tt||d dd d S )Nrx   r   )gnf	@gDq?rr  )r)   rZ   r   r   fligner)rM   r   r/   r/   r0   r  
  s
   
zTestFligner.test_datac           	         sv   t jd  fdd}|t}|t}|t}tj|||dd\}}tj|||ddd\}}t|| t|| d S )	Nr  c                    s    t | d t|    S )Nrf  )r)   asarrayrandnr   tolist)gr   r/   r0   rs    s     z+TestFligner.test_trimmed1.<locals>.<lambda>rO   r  r  r   r  )	r)   r  r   r  r  r  r   r  r   )	rM   _perturbg1_g2_g3_Xsq1r"  Xsq2r$  r/   r  r0   r%    s   

zTestFligner.test_trimmed1c                 C   sb   g d}g d}t j||ddd\}}t j|dd |dd dd	\}}t|| t|| d S )
Nr&  r-  r  r0  r  r   r   rO   r  )r   r  r   )rM   r[   r  r  r"  r  r$  r/   r/   r0   r5  !  s   

$
zTestFligner.test_trimmed2c                 C   r8  r9  )r)   r  rT   r<  r   r  r   r/   r/   r0   r=  ;  r   zTestFligner.test_bad_keywordc                 C   r8  r>  )r)   r  rT   rU   r   r  r   r/   r/   r0   r@  ?  r   z!TestFligner.test_bad_center_valuec                 C   r  ri   )rT   rU   r   r  rV   r/   r/   r0   test_bad_num_argsC  r  zTestFligner.test_bad_num_argsc                 C   s.   t d}tt jt jft||d g  d S )Nrx   r   )r)   rZ   r	   r   r   r  r   r/   r/   r0   r  G  s   
$zTestFligner.test_empty_argN)
r`   ra   rb   r  r%  r5  r=  r@  r  r  r/   r/   r/   r0   r    s    r  c                  c   s    g d} g d}t |D ]S\}}tj|}|d}|jdddd}|jdddd}t||D ]\}}	t|d ||	 D ]}
|| ||
< q>q1|| t|d	\}}||d
g| | R V  qd S )N))g]YCgQɣ?)gme5gϵ5?)g;jgH?)iFeiHcMi%:r   r   r  rx   )r  r  r   r   r   r  )		enumerater)   r  r  integersziprangeshufflesplit)expected_resultsseedssir1  r  xytie_indnum_ties_per_indr^   r   jr[   r  r/   r/   r0   mood_cases_with_tiesL  s    

r  c                   @   s   e Zd Zejde dd Zejdg ddd Zdd	 Z	d
d Z
dd Zdd Zdd Zdd Zejdg ddd ZdS )TestMoodz$x,y,alternative,stat_expect,p_expectc                 C   s4   t j|||d\}}t||dd t||dd dS )am  
        Example code used to generate SAS output:
        DATA myData;
        INPUT X Y;
        CARDS;
        1 0
        1 1
        1 2
        1 3
        1 4
        2 0
        2 1
        2 4
        2 9
        2 16
        ods graphics on;
        proc npar1way mood data=myData ;
           class X;
            ods output  MoodTest=mt;
        proc contents data=mt;
        proc print data=mt;
          format     Prob1 17.16 Prob2 17.16 Statistic 17.16 Z 17.16 ;
            title "Mood Two-Sample Test";
        proc print data=myData;
            title "Data for above results";
          run;
        r  gؗҜ<r!  Nr   moodr   )rM   r[   r  r  stat_expectp_expectrJ   r   r/   r/   r0   test_against_SASd  s   zTestMood.test_against_SASzalternative, expected))rC  )3oXQ?gpL?)r  )r  gԣl?)r  )r  gpL?c                 C   s.   g d}g d}t j|||d}t|| d S )Nr  r  r  r  )rM   r  r  r[   r  r]   r/   r/   r0   test_against_SAS_2  s   	zTestMood.test_against_SAS_2c                 C   s`   t jd t jdd}t jdd}t||\}}t||\}}t||g| |g d S )Nr/  r   r   rY   )r)   r  r1  r  r   r  r   )rM   r   r   z1p1z2p2r/   r/   r0   test_mood_order_of_args  s   z TestMood.test_mood_order_of_argsc                 C   sN   g d}g d}t |}t |}d|_d|_ttj||d dddg d S )N)gg۟?gy<7xg1)E?g9cç?g
FAg<=ӟ	2?g{![?g_l?gڻ{ӿgGSsA0?g1?g 3ggL?g_lg,*.gː53?gvNqG?gW=?)gd"g `Y#?g^g?gNgBagp5_%?g(?gv]lοgg?ggr6ʺ?ḡj?gK!ٿg{g&?g*h}gZ?gyBU?g .ۋ4?g|w;?g&ck @gRj2g}+w(o?g@F?g}_9t?g =g	S?gx(gg7Y?g{ih?)r   r   )rY   r   rn  g4g^}E	?)r)   r   shaper   r   r  )rM   r   r   r/   r/   r0   test_mood_with_axis_none  s   

z!TestMood.test_mood_with_axis_nonec              
   C   s   d}t jd t jd|}t jd|}t||\}}t|D ]}t|| || gt|d d |f |d d |f  q"| }| }tj||dd\}}t|D ]}t|| || gt||d d f ||d d f  qVd S )Nrx   r/  r   rY   r   rn  )	r)   r  r1  r  r   r  r  r   	transpose)rM   nyr   r   	z_vectestpval_vectestr  r^   r/   r/   r0   test_mood_2d  s"   ""zTestMood.test_mood_2dc              	   C   s>  d}t jd t jj| }t jj| }tdD ]}tj|||d\}}ddgddgddgf}t||| d  D ]b}t||| d  D ]U}	|dkr^|d d ||	f }
|d d ||	f }n)|dkru||d d |	f }
||d d |	f }n|||	d d f }
|||	d d f }t|||	f |||	f gt|
| qEq9qd S )N)r   rx   r6   r/  ru   rn  r   r   r   )r)   r  r1  r  r  r   r  r   )rM   r  r   r   ro  r  r  axes_idxr^   r  slice1slice2r/   r/   r0   test_mood_3d  s2   

zTestMood.test_mood_3dc                 C   s   t ttjdgg  d S ri   )rT   rU   r   r  rV   r/   r/   r0   test_mood_bad_arg     zTestMood.test_mood_bad_argc           	      C   s   t jd tjjddd}tjjddd}tj||dd\}}tj||dd\}}tj||d	d\}}||  kr?|ksBJ  J t|dd
d t||d  t|d|d   tj	t
dd tj||dd W d    d S 1 stw   Y  d S )Nr   rM  r   )r   r   g      ?rC  r  r  r  Hz>r!  r   r   zalternative must be...r  z	ekki-ekki)r)   r  r1  r   r   r,   r  r   r  r   rU   )	rM   r[   r  stat1r  stat2r  stat3p3r/   r/   r0   test_mood_alternative  s   "zTestMood.test_mood_alternativer  )rC  r  r  c                 C   sF   t jd}|d}|d}tj|||d}t|j|jf| d S )Nl	   hTt fU6H~ )r   r   )rY   r   r  )	r)   r  r  r   r   r  r	   rJ   r   )rM   r  r  r   r   r]   r/   r/   r0   test_result  s
   

zTestMood.test_resultN)r`   ra   rb   r  r  r  r  r  r  r  r  r  r  r  r  r  r/   r/   r/   r0   r  c  s"    
 
	
r  c                   @   sV   e Zd Zdd Zdd Zdd Zejje	 ddd	d
 Z
dd Zdd Zdd ZdS )TestProbplotc                 C   sl   t jjddd}t j|dd\}}g d}t|t| t|| t j|dd\}}g d}t|| d S )	NrO  90  r  Fr   )g+/gY3gtg5Z0gFjg!2gsR{ܿg7Կg>hǿgcz隇gcz隇?g>h?g7?gsR{?g!2?gFj?g5Z0?gt?gY3?g+/?T)gƝV?g?gà?)r   r   r,   probplotr   r)   sort)rM   r[   osmosrosm_expectedr]   res_fitres_fit_expectedr/   r/   r0   rR     s   
zTestProbplot.test_basicc           
      C   s   t jjddd}t j|d dd\}}t j|ddd\}}t j|ddd\}}t|| t|| t|| t|| t j|ddd\}}	d S )Nr   i@ r  Fsparamsr   r   r/   )r   r   r,   r  r   )
rM   r[   osm1osr1osm2osr2osm3osr3r  r  r/   r/   r0   test_sparams_keyword!  s   



z!TestProbplot.test_sparams_keywordc                 C   s   t jjddd}t j|dddd\}}t j|dt jdd\}}t|| t|| ttt j|dd	 ttt j|g d	 G d
d d}t j|ddd\}}t j|| dd\}}t|| t|| d S )NrO  r  r  Ft)ru   )r   r   r  zwrong-dist-namer   c                   @   s   e Zd ZdZdd ZdS )z3TestProbplot.test_dist_keyword.<locals>.custom_distz6Some class that looks just enough like a distribution.c                 S   s   t jj|ddS )Nr   )r   )r   r   ppf)rM   qr/   r/   r0   r  ;  rt  z7TestProbplot.test_dist_keyword.<locals>.custom_dist.ppfN)r`   ra   rb   ra  r  r/   r/   r/   r0   custom_dist9  s    r  )r   r  )r   r   )	r   r   r,   r  r  r   rT   rU   AttributeError)rM   r[   r  r  r  r  r  r/   r/   r0   test_dist_keyword/  s   


zTestProbplot.test_dist_keywordno matplotlibrZ  c           
      C   s  t  }|d tjjdddd}tj|t d\}}t   tj|d d\}}tj|dt d}t   tj|dd d}tt	|t	|  koXt	|  koXt	|  koXd	kn   t
|| t
|| t
|| t
|| t  }|d}	tj|d|	d t   d S )
Nr  ru   r   t r  plotF)r   r  r   )pltfigureadd_subplotr   r  r,   r  closer   r   r   )
rM   figr[   res1fitres1res2fitres2res3res4axr/   r/   r0   test_plot_kwargC  s$   
@




zTestProbplot.test_plot_kwargc                 C   r   r   )rT   rU   r   r  rV   r/   r/   r0   test_probplot_bad_args\     z#TestProbplot.test_probplot_bad_argsc                 C   sZ   t tjg ddtg tg f t tjg ddtg tg ftjtjdff d S )NFr  Tr   )r	   r   r  r)   r   r   rV   r/   r/   r0   
test_empty`  s   zTestProbplot.test_emptyc                 C   sj   t jdd% ttjdgddt dgt dgft jt jdff W d    d S 1 s.w   Y  d S )Nr   )invalidr   Tr  r   )r)   r   r	   r   r  r   r   rV   r/   r/   r0   test_array_of_size_oneg  s   "z#TestProbplot.test_array_of_size_oneN)r`   ra   rb   rR   r  r  r  r  rc  have_matplotlibr  r  r  r  r/   r/   r/   r0   r    s    
r  c                   @   s   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd Zdd Zejdg dg dg dgdd Zdd ZdS ) TestWilcoxonc                 C   sh   t ttjdgddg t ttjddgddgd t ttjddgddgdd t ttjdgd dd d S )Nr   r   dummyr  r   xyzmode)rT   rU   r   wilcoxonrV   r/   r/   r0   test_wilcoxon_bad_argo  s   z"TestWilcoxon.test_wilcoxon_bad_argc                 C   sR   t d}tttj||ddd tttj||ddd ttj||dddd d S )NrO  wilcoxapproxr  prattzsplit)g     @Z@r   )r)   rZ   rT   rU   r   r  r	   r   r/   r/   r0   test_zero_diffx  s   
zTestWilcoxon.test_zero_diffc                 C   sd   g d}g d}t  }|jtdd tj||ddd}W d    n1 s&w   Y  t|d d S )	Nr  )r   r   ru   rx   Sample size too smallr  r  r  zero_methodr  )r   g|N?)r
   r  r  r   r  r   )rM   r[   r  r  r]   r/   r/   r0   
test_pratt  s   zTestWilcoxon.test_prattc                 C   s<   g d}t j|ddd}t j|ddd}t j|ddd}d S )N)r   r   ru   r   r   ru   r   r   r   r   r   r  r  r   r  r  )r   r  )rM   arrr  r/   r/   r0   test_wilcoxon_arg_type  s   z#TestWilcoxon.test_wilcoxon_arg_typec                 C   s2  g d}t dd}tdd t||D }t|j}tj||ddd\}}t|d	 t|d
 tj||ddd\}}t|d t|d tj||ddd\}}t|d t|d t	g d}t	g d}tj||ddd\}}t
|d t|ddd tj||ddd\}}t
|d t|ddd d S )N)	r   rw   rs  rY   r:   rw   rx   r   r   rx   c                 S   s   g | ]	\}}|g| qS r/   r/   )rq   ur   r/   r/   r0   rr     s    z7TestWilcoxon.test_accuracy_wilcoxon.<locals>.<listcomp>r  r  r  i  g.i?r  i  gFYv]Uj?r  iG  g%^ Ez?rn  r  r5  r        y   r     r     p         w      r-        r  s   r*  rz     F
correctionr  r@  g^҂<?r  r?   Tg:b+?)r  r)   concatenater  zerosr   r   r  r   r   r	   )rM   freqnumsr[   r  r  r,  r/   r/   r0   test_accuracy_wilcoxon  s*   








z#TestWilcoxon.test_accuracy_wilcoxonc                 C   s@   t g d}t g d}tj||ddd}d}t|| d S )Nr'  r.  Fr  r7  r  )r)   r   r   r  r   )rM   r[   r  r]   r\   r/   r/   r0   test_wilcoxon_result_attributes  s
   z,TestWilcoxon.test_wilcoxon_result_attributesc                 C   s   t jd}|d|d}}tj||dd}tj|jd }t|j	| tj||dd}t
|dr7J t||}t
|drDJ d S )Nl   	m$S rY   r  r  r   exact
zstatistic)r)   r  r  r   r  r   r  r   r   r@  hasattr)rM   r  r[   r  r]   r   r/   r/   r0   test_wilcoxon_has_zstatistic  s   z)TestWilcoxon.test_wilcoxon_has_zstatisticc                 C   sn   t jdgd dd\}}d}t|d t||dd t jdgd d	dd
\}}d}t|d t||dd d S )Nr:  r   r  r  g4IťY?r   r  r?   Tr7  gc2_?)r   r  r	   r   )rM   statr,  
expected_pr/   r/   r0   test_wilcoxon_tie  s   	

zTestWilcoxon.test_wilcoxon_tiec                 C   s  g d}g d}t  }|jtdd tj||ddd\}}W d    n1 s(w   Y  t|d t|d	d
d t  }|jtdd tj||dddd\}}W d    n1 sZw   Y  t|d t|dd
d t  }|jtdd tj||ddd\}}W d    n1 sw   Y  t|d t|dd
d t  }|jtdd tj||dddd\}}W d    n1 sw   Y  t|d t|dd
d d S )N)
}   r5  rZ     rG  r5  rG  rF  rG     )
r  z   rF  rn  rG  |   r     rH     r  r  r  r  r  r  rb  g13}?r6   rF   T)r  r8  r  g7a%?r  g?gڏoJT?)r
   r  r  r   r  r	   r   )rM   r[   r  r  r   r,  r/   r/   r0   test_onesided  s<   	



zTestWilcoxon.test_onesidedc                 C   sZ   t ddD ]%}t|}t|}t||d  d d t| tt|d t|| qd S )Nr   rx  r   )r  r   r   r	   r   rv  r   )rM   r   pmf1pmf2r/   r/   r0   test_exact_basic  s   zTestWilcoxon.test_exact_basicc                 C   s  t g d}t g d}tj||ddd\}}t|ddd tj||d	dd\}}t|d
dd tj||ddd\}}t|ddd t ddd }t ddd}tj||ddd\}}t|ddd tj||d	dd\}}t|ddd tj||ddd\}}t|ddd d S )N)
g(\?g=
ףp=?g(\?gQ޿gQ?g{Gz?gp=
ףgq=
ףp?g      gQ?)
r   g?gɿg333333?皙gܿgzGgQοg(\gRQrC  r?  rM  g֔  ?r6   rF   r  gYJ?r  g*  ?r   rO  r  r   g(?g(?gS?)r)   r   r   r  r   rZ   )rM   r[   r  r  r,  r/   r/   r0   test_exact_pval	  s    zTestWilcoxon.test_exact_pvalr[   )r   r   ru   )r   r   r%  rx   )r   r   ru   r%  r  ir9   r:   c                 C   s@   t |\}}t|}||dk  }t|| t|d d S Nr   r   )r   r  r)   r   rv  r	   )rM   r[   r   r,  wtruer/   r/   r0   test_exact_p_1%  s
   

zTestWilcoxon.test_exact_p_1c                 C   s   t ddd }t ddd}tt||tj||dd t dd}t }|jtdd	 t|\}}W d    n1 s@w   Y  ttj|d
d||f t dd}tt|tj|d
d d S )Nr   rT  r  r   r?  r  r;   zExact p-value calculationr  r  r   rt  )r)   rZ   r	   r   r  r
   r  r  )rM   r[   r  rz  r  r   r,  r/   r/   r0   	test_auto/  s   zTestWilcoxon.test_autoN)r`   ra   rb   r  r  r"  r$  r=  r>  rB  rE  rN  rQ  rS  r  r  r  rW  rX  r/   r/   r/   r0   r  n  s$    	
	&
r  c                   @   rc   )
	TestKstatc                    s   t jd t jd  fdddD }g d}t||dd tj d	d
}tj dd
}tj dd
}t|||f|d d ddd d S )Ni}  r  c                    s   g | ]}t  |qS r/   )r   kstat)rq   r   rN   r/   r0   rr   F  s    z>TestKstat.test_moments_normal_distribution.<locals>.<listcomp>r  )gHQ,?glZ)rI?g88}?gcJN?r  r?   r   )momentr   ru   r   g{Gz?{Gz?r  )r)   r  r1  r  r   r   r\  )rM   momentsr  m1m2m3r/   r[  r0    test_moments_normal_distributionC  s   "z*TestKstat.test_moments_normal_distributionc                 C   rS   r(   )rT   rU   r   rZ  rV   r/   r/   r0   rW   Q  rX   zTestKstat.test_empty_inputc                 C   *   t d}t j|d< tt|t j d S NrA   r6   r)   rZ   r   r	   r   rZ  rj   r/   r/   r0   r   T     

zTestKstat.test_nan_inputc                 C   s*   t d}dD ]}tttj||d qd S )Nr   )r   g/$@rT  )r)   rZ   rT   rU   r   rZ  rM   rN   r   r/   r/   r0   test_kstat_bad_argZ  s   
zTestKstat.test_kstat_bad_argN)r`   ra   rb   rb  rW   r   rh  r/   r/   r/   r0   rY  B  s
    rY  c                   @   r3   )TestKstatVarc                 C   rS   r(   rT   rU   r   kstatvarrV   r/   r/   r0   rW   b  rX   zTestKstatVar.test_empty_inputc                 C   rc  rd  re  rj   r/   r/   r0   r   e  rf  zTestKstatVar.test_nan_inputc                 C   s    dg}d}t ttj||d d S )Nr   r   rT  rj  rg  r/   r/   r0   rk   k  s   zTestKstatVar.test_bad_argN)r`   ra   rb   rW   r   rk   r/   r/   r/   r0   ri  a  s    ri  c                   @   sN   e Zd Zdd Zdd Zdd Zejje	 ddd	d
 Z
dd Zdd ZdS )TestPpccPlotc                 C      t ddddd | _d S Nrx     r  r  r1   r[   rV   r/   r/   r0   setup_methods  r2   zTestPpccPlot.setup_methodc                 C   H   d}t j| jdd|d\}}g d}t|tjdd|d t|| d S )Nrx   r   N)g9%	?gxM _?g";9?g®XZ?g?num)r   	ppcc_plotr[   r   r)   r  )rM   ru  svalsppccppcc_expectedr/   r/   r0   rR   v  
   zTestPpccPlot.test_basicc                 C   s   t j| jdddd\}}t j| jddt jd\}}t||dd t||dd t | jdd\}}t||dd t||dd d S )Nrs  r   tukeylambdar   g#B;r?   )r   rx  r[   r}  r   )rM   svals1ppcc1svals2ppcc2svals3ppcc3r/   r/   r0   	test_dist~  s   
zTestPpccPlot.test_distr  rZ  c                 C   Z   t  }|d}tj| jddt d || |d}tj| jdd|d t   d S Nr  irO  r  )r   r  r  r   rx  r[   delaxesr  rM   r  r  r/   r/   r0   r       


zTestPpccPlot.test_plot_kwargc                 C   s2   t ttj| jdd t ttjg ddddd d S )Nr   r   rt   r   r   )rT   rU   r   rx  r[   rV   r/   r/   r0   test_invalid_inputs  s   
z TestPpccPlot.test_invalid_inputsc                 C   s@   t g dd\}}t|tjdddd t|tjdtd d S )Nr   r   r}  rv  r   )r   rx  r   r)   r  r:  float)rM   ry  rz  r/   r/   r0   r    s   zTestPpccPlot.test_emptyN)r`   ra   rb   rq  rR   r  r  r  rc  r  r  r  r  r/   r/   r/   r0   rl  r  s    
rl  c                   @   rc   )
TestPpccMaxc                 C   s   dg}t ttj|dd d S r   )rT   rU   r   ppcc_maxrj   r/   r/   r0   test_ppcc_max_bad_arg  s   z!TestPpccMax.test_ppcc_max_bad_argc                 C   s2   t jjddddddd }tt |dd	d
 d S )Nffffffr   r  '   r        @{r9   rF   )r   r}  r,   r   r  r   r/   r/   r0   test_ppcc_max_basic  s   zTestPpccMax.test_ppcc_max_basicc                 C   sp   t jjddddddd }t j|dd	}t j|t jd	}t|d
dd t|d
dd t |}t|d
dd d S )Nr  r   r  r  r  r   r  r}  r   r  rx   rF   )r   r}  r,   r  r   )rM   r[   max1max2max3r/   r/   r0   r    s   
zTestPpccMax.test_distc                 C   s`   t jjddddddd }ttt j|dd	 tt j|d
d	ddd tt j|dd	ddd d S )Nr  r   r  r  r  r   r  )r   r   r  )brack)r   r   r  r9   rF   )r   r   )r   r}  r,   rT   rU   r  r   r   r/   r/   r0   
test_brack  s   
zTestPpccMax.test_brackN)r`   ra   rb   r  r  r  r  r/   r/   r/   r0   r    s
    r  c                   @   r  )TestBoxcox_llfc                 C   sT   t jjdddd}d}t ||}|j d tt| d  }t	|| d S )Nr  r   1  r   r   r   r   r'  r   )
r   r   r,   
boxcox_llfr   r)   r*   rv  rQ   r   )rM   r[   lmbdallfllf_expectedr/   r/   r0   rR     s
   $zTestBoxcox_llf.test_basicc                 C   D   t jjdddd}d}t ||}t |t|}t||dd d S Nr   r   r  r  r   r  r?   )r   r   r,   r  listr   rM   r[   r  r  llf2r/   r/   r0   test_array_like  
   zTestBoxcox_llf.test_array_likec                 C   P   t jjdddd}d}t ||}t |t||gj}t||g|dd d S r  )r   r   r,   r  r)   vstackr  r   r  r/   r/   r0   test_2d_input  s
   zTestBoxcox_llf.test_2d_inputc                 C      t ttdg  d S ri   )r   r)   isnanr   r  rV   r/   r/   r0   r       zTestBoxcox_llf.test_emptyc                 C   s"   g d}t d|}t|d d S )N)g     h@      m@r  g     x@ig#:x1)r   r  r   )rM   rN   r  r/   r/   r0   test_gh_6873  s   zTestBoxcox_llf.test_gh_6873N)r`   ra   rb   rR   r  r  r  r  r/   r/   r/   r0   r    s    
r  (  iU>  iϵ i i_
 i7 i is iv  i+ i* i-q ir: i  i i i i_ i4  i iD  i i, iE i\  i i| ig  iM iӢ i[  i! i if6 iP i i iI i/ iF& i i i`# i i  i& i i iQ ia i
 iXZ iw  i_ im  iD i- iB  i:C i	 i' i  i % ix i  iz_ iQ i!# i= i  ix i
 iU7 i_ i1k iW9 i& ix iI i  iT i
 i i  irv im i͟  i iO iv i) iP i4 i= ii  iU i i ig> ix	 i` i| i0 i` iM	 i ' i[ i8 i  i3[ i  i) i`N i
	 i  i[ i:+ ip i	% i
  iV i[$ iX  i_ ib ieZ i i`?	 it i  i iE i iT iy i imv i< iZ i! i i(  ir
 i i iɪ i| i i> ii  i i* i# ix i i& i i# i1g iP i) iۆ i{% i? i۽ i| i i& i- iư) i+c i7  iQ i	o i& i|_ iy
 i6 i  i i i:J i#j i8 i  i%  i i%W i( i` i: ina  i  i i% i$ i4 iA( iq  i  i" iy iXZ i izr i i$ i; is igY i i7 ip	 ip iL6" iDH i i!3 i iI( i|C iш i im  iS i  i{8 i  i i iP i2L i( i@ is iu i2 i_ ij i)m igZ i_ iu	 i9 ik  i'; i ii iK i܃ i i= iv i= i ia  ir  i i i}W i*  i  iJ i i  i2  iUF iNj	 iV," ip iA i% i iv9 iI5 i  iT- i  i8~ i?e` i1 iV  i/X i= i4 i iح i^ i i/ ig i( i i|B; ip i ic iE' iE i$
 i i{ ik i: iI iA_ i iP ioP  i6_ i iܲ iO	 iA7 iL  iA i iV  iwm i iZh i|V iԇC iw i& i~ i i'  i!  i i
 i} ie i3 i i i_ i  i@) i  iV ii iF i? i&
 i"Y iw i  iH iS ird ib i@ iK iA; iz| id+ i	 i
8 i"  i	 iO i_ i$l i!9 ii iT iuL iQ'% iqp iC i]K@ i! i i  iIp| i i?	B i i io0 i i

 i  i5% i i i* i i iv0 i* i* iw i  ive i$ i i- iD~ i- i i,4 ib i  i iP0 i) im i	 iR iQ i
 i i c                   @   s   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Ze	j
dg ddd Zdd Ze	j
dg ddd Zdd ZdS )
TestBoxcoxc                 C   s   t ddddd }tj|dd}t||d  tj|dd}t|dd|   tj|dd}t|t| tjt|dd}t|t| ttd	d
}t|t	d	 d S )Nrx   r   r  r  r   r  r   r   r   r   )
r1   r   r   r   r)   r*   r  r   r	   r:  )rM   r[   xtr/   r/   r0   test_fixed_lmbda)  s   zTestBoxcox.test_fixed_lmbdac                 C   sL   d}t jjdddd}|| d |  }t |\}}t|d| dd	 d S )
Ng      @r   iP  i  )r   r   r   r   r   r   rF   )r   r   r,   r   r   )rM   r  r[   x_invr  r   r/   r/   r0   test_lmbda_None;  s
   zTestBoxcox.test_lmbda_Nonec                 C   s   t jd}tdd|dd }tj|dd\}}}t|ddg tj|d	d\}}}t|d
dg tdd|dd }tj|dd\}}}t|ddg tj|dd\}}}t|ddg d S )Nr/  rx   r   r  rM  )alphagx@gv[%@r  gxˉEk?g8Bhk @r9   ro  rY   r  g\?gx̡&@r   g7(U@gp:sY@)r)   r  r   r1   r   r   r   )rM   r  r[   r  rf   r/   r/   r0   
test_alphaE  s   zTestBoxcox.test_alphac                 C   sR   t ddg}tttj| tttjt dg tttjt dgdgg d S )Nr   r   r   )r)   r   rT   rU   r   r   r   r/   r/   r0   test_boxcox_bad_argV  s    zTestBoxcox.test_boxcox_bad_argc                 C      t tg jdk d S N)r   )r   r   r   r  rV   r/   r/   r0   r  _  r2   zTestBoxcox.test_emptyc                 C   s    t t\}}t|ddd d S )NgsHjdrr   r?   r   r   _boxcox_datar   )rM   r  lamr/   r/   r0   r  b  s   zTestBoxcox.test_gh_6873bounds)r   r   )r   r   )r   rR  c                    sH    fdd}t jtd |d\}} d |  k r d k s"J  J d S )Nc                       t j|  ddS Nbounded)r  r  r   minimize_scalarfunr  r/   r0   	optimizerm     zBTestBoxcox.test_bounded_optimizer_within_bounds.<locals>.optimizerr  r  r   r   )r   r   r  )rM   r  r  r  r  r/   r  r0   $test_bounded_optimizer_within_boundsj  s   (z/TestBoxcox.test_bounded_optimizer_within_boundsc                    sj   t jtd d\}}|d |d f ddi fdd}t jtd |d\}}||ks,J t| d	  d S )
Nr  r:  r   xatolr  c                    s   t j|  ddS )Nr  )r  r  optionsr  r  r  r  r/   r0   r    s   zPTestBoxcox.test_bounded_optimizer_against_unbounded_optimizer.<locals>.optimizerr  r   r  )rM   r  r  r  lmbda_boundedr/   r  r0   2test_bounded_optimizer_against_unbounded_optimizert  s   
z=TestBoxcox.test_bounded_optimizer_against_unbounded_optimizerr  )str)r   r   r:  c                 C   sD   t jtdd tjtd |d W d    d S 1 sw   Y  d S )Nz`optimizer` must be a callabler  r  r  r   rU   r   r   r  rM   r  r/   r/   r0   $test_bad_optimizer_type_raises_error  s   "z/TestBoxcox.test_bad_optimizer_type_raises_errorc                 C   sP   dd }d}t jt|d tjtd |d W d    d S 1 s!w   Y  d S )Nc                 S   s   dS ri   r/   r  r/   r/   r0   r    s   zCTestBoxcox.test_bad_optimizer_value_raises_error.<locals>.optimizerz;`optimizer` must return an object containing the optimal...r  r  r  )rM   r  r  r/   r/   r0   %test_bad_optimizer_value_raises_error  s
   "z0TestBoxcox.test_bad_optimizer_value_raises_errorN)r`   ra   rb   r  r  r  r  r  r  r  r  r  r  r  r  r  r/   r/   r/   r0   r  '  s    
	
	
r  c                   @   sh   e Zd Zdd Zdd Zdd Zdd Zej	d	g d
ej	dg ddd Z
dd Zdd ZdS )TestBoxcoxNormmaxc                 C   rm  Nrx   r   r  r  rp  rV   r/   r/   r0   rq    r2   zTestBoxcoxNormmax.setup_methodc                 C      t | j}t|ddd d S )Nm?r  r?   r   boxcox_normmaxr[   r   rM   r   r/   r/   r0   test_pearsonr     zTestBoxcoxNormmax.test_pearsonrc                 C   s<   t j| jdd}t|ddd t | j\}}t|| d S )Nmler  	.!?r  r?   )r   r  r[   r   r   )rM   r   r  maxlog_boxcoxr/   r/   r0   test_mle  s   zTestBoxcoxNormmax.test_mlec                 C   s&   t j| jdd}t|ddgdd d S )Nr   r  r  r  r  r?   r  )rM   
maxlog_allr/   r/   r0   test_all  s   zTestBoxcoxNormmax.test_allr  )r  pearsonrr   r  r  c                    sN    fdd}t j| j||d}t d |k sJ t| d k s%J d S )Nc                    r  r  r  r  r  r/   r0   r    r  zITestBoxcoxNormmax.test_bounded_optimizer_within_bounds.<locals>.optimizer)r  r  r   r   )r   r  r[   r)   r   )rM   r  r  r  r   r/   r  r0   r    s   
z6TestBoxcoxNormmax.test_bounded_optimizer_within_boundsc                    sv   t | j}t|d}t|d |d dG dd d  fdd}t j| j|d}||ks3J t||d	 d S )
Nrx   r]  i  c                   @   s   e Zd ZdS )z?TestBoxcoxNormmax.test_user_defined_optimizer.<locals>.MyResultN)r`   ra   rb   r/   r/   r/   r0   MyResult  s    r  c                    s6   g }D ]	}| | | q  }t| |_|S r(   )appendr)   argminr[   )r  objsr  r]   r  lmbda_ranger/   r0   r    s   z@TestBoxcoxNormmax.test_user_defined_optimizer.<locals>.optimizer)r  r   )r   r  r[   r)   roundr  r   )rM   r  lmbda_roundedr  lmbda2r/   r  r0   test_user_defined_optimizer  s   	z-TestBoxcoxNormmax.test_user_defined_optimizerc                 C   s^   t j}tj| jd |d tjtdd tj| jd|d W d    d S 1 s(w   Y  d S )N)r  r  z,`brack` must be None if `optimizer` is givenr  )g       r'  )r   r  r   r  r[   r  r   rU   r  r/   r/   r0   2test_user_defined_optimizer_and_brack_raises_error  s   
"zDTestBoxcoxNormmax.test_user_defined_optimizer_and_brack_raises_errorN)r`   ra   rb   rq  r  r  r  r  r  r  r  r  r  r/   r/   r/   r0   r    s    r  c                   @   sF   e Zd Zdd Zdd Zejje dddd Z	d	d
 Z
dd ZdS )TestBoxcoxNormplotc                 C   rm  rn  rp  rV   r/   r/   r0   rq    r2   zTestBoxcoxNormplot.setup_methodc                 C   rr  )Nrx   rs  r   rt  )g!X4}?g΁i?g-	15?gW$?g\ܱ{?rv  )r   boxcox_normplotr[   r   r)   r  )rM   ru  lmbdasrz  r{  r/   r/   r0   rR     r|  zTestBoxcoxNormplot.test_basicr  rZ  c                 C   r  r  )r   r  r  r   r  r[   r  r  r  r/   r/   r0   r    r  z"TestBoxcoxNormplot.test_plot_kwargc                 C   s.   t ttj| jdd t ttjddgdd d S )Nr   r   r   )rT   rU   r   r  r[   rV   r/   r/   r0   r    s   z&TestBoxcoxNormplot.test_invalid_inputsc                 C   s   t tg ddjdk d S rU  )r   r   r  r   rV   r/   r/   r0   r    r  zTestBoxcoxNormplot.test_emptyN)r`   ra   rb   rq  rR   r  r  rc  r  r  r  r  r/   r/   r/   r0   r    s    
r  c                   @   r3   )TestYeojohnson_llfc                 C   r  )Nr   r   r  r  r   r  r?   )r   r   r,   yeojohnson_llfr  r   r  r/   r/   r0   r  	  r  z"TestYeojohnson_llf.test_array_likec                 C   r  r  )r   r   r,   r  r)   r  r  r   r  r/   r/   r0   r    s
   z TestYeojohnson_llf.test_2d_inputc                 C   r  ri   )r   r)   r  r   r  rV   r/   r/   r0   r    r  zTestYeojohnson_llf.test_emptyN)r`   ra   rb   r  r  r  r/   r/   r/   r0   r    s    r  c                   @   s   e Zd Zdd Zejdg ddd Zdd Zd	d
 Z	ejde
je
jgdd Zejde
je
je
je
jgdd Zdd ZdS )TestYeojohnsonc                 C   s  t jd}tdd|dd }t |dksJ tj|dd}t|| tj|dd}t|dd|d    tj|dd}t|t |d  tj|dd}t|| tdd|dd }t |dk seJ tj|d	d}t|t | d   tj|dd}t|| tj|d
d}t|d| d  d  tdd|dd	 }t |dk rJ t |dkrJ |dk}tj|dd}t|| ||  tj|dd}t|| dd|| d    tj|dd}t|| t || d  tj|dd}t|| ||  | }tj|d	d}t|| t ||  d   tj|dd}t|| ||  tj|d
d}t|| d||  d  d  d S )Nr  rx   r   r  r   r   r  r   r   ru   )	r)   r  r   r1   r   r   
yeojohnsonr   r*   )rM   r  r[   r  posnegr/   r/   r0   r    sL   


 $zTestYeojohnson.test_fixed_lmbdar  )r   r:  r  r   c                 C   s   dd }d}t jd t jjdd|d}|||}t|\}}t||dd	 tdt j	|| | d
d td|
 dd td| dd d S )Nc                 S   s   t j| j| jd}| dk}t|t dk r"t | | d ||< nt | | | d d| d ||< t|d t dkrZdt d|  | |   d dd|   || < |S dt | |    || < |S )Nr   r   r   r   r   )r)   r:  r  r   absspacingexppower)r[   r  r  r  r/   r/   r0   _inverse_transformQ  s   $
z:TestYeojohnson.test_lmbda_None.<locals>._inverse_transformi N  r  r   r   )r   r   r   r]  r!  r   rF   )r)   r  r1  normalr   r  r   r   linalgr   rO   rQ   )rM   r  r  	n_samplesr[   r  r  r   r/   r/   r0   r  L  s   
zTestYeojohnson.test_lmbda_Nonec                 C   r  r  )r   r   r  r  rV   r/   r/   r0   r  q  r2   zTestYeojohnson.test_emptyc                 C   sD   t jjdddd}t |\}}t t|\}}t||dd d S )Nr   r   r  r  r  r?   )r   r   r,   r  r  r   )rM   r[   xt1r  xt2r/   r/   r0   r  t  s   zTestYeojohnson.test_array_liker   c                 C   sP   t jd|d}d}tjt|d t| W d    d S 1 s!w   Y  d S )Nr6   r   z>Yeo-Johnson transformation is not defined for complex numbers.r  )r)   rZ   r  r   rU   r   r  )rM   r   r[   err_msgr/   r/   r0   test_input_dtype_complexz  s
   "z'TestYeojohnson.test_input_dtype_complexc                 C   sZ   t jd|d}t jdt jd}t|\}}t|\}}t||dd t||dd d S )Nr:   r   r  r?   )r)   rZ   float64r   r  r   )rM   r   x_intx_floatxt_int	lmbda_intxt_floatlmbda_floatr/   r/   r0   test_input_dtype_integer  s   z'TestYeojohnson.test_input_dtype_integerc                 C   sN   t g d}t|\}}t|d \}}t||dd t||dd d S )N)(HAG."A    5fAg(@Ag   VAgHzuoAg)\¿+Ar  g(\^@Ag    RAr  g    #Ag\µ=Ag
ףNAr  gp=oHAg    6bAg{csAg)\tmhAg/BAg   '|Ag(\wipAg    ݃hAgQlBAgףp @Ar   r  r?   )r)   r   r   r  r   r   )rM   r[   xt_yeolam_yeoxt_boxlam_boxr/   r/   r0   test_input_high_variance  s
   z'TestYeojohnson.test_input_high_varianceN)r`   ra   rb   r  r  r  r  r  r  r  r)   	complex64
complex128r  int8uint8int16int32r  r  r/   r/   r/   r0   r    s    /
$

r  c                   @   r3   )TestYeojohnsonNormmaxc                 C   rm  r  rp  rV   r/   r/   r0   rq    r2   z"TestYeojohnsonNormmax.setup_methodc                 C   r  )Ng?r  r?   )r   yeojohnson_normmaxr[   r   r  r/   r/   r0   r    r  zTestYeojohnsonNormmax.test_mlec                 C   s*   g d}t |}tj|dddsJ d S )N)gffffff@g r   r'  gffffff?g333333@r.  gffffff@g?g@r,  r(  g"@r  g      gzG?r  r!  )r   r  r)   allclose)rM   r[   r  r/   r/   r0   test_darwin_example  s   
z)TestYeojohnsonNormmax.test_darwin_exampleN)r`   ra   rb   rq  r  r  r/   r/   r/   r0   r    s    r  c                   @   sB  e Zd Zejdejdfejdfej	dfgdd Z
dd Zejd	ejejfejejfej	ejfgd
d Zdd Zdd Zdd Zejdejdfejdfej	dfgdd Zejdejejej	gdd Zejdejejej	gdd Zejdejejdddfejejdddfej	ejdddfgd d! Zejdejd"eg d#ed$d%gdfejd&eg d'ejd( d)  edd*gdfej	d+eg d,ed-d.gdfgd/d0 Zejdejdfejdfej	dfgd1d2 Zejdejejej	gd3d4 Zejdejejej	gd5d6 Zejd7d8d9d)d:d;d<ejged8d9d)d:d;d<ejgd=d>d?d@ejdAdBgggejdejejej	gdCdD Zejd7d8d9d)d:d;d<ejged8d9d)d:d;d<ejgd=d>d?d@ejdAdBgggejdejejej	gdEdF Z dGdH Z!dIdJ Z"dKdL Z#dMS )NTestCircFuncsztest_func,expectedgA:wv?g @pz?gE|2@c                 C   s(   t g d}t||dd|dd d S Nc  rx   r   r  r   ^  h  r  r  r?   r)   r   r   rM   	test_funcr  r[   r/   r/   r0   test_circfuncs  s   zTestCircFuncs.test_circfuncsc                 C   s   t g d}| }tj|dd}t||dd |t j d  }|d }tj|dd}t||dd |	 }tj
|dd}t||dd d S )	N)rO  ra  rF  rB     g     4@g3333333@r%  r&  r   r?   r(  r'  r  )r)   r   rO   r   circmeanr   pirP   circvarrQ   circstd)rM   r[   M1M2V1V2S1S2r/   r/   r0   test_circfuncs_small  s   z"TestCircFuncs.test_circfuncs_smallztest_func, numpy_funcc                 C   s<   t dgd dgd  }||}||}t||dd d S )Ng˛ɦv9?r   guv9?r   g:0yU>r!  r'  )rM   r)  
numpy_funcr[   circstatr  r/   r/   r0   test_circfuncs_close  s   z"TestCircFuncs.test_circfuncs_closec                       t g dg dg dg tj dd}tj  dd}t||dd tj ddd	} fd
dt jd D }t||dd tj ddd	} fddt jd D }t||dd d S )Nr"  _  r9   rw   `  r7   ]  ie  r7   r:   if  rw   id  r%  r&  r>   r?   r   r  ro  c                       g | ]}t j | d dqS r%  r&  r   r,  rq   r^   r[   r/   r0   rr         z4TestCircFuncs.test_circmean_axis.<locals>.<listcomp>r   c                    &   g | ]}t j d d |f ddqS Nr%  r&  rC  rD  rE  r/   r0   rr        & )r)   r   r   r,  ravelr   r  r  )rM   r0  r1  r/   rE  r0   test_circmean_axis  s   
z TestCircFuncs.test_circmean_axisc                    r:  )Nr"  r;  r?  r%  r&  dy=r?   r   r@  c                    rA  rB  r   r.  rD  rE  r/   r0   rr     rF  z3TestCircFuncs.test_circvar_axis.<locals>.<listcomp>r   c                    rG  rH  rM  rD  rE  r/   r0   rr     rI  )r)   r   r   r.  rJ  r   r  r  )rM   r2  r3  r/   rE  r0   test_circvar_axis     
zTestCircFuncs.test_circvar_axisc                    r:  )Nr"  r;  r?  r%  r&  rL  r?   r   r@  c                    rA  rB  r   r/  rD  rE  r/   r0   rr   	  rF  z3TestCircFuncs.test_circstd_axis.<locals>.<listcomp>r   c                    rG  rH  rP  rD  rE  r/   r0   rr   	  rI  )r)   r   r   r/  rJ  r   r  r  )rM   r4  r5  r/   rE  r0   test_circstd_axis  rO  zTestCircFuncs.test_circstd_axisc                 C   s"   g d}t ||dd|dd d S r!  )r   r(  r/   r/   r0   test_circfuncs_array_like	  s   z'TestCircFuncs.test_circfuncs_array_liker)  c                 C   s   t t|g  d S r(   )r   r)   r  )rM   r)  r/   r/   r0   r  	  r  zTestCircFuncs.test_emptyc                 C   s.   ddddddt jg}tt ||dd d S )	Nr#  rx   r   r  r   r$  r%  r&  r)   r   r   r  rM   r)  r[   r/   r/   r0   test_nan_propagate	  s   z TestCircFuncs.test_nan_propagateg5:v@gfPb?rU  gee?gOQOv?g#q@gauy+@c                 C   s   t g dddddddt jgdt jt jt jt jt jt jgg}| D ]-}||d	|d
}|d u r8tt | q#t|d || dd tt |dd    q#d S )N)r#  rx   r   r  r   r$  r   r<  r9   rw   r=  r7   r>  r   r%  r@  r   r  r?   )r)   r   r   keysr   r  r   r   rM   r)  r  r[   ro  outr/   r/   r0   test_nan_propagate_array	  s   

z&TestCircFuncs.test_nan_propagate_arraygrQvv@)g     v@r+  r(  g     8v@r  g     u@g5@v?g2>Bjv@gG2?)gaF+?z?rZ  gA ~@碲?r[  r(  r   gX҄?gJPm$@)gj  @竃 ?r\  g)h;@{ ?r]  gun2@g+fa @c                 C   s   t ddddddt jgddd	d
ddt jgt jt jt jt jt jt jt jgg}| D ].}||dd|d}|d u rBt||| dd q*t|d d || dd tt |d  q*d S )Nr#  rx   r   r  r   r$  r<  r9   rw   r=  r7   r>  r%  omit)r  
nan_policyro  r  r?   r   )r)   r   r   rV  r   r   r  rW  r/   r/   r0   test_nan_omit_array0	  s   z!TestCircFuncs.test_nan_omit_arrayc                 C   s0   ddddddt jg}t||ddd	|d
d d S )Nr#  rx   r   r  r   r$  r%  r^  r  r_  r  r?   )r)   r   r   r(  r/   r/   r0   test_nan_omitO	  s   
zTestCircFuncs.test_nan_omitc                 C   s2   t jt jt jt jt jg}tt ||dd d S )Nr^  r_  rS  rT  r/   r/   r0   test_nan_omit_allX	  s   zTestCircFuncs.test_nan_omit_allc                 C   sj   t t jt jt jt jt jgt jt jt jt jt jgg}||ddd}tt |  tt|dk d S )Nr^  r   )r_  ro  r   )r)   r   r   r   r  r   r   )rM   r)  r[   rX  r/   r/   r0   test_nan_omit_all_axis^	  s   z$TestCircFuncs.test_nan_omit_all_axisr[   r#  rx   r  r   r$  r<  r9   rw   r=  r7   r>  c                 C      t t||ddd d S )Nr%  raisera  rT   rU   rT  r/   r/   r0   test_nan_raiseg	     zTestCircFuncs.test_nan_raisec                 C   rf  )Nr%  foobarra  rh  rT  r/   r/   r0   test_bad_nan_policyp	  rj  z!TestCircFuncs.test_bad_nan_policyc                 C   s$   d}|}t |}t||dd d S )Nr   r   r?   )r   r,  r   )rM   r[   r0  r1  r/   r/   r0   test_circmean_scalary	  s   
z"TestCircFuncs.test_circmean_scalarc                 C   s@   t tdddtjtj }t|tjk  t|tj k d S )Nr   r   r:  )r   r,  r)   rZ   r-  r   )rM   mr/   r/   r0   test_circmean_range	  s   z!TestCircFuncs.test_circmean_rangec                 C   sZ   t jddgdd}ttj|ddd ttj|dddd	d
 ttj|dddd	d
 d S )N   r   r  r   r(  r&  g     @e@gLwqA?r  r?   g)#p_4@)r)   r   r	   r   r,  r   r.  r/  r   r/   r/   r0   test_circfuncs_uint8	  s   z"TestCircFuncs.test_circfuncs_uint8N)$r`   ra   rb   r  r  r  r   r,  r.  r/  r*  r6  r)   rO   rP   rQ   r9  rK  rN  rQ  rR  r  rU  r   rY  r   r-  r`  rb  rd  re  ri  rl  rm  ro  rq  r/   r/   r/   r0   r     s    








	



r   c                   @   s~   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd Zdd Zejdddgdd ZdS )TestMedianTestc                 C   s   t ttjg d d S rP  rT   rU   r   median_testrV   r/   r/   r0   test_bad_n_samples	  rS  z!TestMedianTest.test_bad_n_samplesc                 C   s   t ttjg g d d S rP  rs  rV   r/   r/   r0   r  	  r  z TestMedianTest.test_empty_samplec                 C   s&   t ttjg dg dddgdd d S )N)r   r   r   r   )r   r   r   r   r   r   tiesrs  rV   r/   r/   r0   test_empty_when_ties_ignored	  s   
z+TestMedianTest.test_empty_when_ties_ignoredc                 C   s8   t ttjg dg d t ttjg dg ddd d S )N)r   r   r   aboverv  rs  rV   r/   r/   r0   test_empty_contingency_row	  s   
z)TestMedianTest.test_empty_contingency_rowc                 C       t ttjg dddgdd d S )Nrt   rw   rx   r  rv  rs  rV   r/   r/   r0   test_bad_ties	     
zTestMedianTest.test_bad_tiesc                 C   r{  )Nrt   rw   rx   rk  rc  rs  rV   r/   r/   r0   rl  	  s    z"TestMedianTest.test_bad_nan_policyc                 C   r{  )Nrt   rw   rx   r  )r  )rT   r<  r   rt  rV   r/   r/   r0   r=  	  r}  zTestMedianTest.test_bad_keywordc                 C   s\   g d}g d}t ||\}}}}t|d t|ddgddgg t|d t|d d S )Nrt   r'  r   r   r   )r   rt  r	   r   )rM   r[   r  rC  r,  medtblr/   r/   r0   test_simple	  s   

zTestMedianTest.test_simplec                 C   s   g d}ddg}g d}t |||\}}}}t|d t|g dg dg t j|||dd\}}}}t|d t|g dg d	g t j|||d
d\}}}}t|d t|g dg d	g d S )Nr  rx   r6   )r9   r:   r7   )r   r   ru   )rw   r   r   r   rv  )rw   r   r   ry  )r   r   ru   )r   rt  r	   )rM   r[   r  zrC  r,  rn  r  r/   r/   r0   test_ties_options	  s   


z TestMedianTest.test_ties_optionsc                 C   s   ddt jg}g d}tj||dd}tj||dd\}}}}t|t jt jt jd f t|d t|d t|d	 t|t d
dgddgg tttj||dd d S )Nr   r   )rw   rx   r6   	propagaterc  r^  g     ?gإvao?r)  r   rg  )	r)   r   r   rt  r	   r   r   rT   rU   )rM   r[   r  mt1r   r,  rn  r  r/   r/   r0   test_nan_policy_options	  s   


z&TestMedianTest.test_nan_policy_optionsc                 C   s2  g d}g d}t ||\}}}}t|d t|ddgddgg t |\}}}	}
t|| t|| t j||dd\}}}}t|d t|ddgddgg t j|dd\}}}	}
t|| t|| t j||dd	\}}}}t|d t|ddgddgg t j|dd	\}}}	}
t|| t|| d S )
Nrv   )r   rw   r6   r:   rw   r   r   r   )lambda_Fr8  )r   rt  r	   chi2_contingencyr   )rM   r[   r  rC  r,  rn  r  exp_statexp_pdofer/   r/   r0   rR   	  s(   







zTestMedianTest.test_basicr8  FTc                 C   s>   g d}g d}t j|||d}t|j|j|j|jf| d S )Nrt   r  )r   rt  r	   rJ   r   r6  table)rM   r8  r[   r  r]   r/   r/   r0   r  
  s   zTestMedianTest.test_resultN)r`   ra   rb   ru  r  rx  rz  r|  rl  r=  r  r  r  rR   r  r  r  r  r/   r/   r/   r0   rr  	  s    rr  c                   @   sz   e Zd Zdd Zejdej d ejd gdfddej gdfgdd	 Z	d
d Z
dd Zdd Zdd Zdd ZdS )TestDirectionalStatsc                 C   s   t t g d }t t g d }t jt |t | t |t | t |fdd}t|}|j}t 	|d}t g d}t
|| d S )N)	g33333su@g      O@g33333sB@g      ;@g     pv@g@g333333I@gYv@r  )	gfffffP@g,Q@gfffffQ@gfffffT@g     S@g     @R@g33333SQ@gffffffM@g33333I@r   rn  rw   )gHP?g&S:g[<)r)   deg2radr   stackcossinr   directional_statsmean_directionr  r   )rM   declinclrN   dirstatsdirectional_meanmean_roundedreference_meanr/   r/   r0   "test_directional_stats_correctness
  s   
z7TestDirectionalStats.test_directional_stats_correctnesszangles, refr   r   r   r   c                 C   sL   t |r||}tjt|t|gdd}dt|j }t|| d S )Nr   rn  )	callabler)   r  r  r  r   r  mean_resultant_lengthr   )rM   anglesr   rN   r]   r/   r/   r0   'test_directional_stats_2d_special_cases&
  s
   z<TestDirectionalStats.test_directional_stats_2d_special_casesc           
      C   s   t jd}dt j |d }t jt |t |fdd}t|}|j	}t 
|d |d }|dt j  }t|}t|| d|j }t|}	t||	 d S )Nl	   }7Qh0P4u&4M r   )r  r   rn  r   )r)   r  r  r-  r  r  r  r   r  r  arctan2r,  r   r  r.  )
rM   r  testdatatestdata_vectorr  r  directional_mean_angler,  directional_varcircular_varr/   r/   r0   test_directional_stats_2d1
  s$   





z.TestDirectionalStats.test_directional_stats_2dc                 C   sf   t g dg dg}t |d}t g dg dgg dg dgg}tj|dd}t||j d S )N+Pz?r  r   )r        r   )r   r   r   r   )r   r   r   r   rn  )r)   r   tiler   r  r   r  )rM   rN   
full_arrayr  r  r/   r/   r0    test_directional_mean_higher_dimE
  s   

z5TestDirectionalStats.test_directional_mean_higher_dimc                 C   sN   g dg dg}t |}t|}t|}t|j|j t|j|j d S )Nr  )r  r  r   )r)   r  r   r  r   r  r  )rM   rN   
data_arrayr]   r   r/   r/   r0   )test_directional_stats_list_ndarray_inputR
  s   


z>TestDirectionalStats.test_directional_stats_list_ndarray_inputc                 C   sR   t d}d}tjtt|d t| W d    d S 1 s"w   Y  d S )N)rx   zIsamples must at least be two-dimensional. Instead samples has shape: (5,)r  )	r)   r   r  r   rU   reescaper   r  )rM   rN   r  r/   r/   r0   test_directional_stats_1d_error\
  s
   
"z4TestDirectionalStats.test_directional_stats_1d_errorc                 C   sh   t g dg dg}tj|dd}|t jj|ddd }tj|dd}t|j|j t|j|j d S )Nr  )g+Pz?g      r   T)	normalizer   )ro  keepdimsF)	r)   r   r   r  r   r   r   r  r  )rM   rN   r]   normalized_datar   r/   r/   r0    test_directional_stats_normalized
  s   
z5TestDirectionalStats.test_directional_stats_normalizeN)r`   ra   rb   r  r  r  r  r)   r-  r  r  r  r  r  r  r/   r/   r/   r0   r  
  s    

r  )Rry   r]  numpyr)   numpy.randomr   numpy.testingr   r   r   r   r   r   r	   r
   r  r   rT   r  scipyr   r   scipy.stats._morestatsr   common_testsr   
_hypotestsr   r   scipy.stats._binomtestr   scipy.stats._distr_paramsr   dict
matplotlibrcParamsmatplotlib.pyplotpyplotr   r  	Exceptionr  r  r  r  r  r  r  r  r	  r
  r1   r4   rd   r   r   r  r  r  r  rB  re  r  r  r  r  r  rY  ri  rl  r  r  r  r  r  r  r  r  r  r   rr  r  r/   r/   r/   r0   <module>   s   (
!kd I E= WD .^ U6%(2uH# e 