A>iD@SrSSKJr SSKJr SSKrSSKJr SSK r SSK J r SSK J r \(a SSKJr SSKJr SS jrSS jrS rS S jrS S!S jjrSrS"SjrSrS#SjrS$SjrS"SjrS%SjrS&SjrSSjr S'Sjr!S(Sjr"Sr#Sr$S)Sjr%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 [/var/www/html/land-tabula/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 UR5nUR5n[U5[U5:Xde[X#SS9H;upEUR5nUR5n[R "Xg5 M= g)zt Check each axes has identical lines Parameters ---------- xp : matplotlib Axes object rs : matplotlib Axes object TstrictN) get_lineslenzip get_xydatatmassert_almost_equal)xprsxp_linesrs_linesxplrslxpdatarsdatas r _check_datar1Fsi||~H||~H x=CM )) )48!! v.9rcSSKJn [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.collectionsr3 isinstancer get_visible) collectionsrr3patchs r_check_visibler9YsD2 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)rfilledrr8s r_check_patches_all_filledr>ms8 D !DZZE::' '' rcUR5n[[X!55nURVs/sHoCUPM sn$s snfN)uniquedictr%values)seriescolorsrAmappedvs r_get_colors_mappedrH~s< ]]_F#f% &F%}} -}!1I} -- -sAcxSSKJn SSKJnJnJn SSKJn URn UbUb[X15nUS[U5n[U5[U5:Xde[XSS9Hup[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[XSS9H{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)rE)r3LineCollectionPolyCollection)Line2DNTr!) matplotlibrEr4r3rJrKmatplotlib.linesrLColorConverterrHr$r%r5 get_colorto_rgbatuple get_edgecolor get_facecolornpndarray)r7 linecolors facecolorsmappingrEr3rJrKrLconvr8colorresultexpecteds r _check_colorsr^s " (  D  +G@J#$6c+&67J;3z?222 ELE%((*f-EN#CDDu224Q78,,.||E*H% %%F  +G@J#$6c+&67J;3z?222 ELE%,,,,.q1,,.&"**--v||E*H% %%F rc[U5(dUR5U:XdegUVs/sHo"R5PM nn[U5[U5:Xde[X1SS9H 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 Tr!N)rget_textr$r%)textsr]trlabeles rrrst   ~~8+++(-.1**,.6{c(m+++FT: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.tickerrfr r5xaxisget_minor_formatterget_xticklabelsr'r( get_fontsize get_rotationyaxisget_yticklabels) r xlabelsizexrot ylabelsizeyrotrfrrrcs r_check_ticks_propsrtsW 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 ri get_scalern)rrirnrs r_check_ax_scalesrw 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.toolsrzr r$ get_childrenset get_position get_pointsaddr'assert_numpy_array_equalfigureget_size_inchesrUarrayfloat64) raxes_numlayoutfigsizerz 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 rry)r}rzr6)rrz axes_ndarrayrs rr r Gs6?%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_errorbarsrWs 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)rBrbothNrBrrmedians)matplotlib.axesr rBrRr5rlinesr rsortedkeysitems get_titlerAssertionError) returned return_type expected_keyscheck_ax_titler typesrkeyvalueliners r_check_box_return_typerss&%4 7E   K(+$67777 & hkk400 00hnnd33 33 !  %h/!!****0 (F++++hmmo&&*????"..*JCe;%788 88f$! ??,333"&! 88--/3666!%((D1111!%++t4444&Y'*yy!>>+s222"%$#+rc$^SSKmU4SjnUc0nSnUGHnTRRSS[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% g7fr@gridliner6.0gs r ;_check_grid_settings..is_grid_on..!@Azz--///+-c3^# UH#oRR5(+v M% g7fr@rrs rrrrr)pyplotgcariget_major_ticksrnall)xticksyticksxoffyoffmpls r is_grid_on(_check_grid_settings..is_grid_onsn!''779!''779@@@@@@MT""rr{rF)gridkindT)rr)piehexbinscatter)rMrsubplotr$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)rcParamsfieldrGs r_unpack_cyclerrs&''89 :9eH9 :: :sc URS$)Nx _shared_axesrs r get_x_axisr ??3 rc URS$)Nyrrs r get_y_axisrrrcSSKJn SSKJn [ U[ [ R45(ap[ U[ 5(a URnURS5H9nS[U5R<3n[ X2[45(aM4U5e gS[U5R<3n[ X[[45(dU5eg)Nr)ArtistrzBone of 'objs' is not a matplotlib Axes instance, type encountered zoobjs is neither an ndarray of Artist instances nor a single ArtistArtist instance, tuple, or dict, 'objs' is a )matplotlib.artistrrr r5rrUrV_valuesreshapetype__name__rBrR)objsrr elmsgs r"assert_is_valid_plot_return_objectrs($$,-- dF # #<rHr^rrtrwrr rrrrrrrrrrrrrrs# /($+0'./&.((".=&@(,G^-$)X  ":?C:%z,^;  <,)X r