U
|
ý°d!È ã @ sî d dl m Z mZ d dlZd dlZd dlZd dlmZmZm Z m
|
Z
|
m Z m Z m Z mZmZmZmZmZmZmZmZmZ d dlmZ d dlmZ dd Z e Z G dd dZ G d d
|
d
|
Z G d d d Z G d d dZ!G dd dZ"G dd dZ#dS )é )ÚdatetimeÚ timedeltaN)Ú CategoricalÚ DataFrameÚ DatetimeIndexÚIndexÚNaTÚPeriodÚ PeriodIndexÚ
|
RangeIndexÚSeriesÚ TimedeltaÚTimedeltaIndexÚ TimestampÚ
|
date_rangeÚisnaÚtimedelta_rangeÚ to_timedelta)Únanopsc
|
s t jdddt jdddt jdddt jdddt jdddjddt jdddt jdddg} tj
|
d¡ fdd| D }| | }|S )Né
|
Úa)Únamez
|
US/Eastern©Útzc s g | ]}t |d dqS )r )Úindexr )r )Ú.0Úidx©Úarr© ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/reductions/test_reductions.pyÚ
|
<listcomp>, s z get_objs.<locals>.<listcomp>) ÚtmZ makeBoolIndexZ makeIntIndexZmakeFloatIndexZ makeDateIndexÚ tz_localizeZmakePeriodIndexZmakeStringIndexÚnpÚrandomÚrandn)ZindexesÚseriesÚobjsr r r Úget_objs s ù
|
r) c @ s e Zd Zej dddg¡ej de¡dd Zej dddg¡ej ddd d
|
ed d d fddg¡dd Z ej dddg¡dd Z
|
ej dddg¡ej dddg¡dd Z dd Z ej ddd gdd gg¡d d Z ej d d!d"g¡d#d$ Zd%d& Zd'S )(ÚTestReductionsÚopnameÚmaxÚminÚobjc C s~ t || }t|ts&t |j| }ntt |j| |jd}t |dd d k rn| d¡ d¡}|j|kszt n ||kszt d S )N)ZordinalÚfreqr úM8[ns]Úint64)
|
ÚgetattrÚ
|
isinstancer
|
Úvaluesr Zasi8r/ ÚastypeÚ_valueÚAssertionError)Úselfr+ r. ÚresultÚexpectedr r r Útest_ops6 s
|
zTestReductions.test_opsz
|
dtype, val)Úobjectç @)Úfloat64r= údatetime64[ns]éÛ é é )ÚInt64é )ÚbooleanTc s |} fdd}|d g d}|t || s2t|t ||ddsJt|g d}|t || sjt|t ||ddst dkrd S |d |g d}t || |ks²t|t ||ddsÊt|d |d g d}t || |ksðt|t ||dds
|
td S )Nc s. dkr| t kS dkr"| tjkS t| S d S )Nr? )rC rE )r ÚpdÚNAr )Úres©Údtyper r Ú check_missingU s
|
|
z4TestReductions.test_nanminmax.<locals>.check_missingrI F©Úskipnar< )r2 r7 )r8 r+ rJ ÚvalÚindex_or_seriesÚklassrK r. r rI r Útest_nanminmaxF s z TestReductions.test_nanminmaxc C sÚ |}|t krd| nd| }|ttdddg}t|| dksDtt||dd}|tkrlt |¡sxtn |dksxt|ttdddtg}t|| dks¢tt||dd}|tkrÊt |¡sÖtn |dksÖtd S ) NÚargr r@ rA rB FrL éÿÿÿÿ)r r r r2 r7 r r$ Úisnan)r8 r+ rO rP Úarg_opr. r9 r r r Útest_nanargminmaxq s z TestReductions.test_nanargminmaxrJ r0 zdatetime64[ns, UTC]c C s¨ |}|t krd| nd| }|g |d}t|| tks<tt||ddtksTttjtdd t|| W 5 Q R X tjtdd t||dd W 5 Q R X d S )NrR r rI FrL zempty sequence©Úmatch)r r2 r r7 ÚpytestÚraisesÚ
|
ValueError)r8 r+ rO rJ rP rU r. r r r Útest_nanops_empty_object s z'TestReductions.test_nanops_empty_objectc C s° t tjddd}| ¡ dks"t| ¡ dks2tt tjdtjdg}| ¡ dksVt| ¡ dksft|jd d
|
d kszt|jd d
|
d kstt tjg}| ¡ d ksªt| ¡ d ksºt|jd d
|
d ksÎt|jd d
|
d ksâtt ttd d dtd d dtg}| ¡ dkst| ¡ dks&t|jd d
|
d ks<t|jd d
|
d ksRtt tg}| ¡ d ksnt| ¡ d kst|jd d
|
d kst|jd d
|
d ks¬td S )Né r1 rI r é rB rD é FrL rS r@ rA ) r r$ ÚarangeÚargminr7 ÚargmaxÚnanr r )r8 r. r r r Útest_argminmax s.
|
z TestReductions.test_argminmaxzop, expected_colr Úbc C s\ t tdddddgd}|j tdd¡|d < t||d
|
d }|| d ¡}t ||¡ d S ) Nz2016-01-01 00:00:00r_ ÚUTC)Úperiodsr r )Úcolumnsi ©Úsecondsre rB ©Úaxis) r r r Úsubtractr r2 Úrenamer" Úassert_series_equal)r8 ÚopZ expected_colÚdfr9 r: r r r Útest_same_tz_min_max_axis_1µ s ÿz*TestReductions.test_same_tz_min_max_axis_1ÚfuncÚmaximumZminimumc C s> |}t dg¡ |¡}t|}tt|||}t ||¡ d S )NZ2019)rF Ú to_datetimer# r r2 r$ r" ro )r8 Ztz_aware_fixturers r rR r: r9 r r r Ú(test_numpy_reduction_with_tz_aware_dtypeÀ s
|
z7TestReductions.test_numpy_reduction_with_tz_aware_dtypec C sV t tddtgddtddtjgddd}ttddd}| ¡ }t ||¡ d S )NrB rD útimedelta64[ns]rI rC ©ÚAÚBr_ ) r r r r$ rc r Úsumr" ro )r8 rq r: r9 r r r Útest_nan_int_timedelta_sumÉ s þÿz)TestReductions.test_nan_int_timedelta_sumN)Ú__name__Ú
|
__module__Ú __qualname__rY ÚmarkÚ parametrizer( r; r rQ rV r\ rd rr rv r| r r r r r* 5 s2 ûþ
|
|
|
|
|
r* c @ s e Zd Zej ddddddg¡dd Zd d
|
Zej d d d g¡dd Zdd Z dd Z
|
ej ddddddg¡dd Z d d Z ej d d d g¡d d Z d d! Zd"d# Zd$d% Zd&d' Zej d d d g¡ej d(g egeeegg¡d)d* Zd+d, Zd-d. Zd/S )0ÚTestIndexReductionszstart,stop,step)r i r_ )iô r iúÿÿÿ)éÀ½ðÿé@B r^ )r r éüÿÿÿ)r r é c C sª t |||}|j ¡ }| ¡ }||ks*t|jdd}||ksBt|j ¡ }| ¡ }||ks`t|jdd}||ksxtt ||| }t| ¡ stt| ¡ s¦td S ©NFrL )r Z_valuesr, r7 r- r )r8 ÚstartÚstopÚstepr r: r9 Zresult2r r r Útest_max_min_rangeÛ s
|
|
z&TestIndexReductions.test_max_min_rangec C s t dddg}|jstt dtjddg}|jr4t||fD ]L}| ¡ tdksTt| ¡ tdksht| ¡ dksxt| ¡ dks<tq<d S )Nú1 daysú2 daysz3 daysr r rD )
|
r Úis_monotonic_increasingr7 r$ rc r- r r, ra rb ©r8 Úidx1Úidx2r r r r Útest_minmax_timedelta64ý s
|
|
z+TestIndexReductions.test_minmax_timedelta64rp r- r, c C s` t g }t|| tks tt tg}t|| tks:tt tttg}t|| tks\td S ©N)r r2 r r7 ©r8 rp r. r r r Ú!test_minmax_timedelta_empty_or_na s
|
z5TestIndexReductions.test_minmax_timedelta_empty_or_nac C s
|
t dddd}t |¡tdks$tt |¡tdks:td}tjt|d tj|dd W 5 Q R X tjt|d tj|dd W 5 Q R X t |¡dks tt
|
|¡d ks²td}tjt|d tj |dd W 5 Q R X tjt|d tj
|
|dd W 5 Q R X d S )
|
Nz
|
16815 daysz
|
16820 daysÚD©r/ ú$the 'out' parameter is not supportedrW r ©Úoutr] ) r r$ r- r r7 r, rY rZ r[ ra rb )r8 ÚtdÚerrmsgr r r Ú test_numpy_minmax_timedelta64 s z1TestIndexReductions.test_numpy_minmax_timedelta64c C s¦ t dd tdD }| ¡ }| ¡ }ttdd}||ks@t| ¡ ¡ }|d |ks\t| d¡}t t
|
dd ¡}||kst| ¡ }td
|
}||kst| ¡ ¡ }|d |ksºt| ¡ }td }||ksÖt| ¡ ¡ }|d |ksòt| ¡ }tt | ¡ j ¡ }||ks t| ¡ ¡ }|d |ks<tt td td g}| ¡ ¡ tddksltt td td tdg}| ¡ ¡ tddks¢td S )Nc S s" g | ]}t d t|| d qS ©Z20130101ri ©r r ©r Úir r r r! 1 s z:TestIndexReductions.test_timedelta_ops.<locals>.<listcomp>r é ri r g¹?i(
|
Úmsz00:00:09z00:01:21z
|
2015-02-03z
|
2015-02-07r^ )Údaysz
|
2015-02-15é )r ÚrangeÚdiffÚmeanr r r7 Úto_frameZquantiler r$ Z timedelta64Úmedianr{ ÚstdÚdropnar4 r )r8 Úsr r9 r: r r r Útest_timedelta_ops- sB ÿ
|
ÿz&TestIndexReductions.test_timedelta_opsr+ ZskewZkurtZsemÚprodÚvarc C s t dd tdD }| ¡ }d d| dd| dd | d
|
g¡}tjt|d t|| W 5 Q R X tjt|d t| ¡ |d d W 5 Q R X d S )Nc S s" g | ]}t d t|| d qS r r r r r r r! e s zDTestIndexReductions.test_invalid_td64_reductions.<locals>.<listcomp>r ú|zreduction operation 'z ' not allowed for this dtypezcannot perform z with type timedelta64\[ns\]z does not support reduction 'ú'rW F)Z numeric_only) r r¦ r§ ÚjoinrY rZ Ú TypeErrorr2 r© )r8 r+ r r Úmsgr r r Ú test_invalid_td64_reductionsb s ÿ
|
|
|
ýÿz0TestIndexReductions.test_invalid_td64_reductionsc C s¢ |}t dddg|d}|js tt dtddtg|d}|jr@t||fD ]T}| ¡ td|dksdt| ¡ td|dks|t| ¡ dkst| ¡ dksHtqHd S )Nú
|
2011-01-01ú
|
2011-01-02ú
|
2011-01-03r r rD ) r r r7 r r- r r, ra rb )r8 Ztz_naive_fixturer r r r r r r Útest_minmax_tzw s
|
ÿ
|
z"TestIndexReductions.test_minmax_tzc C s` t g }tt|| s tt tg}tt|| s:tt tttg}tt|| s\td S r )r r r2 r7 r r r r r Útest_minmax_nat_datetime64 s
|
z.TestIndexReductions.test_minmax_nat_datetime64c C s: t dddg}|j ¡ }t |¡}||ks.t|j ¡ }t |¡}||ksNtd}tjt|d tj|dd W 5 Q R X tjt|d tj|dd W 5 Q R X |j ¡ }t |¡}||ksÂt|j
|
¡ }t
|
|¡}||ksâtd}tjt|d tj
|
|dd W 5 Q R X tjt|d tj |dd W 5 Q R X d S )NrB rD r_ r rW r r ) r r4 r, r$ r7 r- rY rZ r[ rb ra )r8 r r: r9 r r r r Útest_numpy_minmax_integer s.
|
|
|
|
|
|
|
|
z-TestIndexReductions.test_numpy_minmax_integerc C s t ddd}t |¡}|dks"tt |¡}|dks8td}tjt|d tj|dd W 5 Q R X tjt|d tj|dd W 5 Q R X d S )Nr r r_ r¢ r rW r )r r$ r, r7 r- rY rZ r[ )r8 r r9 r r r r Útest_numpy_minmax_rangeµ s
|
|
z+TestIndexReductions.test_numpy_minmax_rangec C s t ddd}t |¡tdks"tt |¡tdks8td}tjt|d tj|dd W 5 Q R X tjt|d tj|dd W 5 Q R X t |¡dkstt
|
|¡d
|
ks°td}tjt|d tj |dd W 5 Q R X tjt|d tj
|
|dd W 5 Q R X d S ) Nú
|
2016-01-15ú
|
2016-01-20©r Úendz2016-01-15 00:00:00z2016-01-20 00:00:00r rW r r r] ) r r$ r- r r7 r, rY rZ r[ ra rb )r8 Zdrr r r r Ú test_numpy_minmax_datetime64È s z0TestIndexReductions.test_numpy_minmax_datetime64c C sÒ t tdddgdd}|jr t|dd
js0tt dtddtgdd}|jrPt||fD ]4}| ¡ tdddkstt| ¡ tdddksXtqX| ¡ dkst| ¡ dks®t| ¡ dks¾t| ¡ d ksÎtd S )
|
Nr· r¸ r¹ r r rB r r_ rD ) r
|
r r r7 r- r r, ra rb r r r r Útest_minmax_periodß s
|
ÿ
|
z&TestIndexReductions.test_minmax_periodÚdatac C s( t |dd}t|| }|tks$td S )NÚMr )r
|
r2 r r7 )r8 rp rÄ r. r9 r r r Ú test_minmax_period_empty_nató s z0TestIndexReductions.test_minmax_period_empty_natc C s t jddd}t |¡tdddks(tt |¡tdddksBtd}tjt |d tj|dd W 5 Q R X tjt |d tj|dd W 5 Q R X t
|
|¡dks¨tt |¡d
|
ksºtd}tjt |d tj
|
|dd W 5 Q R X tjt |d tj |dd W 5 Q R X d S ) Nr¾ r¿ rÀ r r r rW r r r] ) rF Z period_ranger$ r- r r7 r, rY rZ r[ ra rb )r8 Úprr r r r Útest_numpy_minmax_periodû s z,TestIndexReductions.test_numpy_minmax_periodc C s t jtdtddd}d}tjt|d | ¡ W 5 Q R X d}tjt|d | ¡ W 5 Q R X t jtdtddd}| ¡ d kst| ¡ d
|
kstd S ) NZaabbcaZcabF©Ú
|
categoriesÚorderedzwCategorical is not ordered for operation min\nyou can use .as_ordered\(\) to change the Categorical to an ordered one\nrW zwCategorical is not ordered for operation max\nyou can use .as_ordered\(\) to change the Categorical to an ordered one\nTÚcre ) rF ZCategoricalIndexÚlistrY rZ r´ r- r, r7 )r8 Úcirµ r r r Útest_min_max_categorical s ÿÿz,TestIndexReductions.test_min_max_categoricalN)r} r~ r rY r r r r r r r® r¶ rº r» r¼ r½ r rà r rÆ rÈ rÏ r r r r r Ö s: ûþ
|
|
|
5
|
|
r c @ sn e Zd Zdd Zej ddddddg¡ej d d
|
d g¡ej d d dg¡dd Zej dddg¡ej ddddg¡dd Zej dddddg¡dd Z dd Z
|
ej d d
|
d g¡ej dd d g¡d d Z ej d d
|
d g¡ej dd!dg¡d"d# Z d$d% Z ej d&d'd(g¡d)d* Zej dd+¡ej d,d
|
d g¡d-d. Zd/d0 Zd1d2 Zd3d4 Zd5d6 Zd7d8 Zd9d: Zd;d< Zd=d> Zej d?d@dAg¡ej d,d
|
d g¡dBdC Zej d?d@dAg¡ej dDd dEgdEd gd ejgejd gg¡dFdG Zej ddddHdg¡ej d?d@dAg¡ej d,d
|
d g¡ej dIdJdJdJgd d gd d ggfdKdKdKgd
|
d
|
gd
|
d
|
ggfe j e j e j ge j e j gd d
|
ggfdJe j dJge j d gd d ggfdKe j dKgd
|
e j gd
|
d
|
ggfdKe j dJgd
|
d gd
|
d ggfg¡dLdM Z dNdO Z dPdQ Z dRdS Z!ej dTe"g ddUe#fe"dVdWdXge$fe"dYdZge$fe"dVdVdWdWdEejdXge$fg¡d[d\ Z%d]d^ Z&dES )_ÚTestSeriesReductionsc C s¼ t tj d¡}| ¡ }tj|dd
< tj|dd
< t | ¡ ¡sFt tj dd¡
|
d¡}tj|d d
df< t dd¡ t | ¡ | ¡ ¡ W 5 Q R X tj|d d
|
}t |¡ ¡ s¸t d S ) Nr r] é éd Zf4rD úmode.use_inf_as_naTrB rk )r r$ r% r& ÚcopyÚinfrc Úisinfr{ r7 r5 rF Úoption_contextr" Úassert_almost_equalr ÚnansumÚall)r8 r Ús2r rH r r r Ú test_sum_inf) s z!TestSeriesReductions.test_sum_infrJ r> ZFloat32rC rE r< Úuse_bottleneckTFz method, unit)r{ ç )r¯ ç ð?c C sF t d|¡. tg |d}t|| }||ks4tt||dd}||ksPtt||dd}t|sltt||dd}||k t||ddd}||ks¢tt||ddd}t|sÀtt||d dd}||ksÞtt||d dd}t|sütttjg|d}t|| }||ks&tt||dd}||ksDtt||dd}t|sbtt||dd}||k t||ddd}||kstt||ddd}t|sºtttjdg|d}t|| }|d
|
ksætt||dd}|d
|
kstt||dd}|d
|
ks"tt||dd}|d
|
ks@tt||ddd}|d
|
ks`ttt d ¡|d}t||d|k
|
¡ sttdg|d}t||d d}t|sºtt||d d d}t|sÚtttjg|d}t||d d}t|stttjdg|d}t||d d}t|s8tW 5 Q R X d S ) NrÝ rI r )Ú min_countrB TrL )rM rà Frß )r r rD ) rF r× r r2 r7 r r$ rc r ÚemptyrÚ )r8 ÚmethodÚunitrÝ rJ r r9 rq r r r Ú
|
test_empty; sn z TestSeriesReductions.test_emptyrâ r¨ r° ZFloat64c C sT t g |d}t|| }|tjks&tt tjg|d}t|| }|tjksPtd S ©NrI )r r2 rF rG r7 r$ rc )r8 râ rJ Zeserr9 Znserr r r Ú&test_ops_consistency_on_empty_nullable s z;TestSeriesReductions.test_ops_consistency_on_empty_nullablerª r« c C s t ttd| }t|s ttg dd}|dkrjd dddg¡}tjt|d t || W 5 Q R X nt || }|t kstd S ) NrI úm8[ns]r° r± zoperation 'var' not allowedz.cannot perform var with type timedelta64\[ns\]z does not support reduction 'var'rW )
|
r2 r Úfloatr r7 r³ rY rZ r´ r )r8 râ r9 Ztdserrµ r r r Ú test_ops_consistency_on_empty° s ýÿ z2TestSeriesReductions.test_ops_consistency_on_emptyc C s0 t dtjgddgd}t |¡}t |d¡ d S )Nrß r rB ©r )r r$ rc rÙ r" rØ )r8 Úserr9 r r r Útest_nansum_bugletÉ s
|
z'TestSeriesReductions.test_nansum_bugletÚint32r1 c C s t d|¡| tjd|d}t|}|jdd}t||jddksHt|jdd}t|dksdt|j dd}t||d kstW 5 Q R X d S ) NrÝ é@KL rI FrL r1 r rS )
|
rF r× r$ r` r r{ Úintr7 r- r, ©r8 rÝ rJ Úvr r9 r r r Útest_sum_overflow_intÎ s z*TestSeriesReductions.test_sum_overflow_intÚfloat32c C s t d|¡ tjd|d}t|}|jdd}||j|dksDt|jdd}t t |d¡sdt|j
|
dd}t t ||d ¡stW 5 Q R X d S )NrÝ rî rI FrL rÞ rS ) rF r× r$ r` r r{ r7 r- Zallcloserè r, rð r r r Útest_sum_overflow_floatÞ s z,TestSeriesReductions.test_sum_overflow_floatc C s\ d}d}t |g| ¡}t|g| dd}t |¡}| ¡ }|| dksLt|dksXtd S )Nl ;á rÒ rC rI r g Ø
W4vC)r$ Úarrayr r¨ r7 )r8 rN Z
|
n_elementsÚnarë Z result_numpyZ result_maskedr r r Útest_mean_masked_overflowì s
|
z.TestSeriesReductions.test_mean_masked_overflowz ddof, exp)rB g @)r r= c C s` t dddddgdd}t dddddgdd}|j|d }|j|d }||ksPt||ks\td S )
|
NrB rD r_ r^ r] rC rI r1 )Úddof)r r° r7 )r8 rø Úexprë Zser_numpy_dtyper9 Zresult_numpy_dtyper r r Útest_var_masked_arrayø s z*TestSeriesReductions.test_var_masked_array)rç rç r0 z M8[ns, UTC]rM c C s< t g |dj|dtks tt g |dj|dtks8td S )NrI rL )r r- r r7 r, )r8 rJ rM r r r Ú+test_empty_timeseries_reductions_return_nat s z@TestSeriesReductions.test_empty_timeseries_reductions_return_natc C s| t dd¡}t||d}t |¡}t |¡}||ks8t| ¡ }||ksLtd}tjt|d t j||d W 5 Q R X d S ©NrB rA rê r rW r )r$ r` r ra r7 rY rZ r[ ©r8 rÄ r r9 r: rµ r r r Útest_numpy_argmin s
|
|
z&TestSeriesReductions.test_numpy_argminc C s| t dd¡}t||d}t |¡}t |¡}||ks8t| ¡ }||ksLtd}tjt|d t j||d W 5 Q R X d S rü )r$ r` r rb r7 rY rZ r[ rý r r r Útest_numpy_argmax s
|
|
z&TestSeriesReductions.test_numpy_argmaxc C sì t ¡ d¡}tj|dd
< || ¡ | ¡ ks4tt|jddsHt| ¡ }|| ¡ | ¡ ksht|j
|
j ¡
|
| ¡ ¡|j ¡ kst|tj }t| ¡ s¦tttddd}| ¡ }|d ksÊttj|d < | ¡ }|d
|
ksètd S ) Nr' r] é FrL Ú20130102r¥ ©rg r rB )r" ÚmakeStringSeriesrn r$ ÚNaNÚidxminr- r7 r r¬ r r4 Útolistra rc r r ©r8 Z string_seriesZnonaZallnar r9 r r r Ú test_idxmin, s $
|
|
z TestSeriesReductions.test_idxminc C sh t ¡ d¡}tj|dd
< || ¡ | ¡ ks4tt|jddsHt| ¡ }|| ¡ | ¡ ksht|j
|
j ¡
|
| ¡ ¡|j ¡ kst|tj }t| ¡ s¦tttddd}| ¡ }|dksÊttj|d< | ¡ }|d ksèttd
|
d d gd ddg}| ¡ }|dkst| ¡ }|d ks*tt|j
|
|j
|
}| ¡ }|dksNt| ¡ }|d ksdtd S )Nr' r] r FrL r r¥ r r^ rB rD r_ gñ?gÍÌÌÌÌÌ @gÍÌÌÌÌÌ@)r" r rn r$ r Úidxmaxr, r7 r r¬ r r4 r rb rc r r r r r r r Ú test_idxmaxJ s2 $
|
|
z TestSeriesReductions.test_idxmaxc C sD t ¡ }|dk}| ¡ r t| ¡ s(ttddg}| ¡ s@td S )Nr ÚabcT)r" ZmakeTimeSeriesrÚ r7 Úanyr )r8 ÚtsZ bool_seriesr r r r Ú test_all_anyu s z!TestSeriesReductions.test_all_anyc C sJ |dddg}t |¡r tt |¡s*ttdddg}t |¡sFtd S )Nr rB rD r_ )r$ rÚ r7 r r )r8 rO r r r r Útest_numpy_all_any s
|
z'TestSeriesReductions.test_numpy_all_anyc C s` t tjdg}t tjdg}|jdds,t|jdds<t|jddsLt|jddr\td S )NTFrL )r r$ rc rÚ r7 r )r8 Ús1rÛ r r r Útest_all_any_skipna s z(TestSeriesReductions.test_all_any_skipnac C sH t ddddddgddddddgd}|jdds4t|jddrDtd S )NFTr rB rD rê )Ú bool_only)r r r7 rÚ ©r8 r r r r Útest_all_any_bool_only s $z+TestSeriesReductions.test_all_any_bool_onlyÚ bool_agg_funcr rÚ c C s: t dddddgtd}t|||d}d}||ks6td S ) Nr re rÌ ÚdÚerI rL T)r r< r2 r7 )r8 r rM rë r9 r: r r r Útest_any_all_object_dtype s z.TestSeriesReductions.test_any_all_object_dtyperÄ Nc C s8 t |}t||dd}|dko&d |k}||ks4td S )NFrL r )r r2 r7 )r8 rÄ r rë r9 r: r r r Ú!test_any_all_object_dtype_missing¡ s z6TestSeriesReductions.test_any_all_object_dtype_missingZUInt64zdata,expected_datar rB c C sP t ||d}|| |dk }t|||d}|tjkr@|tjksL||ksLtd S )NrI rÚ rL )r r2 rF rG r7 ) r8 r rM rÄ rJ Z expected_datarë r: r9 r r r Ú"test_any_all_nullable_kleene_logic® s z7TestSeriesReductions.test_any_all_nullable_kleene_logicc C s@ t ddgddgd}|jddd}tddg}t ||¡ d S )NTFrB rD rx )rl r )r r r r" ro )r8 rq r9 r: r r r Útest_any_axis1_bool_onlyÈ s z-TestSeriesReductions.test_any_axis1_bool_onlyc C s t dddj}t|}t|}d}tjt|dV | ¡ s>t| ¡ sJt| ¡ sVt| ¡ sbt| ¡ ¡ srt| ¡ ¡ stW 5 Q R X |
|
d¡}t|}t|}tjt|dX | ¡ sÂt| ¡ sÎt| ¡ sÚt| ¡ sæt| ¡ ¡ söt| ¡ ¡ stW 5 Q R X ||d }t|}t|}| ¡ s<t| ¡ rJt| ¡ sXt| ¡ rft| ¡ ¡ sxt| ¡ ¡ rtd S )Nz
|
1995-01-02r_ r z0'(any|all)' with datetime64 dtypes is deprecatedrW rf r ) r Ú_datar r r" Úassert_produces_warningÚ FutureWarningrÚ r7 r r# )r8 Zdtarë rq rµ Ztdar r r Útest_any_all_datetimelikeÏ s<
|
z.TestSeriesReductions.test_any_all_datetimelikec C s t dddd}t|td }| ¡ }|dks2t| ¡ }|dksFttj|d< | ¡ }|dksdt| ¡ }|dksxttt ddd }tt d
|
dd }t|| }t || ¡}t
|
||¡ || ¡ }t
|
||¡ | ¡ }t d }||ksòt| ¡ }t d }||kstd S ) Nz2012-1-1r_ r ©rg r/ Z20120101r rD rB r Z20120102r r )r r r r r7 r r$ rc Úabsr" ro r, r r- )r8 Zdtir r9 r rÛ r: r r r Útest_timedelta64_analyticsþ s0
|
z/TestSeriesReductions.test_timedelta64_analyticsztest_input,error_typerI ÚfooÚbarZbaz)rB )rD c C s d}t j||d | ¡ W 5 Q R X t j||d |jdd W 5 Q R X d}t j||d | ¡ W 5 Q R X t j||d |jdd W 5 Q R X dS )zU
|
Cases where ``Series.argmax`` and related should raise an exception
|
zbreduction operation 'argmin' not allowed for this dtype|attempt to get argmin of an empty sequencerW FrL zbreduction operation 'argmax' not allowed for this dtype|attempt to get argmax of an empty sequenceN)rY rZ r r )r8 Z
|
test_inputZ
|
error_typerµ r r r Ú test_assert_idxminmax_raises' s ÿÿz1TestSeriesReductions.test_assert_idxminmax_raisesc C sÈ t dtj tjtjg}| ¡ dks(tt |jdd¡s>t| ¡ dksNtt |jdd¡sdtt dd¡N | ¡ dkstt |jdd¡st| ¡ dks¨tt |jdd¡ W 5 Q R X d S )Nr rB FrL rD rÓ T)
|
r r$ rÕ rc r r7 rT r rF r× r r r r Útest_idxminmax_with_infG s z,TestSeriesReductions.test_idxminmax_with_inf)'r} r~ r rÜ rY r r rä ræ ré rì rò rô r÷ rú rû rþ rÿ r r
|
r r r r r r$ rc r rF rG r r r r" r r[ r´ r% r& r r r r rÐ $ s ÿ`
|
|
+
|
ÿ " úü
|
/)úþ
|
rÐ c @ s e Zd Zej deeegeeedgeededgg¡dd Z ej de
|
eege
|
eedge
|
ededgg¡dd Z dd Z d
|
d Z d S ) Ú TestDatetime64SeriesReductionsÚnat_serZnatc C sL | ¡ tkst| ¡ tks t|j ddtks4t|jddtksHtd S r ©r- r r7 r, )r8 r( r r r Útest_minmax_nat_series_ s
|
z5TestDatetime64SeriesReductions.test_minmax_nat_seriesÚnat_dfc C s\ | ¡ d tkst| ¡ d tks(t|j ddd tks@t|jddd tksXtd S )Nr FrL r) )r8 r+ r r r Útest_minmax_nat_dataframen s
|
z8TestDatetime64SeriesReductions.test_minmax_nat_dataframec C s t dd}| tj t|¡¡}| ¡ }| ¡ }t|t s>t
|
t|t sLt
|
||d ks\t
|
||d kslt
|
| ¡ |d kst
|
| ¡ |d kst
|
d S )Nú1/1/2000z
|
12/31/2000r rS ) r Ztaker$ r% Z permutationÚlenr- r, r3 r r7 )r8 ÚrngZrng2Zthe_minZthe_maxr r r Ú test_min_max} s
|
z+TestDatetime64SeriesReductions.test_min_maxc
|
C s® t dddd}ddddddddddg
|
}t|tj t|¡|d}|j ¡ }t|jj d }t
|
|tsjt ||ksvt |j ¡ }t|jj d
|
}t
|
|tst ||ksªt d S ) Nr- r Z4hr ry rz ÚC)ÚTSÚVÚLrS r ) r r r$ r% r& r. r2 r, r Ziatr3 r7 r- )r8 r/ Zlvlsrq r9 rù r r r Útest_min_max_series s
|
|
z2TestDatetime64SeriesReductions.test_min_max_seriesN)r} r~ r rY r r r r r r* r r, r0 r5 r r r r r' Z s$
|
ýþ
|
|
ýþ
|
r' c @ s e Zd Zej dddg¡dd Zej dededfededfedej g edfd d
|
d gd d
|
d gfd d
|
d ej gd d
|
d gfg¡ej dddg¡d d Z
|
ej dddg¡ej dddg¡dd Z ej dddg¡ej dddg¡dd Z dS )Ú TestCategoricalSeriesReductionsÚfunctionr- r, c C sL t tddddgdd}d| }tjt|d t|| W 5 Q R X d S ) Nr re rÌ r F)rË z)Categorical is not ordered for operation rW )r r rY rZ r´ r2 )r8 r7 Úcatrµ r r r Ú test_min_max_unordered_raises s
|
z=TestCategoricalSeriesReductions.test_min_max_unordered_raiseszvalues, categoriesr ZcbarB rD r_ c C sJ t t||dd}t||dd}|dkr2|d n|d }||ksFtd S )NTrÉ rL r- r rD )r r r2 r7 )r8 r4 rÊ r7 r8 r9 r: r r r Útest_min_max_ordered¨ s z4TestCategoricalSeriesReductions.test_min_max_orderedrM TFc C s< t ttjgddgdd}t|||d}|tjks8td S )NrB rD TrÉ rL ©r r r$ rc r2 r7 )r8 r7 rM r8 r9 r r r Ú"test_min_max_ordered_with_nan_onlyº s zBTestCategoricalSeriesReductions.test_min_max_ordered_with_nan_onlyc C sh t tddtjdgddgdd}t|||d}|dkrV|dkrDdnd}||ksdtn|tjksdtd S )Nr re TrÉ rL r- r; )r8 r7 rM r8 r9 r: r r r Útest_min_max_skipna s ÿz3TestCategoricalSeriesReductions.test_min_max_skipnaN) r} r~ r rY r r r9 rÍ r$ rc r: r< r= r r r r r6 s&
|
ûþ
|
r6 c @ s e Zd Zej ddeg ejdfdeg ejdfg¡dd Z ej dddddd gdgfddddd d
|
d
|
d
|
gdd
|
gfddddd gdgfddddd d
|
d
|
d
|
gdd
|
gfg¡ej d e
|
ej d ej d ¡dd Z ej dddgfddej gfg¡dd Zej dddgdgdgfddgej gdgfg¡dd Zej dddgdgfddgej gfg¡dd Zej ddd d d gd d gfdej gej d d gfg¡d d! Zej ddd"d#d$gd%d&gfdej gej d%d&gfg¡d'd( Zej ddedd gdd gd)ed*gdd*gd)ed
|
dgd
|
d dgdd+fdeej gdd gd)eej d*gdd*gd)eej d
|
dgd
|
d dgdd+fg¡d,d- Zej ddd.gdd.gfdd.gdd.gfg¡d/d0 Zd1d2 Zd3d4 Zej d5d6d7ddd8d9gedgejdejfd6d7ddd8d9gedgejdejfd8d:d8ged8gejdejfg¡d;d< Zej d5d6d7dd8d9ged=d7d>d8d9gejdejfd8d:d8d:d
|
ged:d8gejdejfg¡d?d@ ZdAS )BÚTestSeriesModezdropna, expectedTrI Fc C s( t g tjd}| |¡}t ||¡ d S rå )r r$ r> Úmoder" ro ©r8 r¬ r: r r9 r r r Útest_mode_emptyÖ s
|
z TestSeriesMode.test_mode_emptyzdropna, data, expectedrB rD r_ ÚdtZ
|
AllIntegerZFloatc C s2 t ||d}| |¡}t ||d}t ||¡ d S rå ©r r? r" ro )r8 r¬ rÄ r: rB r r9 r r r Útest_mode_numericalß s
|
z"TestSeriesMode.test_mode_numericalrß c C s8 t dddtjtjg}| |¡}t |}t ||¡ d S )NrB rD )r r$ rc r? r" ro r@ r r r Útest_mode_numerical_nanñ s
|
z&TestSeriesMode.test_mode_numerical_nanz'dropna, expected1, expected2, expected3re r$ rc c C sÔ dgd dgd }t |dd}| |¡}t |dd}t ||¡ dddtjtjtjg}t |td}| |¡}t |td}t ||¡ dddtjtjtjg}t |td t¡}| |¡}t |td}t ||¡ d S ) Nr rD re r_ rÌ rI r# r$ ) r r? r" ro r$ rc r< r5 Ústr)r8 r¬ Ú expected1Ú expected2Ú expected3rÄ r r9 r r r Útest_mode_str_objø s
|
|
|
z TestSeriesMode.test_mode_str_objz dropna, expected1, expected2r# c C sl t dddg}| |¡}t |}t ||¡ t dddtjtjtjg}| |¡}t |td}t ||¡ d S )NrB r# rI )r r? r" ro r$ rc r< )r8 r¬ rG rH r r9 r: r r r Útest_mode_mixeddtype s
|
|
z#TestSeriesMode.test_mode_mixeddtypeú
|
1900-05-03r¹ ú
|
2013-01-02c C sx t dddddgdd}| |¡}t |dd}t ||¡ t dddddddgdd}| |¡}t |dd}t ||¡ d S )Nr¹ rM rL rc r0 rI rC ©r8 r¬ rG rH r r9 r r r Útest_mode_datetime" s* ÿ
|
ù ö
|
z!TestSeriesMode.test_mode_datetimeú-1 daysú0 daysr ú2 minú1 dayc C sz t dddddgdd}| |¡}t |dd}t ||¡ t dddd d
|
d
|
ddgdd}| |¡}t |dd}t ||¡ d S ) Nr rP rQ rc rw rI rS z-1 dayz -1 day 2 minrR rC rN r r r Útest_mode_timedeltaE s,
|
ÿ
|
ø
|
õ
|
z"TestSeriesMode.test_mode_timedelta)rÊ r rÉ c C sÆ t tddtjtjg}| |¡}t |dd}t ||¡ t tdddtjtjg}| |¡}t |dd}t ||¡ t tdddddtjtjgdddgdd}| |¡}t |dd}t ||¡ d S ) NrB rD ÚcategoryrI r r_ TrÉ )r r r$ rc r? r" ro )r8 r¬ rG rH rI r r9 r r r Útest_mode_categoryg s$
|
|
ÿÿ
|
z!TestSeriesMode.test_mode_categoryì c C sr t dddgtjd}| |¡}t |tjd}t ||¡ t ddgtjd}| |¡}t |tjd}t ||¡ d S )NrB rW rI )r r$ Zuint64r? r" ro rN r r r Útest_mode_intoverflow s
|
|
z$TestSeriesMode.test_mode_intoverflowc C sf t dtjg}t dddtjtjg}t t¡" |jdd}| ¡ jdd}W 5 Q R X t ||¡ d S )Nr# rB F)r¬ T)Zdrop)
|
r r$ rc r" r Ú UserWarningr? Z sort_valuesZ reset_indexro )r8 r: r r9 r r r Útest_mode_sortwarning s z$TestSeriesMode.test_mode_sortwarningc C s> t dddtjgdd}| ¡ }t ddidd}t ||¡ d S )NTFrE rI r )r rF rG r? r" ro )r8 rë r9 r: r r r Útest_mode_boolean_with_na© s z(TestSeriesMode.test_mode_boolean_with_nazarray,expected,dtyper y ð?y ð? ð?y ð? @y @c C s t ||d ¡ }t ||¡ d S rå rC ©r8 rõ r: rJ r9 r r r Ú test_single_mode_value_complex° s z-TestSeriesMode.test_single_mode_value_complexy y ð? c C s t ||d ¡ }t ||¡ d S rå rC r\ r r r Útest_multimode_complexÈ s z%TestSeriesMode.test_multimode_complexN) r} r~ r rY r r r r$ r> rA rÍ Z typecodesrD rc rE rJ rK rO rT r rV rX rZ r[ Z
|
complex128Z complex64r] r^ r r r r r> Ñ sÆ "þ
|
üþ ÿ
|
$þ
|
þ
|
ýúþ
|
þþ
|
üüùþ
|
þ
|
ýýýõþ
|
ü ýùþr> )$r r Únumpyr$ rY ZpandasrF r r r r r r r
|
r r r r r r r r r Zpandas._testingZ_testingr" Z pandas.corer r) r( r* r rÐ r' r6 r> r r r r Ú<module> s* H " P :A6
|
