
p7]c           @   s  d  Z  d d l Z d d l m Z d d l Z d d l m Z m	 Z	 m
 Z
 d d l m Z m Z m Z m Z m Z d d l m Z d e f d     YZ d	 e f d
     YZ d e f d     YZ d e f d     YZ d e f d     YZ d e f d     YZ d e f d     YZ d e f d     YZ d e f d     YZ d e f d     YZ d e f d     YZ d e f d     YZ d   Z  d    Z! e   Z" d! e" _# d" e" _$ e j% d# d$ g  e" _& e j% d% d& g  e" _' d' e" _( d( e" _) d) e" _* d* e" _+ e   Z, d! e, _# d+ e, _$ e j% e j- d, g  e, _& e j% d% d& g  e, _' d' e, _( d- e, _) d) e, _* d* e, _+ e   Z. d! e. _# d. e. _$ e j% d/ e j- g  e. _& e j% d% d& g  e. _' d' e. _( d0 e. _) d) e. _* d* e. _+ e   Z/ d1 e/ _# d2 e/ _$ e j% d# d$ g  e/ _& e j% d% d& g  e/ _' d3 e/ _( d( e/ _) d) e/ _* d* e/ _+ e   Z0 d1 e0 _# d4 e0 _$ e j% d/ e j- g  e0 _& e j% d% d& g  e0 _' d3 e0 _( d0 e0 _) d) e0 _* d* e0 _+ e   Z1 d5 e1 _# d6 e1 _$ e j% e j- d, g  e1 _& e j% d% d& g  e1 _' d7 e1 _( d- e1 _) d) e1 _* d* e1 _+ e   Z2 d8 e2 _# d9 e2 _$ e j% d: d; g  e2 _& d% e2 _' d< e2 _( d( e2 _) d= e2 _* d> e2 _+ e   Z3 d? e3 _# d@ e3 _$ e j% e j- dA g  e3 _& d% e3 _' dB e3 _( d- e3 _) d= e3 _* d> e3 _+ e   Z4 d8 e4 _# dC e4 _$ e j% dD e j- g  e4 _& d% e4 _' d< e4 _( d0 e4 _) d= e4 _* d> e4 _+ i dE d- 6dF d0 6dG d( 6Z5 dH e f dI     YZ6 d S(J   s  tests for weightstats, compares with replication

no failures but needs cleanup
update 2012-09-09:
   added test after fixing bug in covariance
   TODOs:
     - I don't remember what all the commented out code is doing
     - should be refactored to use generator or inherited tests
     - still gaps in test coverage
       - value/diff in ttest_ind is tested in test_tost.py
     - what about pandas data structures?

Author: Josef Perktold
License: BSD (3-clause)

iN(   t   stats(   t   assert_t   assert_almost_equalt   assert_allclose(   t   DescrStatsWt   CompareMeanst	   ttest_indt   ztestt   zconfint(   t   Holdert   CheckExternalMixinc           B   sb   e  Z e d  d   Z e d d   Z d   Z d   Z d   Z d   Z d   Z	 d	   Z
 RS(
   i    c         C   s   t  |  j |  j |  |  _ d  S(   N(   R   t   datat   weightst   descriptive(   t   clst   ddof(    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyt   get_descriptives   s    s   data.csvc         C   s   t  j d t j t |  j    } |  j | d <|  j j d k rS |  j | d <nF xC t |  j j	 d  D]+ } |  j d  d   | f | d | d <qj W| j
 |  d  S(   Nt   indexR   i   t   data1s   data%d(   t   pdt	   DataFramet   npt   aranget   lenR   R   t   ndimt   ranget   shapet   to_csv(   R   t   fnamet   dft   k(    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyt	   save_data#   s    $)c         C   s&   |  j  j } t | |  j d d d  S(   Nt   rtolg-C6?(   R   t   meanR   (   t   selft   mn(    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyt	   test_mean/   s    c         C   s&   |  j  j } t | |  j d d d  S(   NR    g-C6?(   R   t   sumR   (   R"   t   sm(    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyt   test_sum3   s    c         C   s&   |  j  j } t | |  j d d d  S(   NR    g-C6?(   R   t   varR   (   R"   R(   (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyt   test_var7   s    c         C   s&   |  j  j } t | |  j d d d  S(   NR    g-C6?(   R   t   stdR   (   R"   R*   (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyt   test_std<   s    c         C   s9   t  |  d  s d  S|  j j } t | |  j d d d  S(   Nt   semR    g-C6?(   t   hasattrR   t   std_meanR   R,   (   R"   R,   (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyt   test_semA   s    c         C   s|   t  j |  j d t  j } xZ t t f D]L } |  j j |  j d | } t  j | d t  j } t	 | | d d q( Wd  S(   Nt   dtypet   return_pandasR    g-C6?(
   R   t   asarrayt	   quantilest   float64t   Falset   TrueR   t   quantilet   quantile_probsR   (   R"   t   quantR1   t   qtl(    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyt   test_quantilesI   s    	(   t   __name__t
   __module__t   classmethodR   R   R$   R'   R)   R+   R/   R;   (    (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyR
      s   					t   TestSim1c           B   s<   e  Z d  Z d Z d Z d Z e j d
 Z e	 d	    Z
 RS(   g(?geڣ")@gGĔH?g|~?g}\*gI2gY?gg~5?g(rwy@c         C   sd   t  j j d  t  j j d d  |  _ t  j j d d d d |  _ t  j d
 |  _ |  j	   d  S(   Ni5 t   sizei   i    i   g?g      ?g      ?i   (   i    g?g      ?g      ?i   (
   R   t   randomt   seedt   normalR   t   uniformR   t   r_R8   R   (   R   (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyt   setup_class\   s
    (   g}\*gI2gY?gg~5?g(rwy@(   R<   R=   R!   R%   R(   R*   R   RE   R3   R>   RF   (    (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyR?   R   s   t	   TestSim1tc           B   s<   e  Z d  Z d Z d Z d Z e j d
 Z e	 d	    Z
 RS(   gOA4@gޭ,Yc@g.B@#@g~C	@i    i   i   i   i	   c         C   s   t  j j d  t  j j d d d d |  _ |  j d d !|  j d d +|  j d d !|  j d	 d +t  j j d d
 d d |  _ t  j d |  _ |  j	   d  S(   Ni5 i    i
   R@   i   i   i   i   i   i   g?g      ?g      ?i   (   i    g?g      ?g      ?i   (
   R   RA   RB   t   randintR   RD   R   RE   R8   R   (   R   (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyRF   o   s    (   i    i   i   i   i	   (   R<   R=   R!   R%   R(   R*   R   RE   R3   R>   RF   (    (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyRG   e   s   t	   TestSim1nc           B   sB   e  Z d  Z d Z d Z d Z d Z e j d Z	 e
 d
    Z RS(   g	ԿgXhg?g?gpA.?g&jjg	3mJgN],Ͽgŏ1w?g6v?c         C   s   t  j j d  t  j j d d  |  _ t  j j d d d d |  _ |  j d |  j j   9_ t  j d
 |  _	 |  j
 d	  d  S(   Ni  R@   i   i    i   g?g      ?g      ?i   (   i    g?g      ?g      ?i   (   R   RA   RB   RC   R   RD   R   R%   RE   R8   R   (   R   (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyRF      s    (   g&jjg	3mJgN],Ͽgŏ1w?g6v?(   R<   R=   R!   R%   R(   R*   R,   R   RE   R3   R>   RF   (    (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyRI   z   s   t   TestSim2c           B   sj   e  Z d  d g Z d d g Z d d g Z d d g Z e j e j d e j d f  Z	 e
 d    Z RS(   ggs˿g}#;οg݀Zggbc?gă?g<O?g!I.?g뭁lg';zgOeg!u?gCԷL@gtgeI)vgJ)Ŀg4k?gtv28J?c         C   sd   t  j j d  t  j j d d  |  _ t  j j d d d d |  _ t  j d |  _ |  j	   d  S(   Ni  R@   i   i   i    i   g?g      ?g      ?i   (   i   i   (   i    g?g      ?g      ?i   (
   R   RA   RB   RC   R   RD   R   RE   R8   R   (   R   (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyRF      s
    (   g뭁lg';zgOeg!u?gCԷL@(   gtgeI)vgJ)Ŀg4k?gtv28J?(   R<   R=   R!   R%   R(   R*   R   t   column_stackRE   R3   R>   RF   (    (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyRJ      s   
t   TestWeightstatsc           B   sM   e  Z e d     Z d   Z d   Z d   Z d   Z d   Z d   Z	 RS(   c         C   s   t  j j d  d \ } } d \ } } | t  j j |  } | t  j j |  } | t  j j | d  } | t  j j | d  } t  j j d d |  }	 t  j j d d |  }
 | | |  _ |  _ |	 |
 |  _ |  _ | | |  _	 |  _
 d  S(	   Ni5 i   i   g333333?i   i   (   i   i   (   i   g333333?(   R   RA   RB   t   randnRH   t   x1t   x2t   w1t   w2t   x1_2dt   x2_2d(   R   t   n1t   n2t   m1t   m2RN   RO   RR   RS   RP   RQ   (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyRF      s    c         C   s   |  j  |  j } } |  j |  j } } d t j t |   } d t j t |   } t |  } t t	 | | d | | f d  t
 j	 t j | | f t j | | f   d  S(   Ng       @R   i   (   RN   RO   RP   RQ   R   t   onesR   R   R   R   R    RE   (   R"   RN   RO   RP   RQ   t   w1_t   w2_t   d1(    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyt   test_weightstats_1   s    	c   
      C   si  |  j  |  j } } |  j |  j } } t |  } t | d | } t | d | } | j   } | j   }	 t t | | d | | f d  t j | |	  d  t |	 j	 d  | j	 d  t |	 j
   | j
 d  t |	 j   | j d  t t j |	 d d | j d  t | j d  d  t j | d  d  t | j d  d  t j | d  d  d  S(	   NR   i   i   i    t   biasi   i   i   (   RN   RO   RP   RQ   R   t	   asrepeatsR   R   R    R!   R(   R*   R   t   covt
   ttest_meant   ttest_1samp(
   R"   RN   RO   RP   RQ   R[   t   d1wt   d2wt   x1rt   x2r(    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyt   test_weightstats_2   s     "	)c         C   sy  |  j  |  j } } |  j |  j } } t | d | } t | d | } | j   } | j   } t | j d  | j d  t | j d  | j d  t | j	 d  | j	 d  t t
 j | j d d | j d  t t
 j | j  | j d  | j d  \ }	 }
 } t |	 |
 g t j | d  d  t | |  } | j   } t j | |  } t | d  | d  d  S(	   NR   i    i   R]   i   i   i   i   (   RR   RS   RP   RQ   R   R^   R   R!   R(   R*   R   R_   t   Tt   corrcoefR`   R    Ra   R   R   (   R"   RR   RS   RP   RQ   t   d1w_2dt   d2w_2dt   x1r_2dt   x2r_2dt   tt   pt   dt   cmt   ressmt   resss(    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyt   test_weightstats_3   s"    %"c   	      C   sk  |  j  } |  j } t | d | d d } t | d | d d } t | d | d d } | j   } | j   } | j   } t t j | t j | d  t t j | t j | d  | j d  } | j d  } | j d  } t t j | t j | d  t t j | t j | d  | j   } | j   } | j   } t | | d  t | | d  d  S(   NR   R   i    i   i   i   g      ?(   RR   RP   R   R`   R   R   RE   t   tconfint_mean(	   R"   RR   RP   t   d1w_d0t   d1w_d1t   d1w_d2t   res0t   res1t   res2(    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyt   test_weightstats_ddof_tests  s(    		c         C   s   |  j  |  j } } t |  } t |  } t | |  } d d l m } xV t t g D]H } x? d d g D]1 } | j d | d |  }	 t	 t
 |	 |   qj WqW Wt j | |  }
 t	 t | j    t |
 j    k  d  S(   Ni(   t   SimpleTablet   pooledt   unequalt   use_tt   usevar(   RR   RS   R   R   t   statsmodels.iolib.tableR|   R6   R5   t   summaryR   t
   isinstancet	   from_datat   str(   R"   RR   RS   R[   t   d2t   cm1R|   R   R   t   smryt   cm2(    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyt&   test_comparemeans_convenient_interface*  s    c         C   s   |  j  |  j } } t |  } t |  } t | |  } d d l m } xV t t g D]H } x? d d g D]1 } | j d | d |  }	 t	 t
 |	 |   qj WqW Wt j | |  }
 t	 t | j    t |
 j    k  d  S(   Ni(   R|   R}   R~   R   R   (   RN   RO   R   R   R   R|   R6   R5   R   R   R   R   R   (   R"   RR   RS   R[   R   R   R|   R   R   R   R   (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyt)   test_comparemeans_convenient_interface_1d;  s    (
   R<   R=   R>   RF   R\   Rf   Rs   R{   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyRL      s   		#	 		t   CheckWeightstats1dMixinc           B   s,   e  Z d    Z d   Z d   Z d   Z RS(   c         C   s
  |  j  } |  j } t | j d  | j d  t | j d d | j | j d  t | j d d | j | j d  | j d d  } t | j d d d | d  | j d d  } t | j d d d | d  t t	 j
 | j d d | j | j
 d  d  S(   Ni    i   R   i   R]   (   Rd   Rb   R   R!   R(   R   R*   t   var_ddoft   std_ddofR   R_   Rg   (   R"   Rd   Rb   t   var1t   std1(    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyt
   test_basicP  s    		%%c         C   s?   |  j  } |  j } t | j d  d  t j | d  d  d  S(   Ni   i   i   (   Rd   Rb   R   R`   R    Ra   (   R"   Rd   Rb   (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyt
   test_ttest`  s    		c         C   s  |  j  |  j } } |  j |  j } } |  j |  j } } t j | |  } t t | | d | | f d  | d  t	 t
 | d | d d t
 | d | d d  } t | j   d  | d  t	 t
 | d | d d t
 | d | d d  } t | j   d  | d  t	 t
 | d | d d t
 | d | d d  }	 t	 t
 | d | d d t
 | d | d d  }
 t	 t
 | d | d d t
 | d | d d  } |	 j d d  } |
 j d d  } | j d d  } t | | d  t | | d  |	 j d d	  } |
 j d d	  } | j d d	  } t | | d  t | | d  |	 j d d  } |
 j d d  } | j d d  } t | | d  t | | d  d  S(
   NR   i   i   R   i    i   R   R~   R}   (   RN   RO   Rd   Re   RP   RQ   R    R   R   R   R   t   tconfint_diff(   R"   RN   RO   Rd   Re   RP   RQ   t   res_spRp   t   cm0R   R   Rx   Ry   Rz   (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyt   test_ttest_2samplej  sB    
c         C   s   |  j  } d } | j   \ } } | j |  \ } } } t | | t j | j  d  | j |  \ } } } t | | t j | j  d  | j t j | | f   \ } } } t | | t j | j  d  d  S(   Ng?i   (   Rb   Rt   R`   R   R   RX   R   t   vstack(   R"   Rb   t   alphat   lowt   uppRm   Rn   Ro   (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyt   test_confint_mean  s    	  '(   R<   R=   R   R   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyR   N  s   		
	-t   CheckWeightstats2dMixinc           B   s   e  Z d    Z RS(   c         C   s5   |  j  } |  j } t t j | j  | j d  d  S(   Ni   (   Rd   Rb   R   R   Rh   Rg   (   R"   Rd   Rb   (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyt	   test_corr  s    		(   R<   R=   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyR     s   t   TestWeightstats1d_ddofc           B   s   e  Z e d     Z RS(   c   	      C   s  t  j j d  d \ } } d	 \ } } | t  j j | d  } | t  j j | d  } t  j j d d |  } t  j j d d |  } | | |  _ |  _ | | |  _ |  _ t	 | d | d d |  _
 t	 | d | d d |  _ |  j
 j   |  _ |  j j   |  _ d  S(
   Ni5 i   i   g333333?i   R   R   (   i   i   (   i   g333333?(   R   RA   RB   RM   RH   RN   RO   RP   RQ   R   Rb   Rc   R^   Rd   Re   (	   R   RT   RU   RV   RW   RN   RO   RP   RQ   (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyRF     s    (   R<   R=   R>   RF   (    (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyR     s   t   TestWeightstats2dc           B   s   e  Z e d     Z RS(   c   	      C   s  t  j j d  d \ } } d	 \ } } | t  j j | d  } | t  j j | d  } t  j j d d |  } t  j j d d |  } | | |  _ |  _ | | |  _ |  _ t	 | d | |  _
 t	 | d | |  _ |  j
 j   |  _ |  j j   |  _ d  S(
   Ni5 i   i   g333333?i   i   R   (   i   i   (   i   g333333?(   R   RA   RB   RM   RH   RN   RO   RP   RQ   R   Rb   Rc   R^   Rd   Re   (	   R   RT   RU   RV   RW   RN   RO   RP   RQ   (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyRF     s    (   R<   R=   R>   RF   (    (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyR     s   t   TestWeightstats2d_ddofc           B   s   e  Z e d     Z RS(   c   	      C   s  t  j j d  d	 \ } } d
 \ } } | t  j j | d  } | t  j j | d  } t  j j d d |  } t  j j d d |  } | | |  _ |  _ | | |  _ |  _ t	 | d | d d |  _
 t	 | d | d d |  _ |  j
 j   |  _ |  j j   |  _ d  S(   Ni5 i   i   g333333?i   i   R   R   (   i   i   (   i   g333333?(   R   RA   RB   RM   RH   RN   RO   RP   RQ   R   Rb   Rc   R^   Rd   Re   (	   R   RT   RU   RV   RW   RN   RO   RP   RQ   (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyRF     s    (   R<   R=   R>   RF   (    (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyR     s   t   TestWeightstats2d_nobsc           B   s   e  Z e d     Z RS(   c   	      C   s  t  j j d  d \ } } d \ } } | t  j j | d  } | t  j j | d  } t  j j d d |  } t  j j d d |  } | | |  _ |  _ | | |  _ |  _ t	 | d | d	 d
 |  _
 t	 | d | d	 d |  _ |  j
 j   |  _ |  j j   |  _ d  S(   Ni5 i   i   i   g333333?i   i   R   R   i    (   i   i   (   i   g333333?(   R   RA   RB   RM   RH   RN   RO   RP   RQ   R   Rb   Rc   R^   Rd   Re   (	   R   RT   RU   RV   RW   RN   RO   RP   RQ   (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyRF     s    (   R<   R=   R>   RF   (    (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyR     s   c          C   s   d }  d } d } d } t  |  | d	 d
 \ } } } t | | g | | g d  d }  d } d } t  |  | d	 d
 \ } } } t | | g | | g d  d  S(   Ni   i   i   g?g333333@g@g9(?geR   R~   i   i   gJ?g*ʿ(   i   i   i   (   g?g333333@g@(   i   i   i   i   (   R   R   (   t   at   bt   prt   trRm   Rn   R   (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyt   test_ttest_ind_with_uneq_var  s    c    
      C   s  d d  l  j j }  d d g } d d g } |  j d d d d } t | |  } | j d  } t | | d	 d
 d d t | t j |  d d  } | j d  } t	 | | d d | j
 d d  } |  j d d d d  } t	 | d | d d d d d g } d d g } t | |  } t | j   | j    } |  j t j d d g  t j d d g   } t | d | d d	 d
 t | |  j   }	 t |	 d | d d	 d
 t	 |	 | d d d  S(   Nii    i   i   i   g      4@t   valueg      ?R    gQ?t   atolg~jth?g      5@i   t   decimali   g?g333333?i
   (   t   statsmodels.stats.proportionR    t
   proportiont   proportions_ztestR   t
   ztest_meanR   R   t   arrayR   t
   ztost_meant   proportions_ztostR   R^   t   proportions_chisquareR2   R   t	   ztest_ind(
   t   smpropRN   RP   Rz   R[   Ry   R   RO   RQ   t   res1a(    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyt   test_ztest_ztost  s.     g}y4S4@gWCf=gOǯ?gfa?@g)A@g@i    s	   two.sideds   Two-sample z-Tests   x and yg4?g/In@t   lessg   @f=g/3	?t   greatergK͎@gNs?i   gc?g*,g9O*?i   g[-@g띏F)>gj2@gO+@g@s   One-sample z-Testt   xgcg@,j?g q@g@gF)>g&@t   smallert   largers	   two-sidedt	   TestZTestc           B   s    e  Z e d     Z d   Z RS(   c         C   s   t  j d d d d d d d d d	 d
 d g  |  _ t  j d d d d d d d d g  |  _ t |  j  |  _ t |  j  |  _ t |  j |  j  |  _ d  S(   Ng333333@gffffff@g      @g@g333333@g      @g@gffffff@g@gffffff@g333333@g      @g@gffffff@g      @gffffff@g      @(	   R   R   RN   RO   R   R[   R   R   Rp   (   R   (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyRF     s    *c   
   	   C   s{  |  j  |  j } } |  j } xt t t t t t g D]} t	 | | d | j
 d t | j \ } } t | | j d d t | | j d d d d | j d | j
 d t | j  \ } } t | | j d d t | | j d d d d | j j   } t j | d  r%t j | d <n  t j | d  rHt j | d <n  t | | d d d t | j } t | | d d | j d t | j  } t | | d d t | | d | j
 d t | j } t | | | j
 d d q5 W|  j }	 xt t t g D]n} t	 | d | j
 d t | j \ } } t | | j d d t | | j d d d d |	 j d | j
 d t | j  \ } } t | | j d d t | | j d d d d | j j   } t j | d  rt j | d <n  t j | d  rt j | d <n  t | d d d t | j } t | | d d |	 j d t | j  } t | | d d qWd  S(	   NR   t   alternativeR    g|=R   gؗҜ<i    i   (   RN   RO   Rp   t   ztest_t   ztest_smallert   ztest_largert   ztest_mut   ztest_smaller_mut   ztest_larger_muR   t
   null_valuet   alternativesR   R   t	   statistict   p_valueR   t   conf_intt   copyR   t   isnant   infR   t   zconfint_diffR[   t   ztest_mu_1st   ztest_smaller_mu_1st   ztest_larger_mu_1sR   t   zconfint_mean(
   R"   RN   RO   Rp   t   tct   zstatt   pvalt   tc_conf_intt   ciR[   (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyt   test  sZ    		(   R<   R=   R>   RF   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyR     s   	(7   t   __doc__t   numpyR   t   scipyR    t   pandasR   t   numpy.testingR   R   R   t   statsmodels.stats.weightstatsR   R   R   R   R   t   statsmodels.tools.testingR	   t   objectR
   R?   RG   RI   RJ   RL   R   R   R   R   R   R   R   R   R   R   R   R   R   t   estimateR   R   t   methodt	   data_nameR   t   nanR   R   R   R   R   R   R   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/stats/tests/test_weightstats.pyt   <module>   s   (5V			)																																																																		

