3
���h> � @ s� d Z ddlmZ ddlZddlZddljjZ ddl
Z ddl j
ZddlmZmZ ejd�Zeej�Zejjd�ZeejjG dd� d���ZdS ) z test feather-format compat � )� LooseVersionN)� read_feather�
to_feather�pyarrowzignore:The Sparsec @ s� e Zd Zdd� Zdi fdd�Zdd� Zdd � Zd
d � Zd d
� Zdd� Z e
j ddd�dd� �Z dd� Z
dd� Zdd� Zdd � Zd d � Ze
j dd d�d d!� �Ze
j d�ejd"d#� ��ZdS )$� TestFeatherc C s8 t j|��$ tj� �}t||� W d Q R X W d Q R X d S )N)�pytestZraises�tm� ensure_cleanr )�self�df�exc�path� r �>/tmp/pip-build-5_djhm0z/pandas/pandas/tests/io/test_feather.py�check_error_on_write s
z TestFeather.check_error_on_writeNc
K sJ |d kr |}t j� �,}t||f|� t|f|�}t j||� W d Q R X d S )N)r r r r �assert_frame_equal)r
r �expected� write_kwargsZ read_kwargsr
�resultr r r �check_round_trip s
z TestFeather.check_round_tripc C sF x@t jdddg�ddt jd�tjdddg�gD ]}| j|t� q.W d S )N� � � �foo�20130101)�pdZSeries� Timestamp�np�arrayr �
ValueError)r
�objr r r �
test_error+ s zTestFeather.test_errorc C s" t jtd�ttdd��tjdd�jd�tjddd d
�d tjdgd d
d gd tjd
gt jtd��t j tt j
ddd��d d�t j tt j
dddd��d d�t j d�t j t j d�gt j tt j
dddd��d d�d� �}t
td�kr�t jdddd�|d< t jd dd�|d < |jjjjdk�st�| j|� d S ) N�abcr � r � �u1g @g @Zfloat64)�dtypeg �?TFr )�periods)�freqz
US/Eastern)r' �tzZ20130103�ns)r' r( ) �string�intZuint�floatZfloat_with_null�boolZbool_with_null�cat�dt�dttzZ dt_with_nullZdtnsz
0.16.1.devZ2013�M)r( r' r' z1 dayZ
timedeltas)r � DataFrame�list�ranger �arangeZastype�nanZ CategoricalZ
DatetimeIndex�
date_ranger ZNaT�pyarrow_versionr Z period_rangeZtimedelta_ranger1 r& r) �zone�AssertionErrorr )r
r r r r �
test_basic6 s0
zTestFeather.test_basicc C s4 t jtjd�jdd�td�d�j� }| j|t� d S )N� r# r Zaaa)�columns) r r3 r r6 �reshaper4 �copyr r )r
r r r r �test_duplicate_columns] s $z"TestFeather.test_duplicate_columnsc C s, t jtjd�jdd��j� }| j|t� d S )Nr= r# r )r r3 r r6 r? r@ r r )r
r r r r �test_stringify_columnsd s z"TestFeather.test_stringify_columnsc C sP t jtd�ttdd��td�ttdd��d��}ddg}| j||| |d � d S )
Nr" r r# Zxyz� )�col1Zcol2�col3Zcol4rD rE )r r> )r r3 r4 r5 r )r
r r> r r r �test_read_columnsi s z TestFeather.test_read_columnsr z0.17.1)� min_versionc C s4 t jddgddgddgd��}| j|dd gd
� d S ) Nr r �x�yTF)�A�B�CrK rJ )r> )r r3 r )r
r r r r � read_columns_different_orderv s z(TestFeather.read_columns_different_orderc C s$ t jddddgi�}| j|t� d S )N�ar g @)r r3 r � Exception)r
r r r r �test_unsupported_other| s z"TestFeather.test_unsupported_otherc C s4 t jdtjd�i�}| j|dd� | j|dd� d S )NrJ i�� T)Z use_threadsF)r r3 r r6 r )r
r r r r �test_rw_use_threads� s z TestFeather.test_rw_use_threadsc C s� t jddddgi�}| j|� xPdddgt jddd�td�dddgt jjd
ddg�gD ]}||_| j|t � qVW d ddg|_d |j_
| j|t � d ddg|_t jjdg�|_ | j|t � d S )NrJ r r r r# r )r' r" rN �br r )rN r )rN r )rR r )rN r ) r r3 r r8 r4 Z
MultiIndex� from_tuples�indexr r �namer> )r
r rT r r r �test_write_with_index� s
z!TestFeather.test_write_with_indexc C s, t j� j� }t j|jtj�}t j||� d S )N)r �
makeDataFrame� reset_indexZround_trip_pathlibr r r r )r
r r r r r �test_path_pathlib� s z TestFeather.test_path_pathlibc C s, t j� j� }t j|jtj�}t j||� d S )N)r rW rX Zround_trip_localpathr r r r )r
r r r r r �test_path_localpath� s z TestFeather.test_path_localpathz
0.16.1.devc C s$ t j� j� }| j|tdd�d� d S )Nr )�version)r )r rW rX r �dict)r
r r r r �test_passthrough_keywords� s z%TestFeather.test_passthrough_keywordsc C s( d}t j|�}t j|�}tj||� d S )Nzmhttps://raw.githubusercontent.com/pandas-dev/pandas/master/pandas/tests/io/data/feather/feather-0_3_1.feather)r r r r )r
Z feather_file�urlr �resr r r �test_http_path� s
zTestFeather.test_http_path)�__name__�
__module__� __qualname__r r r! r< rA rB rF �tdZ
skip_if_norM rP rQ rV rY rZ r] r �networkr` r r r r r s '
r )�__doc__Zdistutils.versionr Znumpyr r Z pandas.util._test_decorators�utilZ_test_decoratorsrd Zpandasr Zpandas._testingZ_testingr Zpandas.io.feather_formatr r Z importorskipr � __version__r9 �mark�filterwarningsZ
filter_sparseZsingler r r r r �<module> s
|