U
|
ý°dm ã @ s¨ d Z ddlZddlZddlZddlZddlmZ ddlm Z ddl
|
m m Z ddlZddlmZmZmZ ddlmZ ddlmZ ddlmZ e jG dd d eZdS )
|
z Test cases for DataFrame.plot é N)Úis_platform_linux)Únp_version_gte1p24)Ú DataFrameÚSeriesÚ
|
date_range)Ú TestPlotBase)Ú pprint_thingc @ s e Zd Zejjdd Zdd Zdd Zejj ddd d
|
Z
|
d d Z ej d i dddfddidddfg¡dd Z ejjdd Zdd Zdd Zdd Zd d Zejj eo²e d d d!d"d# Zejj eoÒe d d d!d$d% Zd&d' Zd(d) Zd*d+ Zd,d- Zej d.d/d0d1d2d3d4dd5gfg¡ej d6d7d8d9g¡d:d; Zej d<d9d=d>d9d=d?d@dAd=d>dAd=d?d@d9d d>d9d d?d@dAd d>dAd d?d@d9d=dBd9d=d?dCdAd=dBdAd=d?dCd9d=dDdEd9d=d?dDdFdAd=dDdEdAd=d?dDdFd9d dDdEd9d d?dDdFdAd dDdEdAd d?dDdFd9d=dDdGd9d=d?dDdHdAd=dDdGdAd=d?dDdHg¡dIdJ Z ej d<d9d d>d9d=d>dAd d>dAd=d>d9d=dBdAd=dBg¡dKdL Z ej d<d9d d>d9d=d>dAd d>dAd=d>d9d=dBdAd=dBg¡dMdN Z ej dOddPg¡dQdR Z dSdT Z dYdWdXZ!d3S )ZÚTestDataFramePlotsSubplotsc C sð t tj dd¡ttjd d
d}dD ]À}|j|dddd}| j|ddd |j d ks^t
|
t ||j D ] \}}| j |t|gd
|
qj|d d
D ]b}| |j¡ | j| ¡ d d |dkrÔ| j|jddd d | j|j ¡ d d | | ¡ ¡ q| |d j¡ | |d ¡ ¡ | |d jdd¡ | |d j ¡ ¡ | |d ¡ ¡ |j|dd d}|D ]P}| |j¡ | | ¡ ¡ | |jdd¡ | |j ¡ ¡ | | ¡ ¡ qj|j|dd d}|D ]}| ¡ d ksÐt
|
qÐq(d S )Né
|
é ©Úindex)ÚbarÚbarhÚlineÚareaT)ÚkindÚsubplotsÚsharexÚlegend©r é ©Zaxes_numÚlayout©r ©ÚlabelséþÿÿÿF©Zvisibler ©Úminoréÿÿÿÿ©r r r )r r r )r ÚnpÚrandomÚrandÚlistÚstringÚ ascii_lettersÚplotÚ_check_axes_shapeÚshapeÚAssertionErrorÚzipÚcolumnsÚ_check_legend_labelsr Ú_check_visibleÚxaxisÚget_xticklabelsÚ get_labelÚget_yticklabelsZ
|
get_legend)ÚselfÚdfr ÚaxesÚaxÚcolumn© r: úfd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/plotting/frame/test_frame_subplots.pyÚ test_subplots s: $
|
z(TestDataFramePlotsSubplots.test_subplotsc C s¤ t dddd}ttj dd¡|d}dD ]t}|j|ddd }| j|dd
|
d |d d
D ]Z}| |j¡ | j| ¡ d d | j|j ddd d | j|j
|
¡ d d | | ¡ ¡ qZ| |d j¡ | |d ¡ ¡ | |d j dd¡ | |d j
|
¡ ¡ | |d ¡ ¡ | j |dd |j|dd ddd}|D ]b}| |j¡ | | ¡ ¡ | |j dd¡ | |j
|
¡ ¡ | | ¡ ¡ | j |dddd q:q(d S )Nú
|
2014-07-01ÚMr
|
©ÚstartÚfreqÚperiodsr r )r r Tr" r r r Fr r r! r )Úxroté- é )r r r ZrotÚfontsize)Z
|
xlabelsizerC Z
|
ylabelsize) r r r# r$ r% r) r* r0 r1 r2 r3 r4 Z_check_ticks_props)r5 Úidxr6 r r7 r8 r: r: r; Útest_subplots_timeseriesB s2
|
z3TestDataFramePlotsSubplots.test_subplots_timeseriesc C s t dddg¡tjdddtjdddtjdd dgt d
|
¡t d ¡t d ¡gtjd
|
d dtjd d dtjd d dgdddgd}t|}ddddg}|D ]>}|j|d}| ¡ d ¡ d }|| j }||k
|
¡ st qd}t j t|d |jdd W 5 Q R X d S ) Nr é é iöÿÿÿÚs)Úunitr
|
ÚmÚhú2017-08-01 00:00:00ú2017-08-01 02:00:00ú2017-08-02 00:00:00T©ÚutcZThisZshouldZfail)ÚnumericÚ timedeltaÚdatetime_no_tzÚdatetime_all_tzÚtextrT rU rV rW )Úyr zno numeric data to plot©ÚmatchrX )r# ÚarrayÚpdZ TimedeltaÚ to_datetimer r) Ú get_linesÚget_dataÚvaluesÚallr, ÚpytestÚraisesÚ TypeError) r5 ÚdataÚtestdataZy_colsÚcolr8 ÚresultÚexpectedÚmsgr: r: r; Ú test_subplots_timeseries_y_axisb s2 ýý ýï
|
z:TestDataFramePlotsSubplots.test_subplots_timeseries_y_axisz6not support for period, categorical, datetime_mixed_tz)Úreasonc C s t dddg¡tjdddtjdddtjdddgtjd d
|
d gd d
|
d gd d tjdddt d¡t d¡gd}t|}|jddd}| ¡ d ¡ d |d j
|
k ¡ s®t |jddd}| ¡ d ¡ d |d j
|
k ¡ sât |jddd}| ¡ d ¡ d |d j
|
k ¡ st dS )aÉ
|
This test will fail for:
|
period:
|
since period isn't yet implemented in ``select_dtypes``
|
and because it will need a custom value converter +
|
tick formatter (as was done for x-axis plots)
|
|
categorical:
|
because it will need a custom value converter +
|
tick formatter (also doesn't work for x-axis, as of now)
|
|
datetime_mixed_tz:
|
because of the way how pandas handles ``Series`` of
|
``datetime`` objects with different timezone,
|
generally converting ``datetime`` objects in a tz-aware
|
form could help with this problem
|
r rI rJ rO ÚH)rA z2017-08-01 02:00rQ ÚcÚbÚaF)Ú
|
categoriesZorderedTrR rP )rT ÚperiodÚ categoricalÚdatetime_mixed_tzrT rs )ÚxrY r rt ru N) r# r\ r] ZPeriodZ Categoricalr^ r r) r_ r` ra rb r, )r5 rf rg Z ax_periodZax_categoricalZax_datetime_mixed_tzr: r: r; Ú-test_subplots_timeseries_y_axis_not_supported s: ý ÿ ýö ÿ
|
ÿÿ
|
ÿÿzHTestDataFramePlotsSubplots.test_subplots_timeseries_y_axis_not_supportedc C s t tj dd¡ttjd d
d}|jddd}| j|ddd |j dksPt
|
|jddd}| j|ddd |j dks|t
|
|jdd d}| j|ddd |j dks¨t
|
|jdd
|
d}| j|dd
|
d |j d
|
ksÔt
|
|jdd d}| j|dd
|
d |j d
|
kst
|
|jdd d}| j|dd d |j d ks0t
|
d}t j t |d |jddd W 5 Q R X d}t j t |d |jddd W 5 Q R X d S )Nr
|
r r T©rI rI ©r r r ©r! rI ©rI r! )r é )r! r| )r| r! )r| r z1Layout of 1x1 must be larger than required size 3rZ ©r r z1At least one dimension of layout must be positive)r! r! )r r# r$ r% r& r' r( r) r* r+ r, rc rd Ú
|
ValueError)r5 r6 r7 rk r: r: r; Ú!test_subplots_layout_multi_column· s2 $z<TestDataFramePlotsSubplots.test_subplots_layout_multi_columnz:kwargs, expected_axes_num, expected_layout, expected_shaper r} ©r r )r r c C sZ t tj dd¡ttjd d
d}|jf ddi|}| j|||d |j |ksVt
|
d S )Nr
|
r r r Tr ) r r# r$ r% r& r' r( r) r* r+ r, )r5 ÚkwargsZexpected_axes_numZexpected_layoutZexpected_shaper6 r7 r: r: r; Ú"test_subplots_layout_single_columnÜ s $ýz=TestDataFramePlotsSubplots.test_subplots_layout_single_columnc C sf t d ¡R ttj dd¡}|jddd ttj dd¡tdddd}|jddd W 5 Q R X d S ) Néd r| T)r rI ry ú1/1/2000©rB r )ÚtmÚassert_produces_warningr r# r$ Úrandnr) r ©r5 r6 r: r: r; Útest_subplots_warningsð s
|
ÿz1TestDataFramePlotsSubplots.test_subplots_warningsc C sx | j dd¡\}}ttj dd¡ttjd d
d}|j d|d ddd}| j
|
|dd d
|
|j d ksjt |d j |ks|t |j d|d ddd}| j
|
|dd d
|
|j d ks°t |d j |ksÂt | j
|
|d dd
|
t ¡ d}tjt|d& | j dd¡\}}|j d|d W 5 Q R X | j dd¡\}}t ¡ ² t dt¡ ttj dd¡ttjd d
d}|j d|dddd}| j
|
|ddd
|
|j dkst |j d|dddd}| j
|
|ddd
|
|j dksÌt |j d|dddd}W 5 Q R X | j
|
|ddd
|
|j dks
|
t | j d d ¡\}}ttj dd ¡ttjd d
d}|j d|gddd}| j
|
|d dd
|
|j dkstt d S ) NrI r r
|
r Tr F)r r8 r Úsharey)r r r r r é )rI r z@The number of passed axes must be 3, the same as the output plotrZ ©r r8 Úignorer| )rI r )r r8 r r r rx )r| r{ rz r} r )Úpltr r r# r$ r% r& r' r( r) r* r+ r, Úfigurer Úcloserc rd r~ ÚwarningsÚcatch_warningsÚ simplefilterÚ UserWarning)r5 Úfigr7 r6 Úreturnedrk r: r: r; Útest_subplots_multiple_axesü sh $
|
$ ÿ ÿ ÿ$z6TestDataFramePlotsSubplots.test_subplots_multiple_axesc C sX | j jddddd\}}| j jddddd ttj dd ¡td
|
d dd d }t| ¡ D ]\}}|| j
|
|dd qX|dd
¡ D ]}| j | ¡ dd q|d ¡ D ]}| j | ¡ dd q¨|dddgdgf ¡ D ]}| j | ¡ dd qØ|dddgdgf ¡ D ]}| j | ¡ dd q|dddgdgf ¡ D ]}| j | ¡ dd q:d S )Nr T)r r g©?gffffffî?g333333Ó?)ÚleftÚrightZhspaceZwspacer
|
é r= r> r? r rJ )r8 rF r r! Fr r rI )r r Zsubplots_adjustr r# r$ r r Ú enumerateÚravelr) r0 r2 r4 )r5 r r7 r6 Úir8 r: r: r; Útest_subplots_ts_share_axes7 s$ þ z6TestDataFramePlotsSubplots.test_subplots_ts_share_axesc C s° ddddgddddgddddgd }t |td
|
d d }|d dg jdd}|d j|d dd | j|d ¡ dd | j|d ¡ dd | ¡ D ]}| j| ¡ dd qd S )Nç ð?g @g @g @rJ r r r| )ÚAÚBÚCz
|
2014 10 11z
|
2014 10 14r r¡ r¢ T©r r£ r )r8 Ú secondary_yFr )r r r) r0 r2 r r4 )r5 Údr6 r7 r8 r: r: r; Ú'test_subplots_sharex_axes_existing_axesT s $ zBTestDataFramePlotsSubplots.test_subplots_sharex_axes_existing_axesc C sæ t tj dd¡tdd}|jdd}|D ]&}| j|dgd t|jdks*t q*t
|
¡ |jddd }|D ]&}| j|dgd t|jdkslt qlt
|
¡ |jdd
|
}| j|d gd d t|jd ksÎt t|j jdksât d S ) NrJ Zaaaaa©r. Tr¤ rq r r )r r¥ )r¥ z a (right)r ) r r# r$ r% r& r) r/ ÚlenÚlinesr, r r Zright_ax)r5 r6 r7 r8 r: r: r; Útest_subplots_dup_columnsa s z4TestDataFramePlotsSubplots.test_subplots_dup_columnszWeird rounding problemsF)rm Ústrictc C s^ t ddddg¡}tdgd ttddd tdd
|
}|jjd d d }t |j
|
¡ |¡ d S ) Nç¹?r ç $@r r rJ r r ©r¡ r¢ r T)ÚgridÚlog) r# r\ r r& Úranger) r r Úassert_numpy_array_equalÚyaxisÚ get_ticklocs)r5 rj r6 r8 r: r: r; Útest_bar_log_no_subplotsv s &z3TestDataFramePlotsSubplots.test_bar_log_no_subplotsc C sn t ddddddg¡}ttddgtdd ggjjd
|
d
|
d }t |d j ¡ |¡ t |d j ¡ |¡ d S )Nr r r® g Y@g @@g Ã@éÈ i, iô T)r± r r r )
|
r# r\ r r r) r r r³ r´ rµ )r5 rj r8 r: r: r; Útest_bar_log_subplots
s ÿz0TestDataFramePlotsSubplots.test_bar_log_subplotsc C sl |}|j jdd}t|ts t| j|d dddgd dD ],}|j j|dd}| j||dddgd d
|
q:d S ) NTr¤ ÚheightZweightÚcategory)Ú expected_keys)Údictr7 Zboth)Ú return_typer F)r» Zcheck_ax_title)r) ZboxÚ
|
isinstancer r, Z_check_box_return_type)r5 Zhist_dfr6 ri Útr r: r: r; Ú!test_boxplot_subplots_return_type s ÿüz<TestDataFramePlotsSubplots.test_boxplot_subplots_return_typec C s< dd l m} ttj dd¡tdddtdd}|jddd d
|
\}}|j d |d }|D ](}t
|
|j dkslt | j | ¡ d d qV| j |d ¡ d d | j |d jd dd d | j |d ¡ d d | j |d jd dd d t ¡ | dd¡\}}t t¡ |j d |d d}W 5 Q R X |D ],}t
|
|j dks:t | j | ¡ d d q"| j |d ¡ d d | j |d jd dd d | j |d ¡ d d | j |d jd dd d t ¡ | dd¡\}}|j d |d }|D ]T}t
|
|j dksòt | j | ¡ d d | j | ¡ d d | j |jd dd d qÚt ¡ d S )Nr r
|
rI r r
ZAB)r r. r T)r r r Fr )r r8 r )Zmatplotlib.pyplotZpyplotr r# r$ r r r& r r) r© rª r, r0 r4 r2 r r r r )r5 r r6 r r7 r8 r: r: r; Ú$test_df_subplots_patterns_minorticks§ sF
|
ý z?TestDataFramePlotsSubplots.test_df_subplots_patterns_minorticksc C s¦ t tj dd¡}tj|jdd
df< tj|jd d
df< | j dd¡\}}|jj |ddd t
|
d dd
|
¡}t
|
d dd
|
¡}t |d ¡ |¡ t |d ¡ |¡ d S ) Nr
|
rI rJ r r TF)r8 r r g @ç à?ç à¿)r r# r$ r% ÚnanZilocr r r) r Úaranger r³ Z
|
get_xticks)r5 r6 ZfigsZaxsZ expected_ax1Z expected_ax2r: r: r; Útest_subplots_sharex_falseÕ s z5TestDataFramePlotsSubplots.test_subplots_sharex_falsec
|
C s t ddd}ttj dd¡|d}i }t| jjdr<d|d< | jjd |\}}t
|
d ¡: |j |d
|
d t j dd }| j |¡ W 5 Q R X W 5 Q R X d S )NÚnowr
|
)r@ rB r r Zget_constrained_layoutTZconstrained_layoutrI r )r8 )Zreturn_filelike)rI )r r r# r$ r% Úhasattrr ZFigurer r r r) Z ensure_cleanZsavefig)r5 rG r6 r r r7 Úpathr: r: r; Ú test_subplots_constrained_layoutå s z;TestDataFramePlotsSubplots.test_subplots_constrained_layoutz index_name, old_label, new_label)NÚ Únew)ÚoldrÍ rÌ )NrË rË )NrË r NrË rI r r r r c s¬ t ddgddggddgd}||j_|j|dd}td d
|
|D sHttfd d
|
|D sbt|j| dd }t fd d
|
|D stt fdd
|
|D s¨td S )Nr rI rJ zType AzType Br¨ T©r r c s s | ]}| ¡ d kV qdS )rË N)Ú
|
get_ylabel©Ú.0r8 r: r: r; Ú <genexpr> s zSTestDataFramePlotsSubplots.test_xlabel_ylabel_dataframe_subplots.<locals>.<genexpr>c 3 s | ]}| ¡ kV qd S ©N)Ú
|
get_xlabelrÐ )Ú old_labelr: r; rÒ s )r ZylabelZxlabelr c 3 s | ]}| ¡ t kV qd S rÓ )rÏ ÚstrrÐ ©Ú new_labelr: r; rÒ s c 3 s | ]}| ¡ t kV qd S rÓ )rÔ rÖ rÐ r× r: r; rÒ s )r r Únamer) rb r, )r5 r Z
|
index_namerÕ rØ r6 r7 r: )rØ rÕ r; Ú%test_xlabel_ylabel_dataframe_subplotsò s z@TestDataFramePlotsSubplots.test_xlabel_ylabel_dataframe_subplotsr T)r ÚstackedçÍÌÌÌÌÌì?©r rÛ Úwidthr rÎ ©r r rÞ Úedge)r rÛ Úalign)r rÛ rÞ rá )r r rá )r r rÞ rá c C s6 t dgd ttddtdd}| j|f| d S )Nr rJ r¯ r )r r& r² Ú_check_bar_alignment©r5 r r6 r: r: r; Ú test_bar_align_multiple_columns s #$z:TestDataFramePlotsSubplots.test_bar_align_multiple_columnsc C s" t tj d¡}| j|f| d S )NrJ ©r r# r$ r râ rã r: r: r; Ú test_bar_align_single_column4 s z7TestDataFramePlotsSubplots.test_bar_align_single_columnc C s. t tj dd¡}| j|fddd| d S )NrJ rÜ gÉ?)rÞ Úpositionrå rã r: r: r; Útest_bar_barwidth_positionC s z5TestDataFramePlotsSubplots.test_bar_barwidth_positionÚwr c
|
C s t tj dd¡}|jjd|d}|j ¡ }t |t
|
dddddg¡¡ | ¡ d ksXt |j d ¡ d
|
ksnt |j d ¡ d kst d S ) NrJ T)rÛ rÞ r r rI r r| )g è¿g @rà r! g @)r r# r$ r r) r r1 rµ r r³ r\ Úget_xlimr, ÚpatchesÚget_x)r5 ré r6 r8 Zticksr: r: r; Ú test_bar_barwidth_position_intR s
|
z9TestDataFramePlotsSubplots.test_bar_barwidth_position_intc C sp t tj dd¡}| j|dddd | j|dddd | j|dddd | j|dddd | j|dddd d S ) NrJ r Tr rÝ r Frß rå r r: r: r; Ú&test_bar_barwidth_position_int_width_1^ s zATestDataFramePlotsSubplots.test_bar_barwidth_position_int_width_1Úcenterr c C s: |j ||||||dd}| |¡}|D ] } |dkrp| j}
|
| ¡ \} } tdd | jD } tdd | jD }nH|dkr´| j}
|
| ¡ \} } tdd | jD } td d | jD }nt t
|
| | d
|
¡ t
|
| |d
|
¡ | jd }|dkr|dksþ|dkr| ¡ }|| ¡ | }n¸|dkrT|d krT| ¡ | ¡ t|j | }| ¡ }n||dkr|dksr|dkr| ¡ | ¡ | }| ¡ }n@|dkrÌ|d krÌ| ¡ | ¡ t|j | }| ¡ }nt |
|
¡ t t|¡k ¡ sðt|d krt
|
|
|
¡ d |¡ q&|dkr0t
|
|
|
¡ d |¡ q&t q&|S )NT)r rÛ r rá rÞ rç r° r c s s | ]}| ¡ V qd S rÓ )rì ©rÑ Úpr: r: r; rÒ s zBTestDataFramePlotsSubplots._check_bar_alignment.<locals>.<genexpr>c s s | ]}| ¡ | ¡ V qd S rÓ )rì Ú get_widthrð r: r: r; rÒ s r c s s | ]}| ¡ V qd S rÓ )Úget_yrð r: r: r; rÒ s c s s | ]}| ¡ | ¡ V qd S rÓ )ró Ú
|
get_heightrð r: r: r; rÒ s g Ð?r Frï rà )r) Z_flatten_visibler1 rê Úminrë Úmaxr´ Zget_ylimr~ r Zassert_almost_equalrì rò r© r. ró rô rµ r# rÅ rb r, )r5 r6 r rÛ r rá rÞ rç r7 r8 ZaxisZax_minZax_maxZmin_edgeZmax_edgerñ rà rï r: r: r; râ g sZ
|
ù
|
|
|
|
|
|
|
|
|
z/TestDataFramePlotsSubplots._check_bar_alignment)r FFrï r r )"Ú__name__Ú
|
__module__Ú __qualname__rc ÚmarkZslowr< rH rl Zxfailrw r Z parametrizer r r r r§ r« r r r¶ r¸ rÀ rÁ rÆ rÊ rÚ rä ræ rè rí rî râ r: r: r: r; r sØ
|
& "
|
2%
|
þþ
|
|
; ý
|
|
ý
|
|
. ûþ
|
|
|
|
|
|
|
|
|
|
|
|
|
äþ!
|
úþ
|
úþ
|
|
ør ) Ú__doc__r' r Únumpyr# rc Z pandas.compatr Zpandas.compat.numpyr Z pandas.util._test_decoratorsÚutilZ_test_decoratorsÚtdZpandasr] r r r Zpandas._testingZ_testingr Z pandas.tests.plotting.commonr Zpandas.io.formats.printingr Zskip_if_no_mplr r: r: r: r; Ú<module> s
|
