a PfA@sFdZddlmZddlmZddlZddlmZddl Z ddl m Z ddl m ZerlddlmZddlmZd=d d Zd>d d ZddZd?ddZd@ddddddZddZdAddZddZdBddZdCd!d"ZdDd#d$Zdd%d&d'd(ZdEd)d*ZdFd+d,Z ifd-d.Z!dGd0d1Z"d2d3Z#d4d5Z$dHd7d8Z%d9d:Z&d;d<Z'dS)IzF Module consolidating common testing functions for checking plotting. ) annotations) TYPE_CHECKINGN) is_list_like)Series)SequenceAxesTcCsb|r|durtdt|}|D]<}|rL|dus8Jt||q |dus Jq dS)a) 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)axeslabelsvisibleaxrU/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/plotting/common.py_check_legend_labelss rcCsT|r|durtd|r@|\}}dd|D}||ksPJn|dusPJdS)a 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|] }|qSr)Z get_marker).0handlerrr @z(_check_legend_marker..)r Zget_legend_handles_labelsr )rZexpected_markersrZhandles_markersrrr_check_legend_marker/s  rc Csnddlm}|}|}t|t|ks0Jt||D]$\}}|}|}t||q:|ddS)z Check each axes has identical lines Parameters ---------- xp : matplotlib Axes object rs : matplotlib Axes object rNall) matplotlib.pyplotpyplot get_lineslenzipZ get_xydatatmassert_almost_equalclose) ZxprspltZxp_linesZrs_linesZxplrslZxpdataZrsdatarrr _check_dataFs r'cCsBddlm}t||s$t|s$|g}|D]}||ks(Jq(dS)z 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.collectionsr( isinstancer get_visible) collectionsrr(patchrrr_check_visible]s r.zAxes | Sequence[Axes]boolNone)r filledreturncCs0t|}|D]}|jD]}|j|ksJqq dS)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 Zpatchesfill)r r1rr-rrr_check_patches_all_filledqs  r4cs*|}tt||fdd|jDS)Ncsg|] }|qSrrrvZmappedrrrrz&_get_colors_mapped..)uniquedictr values)Zseriescolorsr8rr7r_get_colors_mappedsr<cCsddlm}ddlm}m}m}ddlm}|j} |dur|dur\t ||}|dt |}t |t |kspJt ||D]d\} } t | |r| } | | } n(t | ||frt| d} n| } | | } | | kszJqz|dur|durt ||}|dt |}t |t |ks$Jt ||D]Z\} } t | |rP| d} n| } t | tjrnt| } | | } | | ks.Jq.dS)a 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)r;)r(LineCollectionPolyCollection)Line2DN) matplotlibr;r)r(r=r>Zmatplotlib.linesr?ZColorConverterr<rr r* get_colorZto_rgbatupleZ get_edgecolorZ get_facecolornpZndarray)r,Z linecolorsZ facecolorsmappingr;r(r=r>r?convr-colorresultexpectedrrr _check_colorss>           rIcCs`t|s||ks\JnBdd|D}t|t|ks.N)rrJrr )ZtextsrHrlabelerrrr s r c Csddlm}t|}|D]}|dus,|durt|j|rF|}n||jdd}|D]4}|durzt| ||dur^t| |q^|dus|durt|j |r| }n| |j dd}|D]4}|durt| ||durt| |qqdS)ac 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.tickerrNr r*xaxisZget_minor_formatterZget_xticklabelsr!r"Z get_fontsizeZ get_rotationyaxisZget_yticklabels) r Z xlabelsizeZxrotZ ylabelsizeZyrotrNrrrLrrr_check_ticks_propss*   rRlinearcCs:t|}|D](}|j|ks"J|j|ks Jq dS)z 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 rPZ get_scalerQ)r rPrQrrrr_check_ax_scales s rTc Csddlm}|durd}t|}|durVt||ks8J|D]}t|dks.)rXrV)r rVZ axes_ndarrayrrrr Ks r c Csvt|}|D]d}|j}d}d}|D]4}t|dd}t|dd} |rJ|d7}| r"|d7}q"||ksdJ||ks Jq dS)z 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_yerrrWN)r containersgetattr) r ZxerrZyerrrr`Z xerr_countZ yerr_countcr^r_rrr_check_has_errorbars[s     rcc Csddlm}t|td}|durj|dur,d}t|||s>J|dkrft|j|sVJt|jtsfJn|durt|D]}t||szJqzdSt|tsJt | t |ksJ| D]\}}t|||sJ|dkr|r| |ksJq|dkrB|r|j |ksJt|j|s.Jt|jts|Jq|dkrx|dd} | j } |r|| |ks|JqtqdS) a< 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)r9r bothNr9rdr Zmedians)matplotlib.axesrr9rBr*rlinesr rsortedkeysitemsZ get_titler AssertionError) Zreturned return_typeZ expected_keysZcheck_ax_titlertypesrkeyvalueliner rrr_check_box_return_typews>      rqcsddlfdd}d}|D]l}jddt|||d7}jddd|jfd |i||rnJjjddt|||d7}jdd d|jf|dd ||rJj|d vrjddt|||d7}jdd d|jfd |i||s&Jjjddt|||d7}jddd|jf|d d ||sJjqdS) NrcsNjj}jj}tdd|D}tdd|D}|oJ| S)Ncss|]}|j VqdSNZgridliner+rgrrr rz;_check_grid_settings..is_grid_on..css|]}|j VqdSrrrsrtrrrrvr)rZgcarPZget_major_ticksrQr)ZxticksZyticksZxoffZyoffZmplrr is_grid_ons z(_check_grid_settings..is_grid_onrWr F)gridkindT)r{rz)pieZhexbinZscatter)r@rZsubplotrrcZplotclf)objkindskwsrxZspndxr{rrwr_check_grid_settingss:         rrFcsfdd|dDS)zL Auxiliary function for correctly unpacking cycler after MPL >= 1.5 csg|] }|qSrrr5fieldrrrrz"_unpack_cycler..zaxes.prop_cycler)ZrcParamsrrrr_unpack_cyclersrcCs |jdS)NxZ _shared_axesrrrr get_x_axissrcCs |jdS)Nyrrrrr get_y_axissrFc Ksxddlm}|rt}nt}d}zH|d|}||||fi|D]}t|qJW| |n | |0|S)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. rNr[) rr_gen_default_plot_gen_two_subplotsgetZgcfr~r!Z"assert_is_valid_plot_return_objectr#)fZ default_axeskwargsr%Z gen_plotsretfigrrr_check_plot_workss rcks|fi|VdS)z' Create plot in a default way. Nrrrrrrrr srcks^d|vr|d|fi|V|tjjurs@           @ /  ,  = -  0