o[we@<ddlmZmZddlmZddlZddlZddlZddlm Z m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZmZmZddlmZddlmZdZGddZGd d Z Gd d Z!Gd dZ"GddZ#GddZ$dS))datetime timedelta)DecimalN) Categorical DataFrame DatetimeIndexIndexNaTPeriod PeriodIndex RangeIndexSeries TimedeltaTimedeltaIndex Timestamp date_rangeisnatimedelta_range to_timedelta)nanopsc tjddtjddtjddtjddtjdddtjddtjddg}tj d dfd|D}||z}|S)N a)namez US/Easterntzc4g|]}t|dS)r)indexr)r).0idxarrs Glib/python3.11/site-packages/pandas/tests/reductions/test_reductions.py zget_objs..-s( B B B3fS#... B B B) tm makeBoolIndex makeIntIndexmakeFloatIndex makeDateIndex tz_localizemakePeriodIndexmakeStringIndexnprandom default_rngstandard_normal)indexesseriesobjsr"s @r#get_objsr5!s #&&& %%% "3''' #&&& #&&&22l2CC 2C((( 2C(((G )   " " 2 22 6 6C B B B B' B B BF V D Kr%c (eZdZejdejdddgejdedZejdddgejddd d e d d d fddgdZ ejdddgdZ ejdddgejdddgdZ dZ ejdddgddggdZejdddgdZdZd S)!TestReductionsz8ignore:Period with BDay freq is deprecated:FutureWarningopnamemaxminobjct||}t|tst|j|}n1t t|j||j}t|dd7|dd}|j|ksJdS||ksJdS)N)ordinalfreqrM8[ns]int64) getattr isinstancer valuesr asi8r>astype_value)selfr8r;resultexpecteds r#test_opszTestReductions.test_ops4s &f%%''#{++ R2wsz62244HH&?gch&?&?&A&AQQQH 3d # # /x0077@@H=H,,,,,,X%%%%%%r%z dtype, val)object@)float64rLdatetime64[ns] )Int64r)booleanTc|}fd}|dg}|t||sJ|t||dsJ|g}|t||sJ|t||dsJdkrdS|d|g}t|||ksJ|t||dsJ|d|dg}t|||ksJ|t||dsJdS)Ncddkr |tuSdvr|tjuSt|S)NrN)rRrS)r pdNAr)resdtypes r# check_missingz4TestReductions.test_nanminmax..check_missingVs=(((cz!...be|#Cyy r%rYFskipnarK)rA)rGr8rYvalindex_or_seriesklassrZr;s ` r#test_nanminmaxzTestReductions.test_nanminmaxGs  ! ! ! ! !eTF%(((}1WS&113344444}1WS&11???@@@@@eBe$$$}1WS&113344444}1WS&11???@@@@@ H   FeT3Ku---#wsF##%%,,,,}1WS&11???@@@@@eT3%U333#wsF##%%,,,,}1WS&11???@@@@@@@r%c(|}|turd|znd|z}|ttdddg}t||dksJd}|turd}t jt|5t||d }dddn #1swxYwY|turtj |sJn|d ksJ|ttdddtg}t||dksJt jt|5t||d }dddn #1swxYwY|turtj |sJdS|d ksJdS) Nargr!rOrPrQzNThe behavior of (DatetimeIndex|Series).argmax/argmin with skipna=False and NAsz2The behavior of Series.(idxmax|idxmin) with all-NAmatchFr\) r r rrArr&assert_produces_warning FutureWarningr.isnan)rGr8r_r`arg_opr;msgrHs r#test_nanargminmaxz TestReductions.test_nanargminmaxrs7 #(E>>uv~eS(4Q//011#wsF##%%**** #  F??FC  ' S A A A 8 8)WS&))777F 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 F??8F## # # # #R<<<<eS(4Q//566#wsF##%%****  ' S A A A 8 8)WS&))777F 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 F??8F## # # # # #R<<<<<>uv~eBe$$$#wsF##%%,,,,#wsF##5111S8888 ]:-= > > > # # GC " " " # # # # # # # # # # # # # # # ]:-= > > > / / GC  . . . . / / / / / / / / / / / / / / / / / /s$B''B+.B+ C55C9<C9c  ttjdd}|dksJ|dksJttjdtjdg}|dksJ|dksJd }t jt| 5|d d ksJ dddn #1swxYwYt jt| 5|d d ksJ dddn #1swxYwYttjg}t jt| 5|d ksJ dddn #1swxYwYt jt| 5|d ksJ dddn #1swxYwYt jt| 5|d d ksJ dddn #1swxYwYt jt| 5|d d ksJ dddn #1swxYwYd}tttdddtdddtg}|dksJ|dksJt jt| 5|d d ksJ dddn #1swxYwYt jt| 5|d d ksJ dddn #1swxYwYttg}t jt| 5|d ksJ dddn #1swxYwYt jt| 5|d ksJ dddn #1swxYwYt jt| 5|d d ksJ dddn #1swxYwYt jt| 5|d d ksJ ddddS#1swxYwYdS)Nr@r[rrQrz=The behavior of Index.argmax/argmin with skipna=False and NAsrdFr\rfzEThe behavior of DatetimeIndex.argmax/argmin with skipna=False and NAsrOrP) r r.arangeargminargmaxnanr&rgrhr r)rGr;rks r#test_argminmaxzTestReductions.test_argminmaxsBIaw///00zz||q    zz||q    RVQ*++zz||q    zz||q    M  ' S A A A 2 2::U:++r11111 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2  ' S A A A 2 2::U:++r11111 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2RVHoo  ' S A A A & &::<<2%%%%% & & & & & & & & & & & & & & &  ' S A A A & &::<<2%%%%% & & & & & & & & & & & & & & &  ' S A A A 2 2::U:++r11111 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2  ' S A A A 2 2::U:++r11111 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2VS(4Q//$A1F1FLMMzz||q    zz||q     ' S A A A 2 2::U:++r11111 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2  ' S A A A 2 2::U:++r11111 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2SEll  ' S A A A & &::<<2%%%%% & & & & & & & & & & & & & & &  ' S A A A & &::<<2%%%%% & & & & & & & & & & & & & & &  ' S A A A 2 2::U:++r11111 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2  ' S A A A 2 2::U:++r11111 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2sC::C>C> E  EE F22F69F6HHH&III6J  J$'J$7M!!M%(M%N11N58N5,PPP:Q""Q&)Q&R22R69R6TT Tzop, expected_colrbc4ttddddg}|jt d|d <t ||d }||d}tj||dS) Nz2016-01-01 00:00:00ruUTC)periodsrr)columnsisecondsr{rQaxis) rrrsubtractrrArenamer&assert_series_equal)rGop expected_coldfrHrIs r#test_same_tz_min_max_axis_1z*TestReductions.test_same_tz_min_max_axis_1s ,aE B B BSE   $-- $ 7 7 7883 Ra(((l#**400 vx00000r%funcmaximumminimumc|}tjdg|}t|}t t |||}t j||dS)N2019)rV to_datetimer+rrAr.r&r)rGtz_aware_fixturerrrcrIrHs r#(test_numpy_reduction_with_tz_aware_dtypez7TestReductions.test_numpy_reduction_with_tz_aware_dtypesgnfX&&22266#;;"T""8X66 vx00000r%c ttddtgdtddtjgdd}tt ddd}|}tj||dS)NrQrtimedelta64[ns]r[rRABru) rrr r.ryrsumr&r)rGrrIrHs r#test_nan_int_timedelta_sumz)TestReductions.test_nan_int_timedelta_sums Q3K/@AAAQ26N':::      ! 15566 vx00000r%N)__name__ __module__ __qualname__rnmarkfilterwarnings parametrizer5rJrrarlrqrzrrrr%r#r7r73s! [B [Xu~66 [UHHJJ// & &0/76 & [Xu~66 [   xxb!44 5      AA  76A@ [Xu~66  76 > [Xu~66 [Wx1F&GHH / /IH76 /)2)2)2V [/5#, 1MNN11ON1 [Vi%;<<11=<1 1 1 1 1 1r%r7c eZdZejdgddZdZejdddgdZd Z d Z ejd gd d Z dZ ejdddgdZ dZdZdZdZejdddgejdgegeeeggdZdZdZdS)TestIndexReductionszstart,stop,step))riru)iri)@Brt)rr)rrc$t|||}|j}|}||ksJ|d}||ksJ|j}|}||ksJ|d}||ksJt||| }t |sJt |sJdSNFr\)r _valuesr9r:r)rGstartstopstepr!rIrHresult2s r#test_max_min_rangez&TestIndexReductions.test_max_min_rangesd++;??$$!!!!'''''("""";??$$!!!!'''''(""""te,,CGGIICGGIIr%ctgd}|jsJtdtjddg}|jrJ||fD]}|t dksJ|t dksJ|dksJ|dksJdS)N)1 days2 days3 daysrrr rr) ris_monotonic_increasingr.ryr:rr9rwrxrGidx1idx2r!s r#test_minmax_timedelta64z+TestIndexReductions.test_minmax_timedelta64s<<<==++++x5ABB////$< % %C7799 ( 3 333337799 ( 3 33333::<<1$$$$::<<1$$$$$  % %r%rr:r9cVtg}t||tusJttg}t||tusJttttg}t||tusJdSN)rrAr rGrr;s r#!test_minmax_timedelta_empty_or_naz5TestIndexReductions.test_minmax_timedelta_empty_or_na$sR  wsB!!S((((cU##wsB!!S((((c3_--wsB!!S((((((r%ctddd}tj|tdksJtj|tdksJd}t jt|5tj|ddddn #1swxYwYt jt|5tj|ddddn #1swxYwYtj|dksJtj |d ksJd}t jt|5tj|ddddn #1swxYwYt jt|5tj |dddddS#1swxYwYdS) Nz 16815 daysz 16820 daysDr>$the 'out' parameter is not supportedrdroutrs) rr.r:rr9rnrorprwrx)rGtderrmsgs r#test_numpy_minmax_timedelta64z1TestIndexReductions.test_numpy_minmax_timedelta640s \B!!B%(B%C**C.1C.E))E-0E-F33F7:F7c(tdtdD}|}|}t t d}||ksJ|}|d|ksJ|d}ttj dd}||ksJ| }t d }||ksJ| }|d|ksJ| }t d }||ksJ| }|d|ksJ| }t t|j }||ksJ| }|d|ksJtt!d t!d g}| t d ksJtt!d t!d t!dg}| t dksJdS)NcTg|]%}tdt||zz&S20130101rrrr is r#r$z:TestIndexReductions.test_timedelta_ops..I4 Q Q Q!Yz " "Yq1u%=%=%= = Q Q Qr%r rrg?i( msz00:00:09z00:01:21z 2015-02-03z 2015-02-07rt)daysz 2015-02-15)rrangediffmeanrrto_framequantilerr. timedelta64medianrstddropnarCr)rGsrrHrIs r#test_timedelta_opsz&TestIndexReductions.test_timedelta_opsEs  Q QuRyy Q Q Q  VVXX ! 4 4 455!!!!##%%ayH$$$$S!!R^D$7788!!!! ++!!!!%%''ayH$$$$ ++!!!!""$$ayH$$$$ryy{{'9 : : > > @ @AA!!!!""$$ayH$$$$ Il++Y|-D-DE F Fvvxx  I1$5$5$55555  | $ $i &=&=y?V?V W  vvxx  I1$5$5$5555555r%r8)skewkurtsemprodvarctdtdD}|}dd|dd|dd|d g}t jt | 5t||dddn #1swxYwYt jt | 5t||d ddddS#1swxYwYdS) NcTg|]%}tdt||zz&Srrrs r#r$zDTestIndexReductions.test_invalid_td64_reductions..}rr%r|zreduction operation 'z' not allowed for this dtypezcannot perform z with type timedelta64\[ns\]zdoes not support reduction ''rdF) numeric_only) rrrjoinrnro TypeErrorrAr)rGr8rrrks r#test_invalid_td64_reductionsz0TestIndexReductions.test_invalid_td64_reductionszs  Q QuRyy Q Q Q  VVXXhhLLLLG6GGG8v888   ]9C 0 0 0 " " GB   ! ! ! " " " " " " " " " " " " " " "]9C 0 0 0 ? ? *GBKKMM6 * * > > > > ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?s$:BB#&B#-C??DDc|}tgd|}|jsJtdtddtg|}|jrJ||fD]}|t d|ksJ|t d|ksJ|dksJ|dksJdS)N) 2011-01-01 2011-01-02 2011-01-03rrrrrr)rrr r:rr9rwrx)rGtz_naive_fixturerrrr!s r#test_minmax_tzz"TestIndexReductions.test_minmax_tzs GGGBOOO++++ 3 lC @R   ////$< % %C7799 ,2 > > >>>>>7799 ,2 > > >>>>>::<<1$$$$::<<1$$$$$  % %r%cztg}tt||sJttg}tt||sJttttg}tt||sJdSr)rrrAr rs r#test_minmax_nat_datetime64z.TestIndexReductions.test_minmax_nat_datetime64sB$GC$$&&'''''SE""$GC$$&&'''''S#sO,,$GC$$&&'''''''r%c"tgd}|j}tj|}||ksJ|j}tj|}||ksJd}t jt|5tj|ddddn #1swxYwYt jt|5tj|ddddn #1swxYwY|j}tj|}||ksJ|j }tj |}||ksJd}t jt|5tj |ddddn #1swxYwYt jt|5tj|dddddS#1swxYwYdS)NrQrrurrdrr) r rCr9r.r:rnrorprxrw)rGr!rIrHrs r#test_numpy_minmax_integerz-TestIndexReductions.test_numpy_minmax_integersIII:>>##!!!!:>>##!!!!7 ]:V 4 4 4   F3A                    ]:V 4 4 4   F3A                   :$$&&3!!!!:$$&&3!!!!7 ]:V 4 4 4 " " Icq ! ! ! ! " " " " " " " " " " " " " " " ]:V 4 4 4 " " Icq ! ! ! ! " " " " " " " " " " " " " " " " " "sHB<<CC"DD  D F::F>F> HH Hctddd}tj|}|dksJtj|}|dksJd}t jt |5tj|ddddn #1swxYwYt jt |5tj|dddddS#1swxYwYdS)Nrrrurrrdr)r r.r9r:rnrorp)rGr!rHrs r#test_numpy_minmax_rangez+TestIndexReductions.test_numpy_minmax_rangesYB""{{{{{{{{7 ]:V 4 4 4   F3A                    ]:V 4 4 4   F3A                      s$'B  BB0CCCc~tdd}tj|tdksJtj|tdksJd}t jt|5tj|d dddn #1swxYwYt jt|5tj|d dddn #1swxYwYtj|dksJtj |d ksJd}t jt|5tj|d dddn #1swxYwYt jt|5tj |d ddddS#1swxYwYdS) N 2016-01-15 2016-01-20rendz2016-01-15 00:00:00z2016-01-20 00:00:00rrdrrrs) rr.r:rr9rnrorprwrx)rGdrrs r#test_numpy_minmax_datetime64z0TestIndexReductions.test_numpy_minmax_datetime64s l = = =vbzzY'<======vbzzY'<======7 ]:V 4 4 4   F21                   ]:V 4 4 4   F21                   y}}!!!!y}}!!!!7 ]:V 4 4 4 ! ! Iba  ! ! ! ! ! ! ! ! ! ! ! ! ! ! !]:V 4 4 4 ! ! Iba  ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! !sH=B  B$'B$C))C-0C-E((E,/E,F22F69F6cBttdddgd}|jrJ|ddjsJtdtddtgd}|jrJ||fD]T}|t ddksJ|t ddksJU|dksJ|dksJ|dksJ|d ksJdS) NrrrrrrQrrur)r r rr:r r9rwrxrs r#test_minmax_periodz&TestIndexReductions.test_minmax_periodsGC|\JQTUUU////ABBx//// 3 lC @s   ////$< ? ?C7799|# > > >>>>>7799|# > > >>>>>>{{}}!!!!{{}}!!!!{{}}!!!!{{}}!!!!!!r%datacnt|d}t||}|tusJdS)NMr)r rAr )rGrrr;rHs r#test_minmax_period_empty_natz0TestIndexReductions.test_minmax_period_empty_nat s>$S)))!b!!##}}}}}}r%ctjdd}tj|t ddksJtj|t ddksJd}t jt|5tj|d dddn #1swxYwYt jt|5tj|d dddn #1swxYwYtj |dksJtj |d ksJd}t jt|5tj |d dddn #1swxYwYt jt|5tj |d ddddS#1swxYwYdS) Nrrrrrrrdrrrs) rV period_ranger.r:r r9rnrorprwrx)rGprrs r#test_numpy_minmax_periodz,TestIndexReductions.test_numpy_minmax_periods _<\ B B BvbzzVLs;;;;;;;vbzzVLs;;;;;;;7 ]:V 4 4 4   F21                    ]:V 4 4 4   F21                   y}}!!!!y}}!!!!7 ]:V 4 4 4 ! ! Iba  ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ]:V 4 4 4 ! ! Iba  ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! !sHB))B-0B-C22C69C6E11E58E5F;;F?F?cVtjtdtdd}d}tjt |5|dddn #1swxYwYd}tjt |5|dddn #1swxYwYtjtdtdd}|d ksJ|d ksJdS) NaabbcacabF categoriesorderedzwCategorical is not ordered for operation min\nyou can use .as_ordered\(\) to change the Categorical to an ordered one\nrdzwCategorical is not ordered for operation max\nyou can use .as_ordered\(\) to change the Categorical to an ordered one\nTcr{)rVCategoricalIndexlistrnrorr:r9)rGcirks r#test_min_max_categoricalz,TestIndexReductions.test_min_max_categorical(s  hDKKQV W W W Y ]9C 0 0 0   FFHHH                Y ]9C 0 0 0   FFHHH                hDKKQU V V Vvvxx3vvxx3s$A00A47A4B99B=B=N)rrrrnrrrrrrrrrrrrrrr rrrrr%r#rrs  [       0 % % % [TE5>22 ) )32 )!!!*363636j [X'M'M'MNN??ON?(%%%$ [TE5>22 ( (32 ("""@&!!!."""( [TE5>22 [Vb3%#sC%ABBCB32 !!!*r%rcneZdZdZejdgdejdddgejddd gd Zejd d d gejdgddZejd gddZ dZ ejdddgejdddgdZ ejdddgejdddgdZ dZ ejdddgdZejddejdddgd Zd!Zd"Zd#Zd$Zd%Zd&Zd'Zd(Zd)Zejd*d+d,gejdddgd-Zejd*d+d,gejd.dd/gd/dgdejgejdggd0Zejdgd1ejd*d+d,gejdddgejd2gd3ddgddggfgd4ddgddggfejejejgejejgddggfd5ejd5gejdgddggfd6ejd6gdejgddggfd6ejd5gddgddggfgd7Zd8Z d9Z!d:Z"d;Z#ejd<e$gd=e%fe$gd>e&fe$d?d@ge&fe$dAdAdBdBd/ejdCge&fgdDZ'dEZ(dFZ)dGZ*dHZ+dIZ,dJZ-dKZ.d/S)LTestSeriesReductionscttjdd}|}tj|dd<tj|dd<tj| sJtjdd d}tj|dddf<d}tj t|5tjd d 5tj| | dddn #1swxYwYdddn #1swxYwYt#j|d }tj|sJdS) Nrrrs)drf4"use_inf_as_na option is deprecatedrdmode.use_inf_as_naTrQr)rr.r/r0r1copyinfryisinfrrEr&rgrhrVoption_contextassert_almost_equalrnansumall)rGrs2r"rkrXs r# test_sum_infz!TestSeriesReductions.test_sum_infAs 29((++;;B?? @ @ VVXX!A#&1Q3x     i##A&&66zBBII$OOFAAAqD 2  ' S A A A : :"#7>> : :&quuww999 : : : : : : : : : : : : : : : : : : : : : : : : : : : : : :mCa(((x}}  """""""s6F):E/# F/E3 3F6E3 7FF  F rY)rMFloat32rRrSrKuse_bottleneckTFz method, unit)r)r?ctjd|5tg|}t||}||ksJt||d}||ksJt||d}t |sJt||d}||kt||dd}||ksJt||dd}t |sJt||d d}||ksJt||d d}t |sJtt jg|}t||}||ksJt||d}||ksJt||d}t |sJt||d}||kt||dd}||ksJt||dd}t |sJtt jdg|}t||}|d ksJt||d}|d ksJt||d}|d ksJt||d}|d ksJt||dd}|d ksJtt jd |}t||d|k sJtdg|}t||d }t |sJt||d d }t |sJtt jg|}t||d }t |sJtt jdg|}t||d }t |sJ ddddS#1swxYwYdS) Nr$r[r) min_countrQTr\)r]r(Fr&)rrr) rVrrrArr.ryremptyr )rGmethodunitr$rYrrHrs r# test_emptyzTestSeriesReductions.test_emptyUsP  / @ @] ] r'''A'WQ''))FT>>>>(WQ''!444FT>>>>'WQ''!444F<<   (WQ''t444F dNN(WQ''tqAAAFT>>>>'WQ''tqAAAF<<   'WQ''uBBBFT>>>>'WQ''uBBBF<<   xu---A'WQ''))FT>>>>(WQ''!444FT>>>>'WQ''!444F<<   (WQ''t444F dNN(WQ''tqAAAFT>>>>'WQ''tqAAAF<<   {%000A'WQ''))FS====(WQ''!444FS===='WQ''!444FS====(WQ''t444FS===='WQ''tqAAAFS====28G,,E:::B'GB''**d27799 9 9 9s%(((A'WQ''!444F<<   'WQ''uBBBF<<   xu---A'WQ''!444F<<   {%000A'WQ''!444F<<    {] ] ] ] ] ] ] ] ] ] ] ] ] ] ] ] ] ] sQQ++Q/2Q/r*rr)Float64rRrSctg|}t||}|tjusJttjg|}t||}|tjusJdSNr[)rrArVrWr.ry)rGr*rYeserrHnsers r#&test_ops_consistency_on_empty_nullablez;TestSeriesReductions.test_ops_consistency_on_empty_nullables~b&&&&v&&((rvhe,,,&v&&((r%)rrrrcttt|}t|sJtgd}|dkredgd}t jt|5t||ddddS#1swxYwYdSt||}|tusJdS)Nr[m8[ns]rr)zoperation 'var' not allowedz.cannot perform var with type timedelta64\[ns\]z does not support reduction 'var'rd) rArfloatrrrnrorr )rGr*rHtdserrks r#test_ops_consistency_on_emptyz2TestSeriesReductions.test_ops_consistency_on_emptys4 6e,,,f5577F||r*** U??((Cy444 ) )&v&&((( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ),WUF++--FS======sB,,B03B0ctdtjgddg}tj|}t j|ddS)Nr&rrQr)rr.ryrr&r)rGserrHs r#test_nansum_bugletz'TestSeriesReductions.test_nansum_bugletsDc26]1a&1113 vq)))))r%int32r@ctjd|5tjd|}t |}|d}t ||dksJ|d}t |dksJ|d}t ||dksJ ddddS#1swxYwYdS) Nr$@KLr[Fr\r@rrf) rVrr.rvrrintr:r9rGr$rYvrrHs r#test_sum_overflow_intz*TestSeriesReductions.test_sum_overflow_ints0 / @ @ ( ( '///Aq AUU%U((Fv;;!%%g%"6"66666UU%U((Fv;;!####UU%U((Fv;;!B%''''' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (sCC%%C),C)float32rMctjd|5tjd|}t |}|d}|||ksJ|d}tjt|dsJ| d}tjt||dsJ ddddS#1swxYwYdS)Nr$r>r[Fr\r%rf) rVrr.rvrrr:allcloser5r9r@s r#test_sum_overflow_floatz,TestSeriesReductions.test_sum_overflow_floats6 / @ @ 5 5 '///Aq AUU%U((FQUUU//////UU%U((F;uV}}c22 2 2 2UU%U((F;uV}}ae44 4 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5sCC66C:=C:cd}d}tj|g|z}t|g|zd}tj|}|}||z dksJ|dksJdS)Nl; rrRr[rg؅W4vC)r.arrayrr)rGr^ n_elementsnar: result_numpy result_maskeds r#test_mean_masked_overflowz.TestSeriesReductions.test_mean_masked_overflows% Xsej( ) )cUZ'w777wr{{  |+q0000$$$$$$r%z ddof, exp)rQ@)rrLctgdd}tgdd}||}||}||ksJ||ksJdS)N)rQrrurtrsrRr[r@)ddof)rr)rGrPexpr:ser_numpy_dtyperHresult_numpy_dtypes r#test_var_masked_arrayz*TestSeriesReductions.test_var_masked_arrays}___G444 @@@d##,00d0;;+++++}}}}}}r%)r4r4r?z M8[ns, UTC]r]ctg||tusJtg||tusJdS)Nr[r\)rr:r r9)rGrYr]s r#+test_empty_timeseries_reductions_return_natz@TestSeriesReductions.test_empty_timeseries_reductions_return_natshb&&&**&*99S@@@@b&&&**&*99S@@@@@@r%ctjdd}t||}tj|}tj|}||ksJ|}||ksJd}t jt |5tj||ddddS#1swxYwYdSNrQrPr9rrdr)r.rvrrwrnrorprGrrrHrIrks r#test_numpy_argminz&TestSeriesReductions.test_numpy_argmin#syB 4t $ $ $19T??!!!!!!!!4 ]:S 1 1 1 # # IaT " " " " # # # # # # # # # # # # # # # # # #B44B8;B8ctjdd}t||}tj|}tj|}||ksJ|}||ksJd}t jt |5tj||ddddS#1swxYwYdSrX)r.rvrrxrnrorprYs r#test_numpy_argmaxz&TestSeriesReductions.test_numpy_argmax5syB 4t $ $ $19T??!!!!!!!!4 ]:S 1 1 1 # # IaT " " " " # # # # # # # # # # # # # # # # # #r[ctddtjgtgd}d}t jt |5|d}dddn #1swxYwY|tusJd }t jt |5| d}dddn #1swxYwY|tusJ| }d }t jt |5|d}dddn #1swxYwY|j d ksJ| sJd }t jt |5| d}dddn #1swxYwY|j d ksJ| sJdS) Nr&rL)r z 2015-02-08r r90The behavior of Series.idxmin with all-NA valuesrdFr\0The behavior of Series.idxmax with all-NA valuesz3The behavior of DataFrame.idxmin with all-NA valuesr?z3The behavior of DataFrame.idxmax with all-NA values)rr.ryrr&rgrhidxminr idxmaxrrYrr )rGr:rkrXrs r#test_idxmin_dt64indexz*TestSeriesReductions.test_idxmin_dt64indexFs #rv m4P4P4P&Q&Q   A  ' S A A A + +**E***C + + + + + + + + + + + + + + +czzzz@  ' S A A A + +**E***C + + + + + + + + + + + + + + +czzzz \\^^C  ' S A A A * *))5)))C * * * * * * * * * * * * * * *yH$$$$xxzz~~C  ' S A A A * *))5)))C * * * * * * * * * * * * * * *yH$$$$xxzz~~sH A..A25A2!CC C D..D25D2 F..F25F2cptjd}tj|dd<|||ksJd}tjt|5t|dsJ dddn #1swxYwY| }|||ksJ|j j ||j ksJ|tjz}tjt|5t|sJ dddn #1swxYwYtt!dd }|}|d ksJtj|d <|}|d ksJdS) Nr3rszThe behavior of Series.idxminrdFr\20130102rr~rrQ)r&makeStringSeriesrr.ryrar:rgrhrrrrCtolistrwrrrG string_seriesrknonaallnarrHs r# test_idxminz TestSeriesReductions.test_idxmin`sp+--44X>> !f ad]11334 8I8I8K8KKKKK-  ' S A A A < < ,,E,::;; ; ; ; ; < < < < < < < < < < < < < < <##%%DKKMM"dhhjj0000z ''))// >>$+BTBTBVBVVVVV&  ' S A A A ( ( '' ' ' ' ' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( :j!444 5 5{{{{v!{{{{{{s$&B::B>B>$GG  G ctjd}tj|dd<|||ksJd}tjt|5t|dsJ dddn #1swxYwY| }|||ksJ|j j ||j ksJ|tjz}d}tjt|5t|sJ dddn #1swxYwYtt!dd }|}|dksJtj|d<|}|d ksJtgd gd }|}|dksJ|}|dksJt|j |j }|}|dksJ|}|dksJdS)Nr3rsrer`rdFr\rfrrgrtr)皙?g@@rqrp)r&rhrr.ryrbr9rgrhrrrrCrirxrrrarjs r# test_idxmaxz TestSeriesReductions.test_idxmaxs +--44X>> !f ad]11334 8I8I8K8KKKKK@  ' S A A A < < ,,E,::;; ; ; ; ; < < < < < < < < < < < < < < <##%%DKKMM"dhhjj0000z ''))// >>$+BTBTBVBVVVVV&@  ' S A A A ( ( '' ' ' ' ' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( :j!444 5 5{{{{v!{{{{ 999ooo . .}}}}}}}} 17AG $ $}}}}}}}}}}s$&B::B>B>$GG G ctj}|dk}|rJ|sJt ddg}|sJdS)NrabcT)r&makeTimeSeriesr anyr)rGts bool_seriesrs r# test_all_anyz!TestSeriesReductions.test_all_anysm   1f ??$$$$$      E4= ! !uuwwr%c|gd}tj|rJtj|sJtgd}tj|sJdS)N)rrQrr)r.r rvr )rGr_r!s r#test_numpy_all_anyz'TestSeriesReductions.test_numpy_all_anyshoiii((6#;;vc{{IIIvc{{r%c2ttjdg}ttjdg}|dsJ|dsJ|dsJ|drJdS)NTFr\)rr.ryr rv)rGs1r!s r#test_all_any_skipnaz(TestSeriesReductions.test_all_any_skipnas RVTN # # RVUO $ $vvUv#####vvTv"""""vvUv#####666&&&&&&&r%ctgdgd}|dsJ|drJdS)N)FFTTFT)rrrQrQrrr9T) bool_only)rrvr )rGrs r#test_all_any_bool_onlyz+TestSeriesReductions.test_all_any_bool_onlys^ :::BTBTBT U U Uuutu$$$$$5545(((((((r% bool_agg_funcrvr c~tgdt}t|||}d}||ksJdS)N)rr{r der[r\T)rrKrA)rGrr]r:rHrIs r#test_any_all_object_dtypez.TestSeriesReductions.test_any_all_object_dtypesU...f===,m,,F;;;!!!!!!r%rNc|t|}t||d}|dkod|v}||ksJdS)NFr\rv)rrA)rGrrr:rHrIs r#!test_any_all_object_dtype_missingz6TestSeriesReductions.test_any_all_object_dtype_missingsX Tll,m,,E:::!E)>d$.>!!!!!!r%)rSrRUInt64r-zdata,expected_data)rrr)rQrQrQrrQct||}|||dk}t|||}|tjur|tjus ||ksJdSdS)Nr[r r\)rrArVrW) rGrr]rrY expected_datar:rIrHs r#"test_any_all_nullable_kleene_logicz7TestSeriesReductions.test_any_all_nullable_kleene_logicsy(T''' (%)?@,m,,F;;;"%H$5$5&H:L:L:L:L:L:L$5$5r%ctddgddgd}|dd}tddg}tj||dS)NTFrQrr)rr)rrvrr&r)rGrrHrIs r#test_any_axis1_bool_onlyz-TestSeriesReductions.test_any_axis1_bool_onlys^ dE]!Q88 9 9Q$//4-(( vx00000r%ctddj}t|}t|}d}t jt |5|sJ|sJ|sJ|sJ|sJ|sJ dddn #1swxYwY| d}t|}t|}t jt |5|sJ|sJ|sJ|sJ|sJ|sJ dddn #1swxYwY||dz }t|}t|}|sJ|rJ|sJ|rJ|sJ|rJdS)Nz 1995-01-02rurgz0'(any|all)' with datetime64 dtypes is deprecatedrdr}r) r_datarrr&rgrhr rvr+)rGdtar:rrktdas r#test_any_all_datetimelikez.TestSeriesReductions.test_any_all_datetimelike sq1117Skk s^^@  ' S A A A " "7799   7799   7799   7799   6688<<>> ! ! !6688<<>> ! ! ! ! " " " " " " " " " " " " " " "ooe$$Skk s^^  ' S A A A " "7799   7799   7799   7799   6688<<>> ! ! !6688<<>> ! ! ! ! " " " " " " " " " " " " " " "CFlSkk s^^wwyy7799wwyy7799vvxx||~~6688<<>>!!!!!s&B)DD D !B)HHHcVtjdtddgd}|sJ|rJtddgd}|sJ|sJ|drJtddgd}|rJ|rJtddgd}|sJ|sJdS) Npyarrowrzstring[pyarrow_numpy]r[Fr\r{)rn importorskiprrvr rGr:s r#test_any_all_pyarrow_stringz0TestSeriesReductions.test_any_all_pyarrow_string9s/I&&&b#Y&=>>>wwyy7799dC[(?@@@wwyywwyy77%7(((((dBZ'>???77997799c3Z'>???wwyywwyyr%c^tddd}t|tdz }|}|dksJ|}|dksJt j|d<|}|dksJ|}|dksJttdd }ttd d }t||z }t j||z }tj ||||z }tj ||| }td }||ksJ| }td }||ksJdS) Nz2012-1-1rurr~r>20120101rrrQrg20120102rr) rrrrarbr.ryabsr&rr9rr:)rGdtirrHr}r!rIs r#test_timedelta64_analyticsz/TestSeriesReductions.test_timedelta64_analyticsMsQS999 C[[9Z00 0{{{{{{{{1{{{{{{{{Jz1555 6 6 Jz1555 6 6"r'??R vx000r' vx000X&&!!!!X&&!!!!!!r%ztest_input,error_typer[foobarbazrQrrrrcrtgd}d}tjt|5|dddn #1swxYwYtjt|5|ddddn #1swxYwYd}tjt|5|dddn #1swxYwYtjt|5|dddddS#1swxYwYdS) zU Cases where ``Series.argmax`` and related should raise an exception rMr[z*attempt to get argmin of an empty sequencerdNFr\z*attempt to get argmax of an empty sequence)rrnrorprarb)rG test_input error_typerks r#"test_assert_idxminmax_empty_raisesz7TestSeriesReductions.test_assert_idxminmax_empty_raisesvs)Bi000 : ]:S 1 1 1                      ]:S 1 1 1 , ,   U  + + + , , , , , , , , , , , , , , ,: ]:S 1 1 1                      ]:S 1 1 1 , ,   U  + + + , , , , , , , , , , , , , , , , , ,sGAAA6BB BC""C&)C&D,,D03D0c tgd}|dksJ|ddksJ|dksJ|ddksJtddg}|dksJ|ddksJ|dksJ|ddksJtddd d dtjd g}d }t jt| 5|dddn #1swxYwYt jt| 5|ddddn #1swxYwYd }t jt| 5|dddn #1swxYwYt jt| 5|dddddS#1swxYwYdS)NrrFr\rQrrrrrz8'>' not supported between instances of 'float' and 'str'rdz8'<' not supported between instances of 'float' and 'str')rrbrar.ryrnror)rGr:ser2ser3rks r#test_idxminmax_object_dtypez0TestSeriesReductions.test_idxminmax_object_dtypes****++zz||q    zzz''1,,,,zz||q    zzz''1,,,,tTl##{{}}!!!!{{%{((A----{{}}!!!!{{%{((A----ueUE4GHHH ]9C 0 0 0   KKMMM                ]9C 0 0 0 & & KKuK % % % & & & & & & & & & & & & & & &H ]9C 0 0 0   KKMMM                ]9C 0 0 0 & & KKuK % % % & & & & & & & & & & & & & & & & & &sH8EE E?F""F&)F& G++G/2G/H55H9<H9ctddgddgddgg}|}tddg}tj||dS) NzimmrNbiffr&bidg(@rr)rrbrr&r)rGrrXrQs r#test_idxminmax_object_framez0TestSeriesReductions.test_idxminmax_object_framesW  }udmD E EiikkaVnn sC(((((r%ctgd}|dksJ|dksJ|ddksJ|ddksJdS)N))rQru)rr)rurQrrFr\)rrbrars r#test_idxminmax_object_tuplesz1TestSeriesReductions.test_idxminmax_object_tupless---..zz||q    zz||q    zzz''1,,,,zzz''1,,,,,,r%cttddgtdtdgtdtdgd}|}tdddd}t j|||}|dz }t j||dS)NrrQz8.68z42.23z7.11z79.61)r!xy)rrrbrr&rra)rGrrXrQres2exp2s r#test_idxminmax_object_decimalsz3TestSeriesReductions.test_idxminmax_object_decimalss 1vfooww'7'78fooww'7'78    iikkQQQ//00 sC(((yy{{Qw tT*****r%ctddgd}|dksJ|dksJ|ddksJ|ddksJdS)NrrQrKr[Fr\)rrxrwrs r#test_argminmax_object_intsz/TestSeriesReductions.test_argminmax_object_intssaV8,,,zz||q    zz||q    zzz''1,,,,zzz''1,,,,,,r%c$tdtj tjtjg}|dksJd}t jt|5tj|dsJ dddn #1swxYwY| dksJd}t jt|5tj| dsJ dddn #1swxYwYd }t jt|5tj d d 5|dksJtj|dsJ| dksJtj| ddddn #1swxYwYddddS#1swxYwYdS) NrrQr_rdFr\rr`rrT) rr.rryrar&rgrhrirbrVr)rGrrks r#test_idxminmax_with_infz,TestSeriesReductions.test_idxminmax_with_infs Aw/ 0 0xxzzQ@  ' S A A A 4 48AHHEH2233 3 3 3 3 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4xxzzQ@  ' S A A A 4 48AHHEH2233 3 3 3 3 4 4 4 4 4 4 4 4 4 4 4 4 4 4 43  ' S A A A 1 1"#7>> 1 1xxzzQx 6 677777xxzzQ//000  1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1s[*+B""B&)B&$+DD #D HBG-! H-G1 1H4G1 5HH  H ctdgd}|}tjd}t j||dS)Nl 9Vuint64r[)rrr.rr&r)rGrrHrIs r#test_sum_uint64z$TestSeriesReductions.test_sum_uint64sN () : : :9122 vx00000r%)/rrrr"rnrrr,r2r7r;rBrFrMrTrVrZr]rcrnrrryr{r~rrr.ryrrVrWrrrrrrrprrrrrrrrrrr%r#rr<su ###( [EEE [-e}== [^lM-JKK^ ^ LK>= ^ @ [X77 [W&E&E&EFF  GF87  [X'G'G'GHH!!IH!0***  [-e}== [Ww&899 ( (:9>= ( [-e}== [Wy)&<== 5 5>=>= 5 % % % [[8X*>??@? [W&STT [Xe}55AA65UTA ###$###"   4B---^'''))) [_uen== [Xe}55""65>=" [_uen== [%u rvP"">=" [W&O&O&OPP [_uen== [Xe}55 [  YY%%8 9 YY$d|4 5eRUBE "beRU^eT]$C D]beU^eU^< =]dBE]T4L9 :]dE]T5M: ;   MM  65>=QP M111-"-"-"^('"'"'"R [ VBi ( ( (* 5 V))) * *I 6 VT4L ! !9 - VUE5%rvuE F F R    ,,  , &&&6)))---+++"---111011111r%rc eZdZejdeeegeeedgeededggdZ ejde eege eedge ededggdZ dZ dZ dS) TestDatetime64SeriesReductionsnat_sernatc|tusJ|tusJ|dtusJ|dtusJdSrr:r r9)rGrs r#test_minmax_nat_seriesz5TestDatetime64SeriesReductions.test_minmax_nat_seriessz{{}}####{{}}####{{%{((C////{{%{((C//////r%nat_dfc&|dtusJ|dtusJ|ddtusJ|ddtusJdS)NrFr\r)rGrs r#test_minmax_nat_dataframez8TestDatetime64SeriesReductions.test_minmax_nat_dataframeszz||A#%%%%zz||A#%%%%zzz''*c1111zzz''*c111111r%c.tdd}|tjdt |}|}|}t|tsJt|tsJ||dksJ||dksJ||dksJ||dksJdS)N1/1/2000z 12/31/2000rrrf) rtaker.r/r0 permutationlenr:r9rBr)rGrngrng2the_minthe_maxs r# test_min_maxz+TestDatetime64SeriesReductions.test_min_maxs\22xx --a00<>&%//ssSHX%%%%%%RV######r%N) rrrrnrrrrr.ryrrrrr%r#rr;s  [Z%88%%98% [ T%[[$$u++ & T%[[$$u++ & T%[[BF8 #TT%[[ 1 YY "Arv  *     [Z%88""98  " [Z%88 [Xe}55  6598  [Z%88 [Xe}55 $ $6598 $ $ $r%rc 0eZdZejddegejfdegejfgdZ ejddgddgfdgd dd gfdgddgfdgd dd gfgejd e ej d ej d zdZ ejdddgfddej gfgdZejdddgdgdgfddgej gdgfgdZejdddgdgfddgej gfgdZejddgdddgfdej gej ddgfgdZejddgdddgfdej gej ddgfgd Zejddedd!gdd!g"ed#gdd#g"ed dggd$d%fdeej gdd!g"eej d#gdd#g"eej d dggd$d%fgd&Zejddd'gdd'gfdd'gdd'gfgd(Zd)Zd*Zejd+gd,edgejejfgd,edgejejfgd-ed.gejejfgd/Zejd+gd0egd1ejejfgd2ed3d.gejejfgd4Zd5S)6TestSeriesModezdropna, expectedTr[Fctgtj}||}t j||dSr/)rr.rMmoder&rrGrrIrrHs r#test_mode_emptyzTestSeriesMode.test_mode_emptyvs@ 2RZ ( ( ( vx00000r%zdropna, data, expected)rQrQrQrrQ)rQrQrQrrurururudt AllIntegerFloatct||}||}t||}tj||dSr/rrr&r)rGrrrIrrrHs r#test_mode_numericalz"TestSeriesMode.test_mode_numericalsP 4r " " "("--- vx00000r%r&ctdddtjtjg}||}t|}t j||dS)NrQr)rr.ryrr&rrs r#test_mode_numerical_nanz&TestSeriesMode.test_mode_numerical_nansS Aq!RVRV, - -(## vx00000r%z'dropna, expected1, expected2, expected3r{rrycdgdzdgdzz}t|d}||}t|d}tj||dddtjtjtjg}t|t }||}t|t }tj||dddtjtjtjg}t|t t}||}t|t}tj||dS) Nrrr{rur r[rr) rrr&rr.ryrKrEstr)rGr expected1 expected2 expected3rrrHs r#test_mode_str_objz TestSeriesMode.test_mode_str_objs0 uqyC519$ 4s # # #9C000  vy111ueRVRVRV< 4v & & &9F333  vy111ueRVRVRV< 4v & & & - -c 2 29C000  vy11111r%zdropna, expected1, expected2rctgd}||}t|}tj||tdddtjtjtjg}||}t|t }tj||dS)N)rQrrrQrr[)rrr&rr.ryrK)rGrrrrrHrIs r#test_mode_mixeddtypez#TestSeriesMode.test_mode_mixeddtypes $$$ % %)$$ vx000 AueRVRVRV< = =)6222 vx00000r%) 1900-05-03r 2013-01-02rrc>tgdd}||}t|d}tj||tgdd}||}t|d}tj||dS)N)rrrryryr?r[)rrrrrryryrrGrrrrrHs r#test_mode_datetimez!TestSeriesMode.test_mode_datetimes  D D DH   9H555  vy111     9H555  vy11111r%)-1 days0 daysr2 min1 dayc>tgdd}||}t|d}tj||tgdd}||}t|d}tj||dS)N)rr r ryryrr[)r r z-1 dayz -1 day 2 minr r ryryrrs r#test_mode_timedeltaz"TestSeriesMode.test_mode_timedeltas  9 9 9AR   9,=>>>  vy111    $ 9,=>>>  vy11111r%r)r rrr c ttddtjtjg}||}t|d}t j||ttdddtjtjg}||}t|d}t j||ttdddddtjtjggdd }||}t|d}t j||dS) NrQrcategoryr[rrurTr )rrr.ryrr&r)rGrrrrrrHs r#test_mode_categoryz!TestSeriesMode.test_mode_categorys-$ ;1bfbf566 7 79J777  vy111 ;3RVRV<== > >9J777  vy111  Aq!Q/IIIt      9J777  vy11111r%ctgdtj}||}t|tj}t j||tddgtj}||}t|tj}t j||dS)N)rQrrr[rQr)rr.rrr&rrs r#test_mode_intoverflowz$TestSeriesMode.test_mode_intoverflow,s $$$BI 6 6 69BI666  vy111 Au:RY / / /9BI666  vy11111r%ctdtjg}tdddtjtjg}tjt 5|d}|d}dddn #1swxYwYtj ||dS)NrrQF)rT)drop) rr.ryr&rg UserWarningr sort_values reset_indexr)rGrIrrHs r#test_mode_sortwarningz$TestSeriesMode.test_mode_sortwarning<s5"&/** AueRVRV4 5 5  ' 4 4 A AVV5V))F''))5545@@F A A A A A A A A A A A A A A A vx00000s?B((B,/B,ctdddtjgd}|}tddid}t j||dS)NTFrSr[r)rrVrWrr&r)rGr:rHrIs r#test_mode_boolean_with_naz(TestSeriesMode.test_mode_boolean_with_naIs[dE4/yAAA1d)9555 vx00000r%zarray,expected,dtype)r?rQrQ???@)r@rrcvt||}tj||dSr/rrGrHrIrYrHs r#test_single_mode_value_complexz-TestSeriesMode.test_single_mode_value_complexPs;*U+++0022 vx00000r%)rrrQrr )yry?rr )rr!rr!rur!cvt||}tj||dSr/rr#s r#test_multimode_complexz%TestSeriesMode.test_multimode_complexhs;(U+++0022 vx00000r%N)rrrrnrrrr.rMrr typecodesrryrrrr rrrrrr complex128 complex64r$r&rr%r#rrqs  [ r,,, -vvb 7S7S7S/TU11 1  [ << 1 1  1 [&:::|,  RVHrv|\B C    22  20 [& 222Wg4F G RVHrvw8 9 2226 [1 QF1v666 SEq#h777 QFyyy$GGG   RVH!Q888 RVSMq#h??? RVQNyyy$OOO   "22#"2( [& !U $uug5z&BC 2 2  2 1 1 1111 [.--s"-000   .--s",///   %$$xr}555   (11)(1 [+**777r}MMM  ,++F|2<888    11! 111r%r)%rrdecimalrnumpyr.rnpandasrVrrrr r r r r rrrrrrrrpandas._testing_testingr& pandas.corerr5r7rrrrrrr%r#r0st $$x1x1x1x1x1x1x1x1vKKKKKKKK\ u 1u 1u 1u 1u 1u 1u 1u 1pDDDDDDDDN3$3$3$3$3$3$3$3$lL1L1L1L1L1L1L1L1L1L1r%