U
|
ý°d, ã @ sê d dl Z d dlZd dlZd dlZd dlm mZ d dl m
|
Z
|
m Z d dl m Z d dlmZ d dlmZ e d¡ dd Zej dd d
|
g¡d d Zd dddgdddfddddgdddfdddgdfddd gd fd dd gd fd dd!gd"fd#d$gd%fd&d'd(gd)fd*d'd+gdd,dfd-d'd+gd.fd/d0d)d1gd2fd3d0d)d1gd4fd5d0d)d1gd6fd7d0d)d1gd8fd9d0d)d1gd:fd;d0d)d1gd<fd=d0d)d1gd>fd?d0d)ddgdddfd@d0dAd1gd<fdBd0dAddgdddfd@d0dCd1gd<fdBd0dCddgdddfd@d0d,d1gd<fdBd0d,ddgdddfd@d0d)d1gd<fdBd0d)ddgdddfdDd0d)d1gdEfgZej dd d
|
g¡ej dFe¡dGdH Zej dd d
|
g¡ej dFe¡dIdJ Zd8dKdLdEd4d6dMdNdOd:g
|
Zej dd d
|
g¡ej dPe¡dQdR Z dSdT Z ejj ej dUdV Z dS )Wé N)Ú DataFrameÚ
|
read_excel)Ú ExcelWriter)ÚExcelFormatterZjinja2c C sx | j j|j jkst| jj|jjks(t| jj|jjks<t| jj|jjksPt| j|jks`t| jj|jjksttd S ©N)Ú alignmentÚ__dict__ÚAssertionErrorÚborderÚfillÚfontÚ number_formatZ
|
protection)Úcell1Úcell2© r úWd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/io/excel/test_style.pyÚassert_equal_cell_styles s r ÚengineÚ
|
xlsxwriterÚopenpyxlc
|
|
C sô t | ¡ ttj dd¡}t d¡Ä}t|| d$}|j |dd |j
|
j |dd W 5 Q R X t d¡}t | |¡¡h}t|d j|d jD ]J\}}t|t|ks¬tt||D ]"\}} |j| jksÎtt|| q¶qW 5 Q R X W 5 Q R X d S )Né ú.xlsx©r Ú dataframe©Z
|
sheet_nameZunstyledr )ÚpytestÚ importorskipr ÚnpÚrandomÚrandnÚtmÚ ensure_cleanr Úto_excelÚstyleÚ
|
contextlibÚclosingÚ load_workbookÚzipÚcolumnsÚlenr Úvaluer )
|
r ÚdfÚpathÚwriterr ÚwbZcol1Zcol2r r r r r Ú test_styler_to_excel_unstyled" s
|
|
r/ zbackground-color: #111222r ZfgColorÚrgbZFF111222Ú00111222)r r zcolor: #111222r Úcolorr* zfont-family: Arial;ÚnameZarialzfont-weight: bold;ÚbTzfont-style: italic;Úiztext-decoration: underline;ÚuZsingleznumber-format: $??,???.00;r z
|
$??,???.00ztext-align: left;r Z
|
horizontalÚleftzvertical-align: bottom;ZverticalÚbottomzvertical-align: middle;Úcenterzborder-left: 2pt solid redr
|
r# Zmediumzborder-left: 1pt dotted redZdottedzborder-left: 2pt dotted redZmediumDashDotDotzborder-left: 1pt dashed redZdashedzborder-left: 2pt dashed redZ mediumDashedzborder-left: 1pt solid redZthinzborder-left: 3pt solid redZthickz border-left: 2pt solid #111222zborder: 1pt solid redÚtopzborder: 1pt solid #111222Úrightz1border-left-style: hair; border-left-color: blackZhairzcss, attrs, expectedc
|
sL t | ¡ ttj dd¡}|j fdd¡}t d¡}t
|
|| d"}|j |dd |j |dd W 5 Q R X t d ¡}t | |¡¡(} | d d
|
d
|
¡| d d
|
d
|
¡ }
|
} W 5 Q R X |D ] } t|
|
| d t| | }
|
} q¾t|tr|
|
d ks|
|
|| kst| || ks>tn&|
|
d ks0|
|
|ks0t| |ks>tW 5 Q R X d S ) Né c s S r r ©Úx©Úcssr r Ú<lambda> ó z,test_styler_to_excel_basic.<locals>.<lambda>r r r r Ústyledr r ©r r r r r r r# Úapplymapr r! r r" r$ r% r& ÚcellÚgetattrÚ
|
isinstanceÚdictr ) r r@ ÚattrsÚexpectedr+ Ústylerr, r- r r. Úu_cellÚs_cellÚattrr r? r Útest_styler_to_excel_basic~ s"
|
|
, rP c
|
s2 t | ¡ ttj dd¡}|j}|j fdddd |j fdddd |j}| dd ¡ |jdd dd |jd d dd t
|
d
|
¡}t || d "}|j |d d |j |dd W 5 Q R X t d¡} t | |¡¡J}
|
|
|
d dd¡|
|
d dd¡ } } |
|
d dd¡|
|
d dd¡ } }W 5 Q R X |D ]6}t| |d t| | } } t| |d t|| } }q6t|trØ| d ks| || kst| || ks¨t| d ksÄ| || ksÄt||| ks$tnL| d ksð| |ksðt| |ksþt| d ks| |kst||ks$tW 5 Q R X d S )Nr< c s S r r r= r? r r rA § rB z4test_styler_to_excel_basic_indexes.<locals>.<lambda>r ©Zaxisc s S r r r= r? r r rA ¨ rB c S s dS ©Nz
|
null: css;r r= r r r rA « rB c S s dS rR r r= r r r rA ¬ rB c S s dS rR r r= r r r rA rB r r Z null_styledr rC r r )r r r r r r r# Zapplymap_indexrE r r! r r" r$ r% r& rF rG rH rI r )r r@ rJ rK r+ rL Z null_stylerr, r- r r. Zui_cellZsi_cellZuc_cellZsc_cellrO r r? r Ú"test_styler_to_excel_basic_indexes s:
|
|
", rS Z mediumDashDotZ
|
dashDotDotÚdoubleZdashDotZ slantDashDotÚ border_stylec
|
sf d| d dddg}|}t | ¡ ttj dd¡}|j fdd¡}t d ¡}t
|
|| d
|
"}|j |d d |j |d d W 5 Q R X t d¡}t | |¡¡(} | d dd¡| d dd¡ }
|
} W 5 Q R X |D ] } t|
|
| d t| | }
|
} qØt|tr2|
|
d ks |
|
|| ks t| || ksXtn&|
|
d ksJ|
|
|ksJt| |ksXtW 5 Q R X d S )Nz border-left: z black thinr
|
r7 r# r< c s S r r r= r? r r rA ê rB z3test_styler_to_excel_border_style.<locals>.<lambda>r r r r rC r r rD ) r rU rJ rK r+ rL r, r- r r. rM rN rO r r? r Ú!test_styler_to_excel_border_styleÞ s(
|
|
|
, rV c
|
C s¶ t d¡} dd }ttj dd¡}|j dd ¡}t d¡n}t
|
|dd }t ||d j |d
|
d W 5 Q R X t | |¡¡$}|d
|
d d ¡jjjd kstW 5 Q R X W 5 Q R X d S )Nr c S s ddddiiiS )Nr r2 r0 Z111222r r? r r r Úcustom_converter s z6test_styler_custom_converter.<locals>.custom_converterr< c S s dS )Nzcolor: #888999r r= r r r rA rB z.test_styler_custom_converter.<locals>.<lambda>r r )Zstyle_converterZcustomr r r1 )r r r r r r r# rE r r! r r Úwriter$ r% r& rF r r2 r* r )r rW r+ rL r, r- r. r r r Ú test_styler_custom_converter s
|
ÿrY c C s¼ d\}}t dddgdddgd}|jjdd }|jd
|
| d | |d d }|dd | d¡j ¡ D krnq¸t d¡ |d8 }|dkst dt
|
d
|
| d | d|d}t ||¡ qNd S )N)ú pandas-testz test.xlsxr< r é é é )r> ÚyÚindexrQ zs3://ú/)Ústorage_optionsé c s s | ] }|j V qd S r )Úkey)Ú.0Úobjr r r Ú <genexpr> s z$test_styler_to_s3.<locals>.<genexpr>rZ g¹?r z,Timed out waiting for file to appear on moto)Z index_colra ) r r# Z
|
set_stickyr" ZBucketZobjectsÚallÚtimeÚsleepr r r Zassert_frame_equal)Z s3_resourceZs3soZmock_bucket_nameZ target_filer+ rL ÚtimeoutÚresultr r r Útest_styler_to_s3 s$ ÿ
|
ÿrl )!r$ rh Únumpyr r Z pandas.util._test_decoratorsÚutilZ_test_decoratorsÚtdZpandasr r Zpandas._testingZ_testingr Zpandas.io.excelr Zpandas.io.formats.excelr r r ÚmarkZ parametrizer/ Zshared_style_paramsrP rS Zexcel_border_stylesrV rY Z
|
single_cpuZskip_if_not_us_localerl r r r r Ú<module> sÎ
|
|
þ
|
ýý
|
ý
|
ý
|
ý
|
ý
|
ý
|
ýýÂFþ þ -óþ
|
