3
���h% � @ sP d dl Z d dlZd dlZd dlZd dljZd dl m
Z
dd� Z G dd� d�Z dS )� N)�read_sasc C s< x6| j D ],}| | jtjd�kr| | jtj�| |< qW d S )N�int64)�columnsZdtype�np�astypeZfloat64)�data�v� r �@/tmp/pip-build-5_djhm0z/pandas/pandas/tests/io/sas/test_xport.py�numeric_as_float s r c @ sX e Zd Zejdd�dd� �Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dS )� TestXportT)Zautousec C sr |ddd�| _ tjj| j d�| _tjj| j d�| _ttjj| j d�d�| _tjj| j d�| _tjj| j d�| _ d S ) N�ioZsasr z
DEMO_G.xptz SSHSV1_A.xpt�rbz DRXFCD_G.xptzpaxraw_d_short.xpt)
�dirpath�os�path�join�file01�file02�open�file02b�file03�file04)�selfZdatapathr r r
� setup_method s zTestXport.setup_methodc C sX t j| jjdd��}t|� t| jdd�}tj||� |jd }t| jddd�}|j |d �}|jd |kspt
�|j � t| jddd�}|j d �}|j � tj||j dd �d d �f � t| jdd d
�}|j
� }|j � tj||j dd �d d �f � d}t| jddd
�}x |D ]}||jd 7 }�qW |j � ||k�s>t
�t| j�}tj||� d S ) Nz.xptz.csv�xport)�formatr T)r �iterator�d �
)r � chunksize)�pd�read_csvr �replacer r �tm�assert_frame_equal�shape�read�AssertionError�close�iloc� get_chunk)r �data_csvr Znum_rows�reader�m�xr r r
� test1_basic! s2
zTestXport.test1_basicc C s� t j| jjdd��}|jd�}t|� t| jddd�}tj||dd� t| jdddd �}|j d
�}|j
� tj||j d d
�d d �f dd� t| jddd
d �}|j � }|j
� tj||j d d
�d d �f dd� d S )
Nz.xptz.csv�SEQNr )�indexr F)�check_index_typeT)r2 r r r r )r2 r r )
r! r" r r# � set_indexr r r$ r% r' r) r* r+ )r r, r r- r r r
� test1_indexK s
"zTestXport.test1_indexc C s` t j| jjdd��}|jd�}t|� t| jddd�}t|�}t j|dd�}t j
||dd � d S )
Nz.xptz.csvr1 i� )r2 r r )ZaxisF)r3 ) r! r" r r# r4 r r �list�concatr$ r% )r r, r- Zall_datar r r r
�test1_incrementalc s
zTestXport.test1_incrementalc C s6 t j| jjdd��}t|� t| j�}tj||� d S )Nz.xptz.csv)r! r" r r# r r r$ r% )r r, r r r r
�test2q s
zTestXport.test2c C s: t j| jjdd��}t|� t| jdd�}tj||� d S )Nz.xptz.csvr )r ) r! r" r r# r r r r$ r% )r r, r r r r
� test2_binary{ s zTestXport.test2_binaryc C s2 t j| jjdd��}t| jdd�}tj||� d S )Nz.xptz.csvzutf-8)�encoding)r! r" r r# r r$ r% )r r, r r r r
�test_multiple_types� s z TestXport.test_multiple_typesc C s8 t j| jjdd��}t| jdd�}tj|jd�|� d S )Nz.xptz.csvr )r r )r! r" r r# r r$ r% r )r r, r r r r
� test_truncated_float_support� s z&TestXport.test_truncated_float_supportN)
�__name__�
__module__� __qualname__�pytestZfixturer r0 r5 r8 r9 r: r< r= r r r r
r s *
r )
r Znumpyr rA Zpandasr! Zpandas._testingZ_testingr$ Zpandas.io.sas.sasreaderr r r r r r r
�<module> s
|