2i"UddlmZddlmZddlZddlZddlZddl m Z e jDcgc]}ejgd|c}ZdgeezZded<ee j$Dcgc]}ejgd |c}z Zed d gz Zeejgd d gz Zed gz Zej&eeedeDcgc]}|j*j,c}dZdZdZdZdZdZdZdZdZdZ ejBjEdgddZ#ejBjEdgdd Z$ycc}wcc}wcc}w)!) annotations)AnyN)Ndtyperz list[Any]scalars)g?皙?g333333?Nr )TFTNbooleanFT)strict)paramsidsc|jS)zFixture returning parametrized (array, scalar) tuple. Used to test equivalence of scalars, numpy arrays with array ops, and the equivalence of DataFrame and Series ops. )param)requests a/var/www/html/acx/venv/lib/python3.12/site-packages/pandas/tests/arrays/masked/test_arithmetic.pydatars ==ct|jtjrd|vrt j dyyy)Nsubz$subtract not implemented for boolean) isinstancer pd BooleanDtypepytestskiprop_names r check_skipr"s1$**boo.5G3C :;4D.rc||jjdk(xr"|jdjddvS)Nb_r)powtruedivfloordiv)r kindstriplstriprs ris_bool_not_implementedr*'s= ::??c ! gmmC&8&?&?&DI'rcB|\}}tj|}t||tj|gt |z|j }||j j|fD]}t||r]d}tjt|5|||dddtjt|5|||dddl|||}|||}tj||y#1swYbxYw#1swYxYwNr-operator '.*' not implemented for bool dtypesmatch) tmget_op_from_namerrarraylenr typer*rraisesNotImplementedErrorassert_extension_array_equal) rall_arithmetic_operatorsscalarop scalar_arrayvalmsgresultexpecteds r"test_array_scalar_like_equivalencer@4sLD&  5 6Bt-.88VHs4y0 CL /0 > "4)A BAC2#> 4  2#> '4& ' 'c]F$ -H  + +FH = >   ' 's D  D D D c|\}}tj|}t||tj}tj tjgt |z|j}|jj}t||rPd}tjt|5|||dddtj||jy|||}tj||j|||} tj||jtj|| y#1swYxYwr,)r0r1rrNAr2r3r _maskcopyr*rr5r6assert_numpy_array_equalr7) rr8r"r:r9r;maskr=r>r?s r test_array_NArGIs GD!  5 6Bt-. UUF88RUUGc$i/tzzBL ::?? Dt%=>= ]].c :  tV   ##D$**5 f Fdjj1$ %Hdjj1##FH5  s 6 EEcT|\}}tj|}t||tj|gt |z|j j}tj||j }t||r]d}tjt|5|||dddtjt|5|||dddy|||}|||}tj||y#1swY`xYw#1swYyxYwr,)r0r1rnpr2r3r numpy_dtyperr*rr5r6r7) rr8r9r: numpy_arraypd_arrayr=r>r?s rtest_numpy_array_equivalencerMesLD&  5 6Bt-.((F8c$i/tzz7M7MNKxx 4::6Ht%=>= ]].c : " t[ ! " ]].c :  tX   k "F$!H##FH5 " " s) D DDD'c|\}}tj|}t||tjd|i}t ||r]d}t jt|5|||dddt jt|5|||dddy|||}tjd|||i}tj||y#1swYuxYw#1swYyxYw)NAr-r.) r0r1rr DataFramer*rr5r6assert_frame_equal)rr8r9r:dfr=r>r?s r test_framerS~sLD&  5 6Bt-. sDk "Bt%=>= ]].c :  r6N  ]].c :  tV   F^F||S"T6"234H&(+   s' C% C1%C.1C:c |\}}tj|}t||tj|}|t j |gt|z|jjtj |gt|z|jtj|gt|z|jg}|D]y}t||r0d}tjt|5|||ddd?|||}tj|||} tj|| {y#1swYxYwr,)r0r1rrSeriesrIr2r3r rJr*rr5r6assert_series_equal) rr8r9r:serothersotherr=r>r?s r test_seriesrZs!LD&  5 6Bt-. ))D/C  &CI%TZZ-C-CD &CI%TZZ8 6(SY&djj9 F 5 "4)A BAC2#> 3  U^FyyD%1H  " "68 4 5  s > EE c V|\}}|}t||}|tjd|i}|tusJd}t j t |5|tjt|jdt|dddy#1swYyxYw)NrOz(can only perform ops with 1-d structuresr.) getattrrrPNotImplementedrr5r6rIaranger3reshape)rr8r"r:opar>r=s rtest_error_invalid_objectrbsGD! !B $ CsDk* +F ^ ## # 5C *# 69 BIIc$i ( (SY 78999s >BB(cV|\}}tj|}|gt|dz z}t}dj ddg}|j j dk(r|jddvr t}d}nt||rd }t}|tj|fD]i}tj|| 5|||dddtj |}tj|| 5|||dddky#1swYJxYw#1swYxYw) Nr|zBoperands could not be broadcast together with shapes \(3,\) \(4,\)zBoperands could not be broadcast together with shapes \(4,\) \(3,\)r!r")rrsubznumpy boolean subtract, the `\-` operator, is not supported, use the bitwise_xor, the `\^` operator, or the logical_xor function insteadr-r.)r0r1r3 ValueErrorjoinr r'r( TypeErrorr*r6rIr2rr5rrU) rr8r9r:rYerrr=r<ss rtest_error_len_mismatchrks*LD&  5 6B HD A &E C (( Q Q  C  zz#":"@"@"EJ# W  !'? @=!rxx' ]]3c *  tSM  IIdO ]]3c *  q#J       s D> DD D( r:)__neg____abs__ __invert__c|\}}tj|}|dk(r|jjdk(rd}t j t |5t||dddt j t |5t||dddt j t |5t|j|dddyt||}|jd}d|d<tj||y#1swYxYw#1swYxYw#1swYyxYw)Nrnfz0ufunc 'invert' not supported for the input typesr.T)deepr) rrUr r'rr5rhr]_datarDr0rV)rr:r"rWr=r>r?s r%test_unary_op_does_not_propagate_maskrssGD! ))D/C \djjoo4@ ]]9C 0  GC    ]]9C 0 GD"   ]]9C 0 & #GDJJ # % &  WS"  F{{{%H CF68,     & s$D$ D0D<$D-0D9<Er )Int64Int32Float32Float64ctjgd|}ttj|}t |t rt |dk(sJtjtjtjtjg|}tj|d|tj|d|tj|dtj|ztj|dtj|zt|tj}t |t rt |dk(sJtj|d|tj|d|tj|d|tjztj|d|tjzy)N)rrrrrrr) rrUdivmodrBrtupler3r0rV)r rWresexps rtest_divmod_pdnar}sK ))IU +C  C c5 !c#h!m3 3 ))RUUBEE255) 7C3q63'3q63'3q6255C<03q62553;/ bee C c5 !c#h!m3 33q63'3q63'3q63"%%<03q63;/r)% __future__rtypingrnumpyrIrpandasrpandas._testing_testingr0ALL_INT_EA_DTYPESr2arraysr3r __annotations__FLOAT_EA_DTYPESfixturezipr namerrr*r@rGrMrSrZrbrkmark parametrizersr})r as00rrs" ?A>R>R SU("((?% 0 SS3v;&&REWEW XE8288) 7 XXC:8288-Y ? @@E7 vwt ,2PA177<<2P< >*6862,*5< 9!HCD-E-0"JK0L0e T Y3QsE .E%E*