a nÝEb3Uã@sÔdZddlmZddlZddlmZmZddlZddlZ ddl m Z ddl m mZddlmZddlZddlmZmZmZddlmZeršddlmZejGd d „d ƒƒZdd d„Zdd„Zdd„Z dd„Z!dS)z² Module consolidating common testing functions for checking plotting. Currently all plotting tests are marked as slow via ``pytestmark = pytest.mark.slow`` at the module level. é)Ú annotationsN)Ú TYPE_CHECKINGÚSequence)Úcache_readonly)Ú is_list_like)Ú DataFrameÚSeriesÚ to_datetime©ÚAxesc@sôeZdZdZdd„Zdd„Zedd„ƒZedd „ƒZd9d d „Z d:dd„Z dd„Z d;dd„Z dd d!„Zd?d#d$„Zd@d%d&„Zd'd(„Zd)d*„ZdAd,d-„ZdBd.d/„Zifd0d1„ZdCd3d4„Zd5d6„Zd7d8„Zd S)DÚ TestPlotBasezE This is a common base class used for various plotting tests cCshddl}ddlm}||_| ¡d|_d|_| ¡|_| ¡|_| ¡|_|  ¡|_ d|_ d|_ d|_ d|_ d }t d ¡’tjjd d g|d }tjjgd¢|d }t||tjjdd|d tjjdd|d tjjd|d ttjj|j|j|tjdƒdœƒ|_Wdƒn1s0Yt ¡|_ttjjdd tjjdd t d¡tjjdd dœƒ|_dS)Nr)ÚcompatljÕ+2ZlÊRŠeöZéé)gš™™™™™@g333333@Úleftédé*ZMaleZFemale)Úsize)ÚAÚBÚCéBéé¡é )rÚdtype)ÚgenderÚ classroomÚheightÚweightÚcategoryÚdatetimeé)Ú matplotlibZpandas.plotting._matplotlibr Z rcdefaultsZstart_date_to_int64Zend_date_to_int64Z mpl_ge_2_2_3Z mpl_ge_3_0_0Ú mpl_ge_3_1_0Z mpl_ge_3_2_0Z bp_n_objectsZpolycollection_factorÚdefault_figsizeZdefault_tick_positionÚtmZ RNGContextÚnpÚrandomÚchoicerÚnormalÚrandintr Zint64Zhist_dfZmakeTimeDataFrameZtdfÚuniformZarangeZ hexbin_df)ÚselfÚmethodÚmplr Únrr©r1ú\/home/ja/django-apps/lartica_env/lib/python3.9/site-packages/pandas/tests/plotting/common.pyÚ setup_method*sR      üÿúÿ&   ýÿzTestPlotBase.setup_methodcCs t ¡dS©N)r&Úclose)r-r.r1r1r2Úteardown_methodaszTestPlotBase.teardown_methodcCsddlm}|S©Nr)Úmatplotlib.pyplotÚpyplot)r-Úpltr1r1r2r:ds zTestPlotBase.pltcCsddlm}|jSr7)Zmatplotlib.colorsÚcolorsZcolorConverter)r-r;r1r1r2Úcolorconverterjs zTestPlotBase.colorconverterNTcCsf|r|durtdƒ‚| |¡}|D]>}|rP| ¡dus:J‚| | ¡ ¡|¡q"| ¡dus"J‚q"dS)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_labelsZ get_texts)r-ÚaxesÚlabelsÚvisibleÚaxr1r1r2Ú_check_legend_labelsps  z!TestPlotBase._check_legend_labelscCsT|r|durtdƒ‚|r@| ¡\}}dd„|Dƒ}||ksPJ‚n| ¡dusPJ‚dS)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 TruecSsg|] }| ¡‘qSr1)Z get_marker)Ú.0Úhandler1r1r2Ú ˜óz5TestPlotBase._check_legend_marker..)r=Zget_legend_handles_labelsr?)r-rDZexpected_markersrCZhandlesÚ_Úmarkersr1r1r2Ú_check_legend_marker‡s  z!TestPlotBase._check_legend_markercsP| ¡}| ¡}dd„‰t|ƒt|ƒks,J‚‡fdd„t||ƒDƒt ¡dS)z¤ Check each axes has identical lines Parameters ---------- xp : matplotlib Axes object rs : matplotlib Axes object cSs | ¡}| ¡}t ||¡dSr4)Z get_xydatar&Úassert_almost_equal)ÚxplÚrslZxpdataZrsdatar1r1r2Ú check_line©sz,TestPlotBase._check_data..check_linecsg|]\}}ˆ||ƒ‘qSr1r1)rFrNrO©rPr1r2rH¯rIz,TestPlotBase._check_data..N)Ú get_linesÚlenÚzipr&r5)r-ZxpÚrsZxp_linesZrs_linesr1rQr2Ú _check_datas zTestPlotBase._check_datacCsBddlm}t||ƒs$t|ƒs$|g}|D]}| ¡|ks(J‚q(dS)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.collectionsrWÚ isinstancerÚ get_visible)r-Ú collectionsrCrWÚpatchr1r1r2Ú_check_visible²s zTestPlotBase._check_visiblezAxes | Sequence[Axes]ÚboolÚNone)rAÚfilledÚreturncCs2| |¡}|D]}|jD]}|j|ksJ‚qqdS)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>ZpatchesÚfill)r-rAr`rDr\r1r1r2Ú_check_patches_all_filledÅs  z&TestPlotBase._check_patches_all_filledcs*| ¡}tt||ƒƒ‰‡fdd„|jDƒS)Ncsg|] }ˆ|‘qSr1r1©rFÚv©Zmappedr1r2rHÜrIz3TestPlotBase._get_colors_mapped..)ÚuniqueÚdictrTÚvalues)r-Zseriesr;rgr1rfr2Ú_get_colors_mapped×szTestPlotBase._get_colors_mappedcCs†ddlm}m}m}ddlm}|j} |durÖ|durR| ||¡}|dt|ƒ…}t|ƒt|ƒksfJ‚t ||ƒD]d\} } t | |ƒr–|   ¡} |   | ¡} n(t | ||fƒr¶t |  ¡dƒ} n|  ¡} |   | ¡} | | kspJ‚qp|dur‚|dur| ||¡}|dt|ƒ…}t|ƒt|ƒksJ‚t ||ƒD]Z\} } t | |ƒrH|  ¡d} n|  ¡} t | tjƒrft | ƒ} |   | ¡} | | ks&J‚q&dS)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)rWÚLineCollectionÚPolyCollection)ÚLine2DN)rXrWrkrlZmatplotlib.linesrmr<rjrSrTrYÚ get_colorZto_rgbaÚtupleZ get_edgecolorZ get_facecolorr'Zndarray)r-r[Z linecolorsZ facecolorsÚmappingrWrkrlrmÚconvr\ÚcolorÚresultÚexpectedr1r1r2Ú _check_colorsÞs<          zTestPlotBase._check_colorscCs`t|ƒs| ¡|ks\J‚nBdd„|Dƒ}t|ƒt|ƒks.N)rrvrSrT)r-ZtextsrtrBÚlabelÚer1r1r2r@ s zTestPlotBase._check_text_labelsc Csddlm}| |¡}|D]ô}|dus.|dur–t|j ¡|ƒrH| ¡}n| ¡|jdd}|D]4} |dur|t |   ¡|¡|dur`t |   ¡|¡q`|dus¦|durt|j  ¡|ƒrÀ|  ¡}n|  ¡|j dd}|D]4} |durôt |   ¡|¡|durØt |   ¡|¡qØqdS)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) Zmatplotlib.tickerrzr>rYÚxaxisZget_minor_formatterZget_xticklabelsr&rMZ get_fontsizeZ get_rotationÚyaxisZget_yticklabels) r-rAZ xlabelsizeZxrotZ ylabelsizeZyrotrzrDrBrxr1r1r2Ú_check_ticks_props3s*    zTestPlotBase._check_ticks_propsÚlinearcCs<| |¡}|D](}|j ¡|ks$J‚|j ¡|ksJ‚qdS)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>r|Z get_scaler})r-rAr|r}rDr1r1r2Ú_check_ax_scalescs zTestPlotBase._check_ax_scalesc Cs¢ddlm}|dur|j}| |¡}|durZt|ƒ|ksrSZ get_childrenÚ_get_axes_layoutr&Zassert_numpy_array_equalÚfigureZget_size_inchesr'ÚarrayZfloat64) r-rAZaxes_numZlayoutZfigsizer‚Z visible_axesrDrsr1r1r2Ú_check_axes_shapets    þzTestPlotBase._check_axes_shapecCsVtƒ}tƒ}|D]4}| ¡ ¡}| |dd¡| |dd¡qt|ƒt|ƒfS)Nré)ÚsetÚ get_positionZ get_pointsÚaddrS)r-rAZx_setZy_setrDZpointsr1r1r2r„˜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|‘qSr1)rZ)rFrDr1r1r2rH®rIz1TestPlotBase._flatten_visible..)rƒr‚)r-rAr‚r1r1r2r>¢s zTestPlotBase._flatten_visiblerc Csx| |¡}|D]d}|j}d}d}|D]4}t|ddƒ} t|ddƒ} | rL|d7}| r$|d7}q$||ksfJ‚||ksJ‚qdS)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 rÚhas_xerrFÚhas_yerrrˆN)r>Ú containersÚgetattr) r-rAZxerrZyerrrDrŽZ xerr_countZ yerr_countÚcrŒrr1r1r2Ú_check_has_errorbars±s     z!TestPlotBase._check_has_errorbarsc Cs„ddlm}t|tdœ}|durj|dur,d}t|||ƒs>J‚|dkrft|j|ƒsVJ‚t|jtƒsfJ‚n|dur”| |¡D]}t||ƒs|J‚q|dSt|tƒs¢J‚t |  ¡ƒt |ƒksºJ‚|  ¡D]¼\}} t| ||ƒsÜJ‚|dkrú|rø|   ¡|ksøJ‚qÂ|dkrD|r| j  ¡|ksJ‚t| j|ƒs0J‚t| jtƒs~J‚qÂ|dkrz| dd} | j } |r~|   ¡|ks~J‚qÂt‚qÂdS) 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 )rhrAÚbothNrhr’rAZmedians)Úmatplotlib.axesr rhrorYrDÚlinesr>rÚsortedÚkeysÚitemsZ get_titlerAÚAssertionError) r-ÚreturnedÚ return_typeZ expected_keysZcheck_ax_titler ÚtypesÚrÚkeyÚvalueÚlinerAr1r1r2Ú_check_box_return_typeÌs>     z#TestPlotBase._check_box_return_typecsfddl}‡fdd„}d}|D]B}ˆj ddt|ƒ|¡|d7}|jddd|jfd |i|¤Ž|ƒrnJ‚ˆj ddt|ƒ|¡|d7}|jdd d|jf|dd œ|¤Ž|ƒr¼J‚|d vrˆj ddt|ƒ|¡|d7}|jdd d|jfd |i|¤Ž|ƒsJ‚ˆj ddt|ƒ|¡|d7}|jddd|jf|d d œ|¤Ž|ƒsJ‚qdS) Nrcszˆj ¡j ¡}ˆj ¡j ¡}ˆjrLtdd„|Dƒƒ}tdd„|Dƒƒ}n$tdd„|Dƒƒ}tdd„|Dƒƒ}|ov| S)Ncss|]}|j ¡ VqdSr4©ZgridlinerZ©rFÚgr1r1r2Ú rIzHTestPlotBase._check_grid_settings..is_grid_on..css|]}|j ¡ VqdSr4r¡r¢r1r1r2r¤rIcss|]}|j VqdSr4©ZgridOnr¢r1r1r2r¤rIcss|]}|j VqdSr4r¥r¢r1r1r2r¤rI)r:Zgcar|Zget_major_ticksr}r$Úall)ZxticksZyticksZxoffZyoff©r-r1r2Ú is_grid_on sz5TestPlotBase._check_grid_settings..is_grid_onrˆrrAF)ÚgridÚkindT)rªr©)ÚpieZhexbinZscatter)r#r:ZsubplotrSÚrcZplot)r-ÚobjÚkindsÚkwsr/r¨Zspndxrªr1r§r2Ú_check_grid_settingss2     z!TestPlotBase._check_grid_settingsrrcs‡fdd„|dDƒS)zT Auxiliary function for correctly unpacking cycler after MPL >= 1.5 csg|] }|ˆ‘qSr1r1rd©Úfieldr1r2rH<rIz/TestPlotBase._unpack_cycler..zaxes.prop_cycler1)r-ZrcParamsr²r1r±r2Ú_unpack_cycler8szTestPlotBase._unpack_cyclercCs|j ¡r|jdS|jS)NÚx)r Ú mpl_ge_3_5_0Ú _shared_axesZ_shared_x_axes©r-rDr1r1r2Ú get_x_axis>szTestPlotBase.get_x_axiscCs|j ¡r|jdS|jS)NÚy)r rµr¶Z_shared_y_axesr·r1r1r2Ú get_y_axisAszTestPlotBase.get_y_axis)NT)NT)T)T)NNN)NNNN)rr)NNN)rr)NT)rr)Ú__name__Ú __module__Ú __qualname__Ú__doc__r3r6rr:r<rErLrVr]rcrjrur@r~r€r‡r„r>r‘r r°r³r¸rºr1r1r1r2r $s<7     ÿÿ Bÿ 0  $  ÿ < 0 r ÚalwaysFc Ksddlm}|rt}nt}d}t ¡Øt |¡z¬zt| d| ¡¡}|  ¡|||fi|¤ŽD]}t   |¡q`t j dd}|  |¡Wdƒn1sœ0YWn(tyÐ} z| ‚WYd} ~ n d} ~ 00Wt  |¡n t  |¡0|WdƒS1s0YdS)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. rNr…T)Zreturn_filelike)r8r9Ú_gen_default_plotÚ_gen_two_subplotsÚwarningsÚcatch_warningsÚ simplefilterÚgetZgcfZclfr&Z"assert_is_valid_plot_return_objectZ ensure_cleanZsavefigÚ Exceptionr5) ÚfÚfilterwarningsZ default_axesÚkwargsr:Z gen_plotsÚretÚfigÚpathÚerrr1r1r2Ú_check_plot_worksEs$    ,rÎcks|fi|¤ŽVdS)z' Create plot in a default way. Nr1©rÇrËrÉr1r1r2rÀsrÀcks^d|vr| d¡|fi|¤ŽV|tjjurs.     $ :