U
|
®ý°d
|
ã @ sf d dl m Z d dlmZ d dlZd dlZd dlZd dlmZ d dl m
|
Z d dl m Z G dd dZdS )é )Údatetime)ÚStringION)ÚSeries)Ú
|
get_handlec
|
@ sº e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z e
|
j de dddgdddgdddfe 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 dd d}|j f | | d¡}tj|f| d¡}|d krJd |_|j_|S )Nr )Ú index_colÚheaderr Úcolumns)ÚupdateÚgetÚpdÚread_csvÚsqueezeÚnameÚindex)ÚselfÚpathÚkwargsÚparamsr Úout© r ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/series/methods/test_to_csv.pyr s
|
|
zTestSeriesToCSV.read_csvc
|
|
C s |j d ¡|_ t ¡ p}|j|dd | j|dd}tj||dd |jd ksTt|j jd ksdt|j|dd | j|dd}|jdkst|j|dd | |¡}tj||dd |jd ksÄt|j jd ksÔt|j|dd | j|dd}|jdksþtt |d }|
|
d
|
¡ W 5 Q R X | j|d dd }t t d dddt ddddi} t | |¡ | j|d dd }t ddd} t | |¡ W 5 Q R X d S )NF©r T)Ú parse_dates)Z check_namesr ÚtsÚseriesÚwz 1998-01-01|1.0
|
1999-01-01|2.0ú|)Úsepr iÎ é g ð?iÏ g @)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 s8
|
ÿ z TestSeriesToCSV.test_from_csvc
|
C s~ t ¡ l}|j|dd t|d d}| ¡ }W 5 Q R X |d dksHt|j|ddd t |¡}t ||j ¡ W 5 Q R X d S )NFr )Únewliner Ú
|
)r r )
|
r r! r" r% Ú readlinesr$ ÚnpZloadtxtZassert_almost_equalÚvalues)r r' r ÚfÚlinesZarrr r r Ú test_to_csvD s
|
|
zTestSeriesToCSV.test_to_csvc C sT t }tddgddgd}|j|ddd | d¡ | j|ddd }t ||¡ d S )
|
Nu ×Zd2u שr zUTF-8F©Úencodingr r ©r r4 )r r r" Úseekr r r# )r ÚbufÚsÚs2r r r Útest_to_csv_unicode_indexP s
|
z)TestSeriesToCSV.test_to_csv_unicode_indexc C sZ t ¡ H}tdddg}|j|ddd | |¡}tddd g}t ||¡ W 5 Q R X d S )
|
N翶~úÏ¿?ç9DÜJÎ?ç >éD)â?z%.2fF)Z float_formatr g¸
ëQ¸¾?gq=
|
×£pÍ?g=
|
×£p=â?)r r! r r" r r# )r ÚfilenameZserÚrsZxpr r r Útest_to_csv_float_formatZ s
|
|
z(TestSeriesToCSV.test_to_csv_float_formatc C s0 t ddg}|j d¡}t }|j|dd d S )Nz jack and jillzjesse and frankz \s+and\s+Fr )r ÚstrÚsplitr r" )r r8 rB r7 r r r Útest_to_csv_list_entriesc s z(TestSeriesToCSV.test_to_csv_list_entriesc C s. t dddg}|jd dd}t|ts*td S )Nr é é F)Z path_or_bufr )r r" Ú
|
isinstancerA r$ )r r8 Zcsv_strr r r Útest_to_csv_path_is_nonek s z(TestSeriesToCSV.test_to_csv_path_is_nonez
|
s,encodingr; r< r= ÚAÚBÚCÚX)r r NÚabcÚdefZghi)r ÚasciiZ123u ä½ å¥½u ä¸çu ä¸æÚgb2312u Îειά ÏοÏ
u
|
ÎÏÏμεu ÎλληνικάZcp737c
|
C s t ¡ ü}|j|||dd tj|||dd d¡}t ||¡ t|d||d}|j|j|dd W 5 Q R X tj|||dd d¡}t ||¡ t ||¡&}|
|
¡ |p®d ¡}|j |ksÀt W 5 Q R X t ||¡$}t |tj|d|d
|
d¡¡ W 5 Q R X W 5 Q R X d S ) NT)Ú compressionr4 r r )rP r4 r r r )rP r4 r3 Úutf8r5 )r r! r" r r r r# r ÚhandleZdecompress_fileÚreadÚdecoder r$ ) r r8 r4 rP r> ÚresultZhandlesÚfhÚtextr r r Útest_to_csv_compressions sH
|
üû ÿ üû þz'TestSeriesToCSV.test_to_csv_compressionc C sr t dddgt dd¡d}t d¡D}|j|dd | j|dd
|
}| ¡ }|j t
|
¡|_t ||¡ W 5 Q R X d S ) NZfooÚbarZbazr rE r2 z!__tmp_to_csv_interval_index__.csvFr )r ) r r Zinterval_ranger r! r" r Úcopyr ZastyperA r# )r r8 r rU Úexpectedr r r Útest_to_csv_interval_index¥ s z*TestSeriesToCSV.test_to_csv_interval_index)Ú__name__Ú
|
__module__Ú __qualname__r r) r1 r: r@ rC rG ÚpytestÚmarkZ parametrizer rX r\ r r r r r s& )
|
þøþ
|
%r )r Úior Únumpyr- r` Zpandasr r Zpandas._testingZ_testingr Zpandas.io.commonr r r r r r Ú<module> s
|
