Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/plotting/__pycache__/common.cpython-37.pyc
Назад
B �5�g3U � @ s� d Z ddlmZ ddlZddlmZmZ ddlZddlZ ddl m Z ddl m mZ ddlmZ ddlZddlmZmZmZ ddlmZ er�ddlmZ ej G d d � d ��Z dd d�Z dd� Z dd� 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� e Zd ZdZdd� Zdd� Zedd� �Zedd � �Zd9d d �Z d:dd�Z dd� Z d;dd�Z d<dddd�dd�Z dd� Zd=d d �Zd d � Zd>d d!�Zd?d#d$�Zd@d%d&�Zd'd(� Zd)d*� ZdAd,d-�ZdBd.d/�Zi fd0d1�ZdCd3d4�Zd5d6� Zd7d8� Zd S )D� TestPlotBasezE This is a common base class used for various plotting tests c C sT dd l }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��| _W d Q R X t�� | _ttjjdd �tjjdd �t� d�tjjdd � d��| _ d S )Nr )�compatl j�+2Zl �R�e�Z� � )g������@g333333@�left�d �* ZMaleZFemale)�size�A�B�C�B � � � )r �dtype)�gender� classroom�height�weight�category�datetime� )r r r ) � 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 �nr r � r0 �R/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/plotting/common.py� setup_method* sF zTestPlotBase.setup_methodc C s t �� d S )N)r% �close)r, r- r0 r0 r1 �teardown_methoda s z TestPlotBase.teardown_methodc C s dd l m} |S )Nr )�matplotlib.pyplot�pyplot)r, �pltr0 r0 r1 r7 d s zTestPlotBase.pltc C s dd l m} |jS )Nr )Zmatplotlib.colors�colorsZcolorConverter)r, r8 r0 r0 r1 �colorconverterj s zTestPlotBase.colorconverterNTc C sj |r|dkrt d��| �|�}xF|D ]>}|rR|�� dk s<t�| �|�� �� |� q$|�� 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�axr0 r0 r1 �_check_legend_labelsp s z!TestPlotBase._check_legend_labelsc C sT |r|dkrt d��|r@|�� \}}dd� |D �}||ksPt�n|�� 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 | ] }|� � �qS r0 )Z get_marker)�.0�handler0 r0 r1 � <listcomp>� s z5TestPlotBase._check_legend_marker.<locals>.<listcomp>)r: Zget_legend_handles_labelsr= r< )r, rB Zexpected_markersrA Zhandles�_�markersr0 r0 r1 �_check_legend_marker� s z!TestPlotBase._check_legend_markerc sP |� � }|� � }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 c S s | � � }|� � }t�||� d S )N)Z get_xydatar% �assert_almost_equal)�xpl�rslZxpdataZrsdatar0 r0 r1 � check_line� s z,TestPlotBase._check_data.<locals>.check_linec s g | ]\}}� ||��qS r0 r0 )rD rK rL )rM r0 r1 rF � s z,TestPlotBase._check_data.<locals>.<listcomp>N)� get_lines�lenr= �zipr% r3 )r, Zxp�rsZxp_linesZrs_linesr0 )rM r1 � _check_data� s zTestPlotBase._check_datac C sF ddl m} t||�s$t|�s$|g}x |D ]}|�� |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.collectionsrS � isinstancer � get_visibler= )r, � collectionsrA rS �patchr0 r0 r1 �_check_visible� s zTestPlotBase._check_visiblezAxes | Sequence[Axes]�bool�None)r? �filled�returnc C s: | � |�}x*|D ]"}x |jD ]}|j|ks t�q W qW dS )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\ rB rX r0 r0 r1 �_check_patches_all_filled� s z&TestPlotBase._check_patches_all_filledc s* |� � }tt||��� � fdd�|jD �S )Nc s g | ] }� | �qS r0 r0 )rD �v)�mappedr0 r1 rF � s z3TestPlotBase._get_colors_mapped.<locals>.<listcomp>)�unique�dictrP �values)r, Zseriesr8 rb r0 )ra r1 �_get_colors_mapped� s z TestPlotBase._get_colors_mappedc C s� ddl m}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|�k�s t �xht ||�D ]Z\} } t | |��rN| �� d } n| �� } t | tj��rlt| �} | � | �} | | 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 )rS �LineCollection�PolyCollection)�Line2DN)rT rS rf rg Zmatplotlib.linesrh r9 re rO r= rP rU � get_colorZto_rgba�tupleZ get_edgecolorZ get_facecolorr& Zndarray)r, rW Z linecolorsZ facecolors�mappingrS rf rg rh �convrX �color�result�expectedr0 r0 r1 � _check_colors� s< zTestPlotBase._check_colorsc C sd t |�s|�� |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 | ] }|� � �qS r0 )�get_text)rD �tr0 r0 r1 rF . s z3TestPlotBase._check_text_labels.<locals>.<listcomp>N)r rq r= rO rP )r, Ztextsro r@ �label�er0 r0 r1 r> s z TestPlotBase._check_text_labelsc C s& ddl m} | �|�}�x|D � ]�}|dk s4|dk r�t|j�� |�rN|�� }n|�� |jdd� }x<|D ]4} |dk r�t�| � � |� |dk rht�| � � |� qhW |dk s�|dk r t|j �� |�r�|� � }n|� � |j dd� }x>|D ]6} |dk �rt�| � � |� |dk r�t�| � � |� 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.tickerru r; rU �xaxisZget_minor_formatterZget_xticklabelsr% rJ Z get_fontsizeZ get_rotation�yaxisZget_yticklabels) r, r? Z xlabelsizeZxrotZ ylabelsizeZyrotru rB r@ rs r0 r0 r1 �_check_ticks_props3 s* z TestPlotBase._check_ticks_props�linearc C s@ | � |�}x0|D ](}|j�� |ks&t�|j�� |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)r; rw Z get_scaler= rx )r, r? rw rx rB r0 r0 r1 �_check_ax_scalesc s z TestPlotBase._check_ax_scalesc C s� ddl m} |dkr| j}| �|�}|dk r^t|�|ks<t�x |D ]}t|�� �dksBt�qBW |dk r�| �||��}||ks�t�t� |d 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 )� flatten_axesN)r )�!pandas.plotting._matplotlib.toolsr| r$ r; rO r= Z get_children�_get_axes_layoutr% Zassert_numpy_array_equal�figureZget_size_inchesr& �arrayZfloat64) r, r? Zaxes_numZlayoutZfigsizer| Z visible_axesrB rn r0 r0 r1 �_check_axes_shapet s z TestPlotBase._check_axes_shapec C sZ t � }t � }x<|D ]4}|�� �� }|�|d d � |�|d d � qW t|�t|�fS )Nr � )�setZ get_positionZ get_points�addrO )r, r? Zx_setZy_setrB Zpointsr0 r0 r1 r~ � 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 )r| c S s g | ]}|� � r|�qS r0 )rV )rD rB r0 r0 r1 rF � s z1TestPlotBase._flatten_visible.<locals>.<listcomp>)r} r| )r, r? r| r0 r0 r1 r; � s z TestPlotBase._flatten_visibler c C s� | � |�}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_yerrr� N)r; � containers�getattrr= ) r, r? ZxerrZyerrrB r� Z xerr_countZ yerr_count�cr� r� r0 r0 r1 �_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 | �|�D ]}t||�s~t�q~W dS t|t �s�t�t |� � �t |�ks�t�x�|� � D ]�\}} t| || �s�t�|dk�r|�r�| � � |k�s�t�q�|dk�rP|�r*| j� � |k�s*t�t| j|��s<t�t| jt��s�t�q�|dk�r�| d d } | j} |�r�| � � |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 )r )rc r? �bothNrc r� r? Zmedians)�matplotlib.axesr rc rj rU r= rB �linesr; r �sorted�keys�itemsZ get_titler? ) r, �returned� return_typeZ expected_keysZcheck_ax_titler �types�r�key�value�liner? r0 r0 r1 �_check_box_return_type� s> z#TestPlotBase._check_box_return_typec sl dd l }� fdd�}d}�xL|D �]B}� j�ddt|� |� |d7 }|jddd� |jf d |i|�� |� rrt�� j�ddt|� |� |d7 }|jdd d� |jf |dd �|�� |� r�t�|d kr � j�ddt|� |� |d7 }|jdd d� |jf d |i|�� |� �st�� j�ddt|� |� |d7 }|jddd� |jf |d d �|�� |� s t�q W d S ) Nr c sz � j �� j�� } � j �� j�� }� jrLtdd� | D ��}tdd� |D ��}n$tdd� | D ��}tdd� |D ��}|ov| S )Nc s s | ]}|j �� V qd S )N)�gridlinerV )rD �gr0 r0 r1 � <genexpr> s zHTestPlotBase._check_grid_settings.<locals>.is_grid_on.<locals>.<genexpr>c s s | ]}|j �� V qd S )N)r� rV )rD r� r0 r0 r1 r� s c s s | ]}|j V qd S )N)�gridOn)rD r� r0 r0 r1 r� s c s s | ]}|j V qd S )N)r� )rD r� r0 r0 r1 r� s )r7 Zgcarw Zget_major_ticksrx r# �all)ZxticksZyticksZxoffZyoff)r, r0 r1 � is_grid_on s z5TestPlotBase._check_grid_settings.<locals>.is_grid_onr� r r? F)�grid�kindT)r� r� )�pieZhexbinZscatter)r" r7 ZsubplotrO �rcZplotr= )r, �obj�kinds�kwsr. r� Zspndxr� r0 )r, r1 �_check_grid_settings s2 z!TestPlotBase._check_grid_settingsrm c s � fdd�|d D �S )zT Auxiliary function for correctly unpacking cycler after MPL >= 1.5 c s g | ] }|� �qS r0 r0 )rD r` )�fieldr0 r1 rF < s z/TestPlotBase._unpack_cycler.<locals>.<listcomp>zaxes.prop_cycler0 )r, ZrcParamsr� r0 )r� r1 �_unpack_cycler8 s zTestPlotBase._unpack_cyclerc C s | j �� r|jd S |jS )N�x)r � mpl_ge_3_5_0� _shared_axesZ_shared_x_axes)r, rB r0 r0 r1 � get_x_axis> s zTestPlotBase.get_x_axisc C s | j �� r|jd S |jS )N�y)r r� r� Z_shared_y_axes)r, rB r0 r0 r1 � get_y_axisA s zTestPlotBase.get_y_axis)NT)NT)T)T)NNN)NNNN)rz rz )NNN)r r )NT)rm ) �__name__� __module__� __qualname__�__doc__r2 r4 r r7 r9 rC rI rR rY r_ re rp r> ry r{ r� r~ r; r� r� r� r� r� r� r0 r0 r0 r1 r $ s0 7 A / $ ; 0 r �alwaysFc K s� ddl m} |rt}nt}d}t�� �� t�|� z�y`|�d|�� �}|� � x || |f|�D ]}t � |� q^W t j dd��}|� |� W dQ R X W n& tk r� } z| �W dd} ~ X Y nX W dt �|� X |S Q R X dS )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. r Nr T)Zreturn_filelike)r5 r6 �_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�kwargsr7 Z gen_plots�ret�fig�path�errr0 r0 r1 �_check_plot_worksE s$ r� c k s | f |�V dS )z' Create plot in a default way. Nr0 )r� r� r� r0 r0 r1 r� s r� c k sV d|kr|� d� | f |�V | tjjkr8d|ksFt�n|� d�|d< | f |�V dS )z9 Create plot on two subplots forcefully created. rB �� �� N)Z add_subplot�pdZplottingZbootstrap_plotr= )r� r� r� r0 r0 r1 r� � s r� c C s t j�t j�t��\} }| S )N)�osr� �split�abspath�__file__)�pthrG r0 r0 r1 �curpath� s r� )r� F)"r� � __future__r r� �typingr r r� �numpyr& Zpandas.util._decoratorsr Z pandas.util._test_decorators�utilZ_test_decorators�tdZpandas.core.dtypes.apir Zpandasr� r r r Zpandas._testingZ_testingr% r� r Zskip_if_no_mplr r� r� r� r� r0 r0 r0 r1 �<module> s, % :
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка