3
���h� � @ s� d dl mZ d dlZd dlZd dlZd dlmZmZm Z d dl
j Z d dl
mZ ejd�dd� �Zejd�dd � �Zejd
�ejd�d d � ��Zejd�d
d� �ZdS )� )�BytesION)� DataFrame�
date_range�read_csv)�_test_decoratorsZgcsfsc s� ddl m}m} |jj� tddgdtjgddgtdd d
�d ��� G � fd d
�d
|�}| j d|� t
ddgd�}t j � |� d S )Nr )�AbstractFileSystem�registry� � g @�t�sz
2018-06-18� )�periods)�int�float�str�dtc s e Zd Z� fdd�ZdS )z,test_read_csv_gcs.<locals>.MockGCSFileSystemc s t � jdd�j� �S )NF)�index)r �to_csv�encode)�args�kwargs)�df1� �:/tmp/pip-build-5_djhm0z/pandas/pandas/tests/io/test_gcs.py�open s z1test_read_csv_gcs.<locals>.MockGCSFileSystem.openN)�__name__�
__module__� __qualname__r r )r r r �MockGCSFileSystem s r zgcsfs.GCSFileSystemzgs://test/test.csvr )� parse_dates)
�fsspecr r �target�clearr �np�nanr �setattrr �tm�assert_frame_equal)� monkeypatchr r r �df2r )r r �test_read_csv_gcs s
r+ c s� ddl m}m} |jj� tddgdtjgddgtdd d
�d ��� t � �d d
� �_
G �fdd�d|�}| j d|� � j ddd� � fdd�}| j d|� t
ddgdd�}tj� |� d S )Nr )r r r r
g @r r z
2018-06-18r
)r )r r r r c S s dS )NTr r r r r �<lambda>3 s z!test_to_csv_gcs.<locals>.<lambda>c s e Zd Z� fdd�ZdS )z*test_to_csv_gcs.<locals>.MockGCSFileSystemc s � j d� � S )Nr )�seek)r r )r r r r 6 s
z/test_to_csv_gcs.<locals>.MockGCSFileSystem.openN)r r r r r )r r r r 5 s r zgcsfs.GCSFileSystemzgs://test/test.csvT)r c s t � jdd�j� �d d dfS )NT)r F)r r r )r r )r r r �mock_get_filepath_or_buffer= s z4test_to_csv_gcs.<locals>.mock_get_filepath_or_bufferz'pandas.io.common.get_filepath_or_bufferr )r Z index_col)r! r r r" r# r r$ r% r r �closer& r r r' r( )r) r r r r. r* r )r r r �test_to_csv_gcs% s"
r0 � fastparquetc sx ddl m}m} |jj� tddgdtjgddgtdd d
�d ��}G � fd d
�d
|�}| j d|� |j
ddddd� dS )zCRegression test for writing to a not-yet-existent GCS Parquet file.r )r r r r
g @r r z
2018-06-18r
)r )r r r r c s e Zd Zd� fdd� ZdS )z7test_to_parquet_gcs_new_file.<locals>.MockGCSFileSystem�rc s d|kr t �ttjj� d�|�S )N�wz test.parquet)�FileNotFoundErrorr �os�path�join)�selfr6 �moder )�tmpdirr r r Z s z<test_to_parquet_gcs_new_file.<locals>.MockGCSFileSystem.openN)r2 )r r r r r )r: r r r Y s r zgcsfs.GCSFileSystemzgs://test/test.csvTr1 N)r Zengine� compression) r! r r r" r# r r$ r% r r& Z
to_parquet)r) r: r r r r r )r: r � test_to_parquet_gcs_new_fileI s
r< c
C s4 t jt�� } td� dt| j�ks&t�W d Q R X d S )Nzgs://test/test.csvzgcsfs library is required)�pytestZraises� ImportErrorr r �value�AssertionError)�er r r � test_gcs_not_present_exceptione s rB )�ior r5 Znumpyr$ r= Zpandasr r r Zpandas._testingZ_testingr' Z pandas.utilr �tdZ
skip_if_nor+ r0 r< Zskip_if_installedrB r r r r �<module> s
$ |