U
|
¬ý°d® ã @ s d dl Zd dlZd dlZd dlmZ d dlm Z ej
|
dej ej g¡ej
|
d¡dd Zej
|
dejejejejejg¡dd Zej
|
dejejg¡d d
|
Zd d Zej
|
d d dgd dgg¡dd Zej
|
dddd ifdddifddd ifdddifdi fdi fdi fg¡dd Zdd Z d d Z d d Z ej
|
d d!d"g¡ej
|
d#d d$g¡d%d& Z ej
|
d d!d"g¡ej
|
d'd(d)g¡d*d+ Z ej
|
d d!d"g¡ej
|
d#d d,g¡d-d. Z!ej
|
d/dd0d1gd2fdd0d1dgd2fdgd fg¡d3d4 Z"ej
|
d5d6d7d(d)g¡d8d9 Z#dS ):é N)Ú FloatingArrayÚufuncz7ignore:invalid value encountered in sign:RuntimeWarningc C s t dddtjg¡}| |}t j| | t¡dd}t ||¡ t |¡}| |}t t j| | t¡dd¡}t ||¡ d S )Né é éýÿÿÿÚInt64©Údtype)
|
ÚpdÚarrayÚnpÚnanÚastypeÚfloatÚtmÚ assert_extension_array_equalÚSeriesÚassert_series_equal©r ÚaÚresultÚexpectedÚs© r ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/integer/test_function.pyÚtest_ufuncs_single_int s
|
r c C s t dddtjg¡}tjdd& | |}t| | t¡|jd}W 5 Q R X t
|
||¡ t |¡}tjdd | |}W 5 Q R X t |¡}t ||¡ d S )Nr r r Úignore)Úinvalid)Úmask) r
|
r r r Zerrstater r r Z_maskr r r r r r r r Útest_ufuncs_single_float s "
|
|
r c C s t dddtjg¡}| ||}t j| | t¡| t¡dd}t ||¡ t ddddg¡}| ||}t j| | t¡|dd}t ||¡ | ||}t j| || t¡dd}t ||¡ | |d}t j| | t¡ddd}t ||¡ | d|}t j| d| t¡dd}t ||¡ d S )Nr r r r r é é )r
|
r r r r r r r )r r r r Úarrr r r Útest_ufuncs_binary_int' s"
|
|
|
|
|
r# c C s t ddtjg¡} t | ¡}t | jtjdd¡}t |d ¡t |d ¡f}t|ts\tt |dksltt
|
||D ]\}}t ||¡ qvd S )Nr r r )Zna_valuer r ) r
|
r r r ÚmodfZto_numpyÚ
|
isinstanceÚtupleÚAssertionErrorÚlenÚzipr r )r r r ÚxÚyr r r Útest_ufunc_binary_outputC s
|
r, Úvaluesr c C s2 t | ¡}tj |¡}|jdd}t ||¡ d S )NF©Úskipna)r
|
r r ÚaddÚreduceÚsumr Zassert_almost_equal)r- r" Úresr r r r Útest_ufunc_reduce_raisesP s
|
r4 zpandasmethname, kwargsÚvarZddofZstdZkurtosisZskewZsemc C st t jddddddtjtjgdd}t|| }|f |}t jddddddgdd}t|| }|f |}||ksptd S ) Nr r r r! é é r )Údatar )r
|
r r r Úgetattrr' )ZpandasmethnameÚkwargsr Z
|
pandasmethr Ús2r r r r Útest_stat_methodY s "
|
|
|
|
r< c C s¸ t jdddt jgdd} | jdd}t jddt jgdd}|jdksHtt jdddg|ddd}t ||¡ | jd d}t jddg| d d
ddd}|j
|
j| jks¨tt ||¡ d S )
|
Nr r r r F)ZdropnaÚcount©Úindexr ÚnameT) r
|
r ÚNAÚ value_countsÚIndexr r' r r r r? )r" r Zex_indexr r r r Útest_value_counts_nao s rD c C sX t jg dd} | ¡ }t jg | jd}|j| jks6tt jg |ddd}t ||¡ d S )Nr r r= r> )r
|
r rB rC r r' r r )Úserr Úidxr r r r Útest_value_counts_empty} s rG c C sh t jdddt jgdd} | jdd}t jddg| d d
ddd d
|
}|jj| jksXtt ||¡ d S ) Nr r r r T)Ú normalizeZFloat64Z
|
proportionr> r ) r
|
r rA rB r? r r' r r )rE r r r r r Ú test_value_counts_with_normalize s
|
"rI r/ TFÚ min_countr! c C sT |}t jdddd g|d}|j| |d}| rB|dkrB|dksPtn|t jksPtd S )Nr r r r ©r/ rJ r r7 )r
|
r r2 r' rA ©r/ rJ Úany_int_ea_dtyper r" r r r r Útest_integer_array_sum s rN ÚmethodÚminÚmaxc C s\ |}t jddd g|d}t||}|| d}| rJ||dkr>dndksXtn|t jksXtd S )Nr r r r. rP )r
|
r r9 r' rA )r/ rO rM r r" Úfuncr r r r Útest_integer_array_min_max s
|
|
rS é c C sR |}t jddd g|d}|j| |d}| r@|dkr@|dksNtn|t jksNtd S )Nr r r rK r )r
|
r Úprodr' rA rL r r r Útest_integer_array_prod© s rV zvalues, expectedr r r7 c C s( t j| dd}t |¡}||ks$td S )Nr r )r
|
r r r2 r' )r- r r" r r r r Ú test_integer_array_numpy_sumµ s
|
rW Úopr2 rU c C s< t dt jddgddi¡}| ¡ }t|d tjs8td S )Nr r r r r )r
|
Z DataFramer rQ r% r Úint64r' )rX Zdfr r r r Útest_dataframe_reductions¾ s rZ )$Únumpyr ZpytestZpandasr
|
Zpandas._testingZ_testingr Zpandas.core.arraysr ÚmarkZ parametrizeÚabsÚsignÚfilterwarningsr ÚlogÚexpÚsinÚcosÚsqrtr r0 Úsubtractr# r, r4 r< rD rG rI rN rS rV rW rZ r r r r Ú<module> sZ
|
|
|
|
|
|
|
|
ùþ
|
|
|
|
|
$ÿ
|
|
