U
|
¬ý°d}" ã @ s d Z ddlZddlZddlmZ ddlZddlm Z
|
ddl m Z ej ejddgedejddgedejddgedejdd gedejd
|
d gedejddgd dejddgd dgddd Zej dddddddddddg
|
¡d d Zej ddd d d d!dddddg
|
¡d"d# Zd$d% Zd&d' Zd(d) Zd*d+ Z d,d- Z d.d/ Z d0d1 Z d2d3 Z d4d5 Z!d6d7 Z"d8d9 Z#d:d; Z$d<d= Z%d>d? Z&d@dA Z'dBdC Z(ej dDej)ej*ej+g¡dEdF Z,dGdH Z-dIdJ Z.ej dKdeg¡dLdM Z/dNdO Z0dPdQ Z1ej dKej2ej3g¡dRdS Z4dTdU Z5dS )VzN
|
Additional tests for PandasArray that aren't covered by
|
the interface tests.
|
é N)Ú PandasDtype)Ú PandasArrayÚaÚb©Údtypeé y ð? @TFúdatetime64[ns]ztimedelta64[ns])Úparamsc C s | j S )zk
|
Parametrized fixture for NumPy arrays with different dtypes.
|
|
This excludes string and bytes.
|
)Úparam)Úrequest© r ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/numpy_/test_numpy.pyÚany_numpy_array s r zdtype, expected)ÚboolT)ÚintT)ÚuintT)ÚfloatT)ÚcomplexT)ÚstrF)ÚbytesF)r F)ÚobjectF)ZvoidFc C s t | } | j|kstd S ©N)r Z _is_numericÚAssertionError©r Úexpectedr r r Útest_is_numeric' s r )r F)r F)r F)r Fc C s t | } | j|kstd S r )r Z _is_booleanr r r r r Útest_is_boolean; s r c C s" t t d¡} t| dks td S )NÚint64zPandasDtype('int64'))r Únpr Úreprr r r r r Ú test_reprO s r! c C s( t d¡} t t d¡}| |ks$td S )Nr )r Zconstruct_from_stringr r r )Úresultr r r r Ú test_constructor_from_stringT s
|
r# c C s t | }t |}||ks td S r )r r )Zany_numpy_dtyper r" r r r Útest_dtype_univalentZ s r$ c C s, t jtdd tdddg W 5 Q R X d S )Nz NumPy array©Úmatchr é é )ÚpytestÚraisesÚ
|
ValueErrorr r r r r Ú test_constructor_no_coercione s r, c C s4 t dddg¡} tjt| dd}|j| k s0td S )Nr r' r( T©Úcopy)r ÚarrayÚpdÚSeriesr Úvaluesr )ÚndarrayÚserr r r Ú!test_series_constructor_with_copyj s r5 c C sF t dddg¡} tjt| dd}tjdddgdd}t ||¡ d S ) Nr r' r( Úfloat64r ç ð?g @g @)r r/ r0 r1 r ÚtmÚassert_series_equal)r3 r" r r r r Ú#test_series_constructor_with_astypeq s r: c C sJ t jdddgdd} tj| dd}tt jdddgdd}t ||¡ d S )Nr r' r( r r Úuint64)r r/ r Z_from_sequencer8 Ú assert_extension_array_equal©Úarrr" r r r r Útest_from_sequence_dtypex s r? c C s. t ddg¡} t| dd}t || ¡r*td S )Nr r Tr- )r r/ r r8 Z shares_memoryr )r> r" r r r Útest_constructor_copy s r@ c C s" | }t |}|jj|jks td S r )r r Z numpy_dtyper ©r Znparrr> r r r Útest_constructor_with_data s rB c C st t t dddg¡} | ¡ }|| jks*t| jdd}|| jk sDt| jdd}tjdddgdd}t ||¡ d S )Nr r' r( Tr- Zf8r )r r r/ Úto_numpyÚ_ndarrayr r8 Úassert_numpy_array_equalr= r r r Ú test_to_numpy s rF c C s: t dddg¡} d| jd< t dddg¡}t | |¡ d S )Nr r' r( é
|
r )r0 r1 r/ r8 r9 )r4 r r r r Útest_setitem_series¡ s
|
rH c C s< | }t |dd}|d |d< |d |d< t | ¡ |¡ d S )NTr- r r )r r8 rE rC rA r r r Ú test_setitem¨ s
|
rI c C sH t jdddgdd} t| } d}tjt|d | |¡ W 5 Q R X d S )Nr r' r( r r z)cannot perform not_a_method with type intr% )r r/ r r) r* Ú TypeErrorZ_reduce©r> Úmsgr r r Útest_bad_reduce_raises¶ s
|
rM c C sB t t dddg¡} d}tjt|d | jdd W 5 Q R X d S )Nr r' r( z/the 'keepdims' parameter is not supported .*allr% T)Zkeepdims)r r r/ r) r* r+ ÚallrK r r r Ú$test_validate_reduction_keyword_args¾ s rO c C s~ dddg} t | ¡}|j}| ¡ |d ks.t| ¡ |d ksBttj |¡}||d ks^ttj |¡}||d ksztd S )N)©ÚjÚk©ÚlÚm)rS ©ÚoÚp)rV rP r' )r0 r1 r/ Úmaxr r ÚmaximumÚreduce)Úvalsr4 r> r" r r r Útest_np_max_nested_tuplesÅ s ý
|
r] c C sZ t d¡ dd¡} t| }t jj|dd}t ||d ¡ |jdd}t ||d ¡ d S )Né é r( r )Zaxiséÿÿÿÿ) r ZarangeZreshaper rZ r[ r8 r< rY )Úrawr> ÚresZaltr r r Útest_np_reduce_2dÚ s rc Úufuncc C sf t t dddg¡}| |}t | |j}t ||¡ t t dddg¡}| ||d t ||¡ d S )Nç ð¿ç r7 g "À)Úout)r r r/ rD r8 r< )rd r> r" r rg r r r Útest_ufunc_unaryé s rh c C st t t dddg¡} t | t | d¡¡\}}t | jt | jd¡¡\}}t |}t |}t ||¡ t ||¡ d S )Nre rf r7 r' )r r r/ ÚdivmodÚaddrD r8 r< )r> Zr1Zr2Úe1Úe2r r r Ú
|
test_ufuncö s rm c C s@ t t dddg¡} | | }t t dddg¡}t ||¡ d S )Nr r' r( r_ é ©r r r/ r8 r< )Úxr" r r r r Útest_basic_binop s rq r c C sH t tjdddg| d}d|d< t tjdddg| d}t ||¡ d S )Nr r Úcr Útr ro )r r> r r r r Ú test_setitem_object_typecode
|
s rt c C sf t t dddg¡} tjtdd d| d< W 5 Q R X d| d< t| d ttjfsbt t
|
| d d S ) Nr r' r( r r% r r ç @) r r r/ r) r* r+ Ú
|
isinstancer Úintegerr Útype)r> r r r Útest_setitem_no_coercion s
|
ry c C s t t dddg¡} | ¡ }| d d
}t | ¡}d| d< |d dksJt|d dksZt|d dksjtd| d< d|d< | d dkstd S ) Nr r' r( é r ru r` é )r r r/ ÚviewZasarrayr )r> Zview1Zview2Zview3r r r Ú test_setitem_preserves_views s
|
r} c C sT t tjg | d}t ddg¡}|j|dd}t t tjtjg¡}t ||¡ d S )Nr rf g à?Zlinear)Ú interpolation) r r r/ r0 ZIndexZ _quantileÚnanr8 r< )r r> Úidxr" r r r r Útest_quantile_empty/ s
|
r c C sX t jdddgt jd} t| }| ¡ \}}t | ¡\}}t ||¡ t |t|¡ d S )Nr r' r( r ) r r/ r; r Z factorizer0 r8 rE r< )r> ÚobjZ res_codesZ
|
res_uniqueZ exp_codesZ
|
exp_uniquer r r Útest_factorize_unsigned: s r )6Ú__doc__Únumpyr r) Zpandas.core.dtypes.dtypesr Zpandasr0 Zpandas._testingZ_testingr8 Z pandas.arraysr Zfixturer/ r r r r r r ÚmarkZ parametrizer r r! r# r$ r, r5 r: r? r@ rB rF rH rI rM rO r] rc ÚabsÚnegativeZpositiverh rm rq rt ry r} r r; r r r r r r Ú<module> s ùÿ
|
öþ
|
öþ
|
|
|
|
|
