
p7]c           @   s   d  Z  d d l m 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(
   s>   

Created on Thu Feb 28 13:24:59 2013

Author: Josef Perktold
i(   t
   itervaluesN(   t   assert_almost_equalt   assert_equal(   t	   chisquaret   chisquare_powert   chisquare_effectsize(   t   Holderc       
   C   s{   d d l  m }  xd t |   D]V } t | j | j | j d d | j } t | | j	 d d d t
 t |   q Wd  S(   Ni   (   t   pwr_chisquaret   alphat   decimali   t   err_msg(   t   results.results_powerR   R    R   t   wt   Nt   dft	   sig_levelR   t   powert   reprt   vars(   R   t   caseR   (    (    s?   lib/python2.7/site-packages/statsmodels/stats/tests/test_gof.pyt   test_chisquare_power   s    c    	      C   s  t    }  t    } d |  _ d |  _ d |  _ d |  _ d |  _ t j d d d d	 d
 g  |  _ t j d d d d	 d g  |  _	 t j d d d d d g  |  _
 d | _ d | _ d | _ d | _ d | _ t j d d d d	 d
 g  | _ t j d d d d	 d g  | _	 t j d d d d d g  | _
 t j d d d d	 d
 g  } t j d d d d	 d g  } t j d d d d	 d g  } xj t | | g |  | g  D]M \ } } t | |  \ } } t | | j d d t | | j d d qWd  S(   Ng*{5 @i   gQH?s(   Chi-squared test for given probabilitiest   freqi  i  i  i  ij  i  i  ih  gͮǈ
?gR.EgH9gogka`?gQO(?g?i  g|f𙯿i    R	   i   i   (   R   t	   statistict	   parametert   p_valuet   methodt	   data_namet   npt   arrayt   observedt   expectedt	   residualst   zipR   R   (	   t   res1t   res2R   t   pr1t   pr2t   prt   rest   statt   pval(    (    s?   lib/python2.7/site-packages/statsmodels/stats/tests/test_gof.pyt   test_chisquare   sB    														(c       	   C   s/  t  j d d d d d g  }  t  j d d d d d g  } d } t |  |  } t |  | d	 t } t | | d
 d t | | d d
 d t |  | d	 t d d t |   d f } d } t | |  | d d d d d g } t |  | d	 t d d t |   d f } d } t | | d
 d d  S(   Ni  i  i  i  ih  i  i  gV]g?t   cohenR	   i   i   t
   correctioni  i   i    i2   gb?(   R   R   R   t   FalseR   t   lenR   (   R#   R$   t   es_rt   es1t   es2R!   t   res0t   pr3(    (    s?   lib/python2.7/site-packages/statsmodels/stats/tests/test_gof.pyt   test_chisquare_effectsizeP   s     (   t   __doc__t   statsmodels.compat.pythonR    t   numpyR   t   numpy.testingR   R   t   statsmodels.stats.gofR   R   R   t   statsmodels.tools.testingR   R   R)   R3   (    (    (    s?   lib/python2.7/site-packages/statsmodels/stats/tests/test_gof.pyt   <module>   s   		6