JEf<" dZddlZddlZddlmZddlZddlm Z ddl m Z ejejddgeejddgeejddgeejdd geejd d geejddgd ejddgd gdZej*j-dgddZej*j-dgddZdZdZdZdZdZdZdZdZ dZ!dZ"dZ#d Z$d!Z%d"Z&d#Z'd$Z(ej*j-d%ejRejTejVgd&Z,d'Z-d(Z.ej*j-d)degd*Z/d+Z0d,Z1ej*j-d)ejdejfgd-Z4d.Z5y)/zV Additional tests for NumpyExtensionArray that aren't covered by the interface tests. N NumpyEADtype)NumpyExtensionArrayabdtypey?@TFdatetime64[ns]ztimedelta64[ns])paramsc|jS)zk Parametrized fixture for NumPy arrays with different dtypes. This excludes string and bytes. )param)requests ^/var/www/VTS_Report/venv/lib/python3.12/site-packages/pandas/tests/arrays/numpy_/test_numpy.pyany_numpy_arrayrs" ==zdtype, expected) boolT)intT)uintT)floatT)complexTstrFbytesFr FobjectFvoidFc:t|}|j|usJyN)r _is_numericr expecteds rtest_is_numericr''"  E    (( (r) r)rF)rF)rF)rFrrrrr c:t|}|j|usJyr#)r _is_booleanr%s rtest_is_booleanr+;r(rc`ttjd}t|dk(sJy)Nint64zNumpyEADtype('int64'))rnpr reprrs r test_reprr0Os( '* +E ;1 11 1rcxtjd}ttjd}||k(sJy)Nr-)rconstruct_from_stringr.r )resultr&s rtest_constructor_from_stringr4Ts3  / / 8FBHHW-.H X  rc>t|}t|}||k(sJyr#r)any_numpy_dtyper r3s rtest_dtype_idempotentr7Zs"  )E % F U??rc~tjtd5tgddddy#1swYyxYw)Nz NumPy arraymatchr )pytestraises ValueErrorrrrtest_constructor_no_coercionrBes- z 7'I&'''s3<ctjgd}tjt |d}|j |usJy)Nr;Tcopy)r.arraypdSeriesrvalues)ndarraysers r!test_series_constructor_with_copyrLjs8hhy!G ))'0t r? TypeError_reducerXmsgs rtest_bad_reduce_raisesrrsP ((9G ,C c "C 5C y , Cs AA&cttjgd}d}tjt |5|j ddddy#1swYyxYw)Nr;z/the 'keepdims' parameter is not supported .*allr9T)keepdims)rr.rFr>r?r@allrps r$test_validate_reduction_keyword_argsrvsK bhhy1 2C ;C z - s AA#cNgd}tj|}|j}|j|dusJ|j|dusJtj j |}||dk(sJtj j |}||dk(sJy)N))jklm)r{op)r~rxr<)rGrHrFmaxr.maximumreduce)valsrKrXr3s rtest_np_max_nested_tuplesrs D ))D/C ))C 779A   779A   ZZ  s #F SV   ZZ  s #F SV  rc.tjdjdd}t|}tjj |d}t j||d|jd}t j||dy)N r=r)axis) r.arangereshaperrrrPrVr)rawrXresalts rtest_np_reduce_2drsv ))B-  1 %C c "C **  Ca  (C##CR1 ''q'/C##CR1rufuncc6ttjgd}||}t||j}t j ||ttjgd}|||t j ||y)NgrO)"rr)out)rr.rFrerPrV)rrXr3r&rs rtest_ufunc_unaryrsr bhh'78 9C 3ZF"5#67H##FH5 bhh'9: ;C #3##C2rcttjgd}tj|tj|d\}}tj|j tj|j d\}}t|}t|}t j||t j||y)Nrr<)rr.rFdivmodaddrerPrV)rXr1r2e1e2s r test_ufuncrs bhh'78 9C YYsBFF3N +FB YYs||RVVCLL!%< =FB R B R B##B+##B+rcttjgd}||z}ttjgd}tj||y)Nr;)r<rrr.rFrPrV)xr3r&s rtest_basic_binoprsC BHHY/0A UF"288I#67H##FH5rr cttjgd|}d|d<ttjgd|}tj||y)N)rrcrtr)rrrr)r rXr&s rtest_setitem_object_typecoder sE bhheD EC CF"288O5#IJH##C2rc$ttjgd}tjt d5d|d<dddd|d<t |dttjfsJt|dy#1swYExYw)Nr;rr9rr@) rr.rFr>r?r@ isinstancerintegertype)rXs rtest_setitem_no_coercionrsv bhhy1 2C z /A CF c!fsBJJ/ 0>$s1v,> 0 s BBcttjgd}|j}|dd}tj|}d|d<|ddk(sJ|ddk(sJ|ddk(sJd|d<d|d<|ddk(sJy)Nr; rrr)rr.rFviewasarray)rXview1view2view3s rtest_setitem_preserves_viewsrs bhhy1 2C HHJE FE JJsOE CF 8q== 8q== 8q==CGE"I r7a<H##FH5rc(tjgdtj}t|}|j \}}t j|\}}t j||t j|t|y)Nr;r) r.rFrTr factorizerGrPrfrV)rXobj res_codes res_unique exp_codes exp_uniques rtest_factorize_unsignedr:si ((9BII .C c "CMMOIzLL-Iz 95##J0CJ0OPr)6__doc__numpyr.r>pandas.core.dtypes.dtypesrpandasrGpandas._testing_testingrP pandas.arraysrfixturerFrrrrrrmark parametrizer'r+r0r4r7rBrLrRrYr\rargrjrlrrrvrrabsnegativepositiverrrrrrr-rTrrrArrrs$ 2-#s6*!Qu%!Qs#!VG,$d+!Q/0!Q01    ))  )) 2  ' %-6-0 2"*7*22662;; "DE 3F 3,64.1323 ?"288RYY"78696 Qr