U
|
ý°dLc ã @ s® d Z ddlZddlZddlZddlZddlm mZ ddl
|
m Z m Z m Z mZmZmZ ddlmZ ddlmZmZ ddlmZ e jG dd deZe jG dd d eZdS )
|
z Test cases for .boxplot method é N)Ú DataFrameÚ
|
MultiIndexÚSeriesÚ
|
date_rangeÚplottingÚtimedelta_range)Ú TestPlotBaseÚ_check_plot_works)Ú pprint_thingc @ sv e Zd Zdd Zejjdd Zdd Zdd Z d d
|
Z
|
d d Z d d Z dd Z dd Zdd Zej dddddddddddfddiddifddddddfg¡d d Zej d d d d d!d dfd"d#d#d$d#dfg¡d%d& Zej d'ddd(d)fg¡d*d+ Zej d,d-d.d/d0g¡d1d2 Zej d3d4d5g¡d6d7 Zej d3d4d5g¡d8d9 Zej d3d4d5g¡d:d; Zd<S )=ÚTestDataFramePlotsc C sØ dd l m} d}ttj ddddg|¡tj ddddg|¡tj ddddg|¡dt d|¡d}|jdd d
|
}d d | ¡ D |j
|
¡ kst | t ddd ¡¡ | ¡ dd | ¡ D tt ddd ¡ksÔt d S )Nr éP é é é )ZClinicalZ ConfirmedZ Discarded©ÚindexÚbarT)ÚkindZstackedc S s g | ]}t | ¡ qS © ©ÚintÚget_text©Ú.0Úxr r ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/plotting/test_boxplot_method.pyÚ
|
<listcomp>, s zDTestDataFramePlots.test_stacked_boxplot_set_axis.<locals>.<listcomp>é
|
c S s g | ]}t | ¡ qS r r r r r r r / s )Úmatplotlib.pyplotÚpyplotr ÚnpÚrandomÚchoiceZarangeÚplotÚget_xticklabelsr Zto_listÚAssertionErrorZ
|
set_xticksZdrawÚlist)ÚselfÚpltÚnÚdfÚaxr r r Ú test_stacked_boxplot_set_axis s ý
|
ú ÿz0TestDataFramePlots.test_stacked_boxplot_set_axisc C s\ t tj dd¡ttjd d
ddddgd}dd gd
|
|d < dd dgd |d < t|jdd t|jddgdd t j
|
t dd t|jddgd d W 5 Q R X t|jdd d gd t j
|
t dd t|jd d W 5 Q R X t j
|
t dd t|jd d gd W 5 Q R X tt j j|d dd t|jddd t j
|
t dd t|jd dd W 5 Q R X d S )Né é ÚoneÚtwoÚthreeÚfour©r ÚcolumnsÚfoor r Zindicr Zindic2Údict©Ú return_type©Úcolumnr8 F©Zcheck_stacklevel)r: Úby©r< )Údatar8 r )Únotchr8 )r< r? )r r r! Úrandnr& ÚstringÚ ascii_lettersr ÚboxplotÚtmÚassert_produces_warningÚ UserWarningr Z_core©r' r* r r r Útest_boxplot_legacy13 s(
|
ý z'TestDataFramePlots.test_boxplot_legacy1c C sz t tj dd¡ddgd}tddddddddddg
|
|d< tdgd |d < tjtd
|
d t|j dd W 5 Q R X | j
|
¡ \}}|j dd|d }|j }||ks¤t | j
|
¡ \}}| d ¡j |dd}|j }||d ksÜt | j
|
¡ \}}t t¡ |j ddgd|dd}W 5 Q R X |d ¡ |ks,t | j
|
¡ \}}|j |dd}ttj | ¡ ¡}t| ¡ t|ksvt d S )Nr r ZCol1ZCol2©r4 ÚAÚBÚXÚYFr; r= )r< r+ Úaxes)r+ r8 )r: r< r+ r8 r6 )r r r! Úrandr rD rE rF r rC r( ÚsubplotsrN r% ÚgroupbyZ
|
get_figurer& Ú itertoolsÚchainÚ from_iterableÚvaluesÚlenÚ get_lines)r' r* Úfigr+ rN Zax_axesÚdÚlinesr r r Útest_boxplot_legacy2L s4 ÿz'TestDataFramePlots.test_boxplot_legacy2c C s | ¡ }t|| jjstd S ©N)rC Ú
|
isinstancer( ZAxesr% )r' Úhist_dfÚresultr r r Ú test_boxplot_return_type_nonen s z0TestDataFramePlots.test_boxplot_return_type_nonec C sú t tj dd¡ttjd d
ddddgd}d}tjt |d |j
|
d
|
d W 5 Q R X |
|
¡ }| |d ¡ t d ¡ |j
|
dd }W 5 Q R X | |d¡ t d ¡ |j
|
d d }W 5 Q R X | |d ¡ t d ¡ |j
|
dd }W 5 Q R X | |d¡ d S )Nr- r. r/ r0 r1 r2 r3 z,return_type must be {'axes', 'dict', 'both'}©ÚmatchZ
|
NOT_A_TYPEr7 rN Fr6 Úboth)r r r! r@ r& rA rB ÚpytestÚraisesÚ
|
ValueErrorrC Ú_check_box_return_typerD rE )r' r* Úmsgr_ r r r Ú test_boxplot_return_type_legacys s&
|
ý z2TestDataFramePlots.test_boxplot_return_type_legacyc C sú dd }| ¡ }tj dd|jd ¡|d< |jddgd d
|
\}}||d | ||d | |j|ksjt|jdddgd d
|
}|d |d |d }}}|d }||d | ||d | ||d | |j|ksÚt|j|ksèt|jd ksötd S )Nc S s0 | ¡ \}}|| ¡ ks t|| ¡ ks,td S r\ )Zget_ylimÚminr% Úmax)Úcolr+ Zy_minZy_maxr r r Ú_check_ax_limits s zETestDataFramePlots.test_boxplot_axis_limits.<locals>._check_ax_limitsr é r ZageÚheightÚweightÚcategoryr= )r r )r r )r r ©r r )Úcopyr r! ÚrandintÚshaperC Z_shareyr% ) r' r^ rm r* Z height_axZ weight_axÚpZage_axZdummy_axr r r Útest_boxplot_axis_limits s z+TestDataFramePlots.test_boxplot_axis_limitsc C s8 t tj dd¡}tj|jd d
df< t|jdd d S )Nrn r. r rN r7 )r r r! r@ ÚnanÚlocr rC rG r r r Útest_boxplot_empty_column¨ s z,TestDataFramePlots.test_boxplot_empty_columnc C sV t tj dd¡dddddgd}|jd d
|
d }|jjjd ks@t|jjj d ksRtd S )Nr é rJ rK ÚCÚDÚErI rN )é é )r8 Zfigsizer r )
|
r r r! rO rC ÚfigureZ bbox_inchesÚwidthr% ro )r' r* r_ r r r Ú test_figsize s z TestDataFramePlots.test_figsizec C s6 t dddddddgi}| j|jddd ddd
|
d S ) NÚar r r r. r{ r- é )Úfontsize©Z
|
xlabelsizeZ
|
ylabelsize©r Z_check_ticks_propsrC rG r r r Ú test_fontsize³ s ÿz TestDataFramePlots.test_fontsizec
|
C s t tdddtj d¡tj d¡d tddd t¡tddddtdddd}|jd d
|
}d d | ¡ D d dgks|t
|
d S )Nz
|
2012-01-01éd )Úperiodsr ÚUTC)r Útzz1 days)r ÚbÚcrY ÚeÚfÚbox)r c S s g | ] }| ¡ qS r ©r r r r r r Æ s z@TestDataFramePlots.test_boxplot_numeric_data.<locals>.<listcomp>r r ) r r r r! r@ ÚastypeÚstrr r# r$ r% )r' r* r+ r r r Útest_boxplot_numeric_data¹ s
|
|
|
úÿ
|
z,TestDataFramePlots.test_boxplot_numeric_datazcolors_kwd, expectedÚrr Úgr )ÚboxesÚwhiskersÚmediansÚcapsr c C sN t tj dd¡}|j|dd}| ¡ D ] \}}|| d ¡ |ks(tq(d S )Nr r r6 ©Úcolorr8 r )r r r! rO rC ÚitemsÚ get_colorr% )r' Z
|
colors_kwdÚexpectedr* r_ ÚkÚvr r r Útest_color_kwdÈ s z!TestDataFramePlots.test_color_kwdzscheme,expectedZdark_backgroundz#8dd3c7z#bfbbd9Údefaultz#1f77b4z#2ca02cc C sf t tj dd¡}dd lm} |j |¡ |jj dd}|
|
¡ D ] \}}|| d ¡ |ks@t q@d S )Nr r r r6 r7 ) r r r! rO r r ÚstyleZuser# r r r r% )r' Úschemer¡ r* r( r_ r¢ r£ r r r Útest_colors_in_themeÚ s z'TestDataFramePlots.test_colors_in_themezdict_colors, msg)r Z invalid_keyzinvalid key 'invalid_key'c C s> t tj dd¡}tjt|d |j|dd W 5 Q R X d S )Nr r ra r6 r )r r r! rO rd re rf rC )r' Z dict_colorsrh r* r r r Útest_color_kwd_errorsû s z(TestDataFramePlots.test_color_kwd_errorszprops, expected)Zboxpropsr )Z whiskerpropsr )Zcappropsr )Z medianpropsr c C sN t dd dD }|ddii}|jf ddi|}|| d ¡ dksJtd S ) Nc S s i | ]}|t j d ¡qS )r )r r! )r r¢ r r r Ú
|
<dictcomp> s z?TestDataFramePlots.test_specified_props_kwd.<locals>.<dictcomp>ÚABCr ZC1r8 r6 r )r rC r r% )r' Úpropsr¡ r* Zkwdr_ r r r Útest_specified_props_kwd s z+TestDataFramePlots.test_specified_props_kwdÚvertTFc C sl t tj d¡tj d¡tj ddgd¡d}d\}}|jd|||d}| ¡ |ksXt| ¡ |kshtd S )Nr Úgroup1Úgroup2©r r Úgroup©r Úyr )r r® ÚxlabelÚylabel) r r r! r@ r" r# Ú
|
get_xlabelr% Ú
|
get_ylabel©r' r® r* rµ r¶ r+ r r r Útest_plot_xlabel_ylabel s
|
|
ýÿz*TestDataFramePlots.test_plot_xlabel_ylabelc C sj t tj d¡tj d¡tj ddgd¡d}d\}}|j|||d}| ¡ |ksVt| ¡ |ksftd S )Nr r¯ r° r± r³ )r® rµ r¶ ) r r r! r@ r" rC r· r% r¸ r¹ r r r Útest_boxplot_xlabel_ylabel$ s
|
|
ýÿz-TestDataFramePlots.test_boxplot_xlabel_ylabelc C sÈ t tj d¡tj d¡tj ddgd¡d}d\}}|jd|||d}|D ]$}| ¡ |ks`t| ¡ |ksLtqL| j
|
¡ |jd|d}|D ]*}|r| ¡ n| ¡ }|t dgkstq| j
|
¡ d S ) Nr r¯ r° r± r³ r² )r< r® rµ r¶ )r< r® ) r r r! r@ r" rC r· r% r¸ r( Úcloser
|
)r' r® r* rµ r¶ r+ ZsubplotZ target_labelr r r Ú test_boxplot_group_xlabel_ylabel2 s"
|
|
ýÿ
|
z3TestDataFramePlots.test_boxplot_group_xlabel_ylabelN)Ú__name__Ú
|
__module__Ú __qualname__r, rd ÚmarkÚslowrH r[ r` ri rw rz r r r Ú parametrizer¤ r¨ r© r rº r» r½ r r r r r sx
|
" þúþ
|
üþ
|
üþöþ
|
|
þ
|
üþ
|
|
|
r c @ sü e Zd Zdd Zejjdd Zdd Zdd Z ejjd d
|
Z
|
ejjd d Z ejjd d Z dd Z ej dddddddgfdgdddddgfddddd d gfddgdddddddd dd g
|
fd dddddddd dd g
|
fg¡d d Zd!d" Zd#d$ Zd S )%ÚTestDataFrameGroupByPlotsc C sn |j dd}tjtdd t|jdd}W 5 Q R X | jt|jddd t|jddd
|
}| j|d d d d S ) NÚgenderr= Fr; rN r7 r ©r r ©Zaxes_numÚlayout©rP r8 r rr ) rQ rD rE rF r rC Ú_check_axes_shaper& rU )r' r^ ÚgroupedrN r r r rH K s z.TestDataFrameGroupByPlots.test_boxplot_legacy1c C s¢ t tjd d
td}ttj dd¡t |¡d}|j
|
dd}t j t dd t|jdd }W 5 Q R X | jt|jdd
|
d t|jddd }| j|dd d d S )Nr r r r ©ÚlevelFr; rN r7 )r. r rÇ rÉ rr )ÚziprA rB Úranger r r! rO r Ú from_tuplesrQ rD rE rF r rC rÊ r& rU ©r' Útuplesr* rË rN r r r r[ S s z.TestDataFrameGroupByPlots.test_boxplot_legacy2c C s¬ t tjd d
td}ttj dd¡t |¡d}|j
|
ddj ddd}t j tdd t|jd
|
d }W 5 Q R X | jt|jdd d t|jdd
|
d}| j|ddd d S )Nr r r r rÌ r )rÍ ZaxisFr; rN r7 ©r r rÇ rÉ rr )rÎ rA rB rÏ r r r! rO r rÐ ZunstackrQ rD rE rF r rC rÊ r& rU rÑ r r r Útest_boxplot_legacy3_ s z.TestDataFrameGroupByPlots.test_boxplot_legacy3c C sî d}t tjjdd|d}t tjjdd|d}tj d¡jddg|d}t|||d }| d
|
¡}| ¡ }t | j
|
¡ d kst t |d kst t ¡ |jd d }t | j
|
¡ dks¼t t |d ksÌt t ¡ | d
|
¡ ¡ }t ¡ d S )Nr é¦ rn )Úsizeé< é* ZmaleZfemale)ro rp rÅ rÅ r rN r7 r )r r r! ÚnormalZ RandomStater" r rQ r# rV r( Z get_fignumsr% rD r¼ rC Úhist)r' r) rp ro rÅ r* ÚgbÚresr r r Útest_grouped_plot_fignumsi s
|
z3TestDataFrameGroupByPlots.test_grouped_plot_fignumsc C s( |}|j dd}t|tjs t| j|d dddgd | d¡j dd}| j|dd d
|
gd d ¡ }ttj
|
d d ¡|d}d ¡ }|d |d< dD ]}| d¡j |d}| j||dddgd |j d|d}| j||dddgd | d¡j |d}| j|||d |j d|d}| j|||d qd S )NrÅ r= ro rp rq )Z expected_keysr6 r7 ZMaleZFemalezX B C D A G Y N Q Oé2 r rI zA B C D E F G H I Jr{ )r6 rN rc Ú classroomrJ rK r| )r< r8 ) rC r] r Zndarrayr% rg rQ Úsplitr r! r@ ) r' r^ r* r_ Zcolumns2Zdf2Z categories2ÚtÚreturnedr r r Ú test_grouped_box_return_type s6 ÿ ÿz6TestDataFrameGroupByPlots.test_grouped_box_return_typec C s¦ |}d}t jt|d |jddg|jdd W 5 Q R X d}t jt|d |jdddgd d
|
d W 5 Q R X d }t jt|d |jddg|jd d W 5 Q R X tjtdd t| d¡jdd
|
d}W 5 Q R X | j
|
| j ¡ j ddd tjtdd t| d¡jdd
|
d}W 5 Q R X | j
|
| j ¡ j ddd tjtdd t| d¡jdd
|
d}W 5 Q R X | j
|
| j ¡ j ddd |jdddgddd}| j
|
| j ¡ j ddd |d fD ].}| j| ¡ dd | j|j ¡ gdd q°|d |d fD ]&}| | ¡ ¡ | |j ¡ g¡ qð| d¡jdddgd
|
d}| j
|
| j ¡ j ddd tjtdd t| d¡jdd d
|
d }W 5 Q R X | j
|
| j ¡ j dd d tjtdd t| d¡jdd d
|
d }W 5 Q R X | j
|
| j ¡ j dd d |jdddgdd d}| j
|
| j ¡ j dd d |jdddgdd d}| j
|
| j ¡ j dd d | d¡jdddgd!d
|
d }| j
|
| j ¡ j dd!d | d¡jdddgd"d
|
d }| j
|
| j ¡ j dd#d d S )$Nz1Layout of 1x1 must be larger than required size 2ra rp ro rr )r: r< rÈ z7The 'layout' keyword is not supported when 'by' is Nonerq )r r r6 )r: rÈ r8 z1At least one dimension of layout must be positive)éÿÿÿÿrä Fr; rÅ r9 r rÆ rÇ r. rÓ rß r rN )r: r< r8 )Zvisible)r r )r rä )r. r )rä r )r r )r r. )r rä ©r r )rd re rf rC rÅ rD rE rF r rQ rÊ r( ÚgcfrN Z_check_visibler$ ZxaxisZ get_label)r' r^ r* rh r rN r+ r r r Útest_grouped_box_layout¢ s¸ ý
|
ÿ
|
ÿ
|
ÿ ÿ
|
ÿ
|
ü
|
ü ÿ ÿ
|
ÿ
|
ÿz1TestDataFrameGroupByPlots.test_grouped_box_layoutc
|
C s¶ |}t t¡F | j dd¡\}}| d¡jdd|d | j| j ¡ j ddd W 5 Q R X | j dd ¡\}}t t¡" |jdd
|
dgd d|d d }W 5 Q R X t
|
t |j ¡}| j|d dd t ||d ¡ |d j|ksàtt t¡& | d¡jdd
|
dgd|d d}W 5 Q R X t
|
t |j ¡}| j|d dd t ||d ¡ |d j|ksZtd}tjt|d@ | j dd ¡\}}t t¡ | d¡j|d}W 5 Q R X W 5 Q R X d S )Nr rq ro rN )r: r8 r+ r. rÓ rÇ r rp rÅ r )r: r< r8 r+ rå rß r z@The number of passed axes must be 3, the same as the output plotra )r+ )rD rE rF r( rP rQ rC rÊ ræ rN r Úarrayr& rU Zassert_numpy_array_equalr r% rd re rf )r' r^ r* rX rN râ rh r r r Ú test_grouped_box_multiple_axes s@ " ü
|
ÿ z8TestDataFrameGroupByPlots.test_grouped_box_multiple_axesc C sF t ddddddgddddddgd}| j|jd d
|
d d d d d d S )Nr r r r. r{ r- r )r r r r r
)r< r r r rG r r r r - s & ÿz'TestDataFrameGroupByPlots.test_fontsizezcol, expected_xticklabelr£ z(a, v)z(b, v)z(c, v)z(d, v)z(e, v)Úv1z(a, v1)z(b, v1)z(c, v1)z(d, v1)z(e, v1)Nc C sl t tj tdd¡tj d¡tj d¡d}| d¡}t|jd|dd}dd | ¡ D }||ksht
|
d S )
|
NZabcder )Úcatr£ rê rë FrN )rP r: r8 c S s g | ] }| ¡ qS r r r r r r r h s zQTestDataFrameGroupByPlots.test_groupby_boxplot_subplots_false.<locals>.<listcomp>) r r r! r" r& rO rQ r rC r$ r% )r' rl Úexpected_xticklabelr* rË rN Úresult_xticklabelr r r Ú#test_groupby_boxplot_subplots_false3 s (
|
|
ýÿ
|
ÿz=TestDataFrameGroupByPlots.test_groupby_boxplot_subplots_falsec C sD | d¡}| d¡}d}tjt|d t|jdd W 5 Q R X d S )NÚobjectrÅ z:boxplot method requires numerical columns, nothing to plotra F)rP )r rQ rd re rf r rC )r' r^ r* rË rh r r r Útest_groupby_boxplot_objectk s
|
|
|
z5TestDataFrameGroupByPlots.test_groupby_boxplot_objectc C s¦ ddddddddgddddddddgg}t t| }tj|ddgd }ttj d
|
d ¡d d dg|d}ddg}t|j |dd}ddg}dd |
|
¡ D }||ks¢t d S )Nr Zbazr5 Zquxr/ r0 ÚfirstÚsecond)Únamesr r rJ rK r| r3 )r r/ )r r0 rN r9 z
|
(bar, one)z
|
(bar, two)c S s g | ] }| ¡ qS r r r r r r r s zLTestDataFrameGroupByPlots.test_boxplot_multiindex_column.<locals>.<listcomp>) r& rÎ r rÐ r r r! r@ r rC r$ r% ) r' ZarraysrÒ r r* rl rN rì rí r r r Ú test_boxplot_multiindex_columns s þ z8TestDataFrameGroupByPlots.test_boxplot_multiindex_column)r¾ r¿ rÀ rH rd rÁ r r[ rÔ rÝ rã rç ré r rà rî rð rô r r r r rÄ I s`
|
|
|
"
|
]
|
,öþöþíþ&
|
rÄ ) Ú__doc__rR rA Únumpyr rd Z pandas.util._test_decoratorsÚutilZ_test_decoratorsÚtdZpandasr r r r r r Zpandas._testingZ_testingrD Z pandas.tests.plotting.commonr r Zpandas.io.formats.printingr
|
Zskip_if_no_mplr rÄ r r r r Ú<module> s .
|
