
    Vfd                     @   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             Zej        j        d             Zd	 Zd
 ZdS )    N)SSLError)timeout)	HTTPErrorURLError)assert_assert_array_equal)get_rdatasetwebusecheck_internetutilsc                  r   d} t          |           }|st          j        d           	 t          ddt                    }n$# t
          $ r t          j        d           Y nw xY wt          t          |t          j	                             t          ddt                    }t          |j
                   d S )Nz[https://raw.githubusercontent.com/vincentarelbundock/Rdatasets/master/csv/datasets/cars.csv'Unable to retrieve file - skipping testDuncancarDatacache3Failed with HTTPError or URLError, these are random)r   pytestskipr	   cur_dirIGNORED_EXCEPTIONSr   
isinstancer   Dataset
from_cache)test_urlinternet_availableduncans      Elib/python3.11/site-packages/statsmodels/datasets/tests/test_utils.pytest_get_rdatasetr      s    8H'11 ?=>>>Kh	AAA K K KIJJJJJKJvu}--...(IW===FFs   A A"!A"c                     	 t          ddt                    } n$# t          $ r t          j        d           Y nw xY wt          | j        du            t          ddt                    }t          |j        du            d}t          j        t          j	        
                    t          |                     d}t          j        t          j	        
                    t          |                     d S )	NGuerry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)guerryguerry2fns      r   "test_get_rdataset_write_read_cacher*   "   s    Kh
'BBB K K KIJJJJJK F&'''8Zw???GG$&'''
&BIbgll7B''(((
*BIbgll7B''(((((s    ;;c                  ,   ddl m}  t          j        d | D                       }d}t	          |          }|st          j        d           	 t          d|d          }n$# t          $ r t          j        d	           Y nw xY wt          ||           d S )
Nr   )macrodata_resultc                 ,    g | ]}t          |          S  )list).0rows     r   
<listcomp>ztest_webuse.<locals>.<listcomp>8   s    ;;;3T#YY;;;    Shttps://github.com/statsmodels/statsmodels/raw/main/statsmodels/datasets/macrodata/r   	macrodataF)baseurlas_dfr   )
)statsmodels.iolib.tests.results.macrodatar,   nparrayr   r   r   r
   r   r   )r,   res2base_ghr   res1s        r   test_webuser>   5   s    JJJJJJ8;;*:;;;<<D0G'00 ?=>>>Kk7%@@@ K K KIJJJJJKtT"""""s   A   B Bc                     ddl m}  ddlm} |                                j        }d}t          |          }|st          j        d           	 t          d|          }n$# t          $ r t          j        d           Y nw xY w|                    t                    } | ||                    t                               d S )	Nr   )assert_frame_equal)r5   r4   r   r5   )r6   z(Failed with HTTP Error, these are random)statsmodels.compat.pandasr@   statsmodels.datasetsr5   load_pandasdatar   r   r   r
   r   astypefloat)r@   r5   dtar<   r   r=   s         r   test_webuse_pandasrH   E   s    <<<<<<......



!
!
&C0G'00 ?=>>>@k7333 @ @ @>?????@;;uDtSZZ../////s   A   B B) r#   sslr   socketr   urllib.errorr   r   numpyr9   numpy.testingr   r   r   rB   r	   r
   r   r   r%   dirnameabspath__file__r   UnicodeEncodeErrorr   marksmoker   r*   r>   rH   r.   r3   r   <module>rT      s;   				             , , , , , , , ,     5 5 5 5 5 5 5 5  L L L L L L L L L L L L
'//"'//(33
4
486H     ) ) )$# # # 0 0 0 0 0r3   