
x\c           @   s   d  d l  m Z d  d l Z d  d l Z d  d l j j Z d  d l m	 Z	 e j
 d    Z e j
 d    Z e j
 d    Z e j
 d    Z d S(   i(   t   LooseVersionN(   t   read_csvc         C   s   |  d d d d  S(   s   Path to the tips datasett   iot   parsert   datas   tips.csv(    (   t   datapath(    (    s7   lib/python2.7/site-packages/pandas/tests/io/conftest.pyt	   tips_file   s    c         C   s   |  d d d d  S(   s   Path a JSONL datasetR   R   R   s   items.jsonl(    (   R   (    (    s7   lib/python2.7/site-packages/pandas/tests/io/conftest.pyt
   jsonl_file   s    c         C   s   t  |  d d d d  d d S(   s#   DataFrame with the salaries datasetR   R   R   s   salaries.csvt   seps   	(   R   (   R   (    (    s7   lib/python2.7/site-packages/pandas/tests/io/conftest.pyt   salaries_table   s    c      	   #   sh  t  j d  t  j d  } t  j d  } t | j  t d  k  rV t  j d  n  t j    t j j	 d d  t j j	 d d	  t  j d
  } d |  f d |  d f d |  d f d | f g     f d   } zt | j
   } | j   d } | j d d d     j d |  | |    j d d d d  | d    VWd | j   XWd QXd S(   s/  Fixture for mocking S3 interaction.

    The primary bucket name is "pandas-test". The following datasets
    are loaded.

    - tips.csv
    - tips.csv.gz
    - tips.csv.bz2
    - items.jsonl

    A private bucket "cant_get_it" is also created. The boto3 s3 resource
    is yielded by the fixture.
    t   s3fst   boto3t   botocores   1.11.0s+   botocore is leaking resources before 1.11.0t   AWS_ACCESS_KEY_IDt
   foobar_keyt   AWS_SECRET_ACCESS_KEYt   foobar_secrett   motos   tips.csvs   tips.csv.gzs   .gzs   tips.csv.bz2s   .bz2s   items.jsonlc      
      sR   xK  D]C \ } } t  | d  & }   j |   j d | d |  Wd  QXq Wd  S(   Nt   rbt   Keyt   Body(   t   opent   Buckett
   put_object(   t   bucket_namet   s3_keyt	   file_namet   f(   t   connt   test_s3_files(    s7   lib/python2.7/site-packages/pandas/tests/io/conftest.pyt   add_tips_filesD   s
    s   pandas-testt   s3t   region_names	   us-east-1R   t   cant_get_itt   ACLt   privateN(   t   pytestt   importorskipR    t   __version__t   skipt   tmt!   ensure_safe_environment_variablest   ost   environt
   setdefaultt   mock_s3t   startt   resourcet   create_buckett   stop(   R   R   R   R   R   R   R   t   bucket(    (   R   R   s7   lib/python2.7/site-packages/pandas/tests/io/conftest.pyt   s3_resource   s2    	


	(   t   distutils.versionR    R*   R$   t   pandas.util.testingt   utilt   testingR(   t   pandas.io.parsersR   t   fixtureR   R   R	   R3   (    (    (    s7   lib/python2.7/site-packages/pandas/tests/io/conftest.pyt   <module>   s   