U
|
ý°d»8 ã @ s¬ d dl Z d dlmZ d dlZd dlZd dlZd dlZd dlm Z d dl
|
m Z d dl mZmZ e d¡Zej ddg¡Zdd Zd
|
d Zej d d dg¡dd Zdd Zej dddg¡dd Zej dddgfdd d dgfg¡d d Zej d d!d"d#d$gfd%d&d'gfd(d&d'd$gfg¡d)d* Zej d+d d d,d-gd.d/gfd d&d0d-gd,d/gfd&d d0d,gd.d/gfd&d&d0d-gd.d,gfg¡d1d2 Z ej d3d4d5d6g¡d7d8 Z d9d: Z ej d;d dg¡d<d= Z ej d>d ej d?d&d"d@gej dAdBdCdDgej dEdFdGdHgdIfd"d&d"d@gdBdCdDgdFdGdHgdJfg¡ej dKdLdMdNg¡ej d;d ddg¡dOdP Z!dQdR Z"ej d;d ddg¡dSdT Z#ej d;d ddg¡dUdV Z$dWdX Z%dYdZ Z&d[d\ Z'dS )]é N)ÚPath)Ú DataFrame)Ú ExcelWriterÚ_OpenpyxlWriterÚopenpyxlÚextz.xlsxc C s4 ddl m} dddddddddd d
|
d d d ddddidddd}| d¡}|jd|d}|j|jjd}|j||||d}|jdd d
|
}|jd d d}|j d |d} d}
|
|j
|
ddd} t |¡} | d |ksÚt | d |ksêt | d |ksút | d | ks t | d |
|
ks t | d | ks0t d S ) Nr )ÚstylesÚ00FF0000T©ÚcolorÚboldZthin)ÚtopÚrightÚbottomÚleftÚcenterr )Z
|
horizontalZverticalZsolidZ006666FFg333333Ó?)ZrgbZtint)Z patternTypeZfgColorZ format_codez0.00F)ÚlockedÚhidden)ÚfontÚbordersÚ alignmentÚfillÚ number_formatÚ
|
protection)r r )Ústyler Úborderr r r r )r r ZColorZFontZSider Z BORDER_THINZBorderZ AlignmentZ PatternFillZ
|
Protectionr Ú_convert_to_style_kwargsÚAssertionError) r r ZhstyleZ
|
font_colorr Zsider r Z
|
fill_colorr r r Úkw© r úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/io/excel/test_openpyxl.pyÚ test_to_excel_styleconverter s0 ú
|
|
r! c
|
C sþ ddl m} d}dddii}dddii}|ddd |d
|
|ddd |d
|
g}dd d di}t |¡}|d }|ddddd|dg} t | ¡l}
|
t|
|
,} | j||d | j| |d | j| } W 5 Q R X | d } | d }| j|ksât |j|ksðt W 5 Q R X d S )Nr )Ú ExcelCellZ merge_styledr r r Z0000FF00é é* )ÚcolÚrowÚvalr éc Z000000FFTr
|
Úpandas)r% r& r' Z
|
mergestartZmergeendr ©Ú
|
sheet_nameÚB1ÚA2)
|
Zpandas.io.formats.excelr" r r ÚtmÚ ensure_cleanZ _write_cellsÚsheetsr r )r r" r+ Zsty_b1Zsty_a2Z initial_cellsZ
|
sty_mergedZ
|
sty_kwargsZopenpyxl_sty_mergedZ merge_cellsÚpathÚwriterZwksZxcell_b1Zxcell_a2r r r Ú test_write_cells_merge_styled7 s8 þ
|
ÿÿ
|
r3 Ú iso_datesTFc
|
C sX d|i}t | ¡<}t|d|d"}|jj|ks4tt |¡ W 5 Q R X W 5 Q R X d S )Nr4 r )ÚengineÚ engine_kwargs)r. r/ r Úbookr4 r r Úto_excel)r r4 r6 Úfr2 r r r Útest_engine_kwargs_writeY s
|
r: c C s t | ¡l}tddg |¡ tjtt d¡d: t |ddddid }td
|
gj|d d W 5 Q R X W 5 Q R X W 5 Q R X d S ) NÚhelloÚworldzAload_workbook() got an unexpected keyword argument 'apple_banana'©Úmatchr ÚaZ apple_bananaÚfruit©r5 Úmoder6 ZgoodÚSheet2r* )
|
r. r/ r r8 ÚpytestÚraisesÚ TypeErrorÚreÚescaper )r r9 r2 r r r Ú!test_engine_kwargs_append_invalidd s ÿþ ÿrI zdata_only, expected)Tr )Fú=1+1c
|
C sr t | ¡^}tdg |¡ t|ddd|id.}|jd d j|ksJtt j|dd W 5 Q R X W 5 Q R X d S )
|
NrJ r r? Ú data_onlyrA ZSheet1ÚB2rC r* )r. r/ r r8 r r0 Úvaluer )r rK Úexpectedr9 r2 r r r Ú#test_engine_kwargs_append_data_onlyv s ÿrO z mode,expectedÚwÚbazr? ÚfooÚbarc
|
C sþ t dgdgd}t | ¡Ú}t ¡ }d|jd _d|jd d _| d¡ d|jd d _| |¡ t
|
|d|d }|j |dd
|
d W 5 Q R X t t |¡¡L}d d |jD }||ks¼tt|D ] \} }
|
|j| d j|
|
ksÄtqÄW 5 Q R X W 5 Q R X d S )Nr# rQ ©ÚcolumnsrR r ÚA1rS r )r5 rB F©r+ Úindexc S s g | ]
|
}|j qS r )Útitle)Ú.0Úsheetr r r Ú
|
<listcomp> s z*test_write_append_mode.<locals>.<listcomp>)r r. r/ r ZWorkbookZ
|
worksheetsrY rM Ú create_sheetÚsaver r8 Ú
|
contextlibÚclosingÚ load_workbookr Ú enumerate) r rB rN Údfr9 Úwbr2 Zwb2ÚresultrX Z
|
cell_valuer r r Útest_write_append_mode
s
|
|
rf z#if_sheet_exists,num_sheets,expectedÚnewé ÚappleÚbananaÚreplacer# ÚpearÚoverlayc
|
|
C s
|
t dddgi}t ddgi}t | ¡Ø}|j|dddd t|dd |d
|
}|j|ddd W 5 Q R X t t |¡¡~}t |j
|
|kst |j
|
d dkst t j |ddd } t| d |ksÂt t |j
|
dkròt j ||j
|
d dd } t | |¡ W 5 Q R X W 5 Q R X d S )Nr@ ri rj rl r rR F©r5 r+ rX r? ©r5 rB Úif_sheet_existsrW r ©r5 rh r# )r r. r/ r8 r r_ r` r ra ÚlenZ
|
sheetnamesr ÚpdÚ
|
read_excelÚlistÚassert_frame_equal)
|
r rp Z
|
num_sheetsrN Údf1Údf2r9 r2 rd re r r r Ú!test_if_sheet_exists_append_modes s(
|
ÿry z%startrow, startcol, greeting, goodbyeÚpoopr< ÚgoodbyeÚpeopler; c
|
C s¬ t ddgddgd}t dg}t | ¡x}|j|ddd d
|
t|dd d d }|j|d d |d |dd W 5 Q R X tj|ddd} t ||d}
|
t | |
|
¡ W 5 Q R X d S )Nr; r< r{ r| )Úgreetingr{ rz r ZpooFrn r? rm ro r# )rX ÚheaderÚstartrowÚstartcolr+ )r+ r5 )r r. r/ r8 r rs rt rv ) r r r r} r{ rw rx r9 r2 re rN r r r Ú%test_append_overlay_startrow_startcol¼ s,
|
|
ÿú r zif_sheet_exists,msg)Úinvalidzf'invalid' is not valid for if_sheet_exists. Valid options are 'error', 'new', 'replace' and 'overlay'.)ÚerrorúASheet 'foo' already exists and if_sheet_exists is set to 'error'.)Nr c C s t ddgi}t | ¡`}tjtt |¡d@ |j|ddd t |dd|d}|j|dd W 5 Q R X W 5 Q R X W 5 Q R X d S )
|
Nr@ rl r= rR r rq r? ro r* )
|
r r. r/ rD rE Ú
|
ValueErrorrG rH r8 r )r rp Úmsgrc r9 r2 r r r Útest_if_sheet_exists_raisesÝ s ÿr c C sz t | ¡f}tdt ddd¡i}tdt ddd¡i}tj||gdd}|j dd ¡ ¡ }|j
|
|d d
|
W 5 Q R X d S ) NÚAr# é
|
ÚBé )Zaxisc S s d| dk rdnd S )Nzcolor: r ÚredZblackr )r' r r r Ú<lambda> ó z4test_to_excel_with_openpyxl_engine.<locals>.<lambda>r rq ) r. r/ r ÚnpZlinspacers Úconcatr ZapplymapZ highlight_maxr8 )r Úfilenamerw rx rc Zstyledr r r Ú"test_to_excel_with_openpyxl_engineû s ÿr Ú read_onlyc C sZ | dddd| }t tj||d¡}tj|dd}W 5 Q R X t |¡}t ||¡ d S )NÚioÚdataÚexcelZtest1©r r rq )r_ r` r ra rs rt r. rv )Údatapathr r r rd re rN r r r Útest_read_workbook s ÿ
|
r zheader, expected_datar é r é é é ÚCé é é ©ZTitlez
|
Unnamed: 1z
|
Unnamed: 2)r r r r Zdimension_missingZdimension_smallZdimension_largec
|
C sv | ddd| | }|d kr.t j||d}n0t tj||d¡}t j|d|d}W 5 Q R X t|} t || ¡ d S )Nr r r )r~ r r )r5 r~ © rs rt r_ r` r ra r r. rv )
|
r r r~ Z expected_datar r r1 re rd rN r r r Ú test_read_with_bad_dimension s ÿr¤ c
|
C s t }t | ¡}|j|dd t|dddd}| |¡ W 5 Q R X t| ¡ }| d¡}| d|d ¡}| d|d ¡}|dkr|dkstW 5 Q R X d S ) Nr rq r? rg )rB r5 rp s docProps/app.xmlr# éÿÿÿÿ) r r. r/ r8 r r Ú
|
read_bytesÚfindr )r rc r9 r2 r ÚfirstÚsecondÚthirdr r r Útest_append_mode_file7 s ÿ
|
r« c C s | dddd| }|d kr(t |¡}n.t tj||d¡}t j|dd}W 5 Q R X ttjdd d
|
d gtjd d ddgtjddddgd}t
|
||¡ d S )Nr r r Zempty_trailing_rowsr r rq r r# rh r r r r r r r r r¡ r¢ ) rs rt r_ r` r ra r r Únanr. rv ©r r r r1 re rd rN r r r Ú"test_read_with_empty_trailing_rowsM s ÿýÿr® c C sl | dddd| }|d kr(t |¡}n.t tj||d¡}t j|dd}W 5 Q R X t }t ||¡ d S )Nr r r Zempty_with_blank_rowr r rq r£ r r r r Ú test_read_empty_with_blank_rowc s ÿr¯ c
|
C s` t | ¡L}t|dd4}|ji ks(t|j dd¡}|jd|iksHtW 5 Q R X W 5 Q R X d S )Nr rq Z test_namer )r. r/ r r0 r r7 r] )r r9 r2 r[ r r r Ú test_book_and_sheets_consistentr s
|
r° c C sB | dddd| }t |¡}ttdddgd}t ||¡ d S ) Nr r r Zints_spelled_with_decimalsrh é r# rT )rs rt r Úranger. rv ©r r r1 re rN r r r Ú test_ints_spelled_with_decimals{ s
|
r´ c C s | dddd| }t j|dddgdddgd}ttjd d d gd tjtjtjggt j d
|
d d d g¡t j ddg¡d}t ||¡ d S )Nr r r Zmultiindex_no_index_namesr r# rh )Z index_colr~ Úx)ÚXÚYrV )r¶ r· r- )ÚXXÚYYr, )r¸ r¹ rL )r ZAAZAAA)r ÚBBZBBB)rU rX ) rs rt r r r¬ Z
|
MultiIndexÚ from_tuplesr. rv r³ r r r Ú*test_read_multiindex_header_no_index_names s
|
ÿûr¼ )(r_ Úpathlibr rG Únumpyr rD r) rs r Zpandas._testingZ_testingr. Zpandas.io.excelr r Z importorskipr ÚmarkZ parametrizeZ
|
pytestmarkr! r3 r: rI rO rf ry r r r r r¬ r¤ r« r® r¯ r° r´ r¼ r r r r Ú<module> s
|
!"
|
|
|
ÿ
|
|
ýþ
|
üþ
|
öþ
|
|
ýþ ÷þ ÿ
|
|
|
