o iL@sdZddlmZddlmZmZddlZddlZddl m Z ddl m m ZddlmZddlZddlmZddlmZerEddlmZejGd d d Zdd dZddZddZdS)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)selfmplrW/var/www/edux/Edux_v2/venv/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}|jSr )matplotlib.colorscolorscolorConverter)rrrrrcolorconverter0s 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_labels6s  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_markerMs  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_datacs  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.collectionsrC isinstancer get_visible)r collectionsr&rCpatchrrr_check_visiblews 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)r patchesfill)rr$rKr'rHrrr_check_patches_all_filleds  z&TestPlotBase._check_patches_all_filledcs*|}tt||fdd|jDS)Ncsg|]}|qSrrr+vmappedrrr-r.z3TestPlotBase._get_colors_mapped..)uniquedictr7values)rseriesrrVrrTr_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)rCLineCollectionPolyCollection)Line2DN)rDrCr[r\matplotlib.linesr]rrZr6r7rE get_colorto_rgbatuple get_edgecolor get_facecolornpndarray)rrG linecolors facecolorsmappingrCr[r\r]convrHcolorresultexpectedrrr _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)rrnr6r7)rtextsrlr%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.tickerrsr rExaxisget_minor_formatterget_xticklabelsrr9 get_fontsize get_rotationyaxisget_yticklabels) rr$ xlabelsizexrot ylabelsizeyrotrsr'r%rqrrr_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)r rv get_scaler{)rr$rvr{r'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.toolsrr r6 get_children_get_axes_layoutrassert_numpy_array_equalfigureget_size_inchesrdarrayfloat64) rr$axes_numlayoutfigsizer visible_axesr'rkrrr_check_axes_shape9s    zTestPlotBase._check_axes_shapecCsVt}t}|D]}|}||dd||ddqt|t|fS)Nr)set get_position get_pointsaddr6)rr$x_sety_setr'pointsrrrr]s 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)rF)r+r'rrrr-ssz1TestPlotBase._flatten_visible..)rr)rr$rrrrr gs 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_errorbarsvs    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)rWr$bothNrWrr$medians)matplotlib.axesr rWrarEr'linesr rsortedkeysitems get_titler$AssertionError) rreturned return_type expected_keyscheck_ax_titler typesrkeyvalueliner$rrr_check_box_return_typesH   z#TestPlotBase._check_box_return_typecsbddl}fdd}d}|D]}jddt|||d7}|jddd|jd d |i||r6Jjddt|||d7}|jdd d|jd |dd ||r]J|d vrjddt|||d7}|jdd d|jd d |i||sJjddt|||d7}|jddd|jd |d d ||sJqdS)NrcsNjj}jj}tdd|D}tdd|D}|o%| S)Ncs|] }|j VqdSrgridlinerFr+grrr zHTestPlotBase._check_grid_settings..is_grid_on..csrrrrrrrrr)rgcarvget_major_ticksr{all)xticksyticksxoffyoffrrr is_grid_ons  z5TestPlotBase._check_grid_settings..is_grid_onrr$F)gridkindT)rr)piehexbinscatterr)r rsubplotr6rcplot)robjkindskwsrrspndxrrrr_check_grid_settingss6     z!TestPlotBase._check_grid_settingsrjcsfdd|dDS)zT Auxiliary function for correctly unpacking cycler after MPL >= 1.5 csg|]}|qSrrrRfieldrrr-r.z/TestPlotBase._unpack_cycler..zaxes.prop_cycler)rrcParamsrrrr_unpack_cyclerszTestPlotBase._unpack_cyclercC"ddlm}|r|jdS|jS)Nr mpl_ge_3_5_0x)"pandas.plotting._matplotlib.compatr _shared_axes_shared_x_axesrr'rrrr get_x_axis  zTestPlotBase.get_x_axiscCr)Nrry)rrr_shared_y_axesrrrr get_y_axisrzTestPlotBase.get_y_axis)NT)T)r$rJrKrLrMrN)NNN)NNNN)rr)rr)rj)__name__ __module__ __qualname____doc__rrrrrr(r4rBrIrQrZrmr"rrrrr rrrrrrrrrrr s>       B 0 $  < ) r alwaysFc Ksddlm}|r t}nt}d}tft|zPz:|d|}| |||fi|D]}t |q0t j dd }| |Wdn1sNwYWntyb} z| d} ~ wwWt |nt |w|WdS1s|wYdS)aT Create plot and ensure that plot return object is valid. Parameters ---------- f : func Plotting function. filterwarnings : str Warnings filter. See https://docs.python.org/3/library/warnings.html#warning-filter 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_subplotswarningscatch_warnings simplefiltergetgcfclfr"assert_is_valid_plot_return_object ensure_cleansavefig Exceptionr) ffilterwarnings default_axeskwargsr gen_plotsretfigpatherrrrr_check_plot_works s0     $rcks|di|VdS)z' Create plot in a default way. NrrrrrrrrrEsrcks`d|vr |d|di|V|tjjurd|vsJn|d|d<|di|VdS)z9 Create plot on two subplots forcefully created. r'Nr) add_subplotpdplottingbootstrap_plotrrrrrLs  r)rF)r __future__rtypingrrrnumpyrdpandas.util._decoratorsrpandas.util._test_decoratorsutil_test_decoratorstdpandas.core.dtypes.apirpandasrrpandas._testing_testingrrr skip_if_no_mplr rrrrrrrs*       q: