B x\}|rR|dk s.check_linecsg|]\}}||qSrr).0rCrD)rErr sz,TestPlotBase._check_data..N) get_lineslenr;rr+r2)r-ZxpZrsZxp_linesZrs_linesr)rEr _check_datays zTestPlotBase._check_datacCsFddlm}t||s$t|s$|g}x|D]}||ks*tq*WdS)a Check each artist is visible or not Parameters ---------- collections : matplotlib Artist or its list-like target Artist or its list or collection visible : bool expected visibility r) CollectionN)matplotlib.collectionsrK isinstancer get_visibler;)r- collectionsr?rKpatchrrr_check_visibles  zTestPlotBase._check_visiblecs*|}tt||fdd|jDS)Ncsg|] }|qSrr)rFv)mappedrrrGsz3TestPlotBase._get_colors_mapped..)uniquedictrvalues)r-Zseriesr7rTr)rSr_get_colors_mappedszTestPlotBase._get_colors_mappedcCsddlm}ddlm}m}m}|j} |dk r|dk rR|||}|dt|}t|t|ksft xrt ||D]d\} } t | |r| } | | } n(t | ||frt| d} n| } | | } | | ksrt qrW|dk r|dk r |||}|dt|}t|t|ks t xht ||D]Z\} } t | |rN| d} n| } t | tjrlt| } | | } | | ks,t q,WdS)a3 Check each artist has expected line colors and face colors Parameters ---------- collections : list-like list or collection of target artist linecolors : list-like which has the same length as collections list of expected line colors facecolors : list-like which has the same length as collections list of expected face colors mapping : Series Series used for color grouping key used for andrew_curves, parallel_coordinates, radviz test r)Line2D)rKPolyCollectionLineCollectionN)Zmatplotlib.linesrXrLrKrYrZr8rWrIr;rrMZ get_colorZto_rgbatupleZ get_edgecolorZ get_facecolorr,Zndarray)r-rOZ linecolorsZ facecolorsmappingrXrKrYrZZconvrPcolorresultexpectedrrr _check_colorss<          zTestPlotBase._check_colorscCsdt|s||ks`tnFdd|D}t|t|ks.N)rrar;rIr)r-Ztextsr_r>labelerrrr<s zTestPlotBase._check_text_labelsc Csddlm}||}x|D]}|s(|rt|j|rB|}n||jdd}x<|D]4} |dk rxt| ||dk r\t| |q\W|s|rt|j |r| }n| |j dd}x<|D]4} |dk rt| ||dk rt| |qWqWdS)a Check each axes has expected tick properties Parameters ---------- axes : matplotlib Axes object, or its list-like xlabelsize : number expected xticks font size xrot : number expected xticks rotation ylabelsize : number expected yticks font size yrot : number expected yticks rotation r) NullFormatterT)minorN) Zmatplotlib.tickerrer:rMxaxisZget_minor_formatterZget_xticklabelsr+rBZ get_fontsizeZ get_rotationyaxisZget_yticklabels) r-r=Z xlabelsizeZxrotZ ylabelsizeZyrotrer@r>rcrrr_check_ticks_propss2             zTestPlotBase._check_ticks_propslinearcCs@||}x0|D](}|j|ks&t|j|kstqWdS)a Check each axes has expected scales Parameters ---------- axes : matplotlib Axes object, or its list-like xaxis : {'linear', 'log'} expected xaxis scale yaxis : {'linear', 'log'} expected yaxis scale N)r:rgZ get_scaler;rh)r-r=rgrhr@rrr_check_ax_scales,s  zTestPlotBase._check_ax_scalescCs|dkr|j}||}|dk rRt||ks0tx |D]}t|dks6tq6W|dk rt|t|}||ksttt|dj t j |t j ddS)a Check expected number of axes is drawn in expected layout Parameters ---------- axes : matplotlib Axes object, or its list-like axes_num : number expected number of axes. Unnecessary axes should be set to invisible. layout : tuple expected layout, (expected number of rows , columns) figsize : tuple expected figsize. default is matplotlib default Nr)Zdtype)r*r:rIr;Z get_children_get_axes_layoutr r+Zassert_numpy_array_equalfigureZget_size_inchesr,ZarrayZfloat64)r-r=Zaxes_numZlayoutZfigsizeZ visible_axesr@r^rrr_check_axes_shape=s    zTestPlotBase._check_axes_shapecCsZt}t}x<|D]4}|}||dd||ddqWt|t|fS)Nr)setZ get_positionZ get_pointsaddrI)r-r=Zx_setZy_setr@Zpointsrrrrl_s  zTestPlotBase._get_axes_layoutcCst|}dd|D}|S)z Flatten axes, and filter only visible Parameters ---------- axes : matplotlib Axes object, or its list-like cSsg|]}|r|qSr)rN)rFr@rrrrGssz1TestPlotBase._flatten_visible..)r )r-r=rrrr:is zTestPlotBase._flatten_visiblerc Cs||}xp|D]h}|j}d}d}x<|D]4}t|dd} t|dd} | rP|d7}| r(|d7}q(W||kslt||kstqWdS)a Check axes has expected number of errorbars Parameters ---------- axes : matplotlib Axes object, or its list-like xerr : number expected number of x errorbar yerr : number expected number of y errorbar rhas_xerrFhas_yerrroN)r: containersgetattrr;) r-r=ZxerrZyerrr@rtZ xerr_countZ yerr_countcrrrsrrr_check_has_errorbarsvs       z!TestPlotBase._check_has_errorbarsc Csddlm}t|td}|dkrj|dkr,d}t|||s>t|dkrft|j|sVtt|jtsftn$|dkrx ||D]}t||s~tq~WdSt|t stt | t |kstxt |D]\}} t| ||st|dkr|r| |kstq|dkrP|r*| j |ks*tt| j|s      z#TestPlotBase._check_box_return_typecsvddl}fdd}d}xV|D]L}t|s0q jddt|||d7}|jddd|jfd |i||r|tjddt|||d7}|jdd d|jf|dd ||rt|d kr jddt|||d7}|jdd d|jfd |i||s tjddt|||d7}|jddd|jf|d d ||s tq WdS) NrcsFtddjjD}tddjjD}|oB| S)Ncss|]}|j VqdS)N)gridOn)rFgrrr szHTestPlotBase._check_grid_settings..is_grid_on..css|]}|j VqdS)N)r)rFrrrrrs)allr6ZgcargZget_major_ticksrh)ZxoffZyoff)r-rr is_grid_ons z5TestPlotBase._check_grid_settings..is_grid_onror!r=F)gridrT)rrZpie)r(rr6ZsubplotrIZrcZplotr;)r-objZkindsZkwsr/rZspndxrr)r-r_check_grid_settingss6    z!TestPlotBase._check_grid_settingsr]csfdd|dDS)zT Auxiliary function for correctly unpacking cycler after MPL >= 1.5 csg|] }|qSrr)rFrR)fieldrrrGsz/TestPlotBase._unpack_cycler..zaxes.prop_cycler)r-ZrcParamsrr)rr_unpack_cyclerszTestPlotBase._unpack_cycler)NT)T)NNN)NNNN)rjrj)NNN)rr)NT)r])__name__ __module__ __qualname__r1r3rr6r8rArJrQrWr`r<rirkrnrlr:rwrrrrrrrr0s.!     > 1  !   9 *ralwaysc Ksddlm}d}tt|zy |d}Wntk rN|}YnX||d| d}|f|}t |y| d|d<|f|}Wnt k rYn Xt |t dd}| |WdQRXWdt|X|SQRXdS)Nrrmr@T)Zreturn_filelike)r4r5warningscatch_warnings simplefilterKeyErrorZgcfZclfgetZ add_subplotr Exceptionr Zsavefigr+r2)ffilterwarningskwargsr6ZretZfigr@pathrrr_check_plot_workss.       rcCstjtjt\}}|S)N)osrsplitabspath__file__)Zpth_rrrcurpathsr)r)$rrZnumpyr,rrZ pandas.compatrrZpandas.util._decoratorsrZpandas.util._test_decoratorsutilZ_test_decoratorsZtdZpandas.core.dtypes.apirZpandasrrZpandas.util.testingZtestingr+r r Zpandas.plottingr)Zpandas.plotting._toolsr rrZskip_if_no_mplobjectrrrrrrrs,      N "