w:iA8SrSSKJr SSKJr SSKrSSKJr SSK r SSK J r SSK J r \(a SSKJr SSKJr SS jrSS jrS rSS jrSS S jjrSrS!SjrSrS"SjrS#SjrS!SjrS$SjrS%SjrSSjr 04Sjr!S&Sjr"Sr#Sr$S'Sjr%Sr&Sr'g)(zF Module consolidating common testing functions for checking plotting. ) annotations) TYPE_CHECKINGN) is_list_like)Series)SequenceAxescU(aUc [S5e[U5nUHZnU(a=UR5ce[UR5R 5U5 MGUR5cMZe g)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_labels get_texts)axeslabelsvisibleaxs _/srv/projetos/modelo_ml_acdoc/venv/lib/python3.13/site-packages/pandas/tests/plotting/common.py_check_legend_labelsrskFNHII D !D ==?. .. r}}88:F C==?* ** cU(aUc [S5eU(a9UR5up4UVs/sHoUR5PM nnXa:XdegUR5begs snf)z 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 True)r get_legend_handles_labels get_markerr )rexpected_markersrhandles_handlemarkerss r_check_legend_markerr/sj$,IJJ113 5<=W6$$&W=***}}&&&>sA*c:SSKJn UR5nUR5n[U5[U5:Xde[ X45H;upVUR 5nUR 5n[ R"Xx5 M= URS5 g)zt Check each axes has identical lines Parameters ---------- xp : matplotlib Axes object rs : matplotlib Axes object rNall) matplotlib.pyplotpyplot get_lineslenzip get_xydatatmassert_almost_equalclose) xprspltxp_linesrs_linesxplrslxpdatarsdatas r _check_datar4Fsx$||~H||~H x=CM )) )+!! v., IIercSSKJn [X5(d[U5(dU/nUHnUR 5U:XaMe g)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.collectionsr6 isinstancer get_visible) collectionsrr6patchs r_check_visibler<]sD2 k . .|K7P7P"m   "g---rcv[U5nUH(nURHnURU:XaMe M* g)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)rfilledrr;s r_check_patches_all_filledrAqs8 D !DZZE::' '' rcUR5n[[X!55nURVs/sHoCUPM sn$s snfN)uniquedictr&values)seriescolorsrDmappedvs r_get_colors_mappedrKs< ]]_F#f% &F%}} -}!1I} -- -sAc|SSKJn SSKJnJnJn SSKJn URn UbUb[X15nUS[U5n[U5[U5:Xde[X5Hup[X5(a"U R5n U RU 5n O?[XU45(a[U R!5S5n OU R!5n U RU 5n X:XaMe UbUb[X25nUS[U5n[U5[U5:Xde[X5H{up[X5(aU R#5Sn OU R#5n [U [$R&5(a [U 5n U RU 5n X:XaM{e gg)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)rH)r6LineCollectionPolyCollection)Line2DN) matplotlibrHr7r6rMrNmatplotlib.linesrOColorConverterrKr%r&r8 get_colorto_rgbatuple get_edgecolor get_facecolornpndarray)r: linecolors facecolorsmappingrHr6rMrNrOconvr;colorresultexpecteds r _check_colorsras " (  D  +G@J#$6c+&67J;3z?222 8LE%((*f-EN#CDDu224Q78,,.||E*H% %%9  +G@J#$6c+&67J;3z?222 8LE%,,,,.q1,,.&"**--v||E*H% %%9 rc[U5(dUR5U:XdegUVs/sHo"R5PM nn[U5[U5:Xde[X15H upEXE:XaM e gs snf)z 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 N)rget_textr%r&)textsr`trlabeles rrrsr   ~~8+++(-.1**,.6{c(m+++F-HE: :./sA=c,SSKJn [U5nUGH|nUcUb[URR 5U5(aUR 5nO UR 5UR SS9-nUHUnUb%[R"UR5U5 UcM0[R"UR5U5 MW UcUcM[URR 5U5(aUR5nO UR5URSS9-nUHUnUb%[R"UR5U5 UcM0[R"UR5U5 MW GM g)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.tickerrir r8xaxisget_minor_formatterget_xticklabelsr(r) get_fontsize get_rotationyaxisget_yticklabels) r xlabelsizexrot ylabelsizeyrotrirrrfs r_check_ticks_propsrwsW 0 D !D  !T%5"((668-HH++-++-0B0B0B0NN)**5+=+=+?L#**5+=+=+?F   !T%5"((668-HH++-++-0B0B0B0NN)**5+=+=+?L#**5+=+=+?F  +rc[U5nUHCnURR5U:XdeURR5U:XaMCe g)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 rl get_scalerq)rrlrqrs r_check_ax_scalesrz sN D !Dxx!!#u,,,xx!!#u,,,rclSSKJn UcSn[U5nUb9[U5U:XdeUH"n[UR 55S:aM"e Ub[ 5n[ 5nU"U5HOnUR 5R5n URU SS5 URU SS5 MQ [U5[U54n X:Xde[R"USRR5[R"U[RS95 g)an 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.toolsr}r r% get_childrenset get_position get_pointsaddr(assert_numpy_array_equalfigureget_size_inchesrXarrayfloat64) raxes_numlayoutfigsizer} visible_axesrx_sety_setpointsr_s r_check_axes_shapers?#D)L< H,,,Br()A- --t$B__&113F IIfQil # IIfQil # % e*c%j)Q..0  +rc|SSKJn U"U5nUVs/sHo3R5(dMUPM nnU$s snf)zo Flatten axes, and filter only visible Parameters ---------- axes : matplotlib Axes object, or its list-like rr|)rr}r9)rr} axes_ndarrayrs rr r Ks6?%L% :2)9BD : K ;s99c[U5nUH^nURnSnSnUH7n[USS5n[USS5n U(aUS- nU (dM2US- nM9 X:XdeX&:XaM^e g)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_yerrr~N)r containersgetattr) rxerryerrrr xerr_count yerr_countcrrs r_check_has_errorbarsr[s D !D]]   Aq*e4Hq*e4Ha xa !!!!!!rcSSKJn [U[S.nUc`UcSn[ XU5(deUS:Xa?[ UR U5(de[ UR [5(deggUc%[U5Hn[ Xd5(aMe g[ U[5(de[UR55[U5:XdeUR5Hupx[ XU5(deUS:Xa!U(aUR5U:XdeM?MAUS:XagU(a UR R5U:Xde[ UR U5(de[ UR [5(deMUS:Xa5USSn U Rn U(aU R5U:XdeMM[e g) 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)rErbothNrErrmedians)matplotlib.axesr rErUr8rlinesr rsortedkeysitems get_titlerAssertionError) returned return_type expected_keyscheck_ax_titler typesrkeyvalueliners r_check_box_return_typerws&%4 7E   K(+$67777 & hkk400 00hnnd33 33 !  %h/!!****0 (F++++hmmo&&*????"..*JCe;%788 88f$! ??,333"&! 88--/3666!%((D1111!%++t4444&Y'*yy!>>+s222"%$#+rc^SSKmU4SjnSnUGHnTRRSS[U5-U5 US- nTR SSS9 UR "S SU0UD6 U"5(aeTRR 5 TRRSS[U5-U5 US- nTR SS S9 UR "S USS .UD6 U"5(aeTRR 5 US ;dGMTRRSS[U5-U5 US- nTR SS S9 UR "S SU0UD6 U"5(deTRR 5 TRRSS[U5-U5 US- nTR SSS9 UR "S US S .UD6 U"5(deTRR 5 GM g) Nrc4>TRR5RR5nTRR5RR5n[ SU55n[ SU55nU=(a U(+$)Nc3^# UH#oRR5(+v M% g7frCgridliner9.0gs r ;_check_grid_settings..is_grid_on..!@Azz--///+-c3^# UH#oRR5(+v M% g7frCrrs rrrrr)r#gcarlget_major_ticksrqr!)xticksyticksxoffyoffmpls r is_grid_on(_check_grid_settings..is_grid_onsn!''779!''779@@@@@@MT""rr~rF)gridkindT)rr)piehexbinscatter)rPr#subplotr%rcplotclf)objkindskwsrspndxrrs @r_check_grid_settingsrs# E 1a#e*ne4   vE" "d"c"<<  1a#e*ne4   vD! .d.#.<<  3 3 JJ  q!c%j.% 8 QJE FF6F % HH &$ &# &<< < JJNN  JJ  q!c%j.% 8 QJE FF6F & HH 1$T 1S 1<< < JJNN 9rc<USVs/sHo"UPM sn$s snf)zD Auxiliary function for correctly unpacking cycler after MPL >= 1.5 zaxes.prop_cycler)rcParamsfieldrJs r_unpack_cyclerrs&''89 :9eH9 :: :sc URS$)Nx _shared_axesrs r get_x_axisr ??3 rc URS$)Nyrrs r get_y_axisrrrc :SSKJn U(a[nO[nSnUR SUR 55nUR 5 U"X40UD6Hn[R"U5 M URU5 U$!URW5 f=f)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) r"r#_gen_default_plot_gen_two_subplotsgetgcfrr("assert_is_valid_plot_return_objectr*)f default_axeskwargsr- gen_plotsretfigs r_check_plot_worksrs6$% % Cjj3779-  Q.v.C  1 1# 6/ # J #s ABBc+ # U"S0UD6v g7f)z Create plot in a default way. Nrrrrrs rrr s +f+s c+# SU;aURS5 U"S0UD6v U[RRLa SU;deOURS5US'U"S0UD6v g7f)z1 Create plot on two subplots forcefully created. rNr) add_subplotpdplottingbootstrap_plotrs rrr'se 6  +f+BKK & &&6!!!s+t +f+sA'A))NT)T)rAxes | Sequence[Axes]r@boolreturnNone)NNN)NNNN)linearr)rrrzSequence[Axes])rr)r^)F)(__doc__ __future__rtypingrnumpyrXpandas.core.dtypes.apirpandasrrpandas._testing_testingr(collections.abcrrr rrr4r<rArKrarrwrzrr rrrrrrrrrrrrr s# /($+0'...((".=&@(,G^-$)X  ":?C:%z*,*Z;  -` r