B [Ž0d3Uã@sÔdZddlmZddlZddlmZmZddlZddlZ ddl m Z ddl m mZddlmZddlZddlmZmZmZddlmZeršddlmZejGd d „d ƒƒZdd d„Zdd„Zdd„Z dd„Z!dS)z² Module consolidating common testing functions for checking plotting. Currently all plotting tests are marked as slow via ``pytestmark = pytest.mark.slow`` at the module level. é)Ú annotationsN)Ú TYPE_CHECKINGÚSequence)Úcache_readonly)Ú is_list_like)Ú DataFrameÚSeriesÚ to_datetime)ÚAxesc@sôeZdZdZdd„Zdd„Zedd„ƒZedd „ƒZd9d d „Z d:dd„Z dd„Z d;dd„Z dd d!„Zd?d#d$„Zd@d%d&„Zd'd(„Zd)d*„ZdAd,d-„ZdBd.d/„Zifd0d1„ZdCd3d4„Zd5d6„Zd7d8„Zd S)DÚ TestPlotBasezE This is a common base class used for various plotting tests cCsTddl}ddlm}||_| ¡d|_d|_| ¡|_| ¡|_| ¡|_|  ¡|_ d|_ d|_ d|_ d|_ d }t d ¡Štjjd d g|d }tjjdddg|d }t||tjjdd|d tjjdd|d tjjd|d ttjj|j|j|tjdƒdœƒ|_WdQRXt ¡|_ttjjdd tjjdd t d¡tjjdd dœƒ|_dS)Nr)ÚcompatljÕ+2ZlÊRŠeöZéé)gš™™™™™@g333333@Úleftédé*ZMaleZFemale)ÚsizeÚAÚBÚCéBéé¡é )rÚdtype)ÚgenderÚ classroomÚheightÚweightÚcategoryÚdatetimeé)rrr)Ú matplotlibZpandas.plotting._matplotlibr Z rcdefaultsZstart_date_to_int64Zend_date_to_int64Z mpl_ge_2_2_3Z mpl_ge_3_0_0Ú mpl_ge_3_1_0Z mpl_ge_3_2_0Z bp_n_objectsZpolycollection_factorÚdefault_figsizeZdefault_tick_positionÚtmZ RNGContextÚnpÚrandomÚchoicerÚnormalÚrandintr Zint64Zhist_dfZmakeTimeDataFrameZtdfÚuniformZarangeZ hexbin_df)ÚselfÚmethodÚmplr Únrr©r0úN/var/www/html/venv/lib/python3.7/site-packages/pandas/tests/plotting/common.pyÚ setup_method*sF         zTestPlotBase.setup_methodcCs t ¡dS)N)r%Úclose)r,r-r0r0r1Úteardown_methodaszTestPlotBase.teardown_methodcCsddlm}|S)Nr)Úmatplotlib.pyplotÚpyplot)r,Úpltr0r0r1r7ds zTestPlotBase.pltcCsddlm}|jS)Nr)Zmatplotlib.colorsÚcolorsZcolorConverter)r,r8r0r0r1Úcolorconverterjs zTestPlotBase.colorconverterNTcCsj|r|dkrtdƒ‚| |¡}xF|D]>}|rR| ¡dk s˜sz5TestPlotBase._check_legend_marker..)r:Zget_legend_handles_labelsr=r<)r,rBZexpected_markersrAZhandlesÚ_Úmarkersr0r0r1Ú_check_legend_marker‡s  z!TestPlotBase._check_legend_markercsP| ¡}| ¡}dd„‰t|ƒt|ƒks,t‚‡fdd„t||ƒDƒt ¡dS)z¤ Check each axes has identical lines Parameters ---------- xp : matplotlib Axes object rs : matplotlib Axes object cSs | ¡}| ¡}t ||¡dS)N)Z get_xydatar%Úassert_almost_equal)ÚxplÚrslZxpdataZrsdatar0r0r1Ú check_line©sz,TestPlotBase._check_data..check_linecsg|]\}}ˆ||ƒ‘qSr0r0)rDrKrL)rMr0r1rF¯sz,TestPlotBase._check_data..N)Ú get_linesÚlenr=Úzipr%r3)r,ZxpÚrsZxp_linesZrs_linesr0)rMr1Ú _check_datas zTestPlotBase._check_datacCsFddlm}t||ƒs$t|ƒs$|g}x|D]}| ¡|ks*t‚q*WdS)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.collectionsrSÚ isinstancerÚ get_visibler=)r,Ú collectionsrArSÚpatchr0r0r1Ú_check_visible²s  zTestPlotBase._check_visiblezAxes | Sequence[Axes]ÚboolÚNone)r?ÚfilledÚreturncCs:| |¡}x*|D]"}x|jD]}|j|kst‚qWqWdS)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;ZpatchesÚfillr=)r,r?r\rBrXr0r0r1Ú_check_patches_all_filledÅs   z&TestPlotBase._check_patches_all_filledcs*| ¡}tt||ƒƒ‰‡fdd„|jDƒS)Ncsg|] }ˆ|‘qSr0r0)rDÚv)Úmappedr0r1rFÜsz3TestPlotBase._get_colors_mapped..)ÚuniqueÚdictrPÚvalues)r,Zseriesr8rbr0)rar1Ú_get_colors_mapped×szTestPlotBase._get_colors_mappedcCsŽddlm}m}m}ddlm}|j} |dk rÚ|dk rR| ||¡}|dt|ƒ…}t|ƒt|ƒksft ‚xrt ||ƒD]d\} } t | |ƒr˜|   ¡} |   | ¡} n(t | ||fƒr¸t|  ¡dƒ} n|  ¡} |   | ¡} | | ksrt ‚qrW|dk rŠ|dk r | ||¡}|dt|ƒ…}t|ƒt|ƒks t ‚xht ||ƒD]Z\} } t | |ƒrN|  ¡d} n|  ¡} t | tjƒrlt| ƒ} |   | ¡} | | ks,t ‚q,WdS)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)rSÚLineCollectionÚPolyCollection)ÚLine2DN)rTrSrfrgZmatplotlib.linesrhr9rerOr=rPrUÚ get_colorZto_rgbaÚtupleZ get_edgecolorZ get_facecolorr&Zndarray)r,rWZ linecolorsZ facecolorsÚmappingrSrfrgrhÚconvrXÚcolorÚresultÚexpectedr0r0r1Ú _check_colorsÞs<          zTestPlotBase._check_colorscCsdt|ƒs| ¡|ks`t‚nFdd„|Dƒ}t|ƒt|ƒks.N)rrqr=rOrP)r,Ztextsror@ÚlabelÚer0r0r1r> s zTestPlotBase._check_text_labelsc Cs&ddlm}| |¡}x|D]þ}|dk s4|dk r t|j ¡|ƒrN| ¡}n| ¡|jdd}x<|D]4} |dk r„t |   ¡|¡|dk rht |   ¡|¡qhW|dk s°|dk rt|j  ¡|ƒrÊ|  ¡}n|  ¡|j dd}x>|D]6} |dk rt |   ¡|¡|dk rät |   ¡|¡qäWqWdS)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) Zmatplotlib.tickerrur;rUÚxaxisZget_minor_formatterZget_xticklabelsr%rJZ get_fontsizeZ get_rotationÚyaxisZget_yticklabels) r,r?Z xlabelsizeZxrotZ ylabelsizeZyrotrurBr@rsr0r0r1Ú_check_ticks_props3s*       zTestPlotBase._check_ticks_propsÚlinearcCs@| |¡}x0|D](}|j ¡|ks&t‚|j ¡|kst‚qWdS)a 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;rwZ get_scaler=rx)r,r?rwrxrBr0r0r1Ú_check_ax_scalescs  zTestPlotBase._check_ax_scalesc Cs¦ddlm}|dkr|j}| |¡}|dk r^t|ƒ|ks.)r}r|)r,r?r|r0r0r1r;¢s zTestPlotBase._flatten_visiblerc Cs€| |¡}xp|D]h}|j}d}d}x<|D]4}t|ddƒ} t|ddƒ} | rP|d7}| r(|d7}q(W||kslt‚||kst‚qWdS)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 rÚhas_xerrFÚhas_yerrr‚N)r;Ú containersÚgetattrr=) r,r?ZxerrZyerrrBr‡Z xerr_countZ yerr_countÚcr…r†r0r0r1Ú_check_has_errorbars±s       z!TestPlotBase._check_has_errorbarsc Cs’ddlm}t|tdœ}|dkrj|dkr,d}t|||ƒs>t‚|dkrft|j|ƒsVt‚t|jtƒsft‚n$|dkr˜x | |¡D]}t||ƒs~t‚q~WdSt|t ƒs¦t‚t |  ¡ƒt |ƒks¾t‚xÎ|  ¡D]Â\}} t| ||ƒsât‚|dkr|rŠ|   ¡|ksŠt‚qÈ|dkrP|r*| j  ¡|ks*t‚t| j|ƒs      z#TestPlotBase._check_box_return_typecslddl}‡fdd„}d}xL|D]B}ˆj ddt|ƒ|¡|d7}|jddd|jfd |i|—Ž|ƒrrt‚ˆj ddt|ƒ|¡|d7}|jdd d|jf|dd œ|—Ž|ƒrÀt‚|d kr ˆj ddt|ƒ|¡|d7}|jdd d|jfd |i|—Ž|ƒst‚ˆj ddt|ƒ|¡|d7}|jddd|jf|d d œ|—Ž|ƒs t‚q WdS) Nrcszˆj ¡j ¡}ˆj ¡j ¡}ˆjrLtdd„|Dƒƒ}tdd„|Dƒƒ}n$tdd„|Dƒƒ}tdd„|Dƒƒ}|ov| S)Ncss|]}|j ¡ VqdS)N)ÚgridlinerV)rDÚgr0r0r1ú szHTestPlotBase._check_grid_settings..is_grid_on..css|]}|j ¡ VqdS)N)r™rV)rDršr0r0r1r›scss|]}|j VqdS)N)ÚgridOn)rDršr0r0r1r›scss|]}|j VqdS)N)rœ)rDršr0r0r1r›s)r7ZgcarwZget_major_ticksrxr#Úall)ZxticksZyticksZxoffZyoff)r,r0r1Ú is_grid_on sz5TestPlotBase._check_grid_settings..is_grid_onr‚rr?F)ÚgridÚkindT)r rŸ)ÚpieZhexbinZscatter)r"r7ZsubplotrOÚrcZplotr=)r,ÚobjÚkindsÚkwsr.ržZspndxr r0)r,r1Ú_check_grid_settingss2    z!TestPlotBase._check_grid_settingsrmcs‡fdd„|dDƒS)zT Auxiliary function for correctly unpacking cycler after MPL >= 1.5 csg|] }|ˆ‘qSr0r0)rDr`)Úfieldr0r1rF<sz/TestPlotBase._unpack_cycler..zaxes.prop_cycler0)r,ZrcParamsr§r0)r§r1Ú_unpack_cycler8szTestPlotBase._unpack_cyclercCs|j ¡r|jdS|jS)NÚx)r Ú mpl_ge_3_5_0Ú _shared_axesZ_shared_x_axes)r,rBr0r0r1Ú get_x_axis>szTestPlotBase.get_x_axiscCs|j ¡r|jdS|jS)NÚy)r rªr«Z_shared_y_axes)r,rBr0r0r1Ú get_y_axisAszTestPlotBase.get_y_axis)NT)NT)T)T)NNN)NNNN)rzrz)NNN)rr)NT)rm)Ú__name__Ú __module__Ú __qualname__Ú__doc__r2r4rr7r9rCrIrRrYr_rerpr>ryr{rr~r;rŠr˜r¦r¨r¬r®r0r0r0r1r $s07      A /  $   ; 0 r ÚalwaysFc KsÚddlm}|rt}nt}d}t ¡ªt |¡zŒy`| d| ¡¡}|  ¡x |||f|ŽD]}t   |¡q^Wt j dd}|  |¡WdQRXWn&tk rº} z| ‚Wdd} ~ XYnXWdt  |¡X|SQRXdS)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)Zreturn_filelike)r5r6Ú_gen_default_plotÚ_gen_two_subplotsÚwarningsÚcatch_warningsÚ simplefilterÚgetZgcfZclfr%Z"assert_is_valid_plot_return_objectZ ensure_cleanZsavefigÚ Exceptionr3) ÚfÚfilterwarningsZ default_axesÚkwargsr7Z gen_plotsÚretÚfigÚpathÚerrr0r0r1Ú_check_plot_worksEs$    rÂcks|f|ŽVdS)z' Create plot in a default way. Nr0)r»r¿r½r0r0r1r´sr´cksVd|kr| d¡|f|ŽV|tjjkr8d|ksFt‚n| d¡|d<|f|ŽVdS)z9 Create plot on two subplots forcefully created. rBéÓéÔN)Z add_subplotÚpdZplottingZbootstrap_plotr=)r»r¿r½r0r0r1rµ†s   rµcCstj tj t¡¡\}}|S)N)ÚosrÀÚsplitÚabspathÚ__file__)ÚpthrGr0r0r1Úcurpath•srË)r³F)"r²Ú __future__rrÆÚtypingrrr¶Únumpyr&Zpandas.util._decoratorsrZpandas.util._test_decoratorsÚutilZ_test_decoratorsÚtdZpandas.core.dtypes.apirZpandasrÅrrr Zpandas._testingZ_testingr%rŒr Zskip_if_no_mplr rÂr´rµrËr0r0r0r1Ús,     % :