3 qdCI@sddlZddlZddlZddlmZddlmZddljj Z ddl m Z ddl Zddl mZmZddljZe jGdddZd d d Zd d ZdS)N)random)cache_readonly) is_list_like) DataFrameSeriesc@seZdZddZddZeddZeddZd.d d Zd/d dZ ddZ d0ddZ ddZ d1ddZ ddZd2ddZd3ddZd4ddZd d!Zd"d#Zd5d%d&Zd6d'd(Zifd)d*Zd7d,d-Zd S)8 TestPlotBasecCs ddl}ddlm}|j|j|_|j|_|j|_ |j |_ d|_ d|_ d|_d|_d}tjd htjjd d g|d }tjjd ddg|d }t||tjdd|d tjdd|d tjd|d d|_WdQRXtj|_ttjjdd tjjdd tjdtjjdd d|_dS)Nr)compat皙@333333@leftd*ZMaleZFemale)sizeABCB )gender classroomheightZweightcategory)rrr)r r ) matplotlibZpandas.plotting._matplotlibrZ rcdefaultsZ _mpl_ge_2_2_3Z mpl_ge_2_2_3Z _mpl_ge_3_0_0Z mpl_ge_3_0_0Z _mpl_ge_3_1_0 mpl_ge_3_1_0Z _mpl_ge_3_2_0Z mpl_ge_3_2_0Z bp_n_objectsZpolycollection_factordefault_figsizeZdefault_tick_positiontmZ RNGContextnprchoicerZnormalrandintZhist_dfZmakeTimeDataFrameZtdfuniformZarangeZ hexbin_df)selfmethodmplrnrrr)>/tmp/pip-build-7vycvbft/pandas/pandas/tests/plotting/common.py setup_methods4          zTestPlotBase.setup_methodcCs tjdS)N)r close)r%r&r)r)r*teardown_methodAszTestPlotBase.teardown_methodcCsddlj}|S)Nr)matplotlib.pyplotpyplot)r%pltr)r)r*r0Ds zTestPlotBase.pltcCsddlj}|jS)Nr)Zmatplotlib.colorscolorsZcolorConverter)r%r1r)r)r*colorconverterJs zTestPlotBase.colorconverterNTcCsj|r|dkrtd|j|}xF|D]>}|rR|jdk sxsz5TestPlotBase._check_legend_marker..)r3Zget_legend_handles_labelsr6r5)r%r;Zexpected_markersr:Zhandles_markersr)r)r*_check_legend_markergs  z!TestPlotBase._check_legend_markercsP|j}|j}ddt|t|ks,tfddt||DtjdS)z Check each axes has identical lines Parameters ---------- xp : matplotlib Axes object rs : matplotlib Axes object cSs |j}|j}tj||dS)N)Z get_xydatar assert_almost_equal)xplrslZxpdataZrsdatar)r)r* check_linesz,TestPlotBase._check_data..check_linecsg|]\}}||qSr)r))r=rDrE)rFr)r*r?sz,TestPlotBase._check_data..N) get_lineslenr6zipr r,)r%ZxprsZxp_linesZrs_linesr))rFr* _check_data}s zTestPlotBase._check_datacCsJddlm}t|| r(t| r(|g}x|D]}|j|ks.tq.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.collectionsrL isinstancer get_visibler6)r% collectionsr:rLpatchr)r)r*_check_visibles  zTestPlotBase._check_visiblecs*|j}tt||fdd|jDS)Ncsg|] }|qSr)r))r=v)mappedr)r*r?sz3TestPlotBase._get_colors_mapped..)uniquedictrIvalues)r%Zseriesr1rUr))rTr*_get_colors_mappedszTestPlotBase._get_colors_mappedcCsddlm}m}m}ddlm}|j} |dk r|dk rR|j||}|dt|}t|t|ksft xrt ||D]d\} } t | |r| j } | j | } n(t | ||frt| jd} n| j} | j | } | | ksrt qrW|dk r|dk r |j||}|dt|}t|t|ks t xht ||D]Z\} } t | |rN| jd} n| j} t | tjrlt| } | j | } | | 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)rLLineCollectionPolyCollection)Line2DN)rMrLrYrZZmatplotlib.linesr[r2rXrHr6rIrNZ get_colorZto_rgbatupleZ get_edgecolorZ get_facecolorr!Zndarray)r%rPZ linecolorsZ facecolorsmappingrLrYrZr[convrQcolorresultexpectedr)r)r* _check_colorss<          zTestPlotBase._check_colorscCsdt|s|j|ks`tnFdd|D}t|t|ks.N)rrcr6rHrI)r%Ztextsrar9labeler)r)r*r7s zTestPlotBase._check_text_labelsc Cs&ddlm}|j|}x|D]}|dk s4|dk rt|jj|rN|j}n|j|jdd}x<|D]4} |dk rtj| j ||dk rhtj| j |qhW|dk s|dk rt|j j|r|j }n|j |j dd}x>|D]6} |dk rtj| j ||dk rtj| j |qWqWdS)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.tickerrgr4rNxaxisZget_minor_formatterZget_xticklabelsr rCZ get_fontsizeZ get_rotationyaxisZget_yticklabels) r%r8Z xlabelsizeZxrotZ ylabelsizeZyrotrgr;r9rer)r)r*_check_ticks_propss*       zTestPlotBase._check_ticks_propslinearcCs@|j|}x0|D](}|jj|ks&t|jj|kstqWdS)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)r4riZ get_scaler6rj)r%r8rirjr;r)r)r*_check_ax_scales-s  zTestPlotBase._check_ax_scalesc Csddlm}|dkr|j}|j|}|dk r^t||kss     zTestPlotBase._check_axes_shapecCsZt}t}x<|D]4}|jj}|j|dd|j|ddqWt|t|fS)Nr)set get_positionZ get_pointsaddrH)r%r8Zx_setZy_setr;Zpointsr)r)r*rpbs  zTestPlotBase._get_axes_layoutcCs&ddlm}||}dd|D}|S)z Flatten axes, and filter only visible Parameters ---------- axes : matplotlib Axes object, or its list-like r)rncSsg|]}|jr|qSr))rO)r=r;r)r)r*r?xsz1TestPlotBase._flatten_visible..)rorn)r%r8rnr)r)r*r4ls zTestPlotBase._flatten_visiblerc Cs|j|}xp|D]h}|j}d}d}x<|D]4}t|dd} t|dd} | rP|d7}| r(|d7}q(W||kslt||kstqWdS)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_yerrrtN)r4 containersgetattrr6) r%r8ZxerrZyerrr;rzZ xerr_countZ yerr_countcrxryr)r)r*_check_has_errorbars{s       z!TestPlotBase._check_has_errorbarsc Csddlm}t|td}|dkrj|dkr,d}t|||s>t|dkrft|j|sVtt|jtsftn$|dkrx |j|D]}t||s~tq~WdSt|t stt |j t |kstx|j D]\}} t| ||st|dkr|r| j |kstq|dkrP|r*| jj |ks*tt| j|s      z#TestPlotBase._check_box_return_typecspddl}fdd}d}xP|D]F}jjddt|||d7}|jddd|jfd |i|| sttjjddt|||d7}|jdd d|jf|dd || st|d kr jjddt|||d7}|jdd d|jfd |i||stjjddt|||d7}|jddd|jf|d d ||s tq WdS) Nrcszjjjj}jjjj}jrLtdd|D}tdd|D}n$tdd|D}tdd|D}|ov| S)Ncss|]}|jj VqdS)N)gridlinerO)r=gr)r)r* szHTestPlotBase._check_grid_settings..is_grid_on..css|]}|jj VqdS)N)rrO)r=rr)r)r*rscss|]}|j VqdS)N)gridOn)r=rr)r)r*rscss|]}|j VqdS)N)r)r=rr)r)r*rs)r0ZgcariZget_major_ticksrjrall)ZxticksZyticksZxoffZyoff)r%r)r* is_grid_onsz5TestPlotBase._check_grid_settings..is_grid_onrtrr8F)gridkindT)rrZpie)rr0ZsubplotrHrcZplotr6)r%objZkindskwsr'rZspndxrr))r%r*_check_grid_settingss2    z!TestPlotBase._check_grid_settingsr_csfdd|dDS)zT Auxiliary function for correctly unpacking cycler after MPL >= 1.5 csg|] }|qSr)r))r=rS)fieldr)r*r?sz/TestPlotBase._unpack_cycler..zaxes.prop_cycler))r%ZrcParamsrr))rr*_unpack_cyclerszTestPlotBase._unpack_cycler)NT)NT)T)NNN)NNNN)rlrl)NNN)rr)NT)r_)__name__ __module__ __qualname__r+r-rr0r2r<rBrKrRrXrbr7rkrmrsrpr4r}rrrr)r)r)r*rs(*      = /  $   ; 0ralwayscKsddlj}d}tjtj|zy |d}Wntk rL|j}YnX|j|jd|j d|f|}t j ||t j jkrd|kstn|j d|d<|f|}t j |t jdd}|j|WdQRXWdt j|X|SQRXdS)Nrrqr;T)Zreturn_filelike)r.r/warningscatch_warnings simplefilterKeyErrorZgcfZclfgetZ add_subplotr Z"assert_is_valid_plot_return_objectpdZplottingZbootstrap_plotr6Z ensure_cleanZsavefigr,)ffilterwarningskwargsr0retZfigpathr)r)r*_check_plot_works s,          rcCstjjtjjt\}}|S)N)osrsplitabspath__file__)pthr@r)r)r*curpath,sr)r)rrnumpyr!rZpandas.util._decoratorsrZpandas.util._test_decoratorsutilZ_test_decoratorstdZpandas.core.dtypes.apirZpandasrrrZpandas._testingZ_testingr Zskip_if_no_mplrrrr)r)r)r*s     w #