U
|
ý°djd ã @ s d Z ddlZddlZddlZddlm mZ ddl Z
|
ddl m Z ddl m Z ddlmZmZ ddlmZ ejG dd deZdS )z Test cases for DataFrame.plot é N)Ú DataFrame)Ú TestPlotBaseÚ_check_plot_works)ÚVersionc @ sv e Zd Zej ddddddddd d
|
d g
|
¡d d Zdd Zej dddg¡dd Zdd Z dd Z
|
ej dddgd fdd dd gdd dd gfg¡d d Z d d! Z d"d# Z d$d% Zd&d' Zd(d) Zej d*d+d,g¡d-d. Zd/d0 Zd1d2 Zd3d4 Zd5d6 Zd7d8 Zd9d: Zd;d< Zejd=d> Zejd?d@ Z dAdB Z dCdD Z dEdF Z dGdH Z dIdJ Z!dKdL Z"dMdN Z#dOdP Z$d+S )QÚTestDataFrameColorÚcolorZC0ZC1ZC2ZC3ZC4ZC5ZC6ZC7ZC8ZC9c C s. t tj dd¡dddgd}t|j|d d S )Né
|
é ÚaÚbÚc©Úcolumns©r )r ÚnpÚrandomÚrandnr Úplot©Úselfr Údf© r úcd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/plotting/frame/test_frame_color.pyÚtest_mpl2_color_cycle_str s z,TestDataFrameColor.test_mpl2_color_cycle_strc C s& t ddddgi}t|jdgd d S )NÚAé é r Úredr ©r r r ©r r r r r Ú test_color_single_series_list s z0TestDataFrameColor.test_color_single_series_list)r r r )r r r ç à?c C s, t ddgddgd}t|jdd|d d S ) Nr r r é ©ÚxÚyr$ r% ©r$ r% r r r r r r Útest_rgb_tuple_color" s z'TestDataFrameColor.test_rgb_tuple_colorc C s< t tj dd¡}tjtdd |jdd W 5 Q R X d S )Nr r zInvalid color argument:©ÚmatchÚ r )r r r r ÚpytestÚraisesÚ
|
ValueErrorr r r r r Útest_color_empty_string( s z*TestDataFrameColor.test_color_empty_stringc C s¤ t ddgddgd}|jddgdd gd
|
}d d |jD }|dd gksLtd d |jD }|ddgksltd}tjt|d |jddgddgd
|
W 5 Q R X d S )Nr r r r" r# r Zblackú-ú--©r Ústylec S s g | ] }| ¡ qS r ©Z get_linestyle©Ú.0Úliner r r Ú
|
<listcomp>3 s zETestDataFrameColor.test_color_and_style_arguments.<locals>.<listcomp>c S s g | ] }| ¡ qS r ©Ú get_colorr4 r r r r7 6 s zCannot pass 'style' string with a color symbol and 'color' keyword argument. Please use one or the other or pass 'style' without a color symbolr( zk-zr--)r r ÚlinesÚAssertionErrorr+ r, r- )r r ÚaxZ linestyler Úmsgr r r Ú test_color_and_style_arguments- s ÿz1TestDataFrameColor.test_color_and_style_argumentszcolor, expectedÚgreenr" Zyellowr Úbluec C sn t tj d¡}|j|dd}dd |jD }||ks:ttdd |jD sRttdd |jD sjtd S ) N)é r" zd--r1 c S s g | ] }| ¡ qS r r8 ©r5 Úir r r r7 N s z<TestDataFrameColor.test_color_and_marker.<locals>.<listcomp>c s s | ]}| ¡ d kV qdS )r0 Nr3 rB r r r Ú <genexpr>Q s z;TestDataFrameColor.test_color_and_marker.<locals>.<genexpr>c s s | ]}| ¡ d kV qdS )ÚdN)Z
|
get_markerrB r r r rD R s )r r r r r: r; Úall)r r Úexpectedr r< Úresultr r r Útest_color_and_markerB s z(TestDataFrameColor.test_color_and_markerc s° dd l m} | |j¡}ttj dd¡}|j ¡ }| j
|
|j d d d
|d d
d t ¡ d}|jj |d}| j
|
|j d d d
|d t ¡ ddlm |jj dd} fd d
|
t dd d¡D }| j
|
|j d d d
|d t ¡ |jj jd} fd d
|
t dd d¡D }| j
|
|j d d d
|d t ¡ |jd d
dgf jj d d}| j
|
|j d gd gd t ¡ |jddd}| j
|
|j d d d
dgd d t ¡ d S )Nr é ©Z
|
facecolorsÚrgcbyr ©ÚcmÚjet©Úcolormapc s g | ]} |¡qS r ©rO ©r5 ÚnrM r r r7 g s z6TestDataFrameColor.test_bar_colors.<locals>.<listcomp>r c s g | ]} |¡qS r rR rS rM r r r7 m s Ú
|
DodgerBlueÚbarr? ©Úkindr )Úmatplotlib.pyplotÚpyplotÚ_unpack_cyclerÚrcParamsr r r r r rV Ú _check_colorsÚpatchesÚtmÚcloseÚ
|
matplotlibrN ÚlinspacerO Úloc)r ÚpltÚdefault_colorsr r< Ú custom_colorsÚ rgba_colorsr rM r Útest_bar_colorsT s2
|
" z"TestDataFrameColor.test_bar_colorsc C sb t tdtddddddgd}|jjd|d d }d
|
d |jD }d d d d g}||ks^td S )Nr" r rJ r r@ )r ÚBr r r )r% r c S s g | ] }| ¡ qS r ©Ú get_facecolor)r5 Úpr r r r7 s z;TestDataFrameColor.test_bar_user_colors.<locals>.<listcomp>)ç ð?ç rn rm )rn rn rm rm )r Úranger rV r^ r; )r r r< rH rG r r r Útest_bar_user_colorsy s ÿüz'TestDataFrameColor.test_bar_user_colorsc C sÌ t j d¡}t|dddgd}|jjddd}|jjdddd}dd |j ¡ D }d
|
d |j ¡ D }||ksttd d |j ¡ D }d d |j ¡ D }||ks¨t|j ¡
|
¡ |j ¡
|
¡ ksÈtd S ) N©iè r úA labelúB labelúC labelr r# ©r$ r% r c S s g | ] }| ¡ qS r ©Ú get_visible©r5 Zvisr r r r7 s z\TestDataFrameColor.test_if_scatterplot_colorbar_affects_xaxis_visibility.<locals>.<listcomp>c S s g | ] }| ¡ qS r rv rx r r r r7 s c S s g | ] }| ¡ qS r rv rx r r r r7 s c S s g | ] }| ¡ qS r rv rx r r r r7 s ) r r r r ÚscatterÚxaxisÚget_minorticklabelsr; Úget_majorticklabelsÚ get_labelrw )r Ú random_arrayr Zax1Úax2Zvis1Zvis2r r r Ú5test_if_scatterplot_colorbar_affects_xaxis_visibility s ÿzHTestDataFrameColor.test_if_scatterplot_colorbar_affects_xaxis_visibilityc C s~ t j d¡}t|dddgd}|jjdddd}tdd |j ¡ D sLttd
|
d |j ¡ D sht|j
|
¡ ¡ sztd S ) Nrq rr rs rt r é )Zgridsizec s s | ]}| ¡ V qd S ©Nrv rx r r r rD § s zKTestDataFrameColor.test_if_hexbin_xaxis_label_is_visible.<locals>.<genexpr>c s s | ]}| ¡ V qd S r rv rx r r r rD ¨ s ) r r r r ÚhexbinrF rz r{ r; r| r} rw )r r~ r r< r r r Ú%test_if_hexbin_xaxis_label_is_visible s z8TestDataFrameColor.test_if_hexbin_xaxis_label_is_visiblec
|
C sú dd l m} tj d¡}t|dddgd}| dd¡\}}|jjddd|d d |jjddd|d d | ¡ t d
|
d |j
|
D ¡}|d d
d d
df }|dd d
f |dd d
f }|d d d
f |dd d
f } tj || d d ¡ söt d S )Nr rq rr rs rt r r r )r r< c S s g | ]}| ¡ ¡ qS r )Ú get_positionZ
|
get_points)r5 r< r r r r7 ¶ s z\TestDataFrameColor.test_if_scatterplot_colorbars_are_next_to_parent_axes.<locals>.<listcomp>r gH¯¼ò×z>)Zatol)rY rZ r r r Úsubplotsr ry Z tight_layoutÚarrayÚaxesÚiscloserF r; )
|
r rd r~ r Úfigr ZpointsZ axes_x_coordsZparent_distanceZcolorbar_distancer r r Ú5test_if_scatterplot_colorbars_are_next_to_parent_axes« s zHTestDataFrameColor.test_if_scatterplot_colorbars_are_next_to_parent_axesÚcmapNZGreysc C s¤ t ddgddgddgddgddggd d
|
gd }d d d d dg|d< |d k rxtjtdd |jjdd|dd}W 5 Q R X n|jjddd|d}|jd jd ks td S )Ngffffff@g @g@g @g @g @g@g@ÚlengthÚwidthr ÚrÚgr ZspeciesF)Zcheck_stacklevelr r )r$ r% r r )r$ r% r r ) r r_ Zassert_produces_warningÚ UserWarningr ry Ú collectionsÚcolorbarr; )r r r r< r r r Ú+test_scatter_with_c_column_name_with_colors¼ s þ z>TestDataFrameColor.test_scatter_with_c_column_name_with_colorsc C sà t dddgdddgdddgd}tjtdd |jjddd d
|
d W 5 Q R X | | jj¡}|jjddd d }t
|
|j d ¡ d t | j |d ¡¡¡ |jjdddd}t
|
|j d ¡ d t jddddgt jd¡ d S )Nr r r )r
|
r r z&Specify exactly one of `c` and `color`r( r
|
r r r? )r$ r% r r ru r Zwhiter& )Zdtype)r r+ r, Ú TypeErrorr ry r[ rd r\ r_ Zassert_numpy_array_equalr rk r r ZcolorconverterZto_rgbaÚfloat64)r r re r< r r r Útest_scatter_colorsÌ s " þþz&TestDataFrameColor.test_scatter_colorsc C s¦ dd l m} tdddgdddgdddgd}|d d |d< | ¡ \}}|jddd d
|
d |d |jddd d
|
d |d |jd jjd kst|jd jjd ks¢td S )Nr r r r ru r$ Zx2r% r ry Zcividis)r rX r r< Zmagma) rY rZ r r r r r Únamer; )r rd r r r< r r r Ú$test_scatter_colorbar_different_cmapß s " z7TestDataFrameColor.test_scatter_colorbar_different_cmapc s° ddl m d}ttj dd¡}|j|d}| j| ¡ |d t
|
¡ |j|d}| ¡ }t | ¡ |D ] \}}| ¡ | ¡ ksjt qjt
|
¡ |jdd} fd d
|
t dd t|¡D }| j| ¡ |d t
|
¡ |j jd} fd d
|
t dd t|¡D }| j| ¡ |d t
|
¡ |jd d
dgf jd d}| j|jd gd |jdd}| j| ¡ dgd d t
|
¡ dddddg}|j|d}| j| ¡ |d t
|
¡ d S )Nr rM rL rJ r ©Ú
|
linecolorsrO rP c s g | ]} |¡qS r rR rS rM r r r7 s z7TestDataFrameColor.test_line_colors.<locals>.<listcomp>r c s g | ]} |¡qS r rR rS rM r r r7 s rU r ú#FF0000ú#0000FFú#FFFF00ú#000000ú#FFFFFF)ra rN r r r r r r] Ú get_linesr_ r` Úzipr9 r; rb ÚlenrO rc r: ) r rf r r< r Zlines2Úl1Úl2rg r rM r Útest_line_colorsí s8 z#TestDataFrameColor.test_line_colorsc C s8 dddg}t tj dd¡j|d t|dks4td S )Nr r r r r r r )r r r Úrandr r£ r; )r Úcolorsr r r Útest_dont_modify_colors s
|
z*TestDataFrameColor.test_dont_modify_colorsc
|
sÜ ddl m | | jj¡}ttj dd¡}|j dd}t
|
|t |D ] \}}| j | ¡ |gd qFt ¡ |j ddd}|D ]}| j | ¡ dgd q~t ¡ |j dd d}|D ]}| j | ¡ d gd q²t ¡ d
|
}|j |dd }t
|
|t |D ] \}}| j | ¡ |gd qôt ¡ |j t |dd }t
|
|t |D ] \}}| j | ¡ |gd q:t ¡ d d dddg}|j |dd }t
|
|t |D ] \}}| j | ¡ |gd qt ¡ fddt ddt|¡D }d jfD ]F}|j |dd}t
|
||D ] \}}| j | ¡ |gd qút ¡ qÞ|jd d
dgf j ddd }| j |d jdgd |j ddd}|D ]}| j | ¡ dgd qlt ¡ t d
|
} |j | dd}t
|
|| D ] \}}| j | ¡ |gd q°t ¡ d S )Nr rM rJ T)r r Úk)r r r? rL )r r r r r r r c s g | ]} |¡qS r rR rS rM r r r7 I s zKTestDataFrameColor.test_line_colors_and_styles_subplots.<locals>.<listcomp>r rO )rQ r rU r )r2 r ©ra rN r[ rd r\ r r r r r r¢ Úlistr] r¡ r_ r` rb r£ rO rc r: ©
|
r re r r r< r rf rg r Zstylesr rM r Ú$test_line_colors_and_styles_subplots s\ z7TestDataFrameColor.test_line_colors_and_styles_subplotsc sô ddl m ddlm d}ttj dd¡}|jj |d}| j
|
| ¡ |d fdd | ¡ D }| j
|
||d
|
| ¡ \}}| j
|
||d
|
|D ]}| ¡ d kstqt ¡ |jj d d }fd d t ddt|¡D }| j
|
| ¡ |d fdd | ¡ D }| j
|
||d
|
| ¡ \}}| j
|
||d
|
|D ]}| ¡ d ks0tq0t ¡ |jj jdd}| j
|
| ¡ |d fdd | ¡ D }dd |D } | j
|
|| d
|
| ¡ \}}| }
|
| j
|
|d t|
|
|
d |D ]}| ¡ dksÖtqÖd S )Nr rM ©ÚPolyCollectionrL rJ r r c s g | ]}t | r|qS r ©Ú
|
isinstance©r5 Úor¯ r r r7 k s
|
z7TestDataFrameColor.test_area_colors.<locals>.<listcomp>rK rO rP c s g | ]} |¡qS r rR rS rM r r r7 v s r c s g | ]}t | r|qS r r± r³ r¯ r r r7 x s
|
F)rQ Zstackedc s g | ]}t | r|qS r r± r³ r¯ r r r7 s
|
c S s$ g | ] }|d |d |d dfqS )r r r r! r ©r5 r r r r r7
s r! )ra rN Zmatplotlib.collectionsr° r r r r§ r Úarear] r¡ Z get_childrenZget_legend_handles_labelsZ get_alphar; r_ r` rb r£ rO ) r rf r r< ZpolyÚhandlesÚlabelsÚhÚ
|
jet_colorsZjet_with_alphar r )r° rN r Útest_area_colorsb sB z#TestDataFrameColor.test_area_colorsc s | | jj¡}ttj dd¡}|j ¡ }| j |j
|
d d d
|d d
d t ¡ d}|jj|d}| j |j
|
d d d
|d t ¡ ddl m |jjdd } fd
|
d t dd d¡D }| j |j
|
d d d
|d t ¡ |jj jd } fd d t dd d¡D }| j |j
|
d d d
|d t ¡ |jd d
dgf jjdd}| j |j
|
d gdgd |jddd}| j |j
|
d d d
dgd d t ¡ d S )NrJ r rK rL r r rM rO rP c s g | ]} |¡qS r rR rS rM r r r7 s z7TestDataFrameColor.test_hist_colors.<locals>.<listcomp>r c s g | ]} |¡qS r rR rS rM r r r7 ¥ s rU Úhistr? rW )r[ rd r\ r r r r r r¼ r] r^ r_ r` ra rN rb rO rc )r re r r< rf rg r rM r Útest_hist_colors s.
|
" z#TestDataFrameColor.test_hist_colorsc sØ ddl m d}ttj dd¡}|jj|d}| j| ¡ |d t
|
¡ |jjdd} fd d
|
t dd t |¡D }| j| ¡ |d t
|
¡ |jj jd} fd d
|
t dd t |¡D }| j| ¡ |d d S ) Nr rM rL rJ r r rO rP c s g | ]} |¡qS r rR rS rM r r r7 ¼ s z6TestDataFrameColor.test_kde_colors.<locals>.<listcomp>r c s g | ]} |¡qS r rR rS rM r r r7 Á s )ra rN r r r r§ r Úkder] r¡ r_ r` rb r£ rO )r rf r r< rg r rM r Útest_kde_colors° s z"TestDataFrameColor.test_kde_colorsc
|
sR ddl m | | jj¡}ttj dd¡}|j ddd}t
|
|t |D ] \}}| j | ¡ |gd qHt ¡ |j dddd }|D ]}| j | ¡ dgd qt ¡ |j dd
|
dd }|D ]}| j | ¡ d
|
gd q¸t ¡ d }|j d|dd }t
|
|t |D ] \}}| j | ¡ |gd qüt ¡ fd d t ddt|¡D }d jfD ]H}|j d|dd}t
|
||D ] \}}| j | ¡ |gd qjt ¡ qL|jd d
dgf j dddd }| j |d jdgd |j dddd}|D ]}| j | ¡ dgd qàt ¡ t d } |j d| dd}t
|
|| D ] \}}| j | ¡ |gd q&t ¡ d S )Nr rM rJ r¾ T)rX r r rª )rX r r r rL c s g | ]} |¡qS r rR rS rM r r r7 ã s zJTestDataFrameColor.test_kde_colors_and_styles_subplots.<locals>.<listcomp>r rO )rX rQ r rU r )rX r2 r r« r r rM r Ú#test_kde_colors_and_styles_subplotsÄ sJ z6TestDataFrameColor.test_kde_colors_and_styles_subplotsc s d(fdd } jj¡}ttj dd¡}|jjdd}|||d |d |d |d t
|
¡ d d
|
d d d }|jj|ddd}|||d |d |d |d d t
|
¡ ddd}|jj|dd}|||d dd|d t
|
¡ ddl m |jjddd} fd d t dd d ¡D }|||d |d |d |d t
|
¡ |jj jdd}|||d |d |d |d t
|
¡ |jjd dd}||d d d d |jjd!d dd}||d!d!d!d!d t d"¡}tjt|d# |jjd$d%d&d' W 5 Q R X d S ))Nrª c s° |d kr d} j | d |gt| d d j | d |gt| d d j | d |gt| d d j | d |gt| d d j | d |gt| d d d S )Nrª Úboxesr ÚwhiskersÚmediansZfliersÚcaps)r] r£ )ÚbpZbox_cZ
|
whiskers_cZ medians_cZcaps_cZfliers_c)r r r r] ý s ÿ ÿ z=TestDataFrameColor.test_boxplot_colors.<locals>._check_colorsrJ Údict)Ú return_typer r z#572923z#982042z#804823z#123456)rÁ r rà rÄ zr+)r ÚsymrÇ rÁ r rà rÄ r r Úm)r rà )r rÇ rM rO )rQ rÇ c s g | ]} |¡qS r rR rS rM r r r7 3 s z:TestDataFrameColor.test_boxplot_colors.<locals>.<listcomp>r r rU )r r r zgcolor dict contains invalid key 'xxxx'. The key must be either ['boxes', 'whiskers', 'medians', 'caps']r( r r@ )rÁ Zxxxxr )rª N)r[ rd r\ r r r r r Zboxr_ r` ra rN rb rO ÚreÚescaper+ r, r- )r r] re r rÅ Z dict_colorsrº r= r )rN r r Útest_boxplot_colorsü s` ûüú
|
ÿz&TestDataFrameColor.test_boxplot_colorsc C sr dd l }dd lm} td}| d|¡|jd< ttj dd¡}| ¡ }|
|
|j¡d d
}| j | ¡ |d d S )Nr Zrgbkr úaxes.prop_cyclerJ r r ) ÚcyclerrY rZ r¬ r\ r r r r r r[ r] r¡ )r rÎ rd r¨ r r< rG r r r Útest_default_color_cycleL s z+TestDataFrameColor.test_default_color_cyclec C sb t tjjddtjjddt d¡tjjdd d}|jjddd d}|jd jd ks^t d S )Né ©Úsize©r ri ÚCr ri )r$ r% r r )
|
r r r ÚuniformÚaranger r r r r; )r r r< r r r Útest_no_color_barY s ýÿz$TestDataFrameColor.test_no_color_barc C sn t tjjddtjjddt d¡tjjdd d}d}tjt|d |jj dddd d
|
W 5 Q R X d S ) NrÐ rÑ rÓ z)Only specify one of `cmap` and `colormap`r( r ri ZYlGnZBuGn)r$ r% r rQ )
|
r r r rÕ rÖ r+ r, r r r ©r r r= r r r Ú$test_mixing_cmap_and_colormap_raisesd s ýÿz7TestDataFrameColor.test_mixing_cmap_and_colormap_raisesc C sT dd l }dddg}|j |¡}tdddggjd|d }|d
|
d |jD ksPtd S ) Nr ©çÍÌÌÌÌÌì?r r r ©r rÛ r r ©r r rÛ r r r r rV )rX r c S s g | ] }| ¡ qS r rj rµ r r r r7 v s z=TestDataFrameColor.test_passed_bar_colors.<locals>.<listcomp>)ra r¨ ZListedColormapr r r^ r; )r ÚmplÚ color_tuplesrQ Úbarplotr r r Útest_passed_bar_colorsp s
|
|
z)TestDataFrameColor.test_passed_bar_colorsc C sj dd l }dddg}|jd| d|¡id tdd d
|
ggjd d }W 5 Q R X |d d |jD ksftd S )Nr rÚ rÜ rÝ rÍ r )Úrcr r r rV )rX c S s g | ] }| ¡ qS r rj rµ r r r r7 ~ s z?TestDataFrameColor.test_rcParams_bar_colors.<locals>.<listcomp>)ra Z
|
rc_contextrÎ r r r^ r; )r rÞ rß rà r r r Útest_rcParams_bar_colorsx s
|
|
"z+TestDataFrameColor.test_rcParams_bar_colorsc C s¤ dd l }tdddgdddgd}tddddgi}tj||gdd }| ¡ }| ¡ }t|jtd
|
k rp|j}n|j }t
|
||j D ] \}}| ¡ | ¡ kst qd S ) Nr r r r )r r
|
r
|
r" é )Zaxisz3.7)ra r ÚpdÚconcatr Z
|
get_legendr Ú __version__Z legendHandlesZlegend_handlesr¢ r: r9 r; ) r rÞ r Zdf1Z df_concatrH Zlegendr· r6 r r r Ú%test_colors_of_columns_with_same_name s z8TestDataFrameColor.test_colors_of_columns_with_same_namec C sL t tj dd¡ddgd}d}tjttf|d |jdd W 5 Q R X d S )
|
Nr r r ri r z0(is not a valid value)|(is not a known colormap)r( Zinvalid_colormaprP ) r r r r r+ r, r- ÚKeyErrorr rØ r r r Útest_invalid_colormap s z(TestDataFrameColor.test_invalid_colormap)%Ú__name__Ú
|
__module__Ú __qualname__r+ ÚmarkZ parametrizer r r' r. r> rI rh rp r r r r r r r¦ r© r® r» r½ ÚtdZskip_if_no_scipyr¿ rÀ rÌ rÏ r× rÙ rá rã rè rê r r r r r sV ÿ
|
|
þþ
|
%
|
,D,"
|
|
7P r )Ú__doc__rÊ Únumpyr r+ Z pandas.util._test_decoratorsÚutilZ_test_decoratorsrï Zpandasrå r Zpandas._testingZ_testingr_ Z pandas.tests.plotting.commonr r Zpandas.util.versionr Zskip_if_no_mplr r r r r Ú<module> s
|
