3
���h � @ sv d dl m Z d dlmZ d dlZd dlZd dlZd dlmZm Z d dl
j Z d dl
mZ G dd� d�ZG dd � d �ZdS )
� )�datetime)�StringION)� DataFrame�Series)�
get_handlec @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z e
j j de
dddgdddgdd�dfe
dddgdd �d fe
d d d gd!d �d"fe
d d#d$gd%d �d&fg�d'd(� �Zd)d*� ZdS )+�TestSeriesToCSVc K sN t ddd dd�}|jf |� |jd�}tj|f|�}|d krJd |_|j_|S )NTr )�squeeze� index_col�header� parse_datesr
)�dict�update�get�pd�read_csv�name�index)�self�path�kwargs�paramsr
�out� r �=/tmp/pip-build-5_djhm0z/pandas/pandas/tests/series/test_io.pyr s
zTestSeriesToCSV.read_csvc
C s� |j jd �|_ tj� ��j}|j|dd� | j|�}tj||dd� |jd ksPt�|j jd ks`t�|j|dd� | j|dd�}|jdks�t�|j|dd� | j|�}tj||dd� |jd ks�t�|j jd ks�t�|j|dd� | j|dd�}|jdks�t�t |d��}|j
d � W d Q R X | j|d
d �}t t d d
d
�dt dd
d
�di�} tj| |� | j|d
dd�}t ddd��} tj| |� W d Q R X d S )NF)r
)Z check_namesTr �ts�series�wz 1998-01-01|1.0
1999-01-01|2.0�|)�sepi� � g �?i� g @)r r )z
1998-01-01z
1999-01-01)
r Z
_with_freq�tm� ensure_clean�to_csvr �assert_series_equalr �AssertionError�open�writer r )
r �datetime_seriesZ
string_seriesr r Zts_hr Zseries_h�outfileZ check_seriesr r r �
test_from_csv s6
z TestSeriesToCSV.test_from_csvc C s� dd l }tj� �n}|j|dd� |j|d d��}|j� }W d Q R X |d dksRt�|j|ddd� tj|�}tj ||j
� W d Q R X d S )Nr F)r
)�newliner �
)r r
) �ior r! r" r% � readlinesr$ �npZloadtxtZassert_almost_equal�values)r r' r, r �f�linesZarrr r r � test_to_csvD s
zTestSeriesToCSV.test_to_csvc C sT t � }tddgddgd�}|j|ddd� |jd� | j|ddd �}tj||� d S )
Nu אZd2u ב)r zUTF-8F)�encodingr
r )r r3 )r r r" �seekr r r# )r �buf�s�s2r r r �test_to_csv_unicode_indexR s
z)TestSeriesToCSV.test_to_csv_unicode_indexc C sZ t j� �H}tdddg�}|j|ddd� | j|�}tddd g�}t j||� W d Q R X d S )
Ng��~�Ϛ�?g9DܜJ�?g >�D�)�?z%.2fF)Z float_formatr
g� ��Q��?gq=
ףp�?g=
ףp=�?)r r! r r" r r# )r �filenameZser�rs�xpr r r �test_to_csv_float_format\ s
z(TestSeriesToCSV.test_to_csv_float_formatc C s0 t ddg�}|jjd�}t� }|j|dd� d S )Nz
jack and jillzjesse and frankz \s+and\s+F)r
)r �str�splitr r" )r r6 r> r5 r r r �test_to_csv_list_entriesf s z(TestSeriesToCSV.test_to_csv_list_entriesc C s. t dddg�}|jd dd�}t|t�s*t�d S )Nr � � F)Z path_or_bufr
)r r" �
isinstancer= r$ )r r6 Zcsv_strr r r �test_to_csv_path_is_nonen s z(TestSeriesToCSV.test_to_csv_path_is_nonez
s,encodingg��~�Ϛ�?g9DܜJ�?g >�D�)�?�A�B�C�X)r r N�abc�defZghi)r �asciiZ123u 你好u 世界u 中文�gb2312u Γειά σουu
Κόσμεu ΕλληνικάZcp737c
$ C s t j� ��}|j|||dd� tj|||ddd�}t j||� t|d||d�\}}|� |j||dd� W d Q R X tj|||ddd�}t j||� t j||��&}|j� j |p�d�} |j
| ks�t �W d Q R X t j||�� }t j|tj|dd|d �� W d Q R X W d Q R X d S )
NT)� compressionr3 r
r )rL r3 r r r )rL r3 )r3 r
�utf8)r r r3 ) r r! r" r r r# r Zdecompress_file�read�decoder r$ )
r r6 r3 rL r9 �resultr0 Z_handles�fh�textr r r �test_to_csv_compressionv s4
z'TestSeriesToCSV.test_to_csv_compressionc C st t dddgtjdd�d�}tjd��F}|j|dd � | j|dd
d �}|j� }|jj t
�|_tj ||� W d Q R X d S ) NZfoo�barZbazr rA )r z!__tmp_to_csv_interval_index__.csvF)r
T)r r ) r r Zinterval_ranger r! r" r �copyr Zastyper= r# )r r6 r rP �expectedr r r �test_to_csv_interval_index� s z*TestSeriesToCSV.test_to_csv_interval_index)�__name__�
__module__� __qualname__r r) r2 r8 r< r? rC �pytest�markZ parametrizer rS rW r r r r r s )
"+r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) � TestSeriesIOc C s� d |_ |j� }tj|j|jd�}tj||� d|_ |j� }tjt|jd�|jd�}tj||� |jdd�}tjt|jd�|jd�}tj||� d S )N)r �testname)r^ �
testdifferent)r )r_ ) r �to_framer r r/ r r �assert_frame_equalr )r r' r: r; r r r �
test_to_frame� s zTestSeriesIO.test_to_framec C sN ddl m} |dddd�}ttjjt|��|�}tj|�}|j j
dksJt �d S )Nr )� period_rangez1/1/2011z1/1/2012�M)�freq) �pandasrc r r. �randomZrandn�lenr Zround_trip_pickler re r$ )r rc Zprngr Znew_tsr r r �test_timeseries_periodindex� s
z(TestSeriesIO.test_timeseries_periodindexc C sD x>dddt ddd�d gD ]$}| jtj|d��}|j|kst�qW d S )
Ni g H�@r i� � r r@ )r )r r@ )r �_pickle_roundtrip_namer ZmakeTimeSeriesr r$ )r �n� unpickledr r r �test_pickle_preserve_name� s z&TestSeriesIO.test_pickle_preserve_namec C s, t j� �}|j|� tj|�}|S Q R X d S )N)r r! Z to_pickler Z read_pickle)r �objr rm r r r rk � s
z#TestSeriesIO._pickle_roundtrip_namec sn G � fdd�dt �}G dd� dt�� |dddgdd �}|j� }t|� �sLt�� ddddgi�}tj||� d S )
Nc s e Zd Ze� fdd��ZdS )z>TestSeriesIO.test_to_frame_expanddim.<locals>.SubclassedSeriesc s � S )Nr )r )�SubclassedFramer r �_constructor_expanddim� s zUTestSeriesIO.test_to_frame_expanddim.<locals>.SubclassedSeries._constructor_expanddimN)rX rY rZ �propertyrq r )rp r r �SubclassedSeries� s rs c @ s e Zd ZdS )z=TestSeriesIO.test_to_frame_expanddim.<locals>.SubclassedFrameN)rX rY rZ r r r r rp � s rp r r@ rA rG )r )r r r` rB r$ r ra )r rs r6 rP rV r )rp r �test_to_frame_expanddim� s z$TestSeriesIO.test_to_frame_expanddimN)rX rY rZ rb ri rn rk rt r r r r r] � s
r] )r r, r Znumpyr. r[ rf r r r Zpandas._testingZ_testingr Zpandas.io.commonr r r] r r r r �<module> s
- |