U
    Ãmœd»¨  ã                   @   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 d dlm	Z	 d dl
m	  mZ d dl
mZmZmZmZmZmZmZmZmZmZmZmZmZ dddd	d
dddgZe j de¡dd„ ƒZdd„ Zddddddejfdfddddddejfdfddddddejfdfddddd d!ejfd"fddddd#d!ejfd"fgZ e j d$e ¡d%d&„ ƒZ!G d'd(„ d(ƒZ"ddd	gZ#e j de#¡d)d*„ ƒZ$d+d,gZ%e j de%¡d-d.„ ƒZ&ddd/d0gZ'e j de'¡d1d2„ ƒZ(d3d4„ Z)d5d6d7d8gZ*e j d$e*¡d9d:„ ƒZ+d;d<„ Z,d=d>„ Z-d?d@„ Z.dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVœZ/e j d$e0e/ 1¡ ƒ¡dWdX„ ƒZ2G dYdZ„ dZƒZ3d[d\„ Z4d]d^„ Z5d_d`„ Z6dadb„ Z7dcdd„ Z8dS )eé    N)Úarange)Úassert_allcloseÚassert_equal)Ústats)Úconfint_poissonÚtolerance_int_poissonÚconfint_quantile_poissonÚetest_poisson_2indepÚconfint_poisson_2indepÚnonequivalence_poisson_2indepÚpower_poisson_ratio_2indepÚ power_equivalence_poisson_2indepÚpower_poisson_diff_2indepÚpower_equivalence_neginb_2indepÚpower_negbin_ratio_2indepÚmethod_names_poisson_1sampÚmethod_names_poisson_2indepÚwaldÚscoreúexact-cZwaldccvzsqrt-azsqrt-vúmidp-cÚsqrtÚmethodc                 C   sÆ   d\}}t ||| d�}tj|||d | d�j}tj|||d | d�j}d}| dkrVd}t|d	|d
� t|d	|d
� tj|||d | dd�j}tj|||d | dd�j}t|d|d
� t|d|d
� d S )N©é   i�  ©r   r   ©Úvaluer   é   ç»½×Ùß|Û=)r   ç�íµ ÷Æ°>çš™™™™™©?©ÚrtolÚlarger)r   r   ÚalternativeÚsmallerçš™™™™™™?)r   ÚsmrÚtest_poissonÚpvaluer   )r   ÚcountÚnobsÚciÚpv1Úpv2r#   © r0   úc/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/statsmodels/stats/tests/test_rates_poisson.pyÚtest_rate_poisson_consistency"   s"    ÿÿr2   c                  C   s   d\} }d}d}t j| |ddd�}t| |dd�}t|j|dd	� t||dd	� d
}d}t j| |ddd�}t| |dd�}t|j|dd	� t||dd	� d}t| |dd�}t||dd	� d}t| |dd�}t||dd	� d}t| |dd�}t|d |d dd	� d S )Nr   g·VUŸÔ?)g~¥Æ ~•?gœ¾úßª¯?r!   r   r   r   çê-�™—q=r"   g!s­5ÞÐ?)gæKãé›E—?g�¡y]]®¯?r   )g‡Gs–?g7ÅPÞ®?Újeff)g¢ÍÛcŸ÷’?g–x´–ê¬?r   )g_õ4yë˜?gd¬kùô®?r   r   gñhãˆµøä>)r(   r)   r   r   r*   )r+   r,   r/   Zci2Úrtr-   r0   r0   r1   Útest_rate_poisson_r;   s,    r6   r   r   )é   é    r7   )r   é   )é   é#   r:   )r   é!   é–   éd   )éh   éÈ   él   )r   éÄ   )éj   éÊ   ém   )r   éÆ   )éi   rD   Úcasec              	   C   s`  d}d}| \}}}}}}}	t |||||ddd�}
t|
|ƒ t |||||ddd�}
t|
|	ƒ t |||||ddd�}
t|
|ƒ |dkr¸t |||||d	dd�}
tj ||| | ¡}t|
|ƒ t|||||ddd�}t|d
 |d
 ƒ t|||||ddd�}t|d
 |	d
 ƒ d}t|||||ddd�}t|d |d ƒ t|||||ddd�}t|d |d ƒ d S )Ngffffffî?g333333ï?r!   ú	two-sided)Úexposure_newr   Úalphar%   r$   r&   )r   gwJëÿï?r   r'   r   )r   r   r   ZpoissonÚintervalr   )rH   ZprobZprob_oneÚmethr+   ÚexposurerJ   Úr2ÚrsZrlÚtir-   ZciqZprob_lowr0   r0   r1   Útest_tol_intv   s¬         þ
     þ
     þ
     þ
     þ     þ     þ     þrR   c                   @   s@   e Zd Zej ded ¡dd„ ƒZej ded ¡dd„ ƒZdS )	ÚTestMethodsCompar1samprM   Útestc                 C   s0   d\}}t j|||ddd�}t|jddd� d S )N©é<   ç33333€@çš™™™™™¹?rI   ©r   r   r%   g      Ð?r"   )r(   r)   r   r*   )ÚselfrM   Úcount1Ún1Útstr0   r0   r1   Ú	test_test·   s
    ÿz TestMethodsCompar1samp.test_testÚconfintc                 C   s.   d\}}t |||dd�}t|ddgdd� d S )NrU   r!   )r   rK   gbX9´È¶?gÓMbX9Ä?rX   r"   )r   r   )rZ   rM   r[   r\   r-   r0   r0   r1   Útest_confint¿   s    z#TestMethodsCompar1samp.test_confintN)	Ú__name__Ú
__module__Ú__qualname__ÚpytestÚmarkÚparametrizer   r^   r`   r0   r0   r0   r1   rS   ´   s   
rS   c           	   
   C   sè   d\}}}}t ||||| dd�}tj|||||d | dd�j}tj|||||d | dd�j}d}| dkrld	}t|d
|d� t|d
|d� tj|||||d | ddd�j}tj|||||d | ddd�j}t|d|d� t|d|d� d S )N©é   g      D@é   g      >@Údiff©r   Úcomparer   ©r   r   rl   r   r   )r   r    r!   r"   r$   ©r   r   rl   r%   r&   r'   ©r
   r(   Útest_poisson_2indepr*   r   )	r   r[   r\   Úcount2Ún2r-   r.   r/   r#   r0   r0   r1   Ú"test_rate_poisson_diff_consistencyÍ   s:    ÿ ÿ ÿýýrs   úwald-logú	score-logc           
   
   C   sì   d}d\}}}}t ||||| |d�}tj|||||d | |d�j}tj|||||d | |d�j}d}	| dkrpd	}	t|d
|	d� t|d
|	d� tj|||||d | |dd�j}tj|||||d | |dd�j}t|d|	d� t|d|	d� d S )NÚratiorg   rk   r   rm   r   r   )r   ru   r    r!   r"   r$   rn   r&   r'   ro   )
r   rl   r[   r\   rq   rr   r-   r.   r/   r#   r0   r0   r1   Ú#test_rate_poisson_ratio_consistencyñ   s<    ÿ ÿ ÿýýrw   Úetestú
etest-waldc              	   C   sÒ   d\}}}}t j||||| dd�}t j||||| dd�}t|j|jdd� t j||||| ddd�}t j||||| ddd�}t|j|jdd� t j||||| dd	d�}t j||||| dd	d�}t|j|jdd� d S )
Nrg   rv   rk   rj   ç‚vIhÂ%<=r"   r$   )r   rl   r%   r&   )r(   rp   r   Útuple)r   r[   r\   rq   rr   Út1Út2r0   r0   r1   Ú(test_rate_poisson_diff_ratio_consistency  s@     ÿ ÿ þ þ þ þr~   c               	   C   sb  d\} }}}t j| |||dd�\}}d}t||d ddd� t|d	ddd
� t j| |||dd�\}}d}	t||	d ddd� t|dddd
� t j| |||dd�\}}d}	t||	d ddd� t|dddd
� t j| |||dd�\}}d}	t||	d ddd� t|dddd
� t j| |||dd�\}}d}	t||	d ddd� t|dddd
� d\} }}}t j| |||ddd�\}}d}t||d ddd� t|dddd
� t j| |||ddd�\}}d}	t||	d ddd� t|dddd
� t j| |||ddd�\}}d }	t||	d ddd� t|d!ddd
� t t¡�  t j| |||ddd"�\}}W 5 Q R X d#}	t||	d ddd� t|d$ddd
� t j| |||ddd�\}}d%}	t||	d dd&d� t|d'ddd
� d\} }}}t j| |||dd(d)�\}}d}t||ddd� t j| |||dd(d)�\}}d}	t||	ddd� t j| |||dd(d)�\}}d}	t||	ddd� t j| |||d*d+d(d,�\}}d-}	t||	ddd� t j| |||d.d+d(d,�\}}d/}	t||	ddd� t| |||dd(d)�\}
}d0}t||ddd� t| |||dd(d)�\}
}d0}t||ddd� d\} }}}t j| |||ddd(d,�\}}d}t||ddd� t j| |||ddd(d,�\}}d}	t||	ddd� t j| |||ddd(d,�\}}d%}	t||	ddd� t j| |||d*dd(d,�\}}d1}	t||	ddd� t j| |||d.dd(d,�\}}d2}	t||	ddd� t| |||ddd(d,�\}
}d3}t||ddd� t| |||ddd(d,�\}
}d3}t||ddd� d S )4N©rV   ç    °"é@rh   çffff–„ê@r   r   gXþ|[°T7?é   r   çñhãˆµøÔ>©r#   Úatolg&qVDM@)r…   r#   r   g:‘`ª™µ4?gªGÜÖV@rt   g¼yªCn†;?g-Cëâ¶
@ru   ç-Cëâ6*?g¥½Á&S@ç-Cëâ6
?r   gv‰ê­�­2?gUÞŽpZ�@©é)   éjm  r   éIJ  ç      ø?)r   r   g<½R–!ŽÍ?çü©ñÒMb@?g:ž%Èˆç?g¨WÊ2Ä±Î?grßj�¸œæ?gÉå?¤ß¾Î?gÜ×�sF”æ?)r   Ú
ratio_nullg
h"lxzÍ?gV-²�ç?gxœ¢#¹üÏ?ç{®Gázt?gdxìg±”å?r$   )r   r%   ú
exact-condr   rY   g()° ¦<?ú	cond-midpg©MœÜïP4?g†ÆAœ‡3?g†ÉTÁ¨¤Ò?g\�Âõ(\Ï?gõ¹ÚŠýeÏ?)r(   rp   r   rd   ÚwarnsÚFutureWarningr	   )r[   r\   rq   rr   Ús1r.   Zpv1rÚs2r/   Zpv2rÚ_Zpve1Zpve1rZpve2Zpve2rr0   r0   r1   Útest_twosample_poisson4  s(   ÿ
ÿ
ÿ
ÿ
 ÿ
 ÿÿ
ÿ
ÿ
ÿ
 þ
 þ

þ

þ
 ÿ
 ÿ
 ÿ
 þ
 þ

 þ

 þ
r—   )r   )çúíëÀ9£@gæ?¤ß¾Œ?)çû:pÎˆÒø?g?W[±¿ì®?)r   )ç�€&Â†§ @g¥½Á&“?)ç(~Œ¹k	ø?g»aÛ¢Ì±?)ry   )r˜   gHPüs×’?)r™   çpUjö°?)zetest-score)rš   gøÂdª`T’?)r›   rœ   c           
      C   s†   | \}}}d\}}}}d}t j||||||ddd d�	}	t|	j|	jf|dd� d}t j||||||ddd d�	}	t|	j|	jf|d	d� d S )
Nrˆ   r   rj   r$   )r   r   rl   r%   Z
etest_kwdsga2U0*©C?©r…   r†   gÇº¸�ðF?)r(   rp   r   Ú	statisticr*   )
rH   rM   Úres1Úres2r[   Ú	exposure1rq   Ú	exposure2r   Útr0   r0   r1   Útest_twosample_poisson_diffÛ  s(    
  ý  ýr¤   c            
   	   C   sf  ddl m} m} d\}}}}| }tj||||dd�}t|j|jdd� t|j|j	dd� t
|j|jƒ |}tj||||dd�}t|j|jd	d
d� t|j|j	dd� t
|j|jƒ d}tj||||dddd�}	t|	j|dd� d}tj||||dddd�}	t|	j|dd� d}tj||||dddd�}	t|	j|dd� d}tj||||dddd�}	t|	j|dd� d S )Nr   )Úres_pexact_condÚres_pexact_cond_midpr   r�   r   rz   r"   r‘   r   rƒ   r„   g�€Ž×CÖï?ç333333ó?r&   rY   r3   g¤·¿8Þt?r$   g½Ý "ä>{?gÜBbwãï?)Zresults.results_ratesr¥   r¦   r(   rp   r   r*   Zp_valuerv   Úestimater   rŽ   Z
null_value)
r¥   r¦   r[   r\   rq   rr   r    rŸ   r/   Úrestr0   r0   r1   Útest_twosample_poisson_rò  sH     ÿ ÿ ÿ ÿrª   c               
   C   sŽ  d\} }}}t | |||ddddd�}d}t||dd	� d
}t||dd� t | |||ddddd�}d}t||dd	� t | |||dddd�}d}t||dd	� t | |||dddd�}d}t||dd	� |d }|d }t | |||ddddd�}d}t||dd	� t | |||dddd�}d}t||dd	� t ||| |ddddd�}d}t||dd	� d}t||dd	� t ||| |ddddd�}d}t||dd	� d S )Nr   Zmoverrv   rX   r4   )r   rl   rK   Zmethod_mover)gò°Pkšw÷?g?ÆÜµ„|@r!   r�   )gøí¸áw÷?g%xCx@gü©ñÒMbP?r"   r   )g8øÂdª`÷?gŸÍªÏÕV@Zwaldcc©r   rl   rK   )g[±¿ìž<÷?gÞ	Š@r�   )g‘BYøúÚõ?g«@-
@rƒ   éè  rj   )gc.”áÓÐ?gú8VQï?r�   )gaÚ9ÍÑ?gçß.ûu§ï?)gÒÅïbí¿g÷p­8hÔ¿)g�•C‹lí¿g8gDioÔ¿)g´Yõ¹ÚŠí¿gH¿}8gÔ¿gú~j¼t“x?©r
   r   )r[   r¡   rq   r¢   r-   Zci1r0   r0   r1   Útest_confint_poisson_2indep$  sˆ    
  þ
  þ
  ÿ
  ÿ
  þ
  ÿ
  þ
  þr®   c            	   
   C   s€   d\} }}}d\}}t j| |||||dd�}t|jddd� dd	d
ddg}|D ],}t j| ||||||d�}t|jddd� qNd S )Nr   )gi¾d¹Žoõ?g¹—ìh_@r�   r   r'   r3   r"   r   r   r   r‘   g{®Gáz„?r�   )r(   Útost_poisson_2indepr   r*   )	r[   r\   rq   rr   ÚlowÚuppÚresÚmethodsrM   r0   r0   r1   Útest_tost_poissont  s    ÿÿr´   g6ŸãD²?gžZ4‚µ?gÞ-ûÑ„™´?g   Èÿ§À?g  p£öÞ·?gÅÚ>ƒ+.·?g…³—Ÿå¦±?g6ŸãD¢?gžZ4‚¥?gÞ-ûÑ„™¤?gÿÿ?‡ºðµ?gÿÿ_5cÿ¨?g[S@ý‰f¦?g =¤‰f¦?g@ŸÆ±Ûî?gVºÞ§î?g"Mà²g¶î?g  <ê*žï?g  ªÌ	pî?g@¹k¦ óî?guîû« óî?))rI   r   )rI   r   )rI   r   )rI   r�   )rI   r‘   )rI   rx   )rI   ry   )r$   r   )r$   r   )r$   r   )r$   r�   )r$   r‘   )r$   rx   )r$   ry   )r&   r   )r&   r   )r&   r   )r&   r�   )r&   r‘   )r&   rx   )r&   ry   c           	   	   C   sF   | \}}d\}}}}t j|||||d|d�\}}t|t|  dd� d S )N)é   g     €I@r   g      K@r§   rY   rz   r"   )r(   rp   r   Ú	cases_alt)	rH   ÚaltrM   r[   r\   rq   rr   r–   Úpvr0   r0   r1   Útest_alternative¢  s     ÿ
r¹   c                   @   sÆ   e Zd Zej ddd„ ed d D ƒdd„ ed d D ƒ ¡dd	„ ƒZej dd
d„ ed d D ƒdd„ ed d D ƒ ¡dd„ ƒZej ddd„ ed d D ƒdd„ ed d D ƒ ¡dd„ ƒZ	dS )ÚTestMethodsCompare2indepzcompare, methc                 C   s   g | ]}d |f‘qS ©rv   r0   ©Ú.0rM   r0   r0   r1   Ú
<listcomp>±  s     z#TestMethodsCompare2indep.<listcomp>rT   rv   c                 C   s   g | ]}d |f‘qS ©rj   r0   r¼   r0   r0   r1   r¾   ³  s     rj   c                 C   sÚ   d\}}}}t j||||||d dd�}t|jddd� |dkrRd}d	| | }	}
nd
}| | }	}
t|||||	|
||d�}d|ksŠd|kr�d}nd}t|j|j|d� t t¡�  t j||d|||d dd�}W 5 Q R X d S )N©rV   rW   é(   gV-²ø€@rI   ©r   rl   r   r%   gJ+‡™?gš™™™™™É?r"   rv   ç      ð?r   ç        rk   Úcondrx   rX   r3   r   )r(   rp   r   r*   r   rd   r’   ÚRuntimeWarning)rZ   rM   rl   r[   r\   rq   rr   r]   Úfr°   r±   ÚvÚtst2r#   r0   r0   r1   r^   ¯  s4     þþ þz"TestMethodsCompare2indep.test_testc                 C   s   g | ]}d |f‘qS r»   r0   r¼   r0   r0   r1   r¾   Ø  s     r_   c                 C   s   g | ]}d |f‘qS r¿   r0   r¼   r0   r0   r1   r¾   Ú  s     c           	   	   C   sN   d\}}}}|dkrddg}nddg}t ||||||dd�}t||d	d
� d S )NrÀ   rv   g¤p=
×£ð?g¸…ëQ¸@gÔ+eâXw?g#Ûù~j¼´?r!   r«   rX   r"   r­   )	rZ   rM   rl   r[   r\   rq   rr   Zci_valr-   r0   r0   r1   r`   Ö  s    
 ÿz%TestMethodsCompare2indep.test_confintc                 C   s   g | ]}d |f‘qS r»   r0   r¼   r0   r0   r1   r¾   ê  s     c                 C   s   g | ]}d |f‘qS r¿   r0   r¼   r0   r0   r1   r¾   ì  s     c              
   C   s:  d|krt  d¡ d\}}}}t ||g¡}t ||g¡}d}	tj|||	 gtd�}
t |||	 g¡}|||
|f\}}}}|dkr–d}d| | }}nd}| | }}t||||||||d	�}|jjd
ksÎt‚|j	jd
ksÞt‚d|k�sd|k�st
j||||||d dd�}t|j	|j	dd� |dk�r<d}d| | }}nd}| | }}t
j|d |d |d |d ||||d	�}t
j|d |d |d |d ||||d	�}t
j||||||||d	�}|jjd
k�sÌt‚|j	jd
k�sÞt‚t|jd |jdd� t|j	d |j	dd� t|jd |jdd� t|j	d |j	dd� d S )Nrx   z#nonequivalence etest not vectorizedrÀ   rÃ   )Zdtyperv   r   rÄ   rk   )r‚   rÅ   rI   rÂ   r3   r"   rŒ   ç      à?r   )rd   ÚskipÚnpÚarrayÚintr   rž   ÚshapeÚAssertionErrorr*   r(   rp   r   r¯   )rZ   rM   rl   r[   r\   rq   rr   Zcount1vZn1vZnfactZcount2vZn2vrÇ   r°   r±   rÈ   rÉ   r]   Ztst0Ztst1r0   r0   r1   Útest_test_vectorizedè  sl    
þ þ
 ý ýþz-TestMethodsCompare2indep.test_test_vectorizedN)
ra   rb   rc   rd   re   rf   r   r^   r`   rÑ   r0   r0   r0   r1   rº   ¬  sB   
ÿ
ÿþþ
 
ÿ
ÿþþ

ÿ
ÿþþrº   c                  C   sz   t dƒ} tdddd| d�\}}t|ddd	� td
ddd| d�\}}t|ddd	� tdddddgd�\}}t|ddd	� d S )Nr¬   rV   r€   rh   r�   ©Úy_gridg2D˜Y‰–B?gVçž¯Ò<r�   r‰   rŠ   r   r‹   g¿òÿv6]£?r   gÌ�¿£ªRÁ?)r   r	   r   )rÓ   r–   r¸   r0   r0   r1   Útest_y_grid_regression/  s    rÔ   c               	   C   s®   t  d¡ t jdd��} tdddddgd� W 5 Q R X t| ƒdksFt‚t| d jtƒsZt‚dt	| d j
ƒkspt‚t t¡�}tdddddd� W 5 Q R X d	t	|jƒksªt‚d S )
NÚalwaysT)Úrecordr   )Úygridr   r×   rÒ   rÓ   )ÚwarningsÚsimplefilterÚcatch_warningsr	   ÚlenrÐ   Ú
issubclassÚcategoryr“   ÚstrÚmessagerd   ZraisesÚ
ValueErrorr   )ÚwÚer0   r0   r1   Útest_invalid_y_grid<  s    
rã   c                  C   s2  d\} }d\}}d}d}d\}}d}ddd	d
g}	|	D ]j}
|
\} }}}t | |||||| |||d�	}t||dd� t | |||||| ||d|d�
}t||dd� q4ddddg}	d}|	D ]h}
|
\} }}}t| |||| ||dddd�	}t||dd� t| |||| ||dddd�	}t||dd� q´t| |||| || | dddd�	}t|ddd� ddddg}	d} d}|	D ]Æ}
|
\}}}}t| |||| ||dddd�	}t||dd� t| |||| ||ddddd�
}t||dd� t| |||| ||dddd�	}t||dd� t| |||| ||ddddd�
}t||dd� �qfd S )N)çš™™™™™@rä   )é_   rå   r'   ç      @)gš™™™™™é?g      ô?r   )çffffffþ?éÀ  rè   gv7OuÈÍì?)g       @éö   ré   goG8-xÑì?)rä   rå   rå   gr§t°þÏì?)ræ   éŒ  rê   gÆÜµ„|Ðì?)Ú
nobs_ratiorN   rK   Ú
dispersionr‡   r�   r   )rë   rN   rK   Ú
method_varrì   r�   r"   )gÍÌÌÌÌÌü?é   rî   gŒ¾‚4cÑì?)rç   é'   rï   gRÕQ÷í?)rä   és   rð   g=IºfòÍì?)g333333@é”  rñ   g§/ú
Òì?r§   r&   )rë   rN   r   rK   rì   r%   r!   rI   rä   g«ªªªªªê?r$   )rë   rN   r   rK   rí   rì   r%   )r   r   r   )Úrate1Úrate2Únobs1Únobs2rK   rN   r°   r±   rì   ZcasesrH   ÚpÚpow_Zpow_2r0   r0   r1   Útest_poisson_power_2ratioK  s   ü     ü    úü      ý      ý      ýü      ý      ü      ý      ürø   c                  C   sš   d\} }||  }d\}}t ||| |dddddd�	}t|jd	d
d� t dddddddddd�	}t|jdd
d� t dddddddddd�	}t|jdd
d� d S )N)rµ   é   )r   é
   r!   r   r·   r$   T)rë   rK   r   rí   r%   Úreturn_resultsg±3…Îkê?r‡   r�   g333333ã?éa   rŒ   ç333333Ó?r'   r&   r   )r   rK   r%   rí   rû   g5Ð|ÎÝ®é?é€   gUUUUUUå?gïU+~©é?)r   r   Úpower)rô   rõ   rë   rò   ró   r÷   r0   r0   r1   Útest_power_poisson_equal¿  sN           þ   ú   úr   c                  C   s|  d\} }d\}}d}d}d\}}d}d}	t | |||||| |||dd	�
}
t|
|	d
d� d\}}d}	t | |||||| |||dd	�
}
t|
|	d
d� d}	t | |||||| |||dd	�
}
t|
|	d
d� d\}}} }}d}	t|| ||| |d|ddddd�}
t|
|	dd� d}	|| }}t|| ||| |d|ddddd�}
t|
|	dd� d}	t|| ||| |d|ddddd�}
t|
|	dd� t|| ||| |d|ddddd�}t||	dd� t||
dd� d}	t|| ||| |d|ddddd�}
t|| ||| |d|ddddd�}t||	dd� t||
dd� t| |||| |d|dd ddd�}
t| |||| |d|dd ddd�}t||	dd� t||
dd� d S )!N)ræ   ræ   )éÅ  r  r!   gÍÌÌÌÌÌì?)g      ì?g’$I’$Iò?gffffffÖ?gX�f,šÎì?r·   )rë   rN   rK   rì   rí   r‡   r�   )éÆ  r  g Òo_Îì?ZftotalgûèÔ•Ïì?r   )rý   é2   rÊ   r>   r‚   gš/å2$Ýã?r   rÊ   rI   F)rë   rN   r   rK   rì   r%   rí   rû   g—}w¤â?g uàAX2ç?r   Trz   r"   gµÑvL]ê?r&   r$   )r   r   r   r   )rò   ró   rô   rõ   rK   rN   r°   r±   rì   Zpow1r÷   Zpow_pr0   r0   r1   Útest_power_negbiní  sP        û     û     û	   ø

   ø	    ù	    ù	    ù	    ù	    ù	    ù	r  )9rd   rØ   ÚnumpyrÌ   r   Znumpy.testingr   r   Zscipyr   Zstatsmodels.stats.ratesZratesr(   r   r   r   r	   r
   r   r   r   r   r   r   r   r   r³   re   rf   r2   r6   ÚinfZcases_tolintrR   rS   Zmethods_diffrs   Zmethods_ratiorw   Zmethods_diff_ratior~   r—   Zcases_diff_ngr¤   rª   r®   r´   r¶   ÚlistÚkeysr¹   rº   rÔ   rã   rø   r   r  r0   r0   r0   r1   Ú<module>   s”   <ÿ
3û	
=


 
  û	
2Pé
	 t.