U
    Ãmœdb  ã                   @   sX   d Z ddlZddlmZmZ ddlmZmZm	Z	 ddl
mZ dd„ Zdd	„ Zd
d„ ZdS )z>

Created on Thu Feb 28 13:24:59 2013

Author: Josef Perktold
é    N)Úassert_almost_equalÚassert_equal)Ú	chisquareÚchisquare_powerÚchisquare_effectsize)ÚHolderc                  C   sT   ddl m}  |  ¡ D ]:}t|j|j|jd |jd�}t||j	dt
t|ƒƒd� qd S )Né   )Úpwr_chisquare)Úalphaé   )ÚdecimalÚerr_msg)Zresults.results_powerr	   Úvaluesr   ÚwÚNZdfZ	sig_levelr   ÚpowerÚreprÚvars)r	   Úcaser   © r   úY/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/statsmodels/stats/tests/test_gof.pyÚtest_chisquare_power   s    ÿ

ÿr   c            	      C   sX  t ƒ } t ƒ }d| _d| _d| _d| _d| _t dddd	d
g¡| _t dddd	dg¡| _	t dddddg¡| _
d|_d|_d|_d|_d|_t dddd	d
g¡|_t dddd	dg¡|_	t dddddg¡|_
t dddd	d
g¡}t dddd	dg¡}t dddd	dg¡}t||g| |gƒD ]8\}}t||ƒ\}}t||jdd� t||jdd� �qd S )Ng*ú{5¬ @é   gýQH«ç?z(Chi-squared test for given probabilitiesÚfreqi  é”  éþ  é¤  ij  éü  é²  éh  gÍ®Çˆ
ì?gRúÒ.öEò¿gÜH9÷«æ¼g•o¿¡¨ûè¼gka`Øüº?gQO(¸µŽŽ?gø¿öíÅÿï?é  gÒ|f�ð™¯¿r   é   ©r   é   )r   Z	statisticZ	parameterZp_valueÚmethodZ	data_nameÚnpÚarrayZobservedÚexpectedZ	residualsÚzipr   r   )	Úres1Zres2r   Úpr1Úpr2ÚprÚresÚstatZpvalr   r   r   Útest_chisquare   sz        ÿ    ÿ   þ    ÿ    ÿ   þr/   c                  C   sÒ   t  dddddg¡} t  dddddg¡}d}t| |ƒ}t| |d	d
�}t||dd� t||d dd� t| |d	dt| ƒd fd�}d}t||ƒ |dddddg }t| |d	dt| ƒd fd�}d}t||dd� d S )Nr   r   r   r   r   r    r   gV­]g¥›?F)Úcohené   r"   é   i¸  r   )r0   Z
correctionr   é2   g¶ªÇíb?)r%   r&   r   r   Úlenr   )r*   r+   Zes_rZes1Zes2r)   Zres0Zpr3r   r   r   Útest_chisquare_effectsizeN   s$    
ÿ
ÿr5   )Ú__doc__Únumpyr%   Znumpy.testingr   r   Zstatsmodels.stats.gofr   r   r   Zstatsmodels.tools.testingr   r   r/   r5   r   r   r   r   Ú<module>   s   6