o h<" @sdZddlZddlZddlmZddlZddlm Z ddl m Z ej ejddgedejddgedejddgedejdd gedejd d gedejddgd dejddgd dgdddZejdgdddZejdgdddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Z d(d)Z!d*d+Z"d,d-Z#d.d/Z$d0d1Z%d2d3Z&d4d5Z'd6d7Z(ejd8ej)ej*ej+gd9d:Z,d;d<Z-d=d>Z.ejd?degd@dAZ/dBdCZ0dDdEZ1ejd?ej2ej3gdFdGZ4dHdIZ5dS)JzV Additional tests for NumpyExtensionArray that aren't covered by the interface tests. N NumpyEADtype)NumpyExtensionArrayabdtypey?@TFdatetime64[ns]ztimedelta64[ns])paramscCs|jS)zk Parametrized fixture for NumPy arrays with different dtypes. This excludes string and bytes. )param)requestrW/var/www/html/env/lib/python3.10/site-packages/pandas/tests/arrays/numpy_/test_numpy.pyany_numpy_arraysrzdtype, expected) boolT)intT)uintT)floatT)complexTstrFbytesFr FobjectFvoidFcCt|}|j|us JdSN)r _is_numericrexpectedrrrtest_is_numeric'r%) r)rF)rF)rF)rFrrrrrcCr r!)r _is_booleanr#rrrtest_is_boolean;r&r(cCs"ttd}t|dksJdS)Nint64zNumpyEADtype('int64'))rnprreprrrrr test_reprOsr,cCs(td}ttd}||ksJdS)Nr))rconstruct_from_stringr*r)resultr$rrrtest_constructor_from_stringTs r/cCs t|}t|}||ksJdSr!r)any_numpy_dtyperr.rrrtest_dtype_idempotentZsr1cCs@tjtddtgdWddS1swYdS)Nz NumPy arraymatchr )pytestraises ValueErrorrrrrrtest_constructor_no_coerciones"r:cCs2tgd}tjt|dd}|j|usJdS)Nr4Tcopy)r*arraypdSeriesrvalues)ndarrayserrrr!test_series_constructor_with_copyjsrCcCsBtgd}tjt|dd}tjgddd}t||dS)Nr4float64r)?g@g@)r*r=r>r?rtmassert_series_equal)rAr.r$rrr#test_series_constructor_with_astypeqsrHcCsFtjgddd}tj|dd}ttjgddd}t||dS)Nr4r)ruint64)r*r=r_from_sequencerFassert_extension_array_equalarrr.r$rrrtest_from_sequence_dtypexsrNcCs.tddg}t|dd}t||rJdS)Nrr Tr;)r*r=rrF shares_memory)rMr.rrrtest_constructor_copys rPcCs"|}t|}|jj|jksJdSr!)rr numpy_dtypernparrrMrrrtest_constructor_with_datasrTcCspttgd}|}||jusJ|jdd}||jus!J|jdd}tjgddd}t||dS)Nr4Tr;f8r)rr*r=to_numpy_ndarrayrFassert_numpy_array_equalrLrrr test_to_numpys  rYcCs6tgd}d|jd<tgd}t||dS)Nr4 r)rZr5r6)r>r?r=rFrG)rBr$rrrtest_setitem_seriess r[cCs<|}t|dd}|d|d<|d|d<t||dS)NTr;r r)rrFrXrVrRrrr test_setitems    r\cCs\tjgddd}t|}d}tjt|d||WddS1s'wYdS)Nr4r)rz)cannot perform not_a_method with type intr2)r*r=rr7r8 TypeError_reducerMmsgrrrtest_bad_reduce_raisess  "racCsVttgd}d}tjt|d|jddWddS1s$wYdS)Nr4z/the 'keepdims' parameter is not supported .*allr2T)keepdims)rr*r=r7r8r9allr_rrr$test_validate_reduction_keyword_argss "rdcCs|gd}t|}|j}||dusJ||dus Jtj|}||dks.Jtj|}||dksr?r=maxr*maximumreduce)valsrBrMr.rrrtest_np_max_nested_tupless   rrcCsZtddd}t|}tjj|dd}t||d|jdd}t||ddS)N r6r)axis) r*arangereshaperrorprFrKrn)rawrMresaltrrrtest_np_reduce_2ds  r|ufunccCsbttgd}||}t||j}t||ttgd}|||dt||dS)NgrE)"rr)out)rr*r=rWrFrK)r}rMr.r$rrrrtest_ufunc_unarys  rcCsrttgd}t|t|d\}}t|jt|jd\}}t|}t|}t||t||dS)Nr~r5)rr*r=divmodaddrWrFrK)rMr1r2e1e2rrr test_ufuncs rcCs<ttgd}||}ttgd}t||dS)Nr4)r5rtrr*r=rFrK)xr.r$rrrtest_basic_binopsrrcCsDttjgd|d}d|d<ttjgd|d}t||dS)N)rrcrtr)rrrr)rrMr$rrrtest_setitem_object_typecode srcCsxttgd}tjtdd d|d<Wdn1swYd|d<t|dttjfs:Jt |ddS)Nr4rr2rr@) rr*r=r7r8r9 isinstancerintegertype)rMrrrtest_setitem_no_coercions  (rcCsttgd}|}|dd}t|}d|d<|ddks$J|ddks,J|ddks4Jd|d<d|d<|ddksDJdS)Nr4 rrrv)rr*r=viewasarray)rMview1view2view3rrrtest_setitem_preserves_viewss  rcCsTttjg|d}tddg}|j|dd}tttjtjg}t||dS)Nrrg?linear) interpolation) rr*r=r>Index _quantilenanrFrK)rrMidxr.r$rrrtest_quantile_empty/s rcCsVtjgdtjd}t|}|\}}t|\}}t||t|t|dS)Nr4r) r*r=rIr factorizer>rFrXrK)rMobj res_codes res_unique exp_codes exp_uniquerrrtest_factorize_unsigned:s   r)6__doc__numpyr*r7pandas.core.dtypes.dtypesrpandasr>pandas._testing_testingrF pandas.arraysrfixturer=rrrrrrmark parametrizer%r(r,r/r1r:rCrHrNrPrTrYr[r\rardrrr|absnegativepositiverrrrrrr)rIrrrrrrsn