U
|
ý°dß" ã @ s^ d dl Zd dlZd dlm mZ d dlmZm Z d dl
|
m Z d dl m Z G dd de ZdS )é N)Ú DataFrameÚ
|
date_range)Ú TestPlotBase)ÚVersionc
|
@ s e Zd Zejjdddd Zdd Zej dd Z
|
d d
|
Z d d Z ej d dddejdej dddg¡dd Zdd Zdd ZdS ) ÚTestFrameLegendzLOpen bug in matplotlib https://github.com/matplotlib/matplotlib/issues/11357)Úreasonc C s¶ dd l }ddlm} ddlm} tddddddddg}|jdd d
|
d d
|
d }|jdd dd |dd | ¡ }t|j tdk r|j
|
}n|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 ú3.7)Ú
|
matplotlibZmatplotlib.collectionsr Zmatplotlib.linesr r ÚplotÚ
|
get_legendr Ú __version__Ú legendHandlesÚlegend_handlesÚ
|
isinstanceÚAssertionError)ÚselfÚmplr r Údfr ÚlegendZresult_handles© r" údd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/plotting/frame/test_frame_legend.pyÚtest_mixed_yerr s z TestFrameLegend.test_mixed_yerrc C s¨ dd l }tddgddgd}tdddgi}|jdd d
|
dd d }|jddd i|d | ¡ }t|jtdk rz|j}n|j}dd |D }d d
|
d g}||ks¤td S )Nr r
|
r r )r r Údg @Tr Úgreenr )r! ÚcolorÚ secondary_yÚred)r! r' r r c S s g | ] }| ¡ qS r" )Z get_color)Ú.0Úhandler" r" r# Ú
|
<listcomp>7 s z5TestFrameLegend.test_legend_false.<locals>.<listcomp>) r r r r r r r r r ) r r r Údf2r r! ZhandlesÚresultÚexpectedr" r" r# Útest_legend_false) s
|
z!TestFrameLegend.test_legend_falsec C sr ddddddg}t tj dd¡dd d
|
gd }t tj dd¡d d dgd }t tj dd¡dddgd }t tj dd¡dddgd }|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|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
|
gksntd S ).NÚlineÚbarÚbarhÚkdeÚareaÚhistr r r r ©Úcolumnsr% ÚeÚfÚgÚhÚiÚjÚkÚlT©Úkindr! ©ÚlabelsF)rB r! r Úreverse)r! r( z b (right)©r! r )rB r! r( r z h (right)z1/1/2014)Zperiods)r8 ÚindexZdata1)r Úyr Zdata2)r rH r! r r Zdata3)r rH r r )rH ZLABEL_b)rH r ZLABEL_c)rH r r )r ÚnpÚrandomÚrandr Ú_check_legend_labelsr8 ÚunionÚlistÚreversedr ÚrandnZscatterZ set_indexÚ_check_text_labelsZxaxisZ get_labelÚtolistr ) r Úkindsr r- Zdf3Zdf4rB r r/ ÚindZdf5r" r" r# Útest_df_legend_labels; sb ÿ
|
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 Úrr; r )Údatar8 r )ZnrowsZncolsr Úo)r rH Z linewidthÚmarkerr' r rC ©Zexpected_markers©r ZpltZsubplotsr rL Ú_check_legend_marker)r r Úfigr r" r" r# Ú*test_missing_marker_multi_plots_on_same_ax| 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 )NrV r r r rH r7 ÚgroupZ
|
individualzgroup,individualé TrF ÚnewF) r rI rJ rP Úarrayr8 Únamesr Zlegend_Z get_titlerQ Úname)r Zmultir Z leg_titler r" r" r# Útest_legend_name s& "þ
|
|
|
|
z TestFrameLegend.test_legend_namerB r1 r2 r3 r4 )Zmarksr5 r6 c C s< t tj dd¡dddgd}|j|dd}| j|dd d S ) Nr r r r r7 FrA )Zvisible)r rI rJ rK r rL )r rB 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 )NrW r ÚAÚBÚCr7 r Zsolid)rH r[ Ú linestylerZ Zdotted)rH r[ rl r ú<rC r\ )r rI rJ rP r rL r^ )r r r r" r" r# Útest_missing_markers_legend¼ s z+TestFrameLegend.test_missing_markers_legendc
|
C s t ddddddgddddddgddddddgddddddgd}| j ¡ \}}dD ]}|jd |||d
|
d qT| j|d d dgd | j|d
|
d
|
d
|
gd d S )Nr
|
r r rV rb é )ri rj rk ÚXÚABCrp Ú.)r r Ústyleri rj rk rC r\ r] )r r r_ r rB r" r" r# Ú'test_missing_markers_legend_using_styleÆ s üÿ z7TestFrameLegend.test_missing_markers_legend_using_styleN)Ú__name__Ú
|
__module__Ú __qualname__ÚpytestÚmarkZxfailr$ r0 ÚtdZskip_if_no_scipyrU r` rg Z parametrizeÚparamrh rn rt r" r" r" r# r s, þ
|
|
@úþ
|
|
r )ÚnumpyrI rx Z pandas.util._test_decoratorsÚutilZ_test_decoratorsrz Zpandasr r Z pandas.tests.plotting.commonr Zpandas.util.versionr r r" r" r" r# Ú<module> s
|
