U
|
ý°dÚ+ ã @ sì d dl Zd dlZd dlmZmZmZ e d¡ e d¡ d dlZ d dl
|
m Z ej dd Z ej dd Zej d
|
d Zej d d Zej dddg¡dd Zej dddg¡dd Zej dd ddddgfdddddgfdddddgfg¡ej dddg¡dd Zej d d d d gd d!gd"fd#d$d gd%d!gd&fdd'd!gd(d gd)fg¡d*d+ Zd,d- Zd.d/ Zej d0d dd1gd2d gd3d!gd2d gd3d!gd4fddd1gd2d gd2d gd3d!gd3d!gd4fde d1dgdd1gg¡d3d!gd2d gd2d gd3d!gd4fg¡d5d6 Zej d7dd1d8gd fdd1gdfe dd1gdd1gg¡dfg¡d9d: Zej d;ed1dgdd1ggd<d=gd>d?gd@ed1dgdd1ggd=d<gd?d>gd@edd1gd1dggd<d=gd?d>gd@edd1d8gd1dd8ggd=d<dAgd>d?gd@edd1gd1dgd8d8ggd=d<gd>d?dBgd@g¡ej dCddd1gd1dggfd=gdgd1ggfd<d=gd1dgdd1ggfed>dd
f dd1ggfed?d>gdd
f d1dgdd1ggfg¡dDdE Z ej dFed1dgd?d>gdGd ddgd1d1ggfed1dgd<d=gdGddd1gdd1ggfedd1d8gd>d?dBgdGd ddgd1d1ggfedd1d8gd=d<dAgdGddd1gdd1ggfg¡dHdI Z ej d7edd1gd1dggd=d<gd>d?gd@dfedd1gd1dggd=d<gd>d?gd@d fg¡dJdK Z dLdM Z dNdO Z ej dPd e j!d g¡dQdR Z"dSdT Z#ej dUdVdWg¡dXdY Z$dS )Zé N)Ú DataFrameÚ
|
IndexSliceÚSeriesÚ
|
matplotlibZjinja2©ÚStylerc C s t ddgddggddgdS )Né é é ÚAÚB)Úcolumns©r © r r údd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/io/formats/style/test_matplotlib.pyÚdf s r c C s t | ddS ©Nr )Zuuid_lenr )r r r r Ústyler s r c C s" t ddgddggddgddgdS )Nr r r ÚXÚY©r Úindexr r r r r Údf_blank s r c C s t | ddS r r )r r r r Ú styler_blank! s r ÚfÚbackground_gradientÚ text_gradientc C sf dD ]\}t | ||d ¡ j}tdd | ¡ D s8t|d |d ksLt|d |d kstqd S ) N)NÚYlOrRd)Úcmapc s s | ]}d |d d kV qdS )ú#r r Nr )Ú.0Úxr r r Ú <genexpr>* s z)test_function_gradient.<locals>.<genexpr>©r r ©r r ©r r ©r r )ÚgetattrÚ_computeÚctxÚallÚvaluesÚAssertionError)r r Zc_mapÚresultr r r Útest_function_gradient& s
|
r. c C sV t | |td d ¡ j}|dkr8|d ddgksRtn|dkrR|d dgksRtd S ) N)r r )Úsubsetr r% ©úbackground-colorú#fff7fb©Úcolorz#000000r )r4 r2 )r' r r( r) r, )r r r- r r r Ú test_background_gradient_color/ s
|
r5 zaxis, expectedÚlowÚhighr Úmidc C s |dkr"ddgddgddgd}n|dkr<d gd
|
gd gd}t | |d |d ¡ j}tddddgD ] \}}|| ||| ksdtqdd S )Nr )r1 ú#f7fbffr3 )r1 ú#abd0e6)r1 ú#08306b©r4 z#f1f1f1)r6 r8 r7 r )r4 r9 )r4 r: )r4 r; ZBlues©r Úaxisr# r$ r% r& )r' r( r) Ú enumerater, )r r> Úexpectedr Úcolorsr- ÚiÚcellr r r Ú test_background_gradient_axis8 s
|
ýýrD zcmap, expectedZPuBu)r1 z#86b0d3r3 )r1 z#83afd3r< ))r
|
é )r
|
é r )r1 z#fd913e)r1 z#fd8f3d))r
|
é )r
|
é )r1 z#48c16e)r1 z#4cc26c))é r )rI r c C sR t t d¡ dd¡}|jj| d d ¡ j}| ¡ D ]}|| || ks4t q4d S )Néd é
|
r= )
|
r ÚnpZarangeZreshapeÚstyler r( r) Úkeysr, )r r@ r r- Úkr r r Útest_text_color_thresholdS s rP c C sN t td} | jjddd ¡ j}|d |d ks6t|d |d ksJtd S ) NrE r é )ZvminZvmaxr# r% )r
|
r )rQ r )r ÚrangerM r r( r) r, )r r) r r r Ú"test_background_gradient_vmin_vmaxu s rS c C s t td ¡ } t tddd ¡ }| j ¡ ¡ j}|j ¡ ¡ j}|d |d ksXt|d |d kslt|d |d kstd S )NrQ ÚInt64©Zdtyper# r% )r r )r rR Zto_framerM r r( r) r, ©Zdf1Zdf2Zctx1Zctx2r r r Ú test_background_gradient_int64} s rW zaxis, gmap, expectedr r0 )r1 ú#023858©r# r% r$ r& c C s$ | j ||d ¡ j}||ks td S ©N)r> Úgmap©r r( r) r, )r r> r[ r@ r- r r r Ú#test_background_gradient_gmap_array s %r] z
|
gmap, axisrQ c C sN t dddgdddgg}d}tjt|d |jj| |d ¡ W 5 Q R X d S )Nr z$supplied 'gmap' is not correct shape©Úmatch©r[ r> )r ÚpytestÚraisesÚ
|
ValueErrorrM r r( )r[ r> r Úmsgr r r Ú*test_background_gradient_gmap_array_raises± s re r[ r r r r r ÚCÚZzsubset, exp_gmapc C s< | j d ||d}| j d ||d}| ¡ j| ¡ jks8td S )N)r> r[ r/ r\ )r r[ r/ Úexp_gmapr@ r- r r r Ú-test_background_gradient_gmap_dataframe_align¼ s ri zgmap, axis, exp_gmap©r c C s8 | j d |d ¡ }| j ||d ¡ }|j|jks4td S rZ r\ )r r[ r> rh r@ r- r r r Ú*test_background_gradient_gmap_series_aligná s rk c C s4 d}t jt|d | j||d ¡ W 5 Q R X d S )NzD'gmap' is a DataFrame but underlying data for operations is a Seriesr^ r` )ra rb rc r r( )r r[ r> rd r r r Ú-test_background_gradient_gmap_wrong_dataframeñ s rl c C sH d}t ddgddgd}tjt|d | j|d d ¡ W 5 Q R X d S ) NzD'gmap' is a Series but underlying data for operations is a DataFramer r r r rj r^ r` )r ra rb rc r r( )r rd r[ r r r Ú*test_background_gradient_gmap_wrong_seriesÿ s rm c C sb t dgdgtjggtd} t dgdgd ggdd}| j ¡ ¡ j}|j ¡ ¡ j}||ks^td S )Nr r rU rT ) r rL ÚnanÚfloatrM r r( r) r, rV r r r Ú(test_background_gradient_nullable_dtypes s
|
rp r c C sf t ddgddgg}|jj| d d ¡ j}dddd d
|
}| ¡ D ] \}}||| d d ks@tq@d S ) Nr r rQ r
|
r= z#d0d1e6z#056fafz#73a9cfrX rY )r rM Úbarr( r) Úitemsr, )r Údatar) Z pubu_colorsrO Úvr r r Útest_bar_colormap s üru c C s d}t jt|d | jjddhd ¡ W 5 Q R X t jt|d | jjdddgd ¡ W 5 Q R X d}t jt|d | jjdd d
|
¡ W 5 Q R X d S ) Nz4`color` must be string or list or tuple of 2 stringsr^ ÚaÚb)r4 Úcz'`color` and `cmap` cannot both be givenZ somethingzsomething else)r4 r )ra rb rc rM rq Zto_html)r rd r r r Útest_bar_color_raises" s "ry Ú plot_methodZscatterÚhexbinc C sh t j dddgdddgg¡}| j| j | d< tddd|d}|dkrR| d¡|d< t| j|f | d S ) Nr r rx r r )r! Úyrx Zcolormapr{ rf ) ÚmplrA ZListedColormapr r ÚdictÚpopr' Zplot)r rz r Úkwargsr r r Útest_pass_colormap_instance. s r )%ÚnumpyrL ra Zpandasr r r Z importorskipr r} Zpandas.io.formats.styler Zfixturer r r r ÚmarkZ parametrizer. r5 rD rP rS rW Úarrayr] re ri rk rl rm rp Z colormapsru ry r r r r r Ú<module> s.
|
|
|
|
|
|
|
|
ýþ þþþþþþñþ
|
üý üý üýëþ#
|
0ÿ
|
ÿ ÿ ÿ ÿ ÿóþ"ûþ
|
$$((üþ
|
$$þþ
|
|
þ
|
þ
|
