o Df<@s dZddlmZmZddlZddlZddlZddlZddlZddlZ ddl Z ddl m m ZddlmZddlZddlmZmZmZmZmZmZmZmZmZddlmZddlm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,ddl-m.Z.e /dZ0e /d Z1Gd d d Z2d d Z3dS)z Test cases for DataFrame.plot )datedatetimeN) is_list_like) DataFrameIndex MultiIndex PeriodIndexSeries bdate_range date_rangeoption_contextplotting) _check_ax_scales_check_axes_shape_check_box_return_type _check_colors _check_data_check_grid_settings_check_has_errorbars_check_legend_labels_check_plot_works_check_text_labels_check_ticks_props_check_visible get_y_axis pprint_thing matplotlibzmatplotlib.pyplotc@seZdZejjddZejjddZejjddZejjddZ ejj d d ejjd d Z ejjd dZ ejjej ddgdidgdidddddddgddZejjddZejjddZejjddZejjd d!Zejjej d"d#d$gd%d&Zejjd'd(Zej d)ed*d+d,gd+d,gd+d,gd-d+gd-d+gd+d-gd+d-gd,d-gd,d-gd+d.gd+d.gd+d/gd+d/ggd0d1Zejjej d2gd3d4d5Zejjej d6d7igd8igd9igd:d;d;igdgd?d=d;d>ggd@dAZdBdCZdDdEZdFdGZdHdIZdJdKZdLdMZej dNdOdPgdQdRZej dSgdTdUdVZ dWdXZ!dYdZZ"d[d\Z#d]d^Z$d_d`Z%dadbZ&ej dce'j(e'j)gdddeZ*ej dfe+d;gdgigdhdie+d;djdke'j,dldmdndogidkdjdldpe'j,dldpgdigdqdrZ-dsdtZ.dudvZ/dwdxZ0ej dyd7dzgej d{d|d}gd~dZ1ej dyd7dzgddZ2ddZ3ej dyd7dzgddZ4ej de5de6dd.ddgddZ7ej de5de6dd.ddgddZ8ej de5de6dd.ddgej diddigddZ9ej diddigddZ:ddZ;ejj dddej dddgddZddZ?ej dddgej dddgddZ@ej dddgddZAddZBddZCddZDddZEddZFej deGeHjIgddZJej dd>dgddZKddZLddZMddZNej dddgddZOej ddejPdeQRdádčgej dddgej dgdǢgdȢgddʄZSej dddgej dgd͢gd΢fddЄZTej dgdѢddӄZUej dddgdd؄ZVddڄZWdd܄ZXddބZYddZZddZ[ejjej diddidddddigddZ\ejjddZ]ejjddZ^ejjddZ_ddZ`ddZaddZbddZcejddddZeddZfddZgej dgdddZhddZiddZjddZkddZldd Zmd d Znd d ZoddZpddZqddZrej dde'jsddde'jsddgddZt # # # #dddZuej de've'wgde'wgde've'wgde'wgd e've'wgde'wgd!d"e've'we'j,d|d#d$dd%ge'wgd&e've'wd|e'j,d#d$dd%ge'wgd'e've'wd|d#d$e'j,dd%ge'wgd(d"gd)d*Zxd+d,Zyej d-d.d/d0d֜d.d/dgd1d.d/ggd2d3Zzd4d5Z{ej d6gd7d8d9Z|d:d;Z}ej dye~jjd<d=Zej dydd?Zej dye~jjd@dAZej dyee~jjdzgdBdCZdDdEZej dFd,d-gd+dGfd+gd,d-gdHdIgfgdJdKZdLdMZej dNdOd+gd,fgdPdQZej dRd+d,gdHgdSgfd+d,d-gdHdIgdSdTgfdUd|d#gdVdWgdXdYgfgdZd[Zej d\dUd|d+d,gfd|dUdUd|gfgd]d^Zd_d`ZdadbZej dcd#e'jgdddeZej dfidgfdhdiidifdjdkidkfgdldmZdndoZej d;dpd#gdqdrZdsdtZdudvZdwdxZejjej ddydidddzd{digd|d}Zejjd~dZejjddZejjej dddgddZejjddZejjddZejjej dygdej dee'sddd=de+e'sdde'sddd>gddZejjej dygdej de+e'sdde'sddd>e+e'sdde'sddd>fee'sddd=dee'sddd=dfdgddZejjej dygdddZejj dedddZddZejjej dyd7d8gddZejjddZejjej ddde+gddZej dygdddZddZddZddZddZddZddZddZddZddZej dye~jjddZddZddZdd„ZdÐdĄZdŐdƄZdǐdȄZdɐdʄZej dgd̢d͐d΄ZdϐdЄZdѐd҄Zej dyd7dzgdӐdԄZdՐdքZdאd؄ZdِdڄZej dydۡdܐd݄Zdސd߄ZddZddZej dgdddZÐddZĐddZej dydddZej dddddd#dd|d#gfgej dygdddZej dgdej dydrCr"rrrArDrEr9r9r:"test_plot_subplots_negative_layoutKs z5TestDataFramePlots.test_plot_subplots_negative_layoutcCsbttjddttdtdtddddd }t |j d d d d }t |d dt |ddddS)Nrr r#r$r&r!r'r(r+TF)r=r> use_indexrxrotr"r?rA) rr/r0r1r2rr3r4r rr5rrrEr9r9r:test_plot_subplots_use_indexZs  z/TestDataFramePlots.test_plot_subplots_use_indexzApi changed in 3.6.0)reasoncCs\tddgddgd}d}tjt|d|jjdd WddS1s'wYdS) Nr@rr"xyz''Line2D' object has no property 'blarg'matchT)blarg)rpytestraisesAttributeErrorr5liner7r8msgr9r9r:test_plot_invalid_argjs "z(TestDataFramePlots.test_plot_invalid_argcCsFttjddttjddd}t|jdd}t |dddS) Nrr!rPr!r-TrKrrL) rr/r0r1r3string ascii_lettersrr5rr7r8axr9r9r:test_plot_tick_propsrs z'TestDataFramePlots.test_plot_tick_propskwargsyticks)r@r!xticks)id)ylimxlimTblah)r=r>titlecCs>ttjddttjddd}t|jfi|dS)Nrr^r!r_ rr/r0r1r3rarbrr5r7rfr8r9r9r:test_plot_other_args|s  z'TestDataFramePlots.test_plot_other_argscCsttjddttjddd}|jddd}t|dd d |ddD]#}t |j t | d d t |j dd d d t |j gd d q(|dfD]}t |j t | t |j gt |ddqQdS)Nrr^r!r_Trm)r>rnrPrPr@rAFvisibleminorrrL)rr/r0r1r3rarbr5rrxaxisget_xticklabels get_labelr)r7r8rFrdr9r9r:test_plot_visible_axs"    z'TestDataFramePlots.test_plot_visible_axcCs:ttjddttjddd}t|jdddS)Nrr^r!r_rmrnror6r9r9r:test_plot_titles z"TestDataFramePlots.test_plot_titlecCsVttjddtd}ttjddt |d}t |j dd}t |dddS) Nr!rr^r_Tr`rrL) ziprarbrangerr/r0r1r from_tuplesrr5r)r7tuplesr8rdr9r9r:test_plot_multiindexsz'TestDataFramePlots.test_plot_multiindexcCs^tjgdddgd}tjddgddgd}ttjd d d d ||d }t|jdddS)N))αr)rr@)βr)rrP)γr")rrh)δ)ri0i1)names)baruΔ)ruΕc0c1rrr!)rr+uΣr{) rrrr/r0r1integersrr5)r7r-r,r8r9r9r:test_plot_multiindex_unicodes  z/TestDataFramePlots.test_plot_multiindex_unicoderCN)rHr@cCs>tdtjddi}t|jjd|d}t|ddddS) NrRrr!T)r>rCr@r@r@rA)rr/r0r1rr5rr)r7rCr8rFr9r9r:test_plot_single_column_barsz.TestDataFramePlots.test_plot_single_column_barcCsbtdtjddi}tj\}}|jjd|d}t |dks$J|j }||dus/JdS)NrRrr!Tr>rdr@r) rr/r0r1mplpyplotr>r5rlenrF)r7r8_rdrFresultr9r9r:test_plot_passed_axs z&TestDataFramePlots.test_plot_passed_axz cols, x, yABCDEAr'CDEc Cstgd}tgdgdtjddtjddgtdtj|dd d tj|dd d d d }t||j ||ddS)N)20082009N20112012r@rrPr"rhrrhrPrr$z%Yformati8T)rutcrr'rrrrQ) rr/arraynanr4pd to_datetimeviewrr5)r7colsrRrSdatesr8r9r9r:test_nullable_int_plots z)TestDataFramePlots.test_nullable_int_plotr5)rZrhistpiecCs.tjgddd}t|}tt|j|dS)Nr@rrPr"UInt32r$)rrr rgetattrr5)r7r5arrsr9r9r:test_integer_array_plot_seriessz1TestDataFramePlots.test_integer_array_plot_seriesz plot, kwargsrZrrrrSscatterrRrQhexbincCs<tjgddd}t||d}tt|j|fi|dS)Nrrr$rQ)rrrrrr5)r7r5rfrr8r9r9r:test_integer_array_plot_df sz-TestDataFramePlots.test_integer_array_plot_dfcCs6tgdgdd}|}t|dksJdS)NrRrSzr@rrPrr'r@)rr5r get_linesrcr9r9r:test_nonnumeric_excludesz*TestDataFramePlots.test_nonnumeric_excludecCs@ttjddgdd}|jddd}t|jddS)Nrr^abcr,rrrQ) rr/r0r1r2r5rrwryrcr9r9r:test_implicit_label s z&TestDataFramePlots.test_implicit_labelcCsHttjddddgd}d|j_|jddd|jjdks"JdS) NrrIrrrNAMELABEL)rSlabel)rr/r0r1r2r-namer5r6r9r9r:test_donot_overwrite_index_name's z2TestDataFramePlots.test_donot_overwrite_index_namecCsttjddttdtdtddddd }t |j d d d | d d t |j d d| d  t |j d d|j t |j d dd | d j t |j d d| d  t |j dd|j dS)Nrrhr"r#r$r&rhr'r(r+rr@rQrrRrS) rr/r0r1r2rr3r4r rr5 set_indexr'r6r9r9r: test_plot_xy0s " zTestDataFramePlots.test_plot_xycCsttjddttdtdtddddd }t d t |j d |_ t |j d dd |d d t |j d d |d  t |j d d |d  dS)Nrrr#r$r&rhr'r(r+r@rQrr)rr/r0r1r2rr3r4r arangerr,rr5rr6r9r9r:test_plot_xy_int_cols>s "z(TestDataFramePlots.test_plot_xy_int_colscCsdttjddttdtdtddddd }|j d dd d d }t |j d t |d ddddS)Nrrr#r$r&rhr'r(r+r@Test)r)rRrSrnfigsizer)g0@g @)rBrCr) rr/r0r1r2rr3r4r r5rrnrrcr9r9r:test_plot_xy_figsize_and_titleJs  z1TestDataFramePlots.test_plot_xy_figsize_and_titlezinput_log, expected_log)Tlog)symsymlogcCstdtditdd}|j|d}t||d||ks"J|j|d}t||d||ks6J|j|d}t|||d ||ksKJ||ksSJdS) Nrrjr_logyyaxis)logxrwloglog)rwr)rr/rr5r get_yscale get_xscale)r7 input_log expected_logr8rdr9r9r:test_logscalesXs     z!TestDataFramePlots.test_logscales input_param)rrrcCstdtditdd}d|d}tjt|d|jd i|diWdn1s0wYd|d }tjt |d|jj d d d i|d iWddS1s[wYdS) Nrrjr_z keyword 'z*' should be bool, None, or 'sym', not 'sm'rTsmzPiePlot ignores the 'z ' keywordr>Tr9) rr/rrWrX ValueErrorr5tmassert_produces_warning UserWarningr)r7rr8r\r9r9r:test_invalid_logscaleks  "z(TestDataFramePlots.test_invalid_logscalecCslttjddttdtdtddddd }|j d d }| }t |d  t r.Jt|d ddS)Nrr r#r$r&r!r'r(r+T)x_compatrrL)rr/r0r1r2rr3r4r r5r isinstance get_xdatarrr7r8rdlinesr9r9r: test_xcompatxs  zTestDataFramePlots.test_xcompatcCsrttjddttdtdtddddd }d t j d <| }| }t |d tr1Jt|d ddS)Nrr r#r$r&r!r'r(r+Tz xaxis.compatrrrL)rr/r0r1r2rr3r4r r plot_paramsr5rrrrrrr9r9r:test_xcompat_plot_paramss  z+TestDataFramePlots.test_xcompat_plot_paramscCsttjddttdtdtddddd }d t j d <| }| }t |d tr1Jd }tjt|dt t|d tsHJWddS1sSwYdS)Nrr r#r$r&r!r'r(r+FrrzPeriodDtype\[B\] is deprecatedrT)rr/r0r1r2rr3r4r r rr5rrrrrr FutureWarningr7r8rdrr\r9r9r:!test_xcompat_plot_params_x_compats  "z4TestDataFramePlots.test_xcompat_plot_params_x_compatcCsttjddttdtdtddddd }t j d d "| }| }t|d tr4Jt|d dWddS1sEwYdS)Nrr r#r$r&r!r'r(r+rTrrrL)rr/r0r1r2rr3r4r r ruser5rrrrrrr9r9r:(test_xcompat_plot_params_context_managers "z;TestDataFramePlots.test_xcompat_plot_params_context_managercCsttjddttdtdtddddd }| }| }t |d  t r,Jd }tjt|d t t |d  t sCJWdn1sMwYt|d d dS)Nrr r#r$r&r!r'r(r+rzPeriodDtype\[B\] is deprecated rTrL)rr/r0r1r2rr3r4r r5rrrrrrrrrr9r9r:test_xcompat_plot_periods z+TestDataFramePlots.test_xcompat_plot_periodcCsPttjddttdddtdddddgd}|tjj d d dS) Nr)rr@rrr-r,rr) rr/r0r1r rr5rraxhliner6r9r9r:test_period_compatsz%TestDataFramePlots.test_period_compat index_dtypecCstdtdittddd|dtjd}|}|d}|}t|dddf|dddftjdd }t j ||j d d dS) NrSrjcrHr$)r-r%rr@)r%rF)check_index_type) rr/rrint64r5r get_xydatar rassert_series_equalrS)r7rr8rdrrsr9r9r:test_unsorted_indexs  *z&TestDataFramePlots.test_unsorted_indexr8?@@)rr rrr_r rrg@g@g@rcCs\|}|\}}|}|t|ddksJ|t|ddks,JdS)Nr)r5get_xlimrr/nanminget_datananmaxr7r8rdxminxmaxrr9r9r:test_unsorted_index_limss  "z+TestDataFramePlots.test_unsorted_index_limscCsxtgdgdd}|jddd}|\}}|}|t|ddks+J|t|ddks:JdS)Nr )gV@gV@g@W@gW@)rSrrrSrQr)rr5rrr/rrrrr9r9r:test_unsorted_index_lims_x_ys  "z/TestDataFramePlots.test_unsorted_index_lims_x_ycCsttjddttjddgdd }d}tjt |d|j j dd Wdn1s3wYtjt |d|j j dd WddS1sRwYdS) Nrrr"rrRrSrfourrz+Log-y scales are not supported in area plotrTTrr) rr/r0r1r3rarbrWrXrr5arear[r9r9r:test_negative_logs"z$TestDataFramePlots.test_negative_logcCsZtt|dd}t||D]\}}||d7}|d}t||qdS)Nrr@)r/zerosrrr}rassert_numpy_array_equal)r7 normal_lines stacked_linesbasenlslsyr9r9r:_compare_stacked_y_coods  z*TestDataFramePlots._compare_stacked_y_coodkindrmultr@rHcCsV|ttjddgdd}t|j|dd}t|j|dd}||j|jdS)NrrwrRrSrrFr(stackedTrr/r0r1rr5r'r)r7r(r)r8ax1ax2r9r9r:test_line_area_stackeds z)TestDataFramePlots.test_line_area_stackedcCsttjddtjddtjdd tjdd d}t|j|dd}t|j|dd}||jdd|jdd||jdd|jdddS)Nrrr*Fr,Tr.)r7r(sep_dfr/r0r9r9r:test_line_area_stacked_sep_df s $z0TestDataFramePlots.test_line_area_stacked_sep_dfcCsttjddttjddgdd}t|j ddd}t j t |d |j d dWddS1s9wYdS) Nrrrr*rFr-zWhen stacked is True, each column must be either all positive or all negative. Column 'w' contains both positive and negative valuesrTT) rr/r0r1r2r3rarbrr5rWrXr)r7mixed_dfr\r9r9r:test_line_area_stacked_mixeds"z/TestDataFramePlots.test_line_area_stacked_mixedcCsDttjddgdd}||jd}t|j|ddddS)Nrrr*rr@T)r(rr-)rr/r0r1rr-rr5)r7r(r8df2r9r9r:#test_line_area_stacked_positive_idx)s z6TestDataFramePlots.test_line_area_stacked_positive_idxidxr"z 2023-01-1)r*r)c Csddtjdg}dtjddg}t||d|d}t|j}|jd}|jd}tjgdtjd}t t |j d|tjgd tjd}t t |j d|t |j tgd t |j tgd dS) Nr@rrPrrr_rrr$)rPrr@)FFTF)FTFF)r/rrrr5r get_ydatarfloat64rr deletedatamask) r7r9values1values2r8rdmasked1masked2expr9r9r:test_line_area_nan_df3s z(TestDataFramePlots.test_line_area_nan_dfcCsddtjdg}dtjddg}t||d|d}tjgdtjd}tjgdtjd}t|jd d }t|j d  |t|j d ||dS) Nr@rrPr:r_r@rrrPr$rPrrr@Tr4r) r/rrrr<rr5rr rr;)r7r9r@rAr8 expected1 expected2rdr9r9r:test_line_area_nan_df_stackedHsz0TestDataFramePlots.test_line_area_nan_df_stackedr-Fc Csddtjdg}dtjddg}t||d|d}tjgdtjd}tjgdtjd}t|jjfi|}t |j d  ||rNt |j d |n t |j d ||t|jjd d }t |j d  |t |j d |dS) Nr@rrPr:r_rFr$rGrFr4) r/rrrr<rr5rrr rr;) r7r9rfr@rAr8rHrIrdr9r9r:"test_line_area_nan_df_stacked_areaWsz5TestDataFramePlots.test_line_area_nan_df_stacked_area secondary_ycCs~ttjddgdd}|jdi|}|\}}|}||dddks/J||dddks=JdS)NrrrPrrrrHr9)rr/r0r1r5rrr)r7rfr8rdrrrr9r9r: test_line_limns   z TestDataFramePlots.test_line_limcCsttjddgdd}|jddd}t|ddd |D]6}t|d s(Jt|d r/J|\}}|}||d  d d ksGJ||d  d d ksUJqdS)NrrMrrT)rLr>rPrrrAleft_axright_axrrH) rr/r0r1r5rhasattrrrr)r7r8rFrdrrrr9r9r:test_line_lim_subplotsws z)TestDataFramePlots.test_line_lim_subplotszX2020-12-01 this has been failing periodically on the ymin==0 assertion for a week or so.)strictrOc Csttjddgdd}| }t|jj|d}|\}}|\}}| } || d ddks8J|| d ddksFJ|dksLJt|jj|d}|\}}|dks`JdS)Nrrrrr4rrH) rr/r0r1rr5rrget_ylimrr) r7r-r8neg_dfrdrryminymaxrr9r9r: test_area_lims    z TestDataFramePlots.test_area_limcCsttjddddgd}tjjdddd\}\}}|j|d d |j|d d t| ||s4Jt| ||s>JdS) Nrr"rrRrSrr@T)shareyr)rdr() rr/r0r1rrr>r5rjoined)r7r8figr/r0r9r9r:test_area_sharey_dont_overwrites z2TestDataFramePlots.test_area_sharey_dont_overwritecCsFttjdd}|jj|dd}|jD] }|dks JqdS)Nrrhrh)r- linewidth) rr/r0r1r2r5rpatches get_linewidth)r7r-r8rdrr9r9r:test_bar_linewidths  z%TestDataFramePlots.test_bar_linewidthcCs^ttjdd}|jjddd}t|ddd|D]}|jD] }| dks+Jq!qdS)Nrr^T)r_r>rh)rhr@rA) rr/r0r1r2r5rrr`ra)r7r8rFrdrbr9r9r:test_bar_linewidth_subplotss z.TestDataFramePlots.test_bar_linewidth_subplotsz meth, dim)r get_width)barh get_heightcCsvttjdd}d}t|j|||d}|jD]}|s.t|||t|j ks-Jqt|||ks8JqdS)Nrr^?)r-width) rr/r0r1r2rr5r`rr,)r7methdimr-r8rirdrbr9r9r:test_bar_barwidths  z$TestDataFramePlots.test_bar_barwidthcCs\ttjdd}d}t|j||dd}|D]}|jD] }t|||ks*JqqdS)Nrr^rhT)rir>)rr/r0r1r2rr5r`)r7rjrkr8rirFrdrbr9r9r:test_barh_barwidth_subplotss z.TestDataFramePlots.test_barh_barwidth_subplotscCsttjdd}|jjddd}dd|jD}|dgdks$J|jjd gd d}d d|jdd D}|gd ksBJdS) Nrr^Fr@)r-bottomcSg|]}|qSr9get_y.0pr9r9r: zBTestDataFramePlots.test_bar_bottom_left_bottom..T)rHcSror9rprrr9r9r:rurvrh)rr/r0r1r5rr`r7r8rdrr9r9r:test_bar_bottom_left_bottomsz.TestDataFramePlots.test_bar_bottom_left_bottomcCsttjdd}|jjdtgdd}dd|jD}|dgd ks)J|jjd gd d}d d|jdd D}|gd ksGJdS)Nrr^Fr@r@r@r@r@)r-leftcSror9get_xrrr9r9r:rurvz@TestDataFramePlots.test_bar_bottom_left_left..r@rwTrcSror9rrrr9r9r:rurvrh)rr/r0r1r5rfrr`r|r9r9r:test_bar_bottom_left_leftsz,TestDataFramePlots.test_bar_bottom_left_leftcCsttjdd}|jjddd}|D]}dd|jD}|dgdks(Jq|jjdtgd d }|D]}d d|jD}|d gdksKJq8dS) Nrr^TrH)r>rncSror9rprrr9r9r:rurvzDTestDataFramePlots.test_bar_bottom_left_subplots..rhr~)r>rcSror9rrrr9r9r:rurvr@) rr/r0r1r5rr`rfr)r7r8rFrdrr9r9r:test_bar_bottom_left_subplotssz0TestDataFramePlots.test_bar_bottom_left_subplotscCsRtdtjdggdgdd}|j}gd}dd|jD}||ks'JdS) Nr!rhr!rrrr'r r!rrrhr!rr@rrPcSror9rgrrr9r9r:rurvz3TestDataFramePlots.test_bar_nan..rr/rr5rr`r7r8rdexpectedrr9r9r: test_bar_nans  zTestDataFramePlots.test_bar_nancCsztdtjdggdgdd}|jjdd}gd}d d |jD}||ks)Jd d |jD}gd }||ks;JdS) Nr!rrrrTr4rcSror9rrrr9r9r:rurvz;TestDataFramePlots.test_bar_nan_stacked..cSror9rprrr9r9r:rurv) r r r $@r g4@g.@rgD@rrr9r9r:test_bar_nan_stackeds  z'TestDataFramePlots.test_bar_nan_stackedcCsttjdd|td|tdd}|j}|j }t |t gd| dks3J|jdd ks>J|jd d ksIJ|jjd d }t |t gd| dkscJ|jdd ksnJ|jd dksyJdS)Nr)rrhABCDEFabcder)rr@rrPr"rh)gg@rgпrHg@Tr4g@)rr/r0r1r2r3r5rrw get_ticklocsrr rrr`r)r7r9r8rdticksr9r9r:test_bar_categoricals     z'TestDataFramePlots.test_bar_categoricalzx, yr@rcCsDttjddttjddgdd}t|j j ||ddS)NrrrrrrQ) rr/r0r1r2r3rarbrr5rr7rRrSr8r9r9r:test_plot_scatters z$TestDataFramePlots.test_plot_scattercCsttjddttjddgdd}t d}t j t |d|j jdd Wdn1s5wYt d }t j t |d|j jd d WddS1sYwYdS) Nrrrrrz5scatter() missing 1 required positional argument: 'y'rTrRrz5scatter() missing 1 required positional argument: 'x'rSr)rr/r0r1r2r3rarbreescaperWrX TypeErrorr5rr[r9r9r:test_plot_scatter_error(s  "z*TestDataFramePlots.test_plot_scatter_errorcCsRttjddttjddgdd}|jdddd d }t |d d d dS)NrrrrrrRrSrT)rRrSr(r>r@rrA) rr/r0r1r2r3rarbr5rrEr9r9r:test_plot_scatter_shape5sz*TestDataFramePlots.test_plot_scatter_shapecCsxttjdddgd}tddddj|d<d }tjt |d |j d ddd WddS1s5wYdS) Nrr!rrz 2014-01-01h)startr*r)dtimez9must be a string or a (real )?number, not 'datetime.time'rTr)r(rRrS) rr/r0r1r2r timerWrXrr5r[r9r9r:&test_raise_error_on_datetime_time_data?s "z9TestDataFramePlots.test_raise_error_on_datetime_time_datarvalsrr@cCsTttdddddd}tjdddt|}t||d}t|j j ||d dS) Nir@ W)rr)r*rrrrQ) r rr/r0r1normalrrrr5r)r7rRrSrrr8r9r9r:test_scatterplot_datetime_dataHsz1TestDataFramePlots.test_scatterplot_datetime_data infer_stringpyarrow)marksr:b_colrrPr"rcCsTtd|tgd|d}t|jj||dWddS1s#wYdS)Nzfuture.infer_stringrr:rQ)r rrr5r)r7rrRrSrr8r9r9r:test_scatterplot_object_dataQs "z/TestDataFramePlots.test_scatterplot_object_dataordered categories)setosa versicolor virginica)rrrc Cstddgddgddgddgddggd d gd }tjgd ||d |d<|jjdddd}|j\}|j}tgd}| }t ||tgd} |j } t | | |} dd|j D} tddt| | DsnJdS)Ngffffff@g @g@rg@g @g@g@lengthrir)rrrrr)rrspeciesrr@rRrSr)?g?g@r cSror9get_text)rsir9r9r:ruyrvzLTestDataFramePlots.test_scatterplot_color_by_categorical..css|] \}}||kVqdSNr9)rsrjr9r9r: zszKTestDataFramePlots.test_scatterplot_color_by_categorical..)rr Categoricalr5r collectionscolorbarr/r get_ticksrr  _boundariesrdget_ymajorticklabelsallr}) r7rrr8rdcolorbar_collectionrexpected_ticks result_ticksexpected_boundariesresult_boundariesexpected_yticklabelsresult_yticklabelsr9r9r:%test_scatterplot_color_by_categorical]s*     z8TestDataFramePlots.test_scatterplot_color_by_categorical)rQ)rSrR)rSrScCs2tgdtgdd}t|jj||ddS)Nr)rrrrrQ)rrrrr5rrr9r9r:'test_plot_scatter_with_categorical_data|sz:TestDataFramePlots.test_plot_scatter_with_categorical_datazx, y, cr)rr@rcCszttjdjddddttjddgdd}|jj |||d }|j dj j d ks.J|j dj jd ks;JdS) NrrrjrlowhighsizerrrrGreysr)rr/r0r1rr3rarbr5rrcmaprrrd get_ylabel)r7rRrSrr8rdr9r9r:test_plot_scatter_with_csz+TestDataFramePlots.test_plot_scatter_with_ccCsttjdjddddttjddgdd}d }|jj d d d |d }|j dj j |ks1J|jj d d d dd}|j dj dusEJ|jj dddd}|j dj dusXJt|j dgddS)Nrrrjrrrrr cubehelixrRrSr)rRrSrcolormapF)rRrSrrr@redrrb facecolors)rr/r0r1rr3rarbr5rrrrrr)r7r8cmrdr9r9r:test_plot_scatter_with_c_propssz1TestDataFramePlots.test_plot_scatter_with_c_propscCstddgddgd}gd}gd}t||g}|jjdd |d }t|jd |td d g}|jjdd |dddS)Nr@rrPr"r)rr r r)r rr rrr'rrr rspring)rRrSrr) rr/rr5rrr r get_facecolor)r7r8red_rgba green_rgba rgba_arrayrd float_arrayr9r9r:test_plot_scatter_with_c_arraysz1TestDataFramePlots.test_plot_scatter_with_c_arraycCsVttjdddgdd}|jjdddd }tj|dj|j d  d dS) Nrr^rjrrrrr)rRrSrr)right) rr/r0r1r5rrr valuesr get_sizesrcr9r9r:test_plot_scatter_with_ss "z+TestDataFramePlots.test_plot_scatter_with_scCsXttjdddgdd}tj}|jjddd|d }|j d j |us*JdS) Nrr^rjrrrrr)rRrSrnormr) rr/r0r1rcolorsLogNormr5rrr)r7r8rrdr9r9r:test_plot_scatter_with_norms  z.TestDataFramePlots.test_plot_scatter_with_normcCsttjdddgdd}|jjdddd }|jd j}|j |j f}t j j |}|jD] }||||ks?Jq3dS) Nrr^rjrrrrrrr)rr/r0r1r5rrrrminmaxrr Normalize)r7r8rd plot_norm color_min_max default_normvaluer9r9r:test_plot_scatter_without_norms   z1TestDataFramePlots.test_plot_scatter_without_normlegendr<cCsFttjddttjddgdd}t|j j fi|dS)Nrrronetwothreerr) rr/r0r1r2r3rarbrr5rrpr9r9r: test_plot_bars  z TestDataFramePlots.test_plot_barcCs>ttjddttjddtdd}t |j j dS)Nr)r!r!rr) rr/r0r1r2r3rarbr~rr5rr6r9r9r:test_plot_bar_int_cols z(TestDataFramePlots.test_plot_bar_int_colcCsRtddgddgd}t|jj}t|dd|jjddd}t|dddd dS) Nrr@r:ZrL#r!rotfontsizerM xlabelsize ylabelsize)rrr5rrrcr9r9r:test_plot_bar_ticks   z&TestDataFramePlots.test_plot_bar_tickscCsRtddgddgd}t|jj}t|dd|jjddd}t|dddddS) Nrr@r:)yrot7 r)rrr)rrr5rfrrcr9r9r:test_plot_barh_ticksrz'TestDataFramePlots.test_plot_barh_tickscCsv|}|j}dd|D}t|jj}t||t|j t dt |dt |jdt |ks9JdS)NcSg|]}t|qSr9rrsrr9r9r:ru rvz3TestDataFramePlots.test_boxplot..r@r)_get_numeric_datar,rr5boxrrxrr rwrr/rrrr7hist_dfr8 numeric_colslabelsrdr9r9r: test_boxplots  zTestDataFramePlots.test_boxplotcCs8|}|d}|jjdd}t|dddt|jjdS)Nheight(rrrMr)r5r rr)r7rr8seriesrFr9r9r:test_boxplot_seriess z&TestDataFramePlots.test_boxplot_seriescCsv|}tgd}|jj|d}|j}dd|D}t||t |j |t |j dt |ks9JdS)N)r@rr) positionscSr r9rr r9r9r:ru rvzDTestDataFramePlots.test_boxplot_series_positions..r)r/rr5r r r,rrxrr rwrrr)r7rr8rrdrrr9r9r:test_boxplot_series_positionss z0TestDataFramePlots.test_boxplot_series_positionscCsj|}|j}dd|D}|jjdddd}t|ddddt||t|jd t|ks3JdS) NcSr r9rr r9r9r:ru(rvz.2rF)rrvertrrMrrr) r r,r5r rrget_yticklabelsrrr r9r9r:test_boxplot_vertical%s z(TestDataFramePlots.test_boxplot_verticalzignore:Attempt:UserWarningcCs|}|j}dd|D}t|jjddddd}t|dddt|d d t||D]\}}t| |gt |j d ks@Jq+dS) NcSr r9rr r9r9r:ru4rvzETestDataFramePlots.test_boxplot_vertical_subplots..TF)r=r>rrrP)r@rPrArrr) r r,rr5r rrr}rrrr)r7rr8rrrFrdrr9r9r:test_boxplot_vertical_subplots0s   z1TestDataFramePlots.test_boxplot_vertical_subplotscCsx|}|j}dd|D}tgd}|jj|dd}t||t |j |t |j dt |ks:JdS)NcSr r9rr r9r9r:ruErvzFTestDataFramePlots.test_boxplot_vertical_positions..)rPrrF)rrr)r r,r/rr5r rrrr rrrr)r7rr8rrrrdr9r9r:test_boxplot_vertical_positionsBs z2TestDataFramePlots.test_boxplot_vertical_positionscCstttjddttjddgdd}d}tj t |d|j j dd WddS1s3wYdS) Nrrrrrz2return_type must be {None, 'axes', 'dict', 'both'}rT not_a_type return_type) rr/r0r1r2r3rarbrWrXrr5r r[r9r9r: test_boxplot_return_type_invalidLs"z3TestDataFramePlots.test_boxplot_return_type_invalidr#)dictrFbothcCsJttjddttjddgdd}|jj |d}t ||dS)Nrrrrrr") rr/r0r1r2r3rarbr5r r)r7r#r8rr9r9r:%test_boxplot_return_type_invalid_typeVsz8TestDataFramePlots.test_boxplot_return_type_invalid_typecCsZtdttjdd}t|jdd}dd|j D}t ||dt |d d dS) Nscipyrrjr"kder(cSr r9rr r9r9r:rudrvz2TestDataFramePlots.test_kde_df..rrrL) rW importorskiprr/r0r1r2rr5r,rr)r7r8rdrr9r9r: test_kde_df`s  zTestDataFramePlots.test_kde_dfcCsDtdttjdd}|jdddd}t|dddddS) Nr(rr r*rrh)r(rrr) rWr-rr/r0r1r2r5rrcr9r9r:test_kde_df_roths z"TestDataFramePlots.test_kde_df_rotcCsDtdttjdd}t|jdddd}t |ddd dS) Nr(rr Tr*)r=r(r>r"r?rA) rWr-rr/r0r1r2rr5rrEr9r9r:test_kde_df_subplotsns z'TestDataFramePlots.test_kde_df_subplotscCs@tdttjdd}|jdddd}t|dddS) Nr(rr r*T)r(rr>rr) rWr-rr/r0r1r2r5rrEr9r9r:test_kde_df_logyys z#TestDataFramePlots.test_kde_df_logycCs@tdttjdjdd}tj|jd<t |j dddS)Nr(rr)rrrr*r+) rWr-rr/r0r1uniformrlocrr5r6r9r9r:test_kde_missing_valss  z(TestDataFramePlots.test_kde_missing_valscCspttjdd}t|jj}dd|jD}t ||dt|jjdddd}t |dd d t |d d dS) Nrr)cSr r9rr r9r9r:rurvz3TestDataFramePlots.test_hist_df..r,T)r=r>rr"r?rArr) rr/r0r1r2rr5rr,rrr)r7r8rdrrFr9r9r: test_hist_dfs  zTestDataFramePlots.test_hist_dfcCs6ttjdd}|jjdd}t|ddddS)Nrr!rrrr)r r/r0r1r5rr)r7rrFr9r9r:test_hist_df_seriessz&TestDataFramePlots.test_hist_df_seriescsbddlmttjdd}|jjdddd}fdd |D}t |d  d dS) Nr Rectanglerr!Tr") cumulativebinsdensitycg|] }t|r|qSr9rrsrRr9r9r:ruzMTestDataFramePlots.test_hist_df_series_cumulative_density..rHr matplotlib.patchesr:r r/r0r1r5r get_childrenrassert_almost_equalrgr7rrdrectsr9r9r:&test_hist_df_series_cumulative_densitys z9TestDataFramePlots.test_hist_df_series_cumulative_densitycs`ddlmttjdd}|jjddd}fdd |D}t |d  d dS) Nrr9rr!Tr")r;r<cr>r9r?r@r9r9r:rurAzETestDataFramePlots.test_hist_df_series_cumulative..rxrrBrFr9r9r:test_hist_df_series_cumulatives z1TestDataFramePlots.test_hist_df_series_cumulativecCs<ttjdd}|jjdddd}t|dddddS) Nrr rr horizontal)rr orientationrr)rr/r0r1r2r5rrrEr9r9r:test_hist_df_orientationsz+TestDataFramePlots.test_hist_df_orientationweights皙?)rj)shape)rjrcCsrtttddgtjdd}t|jd|d}t|jdd}dd |j D}d d |j D}t ||dS) Nrr'rrrjr)r(rMr+cSror9rrspatchr9r9r:rurvz8TestDataFramePlots.test_hist_weights..cSsg|]}d|qS)rNrrQr9r9r:ru) rr%r}r/r0r1r2rr5r`rrE)r7rMr8r/r0patch_height_with_weightsexpected_patch_heightr9r9r:test_hist_weightssz$TestDataFramePlots.test_hist_weightsc Cstdd|D}tdd|D}tdd|D}tdd|D} |dur4tj||dd|dur@tj||dd|durLtj||dd|durZtj| |dddSdS)NcSror9rprrr9r9r:rurvz7TestDataFramePlots._check_box_coord..cSror9rrrr9r9r:rurvcSror9rrrr9r9r:rurvcSror9)rerrr9r9r:rurvF) check_dtype)r/rrr ) r7r` expected_y expected_h expected_x expected_wresult_y result_heightresult_x result_widthr9r9r:_check_box_coordsz#TestDataFramePlots._check_box_coordr>rr! rrrrrrrrrrrrbr!rrrPrh)rPr!rbrrr)rrPrrrr)rrrrPrbr!cCst|}|jjdd}|j|jddtgdtgdd|j|jddtgdtgdd|j|jddtgdtgdd|jjdd d }|j|jddtgdtgdd|j|jddtgdtgdd|j|jddtgd tgdd|jjdd d d }|j|d jtgdtgdd|j|djtgdtgdd|j|djtgdtgdd|jjddd}|j|jddtgdtgdd|j|jddtgdtgdd|j|jddtgdtgdd|jjdd dd}|j|jddtgdtgdd|j|jddtgdtgdd|j|jddtgd tgdd|jjdd d dd}|j|d jtgdtgdd|j|djtgdtgdd|j|djtgdtgdddS)Nrh)r<)rrrrrra)rXrYr!rcrdT)r<r-)rr)r<r-r>rr@rrJ)r<rK)rZr[)r<r-rK)r<r-r>rK)rr5rr`r`r/r)r7r>r8rdrFr9r9r:test_hist_df_coords                                                 z%TestDataFramePlots.test_hist_df_coordcCs,ttjdd}t|jdddS)Nrr)Tr)rr/r0r1r2cumsumrr5r6r9r9r:test_plot_int_columns_sz(TestDataFramePlots.test_plot_int_columnsmarkers^+o)rmrnrocCsddlm}|}||dttjd d}|j |d}t | dt |D]\}}|||ks>Jq0dS)Nrorr^)style)matplotlib.pyplotrgcfclf add_subplotrr/r0r1r2r5 enumeraterr get_marker)r7rlpltr\r8rdr9rZr9r9r:test_style_by_columncs    z'TestDataFramePlots.test_style_by_columncCsPtddg}|}|dusJ|jdd}|ddks&JdS)Nr@rTrir)r r5 get_legend get_textsr)r7rrdr9r9r:test_line_label_nonews   z'TestDataFramePlots.test_line_label_nonezprops, expected))boxpropsboxes) whiskerpropswhiskers)cappropscaps) medianpropsmedianscCsPtdddD}|ddii}|jjd ddi|}||ddks&JdS) NcSs i|] }|tjddqSrP)r/r0r1)rskr9r9r: s zHTestDataFramePlots.test_specified_props_kwd_plot_box..ABCcolorC1r#r%rr9)rr5r  get_color)r7propsrr8kwdrr9r9r:!test_specified_props_kwd_plot_boxs z4TestDataFramePlots.test_specified_props_kwd_plot_boxc Cstdddtdddtdddg}gd}tt||dgd}|}|jd }t|tj|t d |jd  }t|t||j }d d |D}t |jsYJd d |D}tj|ddjsjJdS)Nr!r@rbr)rrrtestrrr$cSsg|]}|dqS)r) get_positionr@r9r9r:rurSz8TestDataFramePlots.test_unordered_ts..cSror9rr@r9r9r:rurvz%Y-%m-%dr)rrr/rr5rrrr r4r;rwget_ticklabelsris_monotonic_increasingrr) r7r-rr8rdriydataxlocsxlabelsr9r9r:test_unordered_tss"" z$TestDataFramePlots.test_unordered_tscCs8tdtdgdi}|j|dt|j|dSNr(rRrr+rWr-rr5rr7r(r8r9r9r:test_kind_both_wayss  z&TestDataFramePlots.test_kind_both_wayscCs@tdtdgdi}|jdd|dt|j|dddSrrrr9r9r:test_kind_both_ways_x_ys z*TestDataFramePlots.test_kind_both_ways_x_ycCsPttd}d}tjt|d|j|dWddS1s!wYdS)Nabcdno numeric data to plotrTr+)rr3rWrXrr5r7r(r8r\r9r9r:test_all_invalid_plot_datas "z-TestDataFramePlots.test_all_invalid_plot_datacCs~ttjddtd}d|tjd|jddk<d}tjt |d|j |d WddS1s8wYdS) Nrr!rr$rrrrrTr+) rr/r0r1r2r4rOrWrXrr5rr9r9r:(test_partially_invalid_plot_data_numerics "z;TestDataFramePlots.test_partially_invalid_plot_data_numericcCsZttjdd}d}tjt|d|jddWddS1s&wYdS)Nrrz*invalid_plot_kind is not a valid plot kindrTinvalid_plot_kindr+) rr/r0r1r2rWrXrr5r[r9r9r:test_invalid_kinds "z$TestDataFramePlots.test_invalid_kindzx,y,lblrrrcCs`tddgddgddgd}tjtdd |j|||d WddS1s)wYdS) Nr@rrPr"rhrrx must be a label or positionrTrRrSrrrWrXrr5)r7rRrSlblr8r9r9r:test_invalid_xy_argss "z'TestDataFramePlots.test_invalid_xy_argscCshtddgddgddgd}d}tjt|d |jd d d gd dWddS1s-wYdS)Nr@rrPr"rhrrz.label should be list-like and same length as yrTrr'r bad_labelrrr[r9r9r:test_bad_labels "z!TestDataFramePlots.test_bad_labelzx,yrcCs^tgdgdgtdd}tjtdd|j||dWddS1s(wYdS)N)r@rPrh)rr"rAABrrrTrQ)rr3rWrXrr5rr9r9r:test_invalid_xy_args_dup_colss"z0TestDataFramePlots.test_invalid_xy_args_dup_colszx,y,lbl,colorsrbluerbokehcythongreenyellowcCsjtddgddgddgd}t|jd||d |j||||d }t|jt|ks+Jt||d dS) Nr@rrPr"rhrrrr)rRrSrr linecolors)rrr5rrrr)r7rRrSrrr8rdr9r9r:test_y_listlikes  z"TestDataFramePlots.test_y_listlikez x,y,colnamescCs0tddgddgd}||_t|j||ddS)Nr@rrPr"rrQ)rr,rr5)r7rRrScolnamesr8r9r9r:test_xy_args_integersz'TestDataFramePlots.test_xy_args_integercCsrttjdjddtjdjddtdtjdjddd}|jjdddd}t|j d ks7JdS) Nrrr2rrr'r!)rRrSgridsizer@ rr/r0r1r4rr5rrrrcr9r9r:test_hexbin_basicsz$TestDataFramePlots.test_hexbin_basiccCsttjdjddtjdjddtdtjdjddd}|jjdddd}t|d j j dks:Jt |d d d dS) Nrrr2rrr'T)rRrSr>rr@rrA) rr/r0r1r4rr5rrfigurerFrrEr9r9r:test_hexbin_basic_subplotssz-TestDataFramePlots.test_hexbin_basic_subplotsreduce_CcCstttjdjddtjdjddtdtjdjddd}|jjddd|d}t|j d ks8JdS) Nrrr2rrr'r)rRrSrreduce_C_functionr@r)r7rr8rdr9r9r:test_hexbin_with_csz%TestDataFramePlots.test_hexbin_with_czkwargs, expectedBuGnrrrYlGncCs|ttjdjddtjdjddtdtjdjddd}|jjd ddd|}|jdj j |kss "z"TestDataFramePlots.test_pie_df_errrcCsFttjddgdgdd}t|jj|d}t|j|j dS)Nrrrrr+r) rr/r0r1rr5rrtextsr-)r7rSr8rdr9r9r: test_pie_dfHszTestDataFramePlots.test_pie_dfcCsttjddgdgdd}t|jjddd}t|t|jks&J|D] }t |j |j q(t ||jD] \}}| |ksDJq8dS)Nrrrrr+Tr<)rr/r0r1rr5rrr,rrr-r}r)r7r8rFrdylabelr9r9r:test_pie_df_subplotsRs z'TestDataFramePlots.test_pie_df_subplotscCsttjddgdgdd}gd}gd}t|jjdd||d }t|t|jks0J|D]}t |j |t |j |d q2dS) Nrrrrr+r)rbgrrmT)r=r>rrr) rr/r0r1rr5rrr,rrrr`)r7r8r color_argsrFrdr9r9r:test_pie_df_labels_colorscs& z,TestDataFramePlots.test_pie_df_labels_colorsc Csttjdd}tdD] }tj|j||f<qtjj dd\}}ddi}t d|j j dd|dd|Wdn1sDwYgd}t|D]9\}}t|}d ||<d d |jD} | |kskJd d |D} |d|||d d} | | ksJqQdS)Nr)r"r"r")ncols normalizeT)r>rdr)0123rzcSror9rr@r9r9r:rurvz6TestDataFramePlots.test_pie_df_nan..cSror9rr@r9r9r:rurvr@r9)rr/r0r1r~rilocrrr>rrr5rrvr3rr{r|) r7r8rrrFrf base_expectedrdrr result_labelsexpected_labelsr9r9r:test_pie_df_nanxs$   z"TestDataFramePlots.test_pie_df_nanr)rrrcCsptdtdddd}t|}tddtddd}t|}t|jfd|i|}t|ddd dS) NrrrHrQ皙?皙?yerrrxerrrr/rronesrr5r)r7rfrr8d_errdf_errrdr9r9r:test_errorbar_plots  z%TestDataFramePlots.test_errorbar_plotcCsrtdtdddd}t|}tddtddd}t|}t|dj||dd d }t|d d d dS) NrrrHrQrrr@rT)rrr(rrrr)r7rr8rrrdr9r9r:test_errorbar_plot_barsz)TestDataFramePlots.test_errorbar_plot_barcCsvtdtdddd}t|}t|djtddd}t|ddd t|jtd dd}t|dd d dS) NrrrHrQrSrrr@r)rrr)r/rrrr5rrr7rr8rdr9r9r:test_errorbar_plot_yerr_arrays z0TestDataFramePlots.test_errorbar_plot_yerr_arrayru誤差cCsttdtdddd}t|}tdd||<t|j|d}t|dddt|jd d |d }t|dd ddS) NrrrHrQrrrrrSrR)rSrRrr@r)r7rrr8rdr9r9r:test_errorbar_plot_column_namesz1TestDataFramePlots.test_errorbar_plot_column_namecCsltdtdddd}t|}tt|jtjd ddWddS1s/wYdS)NrrrHrQrrr) r/rrrexternal_error_raisedrr5r0r1r2r7rr8r9r9r:&test_errorbar_plot_external_valueerrors  "z9TestDataFramePlots.test_errorbar_plot_external_valueerrorcCsxtdtdddd}t|}tdgddgdd}tt|j|dWddS1s5wYdS)NrrrHrQzzzr)r/rrrrrr5)r7rr8rr9r9r:%test_errorbar_plot_external_typeerrors  "z8TestDataFramePlots.test_errorbar_plot_external_typeerror)rZrrfy_errrr)rrcCs@ttdtdddd}t|j||d}t|ddddS)NrrrHrQrr(rrrr/rrr5r)r7r(rr8rdr9r9r:!test_errorbar_plot_different_yerrs z4TestDataFramePlots.test_errorbar_plot_different_yerrz y_err, x_err)rrcCsBttdtdddd}t|j|||d}t|ddddS)NrrrHrQ)rrr(rrr)r7r(rx_errr8rdr9r9r:&test_errorbar_plot_different_yerr_xerrsz9TestDataFramePlots.test_errorbar_plot_different_yerr_xerrcCshttdtdddd}ttddtddd}t|jd||d|d}t|d d d dS) NrrrHrQrrT)r=rrr>r(r@r)rr/rrrr5r)r7r(r8rrFr9r9r:/test_errorbar_plot_different_yerr_xerr_subplotss"zBTestDataFramePlots.test_errorbar_plot_different_yerr_xerr_subplotszIterator is consumed)rOrXcCsNtdtdddd}t|}t|jtdt|d}t|ddddS) NrrrHrQrNrrr) r/rrrr5 itertoolsrepeatrrrr9r9r:test_errorbar_plot_iteratorsz.TestDataFramePlots.test_errorbar_plot_iteratorcCsvtttjdd}tttjdd}t|j|d}t|dddt|jddd}t|ddddS)Nrrrrrr@)rSr rr/absr0r1r2rr5rr7r8rrdr9r9r:'test_errorbar_with_integer_column_namess z:TestDataFramePlots.test_errorbar_with_integer_column_namescCsbtttjdd}tttjddddgd}t|j||d}t|ddddS)Nrr^rrrrrr)r7r(r8rrdr9r9r:'test_errorbar_with_partial_columns_kinds z:TestDataFramePlots.test_errorbar_with_partial_columns_kindcCstttjdd}tttjddddgd}tdddd }|j|d d |j|d d t|j |d d }t |ddddS)Nrr^rrr1/1/2000r!MEr(T)inplacerZrr) rr/rr0r1r2r rrr5r)r7r8rixrdr9r9r:&test_errorbar_with_partial_columns_dti#sz9TestDataFramePlots.test_errorbar_with_partial_columns_dtierr_boxcCs|Srr9rr9r9r:0szTestDataFramePlots.cCsdtdtdddd}t|}|tddtddd}t|j|d}t|dd d dS) NrrrHrQrr)rRrrr@rr)r7rrr8errrdr9r9r:&test_errorbar_with_partial_columns_box/s "z9TestDataFramePlots.test_errorbar_with_partial_columns_boxc Cs"tdtdddd}tddtddd}tddd d }t||d }t||d }t|j||d }t|dd dt|j||d }t|dd dt|jd|d|d}t|dddt|jdd|d}t|dddt|j||d }t|dd dt|jd||dd}t|ddddS)NrrrHrQrrrz1/1/2001r)r*r_rrrrSrR)rSrr(r@T)r=r(rr>)r/rrr rrr5r) r7r(rrrtdftdf_errrdrFr9r9r:test_errorbar_timeseries8s.  z+TestDataFramePlots.test_errorbar_timeseriescCstjdd}ttdddj}|j||dd}|jd dj dddf}|ddddft d dg}t ||td }tjt|d |j|jd WddS1scwYdS) Nr)rPrrhrrPrh)rrr@rrHzCAsymmetrical error bars should be provided with the shape (3, 2, 5)rTr)r/r0r1rrreshapeTr5r get_pathsverticesrrrErrrWrXr)r7r r8rdyerr_0_0 expected_0_0r\r9r9r:test_errorbar_asymmetricalZs   "z-TestDataFramePlots.test_errorbar_asymmetricalcCsttjddttjddd}t|jddt|j|dt d&|}t |j dks5Jt ||jt |j dksEJWddS1sPwYdS) Nrr^r!r_T)tablerr@)rr/r0r1r3rarbrr5rrrtablesr rrrcr9r9r: test_tablels "zTestDataFramePlots.test_tablecCstttjddtdddgd}tttjdddtdddgd}t|jj ddd}t |ddd t|jj dd|d }t |d dd t|jj dd|d }t |dd d t|jj dd||d }t |d d d dS)Nr)rhrrhrRrSrrQrr)rRrSrr@)rRrSr)rRrSrr) rr/rr0r1r2r~rr5rrrr9r9r:test_errorbar_scatter{s$z(TestDataFramePlots.test_errorbar_scattercsdfdd }tttjddgdd}|jjdd d d d d tddd|j d dd|j d dd|jjdd d ddtddd|j ddddS)Nhas_xerrcsvg}tfddjD}|D]}t|r||q||qfdd|D}t|t|gt|ddS)Nc3s"|] }t|dr|jVqdS)FN)rrr has_errr9r:rs z`TestDataFramePlots.test_errorbar_scatter_color.._check_errorbar_color..csg|] }|jvr|qSr9)rr@rdr9r:rurAzaTestDataFramePlots.test_errorbar_scatter_color.._check_errorbar_color..r) next containersrextendappendrr/rr)rrrrerrsel err_linesrrr:_check_errorbar_colors   zMTestDataFramePlots.test_errorbar_scatter_color.._check_errorbar_colorr)r!rhrrrrrrr)rRrSrrrr@rrhas_yerrr)rRrSrrr)r) rr/rr0r1r2r5rrr)r7r$r8r9rr:test_errorbar_scatter_colors z.TestDataFramePlots.test_errorbar_scatter_colorcCs\tgddd}tjttfdd|jdddd d WddS1s'wYdS) Nrr"r:z'unknown' is not arTrrunknownr)rRrSrr()rrWrXrKeyErrorr5r6r9r9r:test_scatter_unknown_colormaps"z0TestDataFramePlots.test_scatter_unknown_colormapcCsddlm}|dt\}}tgdgdgdgdd}dd}|D] }|jddd |d d q'|||||dt\}}t t |jd |d d }Wdn1sawY||dS) Nrrr@rrPr"rhrrrrrcSs|D]}t|jdks Jt|ddq|d|dfD]}t|ddt|jddddq|d|dfD]}t|ddt|jddddq;dS) Nr@TrsrrFrurP)rrrrrxrFrdr9r9r:_checksz5TestDataFramePlots.test_sharex_and_ax.._checkrrrnT)rRrSrnrdsharex)r>rdr.) rrrclose_generate_4_axes_via_gridspecrr5 tight_layoutrsrrrr7rxgsrFr8r-rdr9r9r:test_sharex_and_axs*         z%TestDataFramePlots.test_sharex_and_axcCddlm}tgdgdgdgdd}t\}}|D] }|jddd|dq|||D]%}t|jdks>Jt | d d t | d d t |j d d d d q3dS Nrr*r+rrrn)rRrSrnrdr@Trsru rrrrr0r5r1rsrrrrrxr7rxr8r3rFrdr9r9r:test_sharex_false_and_axs$  z+TestDataFramePlots.test_sharex_false_and_axcCsddlm}t\}}tgdgdgdgdd}dd}|D] }|jddd|d d q"|||||d t\}}t t |jd |d d }Wdn1s\wY||||dS) Nrr*r+cSs|D]}t|jdks Jt|ddt|jddddq|d|dfD] }t|ddq(|d|dfD] }t|ddq;dS) Nr@TrsrurrrPF)rrrrxrr,r9r9r:r-sz5TestDataFramePlots.test_sharey_and_ax.._checkrrrnT)rRrSrnrdrZr)r>rdrZ) rrrr0rr5r1rsr/rrrr2r9r9r:test_sharey_and_axs*        z%TestDataFramePlots.test_sharey_and_axcCr5r6r7r8r9r9r:test_sharey_and_ax_tights$  z+TestDataFramePlots.test_sharey_and_ax_tightcCstdi}|dvr6ttjdjddtjdjddtdtjdjddd}ddd }n9|d krVttjdd t t d t d t ddddd }nttjdd t t d t d t ddddd}t|jdd|i|}tdt|dusJdS)z3Check that every plot type gets properly collected.r()rrrrrr2rrr'rQrr r#r$r&r!r(r+r(rNr9)rWr-rr/r0r1r4rr2rr3r4r rweakrefrefr5rxr/gccollect)r7r(argsr8r=r9r9r:test_memory_leak3s6     z#TestDataFramePlots.test_memory_leakc sddlmddlmttjddt dddd}t tjdd|j t d d }fd d }fd d}||fD]H\}}|j |d}t|jdksUJ|j |d}t|jdksdJ||fD]}t|ddt|ddt|jddddqhdqB||fD]C\}}|j d||gd}t|jdksJt|jdksJ|D]}t|ddt|ddt|jddddqdq|\}}tt|j d||gddd}Wdn1swYt|djdksJt|djdksJ||fD] }t|ddqt|dddt|djddddt|dddt|djddddd|\}}tt|j d||gddd}Wdn 1stwYt|djdksJt|djdksJt|dddt|ddd||fD]}t|ddt|jddddqddS)Nrgridspecrr!rr)r_rABrcsLdd}}||ddddf}||dddf}||fS)NrPr@rGridSpecrrur3r\r/r0rCrxr9r:_get_vertical_gridh zSTestDataFramePlots.test_df_gridspec_patterns_vert_horiz.._get_vertical_gridcsLdd}}||ddddf}||dddf}||fS)Nr@rPrrFrHrIr9r:_get_horizontal_gridorKzUTestDataFramePlots.test_df_gridspec_patterns_vert_horiz.._get_horizontal_gridrr@Trsrurrr>rdr.rZFrrCrrrr r/r0r1r2r rr-r3r5rrrrrxr/rrr) r7tsr8rJrLr/r0rdrFr9rIr:$test_df_gridspec_patterns_vert_horizXsr               z7TestDataFramePlots.test_df_gridspec_patterns_vert_horizcsddlmddlmttjddt dddd}fdd }|}t tjdd |j t d d }|j d |d}|D]%}t|jdksLJt|d dt|d dt|jd dd dqAd|}tt|j d |d d d}Wdn1swY|D] }t|jdksJq|d|dfD] }t|d dq|d|dfD] }t|ddq|d|dfD]}t|ddt|jd dddq|d|dfD]}t|d dt|jd dd dqddS)NrrBrr!rrDr_cstdd}}||ddddf}||dddf}||dddf}||d}||||fS)NrPrrIrF)r3r\r/r0ax3ax4rIr9r:_get_boxed_grids  zKTestDataFramePlots.test_df_gridspec_patterns_boxed.._get_boxed_gridr r#rTrr@rsrurrMrPFrN)r7rOrSrFr8rdr9rIr:test_df_gridspec_patterns_boxedsJ      z2TestDataFramePlots.test_df_gridspec_patterns_boxedcCs.ttgdgddtjjdddddS)Nrrr:rrrQ)kws)rrr PlotAccessor_dataframe_kinds)r7r9r9r:test_df_grid_settingss  z(TestDataFramePlots.test_df_grid_settingscCs>tj\}}|gdttjddj|ddS)N)rrrrrr!r) rrr>add_axesr r/r0r1r5)r7r\rdr9r9r:test_plain_axess"z"TestDataFramePlots.test_plain_axescCsXttjddtjddd}tj}|d}|j d|ddddd dS) Nrrr:)rrr@r@rrrhsv)r(rdrRrSrr) rr/r0r1r2rrrrYr5)r7r8r\rdr9r9r:test_plain_axes_dfs  z%TestDataFramePlots.test_plain_axes_dfcCsrtj\}}ddlm}||}|jdddd}ttj ddj |d ttj ddj |d dS) Nr)make_axes_locatablerz5%g?)rpadrr!r) rrr>mpl_toolkits.axes_grid1r] append_axesr r/r0r1r5)r7r\rdr]dividercaxr9r9r:#test_plain_axes_make_axes_locatables  "z6TestDataFramePlots.test_plain_axes_make_axes_locatablecCsjtj\}}ddlm}||dddd}ttjddj |d ttjddj |d dS) Nr) inset_axesz30%rrP)rirr5rr!r) rrr>%mpl_toolkits.axes_grid1.inset_locatorrdr r/r0r1r5)r7r\rdrdiaxr9r9r:test_plain_axes_make_inset_axes s  "z2TestDataFramePlots.test_plain_axes_make_inset_axesmethod)rZrfrcCsvttjddtddjdddjddd }d }d d g}||d d}t|j|di|}t |j |ddS)Nr)rrrErcSs |jSr)r'rjr8r9r9r:r  zBTestDataFramePlots.test_secondary_axis_font_size..rcSs |jdS)Ng?rkrir9r9r:r rj)rrrrT)rLr mark_right)rFrr9) rr/r0r1r2r3assignrr5rrP)r7rhr8rr&rfrdr9r9r:test_secondary_axis_font_size s z0TestDataFramePlots.test_secondary_axis_font_sizecCstgdgdgdd}|jjdd}|ddd d |D}tt||}|d d ks5J|d dks=J|ddksEJdS)N)rPrrP)r*)MondayTuesday Wednesday)salesvisitsdayrvrrHrPcSror9rrstr9r9r:ru, rvzATestDataFramePlots.test_x_string_values_ticks..rqr rrrrsr)rr5rset_xlimrxr%r} get_xticks)r7r8rd xticklabelslabels_positionr9r9r:test_x_string_values_ticks s z-TestDataFramePlots.test_x_string_values_tickscCstddgddgg}ttjddddg|d}|}|d d d d | D}t t || }|d dks?J|ddksGJ|ddksOJ|ddksWJdS)Nrir@rrYrr'r+rHr"cSror9rrwr9r9r:ru> rvzETestDataFramePlots.test_x_multiindex_values_ticks..z (2012, 1)r z (2012, 2)rz (2013, 1)rz (2013, 2)r) r from_productrr/r0r1r2r5ryrxr%r}rz)r7r-r8rdr{r|r9r9r:test_x_multiindex_values_ticks3 s z1TestDataFramePlots.test_x_multiindex_values_tickscCsLtddgddgd}|j|d}|}|ddksJ|ddks$JdS)Nrr"r@r_r+r)rr5r)r7r(r8rdxlimsr9r9r:test_xlim_plot_lineF s  z&TestDataFramePlots.test_xlim_plot_linecCstj\}}gd}tgdgdgdgdd|d}|gdjj|d d |d gj|d d |}|ddks@J|ddksHJdd|D}||ksWJdS)N)k1k2k3k4)irr)iixri )rri@i)r@rPr"rP)s1s2s3rLr_)rrrF)rdr-rLT)rdrLrr@rPcSror9rrwr9r9r:rug rvzWTestDataFramePlots.test_xlim_plot_line_correctly_in_mixed_plot_type..)rrr>rr5rrrx)r7r\rdindexesr8rr{r9r9r:0test_xlim_plot_line_correctly_in_mixed_plot_typeP s" zCTestDataFramePlots.test_xlim_plot_line_correctly_in_mixed_plot_typecCsltdgtd}|js J|}t|dksJ|d}t|dks*Jt|dks4JdS)Nfoo)r,r%r@r)rintemptyr5rrrr;)r7r8rdrZr9r9r:test_plot_no_rowsj s  z$TestDataFramePlots.test_plot_no_rowscCsHtgd}tjtdd |WddS1swYdS)NrrrT)rrWrXrr5r6r9r9r:test_plot_no_numeric_datat s  "z,TestDataFramePlots.test_plot_no_numeric_data)rZrrfrr*r=rrcCstdtdtddtddtdtdd}t|}|jddg|d}t|dks5Jd d gd d gd gf}t||D]\}}|dkrRt||d|dkrat|j t|ksaJqDdS)Nr(r!r@r)rr)rrr>r(rPrrrrrrr,rZ) rWr-r/rrr5rr}rr)r7r(rr8rFrrdrr9r9r:test_group_subploty s$    z%TestDataFramePlots.test_group_subplotcCsRttd}d}tjt|d|jdgdWddS1s"wYdS)Nr@z!An iterable subplots for a SeriesrTrr>)r r~rWrXNotImplementedErrorr5)r7serr\r9r9r:(test_group_subplot_series_notimplemented s "z;TestDataFramePlots.test_group_subplot_series_notimplementedcCsbttdtddgd}d}tjt|d|jdgdWddS1s*wYdS)Nrrrrz6An iterable subplots for a DataFrame with a MultiIndexrTr) rr/eyerrrWrXrr5r[r9r9r:,test_group_subplot_multiindex_notimplemented s "z?TestDataFramePlots.test_group_subplot_multiindex_notimplementedcCs\ttdddgd}d}tjt|d|jdgdWddS1s'wYdS)Nrrrz4An iterable subplots for a DataFrame with non-uniquerTrr)rr/rrWrXrr5r[r9r9r:0test_group_subplot_nonunique_cols_notimplemented s "zCTestDataFramePlots.test_group_subplot_nonunique_cols_notimplementedzsubplots, expected_msg)){z(subplots should be a bool or an iterable)r!each entry should be a list/tuple))r@r)rrcCs^tdtdd}t|}tjt|d|j|dWddS1s(wYdS)Nr!r:rTrr/rrrWrXrr5)r7r> expected_msgrr8r9r9r:test_group_subplot_bad_input s  "z/TestDataFramePlots.test_group_subplot_bad_inputcCs`tdtdd}t|}tjtdd|jdgdWddS1s)wYdS)Nr!r:z Column label\(s\) \['bad_name'\]rT)rbad_namerrrr9r9r:&test_group_subplot_invalid_column_name s "z9TestDataFramePlots.test_group_subplot_invalid_column_namecCsjtdtdtdd}t|}tjtdd|jddgdWddS1s.wYdS)Nr!rzshould be in only one subplotrTr:)rrrrrr9r9r:$test_group_subplot_duplicated_column s "z7TestDataFramePlots.test_group_subplot_duplicated_column)r rrcCsbtdtdd}t|}tjtdd|jdg|dWddS1s*wYdS)Nr!r:z1When subplots is an iterable, kind must be one ofrTrr)r7r(rr8r9r9r:test_group_subplot_invalid_kind s"z2TestDataFramePlots.test_group_subplot_invalid_kindz index_name, old_label, new_label)Nrznew)oldrr)Nrzrz)Nrzr@rz)rZrrcCstddgddggddgd}||j_|j|d}||ks J|dks(J|j|||d }|t|ks:J|t|ksDJdS) Nr@rrhType AType Brr+rz)r(rxlabel)rr-rr5 get_xlabelrstr)r7r( index_name old_label new_labelr8rdr9r9r:(test_xlabel_ylabel_dataframe_single_plot s z;TestDataFramePlots.test_xlabel_ylabel_dataframe_single_plotzxlabel, ylabel))NN)X LabelN)NY Label)rrcCstd}d}tddgddgg||gd}|j|||||d}||dur%|n|ks*J||dur3|n|ks8JdS)Nrrr@rrhr)r(rRrSrr)rr5rr)r7r(rrxcolycolr8rdr9r9r:'test_xlabel_ylabel_dataframe_plane_plot s   z:TestDataFramePlots.test_xlabel_ylabel_dataframe_plane_plot)FTcCsjtdgj|dddgd}|jjD]!}|jr2|dks J|dks(J|ddks2JqdS)Nrr)rrjr)rLrrkrg) rr5rrFr get_visiblerrT get_yticks)r7rLax_dfrdr9r9r:test_secondary_y s    z#TestDataFramePlots.test_secondary_ycCstttjddttdtdtddddd }t d | }|j }WddS1s3wYdS) Nrr r#r$r&r!r'r(r+F) rr/r0r1r2rr3r4r rrr5r)r7r8rr9r9r:test_plot_no_warning s   "z'TestDataFramePlots.test_plot_no_warning)NNNN)__name__ __module__ __qualname__rWmarkslowr;rGrJrNxfailr]re parametrizerqrzr|rrrrr3rrrrrrrrrrrrrrrrrr/rr<r rrrrrr'r1r3r6r8r~r rErJrKrNrRrXr]rcrdrlrmr}rrrrrrCategoricalIndexrrrrrrparamtd skip_if_norrrrrrrrrrrrrrrrrfilterwarningsrr r$r'r.r/r0r1r6r7r8rHrIrLrrVr`rrrhrkryr}rrr rV _common_kindsrrrrrrrrrrrrstdrrrrrrrrrrrrrr rrrrrrrrr r rrrr&r)r4r9r:r; _all_kindsrArPrTrXrZr\rcrgrnr}rrrrrrrrrrrrrrrrrr9r9r9r:r6sT                                                                                            ."""*** l      ",             (((6      !      (  ( $ S 5               rcCsbddlm}tjdd}||d}||d}||d}||d}|||||gfS)Nrrr3)r@rrr)rrrrrCrGsubplot)rxr3ax_tlax_llax_trax_lrr9r9r:r0 s r0)4__doc__rrr>rrrar<numpyr/rWpandas.util._test_decoratorsutil_test_decoratorsrpandas.core.dtypes.apirpandasrrrrrr r r r r pandas._testing_testingrpandas.tests.plotting.commonrrrrrrrrrrrrrpandas.io.formats.printingrr-rrxrr0r9r9r9r:sN , <    t