U
    md                     @   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?   )decimalgY8m?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   