
    !gJ                     R    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 Zd Zd Zy)	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                     ddl m}  | j                         D ]g  }t        |j                  |j
                  |j                  dz   |j                        }t        ||j                  dt        t        |                   i y )N   )pwr_chisquare)alpha   )decimalerr_msg)results.results_powerr   valuesr   wNdf	sig_levelr   powerreprvars)r   caser   s      ]/var/www/dash_apps/app1/venv/lib/python3.12/site-packages/statsmodels/stats/tests/test_gof.pytest_chisquare_powerr      sb    4$$& 6!&*nn6E4::q$(d$4	66    c                  @   t               } t               }d| _        d| _        d| _        d| _        d| _        t        j                  g d      | _        t        j                  g d      | _	        t        j                  g d      | _
        d	|_        d|_        d
|_        d|_        d|_        t        j                  g d      |_        t        j                  g d      |_	        t        j                  g d      |_
        t        j                  g d      }t        j                  g d      }t        j                  g d      }t        ||g| |g      D ]D  \  }}t        ||      \  }}t        ||j                  d       t        ||j                  d       F y )Ng*{5 @   gQH?z(Chi-squared test for given probabilitiesfreq)i        ij  i  i  r!   r"   h  )gͮǈ
?gR.EH9oka`?gQO(?g?i  r    r!   r"   r$   )g|f𙯿r   r%   r&   r'      r      )r   	statistic	parameterp_valuemethod	data_namenparrayobservedexpected	residualszipr   r   )	res1res2r   pr1pr2prresstatpvals	            r   test_chisquarer?      sa   8D8D 'DNDN#DL<DKDNHH 
 DM HH 
 DM XX 
 DN )DNDN$DL<DKDNHH 
 DM HH 
 DM XX 
 DN
 8823D
((1
2C
((1
2CSzD$<0 ;CtR(
dD#--<D#++r:;r   c            	         t        j                  g d      } t        j                  g d      }d}t        | |      }t        | |d      }t        ||d       t        ||dz  d       t        | |dd	t	        |       d
z
  f      }d}t        ||       |g dz   }t        | |dd	t	        |       d
z
  f      }d}t        ||d       y )Nr#   r(   gV]g?F)cohen   r*      i  r
   )rA   
correctionr   )r   r   r   2   rE   gb?)r1   r2   r   r   lenr   )r9   r:   es_res1es2r7   res0pr3s           r   test_chisquare_effectsizerL   M   s    
((1
2C
((1
2C D
sC
(C
sCu
5CT2.T1Wb1  S,0#c(1*+=?DDt

CS,0#c(1*+=?D DdB/r   )__doc__numpyr1   numpy.testingr   r   statsmodels.stats.gofr   r   r   statsmodels.tools.testingr   r   r?   rL    r   r   <module>rS      s/     ;9 9 ,63;l0r   