U
    Kvf                     @   s   d dl Z d dlmZ d dlmZ d dlmZmZ d dlZ	d dl
mZmZ d dlZd dlmZmZmZmZ e je jeZeeeeefZejjdd Zejjd	d
 Zdd Zdd ZdS )    N)SSLError)timeout)	HTTPErrorURLError)assert_assert_array_equal)get_rdatasetwebusecheck_internetutilsc                  C   sx   d} t | }|std ztddtd}W n tk
rJ   td Y nX tt|tj	 tddtd}t|j
 d S )Nz[https://raw.githubusercontent.com/vincentarelbundock/Rdatasets/master/csv/datasets/cars.csv'Unable to retrieve file - skipping testZDuncanZcarDatacache3Failed with HTTPError or URLError, these are random)r
   pytestskipr   cur_dirIGNORED_EXCEPTIONSr   
isinstancer   ZDataset
from_cache)Ztest_urlinternet_availableZduncan r   I/tmp/pip-unpacked-wheel-2v6byqio/statsmodels/datasets/tests/test_utils.pytest_get_rdataset   s    
r   c                  C   s   zt ddtd} W n tk
r0   td Y nX t| jdk t ddtd}t|jdk d}ttj	
t| d}ttj	
t| d S )	NZGuerryZHistDatar   r   FTz\raw.githubusercontent.com,vincentarelbundock,Rdatasets,master,csv,HistData,Guerry-v2.csv.zipz`raw.githubusercontent.com,vincentarelbundock,Rdatasets,master,doc,HistData,rst,Guerry-v2.rst.zip)r   r   r   r   r   r   r   osremovepathjoin)ZguerryZguerry2fnr   r   r   "test_get_rdataset_write_read_cache"   s    r   c                  C   sz   ddl m}  tdd | D }d}t|}|s:td ztd|dd	}W n tk
rj   td
 Y nX t	|| d S )Nr   )macrodata_resultc                 S   s   g | ]}t |qS r   )list).0rowr   r   r   
<listcomp>8   s     ztest_webuse.<locals>.<listcomp>Shttps://github.com/statsmodels/statsmodels/raw/main/statsmodels/datasets/macrodata/r   	macrodataF)baseurlZas_dfr   )
Z)statsmodels.iolib.tests.results.macrodatar    nparrayr
   r   r   r	   r   r   )r    Zres2base_ghr   res1r   r   r   test_webuse5   s    
r,   c                  C   s   ddl m}  ddlm} | j}d}t|}|s<td zt	d|d}W n t
k
rj   td Y nX |t}| ||t d S )	Nr   )assert_frame_equal)r&   r%   r   r&   )r'   z(Failed with HTTP Error, these are random)Zstatsmodels.compat.pandasr-   statsmodels.datasetsr&   Zload_pandasdatar
   r   r   r	   r   Zastypefloat)r-   r&   Zdtar*   r   r+   r   r   r   test_webuse_pandasE   s    


r1   ) r   sslr   socketr   urllib.errorr   r   Znumpyr(   Znumpy.testingr   r   r   r.   r   r	   r
   r   r   dirnameabspath__file__r   UnicodeEncodeErrorr   markZsmoker   r   r,   r1   r   r   r   r   <module>   s"   

