U
|
ý°döÀ ã @ s4 d dl mZm Z mZ d dlmZ d dlmZ d dlZd dlZd dl Z
|
d dl Z d dl m mZ d dlZd dlmZmZmZmZ d dlmZ d dlmZmZm Z m Z m Z d dl m Z e j!dd Z"e j!d
|
d Z#e j$ %d e j&d e 'd¡e 'd¡gde j&de 'd¡e 'd¡gde j&d e 'd¡e 'd¡gde j&de 'd¡dg¡G dd dZ(e j$ %de j&dd e 'd¡e 'd¡gde j&dde 'd¡e 'd¡gde j&dd e 'd¡e 'd¡gde j&dde 'd¡dg¡e j$ )d ¡G dd dZ*G dd dZ+e 'd¡e 'd¡G d d d Z,e j$ %d e -¡ ¡d d Z.dS )!é )ÚdateÚdatetimeÚ timedelta)Úpartial)ÚBytesION)Ú DataFrameÚIndexÚ
|
MultiIndexÚoption_context)Ú ExcelFileÚ ExcelWriterÚ_OpenpyxlWriterÚ _XlsxWriterÚregister_writer)Ú_writersc c s t | ¡ }|V W 5 Q R X dS )z9
|
Fixture to open file for use in each test case.
|
N)ÚtmÚ ensure_clean)ÚextÚ file_path© r úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/io/excel/test_writers.pyÚpath# s r c c s2 d| d¡ d}t|| dV W 5 Q R X dS )aM
|
Fixture to set engine for use in each test case.
|
|
Rather than requiring `engine=...` to be provided explicitly as an
|
argument in each test, this fixture sets a global option to dictate
|
which engine should be used to write Excel files. After executing
|
the test it rolls back said change to the global option.
|
z io.excel.Ú.z.writerN)Ústripr
|
)Úenginer Z option_namer r r Ú
|
set_engine, s
|
r r ú.xlsxÚopenpyxlZxlrd©Zmarksz.xlsmÚ
|
xlsxwriterz.odsZodfc @ s e Zd Zej ddeejgd fdedejgd ifg¡dd Z ej ddedgejgd fdeejgd fg¡d d
|
Z
|
d d Z d d Z dd Z ej dddg¡ej dddg¡ej dddg¡ej dddg¡dd Zdd Zd d ZdS ) Ú TestRoundTripzheader,expectedNé r z
|
Unnamed: 0é c C sx d}t dddgdddgdddgdd d
|
gg}t |¡,}|j||d d d tj||d g|d}W 5 Q R X t ||¡ d S )NZ no_headerÚ é éd é éÈ r" é, r! é F©ÚindexÚheaderr ©Ú
|
sheet_nameZusecolsr, ©r r r Úto_excelÚpdÚ
|
read_excelÚassert_frame_equal©Úselfr r, ÚexpectedÚfilenameÚdfr Úresultr r r Ú!test_read_one_empty_col_no_headerG s ( ÿz/TestRoundTrip.test_read_one_empty_col_no_headerc C sx d}t dddgdddgdddgdd d
|
gg}t |¡,}|j|dd d d tj||dg|d}W 5 Q R X t ||¡ d S )NZ with_headerr# r$ r% r& r' r" r( r! r) FTr* r r- r/ r4 r r r Ú#test_read_one_empty_col_with_headerX s ( ÿz1TestRoundTrip.test_read_one_empty_col_with_headerc
|
C sÔ t ddgddgddggddgd }t |¡}t|(}|j|d
|
d d d |j|d d d W 5 Q R X ddg|_t|2}tj|d
|
d ddgd}tj|d d ddgd}W 5 Q R X t ||¡ t ||¡ W 5 Q R X d S )Nr$ Úfoor& Úbarr" ZbazÚaÚb©ÚcolumnsZ Data_no_headF)r, r+ ZData_with_head©r+ ÚAÚB)r. r, Únames)r. Ú index_colrE )
|
r r r r r0 rA r r1 r2 r3 )r5 r ZrefdfÚpthÚwriterÚreaderZ xlsdf_no_headZxlsdf_with_headr r r Ú"test_set_column_names_in_parameterh s* "
|
|
|
ÿü z0TestRoundTrip.test_set_column_names_in_parameterc
|
|
s¬ dd dddg} fdd|D }t t||}t |¡f}t|$}| ¡ D ]\}}| ||¡ qPW 5 Q R X tj||dd }|D ]} t || || ¡ qW 5 Q R X d S )
|
Nc S s& dddgdddg }}t ||| gdS )Né é é! r$ r& r" r@ )r )Zcol_sheet_nameÚdÚir r r Útdf s zDTestRoundTrip.test_creating_and_reading_multiple_sheets.<locals>.tdfZAAAZBBBZCCCc s g | ] } |qS r r )Ú.0Ús©rP r r Ú
|
<listcomp> s zKTestRoundTrip.test_creating_and_reading_multiple_sheets.<locals>.<listcomp>r ©r. rF )
|
ÚdictÚzipr r r Úitemsr0 r1 r2 r3 )
|
r5 r ÚsheetsÚdfsrG ÚewZ sheetnamer8 Z dfs_returnedrR r rS r Ú)test_creating_and_reading_multiple_sheets s
|
|
z7TestRoundTrip.test_creating_and_reading_multiple_sheetsc C sö t |¡â}tddiddiddiddid}tddiddiddiddid}| |¡ tj|ddgdd}t ||¡ tddiddiddiddid}tddiddiddiddid }| |¡ tj|ddgdd}t ||¡ W 5 Q R X d S )
|
Nr r$ r" é )©ZOneÚx©ÚTwoÚX©ra ÚY)ÚZeror# )r^ r` rc )re zUnnamed: 4_level_1©r, rF ))ÚBegr# ©ZMiddler_ ©ÚTailrb ©rj rd ))rg zUnnamed: 1_level_1rh ri rk ©r r r r0 r1 r2 r3 )r5 r r r8 r6 Úactualr r r Ú&test_read_excel_multiindex_empty_level sF üÿ üÿ
|
üÿ üÿ
|
z4TestRoundTrip.test_read_excel_multiindex_empty_levelÚ c_idx_namesTFÚ r_idx_namesÚ c_idx_levelsr$ Ú r_idx_levelsc C sD t |¡.}|dkr@|r@|dkr&|r@tjjdd}|j |¡ |pJ|dk} t dd||||¡}
|
|
|
|¡ t j
|
|t t |t t |d} t j |
|
| | d tj|
|
jdd d
f< |
|
|¡ t j
|
|t t |t t |d} t j |
|
| | d tj|
|
jd d d
f< |
|
|¡ t j
|
|t t |t t |d} t j |
|
| | d W 5 Q R X d S )
|
Nr$ r" z?Column index name cannot be serialized unless it's a MultiIndex)Úreasoné )rF r, )Ú check_namesr éÿÿÿÿ)r r ÚpytestÚmarkZxfailÚnodeZ
|
add_markerÚmakeCustomDataframer0 r1 r2 ÚlistÚranger3 ÚnpÚnanÚiloc) r5 r ro rp rq rr ÚrequestrG rx ru r8 Zactr r r Ú test_excel_multindex_roundtripÉ sT ÿÿÿ ÿ
|
|
|
ý
|
|
|
ý
|
|
|
ýz,TestRoundTrip.test_excel_multindex_roundtripc
|
C sø t dddgtjdddd}| ¡ }|d j d¡|d< t |¡ª}| |¡ tj |d d
|
}t
|
||¡ tj |dgd d }t
|
||¡ d d }tj t dd tj |dg|d d}W 5 Q R X t
|
||¡ tj |dgdd d}t
|
||¡ W 5 Q R X d S )Nr$ r& r" z
|
2012-01-01)Úperiods)ÚcolÚ date_stringsr ú%m/%d/%Yr ©rF )Ú parse_datesrF c S s t | d¡S )Nr
)r Ústrptime©r_ r r r Ú<lambda> ó z;TestRoundTrip.test_read_excel_parse_dates.<locals>.<lambda>zuse 'date_format' instead©Úmatch)r Ú date_parserrF )r Ú date_formatrF ) r r1 Z
|
date_rangeÚcopyÚdtÚstrftimer r r0 r2 r3 Zassert_produces_warningÚ FutureWarning)r5 r r8 Zdf2rG Úresr r r r Útest_read_excel_parse_datesþ s< ÿ
|
ÿü ÿz)TestRoundTrip.test_read_excel_parse_datesc C s t tdtjt d¡dddg¡}ttd|d}t |¡"}| |¡ tj
|
|ddgd}W 5 Q R X ttdt tdd d
|
d d gg¡}t ||¡ d S ) Nr! z
|
2020-01-01Z6M)Ústartr ÚfreqrB r r$ r z(2020-01-31, 2020-07-31]z(2020-07-31, 2021-01-31]z(2021-01-31, 2021-07-31]z(2021-07-31, 2022-01-31]) r Ú from_arraysr| r1 Zinterval_rangeÚ Timestampr r r r0 r2 r3 )r5 r Zmidxr8 rG r9 r6 r r r Ú"test_multiindex_interval_datetimes s4 ÿþÿ
|
üþÿþz0TestRoundTrip.test_multiindex_interval_datetimes)Ú__name__Ú
|
__module__Ú __qualname__rw rx Ú parametrizer r} r~ r: r; rJ r\ rn r r r r r r r r ; s( *þ
|
,þ
|
/1!r z
|
engine,extc @ sJ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z e
|
j de je je je jg¡dd Ze
|
j de je je jg¡dd Zdd Zdd Zdd Zdd Zd d Zd d Z d d! Z d"d# Z d$d% Z d&d' Z d(d) Z!d*d+ Z"d,d- Z#d.d/ Z$d0d1 Z%d2d3 Z&d4d5 Z'd6d7 Z(d8d9 Z)d:d; Z*d<d= Z+e
|
j d>d?d@g¡e
|
j dAdBdCdDg¡e
|
j dEdBdCdDg¡dFdG Z,dHdI Z-dJdK Z.dLdM Z/e
|
j dNdOdPg¡dQdR Z0dSdT Z1dUdV Z2dWdX Z3dYdZ Z4d[d\ Z5d]d^ Z6d_d` Z7dadb Z8dcdd Z9dedf Z:dgdh Z;didj Z<dkdl Z=e
|
j dmdne>g¡dodp Z?dqdr Z@dsdt ZAdudv ZBdwdx ZCdnS )yÚTestExcelWriterc C s d}d}t j|dfd}t jd|fd}t|}t|}d}tjt|d | |¡ W 5 Q R X tjt|d | |¡ W 5 Q R X d S )Ni i@ r$ )Úshapezsheet is too larger )r} Úzerosr rw ÚraisesÚ
|
ValueErrorr0 ) r5 r Zbreaking_row_countZbreaking_col_countZrow_arrZcol_arrZrow_dfZcol_dfÚmsgr r r Útest_excel_sheet_sizeU s z%TestExcelWriter.test_excel_sheet_sizec C sz t tj dd¡}| |¡ t|}tj|ddd}W 5 Q R X t ||¡ d}t
|
j t |d t |d¡ W 5 Q R X d S )Né
|
r& r rU z Worksheet named '0' not foundr Ú0) r r} ÚrandomÚrandnr0 r r1 r2 r r3 rw r¢ r£ )r5 r ÚgtÚxlr8 r¤ r r r Ú test_excel_sheet_by_name_raisef s
|
|
z.TestExcelWriter.test_excel_sheet_by_name_raisec C s t |8}| |d¡ | ¡ }|jd d d
|_| |d¡ W 5 Q R X t|>}tj|ddd}tj|ddd}t ||¡ t ||¡ W 5 Q R X d S )NZData1rv ZData2r rU ) r r0 r rA r r1 r2 r r3 )r5 Úframer rH Úframe2rI Zfound_dfZ found_df2r r r Ú!test_excel_writer_context_managers s
|
|
z1TestExcelWriter.test_excel_writer_context_managerc C sÒ | ¡ }tj|jd d
|j d¡f< | |d¡ |j|dddgd |j|ddd |j|ddd | |d¡ tj|dd d
|
}t
|
||¡ |j|ddd tj|dd d
|
}|j |_ t
|
||¡ |j|dd d tj|dd d gd }t
|
||¡ |j|ddd tj|dd dgd }t
|
||¡ |j|ddd tj|dd ddgd }t
|
||¡ | |d¡ tj|d d}t
|
||¡ | |d¡ tj|d d}t
|
||¡ |d }| |¡ tj|d d}t
|
| ¡ |¡ d S )Nrt rC Útest1rD r@ F©r, rB r rU ZNA)Zna_rep)r. rF Z na_valuesZ88éX g V@ÚSheet1r r§ ) r r} r~ r rA Úget_locr0 r1 r2 r r3 r+ Zto_frame)r5 r r ÚreconsrR r r r Útest_roundtrip sJ ÿ
|
z TestExcelWriter.test_roundtripc C sP | ¡ }d|d< | |d¡ t|}tj|ddd}W 5 Q R X t ||¡ d S )Nr= r< r° r rU )r r0 r r1 r2 r r3 )r5 r r Z mixed_framerI rµ r r r Ú
|
test_mixed² s
|
zTestExcelWriter.test_mixedc C s` |}t jt |j¡d d}||_| |d¡ t|}t j|ddd}W 5 Q R X t ||¡ d S )N©r r° r rU )
|
r1 Ú DatetimeIndexr} Úasarrayr+ r0 r r2 r r3 )r5 Útsframer r8 r+ rI rµ r r r Ú test_ts_frame» s
|
z TestExcelWriter.test_ts_framec C sh | ¡ }tj|jd d
|j d¡f< | |d¡ |j|dddgd |j|ddd |j|ddd d S ) Nrt rC r° rD r@ Fr± rB )r r} r~ r rA r´ r0 )r5 r r r r r Útest_basics_with_nanÇ s z$TestExcelWriter.test_basics_with_nanÚnp_typec C s t tjjdddd|d}| |d¡ t|}tj|ddd}W 5 Q R X | tj ¡}t
|
||¡ tj|ddd}t
|
||¡ d S ) Néöÿÿÿr¦ )r¦ r& ©Úsize©Údtyper° r rU ) r r} r¨ Úrandintr0 r r1 r2 ÚastypeÚint64r r3 )r5 r¾ r r8 rI rµ Z int_frameZrecons2r r r Útest_int_typesÏ s
|
z TestExcelWriter.test_int_typesc C sZ t tj d¡|d}| |d¡ t| }tj|ddd |¡}W 5 Q R X t
|
||¡ d S )Nr¦ r r° r rU ) r r} r¨ Z random_sampler0 r r1 r2 rÅ r r3 )r5 r¾ r r8 rI rµ r r r Útest_float_typesß s
|
ÿz TestExcelWriter.test_float_typesc C s^ t ddddgtjd}| |d¡ t| }tj|ddd tj¡}W 5 Q R X t ||¡ d S )Nr$ r TFr r° rU )
|
r r} Zbool_r0 r r1 r2 rÅ r r3 ©r5 r r8 rI rµ r r r Útest_bool_typesì s
|
ÿz TestExcelWriter.test_bool_typesc C s\ t dtjfddtj fg}| |d¡ t|}tj|ddd}W 5 Q R X t ||¡ d S )Nr$ )r& r" rt r° r rU ) r r} Úinfr0 r r1 r2 r r3 rÉ r r r Útest_inf_roundtripø s
|
|
z"TestExcelWriter.test_inf_roundtripc C s8 t jt |j¡d d}||_| ¡ }tj|jd d
|j d¡f< |
|
|d¡ |j
|
|dddgd |j
|
|ddd |j
|
|ddd t | }|
|
|d¡ |
|
|d
|
¡ W 5 Q R X t |>}t j |dd d }t ||¡ t j |d
|
d d }t ||¡ W 5 Q R X d t|jks td|jd ks td
|
|jd ks4td S )Nr¸ rt rC r° rD r@ Fr± rB Ztest2r rU r& r$ )r1 r¹ r} rº r+ r r~ r rA r´ r0 r r r2 r r3 ÚlenÚ sheet_namesÚAssertionError)r5 r r» r r+ rH rI rµ r r r Ú test_sheets s&
|
|
zTestExcelWriter.test_sheetsc C sÆ | ¡ }tj|jd d
|j d¡f< | |d¡ |j|dddgd |j|ddd |j|ddd td d
|
d d g}|j|d|d t|}t j
|
|dd d}W 5 Q R X | ¡ }||_t ||¡ d S )Nrt rC r° rD r@ Fr± rB ZAArb rd ÚZr rU ) r r} r~ r rA r´ r0 r r r1 r2 r r3 )r5 r r Z col_aliasesrI ÚrsÚxpr r r Útest_colaliases s
|
z TestExcelWriter.test_colaliasesc C s, | ¡ }tj|jd d
|j d¡f< | |d¡ |j|dddgd |j|ddd |j|ddd ttj d d
|
¡d k}|j|dd g|d t
|
| }t j |dd d tj¡}W 5 Q R X d g|j_|jj|jjksØtttj d d
|
¡d k}|j|dd ddg|d t
|
| }t j |dd d tj¡}W 5 Q R X d g|j_|jj|jjksRtttj d d
|
¡d k}|j|dd |d t
|
| }t j |dd d tj¡}W 5 Q R X d g|j_t || t¡¡ |j|dddddgd|d | ¡ }| ddg¡}t
|
|}t j |dd dgd}W 5 Q R X t ||¡ d S )Nrt rC r° rD r@ Fr± rB r¦ r& r Útest)Z index_labelÚ merge_cellsrU ÚdummyZdummy2ÚCÚD)rA r+ rÖ r$ )r r} r~ r rA r´ r0 r r¨ r© r r1 r2 rÅ rÆ r+ rE rÏ r r3 ÚboolÚ set_index)r5 rÖ r r r8 rI rµ r r r Útest_roundtrip_indexlabels- s^
|
ÿ
|
ü
|
ÿ
|
|
ÿ
|
|
û
|
z*TestExcelWriter.test_roundtrip_indexlabelsc C sr t tj dd¡}d|j_|j||d t| }tj ||j
|
d dd}W 5 Q R X t ||¡ |jjdksnt d S )Nr¦ r! r< ©rÖ r rU )r r} r¨ r© r+ Únamer0 r r1 r2 rÎ r r3 rÏ )r5 rÖ r r8 Úxfr9 r r r Ú test_excel_roundtrip_indexnamef s
|
z.TestExcelWriter.test_excel_roundtrip_indexnamec C sz t jt |j¡d d}||_| ¡ }dd |jD |_|j|d|d t|}t j|ddd}W 5 Q R X t
|
||¡ d S )Nr¸ c S s g | ] }| ¡ qS r )r )rQ r_ r r r rT { s zATestExcelWriter.test_excel_roundtrip_datetime.<locals>.<listcomp>r° rÝ r rU ) r1 r¹ r} rº r+ r r0 r r2 r r3 )r5 rÖ r» r r+ ZtsfrI rµ r r r Ú test_excel_roundtrip_datetimer s
|
z-TestExcelWriter.test_excel_roundtrip_datetimec
|
C s\ t tdddtdddgtddd d
|
d d tdd ddddggddgddgd}t tdddtdddgtddd d
|
d d tdd ddddggddgddgd}t |¡}t|}| |d¡ W 5 Q R X t|ddd}| |d¡ W 5 Q R X t|}tj |ddd} W 5 Q R X t|}
|
tj |
|
ddd} W 5 Q R X W 5 Q R X t
|
| | ¡ t
|
| |¡ d S )NiÞ r$ é iÏ é é iÎ rt é é rM r! r& é é ZDATEZDATETIMErb rd ©r+ rA r° z
|
DD.MM.YYYYzDD.MM.YYYY HH-MM-SS)r Zdatetime_formatr rU ) r r r r r r r0 r r1 r2 r3 ) r5 r r r8 Z df_expectedÚ filename2Zwriter1Zwriter2Zreader1Zrs1Zreader2Zrs2r r r Ú test_excel_date_datetime_format s< "þú"þú
|
ý
|
|
$ z/TestExcelWriter.test_excel_date_datetime_formatc C s t tjjddddtjd}| ¡ }t |d d¡|d< t |d d¡ t ¡|d< |
|
|d¡ t |}tj |ddd }W 5 Q R X t ||¡ d S )
|
Nr¿ r¦ ©é r$ rÀ r r Únewr° rU )r r} r¨ rÄ rÆ r r1 ÚcutrÅ Ústrr0 r r2 r r3 ©r5 r r8 r6 rI rµ r r r Ú test_to_excel_interval_no_labels¯ s
|
z0TestExcelWriter.test_to_excel_interval_no_labelsc C s¦ t tjjddddtjd}| ¡ }tj|d dddd d
|
d d d dddg
|
d}||d< t t |¡|d< |
|
|d¡ t |}tj |ddd}W 5 Q R X t ||¡ d S )Nr¿ r¦ rì rÀ r r rC rD rØ rÙ ÚEÚFÚGÚHÚIÚJ)Úlabelsrî r° rU )r r} r¨ rÄ rÆ r r1 rï ZSeriesr{ r0 r r2 r r3 )r5 r r8 r6 Z intervalsrI rµ r r r Ú test_to_excel_interval_labels¾ s ÿ
|
z-TestExcelWriter.test_to_excel_interval_labelsc C s t tjjdddddgtjd}| ¡ }|d dd ¡|d < |d d
|
d ¡|d < | |d ¡ t|}t j
|
|d d d }W 5 Q R X t ||¡ d S )Nr¿ r¦ rì rÀ rC )rA rà c S s
|
t | dS )N©Úseconds)r r r r r r Ø r z9TestExcelWriter.test_to_excel_timedelta.<locals>.<lambda>rî c S s t | d ¡ d S )Nrû iQ )r Ú total_secondsr r r r r Ú r r° r rU ) r r} r¨ rÄ rÆ r Úapplyr0 r r1 r2 r r3 rñ r r r Útest_to_excel_timedeltaÏ s ÿÿ
|
z'TestExcelWriter.test_to_excel_timedeltac C sX |j ddd ¡ }| |d¡ t|}tj|ddd}W 5 Q R X t || d¡¡ d S )NÚMZperiod)ÚkindZsht1r rU ) ZresampleZmeanr0 r r1 r2 r r3 Z to_period)r5 r» r rÓ rI rÒ r r r Útest_to_excel_periodindexâ s
|
|
z)TestExcelWriter.test_to_excel_periodindexc C s¦ t jt|jd t jd dd¡}tj|ddgd}||_|j|ddd |j|dd
|
d gd |j|d|d t |}t
|
j |dddgd}W 5 Q R X t ||¡ d S )Nr& r rv ÚfirstÚsecond©rE r° Fr± rC rD r@ rÝ r r$ rU )r} ÚarangerÍ r+ rÆ Úreshaper r r0 r r1 r2 r r3 )r5 rÖ r r ÚarraysÚ new_indexrI r8 r r r Útest_to_excel_multiindexë s "
|
z(TestExcelWriter.test_to_excel_multiindexc C sb t d ddgdddgtj d¡d}| ddg¡}|j||d tj|d
|
d gd }t ||¡ d S ) Nr& r" r¦ rí é ©rC rD rØ rC rD rÝ r r$ r )
|
r r} r¨ ÚsamplerÛ r0 r1 r2 r r3 )r5 rÖ r r8 Zdf1r r r Ú"test_to_excel_multiindex_nan_labelú s
|
$z2TestExcelWriter.test_to_excel_multiindex_nan_labelc C sÖ t jt|jd t jd dd¡}tj|ddgd}||_t ddd d
|
g¡}||_ d d g}|sbd }|j
|
|d |d t | }t j |d |d d gd} W 5 Q R X |sÆ|j jdddd}
|
dd t|
|
D |_ t || ¡ d S )Nr& r rv r r r )é( r$ )r r& )é2 r$ )r r& r r$ r° rÝ ©r. r, rF F)ZsparsifyZadjoinrE c S s g | ]}d tt|¡qS )r )ÚjoinÚmaprð )rQ Úqr r r rT s zATestExcelWriter.test_to_excel_multiindex_cols.<locals>.<listcomp>)r} r rÍ r+ rÆ r r r Ú from_tuplesrA r0 r r1 r2 ÚformatrW r r3 ) r5 rÖ r r r r Znew_cols_indexr, rI r8 Zfmr r r Ú test_to_excel_multiindex_cols s( "
|
ÿz-TestExcelWriter.test_to_excel_multiindex_colsc C s |j tjt|j tjdg}t |¡|_ ddg|j _|j|d|d t |}t
|
j |dddgd}W 5 Q R X t ||¡ |j jd kstd S )
|
Nr Útimer< r° rÝ r r$ rU )r r< )r+ r} r rÍ rÆ r r rE r0 r r1 r2 r r3 rÏ )r5 rÖ r» r r rI rµ r r r Ú test_to_excel_multiindex_dates s
|
z.TestExcelWriter.test_to_excel_multiindex_datesc C sz t ddgddgddgd}| ¡ }t dd g¡}||_|j|d
|
d d t|}tj|d
|
d }W 5 Q R X t
|
||¡ d S )Nr¦ rí r r r é< )r> r? Úc)éF éP )éZ r% r° FrB ©r. ) r r r r r+ r0 r r1 r2 r r3 )r5 r Zframe1r® Z multi_indexrI Zframe3r r r Ú'test_to_excel_multiindex_no_write_index( s
|
z7TestExcelWriter.test_to_excel_multiindex_no_write_indexc C st t g dddgd}t g tjg ddgddgd}| |d¡ t|}tj|dd}W 5 Q R X tj||d d d
|
d S ) Nr r$ r& r@ r ré r° r F)Zcheck_index_typeZ check_dtype) r r r r0 r r1 r2 r r3 )r5 r r6 r8 rI r9 r r r Ú test_to_excel_empty_multiindex= s
|
ÿz.TestExcelWriter.test_to_excel_empty_multiindexc C s t dddgdddggddgd d
|
d gd }|j|d dd t|}tj|d dd}W 5 Q R X t dddgdddggddgd d
|
d gd }t ||¡ d S )N翶~úÏ¿?ç9DÜJÎ?ç >éD)â?çÜ.4×i¤(@ç33333þ@çÍÌÌ̤ArC rD rb rd rÑ ré r° ú%.2f©Z float_formatr rU ç¸
ëQ¸¾?çq=
|
×£pÍ?ç=
|
×£p=â?ç¤p=
|
×£(@)r r0 r r1 r2 r r3 )r5 r r8 rI r9 r6 r r r Útest_to_excel_float_formatJ s ý
|
ýz*TestExcelWriter.test_to_excel_float_formatc C sp t dddgdddggddgd d
|
d gd }t d | ¡0}|j|dd tj|ddd}t ||¡ W 5 Q R X d S )Nu Æu Æu Æu Æu Æu Æu AÆrD u XÆrd rÑ ré z __tmp_to_excel_float_format__.Z TestSheetr r rU r/ )r5 r r8 r7 r9 r r r Ú test_to_excel_output_encoding\ s ýz-TestExcelWriter.test_to_excel_output_encodingc
|
C sè t d| ¡}zt|d W 5 Q R X W n tk
|
rH t d¡ Y nX tdddgddd ggd
|
d gd d dgd}|j|ddd t|}t j
|
|ddd}W 5 Q R X W 5 Q R X tdddgddd ggd
|
d gd d dgd}t ||¡ d S )Nu Æu.Úwbz$No unicode file names on this systemr" r# r$ r% r& r' rC rD rb rd rÑ ré r° r( r) r rU r* r+ r, r- ) r r ÚopenÚUnicodeEncodeErrorrw Úskipr r0 r r1 r2 r3 )r5 r r7 r8 rI r9 r6 r r r Ú test_to_excel_unicode_filenamei s( ý
|
$ýz.TestExcelWriter.test_to_excel_unicode_filenameÚ use_headersTFÚ r_idx_nlevelsr$ r& r" Ú c_idx_nlevelsc s( d fdd }|rdnd }|t dg||}|jdks<t|jd tjk sPtd} d}
|
tj| |
|
||d } |d
|
kr d } tj t
|
| d || |d d W 5 Q R X n|| |}|rÆ|j| |
|
| fksàtn|j| d
|
|
|
| fksàtt t |j D ]4} t t |jD ] }|j| |f tjk s tq qîd S )NTr c
|
sH | j | |d t$}tj||jd |dW 5 Q R £ S Q R X d S )N)r, rÖ r+ r ©r. r, )r0 r r1 r2 rÎ )Údatar, Z
|
parser_hdrr+ rß ©rÖ r r r Ú roundtrip s
|
ÿz;TestExcelWriter.test_excel_010_hemstring.<locals>.roundtrip)r$ r& )r r rt r" )r6 r7 r$ z_Writing to Excel with MultiIndex columns and no index \('index'=False\) is not yet implemented.r FrB )Tr T)r r rÏ r r} r~ r rz rw r¢ ÚNotImplementedErrorr| rÍ r+ rA )r5 rÖ r7 r6 r5 r r; Z parser_headerr ZnrowsZncolsr8 r¤ Úrr r r: r Útest_excel_010_hemstring s2 ÿÿ
|
z(TestExcelWriter.test_excel_010_hemstringc C sB t dddgdddgdddggdddgd}| |d¡ t dddgdddgdddggdddgd}tj|dd d
|
}t ||¡ t dddd gd d ddggddddgd}| |d¡ tj|dd d
|
}t dddd gd d ddggddddgd}t ||¡ |j|dddd tj|dd d}t dddd gd d ddgg}t ||¡ d S )Nr$ r& r" rC rD r@ r° zB.1r rU r! rt é r] é úA.1Fr* r8 ©r r0 r1 r2 r r3 ©r5 r r8 r6 r9 r r r Útest_duplicated_columns¹ s( * ÿ (
|
ÿ z'TestExcelWriter.test_duplicated_columnsc C sj t dddgdddgd}|j|dddgd tj|ddd }t |d |d ¡ t |d |d ¡ d S )
|
Nr$ r& ©rC rD r° rD rC r@ r r8 )r r0 r1 r2 r Úassert_series_equal)r5 r Ú write_frameÚ
|
read_framer r r Útest_swapped_columnsÖ s
|
z$TestExcelWriter.test_swapped_columnsc C sz t dddgdddgd}tjtdd |j|dddgd W 5 Q R X tjtd
|
d |j|ddd gd W 5 Q R X d S ) Nr$ r& rE zNot all names specifiedr r° rD rØ r@ z.'passes columns are not ALL present dataframe'rÙ )r rw r¢ ÚKeyErrorr0 )r5 r rG r r r Útest_invalid_columnsà s ÿz$TestExcelWriter.test_invalid_columnsz#to_excel_index,read_excel_index_col)Tr )FNc C sd t dddgdddgdddgd}|j|dddg|d |ddg }tj|d|d }t ||¡ d S )
|
Nr$ r& r" r Zcol_subset_bugrC rD ©rA r+ rU rB )r5 r Zto_excel_indexZread_excel_index_colrG r6 rH r r r Útest_write_subset_columnsì s " ÿ ÿz)TestExcelWriter.test_write_subset_columnsc C sv t dddgdddgd}| |d¡ tj|ddd}d |jd < d |jd
|
< d |jd < tj|dd dd }t ||¡ d S )NÚoneú#oneÚtwoú#tworE Útest_cr rU )r$ r ©r$ r$ )r& r$ ú#©r. ÚcommentrF )r r0 r1 r2 r r r3 ©r5 r r8 Zresult1Zresult2r r r Útest_comment_arg s
|
|
|
z TestExcelWriter.test_comment_argc C sT t dddgdddgd}| |d¡ tj|dd}tj|dd d}t ||¡ d S ) NrN rO rP rQ rE rR r )r. rV rB rW r r r Útest_comment_default s
|
z$TestExcelWriter.test_comment_defaultc C sb t dddgdddgd}| |d¡ t dd dgdd d gd}tj|dddd }t ||¡ d S )
|
NrN rO rP rQ rE rR rT r rU rB rC r r r Útest_comment_used! s
|
z!TestExcelWriter.test_comment_usedc C sT t ddgddgd}|j|dd t dgd gd}tj|d
|
d }t ||¡ d S ) NÚ1z#2Ú2Ú3)r> r? FrB r$ r& rT )rV rB rC r r r Útest_comment_empty_line/ s
|
z'TestExcelWriter.test_comment_empty_linec C sô t ddddddt ddddddt ddddd d
|
t dddd dd t dddd ddt ddddddt ddddddt ddddddt dddddd t ddddddt ddddd d g }td |i}| |d ¡ tj|d dd }t |d |d ¡ d S )!NiÝ r$ rè r& r" é- é8 r! é é1 r? é* r] é9 é# rã é) rç rK é é é é5 é é% r é rí é4 rC r³ r8 )r r r0 r1 r2 r rF )r5 r Z datetimesrG rH r r r Útest_datetimes; s õ z TestExcelWriter.test_datetimesc
|
C sn t ^}ttj dd¡}t||d}| |¡ W 5 Q R X | d¡ tj |dd}t
|
||¡ W 5 Q R X d S )Nr¦ r& ©r r r ) r r r} r¨ r© r r0 Úseekr1 r2 r r3 )r5 r Zbior8 rH Z reread_dfr r r Ú test_bytes_ioQ s
|
z TestExcelWriter.test_bytes_ioc C s~ t dddgdddgdddgd d
|
d gd }| |d ¡ tj|d ddd}| ¡ }|j t¡|_|j d¡|_t
|
||¡ d S )Nr> r? r Úer& )rN Úfr$ g @ZappleZbananaZcherry)ÚmixedÚnumericrð r³ r r rÆ ) r r0 r1 r2 r ru rþ rð rv rÅ r r3 ©r5 r r8 Úreadr6 r r r Útest_write_lists_dict^ s ýÿ z%TestExcelWriter.test_write_lists_dictc C sF t ddgddgd}| |d¡ tj|ddd}|}t ||¡ d S ) Nr$ r& r" r! )Úrenderr9 r³ r r rB rw r r r Útest_render_as_column_namep s
|
z*TestExcelWriter.test_render_as_column_namec C sV t ddggddgd}| ddd¡}| |¡ tj|dgdgd d
|
}t ||¡ d S ) Nr< r= Úcol1Úcol2r@ TF)r< r= r )Z true_valuesZ false_valuesrF )r Úreplacer0 r1 r2 r r3 )r5 r r8 r6 rH r r r Ú!test_true_and_false_value_optionsx s
|
ÿz1TestExcelWriter.test_true_and_false_value_optionsc C sJ t ddgddggddgd}|j|dd d
|
tj|d d }t ||¡ d S ) Nr$ r& r" r! r| r} r@ r³ rS )Z freeze_panesr r rB )r5 r r6 r9 r r r Útest_freeze_panes s z!TestExcelWriter.test_freeze_panesc C sJ t ¡ }t|j|d}ttjdd}t j||d| d}t ||¡ d S ©Nrp r r r< )r )r Ú makeDataFramer r0 r1 r2 Zround_trip_pathlibr3 ©r5 r r r8 rH rI r9 r r r Útest_path_path_lib s
|
z"TestExcelWriter.test_path_path_libc C sJ t ¡ }t|j|d}ttjdd}t j||d| d}t ||¡ d S r )r r r r0 r1 r2 Zround_trip_localpathr3 r r r r Útest_path_local_path s
|
z$TestExcelWriter.test_path_local_pathc C s¬ t t d¡t d¡ft d¡t d¡fg¡}ttjddd|d}| |¡ tj|dd gdd
|
}|j j
|
d d |j d D d d |j d D gdd gd |_ t ||¡ d S )NZ2018Z2018Q1Z2018Q2)r& r& rÆ r r@ r r$ rf c S s g | ] }t |qS r )rð )rQ rO r r r rT ¨ s zCTestExcelWriter.test_merged_cell_custom_objects.<locals>.<listcomp>)Úlevel)r r r1 ZPeriodr r} Zonesr0 r2 rA Z
|
set_levelsÚlevelsr r3 )r5 r Úmir6 r9 r r r Ú test_merged_cell_custom_objects s þÿ
|
&þz/TestExcelWriter.test_merged_cell_custom_objectsrà Nc C s |}t jd|d}t|g|d}tjtdd | |¡ W 5 Q R X | ¡ }t|g|d}tjtdd | |¡ W 5 Q R X d S )NZ2019)Útzr zExcel does not supportr )r1 r r rw r¢ r£ r0 Z to_pydatetime)r5 rà Ztz_aware_fixturer r r9 r8 r r r Ú test_raise_when_saving_timezones s z0TestExcelWriter.test_raise_when_saving_timezonesc C sh t ddgddgd}|j|dddgdd t |¡}t dddgdddggddd
|
gd }t ||¡ d S ) Nr r$ r¦ rK rE rC rD FrL rA r@ rB )r5 r r8 r9 r6 r r r Ú'test_excel_duplicate_columns_with_names» s
|
|
"z7TestExcelWriter.test_excel_duplicate_columns_with_namesc
|
C sJ d}t |¡2}tjtt |¡d t|dd W 5 Q R X W 5 Q R X d S )Nz7if_sheet_exists is only valid in append mode (mode='a')r r~ )Zif_sheet_exists)r r rw r¢ r£ ÚreÚescaper )r5 r r¤ rt r r r Útest_if_sheet_exists_raisesÄ s z+TestExcelWriter.test_if_sheet_exists_raisesc
|
C sZ t |¡F}t||d}t |¡ W 5 Q R X t |¡}t }t ||¡ W 5 Q R X d S ©Nrp )r r r r r0 r1 r2 r3 )r5 r r r rH r9 r6 r r r Ú test_excel_writer_empty_frameÌ s
|
z-TestExcelWriter.test_excel_writer_empty_framec C sF t |¡2}t j||d t |¡}t }t ||¡ W 5 Q R X d S r rl )r5 r r r r9 r6 r r r Útest_to_excel_empty_frameÕ s
|
|
z)TestExcelWriter.test_to_excel_empty_frame)Dr r r r¥ r¬ r¯ r¶ r· r¼ r½ rw rx r r} Zint8Úint16Úint32rÆ rÇ Úfloat16Úfloat32Úfloat64rÈ rÊ rÌ rÐ rÔ rÜ rà rá rë rò rú rÿ r r
|
r r r r r! r. r/ r4 r> rD rI rK rM rX rY rZ r^ ro rr ry r{ r r r r
r Úobjectr r r r r r r r r r > s 1
|
|
9 , 4
|
þþ
|
|
r c
|
@ sZ e Zd Zej dejede d¡deje
|
de d¡dg¡dd Z dd Z d
|
d Z d S ) ÚTestExcelWriterEngineTestsz klass,extr r r r c
|
C s^ t |¡J}t|6}|dkr8t d¡r8t|tsFtnt||sFtW 5 Q R X W 5 Q R X d S )Nr r )r r r ÚtdZ safe_importÚ
|
isinstancer rÏ )r5 Úklassr r rH r r r Útest_ExcelWriter_dispatchß s
|
|
z4TestExcelWriterEngineTests.test_ExcelWriter_dispatchc C s& t jtdd td W 5 Q R X d S )Nz No enginer Znothing)rw r¢ r£ r ©r5 r r r Ú test_ExcelWriter_dispatch_raisesï s z;TestExcelWriterEngineTests.test_ExcelWriter_dispatch_raisesc C s¸ G dd dt }t| tdd` d}t |¡>}t |}t||sLtW 5 Q R X t dd¡}| |¡ W 5 Q R X | ¡ W 5 Q R X t d¡}|j|dd W 5 Q R X | ¡ d S ) Nc @ sP e Zd ZdZdZdZdZdZdd Zdd Z dd Z
|
e d
|
d Z e d d ZdS )zCTestExcelWriterEngineTests.test_register_writer.<locals>.DummyClassF)ZxlsxZxlsr× c S s d S )Nr r r r r Úbookû s zHTestExcelWriterEngineTests.test_register_writer.<locals>.DummyClass.bookc S s dt | _d S ©NT)ÚtypeÚ called_saver r r r Ú_saveþ s zITestExcelWriterEngineTests.test_register_writer.<locals>.DummyClass._savec _ s dt | _d S r¡ )r¢ Úcalled_write_cells)r5 ÚargsÚkwargsr r r Ú _write_cells s zPTestExcelWriterEngineTests.test_register_writer.<locals>.DummyClass._write_cellsc S s dt | _d S r¡ )r¢ Ú called_sheetsr r r r rY s zJTestExcelWriterEngineTests.test_register_writer.<locals>.DummyClass.sheetsc S s. | j s
|
t| jst| jr td| _ d| _d S )NF)r£ rÏ r¥ r© )Úclsr r r Úassert_called_and_reset s
|
|
|
|
z[TestExcelWriterEngineTests.test_register_writer.<locals>.DummyClass.assert_called_and_resetN)r r r r£ r¥ r© Z_supported_extensionsZ_enginer r¤ r¨ ÚpropertyrY Ú classmethodr« r r r r Ú
|
DummyClassô s
|
r® zio.excel.xlsx.writerr× zsomething.xlsxr$ z something.xlsrp )
|
r r r
|
r r r rÏ rz r0 r« )r5 r® r ÚfilepathrH r8 r r r Útest_register_writeró s
|
z/TestExcelWriterEngineTests.test_register_writerN)r r r rw rx r Úparamr r Ú
|
skip_if_nor r r r° r r r r r Þ s þþ
|
r c @ s e Zd Zdd Zdd ZdS )Ú
|
TestFSPathc
|
C s^ t d¡J}tdddgi}| |¡ t|}t |¡}W 5 Q R X ||ksPtW 5 Q R X d S )Núfoo.xlsxrC r$ r& )r r r r0 r ÚosÚfspathrÏ )r5 r r8 r« r9 r r r Útest_excelfile_fspath# s
|
|
z TestFSPath.test_excelfile_fspathc
|
C sD t d¡0}t| }t |¡t|ks,tW 5 Q R X W 5 Q R X d S )Nr´ )r r r rµ r¶ rð rÏ )r5 r rH r r r Útest_excelwriter_fspath+ s
|
z"TestFSPath.test_excelwriter_fspathN)r r r r· r¸ r r r r r³ s r³ r c C s6 dd t tD }dd t | D }| |¡r2td S )Nc S s h | ]}| d ¡s|qS ©Ú_©Ú
|
startswith©rQ rÞ r r r Ú <setcomp>4 s
|
z%test_subclass_attr.<locals>.<setcomp>c S s h | ]}| d ¡s|qS r¹ r» r½ r r r r¾ 5 s
|
)Údirr Úsymmetric_differencerÏ )r Z
|
attrs_baseZ attrs_klassr r r Útest_subclass_attr1 s rÁ )/r r r Ú functoolsr Úior rµ r Únumpyr} rw Z pandas.util._test_decoratorsÚutilZ_test_decoratorsr Zpandasr1 r r r r
|
Zpandas._testingZ_testingr Zpandas.io.excelr r r r r Zpandas.io.excel._utilr Zfixturer r rx r r± r² r Z usefixturesr r r³ ÚvaluesrÁ r r r r Ú<module> s
|
|
ÿúþ yýýýðþ
|
B
|
