3
UdR                 @   s   d Z ddlZddlZddlZddlZddlZddlmZ ddl	j
Zejddd ejd ejjdZejd	d
 Zdd Zdd Zdd Zdd Zdd Zdd ZdS )z test orc compat     N)read_orcZpyarrowz0.13.0)Z
minversionzpyarrow.orcz4ignore:RangeIndex.* is deprecated:DeprecationWarningc             C   s   | dddS )NiodataZorc )Zdatapathr   r   L/var/www/html/virt/lib64/python3.6/site-packages/pandas/tests/io/test_orc.pydirpath   s    r   c          	   C   s   ddddddddd	g	}d
ddddddddg	}t jt jdd}x&t||D ]\}}t j|d||< qJW tjj| 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)dtypezTestOrcFile.emptyFile.orc)columns)pd	DataFrameZ
RangeIndexzipZSeriesospathjoinr   tmassert_equal)r   r   ZdtypesexpectedZcolnamer   	inputfilegotr   r   r   test_orc_reader_empty   s0    r'   c             C   s   t jddgddt jddgddt jdd	gd
dt jddgddt jddgddt jddgddt jddgddt jddgddt jddgddd	}tjj|}tjj| 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   hiZbye)	r   r	   r
   r   r   r   r   r   r   zTestOrcFile.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   ddl m} dtj|d|d|d|d|d|d	|d
|d|d|dg
ddi}tjj|}tjj	| 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   zTestOrcFile.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 jddddddddd	d
g
ddt jtjdddtjdddtjdddtjdddtjdddtjdddtjdddtjdddtjdddtjdddg
ddd}tjj|}tjj	| d}t
|jd d }tj|| d S )Nz1900-05-05 12:34:56.100000z1900-05-05 12:34:56.100100z1900-05-05 12:34:56.100200z1900-05-05 12:34:56.100300z1900-05-05 12:34:56.100400z1900-05-05 12:34:56.100500z1900-05-05 12:34:56.100600z1900-05-05 12:34:56.100700z1900-05-05 12:34:56.100800z1900-05-05 12:34:56.100900zdatetime64[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 jddddddddd	d
g
ddt jtjdddtjdddtjdddtjdddtjdddtjdddtjdddtjdddtjdddtjdddg
ddd}tjj|}tjj	| d}t
|jd d }tj|| d S )Nz2038-05-05 12:34:56.100000z2038-05-05 12:34:56.100100z2038-05-05 12:34:56.100200z2038-05-05 12:34:56.100300z2038-05-05 12:34:56.100400z2038-05-05 12:34:56.100500z2038-05-05 12:34:56.100600z2038-05-05 12:34:56.100700z2038-05-05 12:34:56.100800z2038-05-05 12:34:56.100900zdatetime64[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 jddddddddddg
ddt jddddddddddg
ddd}tjj|}tjj| d}t|j	d d }t
j|| d S )Ni%EijFiA!{i#iY5B
irhi;SinqiMi~br   )r   Zf50dcb8Z382fdaaaZ90758c6Z9e8caf3fZee97332bZd634da1Z2bea4396Zd67d89e8Zad71007eZe8c82066r   )r   r   zTestOrcFile.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   Znumpyr,   ZpytestZpandasr   r   Zpandas._testingZ_testingr"   ZimportorskipmarkfilterwarningsZ
pytestmarkZfixturer   r'   r0   r5   r;   r<   r=   r   r   r   r   <module>   s$   

!))