iDdZddlmZddlmZddlZddlmZddl Z ddl m Z ddl m Zer ddlmZddlmZdd Zdd Zd Zd d Zd d!d ZdZd"dZdZd#dZd$dZd"dZd%dZd&dZ ddZ d'dZ!d(dZ"dZ#dZ$d)dZ%d*dZ&dZ'dZ(y)+zF Module consolidating common testing functions for checking plotting. ) annotations) TYPE_CHECKINGN) is_list_like)Series)SequenceAxesc|r | tdt|}|D]Q}|r;|jJt|jj |@|jQJy)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 q/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/pandas/tests/plotting/common.py_check_legend_labelsrskFNHII D !D ==?. .. r}}88:F C==?* ** c|r | td|r8|j\}}|Dcgc]}|j}}||k(sJy|jJycc}w)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 True)r get_legend_handles_labels get_markerr )rexpected_markersrhandles_handlemarkerss r_check_legend_markerr/sq$,IJJ113 5<=W66$$&W=****}}&&&>sAc|j}|j}t|t|k(sJt||dD];\}}|j}|j}t j ||=y)z 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_datar1Fso||~H||~H x=CM )) )(48S!! vv.9rc|ddlm}t||st|s|g}|D]}|j |k(rJy)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_visibler9Ys@2 k: .|K7P"m   "g---rclt|}|D]$}|jD]}|j|k(rJ&y)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::' '' rc|j}tt||}|jDcgc]}|| c}Scc}wN)uniquedictr%values)seriescolorsrAmappedvs r_get_colors_mappedrH~s@ ]]_F#ff% &F%}} -}!F1I} -- -s AcZddlm}ddlm}m}m}ddlm}|j} ||t||}|dt|}t|t|k(sJt||dD]\} } t| |r"| j} | j| } n;t| ||frt| j!d} n| j!} | j| } | | k(rJ||t||}|dt|}t|t|k(sJt||dD]r\} } t| |r| j#d} n| j#} t| t$j&r t| } | j| } | | k(rrJyy)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  +GZ@J#$6c+&67J;3z?222 ZELE5%(*f-ENN#CDu224Q78,,.||E*HX% %%F  +GZ@J#$6c+&67J;3z?222 ZELE5%,,,.q1,,.&"**-v||E*HX% %%F rct|s|j|k(sJy|Dcgc]}|j}}t|t|k(sJt||dD] \}}||k(r Jycc}w)a 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 rrrsy  ~~8+++(-.1!**,.6{c(m+++FHT:HE1A: :;/sA6cddlm}t|}|D]d}||t|jj |r|j }n#|j |j dz}|D]O}|$tj|j||,tj|j|Q||t|jj |r|j}n#|j|jdz}|D]O}|$tj|j||,tj|j|Qgy)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) matplotlib.tickerrfr r5xaxisget_minor_formatterget_xticklabelsr'r( get_fontsize get_rotationyaxisget_yticklabels) r xlabelsizexrot ylabelsizeyrotrfrrrcs r_check_ticks_propsrtsQ 0 D !D  !T%5"((668-H++-++-0B0B0B0NN)**5+=+=+?L#**5+=+=+?F   !T%5"((668-H++-++-0B0B0B0NN)**5+=+=+?L#**5+=+=+?F  +rct|}|D]@}|jj|k(sJ|jj|k(r@Jy)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,,,rc\ddlm}|d}t|}|5t||k(sJ|D] }t|j dkDr J|t }t }||D]N}|j j} |j| dd|j| ddPt|t|f} | |k(sJtj|djjtj|tjy)a 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  +rclddlm}||}|Dcgc]}|js|}}|Scc}w)z Flatten axes, and filter only visible Parameters ---------- axes : matplotlib Axes object, or its list-like rry)r}rzr6)rrz axes_ndarrayrs rr r Gs8?%L% :2)9BD : K ;s11ct|}|D]P}|j}d}d}|D]+}t|dd}t|dd} |r|dz }| s'|dz }-||k(sJ||k(rPJy)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 a z!!!z!!!rc6ddlm}t|td}|P|d}t |||sJ|dk(r5t |j |sJt |j tsJyy|t|D]}t ||rJyt |tsJt|jt|k(sJ|jD]\}}t |||sJ|dk(r|s|j|k(r3J|dk(rV|r|j j|k(sJt |j |sJt |j trJ|dk(r-|dd} | j} |s| j|k(rJty) 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(E+$6777 & hkk40 00hnnd3 33 !  %h/!!T***0 (F+++hmmo&&*????"..*JCeU;%78 88f$! ??,333&! 88--/3666!%((D111!%++t444&Y'*yy!>>+s222$$#+rcddlfd}|i}d}|D]}jjddt|z||dz }j dd|j d d|i||rJjj jjddt|z||dz }j dd |j d |dd ||rJjj |d vsjjddt|z||dz }j dd |j d d|i||sJjj jjddt|z||dz }j dd|j d |d d ||sJjj y) Nrc"jjjj}jjjj}t d|D}t d|D}|xr| S)Nc3RK|]}|jj !ywr@gridliner6.0gs r z;_check_grid_settings..is_grid_on..!@Aqzz--//%'c3RK|]}|jj !ywr@rrs rrz;_check_grid_settings..is_grid_on..rr)pyplotgcariget_major_ticksrnall)xticksyticksxoffyoffmpls r is_grid_onz(_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 % CHH &$ &# &< < JJNN  JJ  q!c%j.% 8 QJE FF6F & CHH 1$T 1S 1< < JJNN 9rc8|dDcgc]}|| c}Scc}w)zL Auxiliary function for correctly unpacking cycler after MPL >= 1.5 zaxes.prop_cycler)rcParamsfieldrGs r_unpack_cyclerrs(''89 :9AeH9 :: :s c |jdS)Nx _shared_axesrs r get_x_axisr ??3 rc |jdS)Nyrrs r get_y_axisrrrcddlm}ddlm}t |t t jfrdt |t r |j}|jdD]3}dt|j}t ||tfr.J|ydt|j}t ||ttfsJ|y)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