ó šxŠ\c@s/ddlZddlZddlZddlmZddlZddlmZmZddl m Z ddl j j ZddlmZddlmZmZddlj jZddlmZmZddljZddlmZd „Zd „Zejd e fd „ƒYƒZ!d d„Z"d„Z#dS(i˙˙˙˙N(trandom(t iteritemstzip(tcache_readonly(t is_list_like(t DataFrametSeries(t"assert_is_valid_plot_return_objectt ensure_clean(t_flattencCs9yddlm}Wntk r4tjdƒnXdS(Ni˙˙˙˙(t gaussian_kdes*scipy version doesn't support gaussian_kde(t scipy.statsR t ImportErrortpytesttskip(R ((s;lib/python2.7/site-packages/pandas/tests/plotting/common.pyt_skip_if_no_scipy_gaussian_kdes cCs<|dkr8yddlm}Wq8tk r4tSXntS(Ntkdetdensityi˙˙˙˙(R (RR(R R R tFalsetTrue(tkindR ((s;lib/python2.7/site-packages/pandas/tests/plotting/common.pyt_ok_for_gaussian_kde&s   t TestPlotBasecBsţeZd„Zd„Zed„ƒZed„ƒZded„Z d„Z ed„Z d„Z dddd„Z d „Zddddd „Zd d d „Zdddd „Zd„Zd„Zddd„Zded„Zid„Zdd„ZRS(c CsĹddl}|jƒtjjƒ|_tjjƒ|_tjjƒ|_ tjj ƒ|_ tjj ƒ|_ d|_d|_d|_d|_d}tjdƒŠtjjd d gd |ƒ}tjjd d dgd |ƒ}ti|d6|d6tjddd |ƒd6tjddd |ƒd6tjdd |ƒd6ƒ|_WdQXtjƒ|_titjjd dƒd 6tjjd dƒd 6tjdƒtjjd dƒd6ƒ|_dS(Ni˙˙˙˙iigš™™™™™@g333333@tleftidi*tMaletFemaletsizetAtBtCtgendert classroomiBitheightiĄi tweighttcategoryi(gš™™™™™@g333333@( t matplotlibt rcdefaultstplottingt_compatt _mpl_ge_2_0_1t mpl_ge_2_0_1t _mpl_ge_2_1_0t mpl_ge_2_1_0t _mpl_ge_2_2_0t mpl_ge_2_2_0t _mpl_ge_2_2_2t mpl_ge_2_2_2t _mpl_ge_3_0_0t mpl_ge_3_0_0t bp_n_objectstpolycollection_factortdefault_figsizetdefault_tick_positionttmt RNGContexttnpRtchoiceRtnormaltrandintthist_dftmakeTimeDataFramettdftuniformtaranget hexbin_df(tselftmethodtmpltnRR((s;lib/python2.7/site-packages/pandas/tests/plotting/common.pyt setup_method3s2      ! %cCstjƒdS(N(R5tclose(RARB((s;lib/python2.7/site-packages/pandas/tests/plotting/common.pytteardown_methodTscCsddlj}|S(Ni˙˙˙˙(tmatplotlib.pyplottpyplot(RAtplt((s;lib/python2.7/site-packages/pandas/tests/plotting/common.pyRJWscCsddlj}|jS(Ni˙˙˙˙(tmatplotlib.colorstcolorstcolorConverter(RARL((s;lib/python2.7/site-packages/pandas/tests/plotting/common.pytcolorconverter\scCsš|r!|dkr!tdƒ‚n|j|ƒ}xc|D][}|rz|jƒdk s[t‚|j|jƒjƒ|ƒq7|jƒdks7t‚q7WdS(sQ 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 s-labels must be specified when visible is TrueN(tNonet ValueErrort_flatten_visiblet get_legendtAssertionErrort_check_text_labelst get_texts(RAtaxestlabelstvisibletax((s;lib/python2.7/site-packages/pandas/tests/plotting/common.pyt_check_legend_labelsas cCs||jƒ}|jƒ}d„}t|ƒt|ƒks?t‚gt||ƒD]\}}|||ƒ^qOtjƒdS(s¤ Check each axes has identical lines Parameters ---------- xp : matplotlib Axes object rs : matplotlib Axes object cSs,|jƒ}|jƒ}tj||ƒdS(N(t get_xydataR5tassert_almost_equal(txpltrsltxpdatatrsdata((s;lib/python2.7/site-packages/pandas/tests/plotting/common.pyt check_line…s  N(t get_linestlenRSRR5RF(RAtxptrstxp_linestrs_linesRaR]R^((s;lib/python2.7/site-packages/pandas/tests/plotting/common.pyt _check_datays   /cCsfddlm}t||ƒ r9t|ƒ r9|g}nx&|D]}|jƒ|ks@t‚q@WdS(s 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 i˙˙˙˙(t CollectionN(tmatplotlib.collectionsRit isinstanceRt get_visibleRS(RAt collectionsRXRitpatch((s;lib/python2.7/site-packages/pandas/tests/plotting/common.pyt_check_visibleŽs   cCs?|jƒ}tt||ƒƒ}g|jD]}||^q+S(N(tuniquetdictRtvalues(RAtseriesRLRptmappedtv((s;lib/python2.7/site-packages/pandas/tests/plotting/common.pyt_get_colors_mappedĄs cCs-ddlm}ddlm}m}m}|j} |dk r;|dk rr|j||ƒ}|t |ƒ }nt |ƒt |ƒkst ‚x¨t ||ƒD]”\} } t | |ƒrŮ| j ƒ} | j| ƒ} n:t | ||fƒrt| jƒdƒ} n | jƒ} | j| ƒ} | | ks t ‚q Wn|dk r)|dk rx|j||ƒ}|t |ƒ }nt |ƒt |ƒks–t ‚xt ||ƒD]|\} } t | |ƒrÔ| jƒd} n | jƒ} t | tjƒrt| ƒ} n| j| ƒ} | | ksŚt ‚qŚWndS(s3 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 i˙˙˙˙(tLine2D(RitPolyCollectiontLineCollectioniN(tmatplotlib.linesRwRjRiRxRyRNRORvRcRSRRkt get_colortto_rgbattuplet get_edgecolort get_facecolorR7tndarray(RARmt linecolorst facecolorstmappingRwRiRxRytconvRntcolortresulttexpected((s;lib/python2.7/site-packages/pandas/tests/plotting/common.pyt _check_colors¨s<        cCsšt|ƒs'|jƒ|ks–t‚nog|D]}|jƒ^q.}t|ƒt|ƒksdt‚x/t||ƒD]\}}||kstt‚qtWdS(s. 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 N(Rtget_textRSRcR(RAttextsR‡ttRWtlabelte((s;lib/python2.7/site-packages/pandas/tests/plotting/common.pyRTçs c CsŽddlm}|j|ƒ}xh|D]`}|s8|rŮt|jjƒ|ƒr_|jƒ}n|jƒ|jdtƒ}x[|D]P} |dk r­t j | j ƒ|ƒn|dk r‚t j | j ƒ|ƒq‚q‚Wn|sĺ|r&t|j jƒ|ƒr |jƒ}n|jƒ|jdtƒ}x[|D]P} |dk rZt j | j ƒ|ƒn|dk r/t j | j ƒ|ƒq/q/Wq&q&WdS(s— 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 i˙˙˙˙(t NullFormattertminorN(tmatplotlib.tickerRŽRQRktxaxistget_minor_formattertget_xticklabelsRROR5R\t get_fontsizet get_rotationtyaxistget_yticklabels( RARVt xlabelsizetxrott ylabelsizetyrotRŽRYRWRŒ((s;lib/python2.7/site-packages/pandas/tests/plotting/common.pyt_check_ticks_propsús2              tlinearcCsZ|j|ƒ}xD|D]<}|jjƒ|ks7t‚|jjƒ|kst‚qWdS(s 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(RQR‘t get_scaleRSR–(RARVR‘R–RY((s;lib/python2.7/site-packages/pandas/tests/plotting/common.pyt_check_ax_scales,s  cCsć|dkr|j}n|j|ƒ}|dk r}t|ƒ|ksKt‚x/|D]$}t|jƒƒdksRt‚qRWn|dk rł|jt|ƒƒ}||ksłt‚ntj |dj j ƒt j |dt jƒƒdS(sĎ 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 itdtypeN(ROR3RQRcRSt get_childrent_get_axes_layoutR R5tassert_numpy_array_equaltfiguretget_size_inchesR7tarraytfloat64(RARVtaxes_numtlayouttfigsizet visible_axesRYR†((s;lib/python2.7/site-packages/pandas/tests/plotting/common.pyt_check_axes_shape=s    % cCsutƒ}tƒ}xJ|D]B}|jƒjƒ}|j|ddƒ|j|ddƒqWt|ƒt|ƒfS(Nii(tsett get_positiont get_pointstaddRc(RARVtx_setty_setRYtpoints((s;lib/python2.7/site-packages/pandas/tests/plotting/common.pyR˘_s   cCs5t|ƒ}g|D]}|jƒr|^q}|S(s— Flatten axes, and filter only visible Parameters ---------- axes : matplotlib Axes object, or its list-like (R Rl(RARVRY((s;lib/python2.7/site-packages/pandas/tests/plotting/common.pyRQis %ic Cs¸|j|ƒ}x˘|D]š}|j}d}d}xX|D]P}t|dtƒ} t|dtƒ} | ru|d7}n| r8|d7}q8q8W||ksžt‚||kst‚qWdS(s 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 ithas_xerrthas_yerriN(RQt containerstgetattrRRS( RARVtxerrtyerrRYRśt xerr_countt yerr_counttcR´Rľ((s;lib/python2.7/site-packages/pandas/tests/plotting/common.pyt_check_has_errorbarsvs     c Cs9ddlm}itd6|d6td6}|dkr§|dkrLd}nt|||ƒset‚|dkr5t|j|ƒs‰t‚t|jtƒs¤t‚q5nŽ|dkrćx,|j |ƒD]}t||ƒsĂt‚qĂWdSt|t ƒsűt‚t |j ƒƒt |ƒkst‚xt |ƒD]\}} t| ||ƒsQt‚|dkr|r1| jƒ|ks~t‚q1q,|dkrä|rą| jjƒ|ksąt‚nt| j|ƒsÉt‚t| jtƒs1t‚q,|dkr+| dd} | j} |r1| jƒ|ks(t‚q1q,t‚q,WdS( st 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. i˙˙˙˙(tAxesRqRVtbothNtmediansi(tmatplotlib.axesRžRqR}RORkRSRYtlinesRQRtsortedtkeysRt get_titleRV( RAtreturnedt return_typet expected_keystcheck_ax_titleRžttypestrtkeytvaluetlineRV((s;lib/python2.7/site-packages/pandas/tests/plotting/common.pyt_check_box_return_type‘s>     $    csáddl}‡fd†}d}xš|D]ą}t|ƒs@q(nˆjjddt|ƒ|ƒ|d7}|jddtƒ|jd|||ƒ s t‚ˆjjddt|ƒ|ƒ|d7}|jddt ƒ|jd|dt||ƒ st‚|dkr(ˆjjddt|ƒ|ƒ|d7}|jddt ƒ|jd|||ƒsqt‚ˆjjddt|ƒ|ƒ|d7}|jddtƒ|jd|dt ||ƒsŮt‚q(q(WdS( Ni˙˙˙˙cs[td„ˆjjƒjjƒDƒƒ}td„ˆjjƒjjƒDƒƒ}|oY| S(Ncss|]}|j VqdS(N(tgridOn(t.0tg((s;lib/python2.7/site-packages/pandas/tests/plotting/common.pys Ńscss|]}|j VqdS(N(RĐ(RŃRŇ((s;lib/python2.7/site-packages/pandas/tests/plotting/common.pys Ós(tallRJtgcaR‘tget_major_ticksR–(txofftyoff(RA(s;lib/python2.7/site-packages/pandas/tests/plotting/common.pyt is_grid_onĐs   iiRVtgridRtpie( R#RRJtsubplotRctrcRtplotRSR(RAtobjtkindstkwsRCRŘtspndxR((RAs;lib/python2.7/site-packages/pandas/tests/plotting/common.pyt_check_grid_settingsËs6            R…cCsg|dD]}||^q S(sT Auxiliary function for correctly unpacking cycler after MPL >= 1.5 saxes.prop_cycle((RAtrcParamstfieldRu((s;lib/python2.7/site-packages/pandas/tests/plotting/common.pyt_unpack_cyclerősN(t__name__t __module__RERGRRJRNRORRZRhRoRvRˆRTRœRŸRŹR˘RQR˝RĎRâRĺ(((s;lib/python2.7/site-packages/pandas/tests/plotting/common.pyR0s. !     > 1 !  9 *talwaysc Ks!ddlj}d}tjƒútj|ƒzÓy|d}Wntk r_|jƒ}nX|jƒ|j d|j dƒƒ}||}t |ƒy#|j dƒ|d<||}Wnt k rŃn Xt |ƒt dtƒ}|j|ƒWdQXWdtj|ƒX|SWdQXdS(Ni˙˙˙˙R¤RYiÓiÔtreturn_filelike(RHRIROtwarningstcatch_warningst simplefiltertKeyErrortgcftclftgett add_subplotRt ExceptionRRtsavefigR5RF(tftfilterwarningstkwargsRJtrettfigRYtpath((s;lib/python2.7/site-packages/pandas/tests/plotting/common.pyt_check_plot_worksüs.        cCs(tjjtjjtƒƒ\}}|S(N(tosRůtsplittabspatht__file__(tptht_((s;lib/python2.7/site-packages/pandas/tests/plotting/common.pytcurpaths$($RűRętnumpyR7RR t pandas.compatRRtpandas.util._decoratorsRtpandas.util._test_decoratorstutilt_test_decoratorsttdtpandas.core.dtypes.apiRtpandasRRtpandas.util.testingttestingR5RRtpandas.plottingR%tpandas.plotting._toolsR RRtskip_if_no_mpltobjectRRúR(((s;lib/python2.7/site-packages/pandas/tests/plotting/common.pyts(      ˙Ě "