U f/em @sddlZddlZddlZddlmZmZddlmZ ddZ ej dddgdd Z d d Zd d ZddZej dejejgej dddideigddZddZddZddZdS)N) MultiIndexSeriescCsNtdtjgdd}|d}||ks*t||ks:t||ksJtdS)Nrzm8[ns]Zdtype)rpdZNaTmedianAssertionErrorminmax)serexpr G/tmp/pip-unpacked-wheel-tiezk1ph/pandas/tests/series/test_reductions.pytest_reductions_td64_with_nat s rskipnaTFcCs>tgdd}|j|d}t|tjs(t|tdks:tdS)Nztimedelta64[ns]r)rr)rsum isinstancer Timedeltar)rr resultr r r test_td64_sum_emptys  rc 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_rangerrZmeanZTimedeltaIndexZasi8lenrnpZallclosevaluerpytestraises ValueError)r rexpectedmsgs1s2r r r test_td64_summation_overflows&  r#cCs0tdddgtdd}|}t|tr,tdS)Ng?index)rrangeprodrr)r rr r r test_prod_numpy16_bug8sr)c CsXtdgtdgd}tt|jdd}W5QRXtdgdgd}t||dS)Ng$@)r$r%r)levelr*)rr from_tuplestmZassert_produces_warning FutureWarningrZassert_series_equal)objrrr r r test_sum_with_level?s  r0funckwargskeepdimsoutc CsZtddg}t|d}|j}d|d|d}tjt|d||f|W5QRXdS)Nr*rzthe 'z=' parameter is not supported in the pandas implementation of z\(\)r)rlist__name__rrr)r2r1r paramnamer r r r )test_validate_any_all_out_keepdims_raisesHs  r:c Cs<tddg}d}tjt|dtj|ddW5QRXdS)Nr5r*zPthe 'initial' parameter is not supported in the pandas implementation of sum\(\)r )initialrrrrrrr r r r r test_validate_sum_initialXs  r?c Cs:tddg}d}tjt|d|jddW5QRXdS)Nr5r*z[the 'overwrite_input' parameter is not supported in the pandas implementation of median\(\)rT)Zoverwrite_input)rrrrrr>r r r test_validate_median_initialcs  r@c Cs<tddg}d}tjt|dtj|ddW5QRXdS)Nr5r*zQthe 'keepdims' parameter is not supported in the pandas implementation of sum\(\)rT)r3r=r>r r r test_validate_stat_keepdimsps  rA)ZnumpyrrZpandasrrrZpandas._testingZ_testingr-rmarkZ parametrizerr#r)r0anyallobjectr:r?r@rAr r r r s