U
|
ý°dw ã @ s¶ d Z ddlZddlZddlZddlm mZ ddl m
|
Z
|
m Z m Z m Z ddlmZ ddlmZmZ ejdd ZejG dd deZejG d d
|
d
|
eZejG d d d eZdS ) z Test cases for .hist method é N)Ú DataFrameÚIndexÚSeriesÚ to_datetime)Ú TestPlotBaseÚ_check_plot_worksc C s t jddS )NÚts©Úname)ÚtmZmakeTimeSeries© r r ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/plotting/test_hist_method.pyr s r c @ sÐ e Zd Zdd Zdd Zdd Zejjdd Z d d
|
Z
|
d d Z d d Z ej dddddg¡dd Zej dddg¡dd Zej dd d g¡d d Zd d! Zejjd"d#ejd$d% Zejd&d' Zd S )(ÚTestSeriesPlotsc C s8 t |j t |jdd t |jdd tjtdd t |j|jjd W 5 Q R X tjtdd t |j|jjdd W 5 Q R X | j d d ¡\}}t |j|d
|
d t |j||d
|
d t |j|d
|
d t ¡ | j d d¡\}\}}t |j||d
|
d t |j||d
|
d d}t
|
j t |d |j|j|d W 5 Q R X d S )NF©Zgrid©é é
|
©Úfigsize©Zcheck_stacklevel©Úbyé )r Úbinsé T)ÚaxÚ default_axes)r Úfigurer )r r é )r r r z`Cannot pass 'figure' when using the 'by' argument, since a new 'Figure' instance will be created©Úmatch)r r ) r Úhistr Úassert_produces_warningÚ UserWarningÚindexÚmonthÚpltÚsubplotsÚcloseÚpytestÚraisesÚ
|
ValueError)Úselfr Úfigr Úax1Úax2Úmsgr r r Útest_hist_legacy s&
|
ÿz TestSeriesPlots.test_hist_legacyc C s< t tj dd¡}|jddd d }t|jdks8td S )Nr r ©r r )r ÚnpÚrandomÚrandnr! ÚlenÚpatchesÚAssertionError)r, Údfr r r r Útest_hist_bins_legacy8 s z%TestSeriesPlots.test_hist_bins_legacyc C s` |}d}t jt|d |jjdd W 5 Q R X t jt|d |jjddgd W 5 Q R X d S )Nz7The 'layout' keyword is not supported when 'by' is Noner ©r r ©Úlayoutr )r) r* r+ Úheightr! )r, Úhist_dfr9 r0 r r r Útest_hist_layout= s z TestSeriesPlots.test_hist_layoutc C sà |}t jtdd t|jj|jdd}W 5 Q R X | j|ddd t jtdd t|jj|jdd}W 5 Q R X | j|ddd t jtdd t|jj|jd d}W 5 Q R X | j|d
|
d d t jtdd t|jj|jd d}W 5 Q R X | j|d
|
d d t jtdd t|jj|jdd}W 5 Q R X | j|d
|
d d t jtdd t|jj|jdd}W 5 Q R X | j|d
|
dd t jtdd t|jj|j d d}W 5 Q R X | j|dd d |jj|jddd}| j|d
|
ddd d S )NFr ©r r ©r r= r ©Úaxes_numr= )é éÿÿÿÿ)rE r ©é r rH ©r rF ©r r )rE r ©rF rH ©r rH rE ©rH r )é é )r r= r ©rD r= r )
|
r r" r# r r> r! ÚgenderÚ_check_axes_shapeÚcategoryÚ classroom©r, r? r9 Úaxesr r r Útest_hist_layout_with_byF s0 z(TestSeriesPlots.test_hist_layout_with_byc C sp ddl m}m} ttj d¡}ttj d¡}|d | ¡ |d | ¡ | }|j}t |dkslt
|
d S )Nr )ÚgcfÚsubplotr éy éz ) Zmatplotlib.pyplotrX rY r r3 r4 r5 r! rV r6 r8 )r, rX rY ÚxÚyr- rV r r r Útest_hist_no_overlapl s z$TestSeriesPlots.test_hist_no_overlapc C s. |}|j j|jd}t| j ¡ dks*td S )Nr r )r> r! rQ r6 r& Ú get_fignumsr8 rU r r r Útest_hist_by_no_extra_plots| s z+TestSeriesPlots.test_hist_by_no_extra_plotsc C sR ddl m} | }| }| d¡}d}tjt|d |j||d W 5 Q R X d S )Nr )r éo z&passed axis not bound to passed figurer )r r )Zpylabr Z add_subplotr) r* r8 r! )r, r r Zfig1Zfig2r. r0 r r r Ú+test_plot_fails_when_ax_differs_from_figure s
|
z;TestSeriesPlots.test_plot_fails_when_ax_differs_from_figureúhisttype, expected©ÚbarT©Z
|
barstackedT©ÚstepF©Z
|
stepfilledTc C s0 t tj dd¡}|j|d}| j||d d S )Nr r ©Úhisttype©Zfilled)r r3 r4 Úrandintr! Ú_check_patches_all_filled)r, rk ÚexpectedÚserr r r r Útest_histtype_argument s z&TestSeriesPlots.test_histtype_argumentz&by, expected_axes_num, expected_layout)Nr r; )Úbr )r r c C sd ddg ddg }t tj d¡|dd}d|j_t|jdd|d }| j|||d
|
| |d¡ d S ) Né Ú1Ú2é Úa©r$ r
|
rr T)r Úlegendr rC )
|
r r3 r4 r5 r$ r
|
r r! rR Ú_check_legend_labels)r, r Úexpected_axes_numÚexpected_layoutr$ ÚsrV r r r Útest_hist_with_legend s z%TestSeriesPlots.test_hist_with_legendr Nrr c C s` ddg ddg }t tj d¡|dd}d|j_tjtdd |j d
|
|d d W 5 Q R X d S ) Nrs rt ru rv rw rx rr ú Cannot use both legend and labelr TÚc)ry r Úlabel)
|
r r3 r4 r5 r$ r
|
r) r* r+ r! )r, r r$ r} r r r Ú test_hist_with_legend_raises¨ s
|
z,TestSeriesPlots.test_hist_with_legend_raisesc C s® | j ¡ \}}|jjd|d}t|jdks0t| |j ¡ d¡ t
|
¡ | j ¡ \}}|jjd|d}| |j ¡ d¡ t
|
¡ | j ¡ \}}|jjdd|d}t
|
¡ d S ) Nr ©r r Z FrequencyZ
|
horizontal)Z orientationr ÚleftT)ZalignZstackedr ) r& r' Úplotr! r6 r7 r8 Ú_check_text_labelsÚyaxisZ get_labelr r( Zxaxis)r, r Ú_r r r r Útest_hist_kwargs² s z TestSeriesPlots.test_hist_kwargszApi changed in 3.6.0)Úreasonc C sì | j ¡ \}}|jjd|d}| j|dd | ¡ }| |dgt| ¡ | ¡ }| |dgt| ¡ t |jj
|
t |jj | j ¡ \}}|jj
|
d|d}| j|dd | ¡ }| |dgt| ¡ | ¡ }| |dgt| ¡ d S )NT)Úlogyr Úlog©r Ú ) r& r' r
r! Ú_check_ax_scalesZget_xticklabelsr r6 Zget_yticklabelsr ÚkdeÚdensity)r, r r r ZxlabelsZylabelsr r r Ú test_hist_kde s z TestSeriesPlots.test_hist_kdec C s° | j ¡ \}}|jjddd|d}| j|dd t|jdksBt| j|jdgd d | j ¡ \}}|jj dd|d }| j|dd |
|
¡ }t|d
|
kst| |dg¡ d S ) NTr rr )r r Úcolorr r r )Z
|
facecolorsÚr)r r r r ) r& r' r
r! r r6 r7 r8 Z _check_colorsr Ú get_lines)r, r r r Úlinesr r r Útest_hist_kde_colorØ s z#TestSeriesPlots.test_hist_kde_color)Ú__name__Ú
|
__module__Ú __qualname__r1 r: r@ r) ÚmarkÚslowrW r^ r` rb Ú parametrizerq r~ r r ZxfailÚtdZskip_if_no_scipyr r r r r r r s<
|
%
|
üþ
|
ÿ
|
|
r c @ s e Zd Zejjdd Zdd Zdd Zdd Z d d
|
Z
|
ej d d d ddgfdd dgdd dgfg¡dd Z ej dddddg¡dd Z ej dd dg¡ej dd d g¡d d Zej dd dg¡ej dd d g¡d d Zd!d" Zd#d$ Zd%d& Zejd'd( Zd S ))ÚTestDataFramePlotsc s¼ ddl m tjtdd t|j W 5 Q R X ttj
|
dd¡}t tj j dddtj d |d< tjtdd t|jdd
|
}W 5 Q R X | j|d d d |d ¡ r¨tt|dg j ttj
|
dd¡}t|j ttj
|
dd¡}t tj j dddtj d |d< tjtdd t|jdd}W 5 Q R X | j|ddd tjtdd t|jddd W 5 Q R X tjtdd t|jdd W 5 Q R X tjtdd t|jdd W 5 Q R X |d }d\}}d\}}|j||||d}| j|||||d d\}}d\}}|j||||d}| j|||||d t ¡ |jdd dd } fd d | ¡ D }
|
t |
|
d ¡ d!¡ t ¡ |jdd"} | j| d#d$ t ¡ t t¡ |jd%d& W 5 Q R X d S )'Nr ©Ú RectangleFr éd r ì jÕ+2Zì ÊReöZ©ÚsizeÚdtyper rE rJ rC r; r r rM r< é T)ÚsharexÚshareyr r r2 ©é é ©rv é( ©Ú
|
xlabelsizeÚxrotÚ
|
ylabelsizeÚyrotrH )Ú
|
cumulativer r c s g | ]}t | r|qS r ©Ú
|
isinstance©Ú.0r\ r r r Ú
|
<listcomp>2 s
|
z:TestDataFramePlots.test_hist_df_legacy.<locals>.<listcomp>rF ç ð?)r r r re )Úfoo)Úmatplotlib.patchesr¡ r r" r# r r! r r3 r4 r5 r rm Úint64rR Ú get_visibler8 Ú_check_ticks_propsr( Ú get_childrenÚassert_almost_equalÚ
|
get_heightr Úexternal_error_raisedÚAttributeError) r, r? r9 rV rp ÚxfÚyfr² r´ r Úrectsr r r Útest_hist_df_legacyê s üÿ
|
üÿ ÿ ÿ z&TestDataFramePlots.test_hist_df_legacyc C s t tj d¡tj ddd¡ttjjdddtjdttjjdddtjdddd}| t¡}d }t j
|
t |d
|
| ¡ W 5 Q R X d S ) Nr r l ú Ûp¹/ l @©spcÍ/ )r§ T)Úutc)rw rr r ÚdzDhist method requires numerical or datetime columns, nothing to plot.r ) r r3 r4 Zrandrm r r¾ ZastypeÚobjectr) r* r+ r! )r, r9 Zdf_or0 r r r Ú*test_hist_non_numerical_or_datetime_raises@ s4
|
ÿÿ ÿüøÿ
|
z=TestDataFramePlots.test_hist_non_numerical_or_datetime_raisesc C s6 t tj dd¡}ttjjdddtjd|d< d ddddddddd d dd
|
d
|
dd d dd ddd dddddf }|D ],}|j|d d}|d }| j|d|d qd}t j
|
t |d |jdd W 5 Q R X t d¡}t j
|
t |d |jdd W 5 Q R X d}t j
|
t |d |jdd W 5 Q R X d S )Nr¢ r r£ r¤ r¥ rJ )r= Ú expected_sizerG rL )rE rE rK )rH rF )rF r rI r= r< rÎ rE rC z1Layout of 1x1 must be larger than required size 3r r; z)Layout must be a tuple of (rows, columns))r ú1At least one dimension of layout must be positive©rF rF )r r3 r4 r5 r rm r¾ r! rR r) r* r+ ÚreÚescape)r, r9 Zlayout_to_expected_sizeZ layout_testrV ro r0 r r r r@ Y s@ üÿ
|
÷
|
z#TestDataFramePlots.test_hist_layoutc C sT t tj dd¡}ttjjdddtjd|d< t|jdd | j
|
¡ t ¡ d S )Nr¢ r r£ r¤ r¥ T)r ) r r3 r4 r5 r rm r¾ r r! r& Z tight_layoutr r( )r, r9 r r r Útest_tight_layout s üÿ
|
z$TestDataFramePlots.test_tight_layoutc C sJ t dddddgdddddgd}t|jd d
|
d d d d}| j|d d d S )Nç ø?ç à?ç333333ó?çÍÌÌÌÌÌì?rE ÚpigÚrabbit)ÚlengthÚanimalTrÚ rÛ r r )r Úcolumnr r r² ©r² )r r r! rÀ )r, r9 rV r r r Útest_hist_subplot_xrot s þÿúz)TestDataFramePlots.test_hist_subplot_xrotzcolumn, expectedNÚwidthrÚ r> c st t dddddgddddd gd dd
|
d d gd dddddgd}t|jd|dd fddtd D }||ksptd S )Nçffffffæ?çÉ?g333333Ã?gñ?rÔ rÕ rÖ r× rE g333333 @r r )rß rÚ r> rØ rÙ ZduckZchickenZhorse)r$ T©r rE )r rÜ r= c s g | ]} d |f ¡ qS )r )Z get_title)r¹ Úi©rV r r rº À s zGTestDataFramePlots.test_hist_column_order_unchanged.<locals>.<listcomp>)r r r! Úranger8 )r, rÜ ro r9 Úresultr rä r Ú test_hist_column_order_unchanged¦ s
|
ý ú
|
üz3TestDataFramePlots.test_hist_column_order_unchangedrc rd rf rg ri c C s< t tjjddddddgd}|j|d}| j||d d S )
|
Nr r ©r¢ r ©r¦ rw rr ©Úcolumnsrj rl ©r r3 r4 rm r! rn ©r, rk ro r9 r r r r rq à s z)TestDataFramePlots.test_histtype_argumentr r rÜ rr c C sÖ |d kr|d k rdnd}d|f}|p*ddg}|d k r>|gd }t ddg ddg dd }ttj d
|
d¡|ddgd }t|jd d ||d }| j|||d |d kr²|d kr²|d }t||D ]\} }
|
| |
|
| ¡ q¼d S )Nr r rw rr rs rt ru r r rv ©r$ rë T)r ry r rÜ rC r )
|
r r r3 r4 r5 r r! rR Úziprz ) r, r rÜ r{ r| Zexpected_labelsr$ r9 rV Zexpected_labelr r r r r~ Ò s&
|
ûz(TestDataFramePlots.test_hist_with_legendc C sh t ddg ddg dd}ttj dd¡|dd gd
|
}tjtd d |jd ||dd W 5 Q R X d S )Nrs rt ru r r rv r rw rr rî r r TrË )ry r rÜ r ) r r r3 r4 r5 r) r* r+ r! )r, r rÜ r$ r9 r r r r î s z/TestDataFramePlots.test_hist_with_legend_raisesc C sF t tj dd¡}| j ¡ \}}|jjd|d}t|j dksBt
|
d S )Nr r r r ) r r3 r4 r5 r& r' r
r! r6 r7 r8 ©r, r9 r r r r r Útest_hist_df_kwargsø s z&TestDataFramePlots.test_hist_df_kwargsc C s t tj d¡ dd¡ddddgd}d d
|
gd |d < | j ¡ \}}|jjd |d }t |j
|
dksdt | j ¡ \}}|jj|d}t |j
|
dkst d S )Né* r rH ÚAÚBÚCÚDrê r\ r] r ÚEr r¬ )r r¯ ) r r3 r4 Ú RandomStater5 r& r' r
r! r6 r7 r8 rð r r r Ú test_hist_df_with_nonnumericsþ s
|
ÿz0TestDataFramePlots.test_hist_df_with_nonnumericsc C s t tj dd¡tdd}| j ¡ \}}|d jjd|d}|d jj|ddd | j |dd
|
gd |
|
¡ ¡ stt |j
|
¡ ¡ st t ¡ | j ¡ \}}|d jjdd|d }|d jj|ddd | j |jd d
|
gd |j
|
¡ ¡ rît |
|
¡ ¡ sþt t ¡ | j ¡ \}}|d jjdd|d }|d jj|dd | j |jd dgd |j
|
¡ ¡ sft |
|
¡ ¡ sxt t ¡ d S )Nrv rH Zabcdrê rw T)ry r rr )r ry Ú secondary_yz b (right))Úlabels)ry rú r z a (right))r ry )r r3 r4 r5 Úlistr& r' r
r! rz Z get_yaxisr¿ r8 Zright_axr r( Zleft_axrð r r r Útest_hist_secondary_legend s, z-TestDataFramePlots.test_hist_secondary_legendc s| t tjddgdtjtjgdddggtdd}t d dd
|
g¡}t dddgdddggtdd}t dd d gd
|
d
|
d
|
gg¡}d d lm | j ¡ \}}|j j
|
||d fdd| ¡ D }dd |D }| j ¡ \}} |j j
|
| |d fdd| ¡ D }
|
dd |
|
D } t dd t || D s&tt dd gd
|
d
|
gg¡} d} tjt| d$ | j ¡ \}}|j j
|
|| d W 5 Q R X d S )Nrá g333333Ó?gÙ?rà gé?r× Úabcrê g Ð?gÍÌÌÌÌÌÜ?r r )r Úweightsc s g | ]}t | r|qS r r¶ r¸ r r r rº A s
|
zFTestDataFramePlots.test_hist_with_nans_and_weights.<locals>.<listcomp>c S s g | ] }| ¡ qS r ©rà ©r¹ Zrectr r r rº B s c s g | ]}t | r|qS r r¶ r¸ r r r rº E s
|
c S s g | ] }| ¡ qS r r r r r r rº F s c s s | ]\}}||kV qd S )Nr )r¹ Zh0Úh1r r r Ú <genexpr>G s zETestDataFramePlots.test_hist_with_nans_and_weights.<locals>.<genexpr>z?weights must have the same shape as data, or be a single columnr )r r3 Únanrü Úarrayr½ r¡ r& r' r
r! rÁ Úallrï r8 r) r* r+ )r, r9 rÿ Z no_nan_dfZno_nan_weightsr Zax0rÈ Zheightsr. Z no_nan_rectsZno_nan_heightsZidxerror_weightsr0 r/ r r r Ú test_hist_with_nans_and_weights2 s, þ z2TestDataFramePlots.test_hist_with_nans_and_weights)r r r r) r r rÉ rÍ r@ rÓ rÞ r rç rq r~ r rñ rù rý r Úskip_if_no_mplr r r r r r è sB
|
U* þþ
|
üþ
|
&r c @ sj e Zd Zdd Zdd Zejjdd Zdd Z d d
|
Z
|
d d Z d d Z ej dddddg¡dd ZdS )ÚTestDataFrameGroupByPlotsc sú ddl m ddlm} ttj dd¡dgd}ttjj dd dtj
|
d
|
|d < tj dd d¡|d < dgd |d< ||j |j d}| j |d dd t ¡ |j|j d}| j |d dd t ¡ |jddd}| j |ddd | j|dd t ¡ d\}}d\}}||j |j dd ||||dd }| ¡ D ]4} fdd | ¡ D } | d ¡ }
|
t |
|
d ¡ q(| j|||||d t ¡ ||j |j dd }| j|d!d" t ¡ t t¡ ||j |j d#d$ W 5 Q R X d%} tjt| d& |jd d'd( W 5 Q R X d S ))Nr r )Ú _grouped_histiô r ró rê r£ r¤ r¥ rô rH rõ ÚXrö r rJ rC rv )r Zrotr; rÝ r« r® T)r rµ r r± r² r³ r´ r c s g | ]}t | r|qS r r¶ r¸ r r r rº s
|
zFTestDataFrameGroupByPlots.test_grouped_hist_legacy.<locals>.<listcomp>rF r» r° )r r r r re )r r¼ z$Specify figure size by tuple insteadr Údefault)r r ) r½ r¡ Z pandas.plotting._matplotlib.histr
|
r r3 r4 r5 r rm r¾ ró rõ rR r r( r! rÀ ZravelrÁ rà r r rÄ rÅ r) r* r+ ) r, r
|
r9 rV rÆ rÇ r² r´ r rÈ r> r0 r r r Útest_grouped_hist_legacyS sp üÿ÷ ÿ z2TestDataFrameGroupByPlots.test_grouped_hist_legacyc C s d}t tjjdd|d}t tjjdd|d}tj d¡jddg|d}t|||d }| d
|
¡}| ¡ }t |d ks|t
|
t | j ¡ d kst
|
t ¡ d S ) Nr é¦ r¬ ré é< rò r r )r> ÚweightrQ rQ r )r r3 r4 Únormalrø Úchoicer Úgroupbyr! r6 r8 r& r_ r r( )r, Únr r> Z
|
gender_intZdf_intÚgbrV r r r Útest_grouped_hist_legacy2 s
|
z3TestDataFrameGroupByPlots.test_grouped_hist_legacy2c C s: |}d}t jt|d |jd|jdd W 5 Q R X d}t jt|d |jd|jdd W 5 Q R X d }t jt|d |jd|jd
|
d W 5 Q R X tjtd d t |jd|jd d}W 5 Q R X | j
|
|dd d tjtd d t |jd|jdd}W 5 Q R X | j
|
|dd d |jd|jdd}| j
|
|ddd |jd|jdd}| j
|
|ddd |jd|jddd}| j
|
|dddd t ¡ tjtd d t |jdddd}W 5 Q R X | j
|
|ddd tjtd d t |jdd}W 5 Q R X | j
|
|ddd |jd d d }| j
|
|dd d |jddd gd }| j
|
|ddd d S )!Nz1Layout of 1x1 must be larger than required size 2r r r; )rÜ r r= z1Layout of 1x3 must be larger than required size 4r> râ rÏ rÐ Fr rA r rC rI rG rH )rF r rM )rN r )rÜ r r= r rP rT rJ rE r rQ )rE r rB rS )rÜ ) r) r* r+ r! rQ rS r r" r# r rR r( )r, r? r9 r0 rV r r r Útest_grouped_hist_layout¤ s` ÿ ÿ ÿz2TestDataFrameGroupByPlots.test_grouped_hist_layoutc C sæ |}| j dd¡\}}|jdddg|d d}| j|ddd t ||d ¡ |d j|ks`t|jd
|
|d d }| j|ddd t ||d ¡ |d j|ks¤t| j dd¡\}}d }tj t
|
|d |jd|d}W 5 Q R X d S )Nr rE r> r rS r )rÜ r râ rC rT r )r r z@The number of passed axes must be 1, the same as the output plotr ) r& r' r! rR r Zassert_numpy_array_equalr r8 r) r* r+ )r, r? r9 r- rV Zreturnedr0 r r r Ú test_grouped_hist_multiple_axesÛ s z9TestDataFrameGroupByPlots.test_grouped_hist_multiple_axesc C sv |}|j d|jdd\}}| |¡ ||¡s0t| |¡ ||¡sFt| |¡ ||¡r\t| |¡ ||¡rrtd S )Nr> T)rÜ r r© ©r! rQ Ú
|
get_x_axisÚjoinedr8 Ú
|
get_y_axis©r, r? r9 r. r/ r r r Útest_axis_share_xï s z+TestDataFrameGroupByPlots.test_axis_share_xc C sv |}|j d|jdd\}}| |¡ ||¡s0t| |¡ ||¡sFt| |¡ ||¡r\t| |¡ ||¡rrtd S )Nr> T)rÜ r rª )r! rQ r r r8 r r r r r Útest_axis_share_yü s z+TestDataFrameGroupByPlots.test_axis_share_yc C sx |}|j d|jddd\}}| |¡ ||¡s2t| |¡ ||¡sHt| |¡ ||¡s^t| |¡ ||¡sttd S )Nr> T)rÜ r r© rª r r r r r Útest_axis_share_xy s z,TestDataFrameGroupByPlots.test_axis_share_xyrc rd rf rg ri c C s> t tjjddddddgd}|jd|d}| j||d d S )
|
Nr r rè ré rw rr rê )r rk rl rì rí r r r rq s z0TestDataFrameGroupByPlots.test_histtype_argumentN)r r r r r r) r r r r r r r r rq r r r r r Q s" E
|
6 üþ r )Ú__doc__rÑ Únumpyr3 r) Z pandas.util._test_decoratorsÚutilZ_test_decoratorsr Zpandasr r r r Zpandas._testingZ_testingr Z pandas.tests.plotting.commonr r Zfixturer r r r r r r r r Ú<module> s$
|
M j
|
