OwgUddlmZddlmZddlZddlZddlZddl m Z e jDcgc]}ejgd|c}ZdgeezZded<ee j$Dcgc]}ejgd |c}z Zed d gz Zeejgd d gz Zed gz Zej&eeeeDcgc]}|j*j,c}dZdZdZdZdZdZdZdZdZdZ ejBjEdgddZ#ycc}wcc}wcc}w)) annotations)AnyN)Ndtyperz list[Any]scalars)g?皙?g333333?Nr )TFTNbooleanF)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/horilla/myenv/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_arraymsgresultexpecteds r"test_array_scalar_like_equivalencer>2sLD&  5 6Bt-.88VHs4y0 CL4::??623 > "4)A BAC2#> !4  !2#> '4& ' 'f%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+)r/r0rrNAr1r2r _maskcopyr)rr4r5assert_numpy_array_equalr6) rr7r!r9r8r:maskr;r<r=s r test_array_NArEGs 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+)r/r0rnpr1r2r numpy_dtyperr)rr4r5r6) rr7r8r9 numpy_arraypd_arrayr;r<r=s rtest_numpy_array_equivalencerKcsLD&  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-) r/r0rr DataFramer)rr4r5assert_frame_equal)rr7r8r9dfr;r<r=s r test_framerQ|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+)r/r0rrSeriesrGr1r2r rHr)rr4r5assert_series_equal) rr7r8r9serothersotherr;r<r=s r test_seriesrXs!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)NrMz(can only perform ops with 1-d structuresr-) getattrrrNNotImplementedrr4r5rGaranger2reshape)rr7r!r9opar<r;s rtest_error_invalid_objectr`sGD! !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-)r/r0r2 ValueErrorjoinr r&r' TypeErrorr)r5rGr1rr4rrS)rr7r8r9rWerrr;ss rtest_error_len_mismatchris*LD&  5 6B HD A &E C (( Q Q  C  zz#":"@"@"EJ# W  !'? @=!%) ]]3c *  tUO  IIdO ]]3c *  q%L       s D> DD D( r9)__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)Nrlfz0ufunc 'invert' not supported for the input typesr-T)deepr) rrSr r&rr4rfr[_datarBr/rT)rr9r!rUr;r<r=s r%test_unary_op_does_not_propagate_maskrqsGD! ))D/C \djjoo4@ ]]9C 0  GC    ]]9C 0 GD"   ]]9C 0 & #GDJJ # % &  WS"  F{{{%H CF68,     & s$D$ D0D<$D-0D9<E)$ __future__rtypingrnumpyrGrpandasrpandas._testing_testingr/ALL_INT_EA_DTYPESr1arraysr2r __annotations__FLOAT_EA_DTYPESfixturezipr namerrr)r>rErKrQrXr`rimark parametrizerq)r as00rrsT" ?A>R>R SU("((?% 0 SS3v;&&REWEW XE8288) 7 XXC:8288-Y ? @@E7s67+1O1!'',,1OPQ< >*6862,*5< 9!HCD-E-o T Y2PsD:.D?E