U
|
ý°dL ã @ sÄ d dl Zd dlZd dlmZmZmZmZmZm Z m
|
Z
|
e d¡ d dl m Z d dlmZ ejdd Zejdd Zejd
|
d Zejd d Zdd Zej dddg¡ej dddg¡dd Zdd Zdd Zdd Zd d Z d d Z d d! Z d"d# Z ej d$d%d&i fd'd(d)d ifd'd*d)d+ifg¡d,d- Z ej d.d/d0g¡d1d2 Z!d3d4 Z"d5d6 Z#ej d7d8dd9gg¡ej d:d%d'g¡d;d< Z$ej d=d+d>d?d@dAgfdBdCdDdEdFgfdGdHdIdJdKgfg¡dLdM Z%ej d)d d+g¡ej dNd dOdOdPdPgfdQdOdOdPdPgfd+dPdPdOdOgfdRdPdPdOdOgfd d+gdOdOdOdOgfd dQgdOdOdPdPgfd dRgdOdOdOdOgfdRdQgdOdOdOdOgfg¡dSdT Z&dUdV Z'ej d7ddWg¡ej dXdYdZg¡ej d[ddBg¡ej d\d]d^g¡d_d` Z(ej d7ddag¡ej dbddcdZg¡ej d[dddg¡ej d\d]d^g¡dedf Z)dgdh Z*didj Z+dkdl Z,ej dmdndo dpfdqd dridsfdtg¡dudv Z-dwdx Z.dydz Z/d{d| Z0ej d}dd+dBgfdd+dBdGddgfg¡d~d Z1dd Z2dd Z3dd
Z4dS )é N)ÚNAÚ DataFrameÚ
|
IndexSliceÚ
|
MultiIndexÚNaTÚ TimestampÚoption_contextZjinja2©ÚStyler)Ú _str_escapec C s" t ddgddggddgddgd S )
|
Nr g°rhí|ã¿é gÙÎ÷Sã¥ó¿ÚAÚBÚxÚy©ÚdataÚcolumnsÚindex)r © r r ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/io/formats/style/test_format.pyÚdf s
|
ýr c C s t | ddS ©Nr ©Zuuid_lenr )r r r r Ústyler s r c C s@ t t d¡ dd¡t ddgddgg¡t ddgd d
|
gg¡d S ) Né é r r ÚaÚbÚXÚYr r r )r ÚnpZarangeZreshaper Ú from_productr r r r Údf_multi! s
|
ýr# c C s t | ddS r r )r# r r r Ú styler_multi* s r$ c C sp | d¡ dd¡}tdd |d D s,ttdd |d D sFtt|d d d d d
|
¡d ksltd S ) Nú{:0.1f}Tc s s | ]}d d |D V qdS )c S s g | ] }d |kqS )Ú display_valuer ©Ú.0Úcr r r Ú
|
<listcomp>1 s ú1test_display_format.<locals>.<genexpr>.<listcomp>Nr ©r( Úrowr r r Ú <genexpr>1 s z&test_display_format.<locals>.<genexpr>Úbodyc s s$ | ] }d d |dd
D V qdS )c S s g | ]}t |d dkqS )r& é )Úlenr' r r r r* 2 s r+ r Nr r, r r r r. 2 s r r r& ú-r0 )ÚformatÚ
|
_translateÚallÚAssertionErrorr1 Úlstrip©r Úctxr r r Útest_display_format/ s r: r TFr c
|
C s¸ ddg}|r&| j dd dd ddg}d d
|
g}|rH| j d d d d dg}| dd¡}t|d D ] \}}|d d || ks`tq`t|d d d
D ] \}} | d || kstqd S )Nr r c S s | ¡ S ©N©Úupper©Úvr r r Ú<lambda>; ó z+test_display_format_index.<locals>.<lambda>r )Úaxisr r r r z*{}*r z*A*z*B*Tr/ r& Úhead©Ú format_indexr4 Ú enumerater6 )
|
r r r Z exp_indexZ exp_columnsr9 Úrr- r) Úcolr r r Útest_display_format_index6 s rI c C sT | ddd¡ dd¡}|d d d d d ks4t|d d d
|
d d ksPtd S ) Nr% ú{0:.2%})r r Tr/ r r r& z0.0é z-60.90%©r3 r4 r6 r8 r r r Útest_format_dictL s rM c C sP | ddd i¡ dd¡}tddgD ]$\}}|d | d d |ks&tq&d S ) Nr c S s | ¡ S r; r< r> r r r r@ S rA z(test_format_index_dict.<locals>.<lambda>Tr r r/ r& rD )r r9 ÚiÚvalr r r Útest_format_index_dictR s rP c C s | d¡ dd¡}|d d d d dks.t|d d d d d ksJt|d d d d d
|
ksft|d d d d d kstd S ) Nú{:.2f}Tr/ r r r& z0.00rK z-0.61ú1.00z-1.23rL r8 r r r Útest_format_stringX s
|
rS c C s | dd ¡ dd¡}|d d d d dks2t|d d d d d
|
ksNt|d d d d dksjt|d d d d d
|
kstd S ) Nc S s | dk r dS dS )Nr ÚnegÚposr r> r r r r@ a rA z&test_format_callable.<locals>.<lambda>Tr/ r r r& rU rK rT rL r8 r r r Útest_format_callable` s
|
rV c C sP t d d gddggddgd} | jjd dd dd¡}|d d
|
d d dksPt|d d
|
d d dkslt| jjddd dd¡}|d d
|
d d dks t|d d
|
d d dks¼t|d d d d dksØt|d d d d dksôt| jjdddgd dd¡}|d d
|
d d dks.t|d d d d dksLtd S )Ngñ?g333333ó?r r ©r r2 ©Úna_repTr/ r r r& rK z{:.2%}z110.00%z120.00%)rY Úsubset)r Ústyler3 r4 r6 ©r r9 r r r Útest_format_with_na_reph s r] c C s t dddddggdd tjttgd} | jjd ddd d
|
d
|
¡}|d d d d dksZtdD ] }|d d | d dks^tq^d S )Nr rK r0 r é r rW z--)rY rB TrC r r& )rK r0 r r^ ) r r! Únanr r r[ rE r4 r6 )r r9 rN r r r Ú test_format_index_with_na_rep{ s
|
$ r` c C s¬ t d tjdgd ttdgd} | jjd dd dd¡}|d d d d
|
dksTt|d d d d
|
dkspt|d d d d
|
dkst|d d d d
|
dks¨td S ) NZfooZ20120101)ÚobjectÚdatetimer2 rX Tr/ r r r& rK ) r r! r_ r r r[ r3 r4 r6 r\ r r r Útest_format_non_numeric_na s
|
þÿ rc zfunc, attr, kwargsr3 Z_display_funcsrE Z_display_funcs_indexrB Z_display_funcs_columnsr c C sZ dt | |kstt | |d| dt | |ks4tt | |f | dt | |ksVtd S )N)r r rQ )rQ )Úgetattrr6 )r ÚfuncÚattrÚkwargsr r r Útest_format_clear s
|
rh z escape, exp)Úhtmlz!<>&"%$#_{}~^\~ ^ \ )Úlatexz<>\&"\%\$\#\_\{\}\textasciitilde \textasciicircum \textbackslash \textasciitilde \space \textasciicircum \space \textbackslash \space c C s d}t |gg}t|ddjdd d}d| d}|| ¡ ksBtt|ddjd| d}d| d}|| ¡ kstttt |gddd}|jdd d d
|
| d d ¡d d d d d| dksÄt|jd| d d
|
| d d ¡d d d d d| dkstd S )Nz<>&"%$#_{}~^\~ ^ \ r r z&{0}&©Úescapez/<td id="T__row0_col0" class="data row0 col0" >&z&</td>rW r )rl rB TrC r& ú&©r r
|
r3 Úto_htmlr6 rE r4 )rl ÚexpÚcharsr ÚsÚexpectedr r r r Útest_format_escape_html¢ s ,rt c C s t dd gg} t| ddjdddd}d}d }|| ¡ ks>t|| ¡ ksNtt dd gd
|
} t| dd}|jdddd d | d d ¡}|d d d d dks¢t|d d d d dks¾td S )Nz<>&"r r zX&{0}>Xri rm )rl rY zI<td id="T__row0_col0" class="data row0 col0" >X&<>&">X</td>z4<td id="T__row0_col1" class="data row0 col1" >&</td>rW r )rl rY rB TrC r& zX&<>&">XrK rn )r rr ÚexZ expected2r r9 r r r Útest_format_escape_na_repà s rv c C sT | j ddd}dD ]}|| ¡ kstq| j ddd}dD ]}|| ¡ ks:tq:d S )Nú{:.1f}ri rk )z>0.0<z>1.0<ú>-1.2<ú>-0.6<r )Ú precisionrl )z>0<z>1<rx ry )r3 ro r6 )r rr rs r r r Útest_format_escape_floatsÕ s r{ Ú formatterr^ ç @re c C s, t jtdd t| || W 5 Q R X d S )Nzexpected str or callable©Úmatch)ÚpytestÚraisesÚ TypeErrorrd )r r| re r r r Útest_format_raisesà s r zprecision, expectedz1.0z2.0z3.2z4.6rK rR z2.01z3.21z4.57r0 z1.000z2.009z3.212z4.566c C s¢ t ddddggddddgd}t|}|j| d |j| dd | d d ¡}t|D ]H\}}|d
|
d |d d |ks|t|d d |d d |ksTtqTd S )Ng ð?gßOn @gQÚ|a² @gÝ$C@rW ©rz r )rz rB Tr/ r r& rC )r r
|
r3 rE r4 rF r6 )rz rs r r r9 rH rp r r r Útest_format_with_precisionç s
|
r
zlevel, expectedr Ú_ÚzeroÚonec sä t jddgddggddgd}tddgddgg}| d krB||_n||_|jjd
|
d || d }| d d ¡ | d kr¤ fddtdD }| fddtdD 7 }n0 fddtdD }| fddtdD 7 }||ksàt d S )Nr r r ©Únamesr rK r0 r r c S s dS )Nr r r> r r r r@ rA z)test_format_index_level.<locals>.<lambda>)ÚlevelrB Tc s g | ]} d | d d qS )r/ r r& r ©r( rr ©r9 r r r* s z+test_format_index_level.<locals>.<listcomp>c s g | ]} d | d d qS )r/ r r& r r r r r r* s c s$ g | ] } d d |d d qS )rC r r r& r r r r r r* s c s$ g | ] } d d |d d qS )rC r r& r r r r r r* s )
|
r Z from_arraysr r r r[ rE r4 Úranger6 )rB r rs Úmidxr r Úresultr r r Útest_format_index_levelü s r c C sP t ddgddggddgd} | jjdddtd d d
f d
|
d d ¡}d }d }|d d d d |ksjt|d d d d |kst|d d d d dks¢t| jjdtd d d
f d
|
d d ¡}|d d d d |ksât|d d d d |ksþt| jjdtd d
|
d d ¡}|d d d d |ks8t|d d d d dksVt| jjdtd d
|
d d ¡}|d d d d |kst|d d d d |ks®t| jjdtd dgdgf d
|
d d ¡}|d d d d |ksòt|d d d d dkst|d d d d dks.t|d d d d |ksLtd S )NgóSt$¿?gï8EGrùñ?r r rW r% rJ )r r r )rZ Tz0.1z1.123400r/ r r& rK z12.34%z0.123400)r r z1.1)r r[ r3 r r4 r6 )r r9 rs Zraw_11r r r Útest_format_subset s< ÿ þ $ ÿ r z{:,.1f}ÚdecimalÚ.Ú*rz z func, col)r3 r )rE r c C sî t dggdgdj}t||d| ||d dd¡}d|d d | d ksNtt d
|
ggd
|
gdj}t||d| ||d dd¡}d|d d | d kstt d ggd gdj}t||d| ||d dd¡}d|d d | d ksêtd S ) Nçnº5?.A©r r )Ú thousandsr| r rz TZ 1_000_000r/ r r& i@B ù ð?nº5?.A©r r[ rd r4 r6 )r| r rz re rH r r r r r Útest_format_thousands? sB ÿ þ ÿ þ ÿ þr z{:,.4f}r ú,r c C s t dggdgdj}t||d| ||d dd¡}d|d d | d ksNtt d
|
ggd
|
gdj}t||d| ||d dd¡}d|d d | d kstd S ) Nr r r )r r| r rz TZ000_123r/ r r& r r )r| r rz re rH r r r r r Útest_format_decimalW s, ÿ þ ÿ þr c C sZ d} t jt| d tdd W 5 Q R X t jt| d tdg W 5 Q R X tdd d S )Nz2`escape` only permitted in {'html', 'latex'}, got r~ ÚtextZ
|
bad_escaper} )r r Ú
|
ValueErrorr ©Úmsgr r r Útest_str_escape_errori s r¢ c C s t ddgdd gddgd} | j dd¡}|d d d d
|
d ksFttd d 0 | j dd¡}|d d d d
|
d ks|tW 5 Q R X |d d d d
|
dks¢ttdd0 | j dd¡}|d d d d
|
dksØtW 5 Q R X tdd 2 | j dd¡}|d d d d
|
dkstW 5 Q R X |d d d d
|
dksBttdd2 | j dd¡}|d d d d
|
dksztW 5 Q R X |d d d d
|
dks¢t|d d d d
|
dksÀttdd2 | j dd¡}|d d d d
|
dksøtW 5 Q R X tdd 2 | j dd¡}|d d d d
|
d ks:tW 5 Q R X td d d i2 | j dd¡}|d d d d
|
d!kstW 5 Q R X d S )"NiÐ r g¾/Ý$ð?z&<z&~)ÚintÚfloatÚstrTr/ rK r& r_ zstyler.format.na_repÚMISSINGr z1.009000zstyler.format.decimalr Z1_009000zstyler.format.precisionz1.01Ú2000zstyler.format.thousandsZ2_000r0 zstyler.format.escaperi z &<rj z\&\textasciitilde ústyler.format.formatterr£ z{:,.2f}z2,000.00)r r[ r4 r6 r )r r9 Z ctx_with_opr r r Útest_format_optionst s8 & & ( ( ( (r© c C sT t | dd}| dd¡}|d d d d dks4t|d d d d dksPtd S ) Nr r Tr/ rK r& z-1r )r
|
r4 r6 )r r r9 r r r Útest_precision_zero s rª zformatter, expc C s | dS )Nz.3fr )r r r r r@ § rA r@ z9.000)rQ z9.00rw z9.0)NÚ9c C s@ t dgg}td| d| d|j ¡ ks2tW 5 Q R X d S )Né r¨ ú )r r r[ Úto_latexr6 )r| rp r r r r Ú test_formatter_options_validator¤ s
|
r¯ c
|
C sH d} t jt| d, tdddg t j ¡ W 5 Q R X W 5 Q R X d S )Nz Value must be an instance ofr~ r¨ ÚbadÚtype)r r r r r r[ r® r r r r Ú test_formatter_options_raises³ s r² c C sª t jddggdgd} td| ddgd}|j dd¡}|d d d d
|
d ksRt|d d d d dksnt|d d d d
|
d kst|d d d d dks¦td S )Nr rK Ú r éÿÿÿÿr )r r Tr/ r& Ú1Z
|
is_visibleÚ2)r r" r r[ r4 r6 )r r r9 r r r Útest_1level_multiindexº s r· c C sX t ddgg} | j dd¡}|d d d d dks8t|d d d d dksTtd S )NTFr/ r r r& rK )r r[ r4 r6 r\ r r r Útest_boolean_formatÅ s r¸ z hide, labelsc C s@ |r| j dddgd tjtdd | j|d W 5 Q R X d S )Nr ©r r ©r r ©rB rZ z"``labels`` must be of length equalr~ ©Úlabels)Úhider r r Ú relabel_index)r$ r¾ r½ r r r Útest_relabel_raise_lengthÍ s rÀ c C sÎ ddg}| j dddgd | j|d | dd¡}d d
|
d ¡ |d d d ¡ ksXtd dd ¡ |d d d
|
¡ ks~tddd ¡ |d d
|
d ¡ ks¤tddd ¡ |d d
|
d
|
¡ ksÊtd S )N©r rK ©r0 r r r¹ rº r» r¼ Tr r ©Úvaluer& r/ r rK r r0 r r ©r¾ r¿ r4 Úitemsr6 ©r$ r½ r9 r r r Útest_relabel_indexÛ s &&&rÈ c C sÐ ddg}| j dddgd | jd|d | dd¡}d dd
|
¡ |d d d ¡ ksZtdd d
|
¡ |d d d ¡ kstddd
|
¡ |d d d ¡ ks¦tddd
|
¡ |d d d ¡ ksÌtd S )NrÁ r r )r r )r r r» )rB r½ Tr rà rC r r0 r r r rK r rÅ rÇ r r r Útest_relabel_columnsæ s &&&rÉ c C sj | ddg¡ | dd¡}ddd ¡ |d d d ¡ ks@tddd ¡ |d d d ¡ ksftd S ) Nz{}Tr rà r/ r r r )r¿ r4 rÆ r6 r8 r r r Útest_relabel_roundtripñ s &rÊ )5Únumpyr! r Zpandasr r r r r r r Z importorskipZpandas.io.formats.styler
|
Z pandas.io.formats.style_renderr Zfixturer r r# r$ r: ÚmarkZ parametrizerI rM rP rS rV r] r` rc rh rt rv r{ r r
r r r r r¢ r© rª r¯ r² r· r¸ rÀ rÈ rÉ rÊ r r r r Ú<module> sÌ $
|
|
|
|
|
|
ýþ
|
þþ
|
ýþ
|
øþ )
|
|
üþ
|
|
þþ
|
|
