3
���hR � @ s� d Z ddlZddlZddlZddlZddlZddlmZ ddl j
Z ej ddd� ej d� ej
jd�Zejd d
� �Zd d � Zd
d� Zdd� Zdd� Zdd� Zdd� ZdS )z test orc compat � N)�read_orcZpyarrowz0.13.0)Z
minversionz pyarrow.orcz4ignore:RangeIndex.* is deprecated:DeprecationWarningc C s | ddd�S )N�io�dataZorc� )Zdatapathr r �:/tmp/pip-build-5_djhm0z/pandas/pandas/tests/io/test_orc.py�dirpath s r c C s� ddddddddd g }d
d d d
dddddg }t jt jd�d�}x&t||�D ]\}}t j|d�||< qJW tjj| d�}t||d�}t j
||� d S )N�boolean1�byte1�short1�int1�long1�float1�double1�bytes1�string1�bool�int8�int16�int32�int64�float32�float64�objectr )�index)�dtypezTestOrcFile.emptyFile.orc)�columns) �pd� DataFrameZ
RangeIndex�zipZSeries�os�path�joinr �tm� assert_equal)r r Zdtypes�expectedZcolnamer � inputfile�gotr r r �test_orc_reader_empty s0 r' c C s� t jddgdd�t jddgdd�t jdd gd
d�t jd d gd d�t jd
d
gdd�t jddgdd�t jd d gdd�t jddgdd�t jddgdd�d� }tjj|�}tjj| d�}t||j � d �}t
j ||� d S ) NFTr )r � �d r i i r i r l ���� r g �?g @r g .@g @r s � r �hiZbye) r r r
r r r
r r r zTestOrcFile.test1.orc)r g .�g �) �np�arrayr r � from_dictr r r! r �keysr" r# )r r r$ r% r& r r r �test_orc_reader_basic: s r0 c
C s� ddl m} dtj|d�|d�|d�|d�|d�|d �|d
�|d �|d �|d
�g
dd�i}tjj|�}tjj | d�}t
|�j d d� }t j
||� d S )Nr )�DecimalZ_col0z -1000.50000z
-999.60000z
-998.70000z
-997.80000z
-996.90000z
-995.10000z
-994.11000z
-993.12000z
-992.13000z
-991.14000r )r zTestOrcFile.decimal.orc�
)�decimalr1 r, r- r r r. r r r! r �ilocr" r# )r r1 r r$ r% r& r r r �test_orc_reader_decimalN s"
r5 c C s� t jddddddddd d
g
d d �t jtjd
dd�tjd
dd�tjd
dd�tjd
dd�tjd
dd�tjd
dd�tjd
dd�tjd
dd�tjd
dd�tjd
dd�g
dd �d�}tjj|�}tjj | d�}t
|�j d d� }t j
||� d S )Nz1900-05-05 12:34:56.100000z1900-05-05 12:34:56.100100z1900-05-05 12:34:56.100200z1900-05-05 12:34:56.100300z1900-05-05 12:34:56.100400z1900-05-05 12:34:56.100500z1900-05-05 12:34:56.100600z1900-05-05 12:34:56.100700z1900-05-05 12:34:56.100800z1900-05-05 12:34:56.100900zdatetime64[ns])r il � � r )�time�datez TestOrcFile.testDate1900.orcr2 )r, r- �datetimer9 r r r. r r r! r r4 r" r# )r r r$ r% r& r r r �test_orc_reader_date_lowk s8 r; c C s� t jddddddddd d
g
d d �t jtjd
dd�tjd
dd�tjd
dd�tjd
dd�tjd
dd�tjd
dd�tjd
dd�tjd
dd�tjd
dd�tjd
dd�g
dd �d�}tjj|�}tjj | d�}t
|�j d d� }t j
||� d S )Nz2038-05-05 12:34:56.100000z2038-05-05 12:34:56.100100z2038-05-05 12:34:56.100200z2038-05-05 12:34:56.100300z2038-05-05 12:34:56.100400z2038-05-05 12:34:56.100500z2038-05-05 12:34:56.100600z2038-05-05 12:34:56.100700z2038-05-05 12:34:56.100800z2038-05-05 12:34:56.100900zdatetime64[ns])r i� r6 r7 r )r8 r9 z TestOrcFile.testDate2038.orcr2 )r, r- r: r9 r r r. r r r! r r4 r" r# )r r r$ r% r& r r r �test_orc_reader_date_high� s8 r< c C s� t jdddd ddddd d g
d d �t jd
dddddddddg
dd �d�}tjj|�}tjj| d�}t|�j d d� }t
j ||� d S ) Ni��%Ei��jFiA�!{i#��iY5B
i��rhi;��Sin��qi�Mi~b�r )r Zf50dcb8Z382fdaaaZ90758c6Z9e8caf3fZee97332bZd634da1Z2bea4396Zd67d89e8Zad71007eZe8c82066r )r r zTestOrcFile.testSnappy.orcr2 iEAںi�~�i�:�i��(�) r, r- r r r. r r r! r r4 r" r# )r r r$ r% r& r r r �!test_orc_reader_snappy_compressed� s8 r= )�__doc__r: r Znumpyr, ZpytestZpandasr r Zpandas._testingZ_testingr" Z importorskip�mark�filterwarningsZ
pytestmarkZfixturer r' r0 r5 r; r< r= r r r r �<module> s$
! )) |