a Pf<@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'jsddgddZtdddZuej 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)reasoncCsZtddgddgd}d}tjt|d|jjdd Wdn1sL0YdS) Nr@rr"xyz''Line2D' object has no property 'blarg'matchT)Zblarg)rpytestraisesAttributeErrorr5liner7r8msgr9r9r:test_plot_invalid_argjsz(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)ylimZxlimTblah)r=r>titlecCs>ttjddttjddd}t|jfi|dS)Nrr]r!r^ rr/r0r1r3r`rarr5r7rer8r9r9r:test_plot_other_args|s z'TestDataFramePlots.test_plot_other_argscCsttjddttjddd}|jddd}t|dd d |ddD]F}t |j t | d d t |j dd d d t |j gd d qP|dfD]6}t |j t | t |j gt |ddqdS)Nrr]r!r^Trk)r>rlrPrPr@rAFvisibleminorrrL)rr/r0r1r3r`rar5rrxaxisget_xticklabels get_labelr)r7r8rFrcr9r9r:test_plot_visible_axs     z'TestDataFramePlots.test_plot_visible_axcCs:ttjddttjddd}t|jdddS)Nrr]r!r^rkrlrmr6r9r9r:test_plot_titles z"TestDataFramePlots.test_plot_titlecCsVttjddtd}ttjddt |d}t |j dd}t |dddS) Nr!rr]r^Tr_rrL) zipr`rarangerr/r0r1r from_tuplesrr5r)r7Ztuplesr8rcr9r9r:test_plot_multiindexsz'TestDataFramePlots.test_plot_multiindexcCs^tjgdddgd}tjddgddgd}ttjd d d d ||d }t|jdddS)N))αr)rr@)βr)rrP)γr")rrg)δ)rZi0i1)names)baruΔ)ruΕZc0c1rrr!)rr+uΣry) rr}rr/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 |dksHJ|j }||dus^JdS)NrRrr!Tr>rcr@r) rr/r0r1mplpyplotr>r5rlenrF)r7r8_rcrFresultr9r9r:test_plot_passed_axs z&TestDataFramePlots.test_plot_passed_axz cols, x, yZABCDEAr'CDEc Cstgd}tgdgdtjddtjddgtdtj|dd d tj|dd d d d }t||j ||ddS)N)Z2008Z2009NZ2011Z2012r@rrPr"rgrrgrPrr$z%Yformati8T)rutcrr'rrrrQ) rr/arraynanr4pd to_datetimeviewrr5)r7colsrRrSdatesr8r9r9r:test_nullable_int_plots z)TestDataFramePlots.test_nullable_int_plotr5)rYrhistpiecCs.tjgddd}t|}tt|j|dS)Nr@rrPr"UInt32r$)rrr rgetattrr5)r7r5arrsr9r9r:test_integer_array_plot_seriessz1TestDataFramePlots.test_integer_array_plot_seriesz plot, kwargsrYrrrrSscatterrRrQhexbincCs<tjgddd}t||d}tt|j|fi|dS)Nrrr$rQ)rrrrrr5)r7r5rerr8r9r9r:test_integer_array_plot_df sz-TestDataFramePlots.test_integer_array_plot_dfcCs6tgdgdd}|}t|dks2JdS)NrRrSzr@rrPrr'r@)rr5r get_linesrbr9r9r:test_nonnumeric_excludesz*TestDataFramePlots.test_nonnumeric_excludecCs@ttjddgdd}|jddd}t|jddS)Nrr]abcr,rrrQ) rr/r0r1r2r5rrurwrbr9r9r:test_implicit_label s z&TestDataFramePlots.test_implicit_labelcCsHttjddddgd}d|j_|jddd|jjdksDJdS) NrrIrrrNAMEZLABEL)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)Nrrgr"r#r$r&rgr'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&rgr'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&rgr'r(r+r@ZTest)r)rRrSrlfigsizer)g0@g @)rBrCr) rr/r0r1r2rr3r4r r5rrlrrbr9r9r:test_plot_xy_figsize_and_titleJs  z1TestDataFramePlots.test_plot_xy_figsize_and_titlezinput_log, expected_log)Tlog)symZsymlogcCstdtditdd}|j|d}t||d||ksDJ|j|d}t||d||kslJ|j|d}t|||d ||ksJ||ksJdS) Nrrir^logyyaxis)logxruloglog)rur)rr/rr5rZ get_yscaleZ get_xscale)r7Z input_logZ expected_logr8rcr9r9r:test_logscalesXs     z!TestDataFramePlots.test_logscales input_param)rrrcCstdtditdd}d|d}tjt|d$|jfi|diWdn1s`0Yd|d }tjt |d*|jj fd d i|d iWdn1s0YdS) Nrrir^z keyword 'z*' should be bool, None, or 'sym', not 'sm'rTsmzPiePlot ignores the 'z ' keywordr>T) rr/rrVrW ValueErrorr5tmassert_produces_warning UserWarningr)r7rr8r[r9r9r:test_invalid_logscaleks 2 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_xdatarrr7r8rclinesr9r9r: test_xcompatxs  zTestDataFramePlots.test_xcompatcCsrttjddttdtdtddddd }d t j d <| }| }t |d trbJt|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 trbJd }tjt|d*t t|d tsJWdn1s0YdS)Nrr r#r$r&r!r'r(r+FrrzPeriodDtype\[B\] is deprecatedrT)rr/r0r1r2rr3r4r r rr5rrrrrr FutureWarningr7r8rcrr[r9r9r:!test_xcompat_plot_params_x_compats  z4TestDataFramePlots.test_xcompat_plot_params_x_compatcCsttjddttdtdtddddd }t j d d B| }| }t|d trhJt|d dWdn1s0YdS)Nrr r#r$r&r!r'r(r+rTrrrL)rr/r0r1r2rr3r4r r rZuser5rrrrrrr9r9r:(test_xcompat_plot_params_context_managers z;TestDataFramePlots.test_xcompat_plot_params_context_managercCsttjddttdtdtddddd }| }| }t |d  t rXJd }tjt|d *t t |d  t sJWdn1s0Yt|d d dS)Nrr r#r$r&r!r'r(r+rzPeriodDtype\[B\] is deprecated rTrL)rr/r0r1r2rr3r4r r5rrrrrrrrrr9r9r:test_xcompat_plot_periods 8z+TestDataFramePlots.test_xcompat_plot_periodcCsPttjddttdddtdddddgd}|tjj d d dS) Nr)rr@rrr-r,rr) rr/r0r1r rr5rrZaxhliner6r9r9r:test_period_compatsz%TestDataFramePlots.test_period_compat index_dtypecCstdtdittddd|dtjd}|}|d}|}t|dddf|dddftjdd }t j ||j d d dS) NrSricrHr$)r-r%rr@)r%rF)Zcheck_index_type) rr/rrint64r5rZ get_xydatar rZassert_series_equalrS)r7rr8rcrrsr9r9r:test_unsorted_indexs  *z&TestDataFramePlots.test_unsorted_indexr8?@@)rrrrr^rrrg@g@g@rcCs\|}|\}}|}|t|ddks:J|t|ddksXJdS)Nr)r5get_xlimrr/nanminget_datananmaxr7r8rcxminxmaxrr9r9r:test_unsorted_index_limss  z+TestDataFramePlots.test_unsorted_index_limscCsxtgdgdd}|jddd}|\}}|}|t|ddksVJ|t|ddkstJdS)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 Wdn1sf0Ytjt |d|j j dd Wdn1s0YdS) Nrrr"rrRrSrfourrz+Log-y scales are not supported in area plotrTTrr) rr/r0r1r3r`rarVrWrr5arearZr9r9r:test_negative_logs,z$TestDataFramePlots.test_negative_logcCsZtt|dd}t||D]0\}}||d7}|d}t||q$dS)Nrr@)r/Zzerosrrr{rassert_numpy_array_equal)r7Z normal_linesZ stacked_linesbasenlslsyr9r9r:_compare_stacked_y_coods  z*TestDataFramePlots._compare_stacked_y_coodkindrmultr@rHcCsV|ttjddgdd}t|j|dd}t|j|dd}||j|jdS)NrrwrRrSrrFrstackedTrr/r0r1rr5rr)r7rrr8ax1ax2r9r9r:test_line_area_stackeds z)TestDataFramePlots.test_line_area_stackedcCsttjddtjddtjdd tjdd d}t|j|dd}t|j|dd}||jdd|jdd||jdd|jdddS)NrrrFrTr)r7rZsep_dfrrr9r9r:test_line_area_stacked_sep_df s z0TestDataFramePlots.test_line_area_stacked_sep_dfcCs~ttjddttjddgdd}t|j ddd}t j t |d |j d dWdn1sp0YdS) NrrrrrFrzWhen stacked is True, each column must be either all positive or all negative. Column 'w' contains both positive and negative valuesrTT) rr/r0r1r2r3r`rarr5rVrWr)r7Zmixed_dfr[r9r9r:test_line_area_stacked_mixedsz/TestDataFramePlots.test_line_area_stacked_mixedcCsDttjddgdd}||jd}t|j|ddddS)Nrrrrr@T)rrr)rr/r0r1rr-rr5)r7rr8Zdf2r9r9r:#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) r7rvalues1values2r8rcZmasked1Zmasked2expr9r9r: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@rrPrr^r@rrrPr$rPrrr@Trr) r/rrrr rr5rr rr)r7rr$r%r8 expected1 expected2rcr9r9r:test_line_area_nan_df_stackedHsz0TestDataFramePlots.test_line_area_nan_df_stackedrFc Csddtjdg}dtjddg}t||d|d}tjgdtjd}tjgdtjd}t|jjfi|}t |j d  ||rt |j d |nt |j d ||t|jjd d }t |j d  |t |j d |dS) Nr@rrPrr^r(r$r)rFr) r/rrrr rr5rrr rr) r7rrer$r%r8r*r+rcr9r9r:"test_line_area_nan_df_stacked_areaWsz5TestDataFramePlots.test_line_area_nan_df_stacked_area secondary_ycCs~ttjddgdd}|jfi|}|\}}|}||dddks^J||dddkszJdS)NrrrPrrrrH)rr/r0r1r5rrr)r7rer8rcrrrr9r9r: test_line_limns  z TestDataFramePlots.test_line_limcCsttjddgdd}|jddd}t|ddd |D]l}t|d sPJt|d r^J|\}}|}||d  d d ksJ||d  d d ks>Jq>dS)Nrr/rrT)r.r>rPrprAZleft_axright_axrrH) rr/r0r1r5rhasattrrrr)r7r8rFrcrrrr9r9r: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 ddkspJ|| d ddksJ|dksJt|jj|d}|\}}|dksJdS)NrrrrrrrH) rr/r0r1rr5rrget_ylimrr) r7rr8Zneg_dfrcrrZyminZymaxrr9r9r: test_area_lims    z TestDataFramePlots.test_area_limcCsttjddddgd}tjjdddd\}\}}|j|d d |j|d d t| ||shJt| ||s|JdS) Nrr"rrRrSrr@T)shareyr)rcr) rr/r0r1rrr>r5rZjoined)r7r8figrrr9r9r:test_area_sharey_dont_overwrites z2TestDataFramePlots.test_area_sharey_dont_overwritecCsFttjdd}|jj|dd}|jD]}|dks,Jq,dS)Nrrgrg)r linewidth) rr/r0r1r2r5rpatches get_linewidth)r7rr8rcrr9r9r:test_bar_linewidths z%TestDataFramePlots.test_bar_linewidthcCs^ttjdd}|jjddd}t|ddd|D] }|jD]}| dksBJqBq8dS)Nrr;T)r<r>rg)rgr@rA) rr/r0r1r2r5rrr=r>)r7r8rFrcr?r9r9r: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 kspJq4t|||ks4Jq4dS)Nrr;?)rwidth) rr/r0r1r2rr5r=rr,)r7methdimrr8rFrcr?r9r9r:test_bar_barwidths  z$TestDataFramePlots.test_bar_barwidthcCs\ttjdd}d}t|j||dd}|D]$}|jD]}t|||ks)rr/r0r1r2rr5r=)r7rGrHr8rFrFrcr?r9r9r:test_barh_barwidth_subplotss  z.TestDataFramePlots.test_barh_barwidth_subplotscCsttjdd}|jjddd}dd|jD}|dgdksHJ|jjd gd d}d d|jdd D}|gd ksJdS) Nrr;Fr@)rbottomcSsg|] }|qSr9Zget_y.0pr9r9r: zBTestDataFramePlots.test_bar_bottom_left_bottom..T)rHcSsg|] }|qSr9rLrMr9r9r:rPrQrg)rr/r0r1r5rr=r7r8rcrr9r9r:test_bar_bottom_left_bottomsz.TestDataFramePlots.test_bar_bottom_left_bottomcCsttjdd}|jjdtgdd}dd|jD}|dgd ksRJ|jjd gd d}d d|jdd D}|gd ksJdS)Nrr;Fr@r@r@r@r@)rleftcSsg|] }|qSr9get_xrMr9r9r:rPrQz@TestDataFramePlots.test_bar_bottom_left_left..r@rRTrcSsg|] }|qSr9r[rMr9r9r:rPrQrg)rr/r0r1r5rCrr=rWr9r9r:test_bar_bottom_left_leftsz,TestDataFramePlots.test_bar_bottom_left_leftcCsttjdd}|jjddd}|D]&}dd|jD}|dgdks*Jq*|jjdtgd d }|D]&}d d|jD}|d gdkspJqpdS) Nrr;TrH)r>rKcSsg|] }|qSr9rLrMr9r9r:rPrQzDTestDataFramePlots.test_bar_bottom_left_subplots..rgrY)r>rZcSsg|] }|qSr9r[rMr9r9r:rPrQr@) rr/r0r1r5rr=rCr)r7r8rFrcrr9r9r:test_bar_bottom_left_subplotssz0TestDataFramePlots.test_bar_bottom_left_subplotscCsRtdtjdggdgdd}|j}gd}dd|jD}||ksNJdS) Nr!rgr!r_rrr'r r!rr_rgr!r_r@rrPcSsg|] }|qSr9rDrMr9r9r:rPrQz3TestDataFramePlots.test_bar_nan..rr/rr5rr=r7r8rcexpectedrr9r9r: test_bar_nans   zTestDataFramePlots.test_bar_nancCsztdtjdggdgdd}|jjdd}gd}d d |jD}||ksRJd d |jD}gd }||ksvJdS) Nr!r_r`rraTrrbcSsg|] }|qSr9rcrMr9r9r:rPrQz;TestDataFramePlots.test_bar_nan_stacked..cSsg|] }|qSr9rLrMr9r9r:rPrQ) rrr$@rg4@g.@rhgD@rdrer9r9r:test_bar_nan_stackeds  z'TestDataFramePlots.test_bar_nan_stackedcCsttjdd|td|tdd}|j}|j }t |t gd| dksfJ|jdd ks|J|jd d ksJ|jjd d }t |t gd| dksJ|jdd ksJ|jd dksJdS)Nr)rrgZABCDEFZabcder)rr@rrPr"rg)gg@rgпrHg@Trg@)rr/r0r1r2r3r5rru get_ticklocsrr rrr=r\)r7rr8rcZticksr9r9r:test_bar_categoricals     z'TestDataFramePlots.test_bar_categoricalzx, yr@rcCsDttjddttjddgdd}t|j j ||ddS)NrrrrrrQ) rr/r0r1r2r3r`rarr5rr7rRrSr8r9r9r:test_plot_scatters z$TestDataFramePlots.test_plot_scattercCsttjddttjddgdd}t d}t j t |d|j jdd Wdn1sj0Yt d }t j t |d|j jd d Wdn1s0YdS) Nrrrrrz5scatter() missing 1 required positional argument: 'y'rTrRrz5scatter() missing 1 required positional argument: 'x'rSr)rr/r0r1r2r3r`rareescaperVrW TypeErrorr5rrZr9r9r:test_plot_scatter_error(s , z*TestDataFramePlots.test_plot_scatter_errorcCsRttjddttjddgdd}|jdddd d }t |d d d dS)NrrrrrrRrSrT)rRrSrr>r@rrA) rr/r0r1r2r3r`rar5rrEr9r9r:test_plot_scatter_shape5sz*TestDataFramePlots.test_plot_scatter_shapecCsvttjdddgd}tddddj|d<d }tjt |d  |j d ddd Wdn1sh0YdS) Nrr!rrz 2014-01-01h)startr*r)Zdtimez9must be a string or a (real )?number, not 'datetime.time'rTr)rrRrS) rr/r0r1r2r timerVrWrqr5rZr9r9r:&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)rur)r*rrrxrQ) r rr/r0r1normalrrrr5r)r7rRrSrryr8r9r9r:test_scatterplot_datetime_dataHsz1TestDataFramePlots.test_scatterplot_datetime_data infer_stringZpyarrow)Zmarksrb_colrrPr"rcCsRtd|4tgd|d}t|jj||dWdn1sD0YdS)Nzfuture.infer_stringrarrQ)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| | DsJdS)Ngffffff@g @g@rg@g @g@g@lengthrFr)rrrrr)rrZspeciesrr@rRrSr)?g?g@rcSsg|] }|qSr9get_text)rNir9r9r:rPyrQzLTestDataFramePlots.test_scatterplot_color_by_categorical..css|]\}}||kVqdSNr9)rNrjr9r9r: zrQzKTestDataFramePlots.test_scatterplot_color_by_categorical..)rr Categoricalr5r collectionscolorbarr/rZ get_ticksrr Z _boundariesrcZget_ymajorticklabelsallr{) r7rrr8rcZcolorbar_collectionrZexpected_ticksZ result_ticksZexpected_boundariesZresult_boundariesZexpected_yticklabelsZresult_yticklabelsr9r9r:%test_scatterplot_color_by_categorical]s*    z8TestDataFramePlots.test_scatterplot_color_by_categorical)rQ)rSrR)rSrScCs2tgdtgdd}t|jj||ddS)Nr)rrrrrQ)rrrrr5rrmr9r9r:'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 ksvJdS) NrrrirlowhighsizerrrrZGreysr)rr/r0r1rr3r`rar5rrcmaprrrc get_ylabel)r7rRrSrr8rcr9r9r:test_plot_scatter_with_csz+TestDataFramePlots.test_plot_scatter_with_ccCsttjdjddddttjddgdd}d }|jj d d d |d }|j dj j |ksbJ|jj d d d dd}|j dj dusJ|jj dddd}|j dj dusJt|j dgddS)Nrrrirrrrr cubehelixrRrSr)rRrSrcolormapF)rRrSrrr@redrr?Z facecolors)rr/r0r1rr3r`rar5rrrrrr)r7r8cmrcr9r9r: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)rrrr)rrrrrr'rrrrZspring)rRrSrr) rr/rr5rrr rZ get_facecolor)r7r8Zred_rgbaZ green_rgbaZ rgba_arrayrcZ float_arrayr9r9r:test_plot_scatter_with_c_arraysz1TestDataFramePlots.test_plot_scatter_with_c_arraycCsVttjdddgdd}|jjdddd }tj|dj|j d  d dS) Nrr]rirrrrr)rRrSrr)right) rr/r0r1r5rrr valuesrZ get_sizesrbr9r9r:test_plot_scatter_with_ss z+TestDataFramePlots.test_plot_scatter_with_scCsXttjdddgdd}tj}|jjddd|d }|j d j |usTJdS) Nrr]rirrrrr)rRrSrnormr) rr/r0r1rcolorsZLogNormr5rrr)r7r8rrcr9r9r:test_plot_scatter_with_norms  z.TestDataFramePlots.test_plot_scatter_with_normcCsttjdddgdd}|jjdddd }|jd j}|j |j f}t j j |}|jD]}||||ksfJqfdS) Nrr]rirrrrrrr)rr/r0r1r5rrrrminmaxrrZ Normalize)r7r8rcZ plot_normZ color_min_maxZ default_normvaluer9r9r:test_plot_scatter_without_norms   z1TestDataFramePlots.test_plot_scatter_without_normlegendr<cCsFttjddttjddgdd}t|j j fi|dS)Nrrronetwothreerr) rr/r0r1r2r3r`rarr5rrnr9r9r: test_plot_bars z TestDataFramePlots.test_plot_barcCs>ttjddttjddtdd}t |j j dS)Nr)r!r!rr) rr/r0r1r2r3r`rar|rr5rr6r9r9r:test_plot_bar_int_cols z(TestDataFramePlots.test_plot_bar_int_colcCsRtddgddgd}t|jj}t|dd|jjddd}t|dddd dS) Nrr@rZrL#r!rotfontsizerM xlabelsize ylabelsize)rrr5rrrbr9r9r:test_plot_bar_tickss   z&TestDataFramePlots.test_plot_bar_tickscCsRtddgddgd}t|jj}t|dd|jjddd}t|dddddS) Nrr@r)yrot7 r)rrr)rrr5rCrrbr9r9r:test_plot_barh_tickss   z'TestDataFramePlots.test_plot_barh_tickscCsv|}|j}dd|D}t|jj}t||t|j t dt |dt |jdt |ksrJdS)NcSsg|] }t|qSr9rrNrr9r9r:rP rQz3TestDataFramePlots.test_boxplot..r@r)_get_numeric_datar,rr5boxrrvrr rurjr/rrrr7hist_dfr8 numeric_colslabelsrcr9r9r: test_boxplots  zTestDataFramePlots.test_boxplotcCs8|}|d}|jjdd}t|dddt|jjdS)Nheight(rrrMr)r5rrr)r7rr8seriesrFr9r9r:test_boxplot_seriess z&TestDataFramePlots.test_boxplot_seriescCsv|}tgd}|jj|d}|j}dd|D}t||t |j |t |j dt |ksrJdS)N)r@rr) positionscSsg|] }t|qSr9rrr9r9r:rP rQzDTestDataFramePlots.test_boxplot_series_positions..r)r/rr5rrr,rrvrr rurjrr)r7rr8rrcrrr9r9r:test_boxplot_series_positionss z0TestDataFramePlots.test_boxplot_series_positionscCsj|}|j}dd|D}|jjdddd}t|ddddt||t|jd t|ksfJdS) NcSsg|] }t|qSr9rrr9r9r:rP(rQz.2rF)rrvertrrMrrr) rr,r5rrrget_yticklabelsrrrr9r9r: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 ksVJqVdS) NcSsg|] }t|qSr9rrr9r9r:rP4rQzETestDataFramePlots.test_boxplot_vertical_subplots..TF)r=r>rrrP)r@rPrArrr) rr,rr5rrrr{rrrr)r7rr8rrrFrcrr9r9r:test_boxplot_vertical_subplots0s  z1TestDataFramePlots.test_boxplot_vertical_subplotscCsx|}|j}dd|D}tgd}|jj|dd}t||t |j |t |j dt |kstJdS)NcSsg|] }t|qSr9rrr9r9r:rPErQzFTestDataFramePlots.test_boxplot_vertical_positions..)rPrrF)rrr)rr,r/rr5rrrrr rrjrr)r7rr8rrrrcr9r9r:test_boxplot_vertical_positionsBs z2TestDataFramePlots.test_boxplot_vertical_positionscCsrttjddttjddgdd}d}tj t |d|j j dd Wdn1sd0YdS) Nrrrrrz2return_type must be {None, 'axes', 'dict', 'both'}rTZ not_a_type return_type) rr/r0r1r2r3r`rarVrWrr5rrZr9r9r: test_boxplot_return_type_invalidLsz3TestDataFramePlots.test_boxplot_return_type_invalidr)dictrFZbothcCsJttjddttjddgdd}|jj |d}t ||dS)Nrrrrrr) rr/r0r1r2r3r`rar5rr)r7rr8rr9r9r:%test_boxplot_return_type_invalid_typeVsz8TestDataFramePlots.test_boxplot_return_type_invalid_typecCsZtdttjdd}t|jdd}dd|j D}t ||dt |d d dS) Nscipyrrir"kdercSsg|] }t|qSr9rrr9r9r:rPdrQz2TestDataFramePlots.test_kde_df..rrrL) rV importorskiprr/r0r1r2rr5r,rr)r7r8rcrfr9r9r: test_kde_df`s   zTestDataFramePlots.test_kde_dfcCsDtdttjdd}|jdddd}t|dddddS) Nrrr rr_rg)rrrr) rVrrr/r0r1r2r5rrbr9r9r:test_kde_df_roths z"TestDataFramePlots.test_kde_df_rotcCsDtdttjdd}t|jdddd}t |ddd dS) Nrrr Tr)r=rr>r"r?rA) rVrrr/r0r1r2rr5rrEr9r9r:test_kde_df_subplotsns z'TestDataFramePlots.test_kde_df_subplotscCs@tdttjdd}|jdddd}t|dddS) Nrrr rT)rrr>rr) rVrrr/r0r1r2r5rrEr9r9r:test_kde_df_logyys z#TestDataFramePlots.test_kde_df_logycCs@tdttjdjdd}tj|jd<t |j dddS)Nrrrrrrrr) rVrrr/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) NrrcSsg|] }t|qSr9rrr9r9r:rPrQz3TestDataFramePlots.test_hist_df..rT)r=r>rr"r?rArr) rr/r0r1r2rr5rr,rrr)r7r8rcrfrFr9r9r: 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") cumulativebinsdensitycsg|]}t|r|qSr9rrNrRrr9r:rPrQzMTestDataFramePlots.test_hist_df_series_cumulative_density..rHr Zmatplotlib.patchesrr r/r0r1r5rZ get_childrenrassert_almost_equalrDr7rrcZrectsr9rr:&test_hist_df_series_cumulative_densitys  z9TestDataFramePlots.test_hist_df_series_cumulative_densitycs`ddlmttjdd}|jjddd}fdd |D}t |d  d dS) Nrrrr!Tr")rrcsg|]}t|r|qSr9rrrr9r:rPrQzETestDataFramePlots.test_hist_df_series_cumulative..rSrhrrr9rr: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皙?)ri)shape)rircCsrtttddgtjdd}t|jd|d}t|jdd}dd |j D}d d |j D}t ||dS) Nrr'rrrir)rrrcSsg|] }|qSr9rcrNpatchr9r9r:rPrQz8TestDataFramePlots.test_hist_weights..cSsg|]}d|qS)rrcrr9r9r:rPrQ) rrr{r/r0r1r2rr5r=rr)r7rr8rrZpatch_height_with_weightsZexpected_patch_heightr9r9r:test_hist_weightssz$TestDataFramePlots.test_hist_weightsc Cstdd|D}tdd|D}tdd|D}tdd|D} |durhtj||dd|durtj||dd|durtj||dd|durtj| |dddS)NcSsg|] }|qSr9rLrMr9r9r:rPrQz7TestDataFramePlots._check_box_coord..cSsg|] }|qSr9rcrMr9r9r:rPrQcSsg|] }|qSr9r[rMr9r9r:rPrQcSsg|] }|qSr9)rBrMr9r9r:rPrQF)Z check_dtype)r/rrr ) r7r= expected_y expected_h expected_x expected_wZresult_yZ result_heightZresult_xZ result_widthr9r9r:_check_box_coordsz#TestDataFramePlots._check_box_coordr"rr! rrrrrrrrrrrr r!rarrPrg)rPr!r rrr)rrPrrrr)rrrrPr r!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)Nrg)r)rrrrrr )rrr!rrT)rr)rr)rrr>rr@rr)rr)r r )rrr)rrr>r)rr5rr r=r/r)r7r"r8rcrFr9r9r:test_hist_df_coords                                                z%TestDataFramePlots.test_hist_df_coordcCs,ttjdd}t|jdddS)NrrTr)rr/r0r1r2cumsumrr5r6r9r9r:test_plot_int_columns_sz(TestDataFramePlots.test_plot_int_columnsmarkers^+o)rrrcCsddlm}|}||dttjd d}|j |d}t | dt |D]\}}|||ks`Jq`dS)Nrorr])style)matplotlib.pyplotrgcfZclf add_subplotrr/r0r1r2r5 enumeraterrZ get_marker)r7rpltr9r8rcrrYr9r9r:test_style_by_columncs    z'TestDataFramePlots.test_style_by_columncCsPtddg}|}|dus$J|jdd}|ddksLJdS)Nr@rTrr)r r5 get_legend get_textsr)r7rrcr9r9r:test_line_label_nonews   z'TestDataFramePlots.test_line_label_nonezprops, expected))ZboxpropsZboxes)Z whiskerpropsZwhiskers)ZcappropsZcaps)Z medianpropsZmedianscCsPtdddD}|ddii}|jjfddi|}||ddksLJdS) NcSs i|]}|tjddqSr)r/r0r1)rNkr9r9r: rQzHTestDataFramePlots.test_specified_props_kwd_plot_box..ABCcolorZC1rrr)rr5r get_color)r7propsrfr8Zkwdrr9r9r:!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 |jsJd d |D}tj|ddjsJdS)Nr!r@r r)rrrtestrrr$cSsg|]}|dqS)r)Z get_positionrr9r9r:rPrQz8TestDataFramePlots.test_unordered_ts..cSsg|] }|qSr9rrr9r9r:rPrQz%Y-%m-%dr)rrr/rr5rrrr r4rruZget_ticklabelsrZis_monotonic_increasingrr) r7r-rr8rcrhZydataZxlocsZxlabelsr9r9r:test_unordered_tss"" z$TestDataFramePlots.test_unordered_tscCs8tdtdgdi}|j|dt|j|dSNrrRrrrVrrr5rr7rr8r9r9r:test_kind_both_wayss  z&TestDataFramePlots.test_kind_both_wayscCs@tdtdgdi}|jdd|dt|j|dddSr1r2r3r9r9r:test_kind_both_ways_x_ys z*TestDataFramePlots.test_kind_both_ways_x_ycCsNttd}d}tjt|d|j|dWdn1s@0YdS)Nabcdno numeric data to plotrTr)rr3rVrWrqr5r7rr8r[r9r9r:test_all_invalid_plot_datas z-TestDataFramePlots.test_all_invalid_plot_datacCs|ttjddtd}d|tjd|jddk<d}tjt |d|j |d Wdn1sn0YdS) Nrr!rr$rrrr7rTr) rr/r0r1r2r4rrVrWrqr5r8r9r9r:(test_partially_invalid_plot_data_numerics z;TestDataFramePlots.test_partially_invalid_plot_data_numericcCsXttjdd}d}tjt|d|jddWdn1sJ0YdS)Nrr:z*invalid_plot_kind is not a valid plot kindrTZinvalid_plot_kindr) rr/r0r1r2rVrWrr5rZr9r9r:test_invalid_kindsz$TestDataFramePlots.test_invalid_kindzx,y,lblrrrcCs^tddgddgddgd}tjtdd  |j|||d Wdn1sP0YdS) Nr@rrPr"rgrrax must be a label or positionrTrRrSrrrVrWrr5)r7rRrSlblr8r9r9r:test_invalid_xy_argss z'TestDataFramePlots.test_invalid_xy_argscCsftddgddgddgd}d}tjt|d $|jd d d gd dWdn1sX0YdS)Nr@rrPr"rgrraz.label should be list-like and same length as yrTrr'rZ bad_labelr>r?rZr9r9r:test_bad_labelsz!TestDataFramePlots.test_bad_labelzx,yrcCs\tgdgdgtdd}tjtdd|j||dWdn1sN0YdS)N)r@rPrg)rr"rZAABrr=rTrQ)rr3rVrWrr5rmr9r9r:test_invalid_xy_args_dup_colssz0TestDataFramePlots.test_invalid_xy_args_dup_colszx,y,lbl,colorsrbluerZbokehcythongreenyellowcCsjtddgddgddgd}t|jd||d |j||||d }t|jt|ksVJt||d dS) Nr@rrPr"rgrrarr>)rRrSrr*Z linecolors)rrr5rrrr)r7rRrSr@rr8rcr9r9r:test_y_listlikes z"TestDataFramePlots.test_y_listlikez x,y,colnamescCs0tddgddgd}||_t|j||ddS)Nr@rrPr"rrQ)rr,rr5)r7rRrSZcolnamesr8r9r9r:test_xy_args_integersz'TestDataFramePlots.test_xy_args_integercCsrttjdjddtjdjddtdtjdjddd}|jjdddd}t|j d ksnJdS) Nrr_rrarr'r!)rRrSZgridsizer@ rr/r0r1rrr5rrrrbr9r9r:test_hexbin_basicsz$TestDataFramePlots.test_hexbin_basiccCsttjdjddtjdjddtdtjdjddd}|jjdddd}t|d j j dkstJt |d d d dS) Nrr_rrarr'T)rRrSr>rr@rrA) rr/r0r1rrr5rrfigurerFrrEr9r9r:test_hexbin_basic_subplotssz-TestDataFramePlots.test_hexbin_basic_subplotsreduce_CcCstttjdjddtjdjddtdtjdjddd}|jjddd|d}t|j d kspJdS) Nrr_rrarr'r)rRrSrZreduce_C_functionr@rK)r7rOr8rcr9r9r:test_hexbin_with_csz%TestDataFramePlots.test_hexbin_with_czkwargs, expectedZBuGnrrrZYlGncCs|ttjdjddtjdjddtdtjdjddd}|jjfddd|}|jdj j |ksxJdS) Nrr_rrarr'rQr) rr/r0r1rrr5rrrr)r7rerfr8rcr9r9r:test_hexbin_cmap+s z#TestDataFramePlots.test_hexbin_cmapcCsdttjddgdgdd}d}tjt|d|jWdn1sV0YdS)NrrgrPXYZrrrder+z/pie requires either y column or 'subplots=True'rT) rr/r0r1rVrWrr5rrZr9r9r:test_pie_df_err>sz"TestDataFramePlots.test_pie_df_errrUcCsFttjddgdgdd}t|jj|d}t|j|j dS)NrrRrSrWr+r) rr/r0r1rr5rrtextsr-)r7rSr8rcr9r9r: test_pie_dfHszTestDataFramePlots.test_pie_dfcCsttjddgdgdd}t|jjddd}t|t|jksLJ|D]}t |j |j qPt ||jD]\}}| |kspJqpdS)NrrRrSrWr+Tr<)rr/r0r1rr5rrr,rr[r-r{r)r7r8rFrcylabelr9r9r:test_pie_df_subplotsRsz'TestDataFramePlots.test_pie_df_subplotscCsttjddgdgdd}gd}gd}t|jjdd||d }t|t|jks`J|D]}t |j |t |j |d qddS) NrrRrSrWr+r)r?grrmT)r=r>rrr) rr/r0r1rr5rrr,rr[rr=)r7r8rZ color_argsrFrcr9r9r:test_pie_df_labels_colorscs$ z,TestDataFramePlots.test_pie_df_labels_colorsc Csttjdd}tdD]}tj|j||f<qtjj dd\}}ddi}t d*|j j fd|dd|Wdn1s0Ygd}t|D]r\}}t|}d ||<d d |jD} | |ksJd d |D} |d|||d d} | | ksJqdS)Nr)r"r"r")Zncols normalizeT)r>rcr)0123r#cSsg|] }|qSr9rrr9r9r:rPrQz6TestDataFramePlots.test_pie_df_nan..cSsg|] }|qSr9rrr9r9r:rPrQr@)rr/r0r1r|rZilocrrr>rrr5rr r3r[r$r%) r7r8rrrFreZ base_expectedrcrfrZ result_labelsexpected_labelsr9r9r:test_pie_df_nanxs   8 z"TestDataFramePlots.test_pie_df_nanr)rrrcCsptdtdddd}t|}tddtddd}t|}t|jfd|i|}t|ddd dS) Nr{rrHrQ皙?皙?yerrrxerrrkr/rronesrr5r)r7rerXr8d_errdf_errrcr9r9r:test_errorbar_plots z%TestDataFramePlots.test_errorbar_plotcCsrtdtdddd}t|}tddtddd}t|}t|dj||dd d }t|d d d dS) Nr{rrHrQrirjr@rT)rkrmrrrrlrn)r7rXr8rprqrcr9r9r:test_errorbar_plot_barsz)TestDataFramePlots.test_errorbar_plot_barcCsvtdtdddd}t|}t|djtddd}t|ddd t|jtd dd}t|dd d dS) Nr{rrHrQrSrjrkr@rl)rr{r)r/rrrr5rorr7rXr8rcr9r9r:test_errorbar_plot_yerr_arrays z0TestDataFramePlots.test_errorbar_plot_yerr_arrayrku誤差cCsttdtdddd}t|}tdd||<t|j|d}t|dddt|jd d |d }t|dd ddS) Nr{rrHrQrirtrrlrSrR)rSrRrkr@rn)r7rkrXr8rcr9r9r:test_errorbar_plot_column_namesz1TestDataFramePlots.test_errorbar_plot_column_namecCsjtdtdddd}t|}tt*|jtjd ddWdn1s\0YdS)Nr{rrHrQrrrt) r/rrrexternal_error_raisedrr5r0r1r2r7rXr8r9r9r:&test_errorbar_plot_external_valueerrors z9TestDataFramePlots.test_errorbar_plot_external_valueerrorcCsvtdtdddd}t|}tdgddgdd}tt|j|dWdn1sh0YdS)Nr{rrHrQzzzrt)r/rrrrxrqr5)r7rXr8rqr9r9r:%test_errorbar_plot_external_typeerrors  z8TestDataFramePlots.test_errorbar_plot_external_typeerror)rYrrCy_errr{ri)rrjcCs@ttdtdddd}t|j||d}t|ddddS)Nr{rrHrQrkrrrlrr/rrr5r)r7rr}r8rcr9r9r:!test_errorbar_plot_different_yerrs z4TestDataFramePlots.test_errorbar_plot_different_yerrz y_err, x_err)riricCsBttdtdddd}t|j|||d}t|ddddS)Nr{rrHrQ)rkrmrrrlr)r7rr}Zx_errr8rcr9r9r:&test_errorbar_plot_different_yerr_xerrsz9TestDataFramePlots.test_errorbar_plot_different_yerr_xerrcCshttdtdddd}ttddtddd}t|jd||d|d}t|d d d dS) Nr{rrHrQrirjT)r=rkrmr>rr@rl)rr/rrorr5r)r7rr8rqrFr9r9r:/test_errorbar_plot_different_yerr_xerr_subplotss"zBTestDataFramePlots.test_errorbar_plot_different_yerr_xerr_subplotszIterator is consumed)rOrWcCsNtdtdddd}t|}t|jtdt|d}t|ddddS) Nr{rrHrQrrtrrl) r/rrrr5 itertoolsrepeatrrrur9r9r:test_errorbar_plot_iteratorsz.TestDataFramePlots.test_errorbar_plot_iteratorcCsvtttjdd}tttjdd}t|j|d}t|dddt|jddd}t|ddddS)Nrr:rtrrlr@)rSrk rr/absr0r1r2rr5rr7r8rqrcr9r9r:'test_errorbar_with_integer_column_namess z:TestDataFramePlots.test_errorbar_with_integer_column_namescCsbtttjdd}tttjddddgd}t|j||d}t|ddddS)Nrr]r:rrr~rlr)r7rr8rqrcr9r9r:'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]r:rr1/1/2000r!MEr(T)ZinplacerYr~rl) rr/rr0r1r2r rrr5r)r7r8rqixrcr9r9r:&test_errorbar_with_partial_columns_dti#sz9TestDataFramePlots.test_errorbar_with_partial_columns_dtierr_boxcCs|Srr9rr9r9r:0rQzTestDataFramePlots.cCsdtdtdddd}t|}|tddtddd}t|j|d}t|dd d dS) Nr{rrHrQrirj)rRrrtr@rlrn)r7rrXr8errrcr9r9r:&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)Nr{rrHrQrirjrz1/1/2001r)r*r^r~rrlrSrR)rSrkrr@T)r=rrkr>)r/rror rrr5r) r7rrXrprZtdfZtdf_errrcrFr9r9r:test_errorbar_timeseries8s.  z+TestDataFramePlots.test_errorbar_timeseriescCstjdd}ttdddj}|j||dd}|jd dj dddf}|ddddft d dg}t ||td }tjt|d |j|jd Wdn1s0YdS) Nr)rPrrgrrPrg)rkrmr@rrHzCAsymmetrical error bars should be provided with the shape (3, 2, 5)rTrt)r/r0r1rrZreshapeTr5r get_pathsZverticesrrrrorprVrWr)r7rr8rcZyerr_0_0Z expected_0_0r[r9r9r:test_errorbar_asymmetricalZs   z-TestDataFramePlots.test_errorbar_asymmetricalcCsttjddttjddd}t|jddt|j|dt dJ|}t |j dksjJt ||jt |j dksJWdn1s0YdS) Nrr]r!r^T)tablerr@)rr/r0r1r3r`rarr5rrrZtablesr rrrbr9r9r: 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)rgrrgrRrSrrQrrl)rRrSrmr@)rRrSrk)rRrSrmrk) 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|r8||q ||q fdd|D}t|t|gt|ddS)Nc3s |]}t|dr|jVqdS)FN)rrrhas_errr9r:rrQz`TestDataFramePlots.test_errorbar_scatter_color.._check_errorbar_color..csg|]}|jvr|qSr9)rrrcr9r:rPrQzaTestDataFramePlots.test_errorbar_scatter_color.._check_errorbar_color..rH) next containersrextendappendrr/rr)rrfrrZerrselZ err_linesrrr:_check_errorbar_colors  zMTestDataFramePlots.test_errorbar_scatter_color.._check_errorbar_colorr)r!rgrWrrrrXrYr)rRrSrmrkrr@rlrZhas_yerrrF)rRrSrkr*r)r) rr/rr0r1r2r5rrr)r7rr8r9rr:test_errorbar_scatter_colors z.TestDataFramePlots.test_errorbar_scatter_colorcCsZtgddd}tjttfdd"|jdddd d Wdn1sL0YdS) Nrr"rz'unknown' is not arTrrunknownr)rRrSrr)rrVrWrKeyErrorr5r6r9r9r:test_scatter_unknown_colormapsz0TestDataFramePlots.test_scatter_unknown_colormapcCsddlm}|dt\}}tgdgdgdgdd}dd}|D]}|jddd |d d qN|||||dt\}}t t  |jd |d d }Wdn1s0Y||dS) Nrrr@rrPr"rgrrrrrXcSs|D]&}t|jdksJt|ddq|d|dfD](}t|ddt|jddddq<|d|dfD](}t|ddt|jddddqvdS) Nr@TrqrrFrsrP)rrrrrvrFrcr9r9r:_checksz5TestDataFramePlots.test_sharex_and_ax.._checkrrrlT)rRrSrlrcsharex)r>rcr) rrclose_generate_4_axes_via_gridspecrr5 tight_layoutrrrrr7r!gsrFr8rrcr9r9r:test_sharex_and_axs(        .z%TestDataFramePlots.test_sharex_and_axcCsddlm}tgdgdgdgdd}t\}}|D]}|jddd|dq<|||D]J}t|jdks|Jt | d d t | d d t |j d d d d qfdS Nrrrrrrl)rRrSrlrcr@Trqrs rrrrr5rrrrrrrvr7r!r8rrFrcr9r9r:test_sharex_false_and_axs"  z+TestDataFramePlots.test_sharex_false_and_axcCsddlm}t\}}tgdgdgdgdd}dd}|D]}|jddd|d d qD|||||d t\}}t t  |jd |d d }Wdn1s0Y||||dS) NrrrcSs|D]:}t|jdksJt|ddt|jddddq|d|dfD]}t|ddqP|d|dfD]}t|ddqvdS) Nr@TrqrsrrrPF)rrrrvrrr9r9r:rsz5TestDataFramePlots.test_sharey_and_ax.._checkrrrlT)rRrSrlrcr8r)r>rcr8) rrrrr5rrrrrrrr9r9r:test_sharey_and_axs(       .z%TestDataFramePlots.test_sharey_and_axcCsddlm}tgdgdgdgdd}t\}}|D]}|jddd|dq<|||D]J}t|jdks|Jt | d d t | d d t |j d d d d qfdSrrrr9r9r:test_sharey_and_ax_tights"  z+TestDataFramePlots.test_sharey_and_ax_tightcCstdi}|dvrlttjdjddtjdjddtdtjdjddd}ddd }nr|d krttjdd t t d t d t ddddd }n2ttjdd t t d t d t ddddd}t|jfd|i|}tdt|dusJdS)z3Check that every plot type gets properly collected.r)rrrrr_rrarr'rQrr r#r$r&r!r(r+rrN)rVrrr/r0r1rrr2rr3r4r rweakrefrefr5r!rgcZcollect)r7rargsr8rr9r9r:test_memory_leak3s2     z#TestDataFramePlots.test_memory_leakc sddlmddlmttjddt dddd}t tjdd|j t d d }fd d }fd d}||fD]\}}|j |d}t|jdksJ|j |d}t|jdksJ||fD]8}t|ddt|ddt|jddddqЈdq||fD]\}}|j d||gd}t|jdksPJt|jdksdJ|D]:}t|ddt|ddt|jddddqhdq"|\}}tt&|j d||gddd}Wdn1s0Yt|djdksJt|djdks.J||fD]}t|ddq6t|dddt|djddddt|dddt|djddddd|\}}tt&|j d||gddd}Wdn1s0Yt|djdksJt|djdks,Jt|dddt|ddd||fD]*}t|ddt|jddddq\ddS)Nrgridspecrr!rr)r^r:ABrcsLdd}}||ddddf}||dddf}||fS)NrPr@rGridSpecrMrrr9rrrr!r9r:_get_vertical_gridhs  zSTestDataFramePlots.test_df_gridspec_patterns_vert_horiz.._get_vertical_gridcsLdd}}||ddddf}||dddf}||fS)Nr@rPrrrrr9r:_get_horizontal_gridos  zUTestDataFramePlots.test_df_gridspec_patterns_vert_horiz.._get_horizontal_gridrr@Trqrsrrr>rcrr8Frrrrr r/r0r1r2r rr-r3r5rrrrrvrrrr) r7tsr8rrrrrcrFr9rr:$test_df_gridspec_patterns_vert_horizXsn         6    6 z7TestDataFramePlots.test_df_gridspec_patterns_vert_horizcsddlmddlmttjddt dddd}fdd }|}t tjdd |j t d d }|j d |d}|D]J}t|jdksJt|d dt|d dt|jd dd dqd|}tt"|j d |d d d}Wdn1s0Y|D]}t|jdks Jq |d|dfD]}t|d dqL|d|dfD]}t|ddqt|d|dfD]*}t|ddt|jd dddq|d|dfD]*}t|d dt|jd dd dq؈ddS)Nrrrr!rrr^cstdd}}||ddddf}||dddf}||dddf}||d}||||fS)NrPrrIr)rr9rrZax3Zax4rr9r:_get_boxed_grids zKTestDataFramePlots.test_df_gridspec_patterns_boxed.._get_boxed_gridr r#rTrr@rqrsrrrPFr)r7rrrFr8rcr9rr:test_df_gridspec_patterns_boxedsH      2z2TestDataFramePlots.test_df_gridspec_patterns_boxedcCs.ttgdgddtjjdddddS)NrrrrrrQ)kws)rrr PlotAccessorZ_dataframe_kinds)r7r9r9r:test_df_grid_settingss z(TestDataFramePlots.test_df_grid_settingscCs>tj\}}|gdttjddj|ddS)N)ririririrr!r) rrr>add_axesr r/r0r1r5)r7r9rcr9r9r:test_plain_axessz"TestDataFramePlots.test_plain_axescCsXttjddtjddd}tj}|d}|j d|ddddd dS) Nrrr)rrr@r@rrrZhsv)rrcrRrSrr) rr/r0r1r2rrrMrr5)r7r8r9rcr9r9r: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>Zmpl_toolkits.axes_grid1rZ append_axesr r/r0r1r5)r7r9rcrdividerZcaxr9r9r:#test_plain_axes_make_axes_locatables  z6TestDataFramePlots.test_plain_axes_make_axes_locatablecCsjtj\}}ddlm}||dddd}ttjddj |d ttjddj |d dS) Nr) inset_axesz30%rrP)rFrrrr!r) rrr>Z%mpl_toolkits.axes_grid1.inset_locatorrr r/r0r1r5)r7r9rcrZiaxr9r9r:test_plain_axes_make_inset_axes s  z2TestDataFramePlots.test_plain_axes_make_inset_axesmethod)rYrCrcCsvttjddtddjdddjddd }d }d d g}||d d}t|j|fi|}t |j |ddS)Nr)rrrrcSs |jSr)r'rr8r9r9r:r rQzBTestDataFramePlots.test_secondary_axis_font_size..rcSs |jdS)Ng?rrr9r9r:r rQ)rr_rrT)r.rZ mark_right)rFr) rr/r0r1r2r3Zassignrr5rr1)r7rr8rr rercr9r9r:test_secondary_axis_font_size s z0TestDataFramePlots.test_secondary_axis_font_sizecCstgdgdgdd}|jjdd}|ddd d |D}tt||}|d d ksjJ|d dkszJ|ddksJdS)N)rPrrP)r_*)MondayTuesday Wednesday)ZsalesZvisitsdayrrrHrPcSsg|] }|qSr9rrNtr9r9r:rP, rQzATestDataFramePlots.test_x_string_values_ticks..rrrrrr)rr5rset_xlimrvrr{ get_xticks)r7r8rc 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|ddksJ|ddksJ|ddksJdS)Nr.ir@rr7rr'r+rHr"cSsg|] }|qSr9rrr9r9r:rP> rQzETestDataFramePlots.test_x_multiindex_values_ticks..z (2012, 1)rz (2012, 2)rz (2013, 1)rz (2013, 2)r) rZ from_productrr/r0r1r2r5rrvrr{r)r7r-r8rcrrr9r9r:test_x_multiindex_values_ticks3 s z1TestDataFramePlots.test_x_multiindex_values_tickscCsLtddgddgd}|j|d}|}|ddks8J|ddksHJdS)Nrr"r@r^rr)rr5r)r7rr8rcxlimsr9r9r:test_xlim_plot_lineF s  z&TestDataFramePlots.test_xlim_plot_linecCstj\}}gd}tgdgdgdgdd|d}|gdjj|d d |d gj|d d |}|ddksJ|ddksJdd|D}||ksJdS)N)Zk1Zk2Zk3Zk4)irr)iixri )rri@i)r@rPr"rP)s1s2s3r.r^)rrrF)rcrr.T)rcr.rr@rPcSsg|] }|qSr9rrr9r9r:rPg rQzWTestDataFramePlots.test_xlim_plot_line_correctly_in_mixed_plot_type..)rrr>rr5rrrv)r7r9rcZindexesr8rrr9r9r:0test_xlim_plot_line_correctly_in_mixed_plot_typeP s" zCTestDataFramePlots.test_xlim_plot_line_correctly_in_mixed_plot_typecCsltdgtd}|jsJ|}t|dks4J|d}t|dksTJt|dkshJdS)NZfoo)r,r%r@r)rintemptyr5rrrr)r7r8rcrYr9r9r:test_plot_no_rowsj s  z$TestDataFramePlots.test_plot_no_rowscCsFtgd}tjtdd|Wdn1s80YdS)Nrr7rT)rrVrWrqr5r6r9r9r:test_plot_no_numeric_datat s z,TestDataFramePlots.test_plot_no_numeric_data)rYrrCrrrrrcCstdtdtddtddtdtdd}t|}|jddg|d}t|dksjJd d gd d gd gf}t||D]:\}}|dkrt||d|dkrt|j t|ksJqdS)Nrr!r@rW)rrY)rrXr>rrPrrYrrXrrrrY) rVrr/rrr5rr{rr)r7rrXr8rFrgrcrr9r9r:test_group_subploty s     z%TestDataFramePlots.test_group_subplotcCsPttd}d}tjt|d|jdgdWdn1sB0YdS)Nr@z!An iterable subplots for a SeriesrTrr>)r r|rVrWNotImplementedErrorr5)r7Zserr[r9r9r:(test_group_subplot_series_notimplemented s z;TestDataFramePlots.test_group_subplot_series_notimplementedcCs`ttdtddgd}d}tjt|d|jdgdWdn1sR0YdS)Nrrzrlrz6An iterable subplots for a DataFrame with a MultiIndexrTr) rr/eyerr}rVrWrr5rZr9r9r:,test_group_subplot_multiindex_notimplemented sz?TestDataFramePlots.test_group_subplot_multiindex_notimplementedcCsZttdddgd}d}tjt|d|jdgdWdn1sL0YdS)Nrrrz4An iterable subplots for a DataFrame with non-uniquerTrr)rr/rrVrWrr5rZr9r9r:0test_group_subplot_nonunique_cols_notimplemented szCTestDataFramePlots.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|dWdn1sN0YdS)Nr!rrTrr/rrrVrWrr5)r7r>Z expected_msgrXr8r9r9r:test_group_subplot_bad_input s z/TestDataFramePlots.test_group_subplot_bad_inputcCs^tdtdd}t|}tjtdd|jdgdWdn1sP0YdS)Nr!rz Column label\(s\) \['bad_name'\]rT)rZbad_namerrryr9r9r:&test_group_subplot_invalid_column_name sz9TestDataFramePlots.test_group_subplot_invalid_column_namecCshtdtdtdd}t|}tjtdd |jddgdWdn1sZ0YdS)Nr!rzshould be in only one subplotrTr)rrrrryr9r9r:$test_group_subplot_duplicated_column sz7TestDataFramePlots.test_group_subplot_duplicated_column)rrrcCs`tdtdd}t|}tjtdd |jdg|dWdn1sR0YdS)Nr!rz1When subplots is an iterable, kind must be one ofrTrr)r7rrXr8r9r9r:test_group_subplot_invalid_kind s z2TestDataFramePlots.test_group_subplot_invalid_kindz index_name, old_label, new_label)Nr#new)oldrr)Nr#r#)Nr#r@r#)rYrrcCstddgddggddgd}||j_|j|d}||ks@J|dksPJ|j|||d }|t|kstJ|t|ksJdS) Nr@rrgType AType Brrr#)rr]xlabel)rr-rr5 get_xlabelrstr)r7rZ index_nameZ old_labelZ new_labelr8rcr9r9r:(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}||durJ|n|ksTJ||durf|n|kspJdS)Nr r r@rrgr)rrRrSr r])rr5r r)r7rr r]ZxcolZycolr8rcr9r9r:'test_xlabel_ylabel_dataframe_plane_plot s z:TestDataFramePlots.test_xlabel_ylabel_dataframe_plane_plot)FTcCsjtdgj|dddgd}|jjD]B}|jr"|dks@J|dksPJ|ddks"Jq"dS)NrrU)rrir)r.r]rjrf) rr5rMrFrZ get_visiblerr5Z get_yticks)r7r.Zax_dfrcr9r9r:test_secondary_y s    z#TestDataFramePlots.test_secondary_ycCsrttjddttdtdtddddd }t d "| }|j }Wdn1sd0YdS) 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__rVmarkZslowr;rGrJrNZxfailr\rdZ parametrizerorxrzr~rrrr3rrrrrrrrrrrrrrrrrr/rr rrrrrrrrrrrr|r r'r,r-r0r3r6r:r@rArIrJrXr]r^rgrirrZCategoricalIndexrkrnrrrsrwr~paramtdZ skip_if_norrrrrrrrrrrrrrrrrfilterwarningsrrrrrrrrrrrrrrrorr rrrrr"r&r-r0r rZ _common_kindsr4r5r9r;r<rArBrCrIrJrLrNZstdrPrQrZr\r^rarhrrrsrvrwrzr|r rrrrrrrrrrrrrrrrrrrZ _all_kindsrrrrrrrrrrrrrrrrrrrrrrrrrrrr9r9r9r:r6sR                                                                                        ."""***l         " ,           (((6    !      (  ( $ S 5                 rcCsbddlm}tjdd}||d}||d}||d}||d}|||||gfS)Nrrr)r@rrzr)rrrrrZsubplot)r!rZax_tlZax_llZax_trZax_lrr9r9r:r s r)4__doc__rrrrror`rnumpyr/rVZpandas.util._test_decoratorsutilZ_test_decoratorsrZpandas.core.dtypes.apirZpandasrrrrrr r r r r Zpandas._testingZ_testingrZpandas.tests.plotting.commonrrrrrrrrrrrrrZpandas.io.formats.printingrrrr!rrr9r9r9r:sL , <   t