.i/,dZddlZddlZddlZddlmZmZmZm Z m Z ddl m Z ddlmZmZmZmZmZmZmZmZmZmZej2dZej2dZddlmZej<dZGd d Z Gd d Z!Gd dZ"y)zTest cases for .hist methodN) DataFrameIndexSeries date_range to_datetime) _check_ax_scales_check_axes_shape _check_colors_check_legend_labels_check_patches_all_filled_check_plot_works_check_text_labels_check_ticks_props get_x_axis get_y_axis matplotlibzmatplotlib.pyplot) _grouped_histc|ttjdtjt ddddS)Ndtypez 2020-01-01B)periodsfreqtsindexname)rnparangefloat64rd/var/www/app/trading-bot/venv/lib/python3.12/site-packages/pandas/tests/plotting/test_hist_method.pyrr#s0  "BJJ'r<  r#c eZdZejj diddiddigdZejj diddigd Zd Zd Z d Z d Z dZ dZ dZejjejj dgdgdgdgdgdgdgdgdZdZdZdZdZejj dgdd Zejj d!d"d#gd$Zejj d%d&d'gd(Zd)Zd*Zd+Zejj7d,-d.Zd/Zd0Zejj7d,-d1Zd2Z d3Z!y&)4TestSeriesPlotskwargsgridFfigsize c0t|jfi|yN)r histselfrr's r$test_hist_legacy_kwargsz'TestSeriesPlots.test_hist_legacy_kwargs-s"'',V,r#binsctjtd5t|jfd|j j i|dddy#1swYyxYw)NFcheck_stacklevelby)tmassert_produces_warning UserWarningr r/rmonthr0s r$test_hist_legacy_kwargs_warningz/TestSeriesPlots.test_hist_legacy_kwargs_warning1sK ' ' e L D bgg C"((.. CF C D D Ds -AAcztjjdd\}}t|j|dy)NT)ax default_axesmplpyplotsubplotsr r/r1rfigr@s r$test_hist_legacy_axz#TestSeriesPlots.test_hist_legacy_ax7s,**%%a+R"''bt ? 299((+;;A> ? C  C ggixx4yA~~r#c|}|jj|jttj j dk(sJy)Nr8r?)rir/rlrarCrD get_fignums)r1rjrcs r$test_hist_by_no_extra_plotsz+TestSeriesPlots.test_hist_by_no_extra_plotss=  "))$3::))+,111r#ctjd}tjd}|jd}d}tjt |5|j ||dddy#1swYyxYw)Nr?rPoz&passed axis not bound to passed figurerU)r@rJ)rrJ add_subplotrWrXAssertionErrorr/)r1rfig1fig2rQrZs r$+test_plot_fails_when_ax_differs_from_figurez;TestSeriesPlots.test_plot_fails_when_ax_differs_from_figuresdzz!}zz!}s#6 ]]> 5 ) GGs4G ( ) ) )s A66A?histtype, expected)barT) barstackedT)stepF) stepfilledTcttjjdj dd}|j |}t ||y)NrPr?r,histtypefilled)rrr^r_integersr/r )r1rexpectedserr@s r$test_histtype_argumentz&TestSeriesPlots.test_histtype_argumentsDRYY**1-66q"=> XXxX (!"X6r#z&by, expected_axes_num, expected_layout)Nr?rf)brP)r?rPcddgzddgzz}ttjjdj d|d}d|j _t|jd d | }t||| t|dy) Nr412rPr,arrT)rAlegendr8r{) rrr^r_r`rrr r/r r )r1r8expected_axes_numexpected_layoutrsrs r$test_hist_with_legendz%TestSeriesPlots.test_hist_with_legends{ SE AI% 299((+;;B?uSV W !d4BO$):?ST3'r#r8Nrc<ddgzddgzz}ttjjdj d|d}d|j _tjtd 5|jd |d dddy#1swYyxYw)Nr4rrrPr,rrr Cannot use both legend and labelrUTc)rr8label) rrr^r_r`rrrWrXrYr/)r1r8rrs r$test_hist_with_legend_raisesz,TestSeriesPlots.test_hist_with_legend_raisessSE AI% 299((+;;B?uSV W ]]:-O P 2 FF$2SF 1 2 2 2s 4BBctjj\}}|jj d|}t |j dk(sJt|jjdy)Nr4r3r@ Frequency) rCrDrEplotr/rarbryaxis get_labelr1rrMr@s r$test_hist_kwargsz TestSeriesPlots.test_hist_kwargss[ ##%2 WW\\qR\ (2::!###288--/=r#ctjj\}}|jj d|}|jj d|}t |j jdy)Nr4r horizontal) orientationr@r)rCrDrErr/rxaxisrrs r$test_hist_kwargs_horizontalz+TestSeriesPlots.test_hist_kwargs_horizontals[ ##%2 WW\\qR\ ( WW\\lr\ :288--/=r#ctjj\}}|jj d|}|jj dd|}y)Nr4rleftT)alignstackedr@)rCrDrErr/rs r$test_hist_kwargs_alignz&TestSeriesPlots.test_hist_kwargs_alignsF ##%2 WW\\qR\ ( WW\\\ <r#zApi changed in 3.6.0)reasonchtjdtjj \}}|j j d|}t|d|j}t|dgt|z|j}t|dgt|zyNscipyT)logyr@logr) rW importorskiprCrDrErr/rget_xticklabelsrraget_yticklabelsr1rrMr@xlabelsylabelss r$ test_hist_kdezTestSeriesPlots.test_hist_kdesG$ ##%2 WW\\t\ +5)$$&7RD3w<$78$$&7RD3w<$78r#cltjdt|jjyNr)rWrr rkder1rs r$test_hist_kde_plot_worksz(TestSeriesPlots.test_hist_kde_plot_workss G$"''++&r#cltjdt|jjyr)rWrr rdensityrs r$test_hist_kde_density_worksz+TestSeriesPlots.test_hist_kde_density_workss G$"''//*r#chtjdtjj \}}|j j d|}t|d|j}t|dgt|z|j}t|dgt|zyr) rWrrCrDrErrrrrrarrs r$test_hist_kde_logyz"TestSeriesPlots.test_hist_kde_logysG$ ##%2 WW[[dr[ *5)$$&7RD3w<$78$$&7RD3w<$78r#c2tjdtjj \}}|j j ddd|}t|dt|jdk(sJt|jdgdzy) NrTr,r)rr3colorr@rr) facecolors) rWrrCrDrErr/rrarbr rs r$test_hist_kde_color_binsz(TestSeriesPlots.test_hist_kde_color_binsstG$ ##%2 WW\\t"CB\ ?5)2::"$$$bjjcURZ8r#c tjdtjj \}}|j j dd|}t|d|j}t|dk(sJt|dgy)NrTr)rrr@rrr?) rWrrCrDrErrr get_linesrar )r1rrMr@liness r$test_hist_kde_colorz#TestSeriesPlots.test_hist_kde_colorsoG$ ##%2 WW[[d#"[ 55) 5zQecU#r#)"__name__ __module__ __qualname__rWmark parametrizer2r=rHrKrNrSr[rdrkslowrrrrrrrrrrrxfailrrrrrrr"r#r$r&r&,s [[XVUOi=Q'RS-T- [[XVQK'89D:D =IBJ -$ * [[ [[* ) * + , , , ,  F FL 2 ) [[ 77  [[03DFV2W ( ( [[TD#;/202> > =  [[45 96 9'+ [[459699$r#r&ceZdZejj dZejj dZejj dZejj dZ ejj ejjdddddd id d igd Z ejj d Z ejj dZ ejj dZejj dZdZejjddddddddddddddddddddddddddddf dZdZdZd Zejjd!dgd"fgd#gd#fgd$Zejjd%gd&d'Zejjd(dd)gejjd*dd+gd,Zejjd(dd)gejjd*dd+gd-Zd.Zd/Zd0Zd1Zd2Zd3Zd4Zy)5TestDataFramePlotsctjtd5t|jdddy#1swYyxYw)NFr6)r9r:r;r r/)r1rjs r$test_hist_df_legacyz&TestDataFramePlots.test_hist_df_legacys3  ' ' e L , gll + , , ,s ;Acttjjdj d}t tjjdj dddtj|d<tjtd5t|jd }dddtd d |d jrJt|dgjy#1swYFxYw)NrPr]j+2ZReZr,sizerFr6)r(rorur{rf)rrr^r_r`rrint64r9r:r;r r/r get_visibler1rcrs r$test_hist_df_legacy_layoutz-TestDataFramePlots.test_hist_df_legacy_layoutsryy,,Q/??H I II ! !! $ - -""hh .  1 ' ' e L :$RWW59D :$6::))+++"aS',,'  : :s C<<Dcttjjdj d}t |j y)NrPr,r?)rrr^r_r`r r/r1rcs r$test_hist_df_legacy_layout2z.TestDataFramePlots.test_hist_df_legacy_layout2s2 ryy,,Q/??H I"''"r#cttjjdj d}t tjjdj dddtj|d<tjtd 5t|jd }dddtd d y#1swYxYw)NrPr,r4rrr,rr4Fr6rrgr{)rrr^r_r`rrrr9r:r;r r/r rs r$test_hist_df_legacy_layout3z.TestDataFramePlots.test_hist_df_legacy_layout3sryy,,Q/??H I II ! !! $ - -""hh .  1 ' ' e L =$RWWV./!3Oa3%i!se)+#6  II ! !! $ 4 4W =#J ! GG   $):?S :&.7D"%otD"I 5 NB ^ 4 5r#c6tddgzddgzzd}ttjj dj d|dd g }t jtd 5|jd ||ddddy#1swYyxYw)Nr4rrrrbrPr]rrrcrrUTr2)rr8rKr) rrrr^r_r`rWrXrYr/)r1r8rKrrcs r$rz/TestDataFramePlots.test_hist_with_legend_raisessa3%i!se)+#6  II ! !! $ 4 4W =#J  ]]:-O P B GG4BvSG A B B Bs 0BBc"ttjjdj d}t j j\}}|jjd|}t|jdk(sJy)NrPr]r4rr, rrr^r_r`rCrDrErr/rarbr1rcrMr@s r$test_hist_df_kwargsz&TestDataFramePlots.test_hist_df_kwargs(sh ryy,,Q/??H I ##%2 WW\\qR\ (2::"$$$r#c>ttjjdj dgd}ddgdz|d<t j j\}}|jjd| }t|jd k(sJy) NrPr,rsArCDr]rrr4Errrkrls r$test_hist_df_with_nonnumericsz0TestDataFramePlots.test_hist_df_with_nonnumerics.s  II ! !! $ 4 4W =( *q.3 ##%2 WW\\qR\ (2::"$$$r#c<ttjjdj dgd}ddgdz|d<t j j\}}|jj| }t|jd k(sJy) NrProrpr]rrr4rt)r@rrkrls r$%test_hist_df_with_nonnumerics_no_binsz8TestDataFramePlots.test_hist_df_with_nonnumerics_no_bins9s  II ! !! $ 4 4W =( *q.3 ##%2 WW\\R\ 2::"$$$r#cttjjdj dt d}t jj\}}|djjd|}|djj|dd t|dd g |jjsJ|jjjsJy) NrProabcdr]rT)rr@rr@r secondary_y b (right)labels)rrr^r_r`listrCrDrErr/r get_yaxisrright_axrls r$test_hist_secondary_legendz-TestDataFramePlots.test_hist_secondary_legendDs  II ! !! $ 4 4W =tF|   ##%2 W\\  dr  2 3 R$? Rk(:;||~))+++{{$$&22444r#cttjjdj dt d}t jj\}}|djjdd|}|djj|dd t|jd d g |jjjrJ|jjsJy) NrProryr]rTrr{r@rrz a (right)r|r}rrr^r_r`rrCrDrErr/r left_axrrrls r$test_hist_secondary_secondaryz0TestDataFramePlots.test_hist_secondary_secondaryTs  II ! !! $ 4 4W =tF|  ##%2 W\\  d  D 3 R$? RZZk0JK::'')55777||~))+++r#cttjjdj dt d}t jj\}}|djjdd|}|djj|d t|jd dg |jjjsJ|jjsJy) NrProryr]rTrr)r@rrr}rrls r$test_hist_secondary_primaryz.TestDataFramePlots.test_hist_secondary_primarycs  II ! !! $ 4 4W =tF|  ##%2 W\\  d  D 3 R- RZZc0BCzz##%11333||~))+++r#c(ttjddgdtjtjggdgtd}tjgd}tgdgdgtd}tjgd gd g}t j j\}}|jj|| |jDcgc])}t|t jjs(|+}}|D cgc]} | j} } t j j\}} |jj| | | jDcgc])}t|t jjs(|+} }| D cgc]} | j} } td t!| | d DsJtjddgddgg}d}t j j\}}t#j$t&|5|jj|| dddycc}wcc} wcc}wcc} w#1swYyxYw)NrR333333?皙?)rQg?rFabcr])?r?)rrRr)rrr)rrr)r@weightsc3,K|] \}}||k(ywr.r").0h0h1s r$ zETestDataFramePlots.test_hist_with_nans_and_weights..sTB28TsTrdrrz?weights must have the same shape as data, or be a single columnrU)rrnanrarrayrCrDrErr/r"r#rbr$r&allrfrWrXrY)r1rcr no_nan_dfno_nan_weightsrMax0rr'rectheightsrQ no_nan_rectsno_nan_heightsidxerror_weightsrZrRs r$test_hist_with_nans_and_weightsz2TestDataFramePlots.test_hist_with_nans_and_weightsss ffc3 #rvvrvv!6 HK ((,-@$u+V #46H"IJ$$&3  W -,,.Wq*Q @U@U2VWW1674??$77$$&3sN;'') Z3;;;P;P-QA  9EE$//+EET#g~d*STTTT88c4[4,$?@O$$&3 ]]:S 1 B NN  30@  A B BX7 F B Bs0*)I4I4I9)I>1I>;J JJ) rrrrWrrrrrr rr rr(r*r/r6rkr?rBrMrZrrrrmrurwrrrrr"r#r$rrs [[,, [[(($ [[## [[ ; ; [[ [[dd3i5IFTU;W11  [[UU [[<< [[**  [[  2 [[f 5 7 7 7 7 8 8 8 8   =  =%0 ")& [[ 0 1 *,I J "", [[ 77 [[TD#;/ [[Xc{35405< [[TD#;/ [[Xc{3 B40 B% % %5 ,, Br#rceZdZdZdZdZdZdZdZdZ dZ e jje jjd gd gd gd gd Ze jjdZe jje jjdgdgdgdgdZe jje jjdidddgdZe jje jjddddddgdgdid dggd!Zd"Zd#Zd$Zd%Zd&Zd'Ze jjd(gd)d*Zy+),TestDataFrameGroupByPlotscvtjjd}t|j ddg}t |j dddtj|d<|j d d d|d <d gdz|d <t|j|j}t|d dyNr,rrqr]rrrrrrsrrXrsrrur{) rr^r_rr`rrrrrqrrr r1rsrcrs r$test_grouped_hist_legacyz2TestDataFrameGroupByPlots.test_grouped_hist_legacys YY " "2 & r))'2SE B KK""hh   3++aB'3%"*3RTTbdd+$6:r#cltjjd}t|j ddg}t |j dddtj|d<|j d d d|d <d gdz|d <|j|j}t|d dyr) rr^r_rr`rrrr/rrr rs r$*test_grouped_hist_legacy_axes_shape_no_colzDTestDataFrameGroupByPlots.test_grouped_hist_legacy_axes_shape_no_cols YY " "2 & r))'2SE B KK""hh   3++aB'3%"*3ww"$$w$6:r#cttjjd}t|j ddg}t |j dddtj|d <|j d d d|d <d gdz|d<|jdd}t|ddt|dy)NrPrrqr]rrr,rrrrsrrrrsr)r8rotr?rfr{rL) rr^r_rr`rrrr/r rrs r$#test_grouped_hist_legacy_single_keyz=TestDataFrameGroupByPlots.test_grouped_hist_legacy_single_keys YY " "1 % r))'2SE B KK""hh   3++aB'3%"*3ww#2w&$6:4b)r#c tjjd}t|j ddg}t |j dddtj|d <|j d d d|d <d \}}d\}}t|j|jdd ||||d }|jD]m}|jD cgc])} t| tjj s(| +} } | dj#} t%j&| dot)|||||ycc} w)NrPrrqr]rrr,rrrrsrrrrT)r8r r3rrrrrrpr!r)rr^r_rr`rrrrrqrrravelr"r#rCrbr$r&r9r%r) r1rrcrrrrrr@rr'ris r$,test_grouped_hist_legacy_grouped_hist_kwargszFTestDataFrameGroupByPlots.test_grouped_hist_legacy_grouped_hist_kwargss; YY " "1 % r))'2SE B KK""hh   3++aB'3B d DDtt  **, 0B??, 1ckk>S>S0TE2Y))+F  " "63 /  0 4BTbtT s )EEcvtjjd}t|j ddg}t |j dddtj|d <|j d d d|d <d gdz|d<t|j|jd}t|dy)NrPrrqr]rrr,rrrrsrrrrsT)r8rrr) rr^r_rr`rrrrrqrrrrs r$%test_grouped_hist_legacy_grouped_histz?TestDataFrameGroupByPlots.test_grouped_hist_legacy_grouped_hists YY " "1 % r))'2SE B KK""hh   3++aB'3%"*3RTTbdd5U+r#ctjjd}t|j ddg}t |j dddtj|d <|j d d d|d <d gdz|d<tjt5t|j|jddddy#1swYyxYw)NrPrrqr]rrr,rrrrsrrrrsr)r8r,)rr^r_rr`rrrr9r-r.rrqrr)r1rrcs r$%test_grouped_hist_legacy_external_errz?TestDataFrameGroupByPlots.test_grouped_hist_legacy_external_errs YY " "1 % r))'2SE B KK""hh   3++aB'3%"*3  % %n 5 4 "$$244U 3 4 4 4s $#CCctjjd}t|j ddg}t |j dddtj|d <|j d d d|d <d gdz|d<d}tjt|5|jd ddddy#1swYyxYw)NrPrrqr]rrr,rrrrsrrrrsz$Specify figure size by tuple insteadrUdefault)r8r)) rr^r_rr`rrrrWrXrYr/)r1rrcrZs r$$test_grouped_hist_legacy_figsize_errz>TestDataFrameGroupByPlots.test_grouped_hist_legacy_figsize_err s YY " "1 % r))'2SE B KK""hh   3++aB'3%"*34 ]]:S 1 / GGsIG . / / /s (CCc,d}ttjjdj dd|}ttjjdj dd|}tjjdj ddg|}t |||d }|jd }|j}t|dk(sJttjjdk(sJy) Nr,rPrr\<rr?)riweightrlrl) rrr^r_normalchoicergroupbyr/rarCrDr)r1nrri gender_intdf_intgbrs r$test_grouped_hist_legacy2z3TestDataFrameGroupByPlots.test_grouped_hist_legacy2s  --a077Ra7HI --a077BQ7GHYY**1-44aV!4D f*UV ^^H %wwy4yA~~3::))+,111r#zmsg, plot_col, by_col, layout)z1Layout of 1x1 must be larger than required size 2rrlrf)z1Layout of 1x3 must be larger than required size 4rirqrV)r;rirqr<c|}tjt|5|j|t |||dddy#1swYyxYw)NrUrKr8rh)rWrXrYr/r})r1rjrZplot_colby_colrhrcs r$test_grouped_hist_layout_errorz8TestDataFrameGroupByPlots.test_grouped_hist_layout_error'sI2 ]]:S 1 L GG8F(;FG K L L Ls AAc|}tjtd5t|jd|j d}dddt ddy#1swYxYw)NFr6rirmrrPr{)r9r:r;r r/rlr rs r$ test_grouped_hist_layout_warningz:TestDataFrameGroupByPlots.test_grouped_hist_layout_warningDsU   ' ' e L $RYYvD  $6:   s $AA"zlayout, check_layout, figsize)rrrrN))rpr?rrN)rr)rr+cd|}|jd|j||}t|d||y)Nri)rKr8rhr)rsr)r/rqr )r1rjrh check_layoutr)rcrs r$ test_grouped_hist_layout_figsizez:TestDataFrameGroupByPlots.test_grouped_hist_layout_figsizeMs1 wwh2;;vwwW$<Qr#r'riru)rKrhc|}tjtd5t|jfddi|}dddt ddy#1swYxYw)NFr6r8rxrorur{)r9r:r;r r/r )r1rjr'rcrs r$#test_grouped_hist_layout_by_warningz=TestDataFrameGroupByPlots.test_grouped_hist_layout_by_warningWsV  ' ' e L H$RWWGGGD H$6: H Hs AAzkwargs, axes_num, layoutrl)ror4rzrPrKrirrqrocH|}|jdi|}t|||y)Nr{r")r/r )r1rjr'r|rhrcrs r$test_grouped_hist_layout_axesz7TestDataFrameGroupByPlots.test_grouped_hist_layout_axes`s(rww  $&Ar#c|}tjjdd\}}|jgd|d}t |ddt j ||d|dj|usJy)NrProrrrKr@rVr{rCrDrEr/r r9assert_numpy_array_equalrJr1rjrcrGrreturneds r$test_grouped_hist_multiple_axesz9TestDataFrameGroupByPlots.test_grouped_hist_multiple_axesmsp JJ''1- T77"BtAw7O(Qv> ##Hd1g6{!!S(((r#c|}tjjdd\}}|jd|d}t |ddt j ||d|dj|usJy) NrProrxr?)r8r@rVr{rrrs r$'test_grouped_hist_multiple_axes_no_colszATestDataFrameGroupByPlots.test_grouped_hist_multiple_axes_no_colswso JJ''1- T77kd1g76(Qv> ##Hd1g6{!!S(((r#c|}tjjdd\}}d}tjt |5|j d|}dddy#1swYyxYw)NrProz@The number of passed axes must be 1, the same as the output plotrUrir)rCrDrErWrXrYr/)r1rjrcrGrrZs r$%test_grouped_hist_multiple_axes_errorz?TestDataFrameGroupByPlots.test_grouped_hist_multiple_axes_errors[ JJ''1- TP ]]:S 1 577(t74D 5 5 5s A  A)c2|}|jd|jd\}}t|j||sJt|j||sJt |j||rJt |j||rJy)NriT)rKr8r r/rlrjoinedrr1rjrcrQrRs r$test_axis_share_xz+TestDataFrameGroupByPlots.test_axis_share_xs 77(ryy7FS#%%c3///#%%c3///c?))#s333c?))#s3333r#c2|}|jd|jd\}}t|j||sJt|j||sJt |j||rJt |j||rJy)NriT)rKr8r )r/rlrrrrs r$test_axis_share_yz+TestDataFrameGroupByPlots.test_axis_share_ys 77(ryy7FS#%%c3///#%%c3///c?))#s333c?))#s3333r#c4|}|jd|jdd\}}t|j||sJt|j||sJt |j||sJt |j||sJy)NriT)rKr8r r rrs r$test_axis_share_xyz,TestDataFrameGroupByPlots.test_axis_share_xys 77(ryyd7SS#%%c3///#%%c3///#%%c3///#%%c3///r#rrcttjjdj dddddg}|j d| }t || y) NrPr?r,r]r\rrr])r8rrr_r`s r$rz0TestDataFrameGroupByPlots.test_histtype_argumentsW II ! !! $ - -a' - BSRUJ WWhW /!"X6r#N)rrrrrrrrrrrrWrrrrrrrrrrrrrrrr"r#r$rrs;"; *$"UH,"4"/" 2 [[ [['      .L/0L  [[;; [[ [[' !8:STR  R  [[ [[Xx6,R'ST;U; [[ [[" /F ;8 91f E BB ))5 4 4 0 [[ 77r#r)#__doc__r=numpyrrWpandasrrrrrpandas._testing_testingr9pandas.tests.plotting.commonrr r r r r rrrrrrCr pandas.plotting._matplotlib.histrfixturerr&rrr"r#r$rs!     f,'f-.:N$N$bQBQBh l7l7r#