U
|
ý°deX ã @ s¨ d Z ddlZddlZddlm mZ ddlm Z m
|
Z
|
m Z m Z m Z mZ ddlmZ ddlmZmZ ejdd Zdd ZejG d d
|
d
|
eZejG d d d eZdS ) z$ Test cases for misc plot functions é N)Ú DataFrameÚIndexÚSeriesÚ TimestampÚinterval_rangeÚplotting)Ú TestPlotBaseÚ_check_plot_worksc C s6 t dddgi} tjtdd | ¡ W 5 Q R X d S )NÚAé é z#matplotlib is required for plotting©Úmatch)r ÚpytestÚraisesÚ ImportErrorÚplot)Údf© r úVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/plotting/test_misc.pyÚtest_import_error_message s r c C s" t jjj} d}tjt|d | dg g i d W 5 Q R X d}tjt|d | dttddd gi d W 5 Q R X | dt dgd d
|
d d d\}}}}|dks t
|
|d ks¬t
|
|d
|
ks¸t
|
|d d iksÈt
|
| dttdg i d\}}}}|d ksòt
|
|d ksþt
|
|dks t
|
t |dks t
|
d S )Nz@Called plot accessor for type list, expected Series or DataFramer Ú )Ú backend_nameÚdataÚargsÚkwargsz.should not be called with positional arguments)ZdtypeÚlineÚxÚyÚbarF)r ÚkindÚgridr! zpandas.plotting._matplotlibé ) r Z_coreZ PlotAccessorZ_get_call_argsr r Ú TypeErrorr Úobjectr ÚAssertionErrorÚlen)ÚfuncÚmsgr r r r r r r Útest_get_accessor_args s6
|
$
|
ü ü r) c @ s: e Zd Zdd Zej di ddig¡dd Zdd Zd
|
S ) ÚTestSeriesPlotsc C sh ddl m} tjdd}t d ¡, t||d t||jd ||dd}W 5 Q R X | j|dgd d S ) Nr )Úautocorrelation_plotÚts©Úname)ÚseriesZTest)Úlabel)Úlabels)Úpandas.plottingr+ ÚtmÚmakeTimeSeriesÚassert_produces_warningr ÚvaluesZ_check_legend_labels)Úselfr+ ÚserÚaxr r r Útest_autocorrelation_plotD s z)TestSeriesPlots.test_autocorrelation_plotr Zlagé c C s0 ddl m} tjdd}t|fd|i| d S )Nr )Úlag_plotr, r- r/ )r2 r< r3 r4 r )r7 r r< r8 r r r Ú test_lag_plotP s z TestSeriesPlots.test_lag_plotc C s* ddl m} tjdd}t||dd d S )Nr )Úbootstrap_plotr, r- é
|
)r/ Úsize)r2 r> r3 r4 r )r7 r> r8 r r r Útest_bootstrap_plotW s z#TestSeriesPlots.test_bootstrap_plotN) Ú__name__Ú
|
__module__Ú __qualname__r: r ÚmarkÚ parametrizer= rA r r r r r* B s
|
r* c @ sØ e Zd Zejej dddg¡dd Zejj dd Z
|
ejj dd Z ej d
|
¡d d Z d d Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Z d+S ),ÚTestDataFramePlotsÚ pass_axisFTc C s t j}d }|r | j dd¡\}}ttj d¡ dd¡}t j
|
t dd t ||d|d}W 5 Q R X |d d j ¡ }d d
|
d g}| ||¡ | j|d d d dd |d d d |d< t j
|
t dd t ||d|d}W 5 Q R X |d d j ¡ }dddg}| ||¡ | j|d d d dd d S )Né é* éd F)Zcheck_stacklevelg¹?)ÚframeZ range_paddingr9 r z-2Ú0Ú2é éZ )Z
|
xlabelsizeZxrotZ
|
ylabelsizeZyrotr z-1.0z-0.5z0.0)r Úscatter_matrixÚpltÚsubplotsr ÚnpÚrandomZ RandomStateÚrandnr3 r5 Ú UserWarningr ZyaxisZget_majorticklabelsZ_check_text_labelsZ_check_ticks_props) r7 rH rQ r9 Ú_r ZaxesZ axes0_labelsÚexpectedr r r Útest_scatter_matrix_axis` s8 ü
|
ü
|
z+TestDataFramePlots.test_scatter_matrix_axisc s ddl m ddlm} |}t d ¡ t||dd W 5 Q R X d}t||d|d}| j| ¡ d d
||d d d
d d
|
d d g}t||d|d}| j| ¡ d d
||d d d
d t||d j d } fddt
|
dd|d ¡ ¡D }| j| ¡ d d
||d d d
d d}t t
|
j |¡t
|
j |¡t
|
j |¡dg| d}t||dd d}t||d|d}| j| ¡ d d
||d d d
d d
|
d d g}t||d|d}| j| ¡ d d
||d d d
d t||d j d } fddt
|
dd|d ¡ ¡D }| j| ¡ d d
||d d d
d dddg} t dddgdddgdddg| d}||d| d}| ¡ \}
|
} | j|
|
| d d S )Nr ©Úcm)Úandrews_curvesÚName©rL Ú class_column©z#556270z#4ECDC4z#C7F464©rL r` Úcolorr? ©Ú
|
linecolorsÚmappingÚ
|
dodgerblueÚ
|
aquamarineÚseagreen©rL r` Zcolormapc s g | ]} |¡qS r ©Újet©Ú.0Únr[ r r Ú
|
<listcomp>§ s z:TestDataFramePlots.test_andrews_curves.<locals>.<listcomp>r r
|
©r
|
ÚBÚCr^ c s g | ]} |¡qS r rk rm r[ r r rp Ë s ÚbÚgÚrr rI ©rc ©re )Ú
|
matplotlibr\ r2 r] r3 r5 r Ú _check_colorsÚ get_linesrl rT ÚlinspaceÚnuniquer rU ÚrandÚget_legend_handles_labels) r7 Úirisr] r Úrgbar9 ÚcnamesÚcmapsÚlengthÚcolorsÚhandlesr1 r r[ r Útest_andrews_curves s¶ ÿ ÿ
|
ÿ ÿ ÿ$ ÿ
|
|
|
üÿ ÿ ÿ
|
ÿ ÿ ÿ$ ÿ
|
$ z&TestDataFramePlots.test_andrews_curvesc s¤ ddl m ddlm} |}t||dd}t| ¡ }t|j ¡ }d}t||d|d}| j | ¡ d d
||d d d
d d
|
d d g}t||d|d}| j | ¡ d d
||d d d
d t||d j
|
d } fddt dd|d ¡ ¡D } | j | ¡ d d
| |d d d
d t||ddd}t| ¡ || ksJtdddg}
|
tdddgdddgdddg|
|
d}||d|
|
d}| ¡ \} } | j | |
|
d d S )Nr r[ ©Úparallel_coordinatesr^ r_ ra rb r? rd rg rh ri rj c s g | ]} |¡qS r rk rm r[ r r rp õ s z@TestDataFramePlots.test_parallel_coordinates.<locals>.<listcomp>r F)rL r` Zaxvlinesrt ru rv r rI rq rw rx )ry r\ r2 r r r& r{ ÚxaxisZget_ticklabelsrz rl rT r| r} r% r r ) r7 r r r r9 ÚnlinesZnxticksr r r r
r r1 r r[ r Útest_parallel_coordinatesÖ sl ÿ ÿ
|
ÿ ÿ ÿ$ ÿ ÿ
|
$ z,TestDataFramePlots.test_parallel_coordinatesz$ignore:Attempting to set:UserWarningc C sê ddl m} tttddd tdD dd tdD dd tdD d }||d
|
d d }| ¡ \}}td d |D |}t|dd d}tt|dd
t|dd
}|D ],\} }
|
| d |
|
d k rà| d |
|
d k s¸tq¸dS )z
|
For #15908r r é c S s g | ]}d qS )r r ©rn rX r r r rp s zSTestDataFramePlots.test_parallel_coordinates_with_sorted_labels.<locals>.<listcomp>r? c S s g | ]}d qS )rI r r r r r rp s c S s g | ]}d qS )r r r r r r rp s )ZfeatÚclassr T)Z sort_labelsc S s g | ] }| ¡ qS r ©Z get_color)rn Zpolyliner r r rp s c S s | d S )Nr r )r r r r Ú<lambda> ó zQTestDataFramePlots.test_parallel_coordinates_with_sorted_labels.<locals>.<lambda>)Úkeyéÿÿÿÿr N) r2 r r ÚlistÚranger ÚzipÚsortedr% ) r7 r r r9 Z polylinesr1 Zcolor_label_tuplesZordered_color_label_tuplesZprev_next_tupelsÚprevZnxtr r r Ú,test_parallel_coordinates_with_sorted_labels s.
|
ÿþþÿ ÿ ÿ z?TestDataFramePlots.test_parallel_coordinates_with_sorted_labelsc sÀ ddl m ddlm} |}t d ¡ t||dd W 5 Q R X d}t||d|d}dd |jd d
|
D }| j|d d
||d d d
d d ddg}t||d|d dd |jd d
|
D }| j|||d d d
d t||d j d fdd t
|
dd|d ¡ ¡D }dd |jd d
|
D }| j|||d d d
d ddddgddddgddddgg} t dddgdddgdddgddd gd }||d| d }| ¡ \}
|
} | j|
|
| d d S ) Nr r[ )Úradvizr^ r_ ra rb c S s g | ]}| ¡ d kr|qS ©r ©Ú get_label©rn Úpr r r rp - s z2TestDataFramePlots.test_radviz.<locals>.<listcomp>é r? )Ú
|
facecolorsrf rg rh ri c S s g | ]}| ¡ d kr|qS r r r r r r rp 2 s rj c s g | ]} |¡qS r rk rm r[ r r rp 6 s r c S s g | ]}| ¡ d kr|qS r r r r r r rp 7 s g ç ð?ç à?r rI rt ru rv rq rw )r¢ )ry r\ r2 r r3 r5 r Úpatchesrz rl rT r| r} r r ) r7 r r r r r9 r¥ r r r
r r1 r r[ r Ú test_radviz s0 $
|
$ "$ÿ z TestDataFramePlots.test_radvizc C s |j ddd ¡ }t|j}|jd|d}dd |D |ks@td}tjt|d |jd|d
|
g d W 5 Q R X tjt|d |jd|d d
d W 5 Q R X d }tjt|d |jd |d W 5 Q R X |j dddjdd|d d
d}dd |D }||d d
dg kstd S )Nr^ r )ZaxisT)rS Útitlec S s g | ] }| ¡ qS r ©Z get_titler r r r rp I s z:TestDataFramePlots.test_subplot_titles.<locals>.<listcomp>zjThe length of `title` must equal the number of columns if using `title` of type `list` and `subplots=True`r zkittens > puppiesr zNUsing `title` of type `list` is not supported unless `subplots=True` is passedFZ
|
SepalWidth)r r r )rS Zlayoutr§ c S s g | ]}|D ] }| ¡ q qS r r¨ )rn Zsublistr9 r r r rp c s rI r ) ZdropÚheadr Úcolumnsr r% r r Ú
|
ValueError)r7 r r r§ r r( Z
|
title_listr r r Útest_subplot_titlesB s*
|
ÿ ÿ
|
ÿz&TestDataFramePlots.test_subplot_titlesc C sz t t d¡}t |d¡ tj ¡ }t |d¡ tj ¡ }||ksFtddlm} |ddd}|ddd}||ksvtd S )N)r? r? r ©Úget_standard_colorsr rU ©Z
|
color_type) r rT Úzerosr r rU r% Ú!pandas.plotting._matplotlib.styler® )r7 r Zrand1Zrand2r® Úcolor1Úcolor2r r r Ú$test_get_standard_colors_random_seedf s
|
|
z7TestDataFramePlots.test_get_standard_colors_random_seedc C sd ddl m} |ddd}|ddd}|ddd}t|dks@tt|dksPtt|dks`td S )Nr r r Údefaultr¯ é r¡ )r± r® r& r% )r7 r® r² r³ Zcolor3r r r Ú+test_get_standard_colors_default_num_colorsx s z>TestDataFramePlots.test_get_standard_colors_default_num_colorsc s t dddgdddgddd gd
|
d d gd ddgdd dgd}|j ¡ j ¡ }dd | ¡ dd
D t fdd D s~td S )Nz
|
2017-02-03z
|
2017-03-03z
|
2017-01-01z Alice Andersz Bob BakerzCharlie Chaplingáz®GaÀg\Âõ(Ü$@g LÝ@iÒ ix éû i iõ iî é4 é )z account-startÚclientZbalancezdb-idzproxy-idZrankc S s g | ] }| ¡ qS r ©Ú get_facecolor©rn Zrectr r r rp s z=TestDataFramePlots.test_plot_single_color.<locals>.<listcomp>r rI c 3 s | ]}| d kV qdS )r Nr )rn rc ©r
r r Ú <genexpr> s z<TestDataFramePlots.test_plot_single_color.<locals>.<genexpr>)r r» Z value_countsr r Ú get_childrenÚallr% )r7 r r9 r r¿ r Útest_plot_single_color s úÿ
|
z)TestDataFramePlots.test_plot_single_colorc C sª ddl m} ddlm} | td¡}|d|d}t|t|ksFttt j
|
dd¡t d d
|
}| t ddd ¡¡}|jjjd |d }|jd ¡ |jd ¡ ks¦td S )Nr r[ r r; r rw é0 é ÚABCD©rª é )rÈ é )Zfigsizerc é )ry r\ r± r® Zgnuplotr r& r% r rT rU rV r r| r
|
r r r¥ r½ )r7 r\ r® Z color_beforeZ color_afterr Z
|
color_listr r r r Ú%test_get_standard_colors_no_appending s z8TestDataFramePlots.test_get_standard_colors_no_appendingc s ddg}ddg t tj dd¡|d}ddd}|jd|d }d
|
d | ¡ d d d
D }t fddt|D sxt|jd|d }dd | ¡ d d
D }t fddt|D s¾td S )NÚart )r¤ g¸
ëQ¸Î?g333333ã?)g333333Ó?çffffffæ?rÍ r rÇ )rt rÌ r )r rc c S s g | ]}| ¡ d d
qS )r r r¼ r¾ r r r rp ² s z<TestDataFramePlots.test_dictionary_color.<locals>.<listcomp>r rI c 3 s | ]\}}| | kV qd S ©Nr ©rn Úindexrc ©rY r r rÀ ³ s z;TestDataFramePlots.test_dictionary_color.<locals>.<genexpr>r c S s g | ] }| ¡ qS r r r¾ r r r rp · s c 3 s | ]\}}| | kV qd S rÎ r rÏ rÑ r r rÀ ¸ s )
|
r rT rU r~ r rÁ rÂ Ú enumerater% r{ )r7 Ú
|
data_filesZdf1Z dic_colorr9 r
r rÑ r Útest_dictionary_color¦ s
|
z(TestDataFramePlots.test_dictionary_colorc C sn ddl m} |ddd|dddg}tdddgitddgd}|j ¡ }td d
|
t| ¡ |D sjt d S ) Nr ©ÚTextrM r ZTotalrÌ r ©rÐ c s s" | ]\}}| ¡ | ¡ kV qd S rÎ ©Zget_text©rn rÌ rt r r r rÀ È s ÿz3TestDataFramePlots.test_bar_plot.<locals>.<genexpr>)
|
Úmatplotlib.textrÖ r r r r r r Úget_xticklabelsr% )r7 rÖ rY r Zplot_barr r r Ú test_bar_plotº s ÿ
|
ü
|
þz TestDataFramePlots.test_bar_plotc C sp ddl m} tddddddg}|jjdd d}|ddd |dddg}td
|
d t| ¡ |D sltd S ) Nr rÕ r )ÚwordÚvalueZknowledgr rÝ )r ZlegendÚ1c s s" | ]\}}| ¡ | ¡ kV qd S rÎ rØ )rn ÚactualrY r r r rÀ Õ s ÿzPTestDataFramePlots.test_barh_plot_labels_mixed_integer_string.<locals>.<genexpr>) rÚ rÖ r r Zbarhr r Zget_yticklabelsr% )r7 rÖ r Z plot_barhZexpected_yticklabelsr r r Ú*test_barh_plot_labels_mixed_integer_stringÍ s ÿþz=TestDataFramePlots.test_barh_plot_labels_mixed_integer_stringc C sØ t jjj}| j ¡ }| dd¡}|jd|d d d|d d< |jd|d d d|d d< |d d ¡ |d d ¡ ||d d dst ||d d dr¨t ||d d ds¾t ||d d drÔt d S )
|
Nr rÅ éç r r ©Zsharexéé r rI )
|
r Ú _matplotlibÚtoolsÚ_has_externally_shared_axisrR ÚfigurerS Ú add_subplotÚtwinxr% ©r7 r' ÚfigÚplotsr r r Ú&test_has_externally_shared_axis_x_axisÜ s
|
|
z9TestDataFramePlots.test_has_externally_shared_axis_x_axisc C sØ t jjj}| j ¡ }| dd¡}|jd|d d d|d d< |jd|d d d|d d< |d d ¡ |d d ¡ ||d d dst ||d d dr¨t ||d d ds¾t ||d d drÔt d S )
|
NrÅ r éA r r ©ZshareyiE r rI )
|
r rå ræ rç rR rè rS ré Ztwinyr% rë r r r Ú&test_has_externally_shared_axis_y_axisõ s
|
|
z9TestDataFramePlots.test_has_externally_shared_axis_y_axisc C sr t jjj}| j ¡ }| dd¡}|jd|d d d|d d< d}tj t
|
|d ||d d d W 5 Q R X d S )
|
NrÅ r rï r r rð z&needs 'x' or 'y' as a second parameterr Úz) r rå ræ rç rR rè rS ré r r r« )r7 r' rì rí r( r r r Ú4test_has_externally_shared_axis_invalid_compare_axis s
|
|
zGTestDataFramePlots.test_has_externally_shared_axis_invalid_compare_axisc C s´ t tj d¡tj d¡d}| j ¡ }| dd¡}|jd|d d d|d d< |jd |d d d|d d< |d d ¡ }|d d ¡ }|d
|
j |d d d d
|
d ¡ |d
|
j |d d d |d
|
j |d d dd
|
d¡ |d
|
j |d d d |d
|
j |d d dd
|
d ¡ |d
|
j |d d d |d j |dd |d j |dd |d d j ¡ ¡ rxt|d d j ¡ ¡ st|d d j ¡ ¡ r°td S )Niè )rÌ rt r rI râ r r rã rä rÌ zExternal share only)r9 r§ z"this label should never be visible)r9 zInternal share (twin) onlyz#this label should always be visibleZBothrt Úgreen)r9 rc Zyellow)r rT rU rV rR rè rS ré rê r Z
|
set_xlabelr r Z get_visibler% )r7 r rì rí Ztwin_ax1Ztwin_ax2r r r Útest_externally_shared_axes s0
|
ÿÿÿ z.TestDataFramePlots.test_externally_shared_axesc C s: t dgtdgd}t|j tddi}t|jj d S )Nr£ z2022-02-22 22:22:22r× r
|
)r r r r r r )r7 r Úsr r r Ú-test_plot_bar_axis_units_timestamp_conversionJ s þ
|
z@TestDataFramePlots.test_plot_bar_axis_units_timestamp_conversionc C sv ddl m} |ddd|dddg}tddgtddddgd }t|jj td
|
d t|j ¡ ¡ |D srt
|
d S ) Nr rÕ z ([0, 1],)r z ([1, 2],)r Zboth)Úclosedr× c s s" | ]\}}| ¡ | ¡ kV qd S rÎ rØ rÙ r r r rÀ a s ÿzFTestDataFramePlots.test_bar_plt_xaxis_intervalrange.<locals>.<genexpr>) rÚ rÖ r r r r r r r rÛ r% )r7 rÖ rY rö r r r Ú test_bar_plt_xaxis_intervalrangeV s þ þz3TestDataFramePlots.test_bar_plt_xaxis_intervalrangeN) rB rC rD ÚtdZskip_if_no_scipyr rE rF rZ Zslowr r Úfilterwarningsr r¦ r¬ r´ r· rà rË rÔ rÜ rá rî rñ ró rõ r÷ rù r r r r rG ^ s0 '
|
L
|
/
|
|
"$ + rG )Ú__doc__ÚnumpyrT r Z pandas.util._test_decoratorsÚutilZ_test_decoratorsrú Zpandasr r r r r r Zpandas._testingZ_testingr3 Z pandas.tests.plotting.commonr r Z skip_if_mplr r) Zskip_if_no_mplr* rG r r r r Ú<module> s
|
"
|
