o gA@s:dZddlmZddlmZddlZddlmZddl Z ddl m Z ddl m Zer6ddlmZddlmZd>d d Zd>d d ZddZd?ddZd?d@ddZddZdAddZddZdBd d!ZdCd#d$ZdAd%d&ZdDd(d)ZdEd*d+Z d>d,d-Z ifd.d/Z!dFd1d2Z"d3d4Z#d5d6Z$dGd8d9Z%d:d;Z&d@z(_check_legend_marker..)r get_legend_handles_labelsr )rexpected_markersrhandles_markersrrr_check_legend_marker/s  r c Csnddlm}|}|}t|t|ksJt||D]\}}|}|}t||q|ddS)z Check each axes has identical lines Parameters ---------- xp : matplotlib Axes object rs : matplotlib Axes object rNall) matplotlib.pyplotpyplot get_lineslenzip get_xydatatmassert_almost_equalclose) xprspltxp_linesrs_linesxplrslxpdatarsdatarrr _check_dataFs r4cCsBddlm}t||st|s|g}|D] }||ksJqdS)z 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.collectionsr5 isinstancer get_visible) collectionsrr5patchrrr_check_visible]s r;rAxes | Sequence[Axes]filledboolreturnNonecCs0t|}|D]}|jD] }|j|ksJq qdS)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=rr:rrr_check_patches_all_filledqs   rCcs*|}tt||fdd|jDS)Ncsg|]}|qSrrrvmappedrrrrz&_get_colors_mapped..)uniquedictr&values)seriescolorsrHrrFr_get_colors_mappedsrMcCsddlm}ddlm}m}m}ddlm}|j} |durp|dur.t ||}|dt |}t |t |ks8Jt ||D]2\} } t | |rP| } | | } nt | ||fr`t| d} n| } | | } | | ksoJq=|dur|durt ||}|dt |}t |t |ksJt ||D]+\} } t | |r| d} n| } t | tjrt| } | | } | | ksJqdSdS)a 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)rL)r5LineCollectionPolyCollection)Line2DN) matplotlibrLr6r5rNrOmatplotlib.linesrPColorConverterrMr%r&r7 get_colorto_rgbatuple get_edgecolor get_facecolornpndarray)r9 linecolors facecolorsmappingrLr5rNrOrPconvr:colorresultexpectedrrr _check_colorssB          rbcCsbt|s||ks JdSdd|D}t|t|ksJt||D] \}}||ks.Jq$dS)a 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 cSrr)get_text)rtrrrrrz&_check_text_labels..N)rrcr%r&)textsrarlabelerrrr s r c Csddlm}t|}|D]z}|dus|durJt|j|r#|}n ||jdd}|D]}|dur=t| ||durIt| |q/|dusR|durt|j |r_| }n | |j dd}|D]}|duryt| ||durt| |qkq dS)ac 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.tickerrhr r7xaxisget_minor_formatterget_xticklabelsr(r) get_fontsize get_rotationyaxisget_yticklabels) r xlabelsizexrot ylabelsizeyrotrhrrrfrrr_check_ticks_propss0   rvlinearcCs:t|}|D]}|j|ksJ|j|ksJqdS)z 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 rk get_scalerp)rrkrprrrr_check_ax_scales s  ryc Csddlm}|dur d}t|}|dur+t||ksJ|D] }t|dks*Jq|durbt}t}||D]}|} || dd|| ddq9t|t|f} | |ksbJt |dj t j|t jddS)a 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 r flatten_axesN)g@g333333@)dtype)!pandas.plotting._matplotlib.toolsr{r r% get_childrenset get_position get_pointsaddr(assert_numpy_array_equalfigureget_size_inchesrYarrayfloat64) raxes_numlayoutfigsizer{ visible_axesrx_sety_setpointsr`rrr_check_axes_shapes*     rSequence[Axes]cCs&ddlm}||}dd|D}|S)z Flatten axes, and filter only visible Parameters ---------- axes : matplotlib Axes object, or its list-like rrzcSsg|]}|r|qSr)r8)rrrrrrWsz$_flatten_visible..)r~r{)rr{ axes_ndarrayrrrr Ks r c Csvt|}|D]2}|j}d}d}|D]}t|dd}t|dd} |r%|d7}| r+|d7}q||ks2J||ks8JqdS)z 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_yerrr|N)r containersgetattr) rxerryerrrr xerr_count yerr_countcrrrrr_check_has_errorbars[s     rc Cstddlm}t|td}|dur7|durd}t|||sJ|dkr3t|j|s+Jt|jts5JdSdS|durKt|D] }t||sHJq?dSt|tsRJt | t |ks^J| D]U\}}t|||soJ|dkr~|r}| |ks}Jqb|dkr|r|j |ksJt|j|sJt|jtsJqb|dkr|dd} | j } |r| |ksJqbtdS) a< 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)rIrbothNrIrrmedians)matplotlib.axesrrIrVr7rlinesr rsortedkeysitems get_titlerAssertionError) returned return_type expected_keyscheck_ax_titlertypesrkeyvaluelinerrrr_check_box_return_typewsH    rcsddlfdd}d}|D]}jddt|||d7}jddd|jd d |i||r6Jjjddt|||d7}jdd d|jd |dd ||rbJj|d vrˆjddt|||d7}jdd d|jd d |i||sJjjddt|||d7}jddd|jd |d d ||sJjqdS)NrcsNjj}jj}tdd|D}tdd|D}|o%| S)Ncs|] }|j VqdSNgridliner8rgrrr z;_check_grid_settings..is_grid_on..csrrrrrrrrr)r#gcarkget_major_ticksrpr!)xticksyticksxoffyoffmplrr is_grid_ons  z(_check_grid_settings..is_grid_onr|rF)gridkindT)rr)piehexbinscatterr)rQr#subplotr%rcplotclf)objkindskwsrspndxrrrr_check_grid_settingss>         rr_csfdd|dDS)zL Auxiliary function for correctly unpacking cycler after MPL >= 1.5 csg|]}|qSrrrDfieldrrrrz"_unpack_cycler..zaxes.prop_cycler)rcParamsrrrr_unpack_cyclersrcC |jdS)Nx _shared_axesrrrr get_x_axis rcCr)Nyrrrrr get_y_axisrrFc Ksvddlm}|r t}nt}d}z%|d|}||||fi|D]}t|q%W| ||S| |w)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. rNr) r"r#_gen_default_plot_gen_two_subplotsgetgcfrr("assert_is_valid_plot_return_objectr*)f default_axeskwargsr- gen_plotsretfigrrr_check_plot_workss    rcks|di|VdS)z' Create plot in a default way. Nrrrrrrrrr srcks`d|vr |d|di|V|tjjurd|vsJn|d|d<|di|VdS)z9 Create plot on two subplots forcefully created. rNr) add_subplotpdplottingbootstrap_plotrrrrr's  r)NT)T)rr<r=r>r?r@)NNN)NNNN)rwrw)rr<r?r)rr)r_)F)(__doc__ __future__rtypingrnumpyrYpandas.core.dtypes.apirpandasrrpandas._testing_testingr(collections.abcrrrrr r4r;rCrMrbr rvryrr rrrrrrrrrrrrrsB            @  /  ,  = - 0