U
|
ý°dâ ã @ sl d Z ddlZddlZddlm mZ ddlm Z m
|
Z
|
m Z ddl m Z ddlmZ ejG dd deZdS )z Test cases for GroupBy.plot é N)Ú DataFrameÚIndexÚSeries)Ú TestPlotBasec @ sx e Zd Zdd Zdd Zdd Zdd Zd d
|
Zej
|
d d d g¡dd Z ej
|
dddg¡dd Z dd Z dd ZdS )ÚTestDataFrameGroupByPlotsc C s d}t tjjdd|d}t tjjdd|d}tj d¡jddg|d}| |¡ ¡ t ¡ | |¡
|
¡ t ¡ | |¡jd d
|
t ¡ d S ) Né
|
é¦ é )Úsizeé< é* ZmaleZfemaleg à?)Úalpha) r ÚnpÚrandomÚnormalZ RandomStateÚchoiceÚgroupbyÚplotÚtmÚcloseÚhist)ÚselfÚnZweightÚheightZgender© r úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/plotting/test_groupby.pyÚ,test_series_groupby_plotting_nominally_works s zFTestDataFrameGroupByPlots.test_series_groupby_plotting_nominally_worksc C s| t dddddddddg tj d¡ddddddddddg d }| d
|
¡d ¡ t ¡ | d
|
¡d d d ¡ t ¡ d S )Né é é é )ÚdefÚvalg ð?g @g @)Úindexr! r" c S s | ¡ S )N)r )Úxr r r Ú<lambda>+ ó zPTestDataFrameGroupByPlots.test_plotting_with_float_index_works.<locals>.<lambda>) r r r Úrandnr r r r Úapply©r Údfr r r Ú$test_plotting_with_float_index_works" s "þz>TestDataFrameGroupByPlots.test_plotting_with_float_index_worksc C sp t ddd¡}tddgddgddgd }|d
|
j|d |d tdgdgdgd }|d
|
j|d |d d S ) NéP éf r ZAAAZBBBr éU éY )ÚNameÚByColÚMarkr2 r1 )ZbyÚbins)r Zaranger r )r r3 r* r r r Útest_hist_single_row. s
|
z.TestDataFrameGroupByPlots.test_hist_single_rowc C sd t dddddgdddddgtdd}| d¡j d d
|
¡ t ¡ | d¡d j ¡ t ¡ d S ) Nr r r é é Úababa©r$ ÚyÚzr: r$ r9 )r Úlistr r Úscatterr r Úliner) r r r Útest_plot_submethod_works6 s
|
(z3TestDataFrameGroupByPlots.test_plot_submethod_worksc C s t dddddgdddddgtdd}| d¡jd d
|
d d }t|d jdksTt| d¡jjd
|
d d}t|d jdkstd S )Nr r r r5 r6 r7 r8 r: r< r$ r9 )Úkindr$ r9 Úa)r$ r9 )r r; r r ÚlenÚ collectionsÚAssertionErrorr< )r r* Úresr r r Útest_plot_kwargs= s
|
(z*TestDataFrameGroupByPlots.test_plot_kwargszcolumn, expected_axes_num)Nr )Úbr c C s¦ d|f}|pdgdgg}t ddg ddg dd}ttj d d
|
¡|ddgd }| d¡}|jd |d D ]8}| j|||d t|d |D ]\} }
|
| | |
|
¡ qqhd S )Nr r@ rF é Ú1Ú2Úc©Únameé r ©r# ÚcolumnsT)ÚlegendÚcolumn©Zaxes_numZlayoutr )
|
r r r r r' r r Ú_check_axes_shapeÚzipÚ_check_legend_labels) r rQ Zexpected_axes_numZexpected_layoutZexpected_labelsr# r* ÚgZaxesÚaxZexpected_labelr r r Ú#test_groupby_hist_frame_with_legendH s
|
ÿz=TestDataFrameGroupByPlots.test_groupby_hist_frame_with_legendrQ NrF c C sp t ddg ddg dd}ttj dd¡|dd gd
|
}| d¡}tjtd d |j d |dd W 5 Q R X d S )NrG rH rI rJ rK rM r r@ rF rN ú Cannot use both legend and label©ÚmatchTÚd)rP rQ Úlabel©
|
r r r r r' r ÚpytestZraisesÚ
|
ValueErrorr )r rQ r# r* rV r r r Ú*test_groupby_hist_frame_with_legend_raisesY s
|
|
zDTestDataFrameGroupByPlots.test_groupby_hist_frame_with_legend_raisesc C s| t ddg ddg dd}ttj dd¡|dd gd
|
}| d¡}|d jd d D ]$}| j|d dd | |ddg¡ qRd S )NrG rH rI rJ rK rM r r@ rF rN T)rP r )r r rR ) r r r r r' r r rS rU )r r# r* rV rW r r r Ú$test_groupby_hist_series_with_legendc s
|
z>TestDataFrameGroupByPlots.test_groupby_hist_series_with_legendc C sn t ddg ddg dd}ttj dd¡|dd gd
|
}| d¡}tjtd d |j d dd W 5 Q R X d S )NrG rH rI rJ rK rM r r@ rF rN rY rZ Tr\ )rP r] r^ )r r# r* rV r r r Ú+test_groupby_hist_series_with_legend_raisesm s
|
|
zETestDataFrameGroupByPlots.test_groupby_hist_series_with_legend_raises)Ú__name__Ú
|
__module__Ú __qualname__r r+ r4 r> rE r_ ÚmarkZ parametrizerX ra rb rc r r r r r s
|
|
|
r )Ú__doc__Únumpyr r_ Z pandas.util._test_decoratorsÚutilZ_test_decoratorsÚtdZpandasr r r Zpandas._testingZ_testingr Z pandas.tests.plotting.commonr Zskip_if_no_mplr r r r r Ú<module> s
|
