U
    mdb                     @   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.EgH9gogka`?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