ó
áp7]c           @   s  d  d l  m Z m Z d  d l Z d  d l m Z d  d l 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 j e j 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(   iÿÿÿÿ(   t	   HTTPErrort   URLErrorN(   t   SSLError(   t   timeout(   t   assert_t   assert_array_equal(   t   get_rdatasett   webuset   check_internett   utilsc          C   sž   d }  t  |  ƒ } | s( t j d ƒ n  y t d d d t ƒ} Wn t k
 ra t j d ƒ n Xt t | t j	 ƒ ƒ t d d d t ƒ} t | j
 ƒ d  S(   Ns[   https://raw.githubusercontent.com/vincentarelbundock/Rdatasets/master/csv/datasets/cars.csvs'   Unable to retrieve file - skipping testt   Duncant   carDatat   caches3   Failed with HTTPError or URLError, these are random(   R   t   pytestt   skipR   t   cur_dirt   IGNORED_EXCEPTIONSR   t
   isinstanceR	   t   Datasett
   from_cache(   t   test_urlt   internet_availablet   duncan(    (    sD   lib/python2.7/site-packages/statsmodels/datasets/tests/test_utils.pyt   test_get_rdataset   s    c          C   s½   y t  d d d t ƒ}  Wn t k
 r9 t j d ƒ n Xt |  j t k ƒ t  d d d t ƒ} t | j t k ƒ d } t	 j
 t	 j j t | ƒ ƒ d } t	 j
 t	 j j t | ƒ ƒ d  S(   Nt   Guerryt   HistDataR   s3   Failed with HTTPError or URLError, these are randomsY   raw.githubusercontent.com,vincentarelbundock,Rdatasets,master,csv,HistData,Guerry.csv.zips]   raw.githubusercontent.com,vincentarelbundock,Rdatasets,master,doc,HistData,rst,Guerry.rst.zip(   R   R   R   R   R   R   R   t   Falset   Truet   ost   removet   patht   join(   t   guerryt   guerry2t   fn(    (    sD   lib/python2.7/site-packages/statsmodels/datasets/tests/test_utils.pyt"   test_get_rdataset_write_read_cache#   s    c          C   s®   d d l  m }  t j g  |  D] } t | ƒ ^ q ƒ } d } t | ƒ } | s` t j d ƒ n  y t d d | d t	 ƒ} Wn t
 k
 rœ t j d ƒ n Xt | | ƒ d  S(	   Niÿÿÿÿ(   t   macrodata_resultsU   https://github.com/statsmodels/statsmodels/raw/master/statsmodels/datasets/macrodata/s'   Unable to retrieve file - skipping testt	   macrodatat   baseurlt   as_dfs3   Failed with HTTPError or URLError, these are random(   t)   statsmodels.iolib.tests.results.macrodataR$   t   npt   arrayt   listR   R   R   R   R   R   R   (   R$   t   rowt   res2t   base_ghR   t   res1(    (    sD   lib/python2.7/site-packages/statsmodels/datasets/tests/test_utils.pyt   test_webuse6   s    (c          C   s·   d d l  m }  d d l m } | j ƒ  j } d } t | ƒ } | sW t j d ƒ n  y t	 d d | ƒ} Wn t
 k
 r t j d ƒ n X| j t ƒ } |  | | j t ƒ ƒ d  S(	   Niÿÿÿÿ(   t   assert_frame_equal(   R%   sU   https://github.com/statsmodels/statsmodels/raw/master/statsmodels/datasets/macrodata/s'   Unable to retrieve file - skipping testR%   R&   s(   Failed with HTTP Error, these are random(   t   pandas.util.testingR1   t   statsmodels.datasetsR%   t   load_pandast   dataR   R   R   R   R   t   astypet   float(   R1   R%   t   dtaR.   R   R/   (    (    sD   lib/python2.7/site-packages/statsmodels/datasets/tests/test_utils.pyt   test_webuse_pandasF   s    (    t   statsmodels.compat.pythonR    R   R   t   sslR   t   socketR   t   numpyR)   t   numpy.testingR   R   R   R3   R   R   R   R	   R   t   dirnamet   abspatht   __file__R   t   UnicodeEncodeErrorR   t   markt   smokeR   R#   R0   R9   (    (    (    sD   lib/python2.7/site-packages/statsmodels/datasets/tests/test_utils.pyt   <module>   s   "		