U
|
ý°dK9 ã @ sD d dl Zd dlZd dlZd dlmZ d dlm Z G dd de Z
|
dS )é N)ÚBaseExtensionTestsc
|
@ sÜ e Zd Zejdd dd dd dd dd dd dd gd d
|
d d d ddgddd Zdd Zdd Zdd Zdd Z dd Z
|
ej d d d g¡d d! Z d"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- Zej j de d.d.d.d/d/g¡ejd.d.d.d/d/gd0d1ejd.d.d.ejejgd0d1gd2d3d4gd5d6d7 Zd8d9 Zd:d; Zej j d<d=d>d?gejd=d>d?gd@d1e d=d>d?g¡gdAdBd2gd5dCdD Zej j dEd=d>d?ejgd/fej d=d>d?ejgd.ej j dFdGdHejd=d>d?ejgd@d1d/fejd=d>d?ejgd@d1d/fgdIdJdKdLgd5dMdN Z ej dOd.d/g¡ej d d dPg¡dQdR Z ej d d dPg¡dSdT Z dUdV Z!dWdX Z"dYdZ Z#d[d\ Z$d]d^ Z%d_d` Z&dadb Z'dcdd Z(dedf Z)dgdh Z*didj Z+dkdl Z,dmdn Z-dodp Z.dqdr Z/dsdt Z0dudv Z1dPS )wÚBaseSetitemTestsc C s | j S ©N©Úindex©Úx© r úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/extension/base/setitem.pyÚ<lambda> ó zBaseSetitemTests.<lambda>c C s
|
t | jS r )Úlistr r r r r
|
r r c C s t d S r )Úslicer r r r
|
r r c C s t dt| S )Nr )r Úlenr r r r
|
r r c C s t t| S r )Úranger r r r r
|
r r c C s t tt| S r )r r r r r r r
|
r r c C s t jt| tdS ©N©Údtype)ÚnpZonesr Úboolr r r r
|
r r r z list[index]Z
|
null_sliceZ
|
full_slicer z list(range)Úmask)ÚparamsÚidsc C s |j S )z·
|
Fixture for an indexer to pass to obj.loc to get/set the full length of the
|
object.
|
|
In some cases, assumes that obj.index is the default RangeIndex.
|
)Úparam)ÚselfÚrequestr r r
|
Ú full_indexer
|
s z BaseSetitemTests.full_indexerc C s2 |rt |¡}|d |d< |d |d ks.td S ©Né r ©ÚpdÚSeriesÚAssertionError©r ÚdataÚ box_in_seriesr r r
|
Útest_setitem_scalar_series' s
|
z+BaseSetitemTests.test_setitem_scalar_seriesc C sZ |rt |¡}| ¡ }|d |d g|ddg< |d |d ksBt|d |d ksVtd S r )r r! Úcopyr" ©r r$ r% Úoriginalr r r
|
Útest_setitem_sequence- s
|
z&BaseSetitemTests.test_setitem_sequencec C s¢ t |¡}| ¡ }|d g}|r*| |¡}d}tjt| d¡d ||ddg< W 5 Q R X | ||¡ tjt| d¡d ||t d< W 5 Q R X | ||¡ d S )Nr z5cannot set using a {} indexer with a different lengthz list-like©Úmatchr r é )
|
r r! r' Z_from_sequenceÚpytestÚraisesÚ
|
ValueErrorÚformatÚassert_series_equalr )r r$ Zas_arrayÚserr) ÚvalueÚxprr r r
|
Ú.test_setitem_sequence_mismatched_length_raises6 s
|
|
|
z?BaseSetitemTests.test_setitem_sequence_mismatched_length_raisesc C s8 |rt |¡}| ¡ }g |tjg td< | ||¡ d S r )r r! r' r ÚarrayÚintÚ assert_equalr( r r r
|
Útest_setitem_empty_indexerG s
|
|
z+BaseSetitemTests.test_setitem_empty_indexerc C sJ |rt |¡}|d |ddg< |d |d ks2t|d |d ksFtd S )Né r r r r# r r r
|
Ú test_setitem_sequence_broadcastsN s
|
|
z1BaseSetitemTests.test_setitem_sequence_broadcastsÚsetterÚlocÚilocc C s8 t |¡}t||}|d |d< |d |d ks4td S r )r r! Úgetattrr" )r r$ r= Úarrr r r
|
Útest_setitem_scalarU s
|
|
z$BaseSetitemTests.test_setitem_scalarc C sB t t t|¡|d¡}|d |jd< |jd |d ks>td S )N©ÚAÚBr ©r rE )r Ú DataFramer Úaranger r> r" ©r r$ Údfr r r
|
Ú test_setitem_loc_scalar_mixed\ s z.BaseSetitemTests.test_setitem_loc_scalar_mixedc C s6 t d|i¡}|d |jd< |jd |d ks2td S )NrE r ©é
|
rE ©r rG r> r" rI r r r
|
Ú test_setitem_loc_scalar_singlea s z/BaseSetitemTests.test_setitem_loc_scalar_singlec C s8 t ||d¡}|d |jd< |jd |d ks4td S )NrC r rL rN rI r r r
|
Ú,test_setitem_loc_scalar_multiple_homogoneousf s z=BaseSetitemTests.test_setitem_loc_scalar_multiple_homogoneousc C sB t t t|¡|d¡}|d |jd< |jd |d ks>td S )NrC r ©r r rF )r rG r rH r r? r> r" rI r r r
|
Ú test_setitem_iloc_scalar_mixedk s z/BaseSetitemTests.test_setitem_iloc_scalar_mixedc C s6 t d|i¡}|d |jd< |jd |d ks2td S )NrE r )rM r rL ©r rG r? r> r" rI r r r
|
Ú test_setitem_iloc_scalar_singlep s z0BaseSetitemTests.test_setitem_iloc_scalar_singlec C s8 t ||d¡}|d |jd< |jd |d ks4td S )NrC r )rM r rL rS rI r r r
|
Ú-test_setitem_iloc_scalar_multiple_homogoneousu s z>BaseSetitemTests.test_setitem_iloc_scalar_multiple_homogoneousTFÚbooleanr z numpy-arrayz boolean-arrayzboolean-array-na)r c C sX |d d
¡ }| dddddg¡}|r<t |¡}t |¡}|d ||< | ||¡ d S ©Né r r- é ©r' Útaker r! r9 )r r$ r r% rA Úexpectedr r r
|
Útest_setitem_maskz s
|
|
|
z"BaseSetitemTests.test_setitem_maskc C sz t ddg¡}|r t |¡}tjtdd |d ||< W 5 Q R X tj|dd}tjtdd |d ||< W 5 Q R X d S )NTFz wrong lengthr+ r rV r )r r7 r r! r. r/ Ú
|
IndexError©r r$ r% r r r r
|
Útest_setitem_mask_raises s
|
z)BaseSetitemTests.test_setitem_mask_raisesc C sn t jtj|jdddd}d|d d
< t j|dd
< |rBt |¡}|d ||< |d d
|d k ¡ sjtd S )Nr r rV Tr- rX r ) r r7 r ÚzerosÚshapeÚNAr! Úallr" r_ r r r
|
Ú'test_setitem_mask_boolean_array_with_na s
|
z8BaseSetitemTests.test_setitem_mask_boolean_array_with_naÚidxr r r; ZInt64r z integer-arrayc C sX |d d
¡ }| dddddg¡}|r<t |¡}t |¡}|d ||< | ||¡ d S rW rZ )r r$ rf r% rA r\ r r r
|
Útest_setitem_integer_array¦ s
|
|
z+BaseSetitemTests.test_setitem_integer_arrayzidx, box_in_serieszGH-31948)Úreason)Zmarksz
|
list-Falsez list-Truezinteger-array-Falsezinteger-array-Truec C sZ | ¡ }|r,tj|dd tt|D d}d}tjt|d |d ||< W 5 Q R X d S )Nc S s g | ]}t d ¡qS )rY )ÚtmZrands)Ú.0Ú_r r r
|
Ú
|
<listcomp>È s zMBaseSetitemTests.test_setitem_integer_with_missing_raises.<locals>.<listcomp>r z9Cannot index with an integer indexer containing NA valuesr+ r )r' r r! r r r. r/ r0 )r r$ rf r% rA Úmsgr r r
|
Ú(test_setitem_integer_with_missing_raises¶ s z9BaseSetitemTests.test_setitem_integer_with_missing_raisesÚ as_callableNc s t |¡}tjt|td d d d
< |r: fdd}n }|rNt||}n|}|dd
||< |dd
||< |d |d kst|d |d
|
kstd S ) Nr Tr; c s S r r r ©r r r
|
r Ö r z<BaseSetitemTests.test_setitem_mask_aligned.<locals>.<lambda>rX é r r é ©r r! r ra r r r@ r" )r r$ ro r= r3 Zmask2Útargetr rp r
|
Útest_setitem_mask_alignedÎ s
|
z*BaseSetitemTests.test_setitem_mask_alignedc C st t |¡}tjt|td}d|d d
< |r8t||}n|}|d ||< |d |d ks\t|d |d ksptd S )Nr Tr; rM r r rs )r r$ r= r3 r rt r r r
|
Útest_setitem_mask_broadcastç s
|
z,BaseSetitemTests.test_setitem_mask_broadcastc C s t d|i¡}| ¡ }d|d< t |dgt| d¡}| ||¡ | ¡ }d|jd d
df< | ||¡ ||d< t ||d¡}| ||¡ d S ©NrD r rE rC )r rG r' r Úassert_frame_equalr> ©r r$ rJ Úresultr\ r r r
|
Útest_setitem_expand_columnsö s z,BaseSetitemTests.test_setitem_expand_columnsc C sx t ddgt| i¡}| ¡ }||d< t dgt| |d¡}| ||¡ | ¡ }||jd d
df< | ||¡ d S rw )r rG r r' rx r> ry r r r
|
Ú"test_setitem_expand_with_extension s z3BaseSetitemTests.test_setitem_expand_with_extensionc C sh t ddgt| i¡}dt|d d
dt| d}tjt|d |d d
|d< W 5 Q R X d S ) NrD r zLength of values \(rX z$\) does not match length of index \(z\)r+ rE )r rG r r. r/ r0 )r r$ rJ r5 r r r
|
Ú!test_setitem_frame_invalid_length s
|
ÿz2BaseSetitemTests.test_setitem_frame_invalid_lengthc C sP t j|d d
ddgd}t j| ddg¡|jd}|d |d< | ||¡ d S )Nr; )r r rQ r r )r r! r[ r r2 )r r$ r3 r\ r r r
|
Útest_setitem_tuple_index s z)BaseSetitemTests.test_setitem_tuple_indexc C s\ |d d
¡ }| dddddg¡}|r<t |¡}t |¡}|d |d d
< | ||¡ d S rW rZ )r r$ r% rA r\ r r r
|
Útest_setitem_slice s
|
|
z#BaseSetitemTests.test_setitem_slicec C s |d d
¡ }tj|dddddgd}tj| dddd d
|
g¡|jd}| ¡ }|d |jd d
< | ||¡ | ¡ }|d |jd d
< | ||¡ d S ) NrX ÚaÚbÚcÚdÚer r r- rY )r' r r! r[ r r? r9 r> )r r$ rA Úsr\ rz r r r
|
Útest_setitem_loc_iloc_slice* s z,BaseSetitemTests.test_setitem_loc_iloc_slicec C s: |d d
}t t¡ |d d
|d d
< W 5 Q R X d S )NrX r; r )r. r/ r0 ©r r$ rA r r r
|
Ú)test_setitem_slice_mismatch_length_raises7 s z:BaseSetitemTests.test_setitem_slice_mismatch_length_raisesc C s< |d d
¡ }|dd
|d d
< | ||dd
¡ d S )NrX éûÿÿÿ)r' Z assert_extension_array_equalr r r r
|
Útest_setitem_slice_array< s z)BaseSetitemTests.test_setitem_slice_arrayc C s: |d d
¡ }t t¡ |ddg |d< W 5 Q R X d S )NrX r r )r' r. r/ r0 r r r r
|
Ú&test_setitem_scalar_key_sequence_raiseA s z7BaseSetitemTests.test_setitem_scalar_key_sequence_raisec C sL | ¡ }|d d
}|d |d< |d |d ks4t|d |d ksHtd S r )Úviewr" )r r$ Zview1Zview2r r r
|
Ú test_setitem_preserves_viewsF s
|
z-BaseSetitemTests.test_setitem_preserves_viewsc C sP t dt |¡i¡ }}t j|jd}||}|d |j|df< | ||¡ d S )Nr$ r )r rG r! r r> rx )r r$ r rJ r\ rz Úkeyr r r
|
Ú,test_setitem_with_expansion_dataframe_columnO s
|
z=BaseSetitemTests.test_setitem_with_expansion_dataframe_columnc C s t d|d d
i¡}|d |jd< t d|d d
i¡}| ||¡ ||jd< t dt j|d |d |g|jdi¡}| ||¡ d S )Nr$ r )r r$ r; )r; r$ r r )r rG r> rx r! r )r r$ Zna_valuerJ r\ r r r
|
Ú test_setitem_with_expansion_rowY s
|
ÿz0BaseSetitemTests.test_setitem_with_expansion_rowc C s\ t j|dd}t j|jtdd}||}||j|< t j| t¡|jdtd}| ||¡ d S )Nr$ ©Úname)r r r )r r r )r r! r Úobjectr> Zastyper2 )r r$ r r3 rz r r\ r r r
|
Útest_setitem_seriesg s
|
ÿz$BaseSetitemTests.test_setitem_seriesc C sÜ t d|i¡}t|jt jjj}t jjj }|jj
|
d }| ¡ }||j d d
< | ||¡ |j d d
|j d d
< | ||¡ |j|j d d
< | ||¡ |s´|s´|jj
|
d |ks´t|jd d
|j d d
< | ||¡ d S )NrD r éÿÿÿÿ)r rG Ú
|
isinstanceZ_mgrÚcoreZ internalsZ ArrayManagerÚoptionsÚmodeZ copy_on_writeZarraysr' r? rx Úvaluesr" )r r$ rJ Zusing_array_managerZusing_copy_on_writeZblk_dataÚorigr r r
|
Ú test_setitem_frame_2d_valuesv s
|
z-BaseSetitemTests.test_setitem_frame_2d_valuesc C sF t j|dd}t t|¡}t |d¡}|| }|d= | ||¡ d S )Nr$ r r )r r! r rH r Údeleter2 )r r$ r3 Ztakerr\ r r r
|
Útest_delitem_series s z$BaseSetitemTests.test_delitem_seriesc C sX d}t jttf|d ||d< W 5 Q R X t jttf|d ||d d
< W 5 Q R X d S )NÚ r+ r )r. r/ r0 Ú TypeError)r r$ Zinvalid_scalarrm r r r
|
Útest_setitem_invalid s
|
z%BaseSetitemTests.test_setitem_invalidc C s | ¡ }t ||d¡}|jddgd d
f j|jddgd d
f< |jdd d
f |d k ¡ sdt|jdd d
f |d k ¡ std S )N)r r r r )r' r rG r> r rd r" )r r$ r) rJ r r r
|
Útest_setitem_2d_values¦ s
|
*"z'BaseSetitemTests.test_setitem_2d_values)2Ú__name__Ú
|
__module__Ú __qualname__r. Zfixturer r& r* r6 r: r< ÚmarkZ parametrizerB rK rO rP rR rT rU r r7 r rc r] r` re rg r Zxfailrn ru rv r{ r| r} r~ r r r r r r r r r r r r¡ r¢ r r r r
|
r sª ù
|
ùö
|
|
ýù
|
*ý
|
ÿú
|
ö
|
|
|
|
r ) Únumpyr r. Zpandasr Zpandas._testingZ_testingri Z pandas.tests.extension.base.baser r r r r r
|
Ú<module> s
|
|
