dJdZddlmZddlmZmZddlZddlm Z ddl m cm Z ddlmZddlZddlmZddlmZerddlmZe jGd d Zdd Zd ZdZdS)zF Module consolidating common testing functions for checking plotting. ) annotations) TYPE_CHECKINGSequenceN)cache_readonly) is_list_like)SeriesAxesceZdZdZdZdZedZedZd$dZ d$d Z d Z d%d Z d%d&dZ dZ d'dZdZ d(dZd)dZd'dZdZdZd*dZ d$dZifdZd+d!Zd"Zd#ZdS), TestPlotBasezE This is a common base class used for various plotting tests c6ddl}|dSNr) matplotlib rcdefaults)selfmpls X/var/www/html/t/fyr/venv311/lib/python3.11/site-packages/pandas/tests/plotting/common.py setup_methodzTestPlotBase.setup_method!s$     c,tjdSN)tmclose)rs rteardown_methodzTestPlotBase.teardown_method&s  rcddlm}|Sr)matplotlib.pyplotpyplot)rplts rrzTestPlotBase.plt)s'''''' rcddlm}|jS)Nr)colors)rr colorConverter)rr s rcolorconverterzTestPlotBase.colorconverter/s%%%%%%$$rNTc2|r|td||}|D]k}|rQ|J|||U|JldS)aQ 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)raxeslabelsvisibleaxs r_check_legend_labelsz!TestPlotBase._check_legend_labels5s  NLMM M$$T** / /B /}}222'' (A(A(C(CVLLLL}}....  / /rc|r|td|r-|\}}d|D}||ksJdS|JdS)aB 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 z5TestPlotBase._check_legend_marker..]s$AAAvv((**AAAr)r$get_legend_handles_labelsr&)rr,expected_markersr+handles_markerss r_check_legend_markerz!TestPlotBase._check_legend_markerLs  O(0MNN N  +5577JGQAAAAAG.......==??*****rcf|}|}t|t|ksJt||D]B\}}|}|}t j||Ct jdS)z Check each axes has identical lines Parameters ---------- xp : matplotlib Axes object rs : matplotlib Axes object N) get_lineslenzip get_xydatarassert_almost_equalr) rxprsxp_linesrs_linesxplrslxpdatarsdatas r _check_datazTestPlotBase._check_databs<<>><<>>8}}H ----Hh// 3 3HC^^%%F^^%%F  "66 2 2 2 2  rcddlm}t||st|s|g}|D]}||ksJdS)a 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.collectionsrK isinstancer get_visible)r collectionsr+rKpatchs r_check_visiblezTestPlotBase._check_visiblevsw 655555+z22 (< ;T;T (&-K  2 2E$$&&'11111 2 2rr)Axes | Sequence[Axes]filledboolreturnNonech||}|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)rSr,rPs r_check_patches_all_filledz&TestPlotBase._check_patches_all_filleds\$$T** , ,B , ,zV+++++ , , ,rc|}tt||fd|jDS)Nc g|] }| Sr0r0)r2vmappeds rr4z3TestPlotBase._get_colors_mapped..s111aq 111r)uniquedictr>values)rseriesr r_r^s @r_get_colors_mappedzTestPlotBase._get_colors_mappedsEc&&))**11116=1111rc2ddlm}m}m}ddlm}|j} | |-|||}|dt|}t|t|ksJt||D]\} } t| |r*| } | | } nNt| ||fr(t| d} n| } | | } | | ksJ||-|||}|dt|}t|t|ksJt||D]\} } t| |r| d} n| } t| t jrt| } | | } | | ksJdSdS)a3 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)rKLineCollectionPolyCollection)Line2DN)rLrKrerfmatplotlib.linesrgr"rcr=r>rM get_colorto_rgbatuple get_edgecolor get_facecolornpndarray)rrO linecolors facecolorsmappingrKrerfrgconvrPcolorresultexpecteds r _check_colorszTestPlotBase._check_colorssl$          ,+++++"  !"!44WjII '(:#k*:*:(:; {##s:6666 #K < < * * ueV,,3"__..F!\\&11FF'GHH3"5#6#6#8#8#;<.s 222qajjll222rN)rrzr=r>)rtextsrvr*labeles rr'zTestPlotBase._check_text_labelssE"" ">>##x//////22E222Fv;;#h--////11 " "qzzzzz " "rcRddlm}||}|D]}||t|j|r|}n+||dz}|D]T} |'tj| ||'tj| |U||t|j |r| }n+| | dz}|D]T} |'tj| ||'tj| |UdS)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.tickerrr%rMxaxisget_minor_formatterget_xticklabelsrr@ get_fontsize get_rotationyaxisget_yticklabels) rr) xlabelsizexrot ylabelsizeyrotrr,r*r}s r_check_ticks_propszTestPlotBase._check_ticks_propss$ 433333$$T** K KB%)9bh::<>FV#### # O " 2 2 4 4 HWBJ / / /     rcPt}t}|D]j}|}||dd||ddkt |t |fS)Nr)set get_position get_pointsaddr=)rr)x_sety_setr,pointss rrzTestPlotBase._get_axes_layoutZs $ $B__&&1133F IIfQil # # # IIfQil # # # #E CJJ''rc@ddlm}||}d|D}|S)z Flatten axes, and filter only visible Parameters ---------- axes : matplotlib Axes object, or its list-like rrc:g|]}||Sr0)rN)r2r,s rr4z1TestPlotBase._flatten_visible..ps'666rR^^%5%56666r)rr)rr)rs rr%zTestPlotBase._flatten_visibleds> CBBBBB|D!!66T666 rrc||}|D]R}|j}d}d}|D]2}t|dd} t|dd} | r|dz }| r|dz }3||ksJ||ksJSdS)a 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) rr)xerryerrr,r xerr_count yerr_countcrrs r_check_has_errorbarsz!TestPlotBase._check_has_errorbarsss$$T** & &BJJJ $ $"1j%88"1j%88$!OJ$!OJ:%%%%:%%%%% & &rcddlm}t|td}|Y|d}t |||sJ|dkr3t |j|sJt |jtsJdSdS|,||D]}t ||sJdSt |tsJt| t|ksJ| D]\}} t | ||sJ|dkr|r| |ksJ@|dkrU|r| j |ksJt | j|sJt | jtsJ|dkr2| dd} | j } |r| |ksJtdS) at 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 )r`r)bothNr`rr)medians)matplotlib.axesr r`rkrMr,linesr%rsortedkeysitems get_titler)AssertionError) rreturned return_type expected_keyscheck_ax_titler typesrkeyvalueliner)s r_check_box_return_typez#TestPlotBase._check_box_return_types=& )(((((tU;;  "$ hk(:;; ; ; ;f$$!(+t44444!(.$77777%$77"..x88//A%a......h// / / /(--//**f].C.CCCCC&nn.. ) ) U!%{);<<<<<&((%8$00C7777 F**%;$x1133s::::%eh55555%ek4888888 F** +A.D9D%7#~~//36666((# ) )rcddl}fd}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 VdSrgridlinerNr2gs r zHTestPlotBase._check_grid_settings..is_grid_on..3DD1:11333DDDDDDrc3JK|]}|j VdSrrrs rrzHTestPlotBase._check_grid_settings..is_grid_on..rr)rgcarget_major_ticksrall)xticksyticksxoffyoffrs r is_grid_onz5TestPlotBase._check_grid_settings..is_grid_onsX\\^^)99;;FX\\^^)99;;FDDVDDDDDDDDVDDDDDD & &rrr)F)gridkindT)rr)piehexbinscatterr0)rrsubplotr=rcplotclf)robjkindskwsrrspndxrs` r_check_grid_settingsz!TestPlotBase._check_grid_settingss_ !    ' ' ' ' '  D H  QCJJ 6 6 6 QJE FF6F & & & CH & &$ &# & & &!z|| # # # HLLNNN H  QCJJ 6 6 6 QJE FF6F % % % CH 2$U 2 2c 2 2 2!z|| # # # HLLNNN777  AE NE::: vD)))**d*c***!z||###   AE NE::: vE***5d55555!z||### 9  rrtc,fd|dDS)zT Auxiliary function for correctly unpacking cycler after MPL >= 1.5 c g|] }| Sr0r0)r2r]fields rr4z/TestPlotBase._unpack_cycler..s>>>Q%>>>rzaxes.prop_cycler0)rrcParamsrs `r_unpack_cyclerzTestPlotBase._unpack_cyclers%?>>>(+<"=>>>>rc|jdS)Nx _shared_axesrr,s r get_x_axiszTestPlotBase.get_x_axiss##rc|jdS)Nyrrs r get_y_axiszTestPlotBase.get_y_axisrr)NT)T)r)rRrSrTrUrV)NNN)NNNN)rr)rr)rt)__name__ __module__ __qualname____doc__rrrrr"r-r:rIrQrZrcrwr'rrrrr%rrrrrrr0rrr r s ^ %%^% ////.++++,(2222(;?,,,,,$222FJ>*>*>*>*@"""(GK.K.K.K.K`1111"" " " " H(((   &&&&8IM:):):):)x46****X???? $$$$$$$$rr Fc ddlm}|rt}nt}d} |d|}||||fi|D]}tj|tj d5}| |dddn #1swxYwYtj |n#tj |wxYw|S)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. rNrT)return_filelike) rr_gen_default_plot_gen_two_subplotsgetgcfrr"assert_is_valid_plot_return_object ensure_cleansavefigr)f default_axeskwargsr gen_plotsretfigpaths r_check_plot_worksrs26$#####&% % C jj37799--  9Q..v.. 7 7C  1# 6 6 6 6 _T 2 2 2 d KK                     Js0A2C B/# C/B33C6B37CC%c+K|di|VdS)z' Create plot in a default way. Nr0r0r rr s rrr6s& !++f++rc+Kd|vr|d|di|V|tjjurd|vsJn|d|d<|di|VdS)z9 Create plot on two subplots forcefully created. r,Nr0) add_subplotpdplottingbootstrap_plotrs rrr=s 6  !++f++BK &&&6!!!!!s++t !++f++r)F)r __future__rtypingrrnumpyrnpandas.util._decoratorsrpandas.util._test_decoratorsutil_test_decoratorstdpandas.core.dtypes.apirpandasrrpandas._testing_testingrrr skip_if_no_mplr rrrr0rrr)s`#""""" 222222)))))))))//////%$$$$$$d$d$d$d$d$d$d$d$N0000f     r