U
    Ãmœdâ  ã                   @   sZ   d Z ddlZddlmZmZ ddlmZ ddlm	Z	 ddl
m  mZ dd„ Zdd	„ ZdS )
zþTests for sandwich robust covariance estimation

see also in regression for cov_hac compared to Gretl and
sandbox.panel test_random_panel for comparing cov_cluster, cov_hac_panel and
cov_white

Created on Sat Dec 17 08:39:16 2011

Author: Josef Perktold
é    N)Úassert_almost_equalÚassert_allclose)ÚOLS)Úadd_constantc                  C   s$  dd l } | j | j t¡¡}| j |d¡}t |¡}|d d …df }|d d …df  t	¡}|d d …df  t	¡}t
|d d …df ƒ}t||ƒ ¡ }tj|||d�\}	}
}ddg}dd	g}d
dg}ddg}t |
¡}t |¡}t |	¡}t||jdd� t||dd� t||dd� t||dd� d S )Nr   ztest_data.txtéÿÿÿÿé   é   )Zgroup2g†ZÓ¼ã�?gôýÔxé&±?gcîZB>è©?ggÕçj+ö—?gÒo_Î¡?gµ7øÂdª°?gS–!Žuq«?é   )Údecimal)ÚosÚpathÚabspathÚdirnameÚ__file__ÚjoinÚnpZ
genfromtxtZastypeÚintr   r   ÚfitÚswZcov_cluster_2groupsÚse_covr   ZHC0_se)r   Zcur_dirZfpathZpetZendogÚgroupÚtimeZexogÚresZcov01ZcovgZcovtZbse_petwZbse_pet0Zbse_pet1Z	bse_pet01Zbse_0Zbse_1Zbse_01© r   ú^/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/statsmodels/stats/tests/test_sandwich.pyÚtest_cov_cluster_2groups   s*    



r   c                  C   sP  ddl m}  |  ¡ j}dt t |d j¡¡ }dt t |d j¡¡ }ttj	||d d d… jf ƒ}t
||ƒ ¡ }dd	d
gdddgdddgg}dddgdddgdddgg}tj|ddd�}t |¡}	tj|ddd�}
t |
¡}t||ƒ t|
|ƒ tt t |¡¡|	ƒ tt t |¡¡|ƒ tj|dd�}tj|ddd�}t||ƒ d S )Nr   )Ú	macrodatai�  ZrealgdpZrealinvZrealintr   gX»ù-Æ€ö?gÆ²¯Š¦ZÔ¿gpv>ôº	¯¿gÅ²¯Š¦ZÔ¿g<BëQ¼?g”N;è9?gvv>ôº	¯¿ghO;è9?g¬·³Ò'g¶?gRp”Ô7+ö?g•cÀCÔ¿g™Iâq¹“®¿gNª©„¬»?g£³Úx»…9?gœIâq¹“®¿gl³Úx»…9?gÃKàú¶?r	   T)ZnlagsÚuse_correctionF)r   )Zstatsmodels.datasetsr   Zload_pandasÚdatar   ÚdiffÚlogÚvaluesr   Zc_r   r   r   Zcov_hac_simpler   r   ÚsqrtZdiag)r   Zd2Zg_gdpZg_invZexoggZres_olsgZcov1_rZcov2_rZcov1Zse1Zcov2Zse2Zcov3Zcov4r   r   r   Útest_hac_simple:   s2    
 ýý



r#   )Ú__doc__Únumpyr   Znumpy.testingr   r   Z#statsmodels.regression.linear_modelr   Zstatsmodels.tools.toolsr   Z%statsmodels.stats.sandwich_covarianceÚstatsZsandwich_covariancer   r   r#   r   r   r   r   Ú<module>   s   
&