3
���hCI � @ s� d dl Z d dlZd dlZd dlmZ d dlmZ d dljj Z
d dl m Z d dl
Zd dl
mZmZ d dljZe
jG dd� d��Zd
d d
�Zd d � ZdS )� N)�random)�cache_readonly)� is_list_like)� DataFrame�Seriesc @ s� e Zd Zdd� Zdd� Zedd� �Zedd� �Zd.d d �Zd/d
d�Z dd� Z
d0dd�Z dd� Z d1dd�Z
dd� Zd2dd�Zd3d d �Zd4d d �Zd d!� Zd"d#� Zd5d%d&�Zd6d'd(�Zi fd)d*�Zd7d,d-�Zd S )8� TestPlotBasec C s dd l }ddlm} |j� |j� | _|j� | _|j� | _ |j
� | _ d| _ d| _
d| _d| _d}tjd ��h tjjd
d g|d �}tjjd
ddg|d �}t||tjdd|d �tjdd|d �tjd|d �d��| _W d Q R X tj� | _ttjjdd �tjjdd �tj d�tjjdd � d��| _ d S )Nr )�compat� � 皙����@�333333@�left�d �* ZMaleZFemale)�size�A�B�C�B � � � )�gender� classroom�heightZweight�category� )r r r )r r ) �
matplotlibZpandas.plotting._matplotlibr Z
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_factor�default_figsizeZdefault_tick_position�tmZ
RNGContext�npr �choicer Znormal�randintZhist_dfZmakeTimeDataFrameZtdf�uniformZarangeZ hexbin_df)�self�method�mplr �nr r � r) �>/tmp/pip-build-5_djhm0z/pandas/pandas/tests/plotting/common.py� setup_method s4
zTestPlotBase.setup_methodc C s t j� d S )N)r �close)r% r& r) r) r* �teardown_methodA s z TestPlotBase.teardown_methodc C s dd l j}|S )Nr )�matplotlib.pyplot�pyplot)r% �pltr) r) r* r0 D s
zTestPlotBase.pltc C s dd l j}|jS )Nr )Zmatplotlib.colors�colorsZcolorConverter)r% r1 r) r) r* �colorconverterJ s
zTestPlotBase.colorconverterNTc C sj |r|dkrt d��| j|�}xF|D ]>}|rR|j� dk s<t�| j|j� j� |� q$|j� dks$t�q$W dS )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�AssertionError�_check_text_labelsZ get_texts)r% �axes�labels�visible�axr) r) r* �_check_legend_labelsP s
z!TestPlotBase._check_legend_labelsc C sT |r|dkrt d��|r@|j� \}}dd� |D �}||ksPt�n|j� dksPt�dS )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 Truec S s g | ] }|j � �qS r) )Z
get_marker)�.0�handler) r) r* �
<listcomp>x s z5TestPlotBase._check_legend_marker.<locals>.<listcomp>)r3 Zget_legend_handles_labelsr6 r5 )r% r; Zexpected_markersr: Zhandles�_�markersr) r) r* �_check_legend_markerg s
z!TestPlotBase._check_legend_markerc sP |j � }|j � }dd� � t|�t|�ks,t�� fdd�t||�D � tj� dS )z�
Check each axes has identical lines
Parameters
----------
xp : matplotlib Axes object
rs : matplotlib Axes object
c S s | j � }|j � }tj||� d S )N)Z
get_xydatar �assert_almost_equal)�xpl�rslZxpdataZrsdatar) r) r* �
check_line� s z,TestPlotBase._check_data.<locals>.check_linec s g | ]\}}� ||��qS r) r) )r= rD rE )rF r) r* r? � s z,TestPlotBase._check_data.<locals>.<listcomp>N)� get_lines�lenr6 �zipr r, )r% Zxp�rsZxp_linesZrs_linesr) )rF r* � _check_data} s zTestPlotBase._check_datac C sJ ddl m} t||� r(t|� r(|g}x |D ]}|j� |ks.t�q.W dS )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: rL �patchr) r) r* �_check_visible� s
zTestPlotBase._check_visiblec s* |j � }tt||��� � fdd�|jD �S )Nc s g | ] }� | �qS r) r) )r= �v)�mappedr) r* r? � s z3TestPlotBase._get_colors_mapped.<locals>.<listcomp>)�unique�dictrI �values)r% Zseriesr1 rU r) )rT r* �_get_colors_mapped� s z TestPlotBase._get_colors_mappedc C s� ddl m}m}m} ddlm} | j} |dk r�|dk rR| j||�}|dt|�� }t|�t|�ksft �xrt
||�D ]d\}
} t |
|�r�|
j � } | j
| �} n(t |
||f�r�t|
j� d �} n|
j� } | j
| �}
| |
ksrt �qrW |dk �r�|dk �r
| j||�}|dt|�� }t|�t|�k�s t �xht
||�D ]Z\}
} t |
|��rN|
j� d } n|
j� } t | tj��rlt| �} | j
| �}
| |
k�s,t ��q,W dS )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 )rL �LineCollection�PolyCollection)�Line2DN)rM rL rY rZ Zmatplotlib.linesr[ r2 rX rH r6 rI rN Z get_colorZto_rgba�tupleZ
get_edgecolorZ
get_facecolorr! Zndarray)r% rP Z
linecolorsZ
facecolors�mappingrL rY rZ r[ �convrQ �color�result�expectedr) r) r* �
_check_colors� s<
zTestPlotBase._check_colorsc C sd t |�s|j� |ks`t�nFdd� |D �}t|�t|�ks<t�x"t||�D ]\}}||ksHt�qHW 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
c S s g | ] }|j � �qS r) )�get_text)r= �tr) r) r* r? � s z3TestPlotBase._check_text_labels.<locals>.<listcomp>N)r rc r6 rH rI )r% Ztextsra r9 �label�er) r) r* r7 � s z TestPlotBase._check_text_labelsc
C s& ddl m} | j|�}�x|D � ]�}|dk s4|dk r�t|jj� |�rN|j� }n|j� |jdd� }x<|D ]4} |dk r�tj| j � |� |dk rhtj| j
� |� qhW |dk s�|dk r t|j j� |�r�|j � }n|j � |j dd� }x>|D ]6} |dk �rtj| j � |� |dk r�tj| j
� |� q�W q W dS )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.tickerrg r4 rN �xaxisZget_minor_formatterZget_xticklabelsr rC Z get_fontsizeZ get_rotation�yaxisZget_yticklabels)
r% r8 Z
xlabelsizeZxrotZ
ylabelsizeZyrotrg r; r9 re r) r) r* �_check_ticks_props� s*
z TestPlotBase._check_ticks_props�linearc C s@ | j |�}x0|D ](}|jj� |ks&t�|jj� |kst�qW dS )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)r4 ri Z get_scaler6 rj )r% r8 ri rj r; r) r) r* �_check_ax_scales- s
z TestPlotBase._check_ax_scalesc C s� ddl m} |dkr| j}| j|�}|dk r^t|�|ks<t�x |D ]}t|j� �dksBt�qBW |dk r�| j||��}||ks�t�tj |d j
j � t j
|t jd�� dS )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 )�_flattenN)Zdtype)�!pandas.plotting._matplotlib.toolsrn r r4 rH r6 Z get_children�_get_axes_layoutr Zassert_numpy_array_equal�figureZget_size_inchesr! �arrayZfloat64) r% r8 Zaxes_numZlayoutZfigsizern Z visible_axesr; r` r) r) r* �_check_axes_shape> s
z TestPlotBase._check_axes_shapec C sZ t � }t � }x<|D ]4}|j� j� }|j|d d � |j|d d � qW t|�t|�fS )Nr � )�set� get_positionZ
get_points�addrH )r% r8 Zx_setZy_setr; Zpointsr) r) r* rp b s
z TestPlotBase._get_axes_layoutc C s& ddl m} ||�}dd� |D �}|S )z�
Flatten axes, and filter only visible
Parameters
----------
axes : matplotlib Axes object, or its list-like
r )rn c S s g | ]}|j � r|�qS r) )rO )r= r; r) r) r* r? x s z1TestPlotBase._flatten_visible.<locals>.<listcomp>)ro rn )r% r8 rn r) r) r* r4 l s z TestPlotBase._flatten_visibler c C s� | j |�}xp|D ]h}|j}d}d}x<|D ]4}t|dd�} t|dd�}
| rP|d7 }|
r(|d7 }q(W ||kslt�||kst�qW dS )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_yerrrt N)r4 �
containers�getattrr6 ) r% r8 ZxerrZyerrr; rz Z
xerr_countZ
yerr_count�crx ry r) r) r* �_check_has_errorbars{ s
z!TestPlotBase._check_has_errorbarsc C s� ddl m} t|td�}|dkrj|dkr,d}t||| �s>t�|dkrft|j|�sVt�t|jt�sft��n$|dkr�x | j|�D ]}t||�s~t�q~W dS t|t �s�t�t
|j � �t
|�ks�t�x�|j � D ]�\}} t| || �s�t�|dk�r|�r�| j
� |k�s�t�q�|dk�rP|�r*| jj
� |k�s*t�t| j|��s<t�t| jt��s�t�q�|dk�r�| d d }
|
j} |�r�| j
� |k�s�t�q�t�q�W dS ) 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.
r )�Axes)rV r8 �bothNrV r r8 Zmedians)Zmatplotlib.axesr~ rV r\ rN r6 r; �linesr4 r �sorted�keys�itemsZ get_titler8 ) r% Zreturned� return_typeZ
expected_keysZcheck_ax_titler~ �types�r�key�value�liner8 r) r) r* �_check_box_return_type� s>
z#TestPlotBase._check_box_return_typec sp dd l }� fdd�}d}�xP|D �]F}� jjddt|� |� |d7 }|jddd� |jf d |i|�� |� stt�� jjddt|� |� |d7 }|jdd
d� |jf |dd �|�� |� s�t�|d kr � jjddt|� |� |d7 }|jdd
d� |jf d |i|�� |� �st�� jjddt|� |� |d7 }|jddd� |jf |d
d �|�� |� s t�q W d S )
Nr c sz � j j� jj� } � j j� jj� }� jrLtdd� | D ��}tdd� |D ��}n$tdd� | D ��}tdd� |D ��}|ov| S )Nc s s | ]}|j j� V qd S )N)�gridlinerO )r= �gr) r) r* � <genexpr>� s zHTestPlotBase._check_grid_settings.<locals>.is_grid_on.<locals>.<genexpr>c s s | ]}|j j� V qd S )N)r� rO )r= r� r) r) r* r� � s c s s | ]}|j V qd S )N)�gridOn)r= r� r) r) r* r� � s c s s | ]}|j V qd S )N)r� )r= r� r) r) r* r� � s )r0 Zgcari Zget_major_ticksrj r �all)ZxticksZyticksZxoffZyoff)r% r) r* �
is_grid_on� s z5TestPlotBase._check_grid_settings.<locals>.is_grid_onrt r r8 F)�grid�kindT)r� r� Zpie)r r0 ZsubplotrH �rcZplotr6 )r% �objZkinds�kwsr' r� Zspndxr� r) )r% r* �_check_grid_settings� s2 z!TestPlotBase._check_grid_settingsr_ c s � fdd�|d D �S )zT
Auxiliary function for correctly unpacking cycler after MPL >= 1.5
c s g | ] }|� �qS r) r) )r= rS )�fieldr) r* r? s z/TestPlotBase._unpack_cycler.<locals>.<listcomp>zaxes.prop_cycler) )r% ZrcParamsr� r) )r� r* �_unpack_cycler s zTestPlotBase._unpack_cycler)NT)NT)T)NNN)NNNN)rl rl )NNN)r r )NT)r_ )�__name__�
__module__� __qualname__r+ r- r r0 r2 r< rB rK rR rX rb r7 rk rm rs rp r4 r} r� r� r� r) r) r) r* r s( *
=
/
$
; 0r �alwaysc K s� dd l j}d }tj� �� tj|� z�y |d }W n tk
rL |j� }Y nX |j� |jd|j d�� | f |�}t
j |� | t j
jkr�d|ks�t�n|j d�|d< | f |�}t
j |� t
jdd��}|j|� W d Q R X W d t
j|� X |S Q R X d S )Nr rq r; �� �� T)Zreturn_filelike)r. r/ �warnings�catch_warnings� simplefilter�KeyErrorZgcfZclf�getZ add_subplotr Z"assert_is_valid_plot_return_object�pdZplottingZbootstrap_plotr6 Z ensure_cleanZsavefigr, )�f�filterwarnings�kwargsr0 �retZfig�pathr) r) r* �_check_plot_works s,
r� c C s t jjt jjt��\} }| S )N)�osr� �split�abspath�__file__)�pthr@ r) r) r* �curpath, s r� )r� )r� r� Znumpyr! r Zpandas.util._decoratorsr Z pandas.util._test_decorators�utilZ_test_decorators�tdZpandas.core.dtypes.apir Zpandasr� r r Zpandas._testingZ_testingr Zskip_if_no_mplr r� r� r) r) r) r* �<module> s
w
# |