U
    Kvf¡  ã                   @   s4   d Z ddlZddlmZmZ ddlmZ dd„ Z	dS )zExample: minimal OLS

é    N)Úassert_almost_equalÚassert_allclosec                  C   sŠ  t j d¡ d} t  ddd¡}tjt  ||d f¡dd}t  dddg¡}t  ||¡t jj	| d	 }t 
||¡ ¡ }t  ddg¡}|jddg|jd
}t||j|d d …d f |f dd |j|j }|jt  d¡|jd}	t|	j|dd t|	j|jdd |jt  d¡d d… |jd}
|jd d… }t  ddg¡}|j|d d …d f |f }t  |t  t j |¡|¡¡t|ƒ }t|
j|dd d S )Nr   éd   é
   é   F)Úprependé   gš™™™™™¹?)Úsize)ÚcolumnÚcov_pé   )Údecimalé   )r   é   éÿÿÿÿé   )Zrtol)ÚnpÚrandomÚseedZlinspaceÚsmZadd_constantZcolumn_stackÚarrayÚdotÚnormalZOLSZfitZ
cov_paramsZcov_HC0r   ÚparamsZHC0_seZt_testZeyeZtvalueÚsdZf_testZlinalgÚinvÚlenr   Zfvalue)ZnsampleÚxÚXÚbetaÚyÚresultsÚidxZcov12ZtvalsZttestZftestZslopesZ
cov_slopesZfval© r#   úI/tmp/pip-unpacked-wheel-2v6byqio/statsmodels/regression/tests/test_cov.pyÚtest_HC_use   s(    $$r%   )
Ú__doc__Znumpyr   Znumpy.testingr   r   Zstatsmodels.apiÚapir   r%   r#   r#   r#   r$   Ú<module>   s   