gL i:)ddlmZddlZddlZddlZddlZddlm Z ddl m Z dddddgZ ejjde dZejjde d Zejjde d Zejjd ej&d gd feej&ddgddggej(d d gd d gggdfej*ej(d d gddgggdfej,dgdfeej.ej(ddgddggdgdfgdZejjde ejjde dZejjde dZdZdZdZdZdZd Z y)!)partialN)is_extension_array_dtypeint64Int64ABdtypectjddgddggd}tj|ddgddg j |}tj |}tjtj ||j |j j |}tj||y Nrr rr abcolumnsindexrr) nparraypd DataFrameastypepositiverrtmassert_frame_equalr valuesdfresultexpecteds c/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/pandas/tests/frame/test_ufunc.pytest_unary_unaryr$sXXBx!Q( 8F fsCjc C J JQV J WB [[_F|| F288RZZ fUm &(+cnt|st|tr/|jtj j dtjddgddggd}tj|ddgd d g j|}tj|}t|tsJt|d k(sJtj|}t||D]G\}}tj||j |j" }t%j&||Iy)Nz8Extension / mixed with multiple outputs not implemented.reasonr rrrrr rrrr)r isinstancedict applymarkerpytestmarkxfailrrrrrmodftuplelenziprrrr) requestr rr result_pandasexpected_numpyr!rr"s r#test_unary_binaryr7s &*UD*A KK  Q   XXBx!Q( 8F fsCjc C J JQV J WBGGBKM mU ++ + }  "" "WWV_N70 <<2::F fh/0r%ctjddgddggd}tj|ddgddg j |}tj ||}tjtj |||j |j j |}tj||yr ) rrrrraddrrrrrs r# test_binary_input_dispatch_binopr:4sXXBx!Q( 8F fsCjc C J JQV J WB VVB^F|| vvbhh  fUm &(+r%zfunc,arg,expectedr)r)FT)where)rr;r<rr))rr) )r rrr))rrctjddgddgg}tj|}tj|}| |||}n ||||}tj|j dd}t j||tj|}t j||y)Nrr)r;r<)out) rrrr zeros_likereshaperassert_numpy_array_equalr)funcargr"arrr result_inplacer!s r#test_ufunc_passes_argsrL@s( ((QFQF# $C c B]]3'N {bn-b#>2xx!))!Q/H9||H%H&(+r%dtype_adtype_bc Dt|s+t|tst|st|tr/|jtj j dtjddgddgdj|}t|tr4t|tr$|j}|jd|d <tjddgddgd j|}tj||}tjtjddtjgddtjggtjdtjdgdtjdgg}tj|d dggd  }t!j"||tj||j$}tjddgddggddg}t!j"||y)N7Extension / mixed with multiple inputs not implemented.r'rr)r;r<rr C)rrQr)rr rQr?rr)rr*r+r,r-r.r/rrrcopypopr heavisidernanrrr)r4rMrNdf1df2r!r"s r# test_binary_input_aligns_columnsrZds !) gt $ #G , gt $ KK  P   ,,aV1a&1 2 9 9' BC'4 Z%>,,.{{3' ,,aV1a&1 2 9 9' BC \\#s #F|| 1a.1a.12 1bffa.1bffa.12H||HQFOLH&(+ \\#szz *F||c3Z#s4sCjIH&(+r%c t|st|tr/|jtj j dtjddgddgddd g j|}tjddgddgddd g j|}tj||}tjtjddgddgtjtjggtjddgtjtjgddgg}tj|gd d dg}tj||tj||j }tjddgddggd dgdd g}tj||y)NrPr'rr)r;r<rrrrcrrr]rr rrRr)rr*r+r,r-r.r/rrrrrVrrWrrr)r4r rXrYr!r"s r#test_binary_input_aligns_indexr_s{&*UD*A KK  P   ,,aV1a&1#s D K KE RC ,,aV1a&1#s D K KE RC \\#s #F|| 1a&1a&266266"234 1a&266266*QF34H ||HOc3ZPH&(+ \\#szz *F|| sc3Z 3*S#JH&(+r%cZtjdddgi}tjtd5t j ||ddddtjtd5t j |d|dddy#1swYGxYw#1swYyxYw)Nrrr) logaddexpmatch)rrr-raisesNotImplementedErrorrrar s r#test_binary_frame_series_raisesrgs sQFm $B *+ >" RC!" *+ >" RWb!""""""sB2B!B!B*ctjdgdi}tjj |}tjdgdi}t j ||tjgdgdd}tjj |}tjgdgdd}t j ||tjj |d }t j ||tjj |d }tjgd gd d}t j ||y) Nr)rr;r)r<)rr;r;r<)皙?@@@rr)rRrkrkrj)rirjrjrjr)axisr)rRrkrlrj)rRrjrkrj)rrrmaximum accumulaterr)r r!r"s r#test_unary_accumulate_axisrqs sL) *B ZZ " "2 &F||S,/0H&(+ L/CD EB ZZ " "2 &F||"6=QRSH&(+ ZZ " "2A " .F&(+ ZZ " "2A " .F||"6=QRSH&(+r%ctjdddgi}tjtd5t j j||dddy#1swYyxYw)Nrrr)rb)rrr-rdrersubtractouterrfs r#test_frame_outer_disallowedrvsP sQFm $B *" 5" "b!"""s !AA(cdtjgdgdd}tjgdgdd}tjddgddg }tjddgdd g }tjgd gd d}tjd5t j ||}dddtj|t j ||j}tj||t j ||}tjt jgd zgdt jgd zd}tj||t j |j|}tjgd gd d}tj||tjgdgdd}tjd5t j ||}dddtj||t j ||j}tj||tjt jgd zgdt jgd zd}t j ||}tj||d}tjt|5t j ||dddy#1swYWxYw#1swY xYw#1swYyxYw)Nrr)r;r<r=rmrr]rr)rrr\r])r)r<rz) r;)r=r?r^)r)r;r<)rzrr|)g@@g@zFCannot apply ufunc to mixed DataFrame and Series inputs.rb) rrSeriesrassert_produces_warningrr9rrrWr-rdre)rXrYs1s2r"r!msgs r##test_alignment_deprecation_enforcedrs_ ,,YY7 8C ,,YY7 8C Aq6#s ,B Aq6#s ,B||)+>?H # #D )"S!"&(+ VVC $F&(+ VVC F||266(Q,YbffXPQ\RSH&(+ VVCJJ $F||)+>?H&(+||))<=H # #D )!R!&(+ VVC #F&(+||vvhlx!|DHVVC_F&(+ RC *# 6 r3E""&!!s$ L 9L,L& LL#&L/ctjd}|j|j|j|j|jgd}t j gdgdd}t j gdgdd}t j gdgdd}||||}t j t jdt jgd  }tj||tjd5||||}dddt j d d gd dgddggddg }tj||d}tjt|5||||jdddtjd5|||j|j}dddtj||d}tjt|5||j||dddy#1swY xYw#1swYxYw#1swYsxYw#1swYyxYw)Nnumbac||z|zSNxyzs r#my_ufunczAtest_alignment_deprecation_many_inputs_enforced..my_ufuncs1uqyr%rxryrmr{)rr])r;r;r^rSrkg(@rg.@g"@g2@rrzLoperands could not be broadcast together with shapes \(3,3\) \(3,3\) \(3,2\)rbzLoperands could not be broadcast together with shapes \(3,2\) \(3,3\) \(3,3\))r- importorskip vectorizefloat64rrrfullrWrrrrd ValueErrorr)rrrXrYdf3r!r"rs r#/test_alignment_deprecation_many_inputs_enforcedrs    (E __emmEMM5==%--PQRS ,,YY7 8C ,,YY7 8C ,,YY7 8C c3 $F||BGGFBFF3_MH&(+ # #D ))#sC()||c4[3+T{CcSVZXH&(+ X z -'c3::&' # #D )7#szz3::67&(+ X z -'S#&''+))''77''s0! H3 I;I I3H=I  II!cd}tj|dd}tjddgddgg}|||d}tjddgddggt }t j ||tjddg}d }tjttj| 5||||dddy#1swYyxYw) Nc||z|zSrrrs r#add3z2test_array_ufuncs_for_many_arguments..add3's1uqyr%r;rr)r<r=rr?rzTCannot apply ufunc to mixed DataFrame and Series inputs.rb) r frompyfuncrrobjectrrrr-rdrereescape)rufuncr r!r"serrs r#$test_array_ufuncs_for_many_argumentsr%s MM$1 %E 1v1v& 'B 2r1 F||aVaV,F;H&(+ ))QF C 0 *"))C. A b"cs > CC)! functoolsrrnumpyrr-pandasrpandas._testing_testingrpandas.api.typesrdtypesr. parametrizer$r7r:r9rpowerrtnegativerLrZr_rgrqrvrrrrr%r#rs  5    &),*,&)0*0*&),*, L! BFFE4=4-"@ A BHHq!fq!f% &  8288aVaV,-}= a' BKKxrxx%u 0N'O P   $,%$,$F+F+,,,,B&),*,2",*".b*'Zr%