o dJ@sdZddlmZddlmZmZddlZddlm Z ddl m m Z ddlmZddlZddlmZddlmZerAddlmZe jGd d d Zdd d ZddZddZdS)zF Module consolidating common testing functions for checking plotting. ) annotations) TYPE_CHECKINGSequenceN)cache_readonly) is_list_like)SeriesAxesc@seZdZdZddZddZeddZedd Zd;d d Z d;ddZ ddZ dddZd d!Z d?d"d#Zd@d%d&Zd>d'd(Zd)d*Zd+d,ZdAd.d/Z d;d0d1Zifd2d3ZdBd5d6Zd7d8Zd9d:Zd S)C TestPlotBasezE This is a common base class used for various plotting tests cCsddl}|dSNr) matplotlib rcdefaults)selfmplra/var/www/html/visualizacion-main/env/lib/python3.10/site-packages/pandas/tests/plotting/common.py setup_method!s zTestPlotBase.setup_methodcCs tdSN)tmcloserrrrteardown_method&s zTestPlotBase.teardown_methodcCsddlm}|Sr )matplotlib.pyplotpyplot)rpltrrrr)s zTestPlotBase.pltcCsddlm}|jS)Nr)colors)r rcolorConverter)rrrrrcolorconverter/s zTestPlotBase.colorconverterNTcCsf|r |dur td||}|D]}|r(|dusJ|||q|dus0JqdS)aQ Check each axes has expected legend labels Parameters ---------- axes : matplotlib Axes object, or its list-like labels : list-like expected legend labels visible : bool expected legend visibility. labels are checked only when visible is True Nz-labels must be specified when visible is True) ValueError_flatten_visible get_legend_check_text_labels get_texts)raxeslabelsvisibleaxrrr_check_legend_labels5s  z!TestPlotBase._check_legend_labelscCsV|r |dur td|r!|\}}dd|D}||ksJdS|dus)JdS)aB Check ax has expected legend markers Parameters ---------- ax : matplotlib Axes object expected_markers : list-like expected legend markers visible : bool expected legend visibility. labels are checked only when visible is True Nz.Markers must be specified when visible is TruecSg|]}|qSr) get_marker).0handlerrr ]z5TestPlotBase._check_legend_marker..)rget_legend_handles_labelsr )rr&expected_markersr%handles_markersrrr_check_legend_markerLs  z!TestPlotBase._check_legend_markerc Cs`|}|}t|t|ksJt||D]\}}|}|}t||qtdS)z Check each axes has identical lines Parameters ---------- xp : matplotlib Axes object rs : matplotlib Axes object N) get_lineslenzip get_xydatarassert_almost_equalr) rxprsxp_linesrs_linesxplrslxpdatarsdatarrr _check_databs  zTestPlotBase._check_datacCsBddlm}t||st|s|g}|D] }||ksJqdS)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.collectionsrB isinstancer get_visible)r collectionsr%rBpatchrrr_check_visiblevs zTestPlotBase._check_visibler#Axes | Sequence[Axes]filledboolreturnNonecCs2||}|D]}|jD] }|j|ksJq qdS)z Check for each artist whether it is filled or not Parameters ---------- axes : matplotlib Axes object, or its list-like filled : bool expected filling N)rpatchesfill)rr#rJr&rGrrr_check_patches_all_filleds  z&TestPlotBase._check_patches_all_filledcs*|}tt||fdd|jDS)Ncsg|]}|qSrrr*vmappedrrr,r-z3TestPlotBase._get_colors_mapped..)uniquedictr6values)rseriesrrUrrSr_get_colors_mappedszTestPlotBase._get_colors_mappedcCs|ddlm}m}m}ddlm}|j} |durk|dur)|||}|dt|}t|t|ks3Jt ||D]2\} } t | |rK| } | | } nt | ||fr[t | d} n| } | | } | | ksjJq8|dur|dur|||}|dt|}t|t|ksJt ||D]+\} } t | |r| d} n| } t | tjrt | } | | } | | ksJqdSdS)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)rBLineCollectionPolyCollection)Line2DN)rCrBrZr[matplotlib.linesr\rrYr5r6rD get_colorto_rgbatuple get_edgecolor get_facecolornpndarray)rrF linecolors facecolorsmappingrBrZr[r\convrGcolorresultexpectedrrr _check_colorss@         zTestPlotBase._check_colorscCsbt|s||ks JdSdd|D}t|t|ksJt||D] \}}||ks.Jq$dS)a. Check each text has expected labels Parameters ---------- texts : matplotlib Text object, or its list-like target text, or its list expected : str or list-like which has the same length as texts expected text label, or its list cSr(r)get_text)r*trrrr,r-z3TestPlotBase._check_text_labels..N)rrmr5r6)rtextsrkr$labelerrrr!s zTestPlotBase._check_text_labelsc Csddlm}||}|D]z}|dus|durKt|j|r$|}n ||jdd}|D]} |dur>t| ||durJt| |q0|dusS|durt|j |r`| }n | |j dd}|D]} |durzt| ||durt| |qlq dS)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) NullFormatterNT)minor) matplotlib.tickerrrrrDxaxisget_minor_formatterget_xticklabelsrr8 get_fontsize get_rotationyaxisget_yticklabels) rr# xlabelsizexrot ylabelsizeyrotrrr&r$rprrr_check_ticks_propss0    zTestPlotBase._check_ticks_propslinearcCs<||}|D]}|j|ksJ|j|ksJqdS)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)rru get_scalerz)rr#rurzr&rrr_check_ax_scales%s zTestPlotBase._check_ax_scalesc Csddlm}|dur d}||}|dur,t||ksJ|D] }t|dks+Jq|dur=|||}||ks=Jt|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 r flatten_axesN)g@g333333@)dtype) !pandas.plotting._matplotlib.toolsrrr5 get_children_get_axes_layoutrassert_numpy_array_equalfigureget_size_inchesrcarrayfloat64) rr#axes_numlayoutfigsizer visible_axesr&rjrrr_check_axes_shape6s    zTestPlotBase._check_axes_shapecCsVt}t}|D]}|}||dd||ddqt|t|fS)Nr)set get_position get_pointsaddr5)rr#x_sety_setr&pointsrrrrZs zTestPlotBase._get_axes_layoutcCs&ddlm}||}dd|D}|S)z Flatten axes, and filter only visible Parameters ---------- axes : matplotlib Axes object, or its list-like rrcSsg|]}|r|qSr)rE)r*r&rrrr,psz1TestPlotBase._flatten_visible..)rr)rr#rrrrrds zTestPlotBase._flatten_visiblerc Csx||}|D]2}|j}d}d}|D]}t|dd} t|dd} | r&|d7}| r,|d7}q||ks3J||ks9JqdS)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_yerrrN)r containersgetattr) rr#xerryerrr&r xerr_count yerr_countcrrrrr_check_has_errorbarsss    z!TestPlotBase._check_has_errorbarsc Csvddlm}t|td}|dur7|durd}t|||sJ|dkr3t|j|s+Jt|jts5JdSdS|durL||D] }t||sIJq@dSt|tsSJt | t |ks_J| D]U\}} t| ||spJ|dkr|r~| |ks~Jqc|dkr|r| j |ksJt| j|sJt| jtsJqc|dkr| dd} | j } |r| |ksJqctdS) at Check box returned type is correct Parameters ---------- returned : object to be tested, returned from boxplot return_type : str return_type passed to boxplot expected_keys : list-like, optional group labels in subplot case. If not passed, the function checks assuming boxplot uses single ax check_ax_title : bool Whether to check the ax.title is the same as expected_key Intended to be checked by calling from ``boxplot``. Normal ``plot`` doesn't attach ``ax.title``, it must be disabled. rr)rVr#bothNrVrr#medians)matplotlib.axesr rVr`rDr&linesrrsortedkeysitems get_titler#AssertionError) rreturned return_type expected_keyscheck_ax_titler typesrkeyvalueliner#rrr_check_box_return_typesH   z#TestPlotBase._check_box_return_typecsddl}fdd}d}|D]}jddt|||d7}|jddd|jd d |i||r6Jjjddt|||d7}|jdd d|jd |dd ||rbJj|d vrˆjddt|||d7}|jdd d|jd d |i||sJjjddt|||d7}|jddd|jd |d d ||sJjqdS)NrcsNjj}jj}tdd|D}tdd|D}|o%| S)Ncs|] }|j VqdSrgridlinerEr*grrr zHTestPlotBase._check_grid_settings..is_grid_on..csrrrrrrrrr)rgcaruget_major_ticksrzall)xticksyticksxoffyoffrrr is_grid_ons  z5TestPlotBase._check_grid_settings..is_grid_onrr#F)gridkindT)rr)piehexbinscatterr)r rsubplotr5rcplotclf)robjkindskwsrrspndxrrrr_check_grid_settingss>         z!TestPlotBase._check_grid_settingsricsfdd|dDS)zT Auxiliary function for correctly unpacking cycler after MPL >= 1.5 csg|]}|qSrrrQfieldrrr,r-z/TestPlotBase._unpack_cycler..zaxes.prop_cycler)rrcParamsrrrr_unpack_cyclerszTestPlotBase._unpack_cyclercC |jdS)Nx _shared_axesrr&rrr get_x_axis zTestPlotBase.get_x_axiscCr)Nyrrrrr get_y_axisrzTestPlotBase.get_y_axis)NT)T)r#rIrJrKrLrM)NNN)NNNN)rr)rr)ri)__name__ __module__ __qualname____doc__rrrrrr'r3rArHrPrYrlr!rrrrrrrrrrrrrrrr s>       @ 0 $  < , r Fc Ksddlm}|r t}nt}d}zH|d|}||||fi|D]}t|q%tj dd }| |Wdn1sCwYWt ||SWt ||St |w)a Create plot and ensure that plot return object is valid. Parameters ---------- f : func Plotting function. default_axes : bool, optional If False (default): - If `ax` not in `kwargs`, then create subplot(211) and plot there - Create new subplot(212) and plot there as well - Mind special corner case for bootstrap_plot (see `_gen_two_subplots`) If True: - Simply run plotting function with kwargs provided - All required axes instances will be created automatically - It is recommended to use it when the plotting function creates multiple axes itself. It helps avoid warnings like 'UserWarning: To output multiple subplots, the figure containing the passed axes is being cleared' **kwargs Keyword arguments passed to the plotting function. Returns ------- Plot object returned by the last plotting. rNrT)return_filelike) rr_gen_default_plot_gen_two_subplotsgetgcfrr"assert_is_valid_plot_return_object ensure_cleansavefigr)f default_axeskwargsr gen_plotsretfigpathrrr_check_plot_workss&      rcks|di|VdS)z' Create plot in a default way. Nrrrrrrrrr6srcks`d|vr |d|di|V|tjjurd|vsJn|d|d<|di|VdS)z9 Create plot on two subplots forcefully created. r&Nr) add_subplotpdplottingbootstrap_plotrrrrr=s  r)F)r __future__rtypingrrnumpyrcpandas.util._decoratorsrpandas.util._test_decoratorsutil_test_decoratorstdpandas.core.dtypes.apirpandasrrpandas._testing_testingrrr skip_if_no_mplr rrrrrrrs(       j3