1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
U
¬ý°dAã@sZddlZddlZddlmZddlZddlmZddlm    Z
ddl m Z Gdd„dƒZ dS)éN)ÚDatetimeTZDtype)ÚNaT)Ú DatetimeArrayc@säeZdZejddddgddd„ƒZejdd    „ƒZd
d „Zej     d d dg¡ej     dddg¡dd„ƒƒZ
ej     d d dg¡ej     dddg¡dd„ƒƒZ dd„Z dd„Z ej d¡dd„ƒZdd„Zdd „Zej     dddg¡d!d"„ƒZd S)#ÚTestReductionsÚsÚmsÚusÚns)ÚparamscCs|jS)N)Úparam)ÚselfÚrequest©rúdd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/datetimes/test_reductions.pyÚunit szTestReductions.unitcCs>|}|dk    rt|dnt d¡}tjddddddg|d    }|S)
z;Fixture returning DatetimeArray with parametrized timezonesN©ÚtzúM8[ns]z
2000-01-03rú
2000-01-02ú
2000-01-05z
2000-01-04©Údtype)rÚnprrÚ_from_sequence)r Ztz_naive_fixturerrÚarrrrrÚarr1dsú÷ zTestReductions.arr1dcCs¸|}| |¡}|j}| ¡}tjd|d |¡}||ks<t‚|j|jksLt‚| ¡}tjd|d |¡}||kstt‚|j|jks„t‚|jdd}|tksœt‚|jdd}|tks´t‚dS)NrrrF©Úskipna)    Zas_unitrÚminÚpdÚ    TimestampÚAssertionErrorrÚmaxr)r rrrrÚresultÚexpectedrrrÚ test_min_max#s
     zTestReductions.test_min_maxrNz
US/CentralrTFcCs^|dk    rt|dnt d¡}tjg|d}|j|d}|tksBt‚|j|d}|tksZt‚dS)Nrrrr)    rrrrrrrr!r")r rrrrr#rrrÚtest_min_max_empty8s    z!TestReductions.test_min_max_emptycCs´|dk    rt|dnt d¡}tjg|d}|j|d}|tksBt‚| dd¡}|jd|d}t    |ƒjtttg|jd}t
  ||¡|jd|d}t    |ƒjg|jd}t
  ||¡dS)    Nrrrrré©Úaxisré) rrrrrÚmedianrr!ÚreshapeÚtypeÚtmÚ assert_equal)r rrrrr#r$rrrÚtest_median_emptyCs    z TestReductions.test_median_emptycCst|}| ¡}||dkst‚|jdd}|tks4t‚| ¡jdd}||dksTt‚|jdd}||dkspt‚dS)NrFr©r))r+r!rÚdropna)r rrr#rrrÚ test_medianTs   zTestReductions.test_medianc    Cs`|}|jdd| ¡kst‚|jdddtks2t‚d}tjt|d|jddW5QRXdS)Nrr1Fr(z"abs\(axis\) must be less than ndim)Úmatchr*)r+r!rÚpytestZraisesÚ
ValueError)r rrÚmsgrrrÚtest_median_axisbs zTestReductions.test_median_axisz/ignore:All-NaN slice encountered:RuntimeWarningcCsÎ| dd¡}| ¡| ¡ks t‚|jddtks4t‚|jdd}|}t ||¡|jddd}|}t ||¡|jdd}t|ƒ | ¡g¡}t ||¡|jddd}t|ƒjtg|jd}t ||¡dS)    Nr*éÿÿÿÿFrrr1r(r)    r,r+r!rr.r/r-rr)r rrr#r$rrrÚtest_median_2dks      zTestReductions.test_median_2dcCs€|}|ddtjdd}| ¡}||ks0t‚|jdd}|tksHt‚| ¡jdd}||ksdt‚|jdd}||ks|t‚dS)Nrgš™™™™™Ù?r*)ÚdaysFrr1)rÚ    TimedeltaÚmeanr!rr2)r rrr$r#rrrÚ    test_mean†s     zTestReductions.test_meancCs–tjdddd}|j dd¡}|jdd}|d    }t ||¡|jd    d}|dd…dftjd
d }t ||¡|jdd}| ¡}||ks’t‚dS) Nz
2016-01-01éz
US/Pacific)Zperiodsrr'érr1r*é )Úhours)    rZ
date_rangeÚ_datar,r=r.Úassert_datetime_array_equalr<r!)r ZdtiZdtar#r$rrrÚ test_mean_2d—s     zTestReductions.test_mean_2dcCs˜|dd…}|j|dtks t‚| dd¡}|jd|d}tjtttg|jd}t ||¡|jd|d}|}t ||¡|jd|d}|tks”t‚dS)Nrrr'r(rr*)    r=rr!r,rrrr.rD)r rrrZarr2dr#r$rrrÚtest_mean_empty§s    zTestReductions.test_mean_empty)Ú__name__Ú
__module__Ú __qualname__r5Zfixturerrr%ÚmarkZ parametrizer&r0r3r8Úfilterwarningsr:r>rErFrrrrr s&
 
          
 
r)Únumpyrr5Zpandas.core.dtypes.dtypesrZpandasrrZpandas._testingZ_testingr.Zpandas.core.arraysrrrrrrÚ<module>s