U
    Ãmœd¾  ã                   @   sT   d dl Zd dlmZ d dlmZ d dlmZ d dlm	Z	m
Z
 d dlmZ dd„ ZdS )	é    N)Úassert_almost_equal)Ú
statecrime)ÚOLS)Úreset_ramseyÚvariance_inflation_factor)Úadd_constantc                  C   s´   t  ¡ j} t| jt| ddg ƒƒ}| ¡ }t|dd�}t|j	ddd� t|j
ddd� t| jƒ d	¡}t | ¡}t||ƒ}t|d
dd� t| jƒ d	¡}t| |ƒ}t|d
dd� d S )NZmurderZhs_gradé   )ZdegreegR¸…ëQø?é   )ÚdecimalgY†8ÖÅmÌ?ZurbangÆÜµ„|p0@)r   Zload_pandasÚdatar   Zviolentr   Úfitr   r   ZfvalueZpvalueÚlistÚcolumnsÚindexÚnpZasarrayr   )r   ÚmodÚresÚstatZexog_idxZdata_arrZvifZvif_df© r   úh/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/statsmodels/stats/tests/test_outliers_influence.pyÚtest_reset_stata   s    



r   )Únumpyr   Znumpy.testingr   Zstatsmodels.datasetsr   Z#statsmodels.regression.linear_modelr   Z$statsmodels.stats.outliers_influencer   r   Zstatsmodels.toolsr   r   r   r   r   r   Ú<module>   s   