ó
áp7]c           @   sQ   d  d l  Z d  d l Z d  d l m Z m Z m Z d  d l m	 Z	 d   Z
 d S(   i˙˙˙˙N(   t   SlicedInverseRegt   SAVEt   PHD(   t   assert_equalc          C   s  t  j j d  t  j j d d  }  d |  d  d   d f t  j d d  |  d  d   d f |  d  d   d f <d |  d  d   d f t  j d d  |  d  d   d	 f |  d  d   d	 f <t  j d } t  j |  |  } t  j |  } t  j j |  } xt	 d  D]w} | d k r@t
 | |   } | j   } ny| d k rpt | |   } | j d d  } nI| d k rŚt | |  d t } | j d d  } n| d	 k r_t j i | d 6|  d  d   d f d 6|  d  d   d f d 6|  d  d   d f d 6|  d  d   d	 f d 6|  d  d   d f d 6 } t
 j d d | } | j   } nZ | d k rt | |   } | j   } n0 | d k ršt | |   } | j d t  } n  t t  j | j d | j d  d k t  t  j | j  }	 t  j |	 d  d   d f |  }
 |
 t  j t  j |	 d  d   d f d   :}
 |
 t  j t  j | d   :}
 t t  j |
  d k t  qWd  S(   Nię¨  t   sizeiô  i   g      ŕ?i    i   i   i   i˙˙˙˙i   t   slice_nid   t   bct   yt   x0t   x1t   x2t   x3i   t   x4s   y ~ 0 + x0 + x1 + x2 + x3 + x4t   datat   residgffffffî?(   iô  i   g      Đ?g      Đ?(   i    i   i˙˙˙˙i    g      ŕ?(   t   npt   randomt   seedt   normalt   sqrtt   r_t   dott   expt   poissont   rangeR    t   fitR   t   Truet   pdt	   DataFramet   from_formulaR   R   t   abst   eigst   asarrayt   paramst   sum(   t   xmatt   bt   lprt   evR   t   methodt   modelt   rsltt   dfR!   t   q(    (    sG   lib/python2.7/site-packages/statsmodels/regression/tests/test_dimred.pyt   test_poisson   sP    OO."0 (   t   numpyR   t   pandasR   t   statsmodels.regression.dimredR    R   R   t   numpy.testingR   R,   (    (    (    sG   lib/python2.7/site-packages/statsmodels/regression/tests/test_dimred.pyt   <module>   s   