U
|
ý°d²' ã @ s d dl Zd dlZd dlZd dlmZ d dlm Z m
|
Z
|
e ej ej dddej ddej dg
|
¡Z e e ZG d d
|
d
|
ZG d d d ZG d d dZG dd dZdS )é N)Ú SparseArrayÚ SparseDtypeé é é é é é c @ sP e Zd Zdd Zdd Zej dej dd
ej dd
ej dd
ej dd
ej dd
|
ej d d
ej dd
ej d d
ej dd
ej ddd
ej ddd
ej ddd
ej ddd
ej ddd
ej ddd
g¡ej dej
|
gd dgd ej
|
gd dgd g g¡dd Z dd Z dd Z d d Zd d Zd d! ZdS )"Ú TestGetitemc C sF t ¡ }tt D ]0\}}t ||| ¡ t t | || ¡ qd S ©N)ÚarrZto_denseÚ enumerateÚtmZassert_almost_equal)ÚselfÚdenseÚiÚvalue© r ú_d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/sparse/test_indexing.pyÚ test_getitem s zTestGetitem.test_getitemc C s8 t dddg}|dddg }t ddg}t ||¡ d S )Nr r r TF)r r Úassert_sp_array_equal)r r ÚresultÚexpectedr r r Útest_getitem_arraylike_mask s z'TestGetitem.test_getitem_arraylike_maskÚslcNr é
|
éd éýÿÿÿéûÿÿÿéüÿÿÿiôÿÿÿr r éÿÿÿÿéþÿÿÿr éúÿÿÿÚas_denser c C s6 t |¡}t|}|| }t|| }t ||¡ d S r )ÚnpÚarrayr r r )r r r# r r r r r r Ú test_getslice s
|
|
zTestGetitem.test_getslicec C sö t t jdddddt jt jdg ¡}t|}|tdd f }t|dd
}t ||¡ t|dd}|tdd f }t|dd
dd}t ||¡ d}tjt |d |dd
d d
f W 5 Q R X tjt |d |dd
d d
f W 5 Q R X d S )Nr r r r ©Ú
|
fill_valueztoo many indices for array©Úmatch)
|
r$ r% Únanr Úslicer r ÚpytestÚraisesÚ
|
IndexError)r r ÚsparseÚresÚexpÚmsgr r r Útest_getslice_tuple< s " z TestGetitem.test_getslice_tuplec C s0 t dddg}|dddg }|j|jks,td S )Nr r r F)r ÚdtypeÚAssertionError)r r r1 r r r Útest_boolean_slice_emptyQ s z$TestGetitem.test_boolean_slice_emptyc C s´ t ddgd tjdd}t tjdtjddg}t t| |¡ | }t| }t tjddd tjg}t ||¡ t ddtjgd tjtjd}t| }t tjddg}t ||¡ d S )
|
NFTr )r5 r( r r r r r )r r$ Zbool_r+ r r r )r Z spar_boolr2 r1 r r r Ú test_getitem_bool_sparse_arrayV s ÿz*TestGetitem.test_getitem_bool_sparse_arrayc C sL t ddddtjtjgtjd}||dk }t ddgtjd}t ||¡ d S )Nr r r r r' g @g @)r r$ r+ r r )r r r1 r2 r r r Ú,test_getitem_bool_sparse_array_as_comparisonh s z8TestGetitem.test_getitem_bool_sparse_array_as_comparisonc C sê t ddddddddddg
|
dd}t td ¡s2ttd dksBttd dksRt|d dksbt|d dksrt|d dkstd
|
}tjt|d td W 5 Q R X tjt|d td W 5 Q R X td tttd ksætd S )Nr r r r r r r r' é z%must be an integer between -10 and 10r) é iõÿÿÿr ) r r$ Úisnanr r6 r- r. r/ Úlen)r ZzarrÚerrmsgr r r Ú test_get_itemo s zTestGetitem.test_get_item)Ú__name__Ú
|
__module__Ú __qualname__r r r- ÚmarkÚ parametrizer$ Zs_r+ r& r4 r7 r8 r9 r? r r r r r
|
s> ñþ *ÿ r
|
c @ s e Zd Zdd ZdS )Ú TestSetitemc sh t t ¡ fdd} fdd}tjtdd | W 5 Q R X tjtdd | W 5 Q R X d S )Nc s d d< d S )Nr r r r ©r r r Úsetitem s z*TestSetitem.test_set_item.<locals>.setitemc s d dd
< d S )Nr r r r r rF r r Úsetslice s z+TestSetitem.test_set_item.<locals>.setslicezassignment via setitemr) )r Úarr_dataÚcopyr- r. Ú TypeError)r rG rH r rF r Ú test_set_item s zTestSetitem.test_set_itemN)r@ rA rB rL r r r r rE
s rE c @ sf e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
e j dddg¡dd ZdS )ÚTestTakec C s, d}t jt|d t d¡ W 5 Q R X d S )Nz-'indices' must be an array, not a scalar '2'.r) r )r- r. Ú
|
ValueErrorr Útake)r r3 r r r Útest_take_scalar_raises s z TestTake.test_take_scalar_raisesc C s\ t t tddg¡}t t ddg¡|¡ t t tdddg¡}t t dddg¡|¡ d S )Nr r r r ©r r$ rO rI r r r ©r r2 r r r Ú test_take s zTestTake.test_takec C s< t jddgtdd}|jddgdtjd}t ||¡ d S )Nr Úint64)r5 r T©Ú
|
allow_fillr( )Úpdr% r rO r$ r+ r r )r Úar r r r Útest_take_all_empty£ s z TestTake.test_take_all_emptyc C s t dt jdddg¡}t|dd}tt |dg¡dd}t | dg¡|¡ tt |dddg¡dd}t | dddg¡|¡ d S )Nr r r r' r )r$ r% r+ r rO r r )r Údatar0 r2 r r r Útest_take_fill_value¨ s z TestTake.test_take_fill_valuec C sX t t tdg¡}t t dg¡|¡ t t tdddg¡}t t dddg¡|¡ d S )Nr r r r! rQ rR r r r Útest_take_negative² s zTestTake.test_take_negativec C s* t jtdd t dg¡ W 5 Q R X d S )NZboundsr) r; )r- r. r/ r rO )r r r r Ú test_bad_take¹ s zTestTake.test_bad_takec C s t tjtjdtjdg}| t dddg¡¡}t tjtjdg}t ||¡ |jt dddg¡dd}t tjtjtjg}t ||¡ |jt dddg¡ddd}t tjtjdg}t ||¡ d }tjt |d
|
|jt ddd g¡dd W 5 Q R X tjt |d
|
|jt ddd g¡dd W 5 Q R X d }tjt
|
|d
|
| t ddg¡¡ W 5 Q R X tjt
|
|d
|
| t ddg¡¡ W 5 Q R X tjt
|
|d
|
|jt ddg¡dd W 5 Q R X d S )Nr r r r T©rV FrU zInvalid value in 'indices'r) r! r ú out of bounds value in 'indices'r" r © r r$ r+ rO r% r r r- r. rN r/ ©r r0 r r r3 r r r Útest_take_filling½ s, $$ zTestTake.test_take_fillingc C sÈ t tjddddgdd}| t dddg¡¡}t dtjdgdd}t ||¡ |jt dddg¡dd}t dtjtjgdd}t ||¡ |jt dddg¡ddd }t dtjdgdd}t ||¡ d
|
}tjt |d |jt ddd g¡dd W 5 Q R X tjt |d |jt ddd g¡dd W 5 Q R X d}tjt
|
|d | t ddg¡¡ W 5 Q R X tjt
|
|d | t ddg¡¡ W 5 Q R X tjt
|
|d |jt ddg¡dd W 5 Q R X d S )Nr r r r' r Tr^ FrU zInvalid value in 'indices'.r) r! r r_ r" r r` ra r r r Ú test_take_filling_fill_valueÞ s, $$ z%TestTake.test_take_filling_fill_valueÚkindÚblockÚintegerc C s. t tjtjtjtjtjg|d}| t dddg¡¡}t tjtjtjg|d}t ||¡ |jt dddg¡dd}t tjtjtjg|d}t ||¡ d}tjt |d | t dd g¡¡ W 5 Q R X tjt |d | t dd
|
g¡¡ W 5 Q R X tjt |d |jt dd
|
g¡dd W 5 Q R X d S ) N)rd r r r Tr' r_ r) r" r )
|
r r$ r+ rO r% r r r- r. r/ )r rd r0 r r r3 r r r Útest_take_filling_all_nan s z"TestTake.test_take_filling_all_nanN)r@ rA rB rP rS rY r[ r\ r] rb rc r- rC rD rg r r r r rM s
|
!#rM c @ s e Zd Zdd ZdS )Ú TestWherec C sx t tjdgdd}t ddg¡}| | d¡}t ddgdd}t ||¡ t |¡}| | d¡}t
|
|t |¡¡ d S )Ng ð?r r' TFr ) r r$ r+ r% Z_wherer r rW ZSeriesÚwhereZassert_series_equal)r r Úmaskr1 r2 Zserr r r Ú test_where_retain_fill_value s
|
z&TestWhere.test_where_retain_fill_valueN)r@ rA rB rk r r r r rh s rh )Únumpyr$ r- ZpandasrW Zpandas._testingZ_testingr Zpandas.core.arrays.sparser r r% r+ rI r r
|
rE rM rh r r r r Ú<module> s &v
|
