U
|
ý°d" ã @ s d dl mZ d dlZd dlZd dlZd dlmZ d dl
|
m Z d dl m Z G dd de ZG dd deZG d d
|
d
|
eZG d d d eZdS ) é )Ú annotationsN)Úops)ÚBaseExtensionTestsc @ s\ e Zd ZddddZefdddddZd d
|
Zefdddd d Zefdd ddZ dS )Ú BaseOpsUtilÚstr)Úop_namec C s
|
t |¡S ©N)ÚtmÚget_op_from_name)Úselfr © r úVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/extension/base/ops.pyr
|
s z BaseOpsUtil.get_op_from_nameú pd.Series)Úserr c C s | |¡}| |||||¡ d S r )r
|
Ú _check_op)r r r ÚotherÚexcÚopr r r Ú check_opname s
|
zBaseOpsUtil.check_opnamec C sN t |tjr>t|jdkr t|jd d
df ||¡ ¡ }n | ||¡}|S )Né r ) Ú
|
isinstanceÚpdÚ DataFrameÚlenÚcolumnsÚNotImplementedErrorÚilocÚcombineÚto_frame)r Úobjr r Úexpectedr r r Ú_combine s zBaseOpsUtil._combinec C sd |d kr@|||}| |||¡}t|t|s2t| ||¡ n t |¡ ||| W 5 Q R X d S r )r! r ÚtypeÚAssertionErrorZ assert_equalÚpytestÚraises)r r r r r r Úresultr r r r r s
|
zBaseOpsUtil._check_op©r c C s |d kr^|||\}}|t kr2|| || }}n|| || }}| ||¡ | ||¡ n t |¡ t || W 5 Q R X d S r )ÚdivmodÚassert_series_equalr$ r% ) r r r r r Z
|
result_divZ
|
result_modZ expected_divZ expected_modr r r Ú_check_divmod_op* s z BaseOpsUtil._check_divmod_opN)
|
Ú__name__Ú
|
__module__Ú __qualname__r
|
Ú Exceptionr r! r r r* r r r r r s
|
ÿ r c @ s e Zd ZU dZeZded< eZded< eZded< eZ ded< dd Z
|
d d
|
Z d d Z d d Z dd Zdd Zej dejejg¡dd ZdS )ÚBaseArithmeticOpsTestsa?
|
Various Series and DataFrame arithmetic ops methods.
|
|
Subclasses supporting various ops should set the class variables
|
to indicate that they support ops of that kind
|
|
* series_scalar_exc = TypeError
|
* frame_scalar_exc = TypeError
|
* series_array_exc = TypeError
|
* divmod_exc = TypeError
|
ztype[Exception] | NoneÚseries_scalar_excÚframe_scalar_excÚseries_array_excÚ
|
divmod_excc C s, |}t |¡}| j|||jd | jd d S ©Nr ©r )r ÚSeriesr r r0 ©r ÚdataÚall_arithmetic_operatorsr r r r r Ú test_arith_series_with_scalarK s
|
z4BaseArithmeticOpsTests.test_arith_series_with_scalarc C s. |}t d|i¡}| j|||d | jd d S )NÚAr r5 )r r r r1 )r r8 r9 r Zdfr r r Ú test_arith_frame_with_scalarQ s z3BaseArithmeticOpsTests.test_arith_frame_with_scalarc C s< |}t |¡}| j||t |jd gt| ¡| jd d S r4 )r r6 r r r r2 r7 r r r Ú test_arith_series_with_arrayW s
|
ÿz3BaseArithmeticOpsTests.test_arith_series_with_arrayc C s8 t |¡}| j|td| jd | jdtj|| jd d S )Nr r5 )r r6 r* r( r3 r Úrdivmod)r r8 r r r r Ú test_divmod_ s
|
z"BaseArithmeticOpsTests.test_divmodc C sJ t |¡}| |t|¡ |}| |tj|¡ t |¡}| |tj|¡ d S r )r r6 r* r( r r> )r r8 Z data_for_twosr r r r r Útest_divmod_series_arrayd s
|
|
z/BaseArithmeticOpsTests.test_divmod_series_arrayc C s0 t |¡}|| }t || ¡}| ||¡ d S r ©r r6 r) ©r r8 r r& r r r r Ú$test_add_series_with_extension_arrayn s
|
z;BaseArithmeticOpsTests.test_add_series_with_extension_arrayÚboxc C sb t |¡}|t jkr | ¡ }t|dsH|j tjj t
|
|j dd¡ | |¡}|t ks^td S )NÚ__add__z does not implement add)Úreason)r r6 r r ÚhasattrÚnodeZ
|
add_markerr$ ÚmarkZxfailr" r+ rE ÚNotImplementedr# )r Úrequestr8 rD r r& r r r Ú6test_direct_arith_with_ndframe_returns_not_implementedt s
|
|
|
ÿÿ
|
zMBaseArithmeticOpsTests.test_direct_arith_with_ndframe_returns_not_implementedN)r+ r, r- Ú__doc__Ú TypeErrorr0 Ú__annotations__r1 r2 r3 r: r< r= r? r@ rC r$ rI Ú parametrizer r6 r rL r r r r r/ 9 s
|
|
r/ c @ sL e Zd ZdZddddZdd Zdd Zej d
|
e
|
j e
|
j g¡d d Z d S )ÚBaseComparisonOpsTestsz4Various Series and DataFrame comparison ops methods.r r' c
|
C s´ |j dkr.|||}| ||¡}| ||¡ nd }z|||}W n& tk
|
rf } z|}W 5 d }~X Y nX |d kr| ||¡}| ||¡ n&t t|¡ | ||¡ W 5 Q R X d S )N)ÚeqÚne)r+ r r) r. r$ r% r" ) r r r8 r r r& r r Úerrr r r Ú_compare_other s
|
|
z%BaseComparisonOpsTests._compare_otherc C s t |¡}| |||d¡ d S ©Nr )r r6 rU )r r8 Ú comparison_opr r r r Útest_compare_scalar s
|
z*BaseComparisonOpsTests.test_compare_scalarc C s6 t |¡}t |d gt| ¡}| ||||¡ d S rV )r r6 r rU )r r8 rW r r r r r Útest_compare_array¤ s
|
z)BaseComparisonOpsTests.test_compare_arrayrD c C s t |¡}|t jkr | ¡ }t|dr>| |¡}|tksTtnt t
|
|j d¡ t|drv| |¡}|tkstnt t
|
|j d¡ d S )NÚ__eq__z does not implement __eq__Ú__ne__z does not implement __ne__) r r6 r r rG rZ rJ r# r$ Úskipr" r+ r[ )r r8 rD r r& r r r rL © s
|
|
|
|
|
|
zMBaseComparisonOpsTests.test_direct_arith_with_ndframe_returns_not_implementedN)r+ r, r- rM rU rX rY r$ rI rP r r6 r rL r r r r rQ s rQ c @ s6 e Zd Zdd Zej dejej ej
|
g¡dd Z dS )ÚBaseUnaryOpsTestsc C s4 t j|dd}| }t j| dd}| ||¡ d S )NÚname)r^ rA rB r r r Ú test_invert¿ s z BaseUnaryOpsTests.test_invertÚufuncc C s t jdt jdt jdi| }d }zt|| }W nL tk
|
rz } z.|}t t|t f¡ || W 5 Q R X W 5 d }~X Y nX ||}|
|
||¡ d S )NÚ__pos__Ú__neg__Ú__abs__) ÚnpÚpositiveÚnegativeÚabsÚgetattrr. r$ r% r" rN Z assert_extension_array_equal)r r8 r` Úattrr r& rT Zaltr r r Ú#test_unary_ufunc_dunder_equivalenceÅ s ÿ$z5BaseUnaryOpsTests.test_unary_ufunc_dunder_equivalenceN) r+ r, r- r_ r$ rI rP rd re rf rg rj r r r r r] ¾ s r] )Ú
|
__future__r Únumpyrd r$ Zpandasr Zpandas._testingZ_testingr Z pandas.corer Z pandas.tests.extension.base.baser r r/ rQ r] r r r r Ú<module> s -N7
|
