3
Ud]                 @   s  d dl Zd dlZd dlmZmZmZmZ d dlj	Z
d dlmZ eddgdejgddged	d
ddZejddj Zejdd Zdd Zdd Zdd Zejddd Zejddd Zejjdddd gejdd!d" Zejdejdd#d$ Zejd%d&d' Z dS )(    N)	DataFrame
date_rangeread_csvread_parquet)_test_decorators      g       @tsz
2018-06-18   )Zperiods)intfloatstrdtF)indexc              c   s(   t jd} | jd}|V  |jj  d S )NfsspecZmemory)pytestZimportorskip
filesystemstoreclear)r   Zmemfs r   O/var/www/html/virt/lib64/python3.6/site-packages/pandas/tests/io/test_fsspec.py
cleared_fs   s    

r   c             C   s:   ddl m} |td| jd< tddgd}tjt| d S )Nr   )
MemoryFile)dataztest/test.csvzmemory://test/test.csvr   )parse_dates)Zfsspec.implementations.memoryr   textr   r   tmassert_frame_equaldf1)r   r   df2r   r   r   test_read_csv   s    r!   c             C   s   ddl m} ddlm} |jj  tjt }t	d dt
|jksHtW d Q R X d}| j|dd|d	 tjt }t	d
 |t
|jkstW d Q R X d S )Nr   )registry)known_implementationsznosuchprotocol://test/test.csvZnosuchprotocolztest error messgaeZ
couldexistzunimportable.CouldExist)classerrzcouldexist://test/test.csv)r   r"   Zfsspec.registryr#   targetr   r   raises
ValueErrorr   r   valueAssertionErrorsetitemImportError)monkeypatchr   r"   r#   eZerr_mgsr   r   r   test_reasonable_error'   s    
r/   c             C   s.   t jddd tddgdd}tjt | d S )Nzmemory://test/test.csvT)r   r   r   )r   Z	index_col)r   to_csvr   r   r   )r   r    r   r   r   test_to_csv:   s    r1   fastparquetc             C   s   t jddddd dS )zCRegression test for writing to a not-yet-existent GCS Parquet file.zmemory://test/test.csvTr2   N)r   enginecompression)r   
to_parquet)r-   r   r   r   r   test_to_parquet_new_fileA   s    r6   Zs3fsc             C   s@   t jtdt| t jtdt| t jtdt| d S )Nzs3://pandas-test/tips.csvzs3://pandas-test/tips.csv.gzzs3://pandas-test/tips.csv.bz2)r   assert_equalr   )s3_resource	tips_filer   r   r   test_from_s3_csvI   s    r:   protocolZs3Zs3aZs3nc             C   s   t jtd| t| d S )Nz%s://pandas-test/tips.csv)r   r7   r   )r8   r9   r;   r   r   r   test_s3_protocolsQ   s    r<   c             C   s2   d}t j|ddd d t|dd}tjt | d S )Nzs3://pandas-test/test.parquetFr2   )r   r3   r4   )r3   )r   r5   r   r   r7   )r8   fnr    r   r   r   test_s3_parquetY   s    r>   r   c           
   C   s4   t jt } td dt| jks&tW d Q R X d S )Nzmemory://test/test.csvzfsspec library is required)r   r'   r,   r   r   r)   r*   )r.   r   r   r   test_not_present_exceptionb   s    r?   )!Znumpynpr   Zpandasr   r   r   r   Zpandas._testingZ_testingr   Zpandas.utilr   tdnanr   r0   encoder   Zfixturer   r!   r/   r1   Z
skip_if_nor6   r:   markZparametrizer<   r>   Zskip_if_installedr?   r   r   r   r   <module>   s*   
		