Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/plotting/frame/__pycache__/test_frame_legend.cpython-37.pyc
Назад
B �5�g� � @ sH d dl Zd dlZd dlmZmZ d dlmZ ejj Z G dd� de�Z dS )� N)� DataFrame� date_range)� TestPlotBasec @ sZ e Zd Zejjdd�dd� �Zdd� Zdd� Zd d � Z d d � Z d d� Z dd� Z dd� Z dS )�TestFrameLegendzLOpen bug in matplotlib https://github.com/matplotlib/matplotlib/issues/11357)�reasonc C s� ddl m} ddlm} tdddd�dddd�g�}|jdd d d d d �}|jdd dd |dd� |�� }|j}t|d |�s~t �t|d |�s�t �d S )Nr )�LineCollection)�Line2D� )�x�a�b� � r r Zorangeg�������?)�c�yerr�labelr �blue)r r �axr ) Zmatplotlib.collectionsr Zmatplotlib.linesr r �plot� get_legend� legendHandles� isinstance�AssertionError)�selfr r �dfr �legendZresult_handles� r �c/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/plotting/frame/test_frame_legend.py�test_mixed_yerr s z TestFrameLegend.test_mixed_yerrc C s� t ddgddgd��}t dddgi�}|jddd d�d d �}|jddd i|d � |�� }dd� |jD �}dd d g}||ks~t�d S )Nr r r )r r �dg @Tr �greenr )r �color� secondary_y�red)r r! r c S s g | ] }|� � �qS r )� get_color)�.0�handler r r � <listcomp>, s z5TestFrameLegend.test_legend_false.<locals>.<listcomp>)r r r r r )r r �df2r r �result�expectedr r r �test_legend_false$ s z!TestFrameLegend.test_legend_falsec C sv ddddddg}t tj�dd�dd d gd �}t tj�dd�d d dgd �}t tj�dd�dddgd �}t tj�dd�dddgd �}x�|D ]�}|j|dd�}| j||jd� |j|d|d�}| j||jd� |j|d|d�}| j||j�|j�d� |j|d|d�}t|j�|j��tt |j�� }| j||d� q�W |jdd d�}| j|dd d gd� |jd|d �}| j|dd d gd� |jddd|d �}| j|dd d dd dgd� t d dd!�} t tj� dd�dd d g| d"�}t tj� dd�d d dg| d"�}t tj� dd�dddg| d"�}|jdd d�}| j|dd d gd� |jd|d �}| j|dd d gd� |jd|d �}| j|dd d dddgd� |jj dd d#d$�}| j|d#gd� |jj d d dd%|d&�}| j|d#gd� |jj ddd'|d(�}| j|d#d'gd� |� d�} | jd d)�}| j|d gd� | jd d*d+�}| j|d*gd� | �|j�� d� | jd d,|d-�}| j|d*d,gd� | j�� d d gk�srt�d S ).N�line�bar�barh�kde�area�histr r r r )�columnsr �e�f�g�h�i�j�k�lT)�kindr )�labelsF)r; r r �reverse)r r" z b (right))r r )r; r r" r z h (right)z1/1/2014)Zperiods)r2 �indexZdata1)r �yr Zdata2)r r? r r r Zdata3)r r? r r )r? ZLABEL_b)r? r ZLABEL_c)r? r r )r �np�random�randr �_check_legend_labelsr2 �union�list�reversedr �randnZscatterZ set_index�_check_text_labelsZxaxisZ get_label�tolistr ) r �kindsr r( Zdf3Zdf4r; r r* �indZdf5r r r �test_df_legend_labels0 s^ z%TestFrameLegend.test_df_legend_labelsc C s� t ddddgddddggddddgd �}| jjdd d �\}}|jddd d d|d d� |jddddd|d d� |jdddd d|d d� | j|d dddgd� | j|d d dd gd� |jdddd d|d d� |jddd d d|d d� |jddddd|d d� | j|d dddgd� | j|d d d dgd� |jddddd|d d� |jdddd d|d d� |jddd d d|d d� | j|d dddgd� | j|d dd d gd� d S )Nr r � � r �rr5 r )�datar2 r )ZnrowsZncolsr �o)r r? Z linewidth�markerr! r )r< )�expected_markers)r �plt�subplotsr rC �_check_legend_marker)r r �figr r r r �*test_missing_marker_multi_plots_on_same_axq s" (z:TestFrameLegend.test_missing_marker_multi_plots_on_same_axc C s� t tj�dd�t�ddddg�t�ddddg�gd�}ddg|j_|�� }|j� � }| � |d � t tj�d d ��}|jd |d �}|j� � }| � |d � d |j_ |jd|d �}|j� � }| � |d � |jd |d �}|j� � }| � |d � d S )NrM r r r r? )r2 �groupZ individualzgroup,individual� T)r r �newF) r r@ rA rG �arrayr2 �namesr Zlegend_Z get_titlerH �name)r �multir Z leg_titler r r r �test_legend_name� s$ ( z TestFrameLegend.test_legend_namec C sZ ddddddg}t tj�dd�dd d gd �}x(|D ] }|j|d d �}| j|d d� q2W d S )Nr, r- r. r/ r0 r1 r r r r )r2 F)r; r )�visible)r r@ rA rB r rC )r rJ r r; r r r r �test_no_legend� s z TestFrameLegend.test_no_legendc C s� t tj�dd�dddgd�}|jdgddd �}|jdgd d |d � |jdgd d |d � | j|dddgd� | j|dd d gd� d S )NrN r �A�B�C)r2 r Zsolid)r? rR � linestylerQ Zdotted)r? rR rf r �<)r< )rS )r r@ rA rG r rC rV )r r r r r r �test_missing_markers_legend� s z+TestFrameLegend.test_missing_markers_legendc C s� t ddddddgddddddgddddddgddddddgd��}| j�� \}}x dD ]}|jd |||d d � qVW | j|d d dgd� | j|d d d gd� d S )Nr r r rM rZ � )rc rd re �X�ABCrj �.)r r �stylerc rd re )r< )rS )r rT rU r rC rV )r r rW r r; r r r �'test_missing_markers_legend_using_style� s z7TestFrameLegend.test_missing_markers_legend_using_styleN)�__name__� __module__� __qualname__�pytest�markZxfailr r+ rL rX r` rb rh rn r r r r r s A r ) �numpyr@ rr Zpandasr r Z pandas.tests.plotting.commonr rs ZslowZ pytestmarkr r r r r �<module> s
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка