ttj <2ddlmZddlZddlZddlZddlZddlZddl m Z ddl m Z ejejejgdZejejdejjdgd d g d Zejd ZejdejejejgdZejdddgddgdZejdddgddgdZejdddgddgejdddgddgdZejdddgdZejdej gejdddgej!ddZ"dZ#dZ$dZ%ejej&gd ej'!ej&gd d"!ej&gd d#!ej&gd$d%!ej(gd&ej(gd&)d'ej(gd&*d()ej+gd*ej,ej-dd+ej-d,d-ej-d+d,gg d. d/Z.Gd0d1Z/ejd2e0ee1gd3Z2d4Z3d5Z4d6Z5ejd7ej6ej7gd8Z8d9Z9d:Z:dS);)dequeN) SparseArray)paramsc|jSNparamrequests \/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/pandas/tests/series/test_ufunc.pyufuncr s  =T)marksFsparsedense)ridsc|jSrrr s r rrs =rctjddddd}tjddddd}d|ddd<d|ddd <||fS) zO A pair of random, length-100 integer-dtype arrays, that are mostly 0. r dint64dtypeN)nprandom default_rngintegers)a1a2s r arrays_for_binary_ufuncr#s   q ! ! * *1b#W * E EB   q ! ! * *1b#W * E EBBsssGBsssG r6Mrr ctjddddd}d|ddd<|r$t |t jdd}ttj dd}d}t j |||}||}t j ||||}tj ||dS)Nrrrrrnameindexr%) rrrr rpd SparseDtypeliststring ascii_lettersSeriestmassert_series_equal)r rarrr'r%seriesresultexpecteds r test_unary_ufuncr4's )   " " + +Ar2W + E ECC!H A#R^GQ%?%?@@@ %crc* + +E D Ys%d 3 3 3F U6]]Fys5t<<'1#=#= > > > 2>'1#=#= > > > D Yr % % %F EbJ5/K ,j)) {++ y +$777H UK F68,,,,,rc|\}}|rHt|tjdd}t|tjdd}d}tj||}tj||d}||f} ||f} |rt | } t | } tj|| |} || } tj| | dSr9) rr(r)r-Indexastyper;r.r/r<s r test_binary_ufunc_with_indexrDPs %FB ? 2>'1#=#= > > > 2>'1#=#= > > > D Yr % % %F HRd # # # * *7 3 3EbJ5/K ,j)) {++ y +$777H UK F68,,,,,rshuffle unalignedalignedc.|\}}|rHt|tjdd}t|tjdd}d}tj||}tj||} tjdt|} |rY| | } |r!| |dj } n(| | dj } n|j } ||f} || f} |r8tt| } tt| } tj|| | |}|| }tj||dS)Nrrrr%r:rr&)rr(r)r-rrr permutationlentakealignr'tupler;r.r/)r5rErr r#r!r"r%r1r=idxr'r>r?r3r2s r test_binary_ufunc_with_seriesrOks%FB ? 2>'1#=#= > > > 2>'1#=#= > > > D Yr % % %F Ibt $ $ $E )   " " . .s2ww 7 7C 3  1KK''*0EELL''*0EE bJ5/K 38J//00 H[1122 y +5tDDDH UK F68,,,,,rcL|\}}|rt|}d}tj|d}||f}||f} |r8tt |}tt | } tj|| d} ||} t j| | dS)Nrr%r:)rr(r-rMr;r.r/) r rr5r#r0_r=r1r?r>r3r2s r test_binary_ufunc_scalarrRs %FC # E Ys ( ( (F5/KuJ 1H[1122 8J//00 y +&999H UK F68,,,,,rz$ignore:divide by zero:RuntimeWarningc|\}}d||dk<d||dk<|rHt|tjdd}t|tjdd}tj|}tj|}|r|d}|||}t |t sJ|||} t | t sJtj| dtj|dtj| dtj|ddS)Nrrr)frac) rr(r)r-sample isinstancerMr.r/) r rrEr#r!r"s1s2r3r2s r "test_multiple_output_binary_ufuncsrZs:%FBBrQwKBrQwK ? 2>'1#=#= > > > 2>'1#=#= > > > 2B 2B YYAY  uR}}H h & &&& & U2r]]F fe $ $$$ $6!9bi &<&<===6!9bi &<&<=====rc|\}}|rt|}tj|d}tj|}tj|}t |t sJt |t sJtj|dtj|ddtj|dtj|dddS)Nr%r:rrT) rr(r-rmodfrWrMr.r/)rr#r0rQr1r2r3s r test_multiple_output_ufuncr]s%FC # Ys ( ( (F WV__Fws||H fe $ $$$ $ h & &&& &6!9bi &&I&I&IJJJ6!9bi &&I&I&IJJJJJrc|\}}tj|d}tj|d}|||}|jJdS)Nar:b)r(r-r%)r rr#r!r"rXrYr2s r #test_binary_ufunc_drops_series_namerasV $FB 2C B 2C B U2r]]F ;     rc Gdd}tj|d|dg}tj|}t jtj||tjtj||t jtj||dtjtj||ddS)NceZdZddZdZdS)$test_object_series_ok..DummyreturnNc||_dSrvalueselfrhs r __init__z-test_object_series_ok..Dummy.__init__ DJJJrc |j|jzSrrgrjr=s r __add__z,test_object_series_ok..Dummy.__add__s: + +rreN)__name__ __module__ __qualname__rkrorr Dummyrds7     , , , , ,rrurrT)rarrayr(r-r.r/add)rur0sers r test_object_series_okrys,,,,,,,, (EE!HHeeAhh' ( (C )C..C26#s++RYrvc37G7G-H-HIII26#uuQxx00")BF3ars#ag,,rc|jSrrr s r values_for_np_reducers =rc&eZdZdZdZdZdZdS)TestNumpyReductionscp|}|}tjd5||}dddn #1swxYwYt|tjjjr5tj d}| ||j j dvrtj|}|tjur-|d} tj|| dS|tjur#|j} || ksJdS|} || ksJdSdgd} tjt.| 5tj|ddddS#1swxYwYdS) NzSparseArray has no 'prod')reasoniufF numeric_only|)does not support operationunsupported operand typez$ufunc 'multiply' cannot use operandsmatch)r.assert_produces_warningrWr(corearraysrpytestmarkxfail applymarkerrkindrmultiplyreduce DataFrameprodr/rB_valuesjoinraises TypeError) rjrbox_with_arrayr boxvaluesobjrr2r3msgs r test_multiplyz!TestNumpyReductions.test_multiply s1%  ' - -  #f++C                fbgn8 9 9 &;$$,G$HHD    % % % <  % %['',,Fbl""88877&vx88888;++--))))))88::))))))((Cy444 ( ( ""3''' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (s 155> F++F/2F/c|}|}tjd5||}dddn #1swxYwY|jjdvrtj|}|tjur-| d}tj ||dS|tj ur#|j }||ksJdS| }||ksJdSd gd}tjt |5tj|ddddS#1swxYwYdS)NmiufFrr)rrzufunc 'add' cannot use operandsr)r.rrrrrwrr(rsumr/rBrrrrr) rjrrrrrr2r3rs r test_addzTestNumpyReductions.test_add/s%  ' - -  #f++C                <  & &V]]3''Fbl""77766&vx88888;??,,))))))7799))))))((Cy444 # # c""" # # # # # # # # # # # # # # # # # #s 155% EEEc$|}|}d}|tjur|jjdvrd}t jd5||}dddn #1swxYwYt j|}|tj ur-| d} t j || dS|d} |r"|jjdvr| } || ksJ|r"t|t| ksJdSdS)NTifFrrT)r(rBrrr.rrmaximumrrmaxr/itemtype rjrrusing_python_scalarsrr same_typerr2r3s r test_maxzTestNumpyReductions.test_maxMsf% "(??v|0D88I  ' - -  #f++C               ""3'' ",  wwEw22H  "68 4 4 4 4 4ayH# + (9T(A(A#==??X%%%% 6F||tH~~5555 6 655 AAAc$|}|}d}|tjur|jjdvrd}t jd5||}dddn #1swxYwYt j|}|tj ur-| d} t j || dS|d} |r"|jjdvr| } || ksJ|r"t|t| ksJdSdS)NTrFrr)if)r(rBrrr.rrminimumrrminr/rrrs r test_minzTestNumpyReductions.test_mingsf% "(??v|0D88I  ' - -  #f++C               ""3'' ",  wwEw22H  "68 4 4 4 4 4ayH# + (9Z(G(G#==??X%%%% 6F||tH~~5555 6 655rN)rqrrrsrrrrrtrr rr sQ ( ( (D###<666466666rrtype_ctjgdd}|gd}tj||}tjtj||d}t j||dS)NrTrrr%r:)rr)r(r-rrwto_numpyr.r/)rr_r`r2r3s r test_binary_ufunc_other_typesrs} )))&)))A iiiA VAq\\Fy a00v>>>H68,,,,,rc,Gfddtjddg}tj|d}tjddg}t j||dS)Nc:eZdZddZdZdeffd ZdefdZdS)#test_object_dtype_ok..ThingreNc||_dSrrgris r rkz,test_object_dtype_ok..Thing.__init__rlrcdt|d|}t||j|zS)Nrh)getattrrrhrns r roz+test_object_dtype_ok..Thing.__add__s0E7E22E4::dj5011 1rcFt|uo|j|jkSr)rrh)rjr=Things r __eq__z*test_object_dtype_ok..Thing.__eq__s";;%'EDJ%+,E Ercd|jdS)NzThing()rg)rjs r __repr__z,test_object_dtype_ok..Thing.__repr__s)DJ))) )rrp) rqrrrsrkroboolrrr)rsr rrs     2 2 2 F4 F F F F F F *c * * * * * *rrrTrr)r(r-rrwr.r/)sr2r3rs @r test_object_dtype_okrs * * * * * * * * * * 5588UU1XX&''A VAuuQxx Fy%%((EE!HH-..H68,,,,,rctjgd}tjgd}t jt d5tj||ddddS#1swxYwYdS)Nrz^$r) r(r-rrvrrNotImplementedErrorsubtractouter)rxrs r test_outerrs )III  C (999  C *$ 7 7 7$$ #s###$$$$$$$$$$$$$$$$$$s!A66A:=A:ctjgdg}tjgd}tjdg}tj||}t j||dS)N)rTr)dataf)r(rrmatmulr.assert_frame_equal)df1df2r3r2s r test_np_matmulrsn ,[[[M * * *C ,KKK ( ( (C|#'''H YsC F(F+++++rrc:tj|ddg|ddg}|dksJ|r1t|tksJt|dSt|tjksJt|dS)NrTrr)rrrintr)rrr2s r test_np_matmul_1Drs YssAq6{{CCAKK 0 0F Q;;;;6F||s"""DLL"""""F||rx'''f'''''rcd}tj|dd}tjddg}|||d}tjddgt}t j||tjddgg}d}tj ttj |5||||ddddS#1swxYwYdS) Nc||z|zSrrt)ryzs r add3z2test_array_ufuncs_for_many_arguments..add3s1uqyrrrTrrrzTCannot apply ufunc to mixed DataFrame and Series inputs.r) r frompyfuncr(r-objectr.r/rrrrreescape)rr rxr2r3dfrs r $test_array_ufuncs_for_many_argumentsrs. M$1 % %E )QF  C U3Q  Fy!Qv...H68,,, 1vh  B 0 *")C.. A A A c3s5CCCctjgd}tj|}tjgd}t j||dS)N)ggg?g?)gggg?)r(r-rtruncr.r/)rxr2r3s r test_np_truncrsV )*** + +C Xc]]Fy///00H68,,,,,r); collectionsrrr+numpyrrpandasr(pandas._testing_testingr. pandas.arraysrfixturerw logaddexpr r rfails_arm_wheelsrr# parametrizepositivefloorexpr4r@rDrOrRdivmodfilterwarningsrZr]raryrvr to_datetime tz_localize to_period to_timedelta IntervalIndexIntervalrrr*rMrrrrrBr-rrrrtrr r sU  %%%%%% -.../.  FLV[%A B B BE J 7 2;"&"ABB - -CB - $Y 4KLL--ML-.$Y 4KLL--ML-4T5M Y7OPP$Y 4KLL#-#-MLQP#-L$//--0/-.29+..T5M22BCC>>DC32/.><KKK& T T T"(+++'***),,,=111///00///00<->  Aq@Q@QRSS        t6t6t6t6t6t6t6t6n4"677--87----*$$$,,,29 56666766,-----r