U
|
ý°dB ã @ sD d dl Zd dlZd dlZd dlmZ d dlm Z G dd de Z
|
dS )é N)ÚBaseExtensionTestsc @ sè e Zd ZdZdd Zdd Zdd Zdd Zd
|
d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd Zdd Zd d Zejjd d d d!gejd d d!gd"d#e d d d!g¡gd$d%d&gd'd(d) Zejjd d d d!ejgejd d d!ejgd"d#gd$d%gd'd*d+ Zejjd,d-ejjd d d d!ejgejd d d!ejgd"d#gd$d%gd'd.d/ Z d0d1 Z d2d3 Z d4d5 Z d6d7 Z d8d9 Z!d:d; Z"d<d= Z#d>d? Z$d@dA Z%ej dBdCdDg¡dEdF Z&dGdH Z'dIdJ Z(dKdL Z)dMdN Z*dOdP Z+dQdR Z,dSS )TÚBaseGetitemTestsz%Tests for ExtensionArray.__getitem__.c C sX t |¡}|jd d
}t |d d
¡}| ||¡ |jddddg }| ||¡ d S )Né r é é é )ÚpdÚSeriesÚilocÚassert_series_equal©ÚselfÚdataÚserÚresultÚexpected© r úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/extension/base/getitem.pyÚtest_iloc_series s
|
z!BaseGetitemTests.test_iloc_seriesc C s* t |tjt|ddd¡}t d|d d
i¡}|jd d
dgf }| ||¡ |jdddd gdgf }| ||¡ t j|d d
dd
|
}|jd d
df }| ||¡ |jd d
df }| ||¡ |jd d
d d d
f }| ||dg ¡ |d dg jd d
d d d
f }| ||d g ¡ d S ) NÚint64©Údtype©ÚAÚBr r r r r r ©Únamer ) r Ú DataFrameÚnpÚarangeÚlenr
|
Úassert_frame_equalr r ©r r Údfr r r r r Útest_iloc_frame s z BaseGetitemTests.test_iloc_framec C sò t d|i¡}|jd d
d d
f }| ||¡ |jd d
d d
f }| ||¡ |jd d
d d
f }| ||¡ |jd d
d d d
f }| ||¡ |jd d
dd
f }| ||jd d
d d
f ¡ |jd d
dd
f }| ||¡ d S )Nr r r r éÿÿÿÿ)r r r
|
r! )r r r# r r r r Ú test_iloc_frame_single_block1 s z-BaseGetitemTests.test_iloc_frame_single_blockc C sX t |¡}|jd d
}t |d d
¡}| ||¡ |jddddg }| ||¡ d S )Nr r r r r )r r Úlocr r r r r Útest_loc_seriesG s
|
z BaseGetitemTests.test_loc_seriesc C sÎ t |tjt|ddd¡}t d|d d
i¡}|jd d
dgf }| ||¡ |jddd dgdgf }| ||¡ t j|d d
dd
|
}|jd d
df }| ||¡ |jd d
df }| ||¡ d S ) Nr r r r r r r r r r ) r r r r r r' r! r r r" r r r Útest_loc_frameP s z BaseGetitemTests.test_loc_framec C s~ t d|i¡}t j|d gdgd|jd}|jd }| ||¡ t j|d gdgt|d |jd}|jd }| ||¡ d S )Nr r )Úindexr r r% r )r r r r r' r r r
|
r" r r r Ú test_loc_iloc_frame_single_dtypef s
|
|
ÿ
|
z1BaseGetitemTests.test_loc_iloc_frame_single_dtypec C s> |d }t ||jjstt |¡d }t ||jjs:td S ©Nr )Ú
|
isinstancer ÚtypeÚAssertionErrorr r ©r r r r r r Útest_getitem_scalaru s z$BaseGetitemTests.test_getitem_scalarc C sä d}t jt|d |d W 5 Q R X t jt|d |d W 5 Q R X t|}d dddd | d
|
| d |d d | d|d d | g¡}t jt|d ||d W 5 Q R X t jt|d || d W 5 Q R X d S )Nz{only integers, slices \(`:`\), ellipsis \(`...`\), numpy.newaxis \(`None`\) and integer or boolean arrays are valid indices©ÚmatchZfoog @ú|zlist index out of rangezindex out of boundszOut of bounds accessz loc must be an integer between -z and zindex r z' is out of bounds for axis 0 with size zindex -)ÚpytestÚraisesÚ
|
IndexErrorr Újoin)r r ÚmsgZubr r r Útest_getitem_invalid| s( ÿúÿ
|
z%BaseGetitemTests.test_getitem_invalidc C s |d }|||st d S r, )r/ )r Ú data_missingÚna_cmpÚna_valuer r r r Útest_getitem_scalar_na s z'BaseGetitemTests.test_getitem_scalar_nac C sL |g }t |dkstt|t|s*t|tjg dd }| ||¡ d S )Nr r r )r r/ r- r. r ÚarrayÚ assert_extension_array_equal)r r r r r r r Útest_getitem_empty s
|
z#BaseGetitemTests.test_getitem_emptyc C sà t jt|td}|| }t|dks*tt|t|s<tt jt|td}t |¡| }t|dkslt|j |j ks|td|d< || }t|dkstt|t|s®tt |¡| }t|dksÌt|j |j ksÜtd S )Nr r Tr )
|
r Úzerosr Úboolr/ r- r. r r r )r r Úmaskr r r r Útest_getitem_mask¥ s z"BaseGetitemTests.test_getitem_maskc C sr t ddg¡}dt| }tjt|d || W 5 Q R X tj|dd}tjt|d || W 5 Q R X d S )NTFz-Boolean index has wrong length: 2 instead of r2 Úbooleanr )r r? r r5 r6 r7 r )r r rD r9 r r r Útest_getitem_mask_raises½ s z)BaseGetitemTests.test_getitem_mask_raisesc C sÎ t jtj|jdddd}|| }t|dks2tt|t|sDtt |¡| }t|dksbt|j
|
|j
|
ksrtd|d d
< | dddd d
|
g¡}|| }| ||¡ t |¡}t |¡| }| ||¡ d S ) NrC r rF r Té r r r r )r r? r rB Úshaper r/ r- r. r r Útaker@ r )r r rD r r r r r Ú test_getitem_boolean_array_maskÇ s
|
z0BaseGetitemTests.test_getitem_boolean_array_maskc C s t jtj|jdddd}t j|d d
< d|dd
< || }|| d¡ }| ||¡ t |¡}|| }|| d¡ }| ||¡ d S )NrC r rF r Tr F)
|
r r? r rB rI ÚNAZfillnar@ r r )r r rD r r Úsr r r Ú(test_getitem_boolean_na_treated_as_falseÚ s
|
z9BaseGetitemTests.test_getitem_boolean_na_treated_as_falseÚidxr r r ZInt64r Úlistz integer-arrayz numpy-array)Úidsc C sn || }t |dkstt|t|s*t| dddg¡}| ||¡ t |¡}t |¡| }| ||¡ d S )Nr r r r ) r r/ r- r. rJ r@ r r r )r r rO r r r r r Útest_getitem_integer_arrayì s
|
z+BaseGetitemTests.test_getitem_integer_arrayc C s* d}t jt|d || W 5 Q R X d S )Nú9Cannot index with an integer indexer containing NA valuesr2 )r5 r6 Ú
|
ValueError)r r rO r9 r r r Ú(test_getitem_integer_with_missing_raisesü s z9BaseGetitemTests.test_getitem_integer_with_missing_raiseszSTries label-based and raises KeyError; in some cases raises when calling np.asarray)Úreasonc C sJ d}t j|dd tt|D d}tjt|d || W 5 Q R X d S )NrS c S s g | ]}t d ¡qS )r )ÚtmZrands)Ú.0Ú_r r r Ú
|
<listcomp> s zTBaseGetitemTests.test_getitem_series_integer_with_missing_raises.<locals>.<listcomp>©r* r2 )r r Úranger r5 r6 rT )r r rO r9 r r r r Ú/test_getitem_series_integer_with_missing_raises s
|
z@BaseGetitemTests.test_getitem_series_integer_with_missing_raisesc C s@ |t d }t|t|s t|t d }t|t|s<td S )Nr r )Úslicer- r. r/ r0 r r r Útest_getitem_slice s z#BaseGetitemTests.test_getitem_slicec C sÔ |dd d
f }| ||¡ |d d
df }| ||¡ |dd d
f }| ||d d
¡ |d d
df }| ||d d
¡ |dd d d
f }| ||d d d
¡ |d d d
df }| ||d d d
¡ d S )N.r r )r@ r0 r r r Ú test_getitem_ellipsis_and_slice s z0BaseGetitemTests.test_getitem_ellipsis_and_slicec C s¾ t j|dd tt|D d}| d¡|jd ks8t| ddg¡}|jddg }| ||¡ | td¡}|jdd g }| ||¡ | d
|
¡d kst| |j
|
¡ d ¡d ks¶tt j|d d
t d d}| d ¡|jd ksèt| td d¡}|jd ddg }| ||¡ | d¡}|d ks,t| d¡|jd ksFt| d
|
¡|jd
|
ks`t| t|¡d ksxtt |¡}t d ¡ |d d d
}W 5 Q R X | d ¡d ksºtd S )Nc S s g | ] }d | qS )r r )rX Úir r r rZ 5 s z-BaseGetitemTests.test_get.<locals>.<listcomp>r[ r r é r r r r% ZabcdefÚcÚbÚdÚZ)r r r\ r Úgetr
|
r/ r r^ r* ÚmaxrP rW Zassert_produces_warning)r r rM r r Ús2r r r Útest_get3 s0
|
|
zBaseGetitemTests.test_getc C sZ t |¡dddg }|jd |d ks*t|jd |d ks@t|jd |d ksVtd S )Nr r r r )r r r
|
r/ r0 r r r Útest_take_sequenceX s z#BaseGetitemTests.test_take_sequencec C s² | ddg¡}|j|jks t|d |d ks2t|d |d ksFt|j ddgd|d}|d |d ksnt||d |sttjtdd | t|d g¡ W 5 Q R X d S )Nr r% r T)Ú
|
allow_fillÚ
|
fill_valuez out of boundsr2 )rJ r r/ r5 r6 r7 r )r r r= r< r r r r Ú test_take^ s zBaseGetitemTests.test_takec C s |d d
}|j dgdd}||d |s.td}tjt|d | dg¡ W 5 Q R X tjtdd | ddg¡ W 5 Q R X d S ) Nr r% T©rl z;cannot do a non-empty take from an empty axes|out of boundsr2 zcannot do a non-empty taker )rJ r/ r5 r6 r7 )r r r= r< Úemptyr r9 r r r Útest_take_emptyk s z BaseGetitemTests.test_take_emptyc C sJ t |}| d| |d dg¡}| dd|d |d g¡}| ||¡ d S )Nr r r% )r rJ r@ )r r Únr r r r r Útest_take_negativey s z#BaseGetitemTests.test_take_negativec C sX |d }|d }|j |||g|jd}|jddg|dd}| ddg¡}| ||¡ d S )Nr r r r% T©rm rl )Ú_from_sequencer rJ r@ )r r; rm ÚnaÚarrr r r r r Útest_take_non_na_fill_value s ÿz,BaseGetitemTests.test_take_non_na_fill_valuec C s2 t jtdd |jddg|dd W 5 Q R X d S )NÚ r2 r éþÿÿÿTrt )r5 r6 rT rJ )r r r= r r r Ú&test_take_pandas_style_negative_raises s z7BaseGetitemTests.test_take_pandas_style_negative_raisesrl TFc C sB |d d
}t jtdd |jt ddg¡|d W 5 Q R X d S )Nr zout of bounds|out-of-boundsr2 r ro )r5 r6 r7 rJ r Zasarray)r r rl rw r r r Ú test_take_out_of_bounds_raises s z/BaseGetitemTests.test_take_out_of_bounds_raisesc C sb t |¡}| ddg¡}t j|j|d |t|d g|jddt|d gd}| ||¡ d S )Nr r% r r r[ )r r rJ ru r r r )r r rM r r r r r Útest_take_series s
|
"þz!BaseGetitemTests.test_take_seriesc C sÜ t |¡}| dddg¡}t j| dddg¡dddgd}| ||¡ t|}| dd|g¡}t j|j||d |g|jddd|gd}| ||¡ | ||d g¡}t j|j||g|jd||d gd}| ||¡ d S )Nr r r r[ r% r )r r ÚreindexrJ r r ru r )r r r= rM r r rr r r r Ú test_reindex s"
|
þ
|
ÿz BaseGetitemTests.test_reindexc C sn |d }|d }|j ||g|jd}t |¡}|jdddg|d}t |j |||g|jd¡}| ||¡ d S )Nr r r r )rm )ru r r r r~ r )r r; Zvalidrv rw r r r r r r Ú test_reindex_non_na_fill_value³ s
|
ÿz/BaseGetitemTests.test_reindex_non_na_fill_valuec C sd t d|i¡}|jdgdf }|jdks,t|jjd jdksBtt|jdr`|jjjdks`td S )Nr r r Úblocks) r r r' Úndimr/ Z_mgrÚarraysÚhasattrÚ_block)r r r# Úresr r r Ú test_loc_len1À s z BaseGetitemTests.test_loc_len1c C s~ t |¡}|d d
¡ }||d ks*td}tjt|d |d d
¡ W 5 Q R X tjt|d | ¡ W 5 Q R X d S )Nr r z6can only convert an array of size 1 to a Python scalarr2 )r r Úitemr/ r5 r6 rT )r r rM r r9 r r r Ú test_itemÉ s
|
zBaseGetitemTests.test_itemc C s^ G dd dt jj}t d|tjddgtdi¡}|jd d
}| t |d jj
|
d¡ d S ) Nc s e Zd ZdZ fddZ ZS )zBBaseGetitemTests.test_ellipsis_index.<locals>.CapturingStringArrayz6Extend StringArray to capture arguments to __getitem__c s || _ t |¡S )N)Ú last_item_argÚsuperÚ __getitem__)r r ©Ú __class__r r r Ü s zNBaseGetitemTests.test_ellipsis_index.<locals>.CapturingStringArray.__getitem__)Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__r Ú __classcell__r r r r ÚCapturingStringArrayÙ s r Zcol1ZhelloZworldr r zslice(None, 1, None)) r r Z StringArrayr r r? Úobjectr
|
Z assert_equalÚstrr )r r r# rY r r r Útest_ellipsis_indexÖ s ÿz$BaseGetitemTests.test_ellipsis_indexN)-r r r r r r$ r& r( r) r+ r1 r: r> rA rE rG rK rN r5 ÚmarkZ parametrizer r? r rR rL rU Zxfailr] r_ r` rj rk rn rq rs rx r{ r| r} r r r r r r r r r r sj
|
*ý
|
$ý
|
ÿ$ý %
|
r ) Únumpyr r5 Zpandasr Zpandas._testingZ_testingrW Z pandas.tests.extension.base.baser r r r r r Ú<module> s
|
|
