+ NV-jDHRt^RIHt^RIHt^RIt^RIHt^RI t ^RI H t ^RI H t]'d ^RIHt^RIHtR!RltR!R ltR tR"R ltR"R R lltRtR#RltRtR$RltR%RltR#RltRRltR&RltR!Rlt R'Rlt!R(Rlt"Rt#Rt$RRlt%R)Rlt&Rt'R t(R#)*zF Module consolidating common testing functions for checking plotting. ) annotations) TYPE_CHECKINGN) is_list_like)Series)SequenceAxescV'dVf \R4h\V4pVF]pV'd?VP4fQh\VP4P 4V4KIVP4fK]Qh R#)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&&& m/Volumes/fast/ai/experiments/ui-tars-smoke/.venv/lib/python3.14/site-packages/pandas/tests/plotting/common.py_check_legend_labelsrskFNHII D !D ==?. .. r}}88:F C==?* ** cV'dVf \R4hV'd;VP4wr4VUu.uFqUP4NK ppWa8XgQhR#VP4eQhR#uupi)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=***}}&&&>sA0cVP4pVP4p\V4\V48XgQh\W#RR7F;wrEVP4pVP4p\P !Wg4K= R#)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_datar0Fsi||~H||~H x=CM )) )48!! v.9rc^RIHp\W4'g\V4'gV.pVFpVP 4V8XdKQh R#)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 ) CollectionN)matplotlib.collectionsr2 isinstancer get_visible) collectionsrr2patchs&& r_check_visibler8YsD2 k . .|K7P7P"m   "g---rc$V^8dQhRRRRRR/#)rAxes | Sequence[Axes]filledboolreturnNone)formats"r __annotate__rBms#(($9(4(SW(rcz\V4pVF)pVPFpVPV8XdKQh K+ R#)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<rr7s&& r_check_patches_all_filledrFms8 D !DZZE::' '' rcVP4p\\W!44pVPUu.uF qCV,NK up#uupiN)uniquedictr$values)seriescolorsrImappedvs&& r_get_colors_mappedrP~s> ]]_F#f% &F%}} -}!1II} -- -sAc^RIHp^RIHpHpHp^RIHpVPp VeVe\W14pVR\V4p\V4\V48XgQh\WRR7Fwr\W4'd#V P4p V PV 4p MD\WV34'd"\V P!4^,4p MV P!4p V PV 4p W8XdKQh VeVe\W24pVR\V4p\V4\V48XgQh\WRR7Fwr\W4'dV P#4^,p MV P#4p \V \$P&4'd \V 4p V PV 4p W8XdKQh R#R#)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 )rM)r2LineCollectionPolyCollection)Line2DNTr ) matplotlibrMr3r2rRrSmatplotlib.linesrTColorConverterrPr#r$r4 get_colorto_rgbatuple get_edgecolor get_facecolornpndarray)r6 linecolors facecolorsmappingrMr2rRrSrTconvr7colorresultexpecteds&&&& r _check_colorsrfs " (  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\V4'gVP4V8XgQhR#VUu.uFq"P4NK pp\V4\V48XgQh\W1RR7F wrEWE8XdK Qh R#uupi)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$)textsretrlabeles&& rr r st   ~~8+++(-.1**,.6{c(m+++FT:HE: :;/sBc^^RIHp\V4pVEFpVfVe\VPP 4V4'dVP 4pM'VP 4VP RR7,pVFWpVe&\P!VP4V4VfK2\P!VP4V4KY VfVfK\VPP 4V4'dVP4pM'VP4VPRR7,pVFWpVe&\P!VP4V4VfK2\P!VP4V4KY EK R#)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 ) NullFormatterNT)minor) matplotlib.tickerrnr r4xaxisget_minor_formatterget_xticklabelsr&r' get_fontsize get_rotationyaxisget_yticklabels) r xlabelsizexrot ylabelsizeyrotrnrrrks &&&&& r_check_ticks_propsr|sW 0 D !D  !T%5"((668-HH++-++-0B0B0B0NN)**5+=+=+?L#**5+=+=+?F   !T%5"((668-HH++-++-0B0B0B0NN)**5+=+=+?L#**5+=+=+?F  +rc\V4pVFEpVPP4V8XgQhVPP4V8XdKEQh R#)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 rq get_scalerv)rrqrvrs&&& r_check_ax_scalesr sN D !Dxx!!#u,,,xx!!#u,,,rc^RIHpVfRp\V4pVe<\V4V8XgQhVF#p\VP 44^8dK#Qh Ve\ 4p\ 4pV!V4F_pVP 4P4p VPV ^,^,4VPV ^,^,4Ka \V4\V43p W8XgQh\P!V^,PP4\P!V\PR74R#)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  flatten_axesN)dtype)g@g333333@)!pandas.plotting._matplotlib.toolsrr r# get_childrenset get_position get_pointsaddr&assert_numpy_array_equalfigureget_size_inchesr]arrayfloat64) raxes_numlayoutfigsizer visible_axesrx_sety_setpointsrds &&&& r_check_axes_shapers?#D)L< H,,,Br()A- --t$B__&113F IIfQil # IIfQil # % e*c%j)Q..0  +rc V^8dQhRRRR/#)r:rr;r>zSequence[Axes]r@)rAs"rrBrBGs  0 ^ rc|^RIHpV!V4pVUu.uFq3P4'gKVNK ppV#uupi)zo Flatten axes, and filter only visible Parameters ---------- axes : matplotlib Axes object, or its list-like r)rrr5)rr axes_ndarrayrs& rr r Gs8?%L% :2)9BBD : K ;s99c\V4pVFjpVPp^p^pVFAp\VRR4p\VRR4p V'd V^, pV 'gK8V^, pKC W8XgQhW&8XdKjQh R#)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 has_xerrFhas_yerrN)r containersgetattr) rxerryerrrr xerr_count yerr_countcrrs &&& r_check_has_errorbarsrWs D !D]]   Aq*e4Hq*e4Ha xa !!!!!!rc ^RIHpR\RVR\/pVflVfRp\ WV,4'gQhVR8XdC\ VP V4'gQh\ VP \4'gQhR#R#Vf(\V4Fp\ Wd4'dKQh R#\ V\4'gQh\VP44\V48XgQhVP4EFwrx\ WV,4'gQhVR8Xd$V'dVP4V8XgQhKHKJVR8XdlV'd"VP P4V8XgQh\ VP V4'gQh\ VP \4'gQhKVR8XdBVR,^,p V Pp V'dV P4V8XgQhEKEK\h R#)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. rrJrbothNmedians)matplotlib.axesrrJrZr4rlinesr rsortedkeysitems get_titlerAssertionError) returned return_type expected_keyscheck_ax_titlertypesrkeyvalueliners &&&& r_check_box_return_typerss&% T64 7E   K(+$67777 & hkk400 00hnnd33 33 !  %h/!!****0 (F++++hmmo&&*????"..*JCe;%788 88f$! ??,333"&! 88--/3666!%((D1111!%++t4444&Y'*yy!>>+s222"%$#+rca^RIoV3RlpVf/p^pVEF*pSPP^^\V4,V4V^, pSP RRR7VP !R RV/VBV!4'dQhSPP 4SPP^^\V4,V4V^, pSP RRR7VP !R RVRR/VBV!4'dQhSPP 4VR 9gEKSPP^^\V4,V4V^, pSP RRR7VP !R RV/VBV!4'gQhSPP 4SPP^^\V4,V4V^, pSP RRR7VP !R RVRR/VBV!4'gQhSPP 4EK- R#) Nc<SPP4PP4pSPP4PP4p\ ;QJdRV4F 'dK RM RM !RV44p\ ;QJdRV4F 'dK RM RM !RV44pT;'dT'*#)c3^"TF#qPP4'*xK% R#5irHgridliner5.0gs& r ;_check_grid_settings..is_grid_on..!@Azz--///+-FTc3^"TF#qPP4'*xK% R#5irHrrs& rrrrr)pyplotgcarqget_major_ticksrvall)xticksyticksxoffyoffmpls r is_grid_on(_check_grid_settings..is_grid_ons!''779!''779s@@sss@@@s@@sss@@@MMT""rrF)gridkindTrr@)piehexbinscatter)rUrsubplotr#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$ 1T 1S 1<< < JJNN 9rcJVR,Uu.uF q"V,NK up#uupi)zD Auxiliary function for correctly unpacking cycler after MPL >= 1.5 zaxes.prop_cycler@)rcParamsfieldrOs&& r_unpack_cyclerrs(''89 :9eHH9 :: :s c(VPR,#)x _shared_axesrs&r get_x_axisr ??3 rc(VPR,#)yrrs&r get_y_axisrrrcV^8dQhRR/#)r:r>r?r@)rAs"rrBrBs<<rs# /($+0'./&.((".=&@(,G^-$)X  "8:%z,^;  <,)X r