ó
áp7]c           @   sY   d  d l  Z  d  d l Z d  d l m Z d „  Z d „  Z d „  Z d „  Z d „  Z	 d S(   iÿÿÿÿN(   t   datac          C   s~   t  j j d ƒ }  t  j |  d <t j |  ƒ } t j | ƒ \ } } } t  j j | j	 ƒ  d d d d d d d	 d
 d g	 ƒ d S(   s   
    Fixes GH: #144
    i
   i   i   i   i    i   i   i   i   i   i	   N(   i
   i   (   i   i   (
   t   npt   randomt   nant   pandast	   DataFrameR    t   interpret_datat   testingt   assert_equalt   tolist(   t   Xt   dft   valst   cnamest   rnames(    (    s@   lib/python2.7/site-packages/statsmodels/tools/tests/test_data.pyt   test_missing_data_pandas   s
    c          C   sŽ   t  j j d ƒ j d d	 d
 g ƒ }  t j |  ƒ \ } } } t  j j | |  j j ƒ t  j j | |  j t	 d f ƒ ƒ t  j j | d  ƒ d  S(   Ni	   t   var1t   f8t   var2t   var3i   (   i	   (   R   R   (   R   R   (   R   R   (   R   R   t   viewR    R   R   R   t   dtypet   namest   floatt   None(   R
   R   R   R   (    (    s@   lib/python2.7/site-packages/statsmodels/tools/tests/test_data.pyt   test_structarray   s    "c          C   sš   t  j j d ƒ j d d	 d
 g ƒ }  t j |  j t  j ƒ ƒ \ } } } t  j j | |  j j	 ƒ t  j j | |  j t
 d f ƒ ƒ t  j j | d  ƒ d  S(   Ni	   R   R   R   R   i   (   i	   (   R   R   (   R   R   (   R   R   (   R   R   R   R    R   t   recarrayR   R   R   R   R   R   (   R
   R   R   R   (    (    s@   lib/python2.7/site-packages/statsmodels/tools/tests/test_data.pyt   test_recarray   s    $"c          C   s‘   t  j j d ƒ }  t j |  ƒ } t j | ƒ \ } } } t  j j | | j ƒ t  j j | j	 ƒ  | j
 j	 ƒ  ƒ t  j j | | j j	 ƒ  ƒ d  S(   Ni
   i   (   i
   i   (   R   R   R   R   R    R   R   R   t   valuesR	   t   indext   columns(   R
   R   R   R   R   (    (    s@   lib/python2.7/site-packages/statsmodels/tools/tests/test_data.pyt   test_dataframe#   s    "c          C   s—   t  j j d
 ƒ }  t j |  d d d g ƒ} d d l m } | d | ƒ } t  j j t j	 | d  ƒ ƒ | d	 | ƒ } t  j j t j	 | | ƒ ƒ d  S(   Ni
   i   R   R   R   iÿÿÿÿ(   t   dmatrixs   var1 - 1s   var2 - 1(   i
   i   (   R   R   R   R   t   patsyR    R   t   assert_R    t   _is_using_patsyR   (   R
   R   R    t   endogt   exog(    (    s@   lib/python2.7/site-packages/statsmodels/tools/tests/test_data.pyt   test_patsy_577+   s    (
   R   t   numpyR   t   statsmodels.toolsR    R   R   R   R   R&   (    (    (    s@   lib/python2.7/site-packages/statsmodels/tools/tests/test_data.pyt   <module>   s   	
			
	