U W7b@sddlZddlZddlZddlmZmZddlmZ ej dddgddZ dd Z ej d ddgd d Zd dZddZddZej dejejgej dddideigddZddZddZddZdS)N) MultiIndexSeries as_periodTFcCs^tddtddD}|r*|jd}n |jd}|}|j|jksNtt ||dS)NcSsg|]}tdd|qS)i)pdZ Timestamp).0nr G/tmp/pip-unpacked-wheel-rhjceu4d/pandas/tests/series/test_reductions.py sz-test_mode_extension_dtype..Dz US/Central) rrangedtZ to_periodZ tz_localizemodedtypeAssertionErrortmassert_series_equal)rserresr r r test_mode_extension_dtype s rcCsNtdtjgdd}|d}||ks*t||ks:t||ksJtdS)Nrzm8[ns]r)rrZNaTmedianrminmax)rexpr r r test_reductions_td64_with_nats rskipnacCs>tgdd}|j|d}t|tjs(t|tdks:tdS)Nztimedelta64[ns]r)rr)rsum isinstancer Timedeltar)rrresultr r r test_td64_sum_empty$s  r$c Csttjdddd}|dtd7<||}tt||jt| }t |j d|j ds|t d}tjt|d || W5QRX|dd }tjt|d || W5QRX|dd}|| dS) NZ20130101iH)Zperiodsfreqrz1s 1msizoverflow in timedelta operationmatchi')rrZ date_ranger"rZmeanZTimedeltaIndexZasi8lenr npZallclosevaluerpytestraises ValueError)rr#expectedmsgs1s2r r r test_td64_summation_overflow.s&  r3cCs0tdddgtdd}|}t|tr,tdS)Ng?index)rrprodr!r)rr#r r r test_prod_numpy16_bugGsr8c CsXtdgtdgd}tt|jdd}W5QRXtdgdgd}t||dS)Ng$@)r4r5r)levelr9)rr from_tuplesrZassert_produces_warning FutureWarningr r)objr#r/r r r test_sum_with_levelNs  r>funckwargskeepdimsoutc CsZtddg}t|d}|j}d|d|d}tjt|d||f|W5QRXdS)Nr r9rzthe 'z=' parameter is not supported in the pandas implementation of z\(\)r')rlist__name__r,r-r.)r@r?rparamnamer0r r r )test_validate_any_all_out_keepdims_raisesWs  rGc Cs<tddg}d}tjt|dtj|ddW5QRXdS)Nr r9zPthe 'initial' parameter is not supported in the pandas implementation of sum\(\)r' )initialrr,r-r.r*r rr0r r r test_validate_sum_initialgs  rLc Cs:tddg}d}tjt|d|jddW5QRXdS)Nr r9z[the 'overwrite_input' parameter is not supported in the pandas implementation of median\(\)r'T)Zoverwrite_input)rr,r-r.rrKr r r test_validate_median_initialrs  rMc Cs<tddg}d}tjt|dtj|ddW5QRXdS)Nr r9zQthe 'keepdims' parameter is not supported in the pandas implementation of sum\(\)r'T)rArJrKr r r test_validate_stat_keepdimss  rN)Znumpyr*r,ZpandasrrrZpandas._testingZ_testingrmarkZ parametrizerrr$r3r8r>anyallobjectrGrLrMrNr r r r s$