U
    KvfÖ  ã                   @   s8   d dl Zd dlZd dlmZ dd„ Zdd„ Zdd„ ZdS )	é    N)Údatac                  C   sX   t j d¡} t j| d< t | ¡}t |¡\}}}t j | 	¡ ddddddd	d
dg	¡ dS )z
    Fixes GH: #144
    ©é
   é   )é   é   r   r   é   é   r   é   é   é   é	   N)
ÚnpÚrandomÚnanÚpandasÚ	DataFramer   Úinterpret_dataÚtestingÚassert_equalÚtolist©ÚXÚdfÚvalsZcnamesZrnames© r   úE/tmp/pip-unpacked-wheel-2v6byqio/statsmodels/tools/tests/test_data.pyÚtest_missing_data_pandas   s
    

r   c                  C   sf   t j d¡} t | ¡}t |¡\}}}t j ||j¡ t j | 	¡ |j
 	¡ ¡ t j ||j 	¡ ¡ d S )Nr   )r   r   r   r   r   r   r   r   Úvaluesr   ÚindexÚcolumnsr   r   r   r   Útest_dataframe   s    
r!   c                  C   sj   t j d¡} tj| ddgd}ddlm} |d|ƒ}t j t 	|d ¡¡ |d|ƒ}t j t 	||¡¡ d S )	N)r   r   Zvar1Zvar2)r    r   )Údmatrixzvar1 - 1zvar2 - 1)
r   r   r   r   Zpatsyr"   r   Zassert_r   Z_is_using_patsy)r   r   r"   ZendogZexogr   r   r   Útest_patsy_577   s    

r#   )Znumpyr   r   Zstatsmodels.toolsr   r   r!   r#   r   r   r   r   Ú<module>   s
   	