
    M/PhF                     Z   d dl 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 d dlmZmZmZmZ e j                            e j                            e                    ZeeeefZesd dlmZ eefz  Zej        j        d             Zej        j        d	             Zd
 Z d Z!dS )    N)timeout)	HTTPErrorURLError)assert_assert_array_equal)PYTHON_IMPL_WASM)get_rdatasetwebusecheck_internetutils)SSLErrorc                  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      e/var/www/html/test/jupyter/venv/lib/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>:   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?   7   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)r6   r5   r   r6   )r7   z(Failed with HTTP Error, these are random)statsmodels.compat.pandasrA   statsmodels.datasetsr6   load_pandasdatar   r   r   r
   r   astypefloat)rA   r6   dtar=   r   r>   s         r   test_webuse_pandasrI   G   s    <<<<<<......



!
!
&C0G'00 ?=>>>@k7333 @ @ @>?????@;;uDtSZZ../////s   A   B B)"r$   socketr   urllib.errorr   r   numpyr:   numpy.testingr   r   r   statsmodels.compat.pythonr   rC   r	   r
   r   r   r&   dirnameabspath__file__r   UnicodeEncodeErrorr   sslr   marksmoker    r+   r?   rI   r/   r4   r   <module>rV      s]   				       , , , , , , , ,     5 5 5 5 5 5 5 5  6 6 6 6 6 6 L L L L L L L L L L L L
'//"'//(33
4
4+=wG  &8+%    ) ) )$# # # 0 0 0 0 0r4   