
    M/PhJ                     T    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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                      ddl m}  |                                 D ]^}t          |j        |j        |j        dz   |j                  }t          ||j	        dt          t          |                               _d S )N   )pwr_chisquare)alpha   )decimalerr_msg)results.results_powerr   valuesr   wNdf	sig_levelr   powerreprvars)r   caser   s      `/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/statsmodels/stats/tests/test_gof.pytest_chisquare_powerr      s    444444$$&& 6 6!&*n6 6 6E4:q$(d$4$4	6 	6 	6 	6 	66 6    c                  H   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 ]F\  }}t          ||          \  }}t          ||j        d           t          ||j        d           Gd S )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?      s   88D88D 'DNDN#DL<DKDNH 
 
 
  DM H 
 
 
  DM X 
 
 
  DN )DNDN$DL<DKDNH 
 
 
  DM H 
 
 
  DM X 
 
 
  DN
 822233D
(111
2
2C
(111
2
2CSzD$<0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           d S )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11
2
2C
(111
2
2C D
sC
(
(C
sCu
5
5
5CT2....T1Wb111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 9 9 9 9 9 9 9 9 , , , , , ,6 6 63; 3; 3;l0 0 0 0 0r   