U
|
¬ý°dÅ ã @ s d dl Zd dlZd dlmZ d dlZd dlmZ ej
|
dej ej g¡ej
|
d¡dd Zej
|
dejejejejejg¡dd Zej
|
dejejg¡d d
|
Zej
|
d d d gd dgg¡d d Zej
|
je ddej
|
dddd ifddd ifddd ifddd ifdi 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(d)gd*fd d(d)dgd*fdgd+fg¡d,d- Z ej
|
d.d/d0d1d2g¡d3d4 Z!ej
|
d d!d"g¡ej
|
d5d0d1g¡d6d7 Z"ej
|
d d!d"g¡ej
|
d#d d8g¡d9d: Z#dS );é N)ÚIS64Úufuncz7ignore:invalid value encountered in sign:RuntimeWarningc C sp t jdddtjgdd}| |}t j| | t¡dd}t ||¡ t |¡}| |}t |¡}t ||¡ d S )Né é éýÿÿÿÚFloat64©Údtype)
|
ÚpdÚarrayÚnpÚnanÚastypeÚfloatÚtmÚ assert_extension_array_equalÚSeriesÚassert_series_equal©r ÚaÚresultÚexpectedÚs© r úad:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/floating/test_function.pyÚtest_ufuncs_single
|
s
|
|
r c C s® t jdddtjgdd}tjdd& | |}t j| | t¡dd}W 5 Q R X t ||¡ t |¡}tjdd& | |}t j | | t¡dd}W 5 Q R X t
|
||¡ d S )Nç ð?çÉ?ç @r r Úignore)Úinvalid) r
|
r r r Zerrstater r r r r r r r r r Útest_ufuncs_single_float s "
|
"r! c C s$ t jdddtjgdd}| ||}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 r Úarrr r r Útest_ufuncs_binary_float( s"
|
|
|
|
|
r% Úvaluesr c C s6 t j| dd}tj |¡}|jdd}t ||¡ d S )Nr r F©Úskipna)r
|
r r ÚaddÚreduceÚsumr Zassert_almost_equal)r& r$ Úresr r r r Útest_ufunc_reduce_raisesD s r- z GH 36579: fail on 32-bit system)Úreasonzpandasmethname, kwargsÚvarZddofZstdZkurtosisZskewZsemc C st t jddddddtjtjgdd}t|| }|f |}t jddddddgd d}t|| }|f |}||ksptd S )
|
Nç¹?r g333333Ó?gÙ?g à?g333333ã?r )Údatar Úfloat64)r
|
r r r ÚgetattrÚAssertionError)ZpandasmethnameÚkwargsr Z
|
pandasmethr Ús2r r r r Útest_stat_methodM s "
|
|
|
|
r7 c C sª t jdddt jgdd} | jdd}t jddt jg| jd}|j| jksLtt jdddg|d d
|
d }t ||¡ | jd d}t jddg|d d
d d
|
d }t ||¡ d S )Nr0 r r r F)Zdropnar r ÚInt64Úcount©Úindexr ÚnameTéÿÿÿÿ)
|
r
|
r ÚNAÚ value_countsÚIndexr r4 r r r )r$ r Úidxr r r r Útest_value_counts_nad s rB c C sT t jg dd} | ¡ }t jg dd}|jdks2tt jg |ddd}t ||¡ d S )Nr r r8 r9 r: )r
|
r r? r@ r r4 r r )Úserr rA r r r r Útest_value_counts_emptyq s rD c C sh t jdddt jgdd} | jdd}t jddg| d d
dd d
|
d }|jj| jksXtt ||¡ d S ) Nr0 r r r T)Ú normalizer r Z
|
proportionr: r" ) r
|
r r> r? r; r r4 r r )rC r r r r r Ú test_value_counts_with_normalizez s
|
"rF r( TFÚ min_countr# c C sP t jdddd g|d}|j| |d}| r>|dkr>|dksLtn|t jksLtd S )Nr r r" r ©r( rG r ç @)r
|
r r+ r4 r> ©r( rG r r$ r r r r Útest_floating_array_sum s
|
rK zvalues, expectedr r" rI ç c C s( t j| dd}t |¡}||ks$td S )Nr r )r
|
r r r+ r4 )r& r r$ r r r r Ú test_floating_array_numpy_sum s
|
rM Úopr+ ÚminÚmaxÚprodc C s¨ t dddgdd dgt jdd dgddd ¡}t|j| }t|tjsLtt| d
|
¡| }t jt d dg¡t jddgddd t j
|
ddgd
|
d d}t ||¡ d S )Nr Úbr r" r0 r r r )ÚAÚBÚCrS r )rT rU )r< )r; ) r
|
Z DataFramer r3 rU Ú
|
isinstancer r2 r4 Úgroupbyr@ r Zassert_frame_equal)rN Zdfr r r r r Útest_preserve_dtypes s ýÿ þrX Úmethodc C sX t jddd g|d}t||}|| d}| rF||dkr:dndksTtn|t jksTtd S )NrL r r r' rO r r )r
|
r r3 r4 r> )r( rY r r$ Úfuncr r r r Útest_floating_array_min_max® s
|
|
r[ é c C sN t jddd g|d}|j| |d}| r<|dkr<|dksJtn|t jksJtd S )Nr g @r rH r r )r
|
r rQ r4 r> rJ r r r Útest_floating_array_prodº s
|
r] )$Únumpyr ZpytestZ pandas.compatr Zpandasr
|
Zpandas._testingZ_testingr ÚmarkZ parametrizeÚabsÚsignÚfilterwarningsr ÚlogÚexpÚsinÚcosÚsqrtr! r) Úsubtractr% r- Zskipifr7 rB rD rF rK rM rX r[ r] r r r r Ú<module> sZ
|
|
|
|
|
|
|
|
ùþ
|
$ÿ
|
|
|
|
