U
    Kvfb                     @   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	   Zcaser    r   D/tmp/pip-unpacked-wheel-2v6byqio/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   )r/   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$    

r4   )__doc__Znumpyr$   Znumpy.testingr   r   Zstatsmodels.stats.gofr   r   r   Zstatsmodels.tools.testingr   r   r.   r4   r   r   r   r   <module>   s   6