WhLdZddlmZddlmZmZddlZddlZddl m Z ddl m cm ZddlmZddlZddlmZddlmZerddlmZejGd d Zdd ZdZdZdS)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 b/var/www/html/movieo_spanner_bot/venv/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}|jSr)matplotlib.colorscolorscolorConverter)rr!s rcolorconverterzTestPlotBase.colorconverter0s******$$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_labels6s  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_markerMs  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_datacs<<>><<>>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.collectionsrL isinstancer get_visible)r collectionsr,rLpatchs r_check_visiblezTestPlotBase._check_visiblewsw 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*rTr-rQs r_check_patches_all_filledz&TestPlotBase._check_patches_all_filleds\$$T** , ,B , ,zV+++++ , , ,rc|}tt||fd|jDS)Nc g|] }| Sr1r1)r3vmappeds rr5z3TestPlotBase._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)rLLineCollectionPolyCollection)Line2DN)rMrLrfrgmatplotlib.linesrhr#rdr>r?rN get_colorto_rgbatuple get_edgecolor get_facecolornpndarray)rrP linecolors facecolorsmappingrLrfrgrhconvrQcolorresultexpecteds r _check_colorszTestPlotBase._check_colorssl$          ,+++++"  !"!44WjII '(:#k*:*:(:; {##s:6666 #K < < * * ueV,,3"__..F!\\&11FF'GHH3"5#6#6#8#8#;<.s 222qajjll222rN)rr{r>r?)rtextsrwr+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&rNxaxisget_minor_formatterget_xticklabelsrrA get_fontsize get_rotationyaxisget_yticklabels) rr* xlabelsizexrot ylabelsizeyrotrr-r+r~s r_check_ticks_propszTestPlotBase._check_ticks_propss$ 433333$$T** K KB%)9bh::< get_children_get_axes_layoutrassert_numpy_array_equalfigureget_size_inchesroarrayfloat64) rr*axes_numlayoutfigsizer visible_axesr-rvs r_check_axes_shapezTestPlotBase._check_axes_shape9s CBBBBB ? G,,T22  |$$0000" 2 22??,,--11111  **<<+=+=>>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_layout]s $ $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|]}||Sr1)rO)r3r-s rr5z1TestPlotBase._flatten_visible..ss'666rR^^%5%56666r)rr)rr*rs rr&zTestPlotBase._flatten_visiblegs> 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_errorbarsvs$$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 )rar*bothNrarr*medians)matplotlib.axesr rarlrNr-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((# ) )rcPddl}fd}d}|D]}jddt|z||dz }|dd|jd d|i||rJjddt|z||dz }|dd |jd |dd ||rJ|d vrljjddt|z||dz }|dd |jd d|i||sJjddt|z||dz }|dd|jd |d d ||sJdS) Nrc2jj}jj}t d|D}t d|D}|o| S)Nc3JK|]}|j VdSrgridlinerOr3gs 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)piehexbinscatterr1)rrsubplotr>rcplot)robjkindskwsrrspndxrs` r_check_grid_settingsz!TestPlotBase._check_grid_settingss! !    ' ' ' ' ' $ $D H  QCJJ 6 6 6 QJE FF6F & & & CH & &$ &# & & &!z|| # ## H  QCJJ 6 6 6 QJE FF6F % % % CH 2$U 2 2c 2 2 2!z|| # ##777  AE NE::: vD)))**d*c***!z||##|  AE NE::: vE***5d55555!z||##|3 $ $rruc,fd|dDS)zT Auxiliary function for correctly unpacking cycler after MPL >= 1.5 c g|] }| Sr1r1)r3r^fields rr5z/TestPlotBase._unpack_cycler..s>>>Q%>>>rzaxes.prop_cycler1)rrcParamsrs `r_unpack_cyclerzTestPlotBase._unpack_cyclers%?>>>(+<"=>>>>rcJddlm}|r |jdS|jS)Nr mpl_ge_3_5_0x)"pandas.plotting._matplotlib.compatr _shared_axes_shared_x_axesrr-rs r get_x_axiszTestPlotBase.get_x_axis8CCCCCC <>> (?3' '  rcJddlm}|r |jdS|jS)Nrry)rrr_shared_y_axesrs r get_y_axiszTestPlotBase.get_y_axisrr)NT)T)r*rSrTrUrVrW)NNN)NNNN)rr)rr)ru)__name__ __module__ __qualname____doc__rrrrr#r.r;rJrRr[rdrxr(rrrrr&rrrrrrr1rrr r s ^ %%^% ////.++++,(2222(;?,,,,,$222FJ@*@*@*@*D"""(GK.K.K.K.K`1111"" " " " H(((   &&&&8IM:):):):)x46'$'$'$'$R???? !!!!!!!!rr alwaysFc ddlm}|rt}nt}d}t j5t j| |d|}| |||fi|D]}tj |tj d5}| |dddn #1swxYwYn#t$r} | d} ~ wwxYwtj|n#tj|wxYw|cdddS#1swxYwYdS)aT Create plot and ensure that plot return object is valid. Parameters ---------- f : func Plotting function. filterwarnings : str Warnings filter. See https://docs.python.org/3/library/warnings.html#warning-filter 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_subplotswarningscatch_warnings simplefiltergetgcfclfr"assert_is_valid_plot_return_object ensure_cleansavefig Exceptionr) ffilterwarnings default_axeskwargsr gen_plotsretfigpatherrs r_check_plot_worksr s<$#####&% % C  " "n--- **Xswwyy11C GGIII yC22622 ; ;5c::::666 "$ D!!! " " " " " " " " " " " " " " "   I  HSMMMMBHSMMMM#slD3A2C#5C C#C C#C C#"D # C4-C//C44D 7D3 D""D33D7:D7c+K|di|VdS)z' Create plot in a default way. Nr1r1rrrs rr r Es& !++f++rc+Kd|vr|d|di|V|tjjurd|vsJn|d|d<|di|VdS)z9 Create plot on two subplots forcefully created. r-Nr1) add_subplotpdplottingbootstrap_plotr s rr r Ls 6  !++f++BK &&&6!!!!!s++t !++f++r)rF)r __future__rtypingrrr numpyropandas.util._decoratorsrpandas.util._test_decoratorsutil_test_decoratorstdpandas.core.dtypes.apirpandasr%rpandas._testing_testingrrr skip_if_no_mplr rr r r1rrr5si#"""""222222)))))))))//////%$$$$$$k!k!k!k!k!k!k!k!\7777t     r