U md@sddlZddlZddlZddlmZddlmZej dddgddZ ej dd d gd d Z d dZ ej dd d gddZddZddZej dejejgej ddd ideigddZddZddZd d!ZdS)"N)Serieszoperation, expected)mina)maxbcCs,tddgdd}t||}||ks(tdS)Nrrstringdtype)rgetattrAssertionError)Z operationexpectedserZres_operation_serier\/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/pandas/tests/series/test_reductions.pytest_reductions_series_strings s r as_periodTFcCs^tddtddD}|r*|jd}n |jd}|}|j|jksNtt ||dS)NcSsg|]}tdd|qS)i)pd Timestamp).0nrrr sz-test_mode_extension_dtype..Dz US/Central) rrangedtZ to_periodZ tz_localizemoder r tmZassert_series_equal)rr resrrrtest_mode_extension_dtypes r cCsNtdtjgdd}|d}||ks*t||ks:t||ksJtdS)Nrzm8[ns]r)rrZNaTmedianr rr)r exprrrtest_reductions_td64_with_nat s r#skipnacCs>tgdd}|j|d}t|tjs(t|tdks:tdS)Nztimedelta64[ns]r)r$r)rsum isinstancer Timedeltar )r$r resultrrrtest_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%npZallclose_valuer pytestraises ValueError)r r(r msgs1s2rrrtest_td64_summation_overflow3s&  r7cCs0tdddgtdd}|}t|tr,tdS)Ng?)index)rrprodr&r )r r(rrrtest_prod_numpy16_bugLsr;funckwargskeepdimsoutc CsZtddg}t|d}|j}d|d|d}tjt|d||f|W5QRXdS)Nrrzthe 'z=' parameter is not supported in the pandas implementation of z\(\)r,)rlist__name__r1r2r3)r=r<r paramnamer4rrr)test_validate_any_all_out_keepdims_raisesSs  rEc Cs<tddg}d}tjt|dtj|ddW5QRXdS)Nrr@zPthe 'initial' parameter is not supported in the pandas implementation of sum\(\)r, )initialrr1r2r3r/r%r r4rrrtest_validate_sum_initialcs  rJc Cs:tddg}d}tjt|d|jddW5QRXdS)Nrr@z[the 'overwrite_input' parameter is not supported in the pandas implementation of median\(\)r,T)Zoverwrite_input)rr1r2r3r!rIrrrtest_validate_median_initialns  rKc Cs<tddg}d}tjt|dtj|ddW5QRXdS)Nrr@zQthe 'keepdims' parameter is not supported in the pandas implementation of sum\(\)r,T)r>rHrIrrrtest_validate_stat_keepdims{s  rL)numpyr/r1ZpandasrrZpandas._testingZ_testingrmarkZ parametrizerr r#r)r7r;anyallobjectrErJrKrLrrrrs&