E g<" dZddlZddlZddlmZddlZddlm Z ddl m Z ej ejddgeejddgeejddgeejdd geejd d geejddgd ejddgd gdZejdgddZejdgddZdZdZdZdZdZdZdZdZ dZ!dZ"dZ#d Z$d!Z%d"Z&d#Z'd$Z(ejd%ej)ej*ej+gd&Z,d'Z-d(Z.ejd)degd*Z/d+Z0d,Z1ejd)ej2ej3gd-Z4d.Z5dS)/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 Z/var/www/sysmax/venv/lib/python3.11/site-packages/pandas/tests/arrays/numpy_/test_numpy.pyany_numpy_arrayrs " =zdtype, expected) boolT)intT)uintT)floatT)complexTstrFbytesFr FobjectFvoidFc:t|}|j|usJdSN)r _is_numericr expecteds rtest_is_numericr''+   E   ( ( ( ( ( (r) r)rF)rF)rF)rFrrrrr c:t|}|j|usJdSr#)r _is_booleanr%s rtest_is_booleanr+;r(rcrttjd}t|dksJdS)Nint64zNumpyEADtype('int64'))rnpr reprrs r test_reprr0Os7 '** + +E ;;1 1 1 1 1 1 1rctjd}ttjd}||ksJdS)Nr-)rconstruct_from_stringr.r )resultr&s rtest_constructor_from_stringr4Ts@  / 8 8FBHW--..H X      rcRt|}t|}||ksJdSr#r)any_numpy_dtyper r3s rtest_dtype_idempotentr7Zs-  ) )E % F U??????rctjtd5tgdddddS#1swxYwYdS)Nz NumPy arraymatchr )pytestraises ValueErrorrrrtest_constructor_no_coercionrBes z 7 7 7''III&&&''''''''''''''''''s ;??ctjgd}tjt |d}|j|usJdS)Nr;Tcopy)r.arraypdSeriesrvalues)ndarraysers r!test_series_constructor_with_copyrLjsKhyyy!!G )'00t < < r? TypeError_reducerXmsgs rtest_bad_reduce_raisesrrs (999G , , ,C c " "C 5C y , , , CsA((A,/A,cttjgd}d}tjt |5|dddddS#1swxYwYdS)Nr;z/the 'keepdims' parameter is not supported .*allr9T)keepdims)rr.rFr>r?r@allrps r$test_validate_reduction_keyword_argsrvs bhyyy11 2 2C ;C z - - - sA%%A),A)cpgd}tj|}|j}||dusJ||dusJtj|}||dksJtj|}||dksJdS)N))jklm)r{op)r~rxr<)rGrHrFmaxr.maximumreduce)valsrKrXr3s rtest_np_max_nested_tuplesrs   D )D//C )C 7799A     7799A     Z  s # #F SV     Z  s # #F SV      rcNtjddd}t|}tj|d}t j||d|d}t j||ddS)N r=r)axis) r.arangereshaperrrrPrVr)rawrXresalts rtest_np_reduce_2drs )B--  1 % %C c " "C *  Ca  ( (C#CR111 ''q'//C#CR11111rufunccPttjgd}||}t||j}t j||ttjgd}|||t j||dS)NgrO)"rr)out)rr.rFrerPrV)rrXr3r&rs rtest_ufunc_unaryrs bh'7'7'788 9 9C U3ZZF"55#6#677H#FH555 bh'9'9'9:: ; ;C E#3#C22222rcttjgd}tj|tj|d\}}tj|jtj|jd\}}t|}t|}t j||t j||dS)Nrr<)rr.rFdivmodaddrerPrV)rXr1r2e1e2s r test_ufuncrs bh'7'7'788 9 9C YsBF3NN + +FB Ys|RVCL!%<%< = =FB R B R B#B+++#B+++++rcttjgd}||z}ttjgd}tj||dS)Nr;)r<rrr.rFrPrV)xr3r&s rtest_basic_binoprs\ BHYYY//00A UF"28III#6#677H#FH55555rr cttjgd|}d|d<ttjgd|}tj||dS)N)rrcrtr)rrrr)r rXr&s rtest_setitem_object_typecoder sd bheDDD E EC CF"28OOO5#I#I#IJJH#C22222rcNttjgd}tjt d5d|d<dddn #1swxYwYd|d<t |dttjfsJt|ddS)Nr;rr9rr@) rr.rFr>r?r@ isinstancerintegertype)rXs rtest_setitem_no_coercionrs bhyyy11 2 2C z / / /A CF c!fsBJ/ 0 0>>$s1v,,>>>>>sAAAc>ttjgd}|}|dd}tj|}d|d<|ddksJ|ddksJ|ddksJd|d<d|d<|ddksJdS)Nr; rrr)rr.rFviewasarray)rXview1view2view3s rtest_setitem_preserves_viewsrs bhyyy11 2 2C HHJJE FE JsOOE CF 8q==== 8q==== 8q====CGE"I r7a<<<<<ttjg|}tjddg}||d}ttjtjtjg}tj||dS)Nrrg?linear) interpolation) rr.rFrGIndex _quantilenanrPrV)r rXidxr3r&s rtest_quantile_emptyr/s bhr777 8 8C (C:  C ]]3h] 7 7F"28RVRV,<#=#=>>H#FH55555rc2tjgdtj}t|}|\}}t j|\}}t j||t j|t|dS)Nr;r) r.rFrTr factorizerGrPrfrV)rXobj res_codes res_unique exp_codes exp_uniques rtest_factorize_unsignedr:s (999BI . . .C c " "CMMOOIzL--Iz 9555#J0CJ0O0OPPPPPr)6__doc__numpyr.r>pandas.core.dtypes.dtypesrpandasrGpandas._testing_testingrP pandas.arraysrfixturerFrrrrrrmark parametrizer'r+r0r4r7rBrLrRrYr\rargrjrlrrrvrrabsnegativepositiverrrrrrr-rTrrrArrrs 222222------#s6***!Qu%%%!Qs###!VG,,,$d+++!Q/000!Q0111         )))    ))) 222  ''' %%%---666---000 2 2 2"***777*222262; "DEE 3 3FE 3,,,6664.1133213 ? ? ?"28RY"78866986 Q Q Q Q Qr