ttjDdZddlmZddlmZddlZddlmZddl Z ddl m Z ddl m Zer ddlmZddlmZd*d Zd*d Zd Zd+d Zd+d,dZdZd-dZdZd.dZd/dZd-dZd0dZd1dZ d*dZ d2d Z!d3d"Z"d#Z#d$Z$d4d%Z%d5d'Z&d(Z'd)Z(dS)6zF Module consolidating common testing functions for checking plotting. ) annotations) TYPE_CHECKINGN) is_list_like)Series)SequenceAxesTc|r|tdt|}|D]e}|rK|Jt||O|JfdS)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 Z/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/pandas/tests/plotting/common.py_check_legend_labelsrsJFNHIII D ! !D++  +==??... r}}88::F C C C C==??**** ++c|r|td|r-|\}}d|D}||ksJdS|JdS)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 Truec6g|]}|S) get_marker).0handles r z(_check_legend_marker..@s$===66$$&&===r)r get_legend_handles_labelsr )rexpected_markersrhandles_markerss r_check_legend_markerr#/sK$,IJJJ'1133 ==W===*******}}&&&&&rcD|}|}t|t|ksJt||dD]B\}}|}|}t j||CdS)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_datar5Fs||~~H||~~H x==CMM ) ) ) )(4888//S!!!! vv....//rcddlm}t||st|s|g}|D]}||ksJdS)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.collectionsr7 isinstancer get_visible) collectionsrr7patchs r_check_visibler=Ysw211111 k: . .$|K7P7P$"m ..  ""g-----..rrAxes | Sequence[Axes]filledboolreturnNonec\t|}|D]}|jD]}|j|ksJdS)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?rr<s r_check_patches_all_filledrFmsV D ! !D((Z ( (E:''''' (((rc|}tt||fd|jDS)Nc g|] }| Srr)rvmappeds rrz&_get_colors_mapped..s - - -!F1I - - -r)uniquedictr)values)seriescolorsrKrJs @r_get_colors_mappedrP~sE ]]__F#ff%% & &F - - - -v} - - --rc.ddlm}ddlm}m}m}ddlm}|j} ||'t||}|dt|}t|t|ksJt||dD]\} } t| |r*| } | | } nNt| ||fr(t| d} n| } | | } | | ksJ||'t||}|dt|}t|t|ksJt||dD]\} } t| |r| d} n| } t| t$jrt| } | | } | | ksJdSdS)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)rO)r7LineCollectionPolyCollection)Line2DNTr%) matplotlibrOr8r7rRrSmatplotlib.linesrTColorConverterrPr(r)r9 get_colorto_rgbatuple get_edgecolor get_facecolornpndarray)r; linecolors facecolorsmappingrOr7rRrSrTconvr<colorresultexpecteds r _check_colorsrfs~ "!!!!! ('''''  D  +GZ@@J#$6c+&6&6$67J;3z??2222 ZEEE & &LE5%(( /**f--ENN#CDD /u2244Q788,,..||E**HX%%%%%  +GZ@@J#$6c+&6&6$67J;3z??2222 ZEEE & &LE5%,, /,,..q1,,..&"*-- 'v||E**HX%%%%%# & &rct|s||ksJdSd|D}t|t|ksJt||dD] \}}||ksJdS)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 c6g|]}|Sr)get_text)rts rrz&_check_text_labels..s ...1!**,,...rTr%N)rrir(r))textsrerlabeles rrrs   ~~8++++++.....6{{c(mm++++FHT:::  HE1A:::::  rcFddlm}t|}|D]}||t|j|r|}n+||dz}|D]T}|'tj| ||'tj| |U||t|j |r| }n+| | dz}|D]T}|'tj| ||'tj| |UdS)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.tickerror r9xaxisget_minor_formatterget_xticklabelsr+r, get_fontsize get_rotationyaxisget_yticklabels) r xlabelsizexrot ylabelsizeyrotrorrrls r_check_ticks_propsr}s 0///// D ! !DGG  !T%5"(6688-HH O++--++--0B0B0B0N0NN G G)*5+=+=+?+?LLL#*5+=+=+?+?FFF  !T%5"(6688-HH O++--++--0B0B0B0N0NN G G)*5+=+=+?+?LLL#*5+=+=+?+?FFF3GGrlinearct|}|D]@}|j|ksJ|j|ksJAdS)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 rr get_scalerw)rrrrwrs r_check_ax_scalesr sk D ! !D--x!!##u,,,,x!!##u,,,,,--rcddlm}|d}t|}|At||ksJ|D])}t|dksJ*|t }t }||D]j}|} || dd|| ddkt|t|f} | |ksJtj |dj tj|tjdS)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.toolsrr r( get_childrenset get_position get_pointsaddr+assert_numpy_array_equalfigureget_size_inchesr]arrayfloat64) raxes_numlayoutfigsizer visible_axesrx_sety_setpointsrds r_check_axes_shaperst?>>>>>#D))L<  H,,,, . .Br(())A----- ,t$$ $ $B__&&1133F IIfQil # # # IIfQil # # # #e**c%jj)Q..00  +++rSequence[Axes]c@ddlm}||}d|D}|S)z Flatten axes, and filter only visible Parameters ---------- axes : matplotlib Axes object, or its list-like rrc:g|]}||Sr)r:)rrs rrz$_flatten_visible..Ss' : : :2)9)9 :B : : :r)rr)rr axes_ndarrays rr r Gs=?>>>>><%%L : : : : :D Krct|}|D]R}|j}d}d}|D]2}t|dd}t|dd} |r|dz }| r|dz }3||ksJ||ksJSdS)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_yerrrN)r containersgetattr) rxerryerrrr xerr_count yerr_countcrrs r_check_has_errorbarsrWs D ! !D " "]     Aq*e44Hq*e44H a  a z!!!!z!!!!! " "rcddlm}t|td}|Y|d}t |||sJ|dkr3t |j|sJt |jtsJdSdS|&t|D]}t ||sJdSt |tsJt| t|ksJ| D]\}}t |||sJ|dkr|r| |ksJ@|dkrU|r|j |ksJt |j|sJt |jtsJ|dkr2|dd} | j } |r| |ksJtdS) 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)rLrbothNrLrrmedians)matplotlib.axesr rLrZr9rlinesr rsortedkeysitems get_titlerAssertionError) returned return_type expected_keyscheck_ax_titler typesrkeyvalueliners r_check_box_return_typerss7&%$$$$$4 7 7E   K(E+$677777 & hk400 0 00hnd33 3 33 ! 3 3  %h// + +!!T****** F(F+++++hmmoo&&&*?*?????"..** % %JCeU;%788 8 88f$$!4 ??,,3333&&!7 8--//36666!%(D11111!%+t444444&&Y'*y!3>>++s2222$$# % %rc ddlfd}|i}d}|D]}jddt|z||dz }dd|jd d|i||rJjjddt|z||dz }dd |jd |dd ||rJj|d vrjddt|z||dz }dd |jd d|i||sJjjddt|z||dz }dd|jd |d d ||sJjdS) Nrc2jj}jj}t d|D}t d|D}|o| S)Nc3JK|]}|j VdSNgridliner:rgs r z;_check_grid_settings..is_grid_on..3@@Aqz--///@@@@@@rc3JK|]}|j VdSrrrs rrz;_check_grid_settings..is_grid_on..rr)pyplotgcarrget_major_ticksrwall)xticksyticksxoffyoffmpls r is_grid_onz(_check_grid_settings..is_grid_ons!!'7799!!'7799@@@@@@@@@@@@@@MT""rrrF)gridkindT)rr)piehexbinscatterr)rUrsubplotr(rcplotclf)objkindskwsrspndxrrs @r_check_grid_settingsrss##### { E 1a#e**ne444   vE"""""d"c""":<<  1a#e**ne444   vD!!!.d..#...:<<  3 3 3 J  q!c%jj.% 8 8 8 QJE FF6F % % % CH & &$ &# & & &:<<  < JNN    J  q!c%jj.% 8 8 8 QJE FF6F & & & CH 1$T 1 1S 1 1 1:<<  < JNN   9rrcc,fd|dDS)zL Auxiliary function for correctly unpacking cycler after MPL >= 1.5 c g|] }| Srr)rrIfields rrz"_unpack_cycler..s : : :AeH : : :rzaxes.prop_cycler)rcParamsrs `r_unpack_cyclerrs% ; : : :h'89 : : ::rc|jdS)Nx _shared_axesrs r get_x_axisr ?3 rc|jdS)Nyrrs r get_y_axisrrrcddlm}ddlm}t |t t jfrnt |t r|j}| dD]:}dt|j }t ||tfs J|;dSdt|j }t ||ttfs J|dS)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 r9rr]r^_valuesreshapetype__name__rLrZ)objsrr elmsgs r"assert_is_valid_plot_return_objectrs(((((($$$$$$$,--< dF # # r?r@rArB)NNN)NNNN)r~r~)rr>rAr)rrr)rc)rArB)F))__doc__ __future__rtypingrnumpyr]pandas.core.dtypes.apirpandasrrpandas._testing_testingr+collections.abcrrr rr#r5r=rFrPrfrr}rrr rrrrrrrr rrrrrrsc#""""" //////%(((((($$$$$$++++0''''.///&....(((((("...=&=&=&=&@(,G,G,G,G^----$))))X     """":?C:%:%:%:%z,,,,^;;;;      <<<<,))))X     r