U
|
ý°d
P ã @ s d dl Z d dlZd dlZd dlZd dlmZ d dl m
|
Z
|
m Z ej ddgddd Z ej dd gdd
|
d ZG d d d Zej de je jg¡dd Zdd Zdd Zej dejejg¡ej de
|
d d ddge
|
ddddgg¡dd Zej de
|
d d d ge d dd g¡fe
|
d d d gdd e d dd g¡fe
|
d d d gdd e d dd g¡fe
|
d d d gdd e d dd g¡fe
|
d d d gdd e d dd g¡fg¡ej dejejg¡d d Z d d! Z d"d# Z ej d$e eje
|
g¡d%d& Z ej dd'd(d)d*d+d,g¡ej d-ej!d.g¡d/d0 Z"dS )1é N)Ú SparseArrayÚ SparseDtypeÚintegerÚblock)Úparamsc C s | j S )z!kind kwarg to pass to SparseArray©Úparam©Úrequest© r úbd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/sparse/test_arithmetics.pyÚkind s r TFc C s | j S )z}
|
Fixture returning True or False, determining whether to operate
|
op(sparse, dense) instead of op(sparse, sparse)
|
r r r r r Úmix s r c @ s e Zd Zdd ZedddZdd Zdd Zd
|
d Ze j
|
d d ddg¡e j
|
ddd dg¡dd Z dd Z dd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Ze j
|
dd'd(ejg¡d)d* Ze j
|
dd'd(ejg¡d+d, Zd-d. Zd/d0 Zd1d2 Z dS )3ÚTestSparseArrayArithmeticsc C s t ||¡ d S ©N)ÚtmÚassert_numpy_array_equal)ÚselfÚaÚbr r r Ú_assert s z"TestSparseArrayArithmetics._assert)r c C s t |tjr |t |¡|j}n$t |tjr@||t |¡j}nttjddd( |rh||| ¡ }n||| ¡ }W 5 Q R X | ||¡ d S )NÚignore)ÚinvalidÚdivide)
|
Ú
|
isinstanceÚnpÚndarrayÚpdZSeriesÚvaluesÚNotImplementedErrorÚerrstateÚto_denser ) r r r Úa_denseÚb_denser ÚopÚexpectedÚresultr r r Ú_check_numeric_ops" s z-TestSparseArrayArithmetics._check_numeric_opsc C sD t |tstt |jts t|jjtjks0tt |jt s@td S r )
|
r r ÚAssertionErrorÚdtyper Úsubtyper Úbool_Ú
|
fill_valueÚbool)r Úresr r r Ú_check_bool_result4 s z-TestSparseArrayArithmetics._check_bool_resultc C sæ t jddÎ | ||k¡ | ||k ¡ ||k¡ | ||k¡ | ||k ¡ ||k¡ | ||k¡ | ||k ¡ ||k¡ | ||k¡ | ||k ¡ ||k¡ | ||k¡ | ||k ¡ ||k¡ | ||k ¡ | ||k ¡ ||k ¡ | ||k¡ | ||k ¡ ||k¡ | ||k¡ | ||k ¡ ||k¡ | ||k¡ | ||k ¡ ||k¡ | ||k¡ | ||k ¡ ||k¡ | ||k¡ | ||k ¡ ||k¡ | ||k ¡ | ||k ¡ ||k ¡ W 5 Q R X d S )Nr )r )r r r/ r r! ©r r r r" r# r r r Ú_check_comparison_ops: s2 z0TestSparseArrayArithmetics._check_comparison_opsc C s | ||@ ¡ | ||@ ¡ ||@ ¡ | ||B ¡ | ||B ¡ ||B ¡ | ||@ ¡ | ||@ ¡ ||@ ¡ | ||B ¡ | ||B ¡ ||B ¡ d S r )r/ r r! r0 r r r Ú_check_logical_opse s z-TestSparseArrayArithmetics._check_logical_opsÚscalarr é é r, Né c
|
C sN |}t t jdddt jddddt jg
|
¡}t|||d} | | |||||¡ d S )Nr4 r6 r ©r r, ©r ÚarrayÚnanr r' )
|
r r r Úall_arithmetic_functionsr, r3 r
|
r$ r r r r r Útest_float_scalars s $z,TestSparseArrayArithmetics.test_float_scalarc C sà t t jdddt jddddt jg
|
¡}t||d}| |d|d¡ | |d|d¡ | |d|d¡ t||dd}| |d|d¡ | |d|d¡ | |d|d¡ t||dd}| |d|d¡ | |d|d¡ | |d|d¡ d S )Nr4 r6 r ©r r5 r7 ©r r9 r: r r1 )r r r r r r r Ú test_float_scalar_comparison} s $ z7TestSparseArrayArithmetics.test_float_scalar_comparisonc C st |}t ddddddddddg
|
¡}t ddddddddddg
|
¡}t||dd}t||dd}| ||||||¡ d S ) Nç ç ð?ç @ç @ç @ç @r r7 )r r9 r r' © r r r r; r$ r Úrvaluesr r r r r Ú"test_float_same_index_without_nans s z=TestSparseArrayArithmetics.test_float_same_index_without_nansc
|
C s| |}t t jdddt jddddt jg
|
¡}t t jdddt jddddt jg
|
¡}t||d}t||d} | || ||||¡ d S )Nr4 r6 r r5 é r= r8 )
|
r r r r; r
|
r$ r rG r r r r r Ú test_float_same_index_with_nans s $$ z:TestSparseArrayArithmetics.test_float_same_index_with_nansc C sÜ t t jdddt jddddt jg
|
¡}t t jdddt jddddt jg
|
¡}t||d}t||d}| ||||¡ t ddd d
|
dddd ddg
|
¡}t dd d d dddd d dg
|
¡}t||dd }t||dd }| ||||¡ d S )Nr4 r6 r r5 rI r= r@ rA rB rC rD rE r7 r> ©r r r rG r r r r r Ú test_float_same_index_comparison¦ s $$ z;TestSparseArrayArithmetics.test_float_same_index_comparisonc C s& |}t t jdddt jddddt jg
|
¡}t dt jddt jddddt jg
|
¡}t||d}t||d}| ||||||¡ | ||d ||d ||¡ t||dd}t||d}| ||||||¡ t||dd}t||dd}| ||||||¡ t||dd}t||dd}| ||||||¡ d S ©Nr4 r6 r r5 é r= r7 r8 rF r r r Útest_float_array¶ s $$ z+TestSparseArrayArithmetics.test_float_arrayc C s& |}t t jdddt jddddt jg
|
¡}t dt jddt jddddt jg
|
¡}t|dd}t|dd}| ||||||¡ | ||d ||d ||¡ t|ddd }t|dd}| ||||||¡ t|ddd }t|ddd }| ||||||¡ t|ddd }t|ddd }| ||||||¡ d S )
|
Nr4 r6 r r5 rN r r= r r7 r8 )r r r; r$ r rG r r r r r Ú test_float_array_different_kindÍ s $$ z:TestSparseArrayArithmetics.test_float_array_different_kindc C s t t jdddt jddddt jg
|
¡}t dt jddt jddddt jg
|
¡}t||d}t||d}| ||||¡ | ||d ||d ¡ t||dd}t||d}| ||||¡ t||dd}t||dd}| ||||¡ t||dd}t||dd}| ||||¡ d S rM r> rK r r r Útest_float_array_comparisonä s $$ z6TestSparseArrayArithmetics.test_float_array_comparisonc
|
C sÚ |}t j}t jddddddddddg
|
|d}t jddddddddddg
|
|d}t|||d}|jt|ksntt|||d} | jt|kst| || ||||¡ | || d ||d ||¡ t|d||d}|jt|ksàtt|||d} | jt|kst| || ||||¡ t|d||d}|jt|ks:tt|d||d} | jt|ks^t| || ||||¡ t|d||d}|jt|dd kstt|d||d} | jt|dd ksÂt| || ||||¡ d S )
|
Nr r4 r6 ©r) r5 rN ©r) r )r, r) r ©r, )r Úint64r9 r r) r r( r' )
|
r r r r; r$ r) r rG r r r r r Útest_int_arrayù s2 "" z)TestSparseArrayArithmetics.test_int_arrayc C s d}t jddddddddddg
|
|d}t jddddddddddg
|
|d}t|||d}t|||d}| ||||¡ | ||d ||d ¡ t|||dd }t|||d}| ||||¡ t|||dd }t|||dd }| ||||¡ t|||dd }t|||dd }| ||||¡ d S )
|
NrU r r4 r6 rR r5 rN rS )r) r r, )r r9 r r1 )r r r) r rG r r r r r Útest_int_array_comparison s ""z4TestSparseArrayArithmetics.test_int_array_comparisonTFc C sh t jddddgt jd}t jddddgt jd}t||t j|d}t||t j|d}| ||||¡ d S ©NTFrR )r r) r, ©r r9 r+ r r2 ©r r r, r rG r r r r r Útest_bool_same_index5 s
|
z/TestSparseArrayArithmetics.test_bool_same_indexc C sp t jddddddgt jd}t jddddddgt jd}t||t j|d}t||t j|d}| ||||¡ d S rX rY rZ r r r Útest_bool_array_logical@ s
|
z2TestSparseArrayArithmetics.test_bool_array_logicalc C sx |}d}t t jdddt jddddt jg
|
¡}t jddddddddddg
|
|d}t||d} t||d}
|
|
|
jt|ksxt| | |
|
||||¡ | | |
|
d ||d ||¡ t||dd } t||d}
|
|
|
jt|ksÔt| | |
|
||||¡ t||dd } t||dd }
|
|
|
jt|kst| | |
|
||||¡ t||dd } t||dd }
|
|
|
jt|dd
|
ks`t| | |
|
||||¡ d S © NrU r4 r6 r r5 rN rR r= r7 rT )r r9 r: r r) r r( r' ) r r r r; r
|
r$ Úrdtyper rG r r r r r Útest_mixed_array_float_intK s* $" z5TestSparseArrayArithmetics.test_mixed_array_float_intc C s` d}t t jdddt jddddt jg
|
¡}t jddddddddddg
|
|d}t||d}t||d}|jt|kstt| ||||¡ | ||d ||d ¡ t||dd }t||d}|jt|ksÈt| ||||¡ t||dd }t||dd }|jt|kst| ||||¡ t||dd }t||dd }|jt|dd
|
ksLt| ||||¡ d S r] )r r9 r: r r) r r( r1 )r r r^ r rG r r r r r Útest_mixed_array_comparisong s( $" z6TestSparseArrayArithmetics.test_mixed_array_comparisonc C sl t ddddg}t ddddg}||A }tjjj dtjdddgdd¡}t dddg|d }t ||¡ d S )
|
NTFrI r r4 r6 Úint32rR )Z sparse_index)
|
r r ÚcoreZarraysÚsparseZIntIndexr r9 r Úassert_sp_array_equal)r ÚsÚtr& Zsp_indexr% r r r Útest_xor s "z#TestSparseArrayArithmetics.test_xor) Ú__name__Ú
|
__module__Ú __qualname__r r- r' r/ r1 r2 ÚpytestÚmarkÚ parametrizer< r? rH rJ rL rO rP rQ rV rW r r: r[ r\ r_ r` rg r r r r r s0 + $
|
|
|
|
r r$ c C s@ t ddgdd}| |ddg}| |t ddg}t ||¡ d S )Nr r4 rT )r r rd )r$ Úarrr& r% r r r Útest_with_list s ro c C s@ t ddgdd} t ddgddgg¡}| |¡}|tks<td S )Nr r4 rT r6 r5 rI )r r Z DataFrameÚ__add__ÚNotImplementedr( )rn Zdfr& r r r Útest_with_dataframe s
|
rr c C s6 t ddgdd} | t d¡ }| d }t ||¡ d S )Nr r4 rT r6 ©r r r9 r rd )rn r& r% r r r Útest_with_zerodim_ndarray s rt Úufuncrn éÿÿÿÿr4 c C s8 | |}| |j }t| t |¡|d}t ||¡ d S )NrT )r, r r Úasarrayr rd )ru rn r& r, r% r r r Ú test_ufuncs¦ s
|
rx za, br6 rT c C sD | ||}| t |¡t |¡}t|ts.tt t |¡|¡ d S r )r rw r r r( r r )ru r r r& r% r r r Útest_binary_ufuncs± s
|
ry c C sL t ddddg} t ddddg¡}|| 7 }t ddddg¡}t ||¡ d S )Nr r6 r4 r5 )r r r9 r r ©Zsparrayr r% r r r Útest_ndarray_inplaceÄ s
|
r{ c C sN t ddddg} t ddddg¡}| |7 } t ddddgdd}t | |¡ d S )Nr r6 r4 r5 rT rs rz r r r Útest_sparray_inplaceÌ s
|
r| Úconsc C s@ t ddg}| dddg}tjtdd ||@ W 5 Q R X d S )NTz operands have mismatched length)Úmatch)r rk ZraisesÚ
|
ValueError)r} ÚleftÚrightr r r Ú test_mismatched_length_cmp_opÔ s r ÚaddÚsubÚmulÚtruedivÚfloordivÚpowr, r5 c C sz t t| } tj d¡}tj d¡}||d d d
< ||d d d
< t||d}t||d}tjdd | ||}t| | ¡ | ¡ |jd}t |tst
|
t | ¡ | ¡ ¡ | || ¡ }t |tsÊt
|
t ||¡ | | ¡ |} t | tsòt
|
t || ¡ | |d}
|
t |
|
tst
|
z | | ¡ d}| |jd} W n tk
|
rL Y n X t |
|
j| ¡ t |
|
¡ |¡ W 5 Q R X d S )Né r6 r5 rT r )ÚallrI )ÚgetattrÚoperatorr ÚrandomZrandnr r r! r, r r( r Zassert_almost_equalrd r ) r$ r, Zdata1Zdata2ÚfirstÚsecondr. ÚexpZres2Zres3Zres4Zexp_fvr r r Útest_binary_operatorsÜ s<
|
|
ÿ
|
r )#r Únumpyr rk Zpandasr Zpandas._testingZ_testingr Zpandas.core.arrays.sparser r Zfixturer r r rl rm Úeqr ro rr rt Úabsr rx r9 Zgreaterry r{ r| Úlistr r: r r r r r Ú<module> sP
|
|
r
|
ÿ ûþ
|
|
|
