U
|
®ý°d±' ã @ s d dl Z d dlZd dlZd dlZd dlmZ d dlZd dlm Z m
|
Z
|
m Z m Z d dl mZ G dd dZej deeeg¡dd ZdS ) é N)Ú
|
skip_if_no)Ú DataFrameÚIndexÚSeriesÚ
|
date_rangec ' @ s< e Zd Zdd Zdd Zdd Zdd Zd d
|
Zej
|
d e d ¡e d ¡ed ddgd e d ¡e d ¡e d ¡e d ¡e d ¡e d ¡e d ¡eddgedd edD ej edd¡ej eddddgd¡g¡dd Zej
|
d ee d edgg¡d d Z d!d" Z d#d$ Z d%d& Z d'd( Z!d)d* Z"d+d, Z#d-d. Z$ej
|
d/d0e g¡d1d2 Z%ej
|
d/d3d4g¡d5d6 Z&d7d8 Z'e(d9d:d; Z)d<d= Z*ej
|
d>d?d@dAdBg¡dCdD Z+dEdF Z,dGdH Z-ej
|
dIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgdhdidjdkdldmg$¡ej
|
d/e.e/e0e g¡dndo Z1d3S )pÚTestSeriesMiscc C s@ t td}dt|ks tdt|ks,tdt|ks<td S )NZabcdÚstrÚdtÚcat©r ÚlistÚdirÚAssertionError©ÚselfÚs© r úSd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/series/test_api.pyÚtest_tab_completion s z"TestSeriesMisc.test_tab_completionc C sD t tddd}dt|ks tdt|ks0tdt|ks@td S )Nú1/1/2015é ©Úperiodsr r r
|
)r r r r r r r r Útest_tab_completion_dt s z%TestSeriesMisc.test_tab_completion_dtc C sD t tddd}dt|ks tdt|ks0tdt|ks@td S )NZabbcdÚcategory©Údtyper
|
r r r r r r r Útest_tab_completion_cat# s z&TestSeriesMisc.test_tab_completion_catc C sJ t tddd d¡}dt|ks&tdt|ks6tdt|ksFtd S )Nr r r r r
|
r r )r r Úastyper r r r r r Útest_tab_completion_cat_str+ s z*TestSeriesMisc.test_tab_completion_cat_strc C s\ ddddddddd d
|
d g }t td d ¡}tdd |j ¡ D }t |tt|¡ d S )NÚ
|
categoriesÚcodesZorderedZset_categoriesZadd_categoriesZremove_categoriesZrename_categoriesZreorder_categoriesZremove_unused_categoriesZ
|
as_orderedZ as_unorderedZaabbcder c S s h | ]}| d ¡s|qS )Ú_)Ú
|
startswith)Ú.0Úrr r r Ú <setcomp>C s
|
zFTestSeriesMisc.test_tab_completion_with_categorical.<locals>.<setcomp>) r r r Úsortedr
|
Ú__dir__ÚtmÚassert_almost_equalÚset)r Z
|
ok_for_catr Úresultsr r r Ú$test_tab_completion_with_categorical2 s õz3TestSeriesMisc.test_tab_completion_with_categoricalÚindexé
|
ÚfooÚbarZbazé TFc C s g | ]}d | qS )Úar )r$ Úir r r Ú
|
<listcomp>T s zTestSeriesMisc.<listcomp>ée ÚABCDZEFGHr é é c C sh t |td}t|}t|jjddD ]<\}}|dk rVt|trb| ¡ rb||ksbt q&||ks&t q&d S )N©r. r r )Úleveléd )
|
r Úobjectr Ú enumerater. ÚuniqueÚ
|
isinstancer Ú isidentifierr )r r. r Zdir_sr4 Úxr r r Útest_index_tab_completionF s z(TestSeriesMisc.test_index_tab_completionÚserr c C s* d}t jt|d t| W 5 Q R X d S )Nzunhashable type: 'Series'©Úmatch)ÚpytestÚraisesÚ TypeErrorÚhash©r rD Úmsgr r r Útest_not_hashablec s z TestSeriesMisc.test_not_hashablec C s t |j|¡ d S ©N)r) Zassert_contains_allr. )r Údatetime_seriesr r r Ú test_containsi s z TestSeriesMisc.test_containsc C sj t ddtjg}t |jdd|jdd¡ | ¡ d¡dksBt| d¡dksTt| d¡dksftd S )Nr8 r2 Úrows)Zaxisr. r9 r )
|
r ÚnpÚnanr) Úassert_series_equalZdropnaÚsumr Z_get_axis_numberZ_get_axis_namer r r r Útest_axis_aliasl s
|
z TestSeriesMisc.test_axis_aliasc C s t tj¡std S rN )ÚpydocÚgetdocr r. r )r r r r Útest_class_axiss s z TestSeriesMisc.test_class_axisc C sP t tj dd¡dddgtdddd}d d
|
}| |¡}| ¡ }t ||¡ d S ) Niè r9 ÚAÚBÚCz1/1/2000r )Úcolumnsr. c S s | | ¡ S rN )Úidxmax)rB r r r Úf s z-TestSeriesMisc.test_ndarray_compat.<locals>.f) r rR ÚrandomÚrandnr ÚapplyÚmaxr) rT )r Ztsdfr_ ÚresultÚexpectedr r r Útest_ndarray_compatx s
|
ý
|
z"TestSeriesMisc.test_ndarray_compatc C s@ t tj d¡}t t |¡}t dtddd}t ||¡ d S )Nr/ r8 Úfloat64r: )r rR r` ra Z ones_likeÚranger) rT )r r rd re r r r Ú test_ndarray_compat_like_func s z,TestSeriesMisc.test_ndarray_compat_like_funcc C s2 t tj d¡}t |jdd|jjdd¡ d S )Nr/ ÚF)Úorder)r rR r` ra r) r* ZravelÚvaluesr r r r Útest_ndarray_compat_ravel s z(TestSeriesMisc.test_ndarray_compat_ravelc C s t td}|jstd S )Nr )r r= Úemptyr )r Zs_emptyr r r Útest_empty_method s
|
z TestSeriesMisc.test_empty_methodr Úint64c C s t dg|d}|jrtd S )Nr8 r: )r rn r )r r Z full_seriesr r r Ú test_empty_method_full_series s z,TestSeriesMisc.test_empty_method_full_seriesNZInt64c C s" t td|d}|jdks td S )Né r )r rh Úsizer )r r r r r r Útest_integer_series_size s z'TestSeriesMisc.test_integer_series_sizec C sF t ddgdd}|ji ks td|jd< |d }|jddiksBtd S )Nr r8 Úabc)ÚnameÚversion)r Úattrsr )r r rd r r r Ú
|
test_attrs¢ s
|
|
zTestSeriesMisc.test_attrsZjinja2c C s2 t td}tjd dd t |¡ W 5 Q R X d S )Nr F)Zcheck_stacklevel)r r= r) Úassert_produces_warningÚinspectÚ
|
getmembers©r rD r r r Útest_inspect_getmembers© s
|
z&TestSeriesMisc.test_inspect_getmembersc C s\ t jdddd}ttjjdd|d}d|jks4td}tj t
|
|d |j W 5 Q R X d S )
|
Nr r/ Z1s)Ústartr Úfreq)rs )r. r0 z&'Series' object has no attribute 'foo'rE ) ÚpdZtimedelta_ranger rR r` ÚnormalÚ__dict__r rG rH ÚAttributeErrorr0 )r ZtdirD rL r r r Útest_unknown_attribute° s z%TestSeriesMisc.test_unknown_attributeÚopÚyearÚdayÚsecondÚweekdayc C s4 d| d}t jt|d t|| W 5 Q R X d S )Nz"'Series' object has no attribute 'ú'rE )rG rH r Úgetattr)r r rO rL r r r Ú&test_datetime_series_no_datelike_attrs¹ s z5TestSeriesMisc.test_datetime_series_no_datelike_attrsc C s> t dddd}|jdks t|jdks,t|jdks:td S )NéÐ r8 r/ ©r Úmonthr )r r r r r r} r r r Ú)test_series_datetimelike_attribute_accessÀ s z8TestSeriesMisc.test_series_datetimelike_attribute_accessc C s8 t dddd}d}tjt|d |j W 5 Q R X d S )Nr r8 r/ r z*'Series' object has no attribute 'weekday'rE )r rG rH r r rK r r r Ú1test_series_datetimelike_attribute_access_invalidÇ s z@TestSeriesMisc.test_series_datetimelike_attribute_access_invalidzkernel, has_numeric_only)ZskewT)ÚvarT)ÚallF)ÚprodT)ÚanyF)ZidxminF)ÚquantileF)r^ F)ÚminT)ZsemT)ZmeanT)ZnuniqueF)rc T)rU T)ÚcountF)ZmedianT)ZstdT)ZbackfillF)ZrankT)Z
|
pct_changeF)ZcummaxF)ÚshiftF)ZdiffF)ZcumsumF)ZcumminF)ZcumprodF)ÚfillnaF)ÚffillF)ÚpadF)ZbfillF)ÚsampleF)ÚtailF)ÚtakeF)ÚheadF)ÚcovF)ÚcorrFc
|
C sT t dddg|d}|dkr"|f}nn|dkr2|f}n^|dkrB|f}nN|dkrPd}n@|d kr^d
|
}n2|d krld }n$|d kr~dgf}n|d krd}nd}t||}|sÌd}tjt|d ||ddi W 5 Q R X n|tkr d| d}tjt|d ||ddi W 5 Q R X nD||ddi}||ddi} t| t rBt || ¡ n|| ksPt d S )Nr r8 r Zcorrwithr£ r¢ Znth)r r )T)r r r )g à?r zP(got an unexpected keyword argument 'numeric_only'|too many arguments passed in)rE Z numeric_onlyTzSeries.z2 does not allow numeric_only=True with non-numericF)
|
r r rG rH rI r= r@ r) rT r )
|
r ÚkernelZhas_numeric_onlyr rD ÚargsÚmethodrL rd re r r r Útest_numeric_onlyÍ sB ,
|
|
ÿ
|
z TestSeriesMisc.test_numeric_only)2Ú__name__Ú
|
__module__Ú __qualname__r r r r r- rG ÚmarkÚ parametrizer) ZmakeStringIndexZmakeCategoricalIndexr Z makeDateIndexZmakePeriodIndexZmakeTimedeltaIndexZ makeIntIndexZ makeUIntIndexZmakeFloatIndexrh r Z
|
MultiIndexÚ from_tuplesÚziprC r r= rM rP rV rY rf ri rm ro rq rt ry r r~ r
r r r ÚboolÚintÚfloatr§ r r r r r s®
|
òþ
|
|
|
|
|
|
|
Üþ)r Ú converterc C s4 t t¡ | tdg| dks&tW 5 Q R X d S )Nr8 )r) rz Ú FutureWarningr r )r² r r r Útest_float_int_deprecated# s r´ )r{ rW ÚnumpyrR rG Z pandas.util._test_decoratorsr Zpandasr r r r r Zpandas._testingZ_testingr) r r« r¬ r° r± Úcomplexr´ r r r r Ú<module> s
|
