U
|
ý°d¹
|
ã @ s d dl Z d dlZd dlmZ d dlmZ d dlmZ e d¡Z
|
ej ddg¡Z dd Zd d
|
Zej d d d g¡dd Zdd ZdS )é N)Ú DataFrame)Ú ExcelWriterÚ
|
xlsxwriterÚextz.xlsxc
|
C s< t d¡}t | ¡ }tddgddgd}t|B}| |¡ d}|j}| ¡ d }| d|i¡}|
|
dd |¡ W 5 Q R X t | |¡¡4} z | d }
|
W n tk
|
r¼ | jdd }
|
Y nX W 5 Q R X z |
|
d
|
} W n tk
|
rò |
|
d
|
¡} Y nX z
|
| j} W n tk
|
r | jjj} Y nX | |ks.tW 5 Q R X d S ) NÚopenpyxli@â )ÚAÚBz#,##0r Ú
|
num_formatzB:BZSheet1)ÚnameZB2)ÚpytestÚ importorskipÚtmÚ ensure_cleanr r Zto_excelÚbookZ
|
worksheetsZ
|
add_formatZ
|
set_columnÚ
|
contextlibÚclosingZ load_workbookÚ TypeErrorZget_sheet_by_nameÚcellZ number_formatÚAttributeErrorÚstyleZ _format_codeÚAssertionError) r r ÚpathÚframeÚwriterr Zwrite_workbookZwrite_worksheetZ
|
col_formatZ read_workbookZread_worksheetr Zread_num_format© r ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/io/excel/test_xlsxwriter.pyÚtest_column_format s0
|
|
|
|
r c
|
C sF d}t | ¡.}tjt|d t|ddd W 5 Q R X W 5 Q R X d S )Nz-Append mode is not supported with xlsxwriter!)Úmatchr Úa)ÚengineÚmode)r r r ZraisesÚ
|
ValueErrorr )r ÚmsgÚfr r r Ú test_write_append_mode_raises7 s r$ Únan_inf_to_errorsTFc
|
C sP dd|ii}t | ¡0}t|d|d}|jj|ks8tW 5 Q R X W 5 Q R X d S )NÚoptionsr% r )r Ú engine_kwargs)r r r r r% r )r r% r' r# r r r r Útest_engine_kwargs? s r( c
|
C s^ t | ¡J}t|dd2}|ji ks(t|j d¡}|jd|iksFtW 5 Q R X W 5 Q R X d S )Nr )r Z test_name)r r r Zsheetsr r Z add_worksheet)r r# r Zsheetr r r Ú test_book_and_sheets_consistentH s
|
r) )r r Zpandasr Zpandas._testingZ_testingr Zpandas.io.excelr r r ÚmarkZ parametrizeZ
|
pytestmarkr r$ r( r) r r r r Ú<module> s
|
(
|
|
