gL i#+ ddlmZddlZddlZddlZddlmZddlm Z m Z ddl m Z ddlmZmZej"j%dgdej"j%dej&gid ej&d gid ej&gd d id ej&gddid ej&d dgddid gej"j%dddgdZej"j%dgddZej"j%dgdej"j%dddgdZej"j%dgdej"j%dddgdZej"j%dee j0e ej2 d!d"ej4fd#ej4fd$d%d&ej4fd'd(ej4fd)ej4fd*ej4fg e j0e ej4d d+d,ggd-e j0e d.j7gd/d0Zej"j%dee j0e ej2 d1e gej2 fd2e gej2 fge j0e ej4d d+d,gd1e ej4d d+d3gfd2e ej4d d,d3gfge j0e d.j7d2e gd4fgd5Zej"j%d6ee j0e d7e d8 fd"e d8 fd#e d8 fd9e e fd:e e fd&e d8 fd;e d8 fd(e d8 fd)e d8 fd*e d8 fg e j0e ej4d gd d+ggd7e dgfd)e ej4d?gfd*e d d=gfg d@Zej"j%d6ee j0e d1e fd2e fge j0e ej4d gd d+ggd1e ej4d gd d+ggfd2e ej4d gd d,ggfgdAZ ej"j%dedBZ!ej"j%dedCZ"ej"j%dDgdEdFZ#y)G)chainN) is_number) DataFrameSeries)frame_transform_kernelsseries_transform_kernelsfunc)summeanminmaxstdz args,kwdsno_args_or_kwds)idaxis_from_argsaxisaxis_from_kwds numeric_onlyT optional_kwds args_and_kwdshowaggapplyct|dkDr9|dk(r4|jtjj t dt |||g|i|}t |||i|}tj||y)Nrrz>agg/apply signature mismatch - agg passes 2nd argument to funcraisesreason) len applymarkerpytestmarkxfail TypeErrorgetattrtmassert_series_equal)request float_framer argskwdsrresultexpecteds a/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/pandas/tests/apply/test_str.pytest_apply_with_string_funcsr/s 4y1} KK   #   'W[# &t ;d ;d ;F)w{D)4848H68,argcV|j|}t||}||k(sJy)N)rr%)datetime_seriesr1r,r-s r.test_with_string_argsr4.s1  " "3 'F,w,.H X  r0op)r medianrvarctddgddgd}t|||}|dvrddini}ttt||fdd i||j }t j ||y) Nr)ab)rr7ddofrr)index)rr%rnpcolumnsr&r')r5rr)r,kwargsr-s r.test_apply_np_reducerrC5s~1a&1v67K &W[# &r *F.0fa[bFB 6!6v6k>Q>QH68,r0)absceilcoscumsumexplogsqrtsquare transformcd|jd<d}|dvrt}tj|d5t |||}t t ||}dddtj y#1swY xYw)Ng)rr)rIrJF)check_stacklevel)ilocRuntimeWarningr&assert_produces_warningr%r@assert_frame_equal)r)r5rwarnr,r-s r.test_apply_np_transformerrTCs"KT D _ # #D5 A0+c*2."72r?;/ 0 &(+ 00s )A;;Bzseries, func, expected)dtype)r rr r allT)anyFr )prodrrr7r6r9r:) )r )r r:)r rrVrXT)r r9)rYrZ)rr)r7r)r6r9za b c))r abc)r c)r r<rVr[ct|trdnt}tj|d5|j |}dddt |rtj|dsJy|k(sJy#1swY7xYw)Nis currently using Series.*matchT) equal_nan) isinstancestr FutureWarningr&rQrrr@iscloseseriesr r-rSr,s r.test_agg_cython_table_seriesriYsubdC(4mD # #D0M N"D!"zz&(d;;;!!! ""s A88BcumprodrGrZ)r<abr\ct|trdnt}tj|d5|j |}dddtj |y#1swY xYw)Nr_r`rcrdrer&rQrr'rgs r.&test_agg_cython_table_transform_seriesrnsX4dC(4mD # #D0M N"D!"68,""s A!!A*zdf, func, expectedr float64rWrXrYg?g?g۞?g?ct|trdnt}tj|d5|j ||}dddtj |y#1swY xYw)Nis currently using DataFrame.*r`rrmdfr r-rrSr,s r.test_agg_cython_table_framerus_NdC(4mD # #D0P Q)4()68,))s A##A,c|dvr|jd}t|trdnt}t j |d5|j ||}dddt j|y#1swY xYw)N)rArrorqr`rr)astypercrdrer&rQrrRrss r.%test_agg_cython_table_transform_framerxst& ~??9-dC(4mD # #D0P Q)4()&(+))s A88Bc|dk(r4|jtjjtd|dk(rdgng}t j |jd}|dk(rtnd}d}tj||5|j|j|g|}ddd|j|dg|}tj|y#1swY4xYw) Nngroupngroup not valid for NDFramerfillnarz"SeriesGroupBy.fillna is deprecatedr`)r r!r"r# ValueErrorr@onesshaperer&rQgroupbyrLr') r( string_seriesr5r*rrSmsgr-r,s r.$test_transform_groupby_kernel_seriesrs X~ KK  Z8V  W (NC5D 77=&&q) *D(N=D .C # #D 4D8=((.88CdCD $] $ $R 2T 2F68,DDs #C""C+c:|dk(r4|jtjjtd|dk(rdgng}|dvr%t j |jd}d}n$t j |jd }d }tjt| 5|j|| }ddd|dk(rtnd}d } tj|| 5j|g|} ddd|j||g|} tj|  d|jvsJ|dj|d<t!|j"j$d kDsJ|dvr#t j |jd}n"t j |jd }tjt| 5|j|| } ddd|dk(rtnd}d } tj|| 5 j|g|} ddd|j||g|}tj| y#1swYxYw#1swYxxYw#1swYxYw#1swYZxYw)Nrzr{rr|r})rr?rz5The 'axis' keyword in DataFrame.groupby is deprecatedrz+DataFrame.groupby with axis=1 is deprecatedr`rrz%DataFrameGroupBy.fillna is deprecatedEA)r r!r"r#r~r@rrr&rQrerrLrRrAcopyr_mgrarrays)r(rr)r5r*rrgbrSop_msgr-r,gb2 expected2result2s r.#test_transform_groupby_kernel_framersm X~ KK  Z8V  W (NC5D |ww{((+,Eww{((+,; # #M =2  D 12(N=D 4F # #D 7+2<<*T*+#[ " "2t 3d 3F&(+ k)) )) )"3',,.K {&& '! ++ + |ww{((+,ww{((+, # #M =3!!$T!23(N=D 4F # #D 7-!CMM",t, -#k##B4t4G'9-722 ++33--s0,I+.I8"J$J+I58JJJmethod)rDshift pct_changerGrankctdddgi}|j|}tj||}t j ||y)Nrr9)rrLoperator methodcallerr&rR)rrtr,r-s r.test_transform_method_namer@sK C"a> "B \\& !F,x$$V,R0H&(+r0)$ itertoolsrrnumpyr@r!pandas.core.dtypes.commonrpandasrrpandas._testing_testingr&pandas.tests.apply.commonrrr" parametrizeparamr/r4rCrTget_cython_table_paramsronansplitrirnboolrurxrrrr0r.rs / !EF R 12 aS"!12 R&!)9: R.$/OD aY 6?K   01 -2 G - DEF ?@ 01 -2A - Jg 67,8,$ """  $ 266"   #"" BFFAq!$ %   #"" 7==? #  ?),Z "[,Z " """  $F2RZZ896"BJJ78   #"" BFFAq!$ %FBFFAq!#45662661a"345   #"" 7==? #x8J1K&L%M ,--,- """ KY/0Y/0Y/0T*+T*+i01i01Y/0Y/06 23   #"" {QF+ ,Qx()Qx()Qx()d|,-d|,-C)*a)*123}-.61c(+,  !"F-G"F- """ K9ik2Xy{4KL  #"" {QF+ ,I{QF&;<=9rvvqkAq6%:;<   , ,78-9-"67+.8+.\#ST,U,r0