gL i dZddlZddlZddlZddlZddlmZ ejejejjejjdZGddZy)zX Tests for Series cumulative operations. See also -------- tests.frame.test_cumulative N)cumsumcumprodcummincummaxc& eZdZejj dejejgdZ ejj dddgdZ ejj de jde jd e jd jd gejj d dd gd gdd gdgddgdgddgdggdZejj dde j ddfde j ddfgdZejj dgdgdgejj dddgddgejj dej'd Zejj d!d"e j*dd#ej,d#ge$gd%e j*ddej,dggde j*ddej,dggde j*dd ej,d gggd&Zd'Zejj d(gd"d gfgd"dgfgd)d"d gd*fgd)d"dgd*fd+e j4d,gd"d d+e j4d-gfd+e j4d,gd"dd+e j4e j4gfe j4d+d,gd"d e j4d+d-gfe j4d+d,gd"de j4e j4e j4gfe j4e j4e j4gd"d e j4e j4e j4gfe j4e j4e j4gd"de j4e j4e j4gfgdd gfgddgfgd.dd gd/fgd.ddgd/fd,e j4d+gdd d,e j4d+gfd,e j4d+gddd,e j4e j4gfe j4d,d+gdd e j4d,d+gfe j4d,d+gdde j4e j4e j4gfe j4e j4e j4gdd e j4e j4e j4gfe j4e j4e j4gdde j4e j4e j4gfgdd gfgddgfgd)dd gd0fgd)ddgd0fd+e j4d,gdd d+e j4d,gfd+e j4d,gddd+e j4e j4gfe j4d+d,gdd e j4d+d,gfe j4d+d,gdde j4e j4e j4gfe j4e j4e j4gdd e j4e j4e j4gfe j4e j4e j4gdde j4e j4e j4gfgd1Zd2Zy3)4TestSeriesCumulativeOpsfuncctj||j|tj|d|j }tj |ddd<||ddd}|tj|j}tj|j|dy)NT) check_dtypeF)tmassert_numpy_array_equalvaluesnparraycopynandropna)selfdatetime_seriesr tsresultexpecteds i/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/pandas/tests/series/test_cumulative.pytest_datetime_seriesz,TestSeriesCumulativeOps.test_datetime_seriess ##  ! ( ( /* +  ! ! #&&3Q3b!$Q$-. ##FMM8Omethodrrct|}t||j}|tj|}t j |||j}tj|ddd<t||ddd}||j}|jjd|_ t j||y)Nr r ) methodsgetattrrrrrrrrrindex _with_freqassert_series_equal)rrrufuncrrrs rtest_cummin_cummaxz*TestSeriesCumulativeOps.test_cummin_cummax+s1&13::/23 ##FH5  ! ! #&&3Q3$V$&qt!t,%||..t4  vx0rrrz 1999-12-31z US/Pacificzmethod, skipna, exp_tdiT)NaT2 daysr'r(r'3 days)r'r(r'1 daysr'r*F)r'r'r'r'r'r'ctjgd}tj||z}tj|}tj||z}t|||}t j ||y)N)r'r(r'r*r'r)skipna)pd to_timedeltaSeriesr!rr$) rrrr-exp_tditdiserrrs rtest_cummin_cummax_datetimelikez7TestSeriesCumulativeOps.test_cummin_cummax_datetimelike;sh:ooQRiib!//'*99Wr\*%f%V4 x0rz func, exp2012-1-1Dfreq2012-1-2cPtjtjddtjtjddg}t ||d}tjtjddtjtjg}t j ||t ||d}tjtjddtj|g}t j ||y)Nr5r6r7r9Fr,T)r.r0Periodr'r!rr$)rr expr3rrs rtest_cummin_cummax_periodz1TestSeriesCumulativeOps.test_cummin_cummax_period`sii YYz ,bffbii QT6U V $d#5199bii =rvvrvvNO vx0#d#4099bii =rvvsKL vx0rarg)FFFTTFF)FFFFFFFc|SNxs rz TestSeriesCumulativeOps.|s1rc|Sr@rArBs rrDz TestSeriesCumulativeOps.|sridentityinverse)idsc|tj|}t|}||j}tj|}t ||}t j ||yr@)r.r0r rr!rr$) rr>r rr3r%exp_valsrrs rtest_cummethods_boolz,TestSeriesCumulativeOps.test_cummethods_boolts\299S>"$99X&%f%' vx0rzmethod, expectedrr dtyperctjddtjdg}t ||}t j ||y)NFT)r.r0rrr!rr$)rrrr3rs r$test_cummethods_bool_in_object_dtypezzdata, op, skipna, expected_data)rCzy)rCxzxzyrCr[xy)r[rZrC)r[r[rC)rCrZrZctj||}t||}tj||}||} tj| |y)NrLr,)r.r0r!rr$) rstring_dtype_no_objectdataopr- expected_datar3rrrs rtest_cum_methods_ea_stringsz3TestSeriesCumulativeOps.test_cum_methods_ea_stringssMPii$:;b!99]2HIv& vx0rctjtd|}tjd|j d}t jt|5|j|dddy#1swYyxYw)NxyzrLz-operation 'cumprod' not supported for dtype ''rSr,) r.r0listreescaperMrVrWrXr)rpyarrow_string_dtyper-r3msgs rtest_cumprod_pyarrow_stringsz4TestSeriesCumulativeOps.test_cumprod_pyarrow_stringsseiiU +?@iiG {RSTU ]]9C 0 ' KKvK & ' ' 's A;;BN)__name__ __module__ __qualname__rVmark parametrizerrrrr&r.rU Timestamp tz_localizer4r;r=r keysrKr0robjectrOrYNArdrmrArrrrsG [[Vbii%<=P>P  [[X(';< 1= 1 [[ BLLO BLL & BLL & 2 2< @  [[! tP Q tP Q:  :  " 1#2 1 [[ yryy#6 7 yryy#6 7  1 1 [[ ; =  [[l+*i1H [[Xw||~6 17 1 [[ yryy!Q!2&A B   5!RVVQ"78 9 yryy%!>? @ yryy%rvvt!<= >  11   [[) 4 $ 5" % h.@ A h/A B255# $beeT0B C255# %#ruubee1D EeeS# $T0B CeeS# %"%%1F GeeRUUBEE "HdRUUBEE2554I JeeRUUBEE "HebeeRUUBEE5J K 4 $ 5" % ho > h ?255# $beeS0A B255# %#ruubee1D EeeS# $S0A BeeS# %"%%1F GeeRUUBEE "HdRUUBEE2554I JeeRUUBEE "HebeeRUUBEE5J K 4 $ 5" % ho > h ?255# $beeS0A B255# %#ruubee1D EeeS# $S0A BeeS# %"%%1F GeeRUUBEE "HdRUUBEE2554I JeeRUUBEE "HebeeRUUBEE5J K= "F 1G"F 1'rr)__doc__rinumpyrrVpandasr.pandas._testing_testingrrrminimum accumulatemaximumr rrArrrsX  iizzjj##jj##  {'{'r