U
|
ý°d¥À ã @ s¨ d dl Z d dlmZ d dlZd dlZd dlZd dlmZ d dl Z
|
d dl m Z m Z m Z mZmZmZmZmZmZ d dlmZ d dlm mZ d dlm Z G dd dZ dS )é N)ÚStringIO)Ú ParserError) Ú DataFrameÚIndexÚ
|
MultiIndexÚNaTÚSeriesÚ TimestampÚ
|
date_rangeÚread_csvÚ to_datetime)Ú
|
get_handlec @ sP e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z e
|
j ddddgg¡dd Z e
|
j jdd Zd·ddZe
|
j je
|
j ddd d d d d d!d"d#d$d%d&d'd(g¡d)d* Ze
|
j je
|
j ddd d d d d d!d"d#d$d%d&d'd(g¡e
|
j d+d,d-d.d/g¡e
|
j d0d1dd2d3g¡d4d5 Ze
|
j je
|
j dd d6d d d d d!d"d#d$d%d&d'd(g¡e
|
j d0d1dd2d3g¡d7d8 Ze
|
j je
|
j dd d6d d d d g¡d9d: Ze
|
j jd;d< Ze
|
j jd=d> Ze
|
j je
|
j ddd d d d d d!d"d#d$d%d&d'd(g¡e
|
j d0dd2d3g¡e
|
j d?d@didAdigdBdidCdigdddDdddEgg¡dFdG ZdHdI ZdJdK ZdLdM ZdNdO ZdPdQ Z dRdS Z dTdU Z dVdW Z dXdY Z dZd[ Z!d\d] Z"e
|
j d^d_d`dag¡dbdc Z#e
|
j jddde Z$dfdg Z%dhdi Z&djdk Z'dldm Z(dndo Z)dpdq Z*drds Z+e
|
j dtddug¡dvdw Z,dxdy Z-dzd{ Z.d|d} Z/d~d Z0e
|
j de1dddgdd
dggddgdddgddfe1dddggdddgddfe1ddddgg dddgddfe1ddddgg dddgddfg¡dd Z2dd Z3e
|
j de4 5d¡e4 5d ¡g¡d¡d¢ Z6d£d¤ Z7d¥d¦ Z8d§d¨ Z9d©dª Z:d«d¬ Z;dd® Z<d¯d° Z=d±d² Z>d³d´ Z?dµd¶ Z@dS )¸ÚTestDataFrameToCSVc K s ddi}|j f | t|f|S )NÚ index_colr )Úupdater )ÚselfÚpathÚkwargsÚparams© r ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_to_csv.pyr s zTestDataFrameToCSV.read_csvc C sh t d¡R}tj|jd d
|j d¡f< | |¡ |j|ddgd |j|dd |j|dd |j d ¡|_| |¡ | j
|
|d d
|
}t ||¡ |j|d d | j
|
|d d d }t |jt |jd ksÎt |j|dd | j
|
|d d d }t |j|j¡ tttdtjdtjddttdtjdtjddd}| |¡ |
|
|¡}t ||¡ W 5 Q R X d S )NZ__tmp_to_csv_from_csv1__é ÚAÚB©ÚcolumnsF©Úheader©ÚindexT)Ú parse_datesr ©Z index_label©r r é é ©Údtypeé )Ús1Ús2)ÚtmÚ ensure_cleanÚnpÚnanÚilocr Zget_locÚto_csvr Ú
|
_with_freqr Úassert_frame_equalÚlenÚAssertionErrorÚassert_almost_equalÚvaluesr r ÚrangeÚarangeÚint64)r Ú float_frameÚdatetime_framer ÚreconsÚdmr r r Útest_to_csv_from_csv1# s0
|
|
þÿ
|
|
z(TestDataFrameToCSV.test_to_csv_from_csv1c
|
|
C s0 t d¡}ttj dd¡dddgdddgd}| |¡ | |¡}t ||¡ t
|
d d d
|
g¡}ttj dd¡|dddgd}| |¡ | j|d d d gdd}t j||dd t ddddg}|j||d | |¡}| ¡ }||_ t ||¡ d} tjt| d |j|ddgd W 5 Q R X W 5 Q R X d S )NZ__tmp_to_csv_from_csv2__r$ ÚaÚbÚxÚyÚz©r r )r r# r' )r r# r' r r# r' Fr" ©Z check_namesZAAÚXÚYÚZr z Writing 4 cols but got 2 aliases©Úmatch)r* r+ r r, ÚrandomÚrandnr/ r r1 r Ú from_tuplesr Úcopyr ÚpytestÚraisesÚ
|
ValueError)
|
r r9 r ÚdfÚresultZmidxZ col_aliasesÚrsÚxpÚmsgr r r Útest_to_csv_from_csv2I s. ÿ
|
|
|
|
z(TestDataFrameToCSV.test_to_csv_from_csv2c C s¢ t d¡}ttj dd¡}ttj dd¡}| |¡ |j|ddd t ||g¡}t |dd}d d
|
|j
|
D |_
|
d d
|
|j
|
D |_
|
t ||¡ W 5 Q R X d S ) NZ__tmp_to_csv_from_csv3__r$ r# r> F)Úmoder r ©r c S s g | ] }t |qS r ©Úint©Ú.0Úlabelr r r Ú
|
<listcomp>p s z<TestDataFrameToCSV.test_to_csv_from_csv3.<locals>.<listcomp>c S s g | ] }t |qS r rY r[ r r r r^ q s ) r* r+ r r, rJ rK r/ ÚpdÚconcatr r r1 )r r Zdf1Údf2rT rS r r r Útest_to_csv_from_csv3g s
|
z(TestDataFrameToCSV.test_to_csv_from_csv3c s¨ t d¡}tjdd td fddtdD it fddtdD d d
|
d }| |¡ t|d d }t |j
|
¡|_
|
t |d ¡|d< t j ||d d W 5 Q R X d S )NZ__tmp_to_csv_from_csv4__r# )ÚsecondsZdt_datac s g | ] }| qS r r ©r\ Úi©Údtr r r^ y s z<TestDataFrameToCSV.test_to_csv_from_csv4.<locals>.<listcomp>r$ c s g | ] }| qS r r rd rf r r r^ z s Údt_index©Únamer rX T)Zcheck_index_type) r* r+ r_ Ú Timedeltar r6 r r/ r Z to_timedeltar r1 ©r r rQ rR r rf r Útest_to_csv_from_csv4t s þ
|
z(TestDataFrameToCSV.test_to_csv_from_csv4c sf t d¡R} |¡ t|ddgd fdd}|d d< |d d< t ¡ W 5 Q R X d S ) NZ__tmp_to_csv_from_csv5__r r r" c s$ t | j d¡j | jj¡S )NÚUTC)r rg Ú
|
tz_convertÚtz)Úc©rR Útimezone_framer r Ú<lambda> s ÿ
|
þz:TestDataFrameToCSV.test_to_csv_from_csv5.<locals>.<lambda>r ÚC)r* r+ r/ r r1 )r rs r Ú converterr rr r Útest_to_csv_from_csv5 s
|
ÿ z(TestDataFrameToCSV.test_to_csv_from_csv5c C sz d}t |d }t |d¡}|j}|d |d g}t ¡ "}|j|||d t|dd}W 5 Q R X tj|| |dd d S )
|
Nr ç @r$ r' r ©r Ú chunksizerX FrD )rZ r* ÚmakeCustomDataframer r+ r/ r r1 )r rz ÚNrQ ÚcsÚcolsr Úrs_cr r r Útest_to_csv_cols_reordering s
|
z.TestDataFrameToCSV.test_to_csv_cols_reorderingr~ Nr? r> c C sî d}t |d }t |d¡}dddg|_t ¡ ´}|j|||d t|dd}|d k rÈ|jjrf||_n |j |¡\}}|j |¡|_|D ]<} || }
|
|| } t
|
|
|
t r´t |
|
| ¡ qtj |
|
| d d
|
qn|j|_tj ||d d
|
W 5 Q R X d S ) Nr rx r$ r> r? ry r rX FrD )rZ r* r{ r r+ r/ r Z is_uniqueZget_indexer_non_uniqueZtakeÚ
|
isinstancer Zassert_series_equalr1 ) r r~ rz r| rQ r r ZindexerÚmissingrq Zobj_dfZobj_rsr r r Útest_to_csv_new_dupe_cols¢ s(
|
|
z,TestDataFrameToCSV.test_to_csv_new_dupe_colsc C s~ d dd}d}||d }||d d}t d¡B}t||d}|j||d | |¡ t¡}t j||d d
|
W 5 Q R X d S ) Nc S sn |d krt | d }ttdd| d}|rjtj dt||¡D ] }t||< q>tj d¡}t|| < t||< |S )Ng¹?Ú2000Z5min)ÚfreqÚperiodsr éd )rZ Úlistr
|
r, rJ Úrandintr2 r )ÚnZnnatÚsre r r r Úmake_dtnat_arrÇ s
|
|
z<TestDataFrameToCSV.test_to_csv_dtnat.<locals>.make_dtnat_arréè r r z1.csv©r> r? ©rz FrD )N)r* r+ r r/ r Úapplyr r1 )r r rz r( r) ÚpthrQ r; r r r Útest_to_csv_dtnatÄ s
|
z$TestDataFrameToCSV.test_to_csv_dtnatFc st ddi}|rj|d k r$t t||d< t t||d< t d¡$} |j| d|d | j| f|W 5 Q R X n<d|d< t d¡$} |j| d|d | j| f|W 5 Q R X d d
|
|rº|j_|r|sfd d t|d D }
|
t t j g|
|
¡} | _ j
|
d d
|d d
f ddddddd} |rH|dkr~d}t j fdd j D |d_ t j fdd |j D |d|_ nÊ|dkrÆd}t j dd j D |d_ t j dd |j D |d|_ n|dkrd}t j } t j dd | D |d_ t j t tt|j ¡ |d|_ n.| |¡}t j j |d_ t j |j |d|_ |rl|dkrd}t j fdd jD |d_t j fd d |jD |d|_nÎ|dkræd}t j d d jD |d_t j d d |jD |d|_n|dkr>d}t j}t j d d |D |d_|j ¡ }t j d d |D |d|_n.| |¡}t j j|d_t j |j|d|_|fS )!Nr Fr r Z__tmp_to_csv_moar__Úutf8)Úencodingrz r c S s t | ts| d¡S | S )Nr )r ÚstrÚdecode©r@ r r r Ú_to_uniø s
|
|
z;TestDataFrameToCSV._return_result_expected.<locals>._to_unic s g | ]} j d d
|f jqS )N)r. r5 rd )r; r r r^ s z>TestDataFrameToCSV._return_result_expected.<locals>.<listcomp>r# re ÚfÚO)re r r Úurg Úpr c s g | ] } |qS r r r[ ©r r r r^ s r% c s g | ] } |qS r r r[ r r r r^ s rg c S s g | ] }t |qS r ©r r[ r r r r^ s c S s g | ] }t |qS r r r[ r r r r^ s r c S s g | ] }t |qS r r r[ r r r r^ s c s g | ] } |qS r r r[ r r r r^ * s c s g | ] } |qS r r r[ r r r r^ - s c S s g | ] }t |qS r r r[ r r r r^ 2 s c S s g | ] }t |qS r r r[ r r r r^ 5 s c S s g | ] }t |qS r r r[ r r r r^ ; s c S s g | ] }t |qS r r r[ r r r r^ ? s )r r6 r* r+ r/ r r r Ú from_arraysr r. r, Úarrayr Úmapr Z to_timestampÚget)r rQ rz Zr_dtypeZc_dtypeÚrnlvlÚcnlvlÚdupe_colr r Z delta_lvlÚixZtype_mapZidx_listZcol_listr )r r; r Ú_return_result_expectedÞ sº
|
|
ÿ ÿ
|
|
ÿ ÿ
|
|
|
ÿ ÿ
|
|
|
ÿ ÿ
|
|
ÿ ÿ
|
|
|
ÿ
|
ÿ
|
|
z*TestDataFrameToCSV._return_result_expectedÚnrowsr' é
|
éc r ée éf éÆ éÇ éÈ éÉ éÊ éù éú éû c C s: t j|dddd}| |ddd¡\}}t j||dd d S )Né rg r ©Ú
|
r_idx_typeÚ
|
c_idx_typer FrD ©r* r{ r§ r1 )r r¨ rQ rR Úexpectedr r r Útest_to_csv_nrowsG s z$TestDataFrameToCSV.test_to_csv_nrowszr_idx_type, c_idx_type)re re )r r )r rg )r r Úncolsr# r$ rµ c C s: t j||||d}| |d||¡\}}t j||dd d S )Nr¶ r FrD r¹ )r r¨ r· r¸ r¼ rQ rR rº r r r Útest_to_csv_idx_typesP s ÿüz(TestDataFrameToCSV.test_to_csv_idx_typeséb c C s0 t ||¡}| |d¡\}}t j||dd d S ©Nr FrD r¹ )r r¨ r¼ rQ rR rº r r r Útest_to_csv_idx_ncolsd s z(TestDataFrameToCSV.test_to_csv_idx_ncolsc C s t |d¡}t|j}ddg|d d
< ddg|dd
< t|j}ddg|d d
< ddg|dd
< ||_||_| j|ddd\}}t j||d d
|
d S ) Nr$ Zduper' éþÿÿÿZrduper T)r¥ FrD )r* r{ r r r r§ r1 )r r¨ rQ r~ r¦ rR rº r r r Útest_to_csv_dup_colsn s
|
|
z'TestDataFrameToCSV.test_to_csv_dup_colsc C s: t tjdtjdd}| |d¡\}}tj||dd d S )Nr© r% r r F)Zcheck_column_type)r r, r7 r8 r§ r* r1 )r rQ rR rº r r r Útest_to_csv_empty} s z$TestDataFrameToCSV.test_to_csv_emptyc C sD d}t j|d d ddd}| j||dd\}}t j||dd d S )Nr r' r# )Ú r_idx_nlevels)r£ FrD r¹ )r rz rQ rR rº r r r Útest_to_csv_chunksize s z(TestDataFrameToCSV.test_to_csv_chunksizezdf_params, func_paramsrÄ r£ Ú c_idx_nlevelsr¤ ©rÄ rÆ )r£ r¤ c C s8 t j||f|}| j|df|\}}t j||dd d S r¿ r¹ )r r¨ Z df_paramsZ func_paramsr¼ rQ rR rº r r r Útest_to_csv_params s z%TestDataFrameToCSV.test_to_csv_paramsc C sr t j|d< dd }|j |¡|d< t ¡ >}| |¡ | |¡}t ||¡ t t |¡t |¡¡ W 5 Q R X d S )NÚGc S s t jt jgt j ¡ dk S )Nç à?)r, Úinfr- rJ Zrandr r r r rt ó zETestDataFrameToCSV.test_to_csv_from_csv_w_some_infs.<locals>.<lambda>ÚH)
|
r, r- r r¡ r* r+ r/ r r1 Úisinf)r r9 r r r; r r r Ú test_to_csv_from_csv_w_some_infs s
|
|
|
|
z3TestDataFrameToCSV.test_to_csv_from_csv_w_some_infsc C sf t j|d< t j |d< t ¡ >}| |¡ | |¡}t ||¡ t t |¡t |¡¡ W 5 Q R X d S )NÚEÚF)r, rË r* r+ r/ r r1 rÎ )r r9 r r; r r r Ú test_to_csv_from_csv_w_all_infs© s
|
|
|
|
z2TestDataFrameToCSV.test_to_csv_from_csv_w_all_infsc C s t d¡z}tdddgdddgd}|j|d d
|
t|}t ||¡ td d d gdd|d< |j|d d
|
t|}t ||¡ W 5 Q R X d S )NZ__tmp_to_csv_no_index__r# r' r$ rµ r é )Zc1Úc2Fr é é é r8 r% Úc3)r* r+ r r/ r r1 r rl r r r Útest_to_csv_no_indexµ s z'TestDataFrameToCSV.test_to_csv_no_indexc C sD t dddgdddgd}d|d < | ¡ |jd
|
d d gd ks@td S ) Nr> r? rq ÚaaZbbÚcc)r r# ÚtxtÚtestr r# r )r r/ r3 )r rQ r r r Ú test_to_csv_with_mix_columnsÁ s z/TestDataFrameToCSV.test_to_csv_with_mix_columnsc C s¼ t ddgddggddgd}t ddgddggdd gd}t d
|
¡p}|j|dd gd | |¡}t ||¡ |j|d dd gd | |¡}|jdd}|d ks¢tt ||¡ W 5 Q R X d S )Nr# r' r$ rµ r r r rE rF Z__tmp_to_csv_headers__r F)r r T)Zinplace)r r* r+ r/ r r1 Z reset_indexr3 )r Zfrom_dfZto_dfr r; Z return_valuer r r Útest_to_csv_headersÉ s
|
|
z&TestDataFrameToCSV.test_to_csv_headersc C sf |}|j }tjt|d tjd dd¡}tj|ddgd}||_ t d¡T}|j
|
|dd |j
|
|d
|
d gd |
|
|¡ | j |d dgdd}tj ||dd |j j |j j ks²t||_ |} | j }|tjt|tjdg}t |¡| _ | j
|
|ddgd tjtdd | j |d dgdd}
|
W 5 Q R X tj | |
|
dd |
|
|¡ | j |d d}
|
t|
|
jt| jd kslt| j
|
|dd | j |d d}
|
t |
|
j|j¡ ||_ W 5 Q R X t d¡B}d%dd} tjdd dd d }|
|
|¡ t |d ddd gd dgd } t || ¡ tjdd dd d }|
|
|¡ t |d ddd gd d } t || ¡ tjdd d d d }|
|
|¡ t |d ddd gd ddgd } t || ¡ | }|j
|
|dd t |d dgd } t || ¡ | d}|j
|
|dd t |d dgd } tj| jj sît|jj | j_ t || ¡ | }|
|
|¡ t |d dgd gd } t || ¡ | d}|
|
|¡ t |d dgd gd } t || ¡ | d}|
|
|¡ d D ]B} d!| d"}tjt|d t |tt| d d W 5 Q R X q~d#}tjt |d |j
|
|dd$gd W 5 Q R X W 5 Q R X t d¡V}| d d
|
|¡ | |¡}
|
| d d
}g |_ t |
|
j|j¡ t|
|
d ksXtW 5 Q R X d S )&Nr' r% éÿÿÿÿÚfirstÚsecond©ÚnamesZ__tmp_to_csv_multiindex__Fr r r r r r# r" rD ÚtimeÚfoor! zCould not infer formatrH TrX r c S s< | dkrddg} t tjjddddtjdd d
|
g| d d d S )NTrá râ r r© )r$ r$ )Úsize)Úbahræ )rè Úbar)ZbanÚbazrã r8 ©r r& )r r, rJ r r rL rã r r r Ú _make_frame s ÿûz>TestDataFrameToCSV.test_to_csv_multiindex.<locals>._make_framer r$ rµ rÇ )r r )rÓ rÕ zlen of z, but only 5 lines in filez%cannot specify cols with a MultiIndexré )N) r r, r7 r2 r8 Zreshaper r r* r+ r/ r r1 rä r3 Zassert_produces_warningÚ UserWarningr r4 r5 r{ ÚcomZall_nonerN rO r r r6 Ú TypeErrorZassert_index_equal)r r9 r: ÚframeZ old_indexZarraysZ new_indexr rQ Ztsframer; rì rR re rU Úexpr r r Útest_to_csv_multiindexÛ s
|
ÿ
|
|
|
|
|
|
|
|
$&
|
z)TestDataFrameToCSV.test_to_csv_multiindexc C sv t tdtddt dd¡d}t d¡@}| |¡ | j|dd}| ¡ }|j
|
t ¡|_
|
t ||¡ W 5 Q R X d S )NÚabcr$ ©r r r r z!__tmp_to_csv_interval_index__.csvrX )r r r6 r_ Zinterval_ranger* r+ r/ r rM r Úastyper r1 )r rQ r rR rº r r r Útest_to_csv_interval_index` s "
|
z-TestDataFrameToCSV.test_to_csv_interval_indexc
|
C s t tj dd¡ tj¡}tj|d< t d¡J}|j |dd t
|
|(}| ¡ }|d d¡d dksjt W 5 Q R X W 5 Q R X d S ) Nr# rµ z!__tmp_to_csv_float32_nanrep__.csviç )Úna_repú,r' Z999)r r, rJ rK rõ Úfloat32r- r* r+ r/ ÚopenÚ readlinesÚsplitr3 )r rQ r r Úlinesr r r Útest_to_csv_float32_nanrepn s
|
|
z-TestDataFrameToCSV.test_to_csv_float32_nanrepc C sT t dddgdddgd}t d¡&}| |¡ | |¡}t ||¡ W 5 Q R X d S ) Nr# r' r$ z5,6z7,8z9,0rô z __tmp_to_csv_withcommas__.csv©r r* r+ r/ r r1 ©r rQ r ra r r r Útest_to_csv_withcommasy s
|
|
|
z)TestDataFrameToCSV.test_to_csv_withcommasc C s> dd }t tj dd¡d|dd}t tj dd¡ d¡d|d d}t d
|
|j|d d }t d |j|dd }t td|j|dd }tj|jdd
dd
f< t j
|
|||||gdd}i }dtj fd tj fd tj fdtffD ] \} }
|
|| D ] } |
|
|| < qìqÜt ¡ 0} | | ¡ t| d||dd} t | |¡ W 5 Q R X d S )Nc s fddt dD S )Nc s g | ]} |d qS )Z03dr rd ri r r r^ s zMTestDataFrameToCSV.test_to_csv_mixed.<locals>.create_cols.<locals>.<listcomp>r )r6 ri r ri r Ú create_cols s z9TestDataFrameToCSV.test_to_csv_mixed.<locals>.create_colsr r Úfloat64Úfloat)r& r r8 rZ TÚboolrC ræ ÚobjectÚ20010101Údateé é2 r# r$ )Úaxisr )r r& r )r r, rJ rK rõ r r r- r. r_ r` r r8 Zbool_r r* r+ r/ r r1 )r r Údf_floatÚdf_intÚdf_boolÚ df_objectÚdf_dtrQ Zdtypesr r& rq ÚfilenamerS r r r Útest_to_csv_mixed sR ÿý ÿ ÿü
|
|
ÿz$TestDataFrameToCSV.test_to_csv_mixedc C s t tj dd¡ttdttd dd}t ¡ 0}| |¡ t |dd}|j
|
|_
|
t ||¡ W 5 Q R X t tj dd¡dd }t tj dd¡ d
|
¡}t d |j tdd }t d |j tdd }t td|j tdd }tj|||||gdd d}dddgd |_
|
t ¡ L}| |¡ t |dd}dD ]} t|| || < q&|j
|
|_
|
t ||¡ W 5 Q R X d}
|
t |
|
d¡}dddg|_
|
t ¡ 8}| |¡ t |dd}|jddid}t ||¡ W 5 Q R X d S )Nr r é r rë r rX r$ r% r8 TrC ræ r r# )r Z ignore_indexr' r )z0.4z1.4z2.4r© r> r? za.1r )r r, rJ rK r r6 r* r+ r/ r r r1 rõ r r r_ r` r r{ Úrename) r rQ r rR r r r r r re r| r r r Útest_to_csv_dups_cols° sH ý
|
|
ÿ
|
|
|
|
z(TestDataFrameToCSV.test_to_csv_dups_colsrz i' iPÃ é c C sx t dtdi}|jd |d< |jd |d< |jd |d< t ¡ ,}|j||d t|d
|
d }t ||¡ W 5 Q R X d S ) Nr r ç ð?r g @ru g @ÚDr r rX )r r6 r r* r+ r/ r r1 )r rz rÚ r rS r r r Útest_to_csv_chunkingá s
|
z'TestDataFrameToCSV.test_to_csv_chunkingc C sX t tj dd¡d d d}t ¡ .}|j|ddd t|d d}t ||¡ W 5 Q R X d S )Nr# iª ©r r F)r r r ) r r, rJ rK r* r+ r/ r r1 )r rQ r rS r r r Ú!test_to_csv_wide_frame_formattingí s
|
|
z4TestDataFrameToCSV.test_to_csv_wide_frame_formattingc C sj t d}| j|d d}td||jd i}t ¡ ,}| |¡ t|dd}tj||dd W 5 Q R X d S )Nz a,1.0
|
b,2.0r Útr rX FrD )r r r r r* r+ r/ r1 )r Úf1rQ Znewdfr r; r r r Útest_to_csv_bugö s
|
|
z"TestDataFrameToCSV.test_to_csv_bugc C s| t ddddgi}t ¡ X}|j|dd t|ddd}t ||¡ |j|dd d
|
t|d dd}t ||¡ W 5 Q R X d S ) Nu c/Ïr# r' r$ úUTF-8©r r ©r r F©r r rÿ r r r r Útest_to_csv_unicode s
|
z&TestDataFrameToCSV.test_to_csv_unicodec C sl t d}tddddgdddd ggdd
|
d d gdd
|
gd }|j|dd | d¡ t|ddd}t ||¡ d S )NÚ u ×Zd2Zd3Zd4Za1Za2Úa3Za4u ×u ×u ×r r r r r! )r r r/ Úseekr r* r1 )r ÚbufrQ ra r r r Ú test_to_csv_unicode_index_col s
|
ý
|
z0TestDataFrameToCSV.test_to_csv_unicode_index_colc C s6 t }| |¡ | d¡ t|dd}t ||¡ d S )Nr rX )r r/ r& r r* r1 )r r9 r' r; r r r Útest_to_csv_stringio s
|
|
|
z'TestDataFrameToCSV.test_to_csv_stringioc C s t dddgdddggddgd d
|
d gd }t ¡ T}|j|d d t|dd}t dddgdddggddgd d
|
d gd }t ||¡ W 5 Q R X d S )N翶~úÏ¿?ç9DÜJÎ?ç >éD)â?çÜ.4×i¤(@ç33333þ@çÍÌÌ̤Ar r rE rF rG rC z%.2f)Ú float_formatr rX g¸
ëQ¸¾?gq=
|
×£pÍ?g=
|
×£p=â?g¤p=
|
×£(@rÿ )r rQ r rS rT r r r Útest_to_csv_float_format" s ý
|
ýz+TestDataFrameToCSV.test_to_csv_float_formatc C sH t dddgi}|jddd dd}dd d
|
g}t |¡}||ksDtd S ) Nr> rÊ r rø c S s t j| ddS )Nú-)Ztrim)r, Zformat_float_positionalr r r r rt 9 rÌ zJTestDataFrameToCSV.test_to_csv_float_format_over_decimal.<locals>.<lambda>F)Údecimalr0 r z0.5Ú1)r r/ r* Ú convert_rows_list_to_csv_strr3 ©r rQ rR Ú expected_rowsrº r r r Ú%test_to_csv_float_format_over_decimal4 s ý
|
|
z8TestDataFrameToCSV.test_to_csv_float_format_over_decimalc C sb t dddgdddgd}t }|j|dtjd d
|
| ¡ }d d d dg}t |¡}||ks^td S )Nr# r' r$ ræ ré rê rô Fúutf-8)r Úquotingr z"A","B"z1,"foo"z2,"bar"z3,"baz") r r r/ ÚcsvÚQUOTE_NONNUMERICÚgetvaluer* r5 r3 )r rQ r' rR r7 rº r r r Ú!test_to_csv_unicodewriter_quoting@ s
|
z4TestDataFrameToCSV.test_to_csv_unicodewriter_quotingr r9 c C sV t dddgi}t }|j|tj|dd | ¡ }dddg}t |¡}||ksRtd S )Nr Zhelloz {"hello"}F)r: r r ) r r r/ r; Ú
|
QUOTE_NONEr= r* r5 r3 )r r rQ r' rR r7 rº r r r Útest_to_csv_quote_noneK s
|
|
z)TestDataFrameToCSV.test_to_csv_quote_nonec C sb t dddgdddgddd d
|
gd }t }|j|d d ddddg}t |¡}| ¡ |ks^td S )Nr# r' r$ rµ r rÓ rô ÚoneÚtwoÚthreer Fr! zA,Bzone,1,4ztwo,2,5z three,3,6)r r r/ r* r5 r= r3 )r rQ r' r7 rº r r r Ú"test_to_csv_index_no_leading_commaW s $
|
z5TestDataFrameToCSV.test_to_csv_index_no_leading_commac
|
C s8 t dddgdddgddd d
|
gd }t ¡ @}|j|d d d}t|dd}| ¡ |ks^tW 5 Q R X W 5 Q R X t ¡ @}|j|dd d}t|dd}| ¡ |ks¬tW 5 Q R X W 5 Q R X t ¡ f}| |¡ tj d¡}d| d | d | d | }t|dd}| ¡ |ks tW 5 Q R X W 5 Q R X d S )Nr# r' r$ rµ r rÓ rô rA rB rC r ú
|
©Úlineterminators# ,A,B
|
one,1,4
|
two,2,5
|
three,3,6
|
Úrb)rW Ú
|
s ,A,B
|
one,1,4
|
two,2,5
|
three,3,6
|
r9 s ,A,Bs one,1,4s two,2,5s three,3,6)
|
r r* r+ r/ rú Úreadr3 ÚosÚlinesepÚencode)r rQ r rº r Z
|
os_linesepr r r Útest_to_csv_lineterminatorsa s@ $
|
$
|
$
|
|
ÿþýüûúùÿ z.TestDataFrameToCSV.test_to_csv_lineterminatorsc C s t t ddddddddg¡}t ddddddddg}t }|j|dd t }|j|dd | ¡ | ¡ ksrttd|i}td|i}t }| |¡ t }| |¡ | ¡ | ¡ ks¾td S )Nr> r? rq Fr r )r r_ Z Categoricalr r/ r= r3 r )r r r) Úresrñ rQ ra r r r Ú test_to_csv_from_csv_categorical s
|
|
z3TestDataFrameToCSV.test_to_csv_from_csv_categoricalc C s: |j d d}t|tsttt|dd}t ||¡ d S )N)Z path_or_bufr rX )r/ r r r3 r r r* r1 )r r9 Zcsv_strr; r r r Útest_to_csv_path_is_none s z+TestDataFrameToCSV.test_to_csv_path_is_nonez df,encodingr* r+ r, r- r. r/ r r rE rF rG rC ró ÚdefZghir Úasciir é{ u ä½ å¥½u ä¸çÚgb2312u Îειά ÏοÏ
u
|
ÎÏÏμεZcp737c
|
|
C s t ¡ ü}|j|||d t||d|d}t ||¡ t|d||d"}|j|j|d |jjrdtW 5 Q R X t|||dd d¡}t ||¡ t
|
||¡0}| ¡ |p¬d¡}|j D ]} | |ks¶tq¶W 5 Q R X t
|
||¡ }t |t|d|d ¡ W 5 Q R X W 5 Q R X d S )
|
N)Ú compressionr r )rV r r Úwr )rV r r r r r! )r* r+ r/ r r1 r ÚhandleÚclosedr3 ZsqueezeÚdecompress_filerJ r r )
|
r rQ r rV r rR ZhandlesÚfhÚtextÚcolr r r Útest_to_csv_compression¨ sB
|
ÿ ÿüû
|
z*TestDataFrameToCSV.test_to_csv_compressionc
|
C sf t d¡P}|j}t|| d¡d|d}|j|dd t|dd}| d d
|
¡}|j d d
|
¡|_t ||¡ |j|d d t|dd}| d d
|
¡}|j dd
|
¡|_t ||¡ |j
|
}|j|dd t|dd}| dd
|
¡}|j dd
|
¡|_ t ||¡ t dgd dddg }td|i|d} | j|d d t|ddgdd}t || ¡ W 5 Q R X d S )NZ__tmp_to_csv_date_format__r# rô r ú%Y%m%d©Ú date_formatr rX c S s t | d¡S ©Nr_ ©rZ Ústrftimer r r r rt é rÌ z<TestDataFrameToCSV.test_to_csv_date_format.<locals>.<lambda>c S s t | d¡S rb rc r r r r rt ì rÌ ú%Y-%m-%dc S s
|
| d¡S ©Nre ©rd r r r r rt ö rÌ c S s
|
| d¡S rf rg r r r r rt ù rÌ c S s t | d¡S rb rc r r r r rt rÌ c S s
|
| d¡S rb rg r r r r rt rÌ r r© ú
|
2000-01-01r )r r ) r* r+ r r Úshiftr/ r Zapplymapr¡ r1 ÚTr r )
|
r r: r rh rÝ Zdatetime_frame_intZdatetime_frame_strZdatetime_frame_columnsZ nat_indexZ nat_framer r r Útest_to_csv_date_formatÝ sR ÿ ÿÿ ÿÿ ÿÿ ÿz*TestDataFrameToCSV.test_to_csv_date_formatÚtdr Z10sc C s t d¡}tdddddd}|| }| d ¡}tjtt|dd }td
|
|i|d }|j |d d t
|
|d d}t |j d d d¡|_ t ||¡ W 5 Q R X d S )NÚcsv_date_format_with_dstz2013-10-26 23:00z2013-10-27 01:00z Europe/LondonrÍ Zinfer)rp r
Z ambiguousr8 r% r r Tr rX ©Úutc)r* r+ r
|
r0 r, r r6 r2 r r/ r r r ro r1 )r rl r Útimesre Z
|
time_rangerQ rR r r r Ú test_to_csv_with_dst_transitions s$ û
|
ÿz3TestDataFrameToCSV.test_to_csv_with_dst_transitionsc C sä t ddddd}| d ¡}d |j_td|d|d}t d ¡\}|j|d
|
d t|d d }t |j
|
d
|
d d¡|_
|
t |d d
|
d d¡|d< t ||¡ W 5 Q R X | t¡ t d ¡&}| |¡ t |¡}t ||¡ W 5 Q R X d S )Nz
|
2015-01-01z
|
2015-12-31rÍ z Europe/Paris)r
rp r# )r5 Úidxr rm Tr rX rn rr z datetime64[ns, Europe/Paris])r
|
r0 Ú_dataZ_freqr r* r+ r/ r r r ro rõ r1 r Z to_pickler_ Z read_pickle)r rr rQ r rR r r r Ú,test_to_csv_with_dst_transitions_with_pickle1 s&
|
ÿÿ
|
|
|
z?TestDataFrameToCSV.test_to_csv_with_dst_transitions_with_picklec C sL t ddgddgdtjgddgd}d d
|
d g}t |¡}| ¡ }||ksLt|jd d }||ksdtd d
|
d g}t |¡}|jtjd }||kstd ddg}t |¡}|jtj d }||ksÀtd ddg}t |¡}|jtj
|
d }||ksîtd}t j tj |d |jtjd W 5 Q R X t j tj |d |jtjd d W 5 Q R X d d
|
dg}t |¡}|jtjdd}||ksztdd
|
dg}t |¡}|jtjdd}||ks¬tddg}t |¡}tt|}t }|j|d dd | ¡ |ksòtt d d gd d!gd"d#gd$}| dd%g¡}d&d'd(g}t |¡}|jtj d |ksHtd S ))NTFr g @é* r> zb,c)Úc_boolÚc_floatÚc_intZc_stringz ,c_bool,c_float,c_int,c_stringz0,True,1.0,42.0,az1,False,3.2,,"b,c")r: z("","c_bool","c_float","c_int","c_string"z"0","True","1.0","42.0","a"z"1","False","3.2","","b,c"z0,True,1.0,42.0,"a"z1,False,3.2,"","b,c"z%need to escape, but no escapechar setrH )r: Ú
|
escapecharz1,False,3.2,,b!,cú!z ,c_bool,c_ffloat,c_int,c_stringz1,False,3.2,,bf,cr za,b,cz 1,"test
|
",3r9 r" r# r' r$ rµ r rÓ )r> r? rq r? z "a","b","c"z "1","3","5"z "2","4","6")r r, r- r* r5 r/ r3 r; Ú QUOTE_MINIMALÚ QUOTE_ALLr< rN rO ÚErrorr? r r r= Z set_index) r rQ r7 rº rR rU Z text_rowsr\ r' r r r Útest_to_csv_quotingJ s üÿ
|
ý
|
ý
|
ý
|
ý
|
ý
|
ý
|
|
|
|
z&TestDataFrameToCSV.test_to_csv_quotingc C sÞ dddg}t j|dd}tdddg|d }| ¡ }d
|
d d d g}t |¡}||ksTtd}|j|d}d
|
dddg}t |¡}||kstdtdg}t j|dd}tdddg|d }| ¡ }d
|
d dd g}t |¡}||ksÚtd S )Nz
|
1990-01-01rh z
|
3005-01-01r )r
rµ r rÓ r z,0z 1990-01-01,4z 2000-01-01,5z 3005-01-01,6z%m-%d-%Yr` z 01-01-1990,4z 01-01-2000,5z 01-01-3005,6z,5)r_ Z PeriodIndexr r/ r* r5 r3 r )r Údatesr rQ rR r7 rº ra r r r Ú test_period_index_date_overflow¬ s&
|
|
|
|
|
z2TestDataFrameToCSV.test_period_index_date_overflowc C sp t ddddg¡}tddddgd d
|
d d gg}||_d dddg}|j|d}dddg}t |¡}||ksltd S )N)r> r# )r> r' )r? r# )r? r' r# r' r$ rµ r rÓ rÕ rÖ r> r? rq Údr z,a,b,c,dz 0,1,2,3,4z 1,5,6,7,8)r rL r r r/ r* r5 r3 )r r rQ r rR r7 rº r r r Útest_multi_index_headerË s
|
|
z*TestDataFrameToCSV.test_multi_index_headerc C sR t dddg}tdddgg|d}|jddgd}d}|jd d
|
}t ||¡ d S ) N)r# )r' )r$ r# r' r$ r z ,1,3
|
0,1,3
|
rI rF )r r Zreindexr/ r* r4 )r r rQ rº rR r r r Ú$test_to_csv_single_level_multi_indexØ s z7TestDataFrameToCSV.test_to_csv_single_level_multi_indexc
|
C s t dddgi}dddg}t |¡}t d¡<}|j|dd tj|d d
|
}| ¡ d ¡}W 5 Q R X W 5 Q R X ||ks|td S ) Nr> r# r' r4 Ú2z__test_gz_lineend.csv.gzFr Úgzip)rV r9 ) r r* r5 r+ r/ rZ rJ r r3 )r rQ r7 rº r r rR r r r Útest_gz_lineendá s
|
|
"z"TestDataFrameToCSV.test_gz_lineendc C s< t dtdddi}t }| |¡ | ¡ }d|ks8td S )Nr> z1/1/2000r© )r rh )r r
|
r r/ r= r3 )r rð r' rR r r r Útest_to_csv_numpy_16_bugí s
|
|
z+TestDataFrameToCSV.test_to_csv_numpy_16_bugc C s6 t d d gjd dddd dd¡}d}||ks2td S )NFr$ )r r r÷ rE rI z""
|
""
|
)r r/ Úreplacer3 )r rR rº r r r Útest_to_csv_na_quotingö s ÿ þÿz)TestDataFrameToCSV.test_to_csv_na_quotingc C sd t ddtjgd}|d d¡|d< |d d¡|d< | ¡ }ddd g}t |¡}||ks`td S )
|
Nr@ r# r r? ZInt16Úcategoryz,a,bz0,x,1z1,x,)r r_ ZNArõ r/ r* r5 r3 r6 r r r Ú test_to_csv_categorical_and_ea s
|
|
z1TestDataFrameToCSV.test_to_csv_categorical_and_eac C s^ t dtjtdtdddgi}|d d¡|d< | ¡ }ddg}t |¡}||ksZtd S ) Nr> z
|
2020-01-01z
|
2020-01-02Zboth)rY r z,az 0,"[2020-01-01, 2020-01-02]") r r_ ZIntervalr rõ r/ r* r5 r3 r6 r r r Ú$test_to_csv_categorical_and_interval s ýÿÿÿ
|
z7TestDataFrameToCSV.test_to_csv_categorical_and_interval)NNNNF)AÚ__name__Ú
|
__module__Ú __qualname__r r= rV rb rm rw r rN ÚmarkZ parametrizer Zslowr r§ r» r½ rÀ r rà rÅ rÈ rÏ rÒ rÙ rÞ rß rò rö rþ r r r r r r r# r( r) r1 r8 r> r@ rD rN rP rQ r r^ rk r_ rk rq rt r~ r r r r r r r r r r r r r sò &
|
!
|
ø
|
i ÿ ÿ
|
ÿ ÿ
|
|
ÿýþ .1
|
|
|
|
& ýú þôþ
|
!<
|
b
|
r ) r; Úior rK Únumpyr, rN Z pandas.errorsr Zpandasr_ r r r r r r r
|
r r Zpandas._testingZ_testingr* Zpandas.core.commonÚcoreÚcommonrî Zpandas.io.commonr r r r r r Ú<module> s ,
|
