U
|
ý°d* ã @ sâ d dl mZ d dlZd dlZd dlZd dlm mZ d dl
|
Z d dl m Z d dlmZ dddddgZej de¡dd Zej de¡d
|
d Zej de¡d d Zej dejdddddgfeejddgddggde ddgddgg¡d ddd gfeje ddgddgg¡ddddgfejddd ddgfeeje ddgddgg¡ddd ddd gfg¡d d Z ej d e¡ej d e¡d d! Z ej de¡d"d# Z d$d% Z d&d' Z d(d) Z!d*d+ Z"e #d,¡d-d. Z$d/d0 Z%dS )1é )ÚpartialN)Úis_extension_array_dtypeÚint64ZInt64©ÚAÚBÚdtypec C st t jddgddggdd}tj|ddgddgd j| d}t |¡}tjt |¡|j|jd
|
| ¡}t ||¡ d S © Néÿÿÿÿé r ©r r r ÚaÚb©ÚcolumnsÚindex©r r )
|
ÚnpÚarrayÚpdÚ DataFrameÚastypeZpositiver r ÚtmÚassert_frame_equal©r ÚvaluesÚdfÚresultÚexpected© r úTd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/test_ufunc.pyÚtest_unary_unary s
|
ÿþr! c C sÌ t |st|tr(| j tjjdd¡ tj ddgddggdd}t
|
j |ddgd d
|
gd j |d}t |¡}t|tsztt|d kstt |¡}t||D ](\}}t
|
j ||j|jd }t ||¡ qd S )Nz8Extension / mixed with multiple outputs not implemented.©Úreasonr
|
r r r r r r r r é r )r Ú
|
isinstanceÚdictÚnodeÚ
|
add_markerÚpytestÚmarkÚxfailr r r r r ÚmodfÚtupleÚAssertionErrorÚlenÚzipr r r r ) Úrequestr r r Z result_pandasZexpected_numpyr r r r r r Útest_unary_binary s ÿÿ
|
|
r2 c C sx t jddgddggdd}tj|ddgddgd j| d}t ||¡}tjt ||¡|j|jd
|
| ¡}t ||¡ d S r )
|
r r r r r Úaddr r r r r r r r Ú test_binary_input_dispatch_binop6 s
|
ÿþr4 zfunc,arg,expectedr r$ é é é FT)Úwhereé é r
|
éþÿÿÿéýÿÿÿc C s t ddgddgg¡}t |¡}t |¡}|d kr@| ||d}n| |||d}t |¡ dd¡}t ||¡ t |¡}t ||¡ d S )Nr r$ r5 r6 )Úout) r r r r Z
|
zeros_likeZreshaper Zassert_numpy_array_equalr )ÚfuncÚargr Zarrr Zresult_inplacer r r r Útest_ufunc_passes_argsB s
|
|
|
r@ Údtype_aÚdtype_bc C sF t |s$t|ts$t |s$t|tr:| j tjjdd¡ t ddgddgd¡
|
|¡}t|trzt|trz| d¡|d < t ddgddgd
|
¡
|
|¡}t ||¡}t t ddt jgddt jgg¡t dt jdgdt jdgg¡¡}tj |d dgd dd gd }t ||¡ t ||j¡}tj ddgddggd dgd}t ||¡ d S )Nú7Extension / mixed with multiple inputs not implemented.r" r r$ r5 r6 r r ÚC)r rD r r r ç ð?©r )r r% r& r' r( r) r* r+ r r r Úpopr Ú heavisider Únanr r r )r1 rA rB Údf1Údf2r r r r r Ú test_binary_input_aligns_columnsf s6 ÿþýüÿÿ þ rL c C s, t |st|tr(| j tjjdd¡ tj ddgddgddd gd
|
|
|¡}tj ddgddgddd gd
|
|
|¡}t ||¡}t t ddgddgt jt jgg¡t ddgt jt jgddgg¡¡}tj |dd d gd d gd}t ||¡ t ||j¡}tj ddgddggd d gdd gd}t ||¡ d S )NrC r" r r$ r5 r6 r r r ©r Úcr r r rE r )r r% r& r' r( r) r* r+ r r r r rH r rI r r r )r1 r rJ rK r r r r r Ú test_binary_input_aligns_index s, ÿÿ&& þ ÿrO c C sj t dddgi¡} tjtdd t | | d ¡ W 5 Q R X tjtdd t | d | ¡ W 5 Q R X d S )Nr r r$ Ú logaddexp©Úmatch)r r r) ÚraisesÚNotImplementedErrorr rP ©r r r r Ú test_binary_frame_series_raises¢ s
|
rV c C sø t dddddgi¡} tj | ¡}t dddddgi¡}t ||¡ t ddddgdddd gd
|
¡} tj | ¡}t d dddgddddgd
|
¡}t ||¡ tjj| d d }t ||¡ tjj| dd }t d dd dgd dddgd
|
¡}t ||¡ d S )Nr r r5 r$ r6 g¹?g @ç @g @©r r rE r )Zaxis)r r r ÚmaximumÚ
|
accumulater r )r r r r r r Útest_unary_accumulate_axis¬ s r[ c C s> t dddgi¡} tjtdd tj | | ¡ W 5 Q R X d S )Nr r r$ Ú rQ )r r r) rS rT r ÚsubtractÚouterrU r r r Útest_frame_outer_disallowedÁ s r_ c C s t dddgdddgd¡} t dddgdddgd¡}t jddgd d
|
gd }t jddgd
|
d gd }t dddgd ddgd¡}t d ¡ t | | ¡}W 5 Q R X t ||¡ t | |j¡}t ||¡ t | |¡}t tj gd dddgtj gd d¡}t ||¡ t | j|¡}t dddgd ddgd¡}t ||¡ t dddgddd gd¡}t d ¡ t | |¡}W 5 Q R X t ||¡ t | |j¡}t ||¡ t tj gd dddgtj gd d¡}t | |¡}t ||¡ d}t
|
j t |d t || ¡ W 5 Q R X d S )Nr r$ r5 r6 r7 é rX ©r rN r r rM rN é é
|
é é r9 )r r rN g @ç @g @zFCannot apply ufunc <ufunc 'add'> to mixed DataFrame and Series inputs.rQ ) r r ÚSeriesr Úassert_produces_warningr r3 r r rI r) rS rT )rJ rK Ús1Ús2r r Úmsgr r r Ú#test_alignment_deprecation_enforcedÈ s< ( ÿ rl Únumbac C s ddl m} m} || | | | gdd }t dddgdd d
|
gd ¡}t dddgdd d
|
gd ¡}t dddgdd d
|
gd ¡}||||}tjt dtj¡dddgd}t ||¡ t
|
d ¡ ||||}W 5 Q R X tjddgddgddggddgd}t ||¡ d}t j t |d ||||j W 5 Q R X t
|
d ¡ |||j|j}W 5 Q R X t ||¡ d}t j t |d ||j|| W 5 Q R X d S ) Nr )Úfloat64Ú vectorizec S s | | | S ©Nr ©ÚxÚyÚzr r r Úmy_ufunc s zAtest_alignment_deprecation_many_inputs_enforced.<locals>.my_ufuncr r$ r5 r6 r7 r` rX ra )r rN )r5 r5 r r rN rF rW g (@rf g .@g "@g 2@zLoperands could not be broadcast together with shapes \(3,3\) \(3,3\) \(3,2\)rQ zLoperands could not be broadcast together with shapes \(3,2\) \(3,3\) \(3,3\))rm rn ro r r r ÚfullrI r r rh r) rS Ú
|
ValueErrorr ) rn ro ru rJ rK Zdf3r r rk r r r Ú/test_alignment_deprecation_many_inputs_enforcedù s0
|
$ ÿ ÿrx c C s dd } t | dd¡}t ddgddgg¡}|||d}tjddgdd ggtd
|
}t ||¡ t ddg¡}d }tj t
|
t |¡d |||| W 5 Q R X d S ) Nc S s | | | S rp r rq r r r Úadd3, s z2test_array_ufuncs_for_many_arguments.<locals>.add3r5 r r$ r6 r7 re r9 r zTCannot apply ufunc <ufunc 'add3 (vectorized)'> to mixed DataFrame and Series inputs.rQ ) r Z
|
frompyfuncr r Úobjectr r rg r) rS rT ÚreÚescape)ry Zufuncr r r Zserrk r r r Ú$test_array_ufuncs_for_many_arguments* s ÿr} )&Ú functoolsr r{ Únumpyr r) Z pandas.util._test_decoratorsÚutilZ_test_decoratorsÚtdZpandasr Zpandas._testingZ_testingr Zpandas.api.typesr Zdtypesr* Z parametrizer! r2 r4 r3 r Úpowerr] Únegativer@ rL rO rV r[ r_ rl Z
|
skip_if_norx r} r r r r Ú<module> sZ ý
|
|
|
|
ý$
|
ý÷þ
|
|
|
1
|
0
|
