3
���h] � @ s d dl Zd dlZd dlmZmZmZmZ d dlj Z
d dl m Z
eddgdejgddged d
d �d ��Zejd
d�j� Zejdd� �Zdd� Zdd� Zdd� Ze
jd�dd� �Ze
jd�dd � �Zejj d d d d g�e
jd�d!d"� ��Z e
jd�e
jd�d#d$� ��Z e
j d%�d&d'� �Z dS )(� N)� DataFrame�
date_range�read_csv� read_parquet)�_test_decorators� � g @�t�sz
2018-06-18� )Zperiods)�int�float�str�dtF)�indexc c s( t jd�} | jd�}|V |jj� d S )N�fsspecZmemory)�pytestZ importorskip�
filesystem�store�clear)r Zmemfs� r �=/tmp/pip-build-5_djhm0z/pandas/pandas/tests/io/test_fsspec.py�
cleared_fs s
r c C s: ddl m} |td�| jd< tddgd�}tjt|� d S )Nr )�
MemoryFile)�dataz
test/test.csvzmemory://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� ddl m} ddlm} |jj� tjt�� }t d� dt
|j �ksHt �W d Q R X d}| j
|dd|d �� tjt�� }t d
� |t
|j �ks�t �W d Q R X d S ) Nr )�registry)�known_implementationsz nosuchprotocol://test/test.csvZnosuchprotocolztest error messgaeZ
couldexistzunimportable.CouldExist)�class�errzcouldexist://test/test.csv)r r" Zfsspec.registryr# �targetr r �raises�
ValueErrorr r �value�AssertionError�setitem� ImportError)� monkeypatchr r" r# �eZerr_mgsr r r �test_reasonable_error' s
r/ c C s. t jddd� tddgdd�}tjt |� d S )Nzmemory://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 jddddd� dS )zCRegression test for writing to a not-yet-existent GCS Parquet file.zmemory://test/test.csvTr2 N)r �engine� compression)r �
to_parquet)r- r r r r �test_to_parquet_new_fileA s r6 Zs3fsc C s@ t jtd�t|�� t jtd�t|�� t jtd�t|�� d S )Nzs3://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: �protocolZs3Zs3aZs3nc C s t jtd| �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|ddd d� t|dd�}tjt |� 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 jt�� } td� dt| j�ks&t�W d Q R X d S )Nzmemory://test/test.csvzfsspec library is required)r r' r, r r r) r* )r. r r r �test_not_present_exceptionb s r? )!Znumpy�npr Zpandasr r r r Zpandas._testingZ_testingr Z pandas.utilr �td�nanr r0 �encoder Zfixturer r! r/ r1 Z
skip_if_nor6 r: �markZ parametrizer< r> Zskip_if_installedr? r r r r �<module> s*
|