ó
áp7]c           @   sD   d  Z  d d l Z d d l j Z d d l m Z m Z d „  Z	 d S(   s   Example: minimal OLS

iÿÿÿÿN(   t   assert_almost_equalt   assert_allclosec          C   s-  t  j j d ƒ d }  t  j d d d ƒ } t j t  j | | d f ƒ d t ƒ} t  j d d d g ƒ } t  j	 | | ƒ t  j j
 d |  ƒ } t j | | ƒ j ƒ  } t  j d d g ƒ } | j | j d	 d d g d
 | j ƒ } t | | j | d  d  … d  f | f d d ƒ| j | j } | j t  j d ƒ d
 | j ƒ}	 t |	 j | d d ƒt |	 j | j d d ƒ| j t  j d ƒ d  d
 | j ƒ}
 | j d  } t  j d d g ƒ } | j | d  d  … d  f | f } t  j	 | t  j	 t  j j | ƒ | ƒ ƒ t | ƒ } t |
 j | d d ƒd  S(   Ni    id   i
   i   t   prependi   gš™™™™™¹?t   sizet   columnt   cov_pt   decimali   i   i   iÿÿÿÿt   rtoli   (   t   npt   randomt   seedt   linspacet   smt   add_constantt   column_stackt   Falset   arrayt   dott   normalt   OLSt   fitt   HC0_set
   cov_paramst   cov_HC0R    t   Nonet   paramst   t_testt   eyet   tvaluet   sdt   f_testt   linalgt   invt   lenR   t   fvalue(   t   nsamplet   xt   Xt   betat   yt   resultst   idxt   cov12t   tvalst   ttestt   ftestt   slopest
   cov_slopest   fval(    (    sD   lib/python2.7/site-packages/statsmodels/regression/tests/test_cov.pyt   test_HC_use
   s*    (%!0!%#4(
   t   __doc__t   numpyR   t   statsmodels.apit   apiR   t   numpy.testingR    R   R1   (    (    (    sD   lib/python2.7/site-packages/statsmodels/regression/tests/test_cov.pyt   <module>   s   