U
|
ý°dR ã @ sè d Z ddlmZmZ ddlZddlZddlZddlZddlZddlZddl Z
|
ddl Z ddl m mZ ddlmZ ddlZddlmZmZmZmZmZmZmZ ddlm Z ddl m Z m Z ddl!m"Z" ej#G dd d e Z$d
|
d Z%dS ) z Test cases for DataFrame.plot é )ÚdateÚdatetimeN)Ú is_list_like)Ú DataFrameÚ
|
MultiIndexÚ PeriodIndexÚSeriesÚ bdate_rangeÚ
|
date_rangeÚplotting)Ú TestPlotBaseÚ_check_plot_works©Ú pprint_thingc @ sp e Zd Zejjddejjdd Zdd Zejjdd Z d d
|
Z
|
d d Z d d Z dd Z ej dddg¡dd Zej ddddg¡dd Zd d Zd d Zd d! Zd"d# Zd$d% Zd&d' Zej d(d)d*g¡d+d, Zd-d. Zd/d0 Zejjd1d2d3ej d4d5d1g¡d6d7 Zd8d9 Zd:d; Z d<d= Z d>d? Z d@dA Z dBdC Z dDdE Z!dFdG Z"dHdI Z#dJdK Z$ej dLd5d1g¡ej dMdNdOdPgdOdPdNgf¡dQdR Z%ej dSdTdUdVg¡dWdX Z&dYdZ Z'd[d\ Z(d]d^ Z)d_d` Z*ejjdadb Z+dcdd Z,dedf Z-dgdh Z.e/j0didj Z1e/j0dkdl Z2dmdn Z3ej dodpe4j5dqdr dpe4j5dsdr g¡dtdu Z6d dwdxZ7dydz Z8d{d| Z9d}d~ Z:dd Z;ej ddddd
g¡dd Z<dd Z=e/j0dd Z>dd Z?dd Z@dd ZAej dddgddfdgddgddgfg¡dd ZBdd ZCej dddgdfg¡dd ZDej d¡ddgdgd¢gfdddgddgd¢d£gfd¤d¥d¦gd§d¨gd©dªgfg¡d«d¬ ZEej dd¤d¥ddgfd¥d¤d¤d¥gfg¡d®d¯ ZFd°d± ZGd²d³ ZHej d´i dµfd¶d·id·fd¸d¹id¹fg¡dºd» ZId¼d½ ZJd¾d¿ ZKejjdÀdÁ ZLejjej d(d)dÂdÃg¡dÄdÅ ZMejjdÆeNdÇdÈdÉ ZOdÊdË ZPejjdÌdÍ ZQej d(d)dÂdÃg¡dÎdÏ ZRdÐdÑ ZSdÒdÓ ZTdÔdÕ ZUdÖd× ZVdØdÙ ZWdÚdÛ ZXe/j0dÜdÝ ZYdÞdß ZZdàdá Z[dâdã Z\ej däd)dÃdÂg¡dådæ Z]dçdè Z^dédê Z_ej d(d)d*g¡dëdì Z`dídî Zadïdð Zbdñdò Zce/j0ej d(dó¡dôdõ Zddöd÷ Zedødù Zfdúdû Zgej düdýdþdÿd g¡dd Zhdd Zidd Zjej d(d¡dd Zkej d
|
d d d ddvdd¥d¦gfg¡ej d(d)d*dÂg¡dd Zlej dddddg¡ej d(ddg¡dd Zmej dd ¡d d ZndvS ( ÚTestDataFramePlotszApi changed in 3.6.0)Úreasonc
|
C s| t ¡ }t|jdd t|jddd}| j|ddd t|jdddd }| j|dd
|
d t|jdddd }| j|d d | j|ddd tddgddgd}d}tjt |d |jj
|
dd W 5 Q R X tt j dd¡ttjd d
d}t|jdd}| j|d d t|jdddgd t|jdddgd t|jddd t|jddd d |jdd d }| j|dd d |d d
D ]P}| |j¡ | j| ¡ dd! | j|jdd"dd! | j|j ¡ gdd! q~|d fD ]@}| |j¡ | | ¡ ¡ | |j ¡ g¡ | j|d d qÚt|jd d# ttjd d
td}tt j dd¡t |¡d}t|jdd}| j|d d tjd$d%d&d'd(d)d*d+gd,d-gd.}tjd/d0gd1d2gd.}tt j d dd3¡||d4}t|jd5d# td6t j d¡i}t|jjdd7}| j|dd8d t|jjdd9d:}| j|dd8d | j ¡ \}}|jjd|d;}t |dks`t |j } | |d ksxt d S )<NF)ZgridT©Ú default_axesÚsubplotsé ©r é ©Úaxes_numÚlayout)éÿÿÿÿé )r r r ©r r )r r Ú use_indexr ©Úxrotr r é ©ÚxÚyz''Line2D' object has no property 'blarg'©Úmatch)Zblargé
|
©Úindex)r é )Úyticks)Úxticks)iÿÿÿéd )ÚylimZxlimZblah)r r Útitle)r r/ ©r! r ©Zvisible©Úminor)r/ )õ αr )r4 r )õ βr )r5 r! )õ γr )r6 r* )õ δé )r7 é Zi0Úi1)Únames)Úbaru Î)r< u ÎZc0Zc1)é r ©Úcolumnsr) u Σr# ©r ©r r )r r )r r ©r Úax) ÚtmÚmakeTimeDataFramer ÚplotÚ_check_axes_shapeÚ_check_ticks_propsr ÚpytestÚraisesÚAttributeErrorÚlineÚnpÚrandomÚrandÚlistÚstringÚ ascii_lettersÚ_check_visibleÚxaxisÚget_xticklabelsÚ get_labelÚzipÚranger Ú from_tuplesÚrandintr< Úpltr ÚlenÚAssertionErrorÚaxes)
|
ÚselfÚdfr^ ÚmsgrC Ztuplesr) r? ÚfigÚresult© rd ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/plotting/frame/test_frame.pyÚ test_plot) s üü$ ø
|
õ ÿz TestDataFramePlots.test_plotc
|
C s ddd ddg}t ddddd gddddd gtjd
|
d tjddgtd tj|d d d¡tj|d dd d¡d}t|j ddd t|ddg j ddd t|ddg j ddd t|ddg j ddd t|ddg j ddd t|ddg j ddd t|ddg j ddd d S )NZ2008Z2009Z2011Z2012r r r! r r* r9 ©Údtypez%Y)ÚformatÚi8T)ri Úutc)ÚAÚBÚCÚDÚErl rm r" rn ro rp )
|
r rM ÚarrayÚnanÚobjectÚpdÚ to_datetimeÚviewr rF )r_ Údatesr` rd rd re Útest_nullable_int_plot s ûÿ
|
z)TestDataFramePlots.test_nullable_int_plotc C s¸ t jddddgdd}t|}t|jj t|jj t|jj t|jj t ||d}t|jj t|jj t|jj t|jjdd t|jj
|
d
|
dd t|jj d
|
dd d S ) Nr r r! r ZUInt32rg r" r$ ©r$ r# ) rt rq r r rF rL r< ÚhistÚpier ÚscatterÚhexbin)r_ ZarrÚsr` rd rd re Útest_integer_array_plot¦ s z*TestDataFramePlots.test_integer_array_plotc C s: t dddgdddgd}| ¡ }t| ¡ dks6td S )Nr# r$ Úzr r r! ©rl rm )r rF r\ Ú get_linesr] ©r_ r` rC rd rd re Útest_nonnumeric_exclude¹ s z*TestDataFramePlots.test_nonnumeric_excludec C s@ t tj dd¡dddgd}|jddd}| |j ¡ d¡ d S )Nr' r! ÚaÚbÚc©r? r" )r rM rN ÚrandnrF Ú_check_text_labelsrT rV r rd rd re Útest_implicit_label¾ s z&TestDataFramePlots.test_implicit_labelc C sD t tj dd¡ddgd}d|j_|jddd |jjdks@td S )Nr r
r r ÚNAMEZLABEL)r$ Úlabel)r rM rN r r) ÚnamerF r] ©r_ r` rd rd re Ú test_donot_overwrite_index_nameà s z2TestDataFramePlots.test_donot_overwrite_index_namec C sj t ¡ }| |jddd| d¡d ¡ ¡ | |jdd| d¡ ¡ ¡ | |jdd|j ¡ ¡ | |jddd| d¡j ¡ ¡ | |jdd| d¡ ¡ ¡ | |jdd|j ¡ ¡ t dt|j d ¡|_ | |jddd| d¡d ¡ ¡ | |jdd| d¡ ¡ ¡ | |jdd|d ¡ ¡ |jddd d
|
d }|
|
|j d ¡ | j |dd d d d S )Nr r r" rl rm ©r# ry r ZTest)é r= )r# r$ r/ ÚfigsizerA )g 0@g @)r r r ) rD rE Z _check_datarF Ú set_indexrm rM Úaranger\ r? r r/ rG r rd rd re Ú test_plot_xyÊ s $ " $ z TestDataFramePlots.test_plot_xyzinput_log, expected_log)TÚlog)ÚsymZsymlogc C s° t dt d¡it d¡d}|j|d}| j||d | ¡ |ksFt|j|d}| j||d | ¡ |kspt|j|d}| j|||d | ¡ |kst| ¡ |ks¬td S )
|
Nr
r- r( ©Úlogy©Úyaxis)Úlogx©rT ©Úloglog)rT r )r rM r rF Ú_check_ax_scalesZ
|
get_yscaler] Z
|
get_xscale)r_ Z input_logZ expected_logr` rC rd rd re Útest_logscalesâ s z!TestDataFramePlots.test_logscalesÚ input_paramr r r c C sN t dt d¡it d¡d}d}tjt|d |jf |di W 5 Q R X d S )Nr
r- r( z9Boolean, None and 'sym' are valid options, 'sm' is given.r% Úsm)r rM r rI rJ Ú
|
ValueErrorrF )r_ r£ r` ra rd rd re Útest_invalid_logscaleõ s z(TestDataFramePlots.test_invalid_logscalec C s t ¡ }|jdd}| ¡ }t|d ¡ tr2t| j|dd t ¡ dt
|
j d< | ¡ }| ¡ }t|d ¡ trxt| j|dd t ¡ dt
|
j d< | ¡ }| ¡ }t|d ¡ tr¾ttt|d ¡ tsØtt ¡ t
|
j dd¡< | ¡ }| ¡ }t|d ¡ trt| j|dd W 5 Q R X t ¡ | ¡ }| ¡ }t|d ¡ tr`ttt|d ¡ ts|t| j|dd d S ) NT)Úx_compatr é r z xaxis.compatFr§ ) rD rE rF r Ú
|
isinstanceÚ get_xdatar r] rH Úcloser Z plot_paramsZuse)r_ r` rC Úlinesrd rd re Ú test_xcompatþ s:
|
|
z TestDataFramePlots.test_xcompatc C sT t tj dd¡ttdddtdddddgd}| ¡ | jjd d
|
t
|
¡ d S ) Né r éÐ r é r
r ©r) r? r ry ) r rM rN rO r r rF r[ ZaxhlinerD r« r rd rd re Útest_period_compat# s ýz%TestDataFramePlots.test_period_compatc C sø t dt d¡it ddd¡tjd}| ¡ }| ¡ d }| ¡ }t|d d
df |d d
df tjdd}tj ||j
|
d d
|
t ¡ t j t ddd¡tjd |_| ¡ }| ¡ d }| ¡ }t|d d
df |d d
df tjdd}t ||j
|
¡ d S ) Nr$ r- éc r )r) rh r r )rh r F)Zcheck_index_typerg )r rM r Úint64rF r Z
|
get_xydatar rD Zassert_series_equalr$ r« rt ÚIndexÚfloat64r) )r_ r` rC r¬ Úrsrd rd re Útest_unsorted_index0 s" ÿ * *z&TestDataFramePlots.test_unsorted_indexc C s t dddddgiddddgd}| ¡ }| ¡ \}}| ¡ }|t |d ¡ d ¡ksZt|t |d ¡ d ¡ksxtt dddtj ddd d
|
giddddtj ddgd}| ¡ }| ¡ \}}| ¡ }|t |d ¡ d ¡ksât|t |d ¡ d ¡kstt ddddgd d d dgd}|jddd}| ¡ \}}| ¡ }|t |d ¡ d ¡ksbt|t |d ¡ d ¡kstd S )Nr$ ç ç ð?ç @ç @r( r g @g @g @g ÀV@g V@g @W@g W@)r$ r r r" )
|
r rF Úget_xlimr rM ZnanminÚget_datar] Znanmaxrr )r_ r` rC ÚxminÚxmaxr¬ rd rd re Útest_unsorted_index_limsB s* þ z+TestDataFramePlots.test_unsorted_index_limsc C s t tj dd¡ttjd d
ddddgd }d}tjt |d |j
|
j d
|
d W 5 Q R X tjt |d |j
|
j d
|
d W 5 Q R X d S ) Nr8 r r# r$ r Úfourr± z+Log-y scales are not supported in area plotr% Tr r ) r rM rN rO rP rQ rR rI rJ r¥ rF Úarea©r_ r` ra rd rd re Útest_negative_log[ s
|
ýz$TestDataFramePlots.test_negative_logc C sZ t t|d ¡ d ¡}t||D ]0\}}|| ¡ d 7 }| ¡ d }t ||¡ q$d S )Nr r )rM Úzerosr\ r¾ rW rD Úassert_numpy_array_equal)r_ Z normal_linesZ stacked_linesÚbaseÚnlÚslÚsyrd rd re Ú_compare_stacked_y_coodg s
|
z*TestDataFramePlots._compare_stacked_y_coodÚkindrL rà c C s¦ t j d¡}t| dd¡ddddgd}| }t| d¡| d¡| d¡ | d¡ d }t| dd¡ttjd d
ddddgd
|
}t |j
|
|d d }t |j
|
|d d }| |j |j ¡ t |j
|
|d d }t |j
|
|d d }| |j |j ¡ t |j
|
|d d }t |j
|
|d d }| |j d d
|j d d
¡ | |j dd
|j dd
¡ t |j
|
d d d} t jt| d |j
|
d d W 5 Q R X | |jd ¡}
|
t |
|
j
|
|d d d d S )Né* r8 r Úwr# r$ r r )rÏ r# r$ r r± F)rÍ ÚstackedTr ©rÐ zWhen stacked is True, each column must be either all positive or all negative. Column 'w' contains both positive and negative valuesr% r )rÍ r rÐ )rM rN Ú RandomStater rO r rP rQ rR r rF rÌ r¬ rI rJ r¥ r r) ) r_ rÍ Z np_randomr` Úneg_dfZsep_dfZmixed_dfÚax1Úax2ra Údf2rd rd re Útest_line_area_stackedn s@
|
|
üÿ
|
|
ý ÿz)TestDataFramePlots.test_line_area_stackedc
|
C sô ddt jdg}dt jddg}t||d}t||dtjddd}||fD ] }t|j}|jd ¡ }|jd ¡ }t j dddgt j
|
d } t t |j d¡| ¡ t j dddgt j
|
d } t t |j d¡| ¡ t |jt d
|
d
|
d d
|
g¡¡ t |jt d
|
d d
|
d
|
g¡¡ t j ddddgt j
|
d }
|
t j ddddgt j
|
d } t|jd d }t |jd ¡ |
|
¡ t |jd ¡ |
|
| ¡ t|jj}t |jd ¡ |
|
¡ t |jd ¡ |
|
| ¡ t|jjd
|
d }t |jd ¡ |
|
¡ t |jd ¡ | ¡ qLd S ) Nr r r! ©r
r r )Úkr( r rg FTrÑ )rM rr r rD Z makeDateIndexr rF r¬ Ú get_ydatarq r¶ rÇ ÚdeleteÚdataÚmaskrà ) r_ Zvalues1Zvalues2r` ÚtdfÚdrC Zmasked1Zmasked2ÚexpZ expected1Z expected2rd rd re Útest_line_area_nan_df s>
|
ÿ ÿ z(TestDataFramePlots.test_line_area_nan_dfc C s` t tj dd¡dddgd}| ¡ }| ¡ \}}| ¡ }||d ¡ d d ksTt||d ¡ d d kspt|jd d
|
}| ¡ \}}| ¡ }||d ¡ d d ks¬t||d ¡ d d ksÈt|jd d d }| j |dd d |D ]p}t
|
|dsütt
|
|dr t| ¡ \}}| ¡ }||d ¡ d d ks>t||d ¡ d d ksêtqêd S )Nr8 r! r# r$ r r r r T)Ú secondary_y)râ r r0 r Zleft_axÚright_ax) r rM rN rO rF r½ r r¾ r] rG Úhasattr)r_ r` rC r¿ rÀ r¬ r^ rd rd re Ú test_line_limÅ s( z TestDataFramePlots.test_line_limFzX2020-12-01 this has been failing periodically on the ymin==0 assertion for a week or so.)Ústrictr rÐ Tc
|
C sÄ t tj dd¡ddddgd}| }t|jj|d}| ¡ \}}| ¡ \}}| ¡ } || d
|
¡ d d kspt || d
|
¡ d d
|
kst |d kst t|jj|d}| ¡ \}}|d ksÀt d S ) Nr8 r r# r$ r r r rÑ r r ) r rM rN rO r rF rà r½ Úget_ylimr r¾ r] )
|
r_ rÐ r` rÓ rC r¿ rÀ ZyminZymaxr¬ rd rd re Ú test_area_limÝ s z TestDataFramePlots.test_area_limc C s t tj dd¡ddgd}| jjdddd\}\}}|j|d d
|
|j|d d
|
| |¡ ||¡sft | |¡ ||¡s|t d S ) Nr r r# r$ r r T)Úshareyrà )rC rÍ )
|
r rM rN rO r[ r rF Z
|
get_y_axisZjoinedr] )r_ r` rb rÔ rÕ rd rd re Ú test_area_sharey_dont_overwriteô s z2TestDataFramePlots.test_area_sharey_dont_overwritec C s² t tj dd¡}|jjdd}|jD ]}| ¡ dks&tq&|jjddd}|jD ]}| ¡ dksRtqR|jjddd}| j |ddd |D ] }|jD ]}| ¡ dkstqqd S ) Nr* r )Ú linewidthT)rÐ rë )rë r )r* r r )
|
r rM rN r rF r< ÚpatchesZ get_linewidthr] rG )r_ r` rC Úrr^ rd rd re Útest_bar_linewidthÿ s
|
|
|
z%TestDataFramePlots.test_bar_linewidthc C sL t tj dd¡}d}|jj|d}|jD ] }| ¡ |t|j ks*t
|
q*|jjd|d}|jD ]}| ¡ |ks`t
|
q`|jj |d}|jD ] }| ¡ |t|j kst
|
q|jj d|d}|jD ]}| ¡ |ksÀt
|
qÀ|jj|dd}|D ] }|jD ]}| ¡ |ksôt
|
qôqê|jj |dd}|D ]&}|jD ]}| ¡ |ks*t
|
q*q d S )Nr* gÍÌÌÌÌÌì?)ÚwidthT)rÐ rï )rï r ) r rM rN r rF r< rì Ú get_widthr\ r? r] ÚbarhÚ
|
get_height)r_ r` rï rC rí r^ rd rd re Útest_bar_barwidth s,
|
|
|
|
|
|
z$TestDataFramePlots.test_bar_barwidthc C s¬ t tj dd¡}|jjddd}dd |jD }|dgd ksDt|jjdd d
|
d d d gd}dd |jd d
D }|d d
|
d d d gkst|jjdt dddddg¡d}dd |jD }|dgd ksÎt|jjddddddgd}dd |jd d
D }|dddddgkst|jjdd d}|D ]*}dd |jD }|d gd ks,tq,|jjdt dddddg¡d}|D ]*}dd |jD }|dgd ks|tq|d S )Nr* Fr )rÐ Úbottomc S s g | ] }| ¡ qS rd ©Zget_y©Ú.0Úprd rd re Ú
|
<listcomp>; s z;TestDataFramePlots.test_bar_bottom_left.<locals>.<listcomp>é Tr éþÿÿÿéýÿÿÿéüÿÿÿéûÿÿÿc S s g | ] }| ¡ qS rd rõ rö rd rd re rù ? s )rÐ Úleftc S s g | ] }| ¡ qS rd ©Úget_xrö rd rd re rù C s r r! r c S s g | ] }| ¡ qS rd r rö rd rd re rù G s )r rô c S s g | ] }| ¡ qS rd rõ rö rd rd re rù L s )r rÿ c S s g | ] }| ¡ qS rd r rö rd rd re rù Q s )
|
r rM rN rO rF r< rì r] rñ rq )r_ r` rC rc r^ rd rd re Útest_bar_bottom_left8 s* z'TestDataFramePlots.test_bar_bottom_leftc C sÀ t dtjdgdddgdddgd}|j ¡ }dddddddddg }d d
|
|jD }||ks`t|jjd d }d d
|
|jD }||kstdd
|
|jD }dddddddddg }||ks¼td S )Nr' é r* r r r! ©rl rm rn r c S s g | ] }| ¡ qS rd ©rò rö rd rd re rù X s z3TestDataFramePlots.test_bar_nan.<locals>.<listcomp>TrÑ c S s g | ] }| ¡ qS rd r rö rd rd re rù \ s c S s g | ] }| ¡ qS rd rõ rö rd rd re rù _ s r¹ g $@g 4@g .@g D@)r rM rr rF r< rì r] )r_ r` rC Úexpectedrc rd rd re Ú test_bar_nanT s $
|
z TestDataFramePlots.test_bar_nanc C sD t tj dd¡t td¡t td¡d}t tj dd¡t td¡t td¡d}||fD ]Þ}|j ¡ }|j
|
¡ }t |t dddd d
|
dg¡¡ | ¡ d ks¦t|jd ¡ d ks¼t|jd ¡ dksÒt|jj dd}t |t dddd d
|
dg¡¡ | ¡ d kst|jd ¡ d ks(t|jd ¡ dks`tq`d S )Nr8 r* ZABCDEFZabcder± r r r r! r )g à¿g @g пr g@TrÑ g @)r rM rN r rt rµ rP ZCategoricalIndexrF r< rT Ú get_ticklocsrD rÇ rq r½ r] rì r )r_ Zdf1rÖ r` rC Zticksrd rd re Útest_bar_categoricalc s, ý ý
|
|
z'TestDataFramePlots.test_bar_categoricalc C sÜ t tj dd¡ttjd d
ddddgd}t|jj ddd t|jj d d
|
d t
|
d ¡}t j t|d |jj dd W 5 Q R X t
|
d¡}t j t|d |jj dd W 5 Q R X |jddddd}| j|d dd d S )Nr8 r r# r$ r r r± r" r r z5scatter() missing 1 required positional argument: 'y'r% r z5scatter() missing 1 required positional argument: 'x'ry r| T)r# r$ rÍ r rA r )r rM rN r rP rQ rR r rF r| ÚreÚescaperI rJ Ú TypeErrorrG )r_ r` ra r^ rd rd re Útest_plot_scatter s
|
ý
|
|
z$TestDataFramePlots.test_plot_scatterc C s\ t tj d¡dgd}tddddj|d< d}tjt|d |j d
|
ddd W 5 Q R X d S ) Nr' r
r z
|
2014-01-01Úh)ÚstartÚfreqÚperiodsZdtimez9must be a string or a (real )?number, not 'datetime.time'r% r| )rÍ r# r$ )
|
r rM rN r r
|
ÚtimerI rJ r rF rÄ rd rd re Ú&test_raise_error_on_datetime_time_data s
|
z9TestDataFramePlots.test_raise_error_on_datetime_time_datac C s` t tdddddd}tj ddt|¡}t||d}t|jj dd d
|
t|jj ddd
|
d S ) Niã r é ÚW)r r r r )rw Úvalsrw r r" )
|
r
|
r rM rN Únormalr\ r r rF r| )r_ rw r r` rd rd re Ú test_scatterplot_datetime_data s
|
z1TestDataFramePlots.test_scatterplot_datetime_datac C s t dddgdddgd}t|jjdd d
|
t|jjd d d
|
t dddgdd d gd}t|jjdd d
|
t|jjd d d
|
d S )Nrl rm rn r r! r rØ r
r r" r r r )r r rF r| r rd rd re Ú test_scatterplot_object_data§ s z/TestDataFramePlots.test_scatterplot_object_dataÚorderedÚ
|
categoriesÚsetosaÚ
|
versicolorÚ virginicac C sì t ddgddgddgddgddggd d
|
gd }tjd d d d dg||d|d< |jjdddd}|j\}|j}t dddg¡}| ¡ }t
|
||¡ t ddddg¡} |j }
|
t
|
|
|
| ¡ |} dd |j ¡ D } td d t| | D sètd S ) Ngffffff@g @g@r¼ g @g @g@g@Úlengthrï r r r r )r r Zspeciesr r ©r# r$ r ç à?g ø?g @r¹ rº r» c S s g | ] }| ¡ qS rd ©Úget_text)r÷ Úird rd re rù Ï s zLTestDataFramePlots.test_scatterplot_color_by_categorical.<locals>.<listcomp>c s s | ]\}}||kV qd S ©Nrd )r÷ r$ Újrd rd re Ú <genexpr>Ð s zKTestDataFramePlots.test_scatterplot_color_by_categorical.<locals>.<genexpr>)r rt Ú CategoricalrF r| Ú collectionsÚcolorbarrM rq Z get_ticksrD rÇ Z _boundariesrC Zget_ymajorticklabelsÚallrW r] ) r_ r r r` rC Zcolorbar_collectionr* Zexpected_ticksZ result_ticksZexpected_boundariesZresult_boundariesZexpected_yticklabelsZresult_yticklabelsrd rd re Ú%test_scatterplot_color_by_categorical³ s* þ ý
|
z8TestDataFramePlots.test_scatterplot_color_by_categoricalzx, yr" )r$ r# )r$ r$ c C s: t ddddgt ddddg¡d}t|jj||d d S ) Nr r r! r r
r r r" )r rt r( r rF r| ©r_ r# r$ r` rd rd re Ú'test_plot_scatter_with_categorical_dataÒ s $z:TestDataFramePlots.test_plot_scatter_with_categorical_datac C sª t tjjddddttjd d
dddd gd
|
}|jjdddd |jjdd d d g}|D ]4}|j d j
|
j dkstt |j d j j ¡ dksZt qZd}|jjddd|d}|j d j
|
j |ks¾t |jjddddd}|j d j d ksæt |jjdd dd }|j d j d kst | j|j dgd t d d gddgd}ddddg}ddddg}t ||g¡}|jjdd |d }t |j d ¡ |¡ t ddg¡}|jjdd |d d d S ) Nr r- )r8 r )ÚlowÚhighÚsizer8 r# r$ r r r± r r r ZGreysÚ cubehelix)r# r$ r ÚcolormapF)r# r$ r r* Úredrí ©Z
|
facecolorsr! r r rº r¹ rl rm Zspring)r# r$ r Úcmap)r rM rN rZ rP rQ rR rF r| r) r6 r r] r* rC Ú
|
get_ylabelÚ _check_colorsrq rD rÇ Z get_facecolor) r_ r` r^ rC ÚcmZred_rgbaZ
|
green_rgbaZ
|
rgba_arrayZ float_arrayrd rd re Útest_plot_scatter_with_cÙ s2
|
ý$ z+TestDataFramePlots.test_plot_scatter_with_cc C sR t tj d¡d dddgd}|jjdddd}tj|d j|jd ¡ d d S )
|
N©r' r! r- r
r r r )r# r$ r~ r )Úright)
|
r rM rN rF r| rD rÇ Úvaluesr) Z get_sizesr rd rd re Útest_plot_scatter_with_s
|
s z+TestDataFramePlots.test_plot_scatter_with_sc C s\ dd l }ttj d¡d dddgd}|j ¡ }|jjddd|d}|jd j |ksXt
|
d S ) Nr r; r- r
r r r )r# r$ r Únorm) Ú
|
matplotlibr rM rN ÚcolorsZLogNormrF r| r) r? r] )r_ Úmplr` r? rC rd rd re Útest_plot_scatter_with_norm s
|
|
z.TestDataFramePlots.test_plot_scatter_with_normc C s dd l }ttj d¡d dddgd}|jjdddd}|jd j}|j ¡ |j
|
¡ f}|j j | }|jD ]}||||ksjt qjd S ) Nr r; r- r
r r r r )r@ r rM rN rF r| r) r? r ÚminÚmaxrA Z Normalizer] )r_ rB r` rC Z plot_normZ color_min_maxZ default_normÚvaluerd rd re Ú test_plot_scatter_without_norm s
|
z1TestDataFramePlots.test_plot_scatter_without_normc C s4 t tj dd¡ttjd d
ddddgd}t|jj t|jj dd t|jj d
|
d
|
d t|jj d
|
d t tj d d¡ttjd d
t
|
dd}t|jj t ddgddgd}t|jj }| j |dd |jj dd d}| j |dd d d t|jj }| j |dd |jj ddd}| j |dddd d S )Nr8 r ÚoneÚtwoÚthreer r± F©ÚlegendTr rÑ r' é r r rØ éZ r é# )ÚrotÚfontsize©r Ú
|
xlabelsizeÚ
|
ylabelsize)Úyroté7 é )rU rT rS ) r rM rN r rP rQ rR r rF r< rX rH rñ r rd rd re Ú test_plot_bar& s0
|
ý ý z TestDataFramePlots.test_plot_barc C s. |}|d }| ¡ j}dd |D }t|jj}| | ¡ |¡ t |j
|
¡ t dt |d ¡¡ t |jdt | ks|tt ¡ |jjdd}| j|ddd t ¡ t|jj}t dd
|
dg¡}|jj|d }| ¡ j}d d |D }| | ¡ |¡ t |j
|
¡ |¡ t |jdt | ks*td S ) NÚheightc S s g | ] }t |qS rd r ©r÷ r rd rd re rù K s z3TestDataFramePlots.test_boxplot.<locals>.<listcomp>r r9 é( ©rP r ©r rU r8 )Ú positionsc S s g | ] }t |qS rd r rZ rd rd re rù ^ s )Ú_get_numeric_datar? r rF Úboxr rU rD rÇ rT r rM r r\ r¬ r] r« rH rq ) r_ Úhist_dfr` ÚseriesÚ numeric_colsÚlabelsrC r^ r^ rd rd re Ú test_boxplotG s.
|
ÿ
|
z TestDataFramePlots.test_boxplotc C s8 |}| ¡ j}dd |D }|jjdddd}| j|dddd | | ¡ |¡ t|jd t| ksjt t
|
|jjd
|
d
|
dd
|
d }| j |d d d | j |dd t ||D ],\}}| | ¡ |g¡ t|jd ks¨t q¨t d ddg¡}|jj|dd}| | ¡ |¡ t |j ¡ |¡ t|jd t| ks4t d S )Nc S s g | ] }t |qS rd r rZ rd rd re rù f s z<TestDataFramePlots.test_boxplot_vertical.<locals>.<listcomp>é2 r= F)rP rQ Úvertr ©r rU rT r9 T)r r rg r r! )r r! r r r r )r^ rg )r_ r? rF r` rH r Úget_yticklabelsr\ r¬ r] r rG r¡ rW rM rq rD rÇ r r ) r_ ra r` rc rd rC r^ r r^ rd rd re Útest_boxplot_verticalc s0
|
ûz(TestDataFramePlots.test_boxplot_verticalc C s t tj dd¡ttjd d
ddddgd}d}tjt |d |j
|
j d
|
d W 5 Q R X |j
|
j d d }| |d ¡ |j
|
j d d }| |d ¡ |j
|
¡ }| |d ¡ |j
|
j dd }| |d¡ d S )Nr8 r rH rI rJ r r± z2return_type must be {None, 'axes', 'dict', 'both'}r% Z
|
not_a_type)Ú return_typeÚdictr^ Zboth) r rM rN r rP rQ rR rI rJ r¥ rF r` Z_check_box_return_type)r_ r` ra rc rd rd re Útest_boxplot_return_type s
|
ý
|
z+TestDataFramePlots.test_boxplot_return_typec C s² t tj dd¡}t|jdd}dd |jD }| j||d | j|dd |jdd
|
d d }| j|d
|
d d d t|jdddd}| j |ddd |jdddd}| j
|
|dd d S )Nr- r Úkde©rÍ c S s g | ] }t |qS rd r rZ rd rd re rù s z2TestDataFramePlots.test_kde_df.<locals>.<listcomp>©rd r r r r* )rÍ rP rQ rR T)r rÍ r r r )rÍ r r r r ) r rM rN r r rF r? Ú_check_legend_labelsrH rG r¡ )r_ r` rC r r^ rd rd re Ú test_kde_df s üz TestDataFramePlots.test_kde_dfc C s0 t tjjdd}tj|jd< t|jdd d S )N)r- r ©r1 ©r r rn ro )r rM rN Úuniformrr Úlocr rF r rd rd re Útest_kde_missing_vals s z(TestDataFramePlots.test_kde_missing_valsc sV ddl m ttj dd¡}|d }t|jj}dd |j D }| j
|
||d t|jjdddd }| j |dd
|
d | j |d d |jjdd}| j |ddd t ¡ |jjdddd} fdd| ¡ D }t |d ¡ d¡ t ¡ |jjddd} fdd| ¡ D }t |d ¡ d¡ t ¡ |jjdddd }| j |dddd d S ) Nr ©Ú Rectangler- r c S s g | ] }t |qS rd r rZ rd rd re rù º s z3TestDataFramePlots.test_hist_df.<locals>.<listcomp>rp T)r r r r r r r r[ r\ r] )Ú
|
cumulativeÚbinsÚdensityc s g | ]}t | r|qS rd ©r© ©r÷ r# rx rd re rù Ì s
|
r rº )rz r{ c s g | ]}t | r|qS rd r} r~ rx rd re rù Ñ s
|
rû g Y@rf r= Ú
|
horizontal)rP rQ Ú orientationrh )Zmatplotlib.patchesry r rM rN r r rF rz r? rq rG r¡ rH rD r« Z get_childrenÚassert_almost_equalrò )r_ r` rb rC r r^ Zrectsrd rx re Ú test_hist_df³ s6 üz TestDataFramePlots.test_hist_dfÚweightsç¹?©r- )Úshape)r- r c C sz t j d¡ tttddgt j dd¡}t|jd|d}t|jdd}d d
|
|j D }d d
|
|j D }t
|
||¡ d S ) Nr rl rm r r- rz )rÍ r ro c S s g | ] }| ¡ qS rd r ©r÷ Úpatchrd rd re rù å s z8TestDataFramePlots.test_hist_weights.<locals>.<listcomp>c S s g | ]}d | ¡ qS )r r r rd rd re rù é s ) rM rN Úseedr rl rW r r rF rì rD r )r_ r r` rÔ rÕ Zpatch_height_with_weightsZexpected_patch_heightrd rd re Útest_hist_weightsÚ s z$TestDataFramePlots.test_hist_weightsNc
|
C s´ t dd |D ¡}t dd |D ¡}t dd |D ¡}t dd |D ¡} |d k rhtj||dd |d k rtj||dd |d k rtj||dd |d k r°tj| |dd d S )Nc S s g | ] }| ¡ qS rd rõ rö rd rd re rù õ s z7TestDataFramePlots._check_box_coord.<locals>.<listcomp>c S s g | ] }| ¡ qS rd r rö rd rd re rù ö s c S s g | ] }| ¡ qS rd r rö rd rd re rù ÷ s c S s g | ] }| ¡ qS rd )rð rö rd rd re rù ø s F)Z check_dtype)rM rq rD rÇ )
|
r_ rì Ú
|
expected_yÚ
|
expected_hÚ
|
expected_xÚ
|
expected_wZresult_yZ result_heightZresult_xZ result_widthrd rd re Ú_check_box_coordí s z#TestDataFramePlots._check_box_coordc C s® t t t dddddg¡t dddd d
|
g¡¡t t dddddg¡t dddddg¡¡t t dddddg¡t d
|
d dddg¡¡d d d dgd}t t t tjdddddg¡t ddddd d
|
g¡¡t t dtjddddg¡t ddddddg¡¡t t dddtjddg¡t d
|
d ddddg¡¡d d d dgd}||fD ]h}|jjdd}| j|jd d
t dddddg¡t dddd d
|
g¡d | j|jdd
t dddddg¡t dddddg¡d | j|jdd
t dddddg¡t d
|
d dddg¡d |jjddd}| j|jd d
t dddddg¡t dddd d
|
g¡d | j|jdd
t dddd d
|
g¡t dddddg¡d | j|jdd
t dddddg¡t d
|
d dddg¡d |jjdddd}| j|d jt dddddg¡t dddd d
|
g¡d | j|d jt dddddg¡t dddddg¡d | j|d jt dddddg¡t d
|
d dddg¡d |jjddd }| j|jd d
t dddddg¡t dddd d
|
g¡d | j|jdd
t dddddg¡t dddddg¡d | j|jdd
t dddddg¡t d
|
d dddg¡d |jjdddd }| j|jd d
t dddddg¡t dddd d
|
g¡d | j|jdd
t dddd d
|
g¡t dddddg¡d | j|jdd
t dddddg¡t d
|
d dddg¡d |jjddddd }| j|d jt dddddg¡t dddd d
|
g¡d | j|d jt dddddg¡t dddddg¡d | j|d jt dddddg¡t d
|
d dddg¡d q>d S ) Nr r r! r r* r' é r= r9 r8 r rl rm rn r )r{ r )r r T)r{ rÐ é é r rM é )r{ rÐ r r )r{ r )r r )r{ rÐ r )r{ rÐ r r ) r rM Úrepeatrq rr rF rz r rì )r_ Z normal_dfZnan_dfr` rC r^ rd rd re Útest_hist_df_coord sú ***ýú ÿ ÿ ÿù ô ý ý ý ý ý ýýýý ý ý ý ý ý ý ÿýýýz%TestDataFramePlots.test_hist_df_coordc C s( t tj dd¡ ¡ }t|jdd d S )Nr- r TrK )r rM rN r Úcumsumr rF r rd rd re Útest_plot_int_columns s z(TestDataFramePlots.test_plot_int_columnsc C s¨ dd l m} | ¡ }ttj dd¡}ddddddddddgddgfD ]X}| ¡ | d ¡ |j |d
|
}t
|
| ¡ d t |
D ] \}}| ¡ || kstqqJd S ) Nr r- r! ú^ú+Úo)r r r ©r r éo )Ústyle)Úmatplotlib.pyplotÚpyplotÚgcfr rM rN r ZclfÚ add_subplotrF Ú enumerater r\ Z
|
get_markerr] )r_ r[ rb r` ÚmarkersrC ÚidxrL rd rd re Útest_style_by_column s
|
ü
|
z'TestDataFramePlots.test_style_by_columnc C sP t ddg}| ¡ }| ¡ d ks$t|jdd}| ¡ ¡ d ¡ dksLtd S )Nr r TrK r Ú )r rF Ú
|
get_legendr] Ú get_textsr# )r_ r~ rC rd rd re Útest_line_label_none s
|
z'TestDataFramePlots.test_line_label_nonezprops, expected)ZboxpropsZboxes)Z whiskerpropsZwhiskers)ZcappropsZcaps)Z medianpropsZmediansc C sP t dd dD }|ddii}|jjf ddi|}|| d ¡ dksLtd S ) Nc S s i | ]}|t j d ¡qS r
)rM rN )r÷ rÙ rd rd re Ú
|
<dictcomp>° s zHTestDataFramePlots.test_specified_props_kwd_plot_box.<locals>.<dictcomp>ÚABCÚcolorZC1rk rl r )r rF r` Z get_colorr] )r_ Úpropsr r` Zkwdrc rd rd re Ú!test_specified_props_kwd_plot_box¥ s z4TestDataFramePlots.test_specified_props_kwd_plot_boxc C s t t dddg¡tdddtdddtdddgd gd
|
}| ¡ }|jd ¡ }|d |d k sdt|jd ¡ }t
|
|t dddg¡¡ d S ) Nr¼ r» rº éÜ r' r r r= Útestr± r ) r rM rq r rF r¬ rª r] rÚ rD rÇ )r_ r` rC r, Zydatard rd re Útest_unordered_ts¶ s ýz$TestDataFramePlots.test_unordered_tsc C sj t ddddgi}tjjD ] }|j|d t|j| qdD ]&}|jdd|d t|j|dd q>d S )Nr# r r r! ro )r| r} )r r Ú PlotAccessorÚ _common_kindsrF Úgetattr)r_ r` rÍ rd rd re Útest_kind_both_ways s z&TestDataFramePlots.test_kind_both_waysc
|
C sH t td}tjjD ].}d}tjt|d |j|d W 5 Q R X qd S )NZabcdúno numeric data to plotr% ro ) r rP r r² r³ rI rJ r rF ©r_ r` rÍ ra rd rd re Útest_all_invalid_plot_dataÌ s
|
z-TestDataFramePlots.test_all_invalid_plot_datac
|
C sÎ t tj d¡ dd¡td}d|tj |jd ¡dk< tj j
|
D ].}d}t j t |d |j|d
|
W 5 Q R X q>t tj d¡ dd¡td}d|tj |jd ¡dk< t j t dd |jd d
|
W 5 Q R X d S ) NrÎ r' r rg r
r r! r¶ r% ro rà )r rM rN rÒ r rs rO r r r² r³ rI rJ r rF r· rd rd re Ú test_partially_invalid_plot_dataÓ s z3TestDataFramePlots.test_partially_invalid_plot_datac C s@ t tj dd¡}d}tjt|d |jdd W 5 Q R X d S )Nr' r z*invalid_plot_kind is not a valid plot kindr% Zinvalid_plot_kindro )r rM rN r rI rJ r¥ rF rÄ rd rd re Útest_invalid_kindá s z$TestDataFramePlots.test_invalid_kindzx,y,lblrm rn rl r
r r c C sJ t ddgddgddgd}tjtdd |j|||d
|
W 5 Q R X d S ) Nr r r! r r* r8 r ú x must be a label or positionr% ©r# r$ r ©r rI rJ r¥ rF )r_ r# r$ Úlblr` rd rd re Útest_invalid_xy_argsç s z'TestDataFramePlots.test_invalid_xy_argsc C sR t ddgddgddgd}d}tjt|d |jd
|
d d gd d W 5 Q R X d S )Nr r r! r r* r8 r z.label should be list-like and same length as yr% rl rm rn Z bad_labelr¼ r½ rÄ rd rd re Útest_bad_labelô s z!TestDataFramePlots.test_bad_labelzx,yr c C sL t dddgdddggtdd}tjtd d
|
|j||d W 5 Q R X d S ) Nr r! r* r r r8 ZAABr r» r% r" )r rP rI rJ r¥ rF r- rd rd re Ú test_invalid_xy_args_dup_colsú s z0TestDataFramePlots.test_invalid_xy_args_dup_colszx,y,lbl,colorsr4 Úbluer r r ZbokehZcythonÚgreenZyellowc C sl t ddgddgddgd}t|jd||d |j||||d
|
}t|jt|ksVt| j| ¡ |d d S ) Nr r r! r r* r8 r rl r¼ )r# r$ r r¬ ©Z
|
linecolors)r r rF r\ r¬ r] r8 r )r_ r# r$ r¾ rA r` rC rd rd re Útest_y_listlike s
|
|
z"TestDataFramePlots.test_y_listlikez x,y,colnamesc C s0 t ddgddgd}||_t|j||d d S )Nr r r! r r r" )r r? r rF )r_ r# r$ Zcolnamesr` rd rd re Útest_xy_args_integer s z'TestDataFramePlots.test_xy_args_integerc C s t tjjddtjjddt d¡tjjdd d}|jjdddd}t|jdks\t |jjddd d
|
}t|d j
|
j d kst | j |dd d d S )Nr rs r rl rm r' )r# r$ Zgridsizer T)r# r$ r r r rA r ) r rM rN ru r rF r} r\ r) r] Úfigurer^ rG )r_ r` rC r^ rd rd re Útest_hexbin_basic s ýÿz$TestDataFramePlots.test_hexbin_basicc C s t tjjddtjjddt d¡tjjdd d}|jjdddd}t|jdks\t |jjdddtj
|
d }t|jdkst d S )
|
Nr rs r rl rm rn )r# r$ rn r )r# r$ rn Zreduce_C_function) r rM rN ru r rF r} r\ r) r] Zstdr rd rd re Útest_hexbin_with_c. s ýÿz%TestDataFramePlots.test_hexbin_with_czkwargs, expectedZBuGnr3 r2 r6 ZYlGnc C sj t tjjddtjjddt d¡tjjdd d}|jjf ddd|}|jd jj |ksft
|
d S )Nr rs r rl rm r" r ) r rM rN ru r rF r} r) r6 r r] )r_ Úkwargsr r` rC rd rd re Útest_hexbin_cmap= s ýÿz#TestDataFramePlots.test_hexbin_cmapc C sp t tj dd¡dddgdddd d
|
gd }d }tjt|d |j ¡ W 5 Q R X t |jjdd}|
|
|j |j ¡ t |jjdd}|
|
|j |j ¡ t |jjddd}t |t |jks¸t|D ]}|
|
|j |j ¡ q¼t||jD ]\}}| ¡ |ksÞtqÞdddddg}dddddg}t |jjdd||d}t |t |jksBt|D ]$}|
|
|j |¡ | j|j|d qFd S ) Nr* r! ÚXÚYÚZr
r r rß Úer> z/pie requires either y column or 'subplots=True'r% ry r Tr rl rm rn ro rp rí ÚgÚm)r r rd rA r5 )r rM rN rO rI rJ r¥ rF r{ r r Útextsr) r\ r? r] rW r7 r8 rì )r_ r` ra rC r^ Úylabelrd Z
|
color_argsrd rd re Ú test_pie_dfP sF ýýûz TestDataFramePlots.test_pie_dfc C s dd l }ttj dd¡}tdD ]}tj|j||f< q"| jj dd\}}i }|j
|
dkr`ddi}t d ¡ |j jf d|dd| W 5 Q R X dd d
|
d g}t|D ]r\}}t|} d | |< d d |jD }
|
|
|
| ksØtdd | ¡ ¡ D } |d |
||d d
} | | ks¤tq¤d S )Nr r )Zncolsz3.3Ú normalizeT)r rC rL Ú0Ú1Ú2Ú3r¦ c S s g | ] }| ¡ qS rd r" r~ rd rd re rù s z6TestDataFramePlots.test_pie_df_nan.<locals>.<listcomp>c S s g | ] }| ¡ qS rd r" r~ rd rd re rù s r )r@ r rM rN rO rX rr Zilocr[ r Ú __version__rD Úassert_produces_warningrF r{ r¢ rP rÒ r] r§ r¨ ) r_ rB r` r$ rb r^ rÊ Z base_expectedrC r rc Z result_labelsÚexpected_labelsrd rd re Útest_pie_df_nanz s&
|
$ z"TestDataFramePlots.test_pie_df_nanc C sö t d¡t ddd¡d}t|}t d¡d t d¡d d}t|}t|j|dd}| j|dd d
|
t|j|ddd }| j|dd d
|
t|j|dd }| j|dd d
|
t|d j||ddd}| j|d d d
|
t|d jt d¡d d}| j|dd d
|
t|jt d¡d d}| j|dd d
|
dD ]`}| ¡ }t d¡d ||< t|j|d}| j|dd d
|
t|jdd|d}| j|dd d
|
q*t t
|
¡ |jt j d¡d W 5 Q R X tdgd dgd d}t t ¡ |j|d W 5 Q R X d S )Nr r r r" çÉ?çÙ?T)Úyerrr r ©Úxerrrà )rà r r )rà r r r< )rà râ rÍ r r$ ©rà )r r )rà u 誤差r# )r$ r# rà rW Zzzz)rM r r Úonesr rF Ú_check_has_errorbarsÚcopyrD Zexternal_error_raisedr¥ rN r r )r_ rß r` Úd_errÚdf_errrC rà Zs_dfrd rd re Útest_errorbar_plot sD ÿ z%TestDataFramePlots.test_errorbar_plotr< rñ c C s$ t d¡t ddd¡d}t|}t d¡d t d¡d d}t|}t|j|d |d}| j|dd d
|
t|j||d}| j|dd d
|
t|j|||d }| j|d d d
|
t|j|d |d |d }| j|d d d
|
t|jdd|d }| j|d d d
|
t|jd ||d |d}| j|ddd
|
d S )Nr r r r" rÞ rß r# ©rà rÍ r rá )rà râ rÍ )râ rà rÍ T)r rà râ r rÍ r )rM r r rä r rF rå )r_ rÍ rß r` rç rè rC r^ rd rd re Ú"test_errorbar_plot_different_kindsÇ s. úz5TestDataFramePlots.test_errorbar_plot_different_kindszIterator is consumed)r rJ c C sd t ¡ R t d¡t ddd¡d}t|}t|jt dt |¡d}| j
|
|ddd W 5 Q R X d S ) Nr r r r" r rã r rá ) ÚwarningsÚcatch_warningsrM r r r rF Ú itertoolsr r\ rå )r_ rß r` rC rd rd re Útest_errorbar_plot_iteratorè s
|
|
z.TestDataFramePlots.test_errorbar_plot_iteratorc C sr t t tj dd¡¡}t t tj dd¡¡}t|j|d}| j|ddd t|jddd}| j|ddd d S )Nr' r rã r rá r )r$ rà )r rM ÚabsrN r r rF rå )r_ r` rè rC rd rd re Ú'test_errorbar_with_integer_column_namesò s z:TestDataFramePlots.test_errorbar_with_integer_column_namesc
|
C s. t t tj dd¡¡}t t tj dd¡¡ddgd}ddg}|D ]$}t|j||d}| j|ddd qDtd
|
dd d }|j |d d |j |d d t|j|dd}| j|ddd t
|
d¡t
|
ddd¡d}t |}t d¡d t d¡d d}t |}||fD ]$} t|j| d}| j|ddd qd S )Nr' r! r r r rL r< rê rá ú1/1/2000ÚM)r r T)Zinplacer r r" rÞ rß )r# r rã r ) r rM rð rN r r rF rå r
|
r r rä )
|
r_ r` rè ÚkindsrÍ rC Úixrß rç Úerrrd rd re Ú"test_errorbar_with_partial_columnsû s$ z5TestDataFramePlots.test_errorbar_with_partial_columnsc C s. t d¡t ddd¡d}t d¡d t d¡d d}tddd d
|
}t||d }t||d }t|j||d }| j|dd d t|j||d }| j|dd d t|jd|d |d}| j|ddd t|jdd|d}| j|ddd t|j||d }| j|dd d t|jd||dd}| j|ddd d S )Nr r r r" rÞ rß rò z1/1/2001ró )r r( rê r rá r$ r# )r$ rà rÍ r T)r rÍ rà r )rM r rä r
|
r r rF rå ) r_ rÍ rß rç rõ rÞ Ztdf_errrC r^ rd rd re Útest_errorbar_timeseries s. ûz+TestDataFramePlots.test_errorbar_timeseriesc C sÐ t j d¡ t j ddd¡}tt d¡ dd¡j}|j||d d}|j d
|
¡ d j d d
df }|dd d
df t ddg¡ }t ||¡ t d ¡}tjt|d
|
|j|jd W 5 Q R X t ¡ d S ) Nr r! r r* rM )rà râ r r zCAsymmetrical error bars should be provided with the shape (3, 2, 5)r% rã )rM rN r rO r r ZreshapeÚTrF r) Ú get_pathsZverticesrq rD r r
|
r rI rJ r¥ r« )r_ rö r` rC Zyerr_0_0Z expected_0_0ra rd rd re Útest_errorbar_asymmetrical4 s ÿz-TestDataFramePlots.test_errorbar_asymmetricalc C s t tj dd¡ttjd d
d}t|jdd t|j|d t
|
d ¡@ | ¡ }t |j dksft t ||j¡ t |j dkst W 5 Q R X d S )Nr' r! r( T)Útabler r )r rM rN rO rP rQ rR r rF rD rÛ r\ Ztablesr] r rü rù r rd rd re Ú
|
test_tableI s $ z TestDataFramePlots.test_tablec s t t tj dd¡¡tdddgd}t t tj dd¡¡d tdddgd}t|jjddd j ddd t|jjdd|d j d
|
dd t|jjdd|d j dd
|
d t|jjdd||d j d
|
d
|
d d fdd }t t tj dd¡¡dddddgd}|jjdddddd j d
|
d
|
d | j
|
dd d | j
|
ddd |jjddddd j dd
|
d | j
|
ddd d S ) Nr* r r# r$ r± r" r rá )r# r$ râ r )r# r$ rà )r# r$ râ rà Úhas_xerrc sx g } fddj D d }|D ]"}t|r8| |¡ q | |¡ q fdd|D }j|t |gt| ¡d d S )Nc s g | ]}t | d r|jqS )F)r´ r¬ rZ ©Úhas_errrd re rù i s z[TestDataFramePlots.test_errorbar_scatter.<locals>._check_errorbar_color.<locals>.<listcomp>r c s g | ]}| j kr|qS rd )r) r~ ©rC rd re rù o s
|
rÄ )Ú
|
containersr ÚextendÚappendr8 rM rq r\ )r r r r¬ ZerrsÚelZ err_lines©rC r_ rÿ re Ú_check_errorbar_colorg s ÿzGTestDataFramePlots.test_errorbar_scatter.<locals>._check_errorbar_colorr' r
r r rß rÏ r r4 )r# r$ râ rà r rÿ Zhas_yerrrà )r# r$ rà r¬ )rþ ) r rM rð rN r rX r rF r| rå r )r_ r` rè r rd r re Útest_errorbar_scatterU s< ÿ ÿ ÿz(TestDataFramePlots.test_errorbar_scatterc C sH t dddgdd}tjttfdd |jdd d
|
d d W 5 Q R X d S ) Nr r r! r rØ z'unknown' is not ar% r
r Úunknownr| )r# r$ r3 rÍ )r rI rJ r¥ ÚKeyErrorrF r rd rd re Ú test_scatter_unknown_colormap s z0TestDataFramePlots.test_scatter_unknown_colormapc
|
s dd l m} | d¡ t \}}tddddddgddddddgddddddgddddddgd } fd
|
d }|D ]}|jd d d|dd qr| | ¡ ¡ || t ¡ t \}}t t
|
¡ |jd|dd}W 5 Q R X || t ¡ t \}}|D ]}|jd d d|d qø| | ¡ ¡ |D ]T}t |j dks:t j| ¡ dd j| ¡ dd j|jdddd q"t ¡ d S )Nr r+ r r r! r r* r8 ©r
r r rß c s® | D ](}t |jdkst j| ¡ dd q| d | d fD ],} j| ¡ dd j|jdddd q>| d | d fD ],} j| ¡ dd j|jdddd q|d S ) Nr Tr1 r r Fr2 r! )r\ r¬ r] rS ri rU ©r^ rC ©r_ rd re Ú_check s z5TestDataFramePlots.test_sharex_and_ax.<locals>._checkr
r r/ T)r# r$ r/ rC Úsharex)r rC r ©r# r$ r/ rC r1 r2 )r r r« Ú _generate_4_axes_via_gridspecr rF Ú tight_layoutr rD rÛ Ú UserWarningr\ r¬ r] rS ri rU ©r_ r[ Úgsr^ r` r rC rd r re Útest_sharex_and_ax s>
|
|
üÿ
|
|
z%TestDataFramePlots.test_sharex_and_axc
|
s dd l m} t \}}tddddddgddddddgddddddgddddddgd} fd d
|
}|D ]}|jd d d |dd qh| | ¡ ¡ || t ¡ t \}}t t
|
¡ |jd|dd}W 5 Q R X | | ¡ ¡ || t ¡ t \}}|D ]}|jd d d |d qü| | ¡ ¡ |D ]T}t |j dks>t j| ¡ dd j| ¡ dd j|jdddd q&d S )Nr r r r! r r* r8 r c s | D ]>}t |jdkst j| ¡ dd j|jdddd q| d | d fD ]} j| ¡ dd qT| d | d fD ]} j| ¡ dd q|d S ) Nr Tr1 r2 r r r! F)r\ r¬ r] rS rU ri r r rd re r Î s z5TestDataFramePlots.test_sharey_and_ax.<locals>._checkr
r r/ T)r# r$ r/ rC ré )r rC ré r r1 r2 )r r r r rF r r rD r« rÛ r r\ r¬ r] rS ri rU r rd r re Útest_sharey_and_ax½ s<
|
üÿ
|
|
|
z%TestDataFramePlots.test_sharey_and_axc
|
C sä i }t jjD ]}i }|dkr`ttjjddtjjddt d¡tjjdd d}ddd}n |dkrvt ¡
|
¡ }nt ¡ }t |j f d |i|¡||< q t ¡ t ¡ d
|
}| ¡ D ]$}tjt|d |j W 5 Q R X qºd S ) z3Check that every plot type gets properly collected.)r} r| r{ r rs r rl rm r" rà rÍ z)weakly-referenced object no longer existsr% N)r r² Z
|
_all_kindsr rM rN ru r rD rE rð ÚweakrefÚproxyrF r« ÚgcZcollectr= rI rJ ÚReferenceErrorr¬ )r_ ÚresultsrÍ Úargsr` ra Z result_valuerd rd re Útest_memory_leakò s* ýÿ z#TestDataFramePlots.test_memory_leakc
|
sB ddl m dd lm ttj d¡tdddd}t tj dd¡|j
|
t dd } fd
|
d } fd d }| | fD ]\}}|j |d}t |jdks t|j |d}t |jdks¾t||fD ]>}| j| ¡ dd | j| ¡ dd | j|jdddd qÆt ¡ qz| | fD ]\}}|j d||gd}t |jdksJtt |jdks^t|D ]@}| j| ¡ dd | j| ¡ dd | j|jdddd qbt ¡ q | \}}t t¡ |j d||gddd}W 5 Q R X t |d jdksþtt |d jdkst||fD ]}| j| ¡ dd q | j|d ¡ dd | j|d jdddd | j|d ¡ dd | j|d jdddd t ¡ | \}}t t¡ |j d||gddd}W 5 Q R X t |d jdksîtt |d jdkst| j|d ¡ dd | j|d ¡ dd ||fD ].}| j| ¡ dd | j|jdddd q:t ¡ fdd} | }t tj dd¡|j
|
t dd }|j d|d}|D ]T}t |jdksÎt| j| ¡ dd | j| ¡ dd | j|jdddd q¶t ¡ | }t t¡ |j d|ddd}W 5 Q R X |D ]}t |jdksFtqF|d |d fD ]}| j| ¡ dd qr|d |d fD ]}| j| ¡ dd q|d |d fD ].}| j| ¡ dd | j|jdddd qÆ|d |d fD ].}| j| ¡ dd | j|jdddd qt ¡ d S )Nr )Úgridspecr' rò )r r( r ÚABr± c sL dd¡} ¡ }| | d d
d d
f ¡}| | dd d
f ¡}||fS )Nr! r r ©ÚGridSpecrÇ r¡ ©r rb rÔ rÕ ©r r[ rd re Ú_get_vertical_grid s
|
zHTestDataFramePlots.test_df_gridspec_patterns.<locals>._get_vertical_gridc sL dd¡} ¡ }| | d d
d d
f ¡}| | d d
df ¡}||fS )Nr r! r r" r$ r% rd re Ú_get_horizontal_grid% s
|
zJTestDataFramePlots.test_df_gridspec_patterns.<locals>._get_horizontal_gridr r Tr1 r2 rB )r rC r ré Fc st dd¡} ¡ }| | d d
d d
f ¡}| | d d
df ¡}| | dd d
f ¡}| | d ¡}||||fS )Nr! r r r" )r rb rÔ rÕ Zax3Zax4r% rd re Ú_get_boxed_gridb s zETestDataFramePlots.test_df_gridspec_patterns.<locals>._get_boxed_gridr ÚABCDr! )r@ r r r r rM rN r r
|
r r) rP rF r\ r¬ r] rS ri rU rD r« rÛ r )
|
r_ Útsr` r& r' rÔ rÕ rC r^ r( rd r% re Útest_df_gridspec_patterns s
|
|
|
z,TestDataFramePlots.test_df_gridspec_patternsc C s4 | j tdddgdddgdtjjdddd d S )
|
Nr r r! r rØ r
r r" )Úkws)Z_check_grid_settingsr r r² Z_dataframe_kindsr rd rd re Útest_df_grid_settings s
|
ýz(TestDataFramePlots.test_df_grid_settingsc C s@ | j ¡ \}}| ddddg¡ ttj d¡j|d ttj d¡tj d¡d}| j
|
¡ }| d¡}|jd|dd dd
|
d | j ¡ \}}d d l m } ||}|j dddd}ttj d¡j|d ttj d¡j|d | j ¡ \}}d dlm} ||dddd}ttj d¡j|d ttj d¡j|d d S )NrÞ r' r r= rØ )r r r r r| r
r Zhsv)rÍ rC r# r$ r r6 r )Úmake_axes_locatabler< z5%g©?)r1 Úpad)Ú
|
inset_axesz30%rº r! )rï rY rv )r[ r Zadd_axesr rM rN rO rF r r rÇ Zmpl_toolkits.axes_grid1r. Z append_axesZ%mpl_toolkits.axes_grid1.inset_locatorr0 ) r_ rb rC r` r. ÚdividerZcaxr0 Ziaxrd rd re Útest_plain_axes s$
|
|
z"TestDataFramePlots.test_plain_axesÚmethodc C sp t tj dd¡tddjdd djdd d }d
|
}d d g}||d d}t|j|f |}| j|j |d d S )NrM r r! r c S s
|
| j ¡ S r% )rm r ©r` rd rd re Ú<lambda>³ ó zBTestDataFramePlots.test_secondary_axis_font_size.<locals>.<lambda>©rn c S s
|
| j d S )Ngñ?r7 r4 rd rd re r5 ´ r6 )ro r rn ro T)râ rQ Z
|
mark_right)r^ rT )
|
r rM rN r rP Zassignr´ rF rH rã )r_ r3 r` rQ rË rÊ rC rd rd re Ú test_secondary_axis_font_size® s ÿþÿ z0TestDataFramePlots.test_secondary_axis_font_sizec C s t dddgdddgdddgd }|jjd
|
d }| d d¡ d d | ¡ D }tt|| ¡ }|d dkspt|d dkst|d dkstd S )Nr! r r rÎ é ZMondayZTuesdayZ Wednesday)ZsalesZvisitsÚdayr: r r c S s g | ] }| ¡ qS rd r" ©r÷ Útrd rd re rù Ê s zATestDataFramePlots.test_x_string_values_ticks.<locals>.<listcomp>r¹ rº r» ) r rF rà Úset_xlimrU rl rW Ú
|
get_xticksr] )r_ r` rC Ú xticklabelsÚlabels_positionrd rd re Útest_x_string_values_ticks¾ s ýÿ z-TestDataFramePlots.test_x_string_values_ticksc C s® t ddgddgg¡}ttj dd¡ddg|d}| ¡ }| d d¡ d
|
d | ¡ D }t t
|
|| ¡ }|d d kszt |d dkst |d dkst |d dksªt d S )Nr¯ iÝ r r r rl rm r> r c S s g | ] }| ¡ qS rd r" r; rd rd re rù Ø s zETestDataFramePlots.test_x_multiindex_values_ticks.<locals>.<listcomp>z (2012, 1)r¹ z (2012, 2)rº z (2013, 1)r» z (2013, 2)r¼ ) r Z from_productr rM rN r rF r= rU rl rW r> r] )r_ r) r` rC r? r@ rd rd re Ú test_x_multiindex_values_ticksÑ s z1TestDataFramePlots.test_x_multiindex_values_ticksc C sL t ddgddgd}|j|d}| ¡ }|d dk s8t|d dksHtd S )Nr r r r( ro r )r rF r½ r] )r_ rÍ r` rC Úxlimsrd rd re Útest_xlim_plot_lineà s
|
z&TestDataFramePlots.test_xlim_plot_linec C sÈ | j ¡ \}}ddddg}tddddgdd dd
|
gddd d gd dddgd|d}|dddg jj|dd |dg j|dd | ¡ }|d dk st|d dks¦tdd | ¡ D }||ksÄtd S ) NZk1Zk2Zk3Zk4iè r¯ iÜ i ix i¸ i@ i° r r! r )Ús1Ús2Ús3râ r( rE rF rG F)rC rÐ râ T)rC râ r c S s g | ] }| ¡ qS rd r" r; rd rd re rù s zWTestDataFramePlots.test_xlim_plot_line_correctly_in_mixed_plot_type.<locals>.<listcomp>)r[ r r rF r< r½ r] rU )r_ rb rC Zindexesr` rC r? rd rd re Ú0test_xlim_plot_line_correctly_in_mixed_plot_typeê s"
|
|
|
|
üù zCTestDataFramePlots.test_xlim_plot_line_correctly_in_mixed_plot_typec C sl t dgtd}|jst| ¡ }t| ¡ dks4t| ¡ d }t| ¡ dksTtt| ¡ dkshtd S )NZfoo)r? rh r r ) r ÚintÚemptyr] rF r\ r rª rÚ )r_ r` rC rL rd rd re Útest_plot_no_rows s
|
z$TestDataFramePlots.test_plot_no_rowsc C s4 t dddg}tjtdd | ¡ W 5 Q R X d S )Nr
r r r¶ r% )r rI rJ r rF r rd rd re Útest_plot_no_numeric_data s z,TestDataFramePlots.test_plot_no_numeric_data)rL r< rñ rz rn r| rà r{ c C sÀ t d¡t d¡d t d¡d t d¡t d¡d}t|}|jddg|d}t|dks`tdd gd
|
d gd gf}t||D ]<\}}|d kr| j||d |dkr~t|jt|ks~tq~d S )Nr' r )r
r r rß rÏ )r rÏ )r rß ©r rÍ r! r rÏ r rß r
r{ rp rL ) rM r r rF r\ r] rW rq r¬ )r_ rÍ rß r` r^ rÜ rC rd rd rd re Útest_group_subplot s ûz%TestDataFramePlots.test_group_subplotc C s< t td}d}tjt|d |jdgd W 5 Q R X d S )Nr z!An iterable subplots for a Seriesr% ©r
r@ )r rX rI rJ ÚNotImplementedErrorrF )r_ Zserra rd rd re Ú(test_group_subplot_series_notimplemented+ s z;TestDataFramePlots.test_group_subplot_series_notimplementedc C sL t t d¡t ddg¡d}d}tjt|d |jdgd W 5 Q R X d S )Nr r )r r r z6An iterable subplots for a DataFrame with a MultiIndexr% r@ ) r rM Úeyer rY rI rJ rP rF rÄ rd rd re Ú,test_group_subplot_multiindex_notimplemented1 s z?TestDataFramePlots.test_group_subplot_multiindex_notimplementedc C sF t t d¡ddgd}d}tjt|d |jdgd W 5 Q R X d S )Nr r
r z4An iterable subplots for a DataFrame with non-uniquer% rO r@ )r rM rR rI rJ rP rF rÄ rd rd re Ú0test_group_subplot_nonunique_cols_notimplemented7 s zCTestDataFramePlots.test_group_subplot_nonunique_cols_notimplementedzsubplots, expected_msg)é{ z(subplots should be a bool or an iterable)r
ú!each entry should be a list/tuple))r rV )rO rV c C sH t d¡t d¡d}t|}tjt|d |j|d W 5 Q R X d S )Nr' rØ r% r@ ©rM r r rI rJ r¥ rF )r_ r Z expected_msgrß r` rd rd re Ú test_group_subplot_bad_input= s z/TestDataFramePlots.test_group_subplot_bad_inputc C sJ t d¡t d¡d}t|}tjtdd |jdgd W 5 Q R X d S )Nr' rØ z Column label\(s\) \['bad_name'\]r% )r
Zbad_namer@ rW ©r_ rß r` rd rd re Ú&test_group_subplot_invalid_column_nameP s z9TestDataFramePlots.test_group_subplot_invalid_column_namec C sT t d¡t d¡t d¡d}t|}tjtdd |jddgd W 5 Q R X d S )Nr' )r
r r z should be in only one subplotr% rØ )r
r r@ rW rY rd rd re Ú$test_group_subplot_duplicated_columnW s z7TestDataFramePlots.test_group_subplot_duplicated_column)r` r| r} c C sL t d¡t d¡d}t|}tjtdd |jdg|d W 5 Q R X d S )Nr' rØ z1When subplots is an iterable, kind must be one ofr% rM rW )r_ rÍ rß r` rd rd re Ú test_group_subplot_invalid_kind^ s ÿz2TestDataFramePlots.test_group_subplot_invalid_kindz index_name, old_label, new_label)Nr¦ Únew)Úoldr^ r] )Nr¦ r¦ )Nr¦ r r¦ c C s t ddgddggddgd}||j_|j|d}| ¡ |ks@t| ¡ dksPt|j|||d }| ¡ t|kstt| ¡ t|kstd S )
|
Nr r r* úType AúType Br ro r¦ )rÍ rÓ Úxlabel)r r) r rF Ú
|
get_xlabelr] r7 Ústr)r_ rÍ Z
|
index_nameZ old_labelZ new_labelr` rC rd rd re Ú(test_xlabel_ylabel_dataframe_single_plotg s z;TestDataFramePlots.test_xlabel_ylabel_dataframe_single_plotzxlabel, ylabel)NN)úX LabelN)NúY Label)re rf r| r} c C st d}d}t ddgddgg||gd}|j|||||d}| ¡ |d krJ|n|ksTt| ¡ |d krf|n|ksptd S )Nr_ r` r r r* r )rÍ r# r$ ra rÓ )r rF rb r] r7 )r_ rÍ ra rÓ ZxcolZycolr` rC rd rd re Ú'test_xlabel_ylabel_dataframe_plane_plot s z:TestDataFramePlots.test_xlabel_ylabel_dataframe_plane_plotrâ )FTc C sj t dgj|dddgd}|jjD ]B}|j ¡ r"| ¡ dks@t| ¡ dksPt| ¡ d dks"tq"d S )Nr rÍ )r r- r³ )râ rÓ r. r+ )
|
r rF rÇ r^ r Z get_visibler7 r] rç Z
|
get_yticks)r_ râ Zax_dfrC rd rd re Útest_secondary_y s
|
ÿ
|
z#TestDataFramePlots.test_secondary_y)NNNN)oÚ__name__Ú
|
__module__Ú __qualname__rI ÚmarkZxfailZslowrf rx r r r r 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> rC rG rX re rj rm ÚtdZskip_if_no_scipyrr rw r rM 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ë r¥ rï rñ r÷ rø rû rý r r r r r r+ r- r2 r8 rA rB rD rH rK rL rN rQ rS rT rX rZ r[ r\ rd rg rh rd rd rd re r ' s f
|
ÿ
|
|
%
|
/'þ % þ
|
1
|
|
|
' ÿ
|
ú
|
üþ
|
|
þþ
|
|
ýþ
|
$
|
|
|
ýþ
|
*
|
.
|
|
|
! ,65
|
"s
|
|
|
ÿ üþ
|
ûþ
|
üþ r c C sr dd l } dd l}dd lm} | j dd¡}| |d ¡}| |d ¡}| |d ¡}| |d ¡}|||||gfS )Nr r rt )r r r rA )r@ Zmatplotlib.gridspecr r r r# Zsubplot)rB r@ r[ r Zax_tlZax_llZax_trZax_lrrd rd re r ¤ s r )&Ú__doc__r r r rî r
|
rQ rì r ÚnumpyrM rI Z pandas.util._test_decoratorsÚutilZ_test_decoratorsrm Zpandas.core.dtypes.apir Zpandasrt r r r r r r
|
r Zpandas._testingZ_testingrD Z pandas.tests.plotting.commonr r Zpandas.io.formats.printingr Zskip_if_no_mplr r rd rd rd re Ú<module> sH $
|
