U
|
ý°d"´ ã @ s@ d dl m Z d dlZd dlZd dlm mZ d dlm Z
|
d dl m Z m Z mZ d dlmZmZmZmZ d dlZd dlmZmZmZmZmZmZm Z m Z m Z m Z m Z m!Z!m"Z"m#Z#m$Z$m%Z% d dl&m'Z( d dl)m*Z* d dl+m,Z, G d d
|
d
|
Z-G d d d Z.G d d dZ/G dd dZ0G dd dZ1G dd dZ2G dd dZ3dS )é )ÚdatetimeN)Ú _registry)Úis_categorical_dtypeÚis_interval_dtypeÚis_object_dtype)ÚCategoricalDtypeÚDatetimeTZDtypeÚ IntervalDtypeÚ PeriodDtype)Ú CategoricalÚ DataFrameÚ DatetimeIndexÚIndexÚIntervalÚ IntervalIndexÚ
|
MultiIndexÚNaTÚPeriodÚ PeriodIndexÚSeriesÚ TimestampÚcutÚ
|
date_rangeÚnotnaÚ period_range)Ú SparseArray)ÚBDayc @ sü e Zd Zdd Zej ddddddd g¡d
|
d Zd d Zdd Z dd Z
|
dd Z dd Z dd Z dd Zdd Zd d Zd d Zd d! Zd"d# Zej d$ed%ed&fed'd(d)edd*fed+d,d-ed,d-fg¡d.d/ Zej d0d1d2 ej D d3d4g ¡d5d6 Z d7d8 Z ej d9d:d;d<d=d>d&d?g¡d@dA Z ej d9d:d;d<d=d>d&d?g¡dBdC Z dDdE Z!dFdG Z"dHdI Z#ej ddJdKdLg¡dMdN Z$ej dOdPdPgg¡dQdR Z%dSdT Z&dUdV Z'ej dWdXdYdZgdXdXdXgg¡d[d\ Z(d]d^ Z)d_d` Z*dadb Z+dcdd Z,dedf Z-ej dgdhd>d>dXgdidjdkdlgdlfd>dhd>dXgdidjdkdlgdlfdhdPdPdXgdidjdkdlgdlfdhdPdXgdidjdkgdkfdPdhdXgdkdjdigdifdhdXdPgdkdjdigdjfg¡dmdn Z.dodp Z/ej dqdrdPdhd>gdse0dsdsdsdsgdsdsdsdsgdsdsdsdsggdrdPdhd>gdtfdhd>gdsduge0didjdsdugdkdldsdugd(dvdsduggdrdPdhd>gdtfdrdPdhge1j2dsdudwge1j3dxe0dsdudwgdsdudwgdsdudwggdrdPdhgdtfdPdhd>gdsdudwgdydzd{gd|d}d~gge0didsdudwgdkdydzd{gd(d|d}d~ggdrdPdhd>gdtfdhdrd>ge1j2dsdudwgdydzd{gd|d}d~gge1j3dxe0dudjdsdwgdzdldyd{gd}dvd|d~ggdrdPdhd>gdtfdrdhge0dsdugdwdygdzd{ggdrdhgdte0dsdjdugdwdldygdzdvd{ggdrdPdhgdtfg¡dd Z4dd Z5dd Z6d
d Z7dd Z8dd Z9dd Z:e;j<dd Z=ej di ddXig¡dd Z>dd Z?dd Z@dd ZAdS )ÚTestDataFrameSetItemc C sv G dd dt }dg}t|}tddgi|d}d|d< d ||d
|
< tddgddg|d
|
d gi|d}t ||¡ d S ) Nc @ s e Zd ZdS )z@TestDataFrameSetItem.test_setitem_str_subclass.<locals>.mystringN)Ú__name__Ú
|
__module__Ú __qualname__© r! r! ú_d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/indexing/test_setitem.pyÚmystring1 s r# z2020-10-22 01:21:00+00:00Úaé ©Úindexé Úbé Úc)Ústrr r ÚtmZ assert_equal)Úselfr# Údatar' ÚdfÚexpectedr! r! r" Útest_setitem_str_subclass/ s "z.TestDataFrameSetItem.test_setitem_str_subclassÚdtypeÚint32Úint64Úuint32Zuint64Úfloat32Úfloat64c C s: t j t|¡}t j||d||< || jj|ks6td S )N©r3 )ÚnpÚrandomÚrandnÚlenÚarrayr3 ÚnameÚAssertionError)r. r3 Ú float_frameÚarrr! r! r" Útest_setitem_dtype< s z'TestDataFrameSetItem.test_setitem_dtypec C s8 t j t|d¡}||ddg< t |ddg j|¡ d S )Nr( ÚAÚB)r: r; r< r= r- Zassert_almost_equalÚvalues)r. rA r/ r! r! r" Ú test_setitem_list_not_dataframeE s z4TestDataFrameSetItem.test_setitem_list_not_dataframec C sÐ t dddgdddgdtdd d
|
gd d d }tddddgtdd d
|
dgd d dd}d}tjt|d ||d< W 5 Q R X t tj ddd¡dd d
|
dgd}d}tjt|d | d d
|
g¡
|
¡ |d< W 5 Q R X d S ) Nr% r( r* ÚdÚeÚf)ÚbarZbazr$ r) r+ Úfoo©r? r& ÚgÚhÚiÚjZfiz©r' r? z/cannot reindex on an axis with duplicate labels©ÚmatchZnewcolr )é rU ©ÚcolumnszDCannot set a DataFrame with multiple columns to the single column grZgr) r r r ÚpytestÚraisesÚ
|
ValueErrorr: r; ÚrandintÚgroupbyÚcount)r. r0 ÚserÚmsgr! r! r" Útest_setitem_error_msmgsJ s þ
|
ý z-TestDataFrameSetItem.test_setitem_error_msmgsc C sh d}d}t t|d}tj |¡}t|D ] }|||< q*t t ||¡ ||¡t|d}t ||¡ d S )Né
|
é r& ) r Úranger: r; r< ÚrepeatZreshaper- Úassert_frame_equal)r. ÚNÚKr0 Znew_colrP r1 r! r! r" Útest_setitem_benchmark` s
|
z+TestDataFrameSetItem.test_setitem_benchmarkc C s^ t tj dd¡t d¡dddgd}| dd|d ¡ | d d
|
|d ¡ |d d ¡|d < |j}tt d ¡gd t d ¡g ddd
|
ddd gd}t
|
||¡ |d d ¡|d< |j}tt d ¡gd t d ¡gd ddd
|
ddd gd}t
|
||¡ |d d¡|d< |j}tt d ¡gd t d ¡gd t d¡g ddd
|
ddd dgd}t
|
||¡ d S )Nrb r* r+ r) r$ ©r' rW r rL r( rK r7 Úxr8 r& rU r4 Úy) r r: r; r< ÚarangeÚinsertÚastypeÚdtypesr r3 r- Úassert_series_equal©r. r0 Úresultr1 r! r! r" Ú test_setitem_different_dtypek s8 ÿþ þ *þz1TestDataFrameSetItem.test_setitem_different_dtypec C sT t dddgd}|j|d< dddg|d< t ddddgidddgd }t ||¡ d S )
|
NrD rE ÚCr& ÚXrj rk Úz)r/ r' )r r' r- re ©r. r0 Úexpr! r! r" Útest_setitem_empty_columns s
|
|
z/TestDataFrameSetItem.test_setitem_empty_columnsc C sF t dddd}tt t|¡d}||d< |d jt d¡ksBtd S )Nz1/1/2000 00:00:00z1/1/2000 1:59:50Z10s)Úfreqr& rD úM8[ns])r r r: rl r= r3 r@ )r. Úrngr0 r! r! r" Ú%test_setitem_dt64_index_empty_columns s z:TestDataFrameSetItem.test_setitem_dt64_index_empty_columnsc C sT t tdd}tddd|d< t tdddggd ddd gdgd
|
}t ||¡ d S ) Nr* r& Z20130101ÚUTC©ÚtzÚnowr r% r( ri )r rc r r- re ©r. r0 r1 r! r! r" Ú$test_setitem_timestamp_empty_columns s ÿz9TestDataFrameSetItem.test_setitem_timestamp_empty_columnsc C sp t ddddddgdddg¡}ttddgd }d
|
t| d t| d }tjt|d ||d< W 5 Q R X d S )Nr r% r( r$ r) r+ ra rK rV zLength of values \(z$\) does not match length of index \(z\)rS rL )r Z
|
from_codesr rc r= rX rY rZ )r. Úcatr0 r_ r! r! r" Ú2test_setitem_wrong_length_categorical_dtype_raises§ s ÿzGTestDataFrameSetItem.test_setitem_wrong_length_categorical_dtype_raisesc C sP t dddgdddgd}tddd g}||d
|
< t|d
|
d }t |d
|
|¡ d S ) Nr$ r) r+ ç ð?ç @ç @©Zc_1Zn_1r r% Ú
|
new_columnrM )r r r r- rp )r. r0 Zsp_arrayr1 r! r! r" Ú test_setitem_with_sparse_value³ s
|
z3TestDataFrameSetItem.test_setitem_with_sparse_valuec C sh t dddgdddgd}ttddd gd
|
d dgd }||d < ttd ddgd d }t |d |¡ d S )Nr$ r) r+ r r r r r r% r( r& r rM )r r r r- rp )r. r0 Z sp_seriesr1 r! r! r" Ú(test_setitem_with_unaligned_sparse_value¼ s
|
z=TestDataFrameSetItem.test_setitem_with_unaligned_sparse_valuec C s8 t ddg}tg }||d< td|i}t ||¡ d S )Nz2003-12ÚDr$ )r r r- re )r. r/ rr r1 r! r! r" Ú#test_setitem_period_preserves_dtypeÄ s
|
z8TestDataFrameSetItem.test_setitem_period_preserves_dtypec C sÐ t tdddgddtdddgtdtdddgtdtdddgddd}t tg ddtg tdtg tdtg ddd}tdddgD ]4\}}t|t|t|t |¡d|j|jd < qt
|
||¡ d S ) Nr r% r( r5 r9 r* r6 ©r$ r) r+ rH ) r r ÚfloatÚ enumerateÚintr: r6 ÚlocÚshaper- re )r. r1 r0 Úidxr) r! r! r" Ú"test_setitem_dict_preserves_dtypesÎ s* üÿ
|
|
|
|
üÿüz7TestDataFrameSetItem.test_setitem_dict_preserves_dtypesz obj,dtypez2020-01ÚMr rb )ÚleftÚrightr z
|
2011-01-01z
|
US/Easternr c C sL t dddgt|gd |dd}t ddddgi}||d< t ||¡ d S )Nr% r( r* r9 )r Úobjr r ©r r r- re )r. r r3 r1 r0 r! r! r" Ú test_setitem_extension_typesé s "z1TestDataFrameSetItem.test_setitem_extension_typesÚea_namec C s g | ]}t |jts|jqS r! )Ú
|
isinstancer? Úproperty)Ú.0r3 r! r! r" Ú
|
<listcomp>ÿ s ýz TestDataFrameSetItem.<listcomp>zdatetime64[ns, UTC]z period[D]c C s8 t dg}dg||< t ddg|dgi}t ||¡ d S )Nr r% ©r r- re )r. r rr r1 r! r! r" Útest_setitem_with_ea_nameý s
|
|
z.TestDataFrameSetItem.test_setitem_with_ea_namec C s t jddgdd}t| ¡ }||d< tdd gdd gddd}t ||¡ t jddgdd}||d< t |d |d ¡ t |d ¡ |¡ d S ) Nr% Znatúdatetime64[ns]r9 Únew)r r¥ ú datetime64[s]r )
|
r: r> r Zto_framer r- re rp Úassert_numpy_array_equalZto_numpy)r. Zdata_nsrr r1 Zdata_sr! r! r" Ú6test_setitem_dt64_ndarray_with_NaT_and_diff_time_units s zKTestDataFrameSetItem.test_setitem_dt64_ndarray_with_NaT_and_diff_time_unitsÚunitrO ÚmÚsÚmsr ÚYc C s d}t d| d¡}t j|t jd |¡}|dkr:|}n
|
| d¡}tdt |¡it |¡d}|||< || j|jks|t|| j|k ¡ std S ) Néd úM8[ú]r9 )r« r¬ r¦ Úintsr& )
|
r: r3 rl r5 Úviewrn r r@ rF Úall©r. r© Únr3 ÚvalsZex_valsr0 r! r! r" Ú-test_frame_setitem_datetime64_col_other_units s
|
zBTestDataFrameSetItem.test_frame_setitem_datetime64_col_other_unitsc C s d}t d| d¡}t j|t jd |¡}| d¡}tdt |¡it |¡d}t j|t jd d¡|d < ||d < |d j|k ¡ st d S )
|
Nr® r¯ r° r9 r¤ r± r& r{ Údates)
|
r: r3 rl r5 r² rn r rF r³ r@ r´ r! r! r" Ú6test_frame_setitem_existing_datetime64_col_other_units1 s
|
zKTestDataFrameSetItem.test_frame_setitem_existing_datetime64_col_other_unitsc
|
C s |}|d d¡}||d< t |d t|dd¡ d|d< ||d< t |d t|dd¡ |d= |jjd }|jjd }t ||¡ |jj}|jj}|d ks°t |t |ks°t
|
| ¡ }t |j d< t |j d < |d } t t| td
|
d d
|
gdd¡ t |j|j¡ d S ) NrE rL rt rM r r% r( )r% r% ©r% r( TF)Úrenamer- rp r Ú_mgrZarraysZ assert_extension_array_equalZ_ndarrayÚbaseÚidr@ Úcopyr Úilocr ro )
|
r. Ztimezone_framer0 r Zv1Zv2Zv1baseZv2baseÚdf2rr r! r! r" Útest_setitem_dt64tzB s(
|
|
z(TestDataFrameSetItem.test_setitem_dt64tzc C s t dddd}ttj dd¡|d}||d< t|d }tj||dd |jdksVt |jdksdt |
|
¡ d¡}t |j tst t |j |¡ d S )
|
Nz1/1/2000rb r' )Úperiodsr? r* r& r F©Z check_names)r r r: r; r< r r- Úassert_index_equalr? r@ Z reset_indexZ set_indexr r' r )r. r| r0 Úrsr! r! r" Útest_setitem_periodindex` s z-TestDataFrameSetItem.test_setitem_periodindexc C sÌ t dddgdddgd}t ddgddgddgg¡}||d d
|
g< t dddgdddgdddgdddgd }|d |j¡|d < |d
|
|j¡|d
|
< |d j|jks¨t|d
|
j|jks¼tt ||¡ d S ) NZoneZtwoZthreer% r( r* ©r$ r) rb r+ rH r )r r: r> rn r3 r@ r- re ©r. r0 rB r1 r! r! r" Ú'test_setitem_complete_column_with_arrayn s üÿz<TestDataFrameSetItem.test_setitem_complete_column_with_arrayÚf8Úi8Úu8c C sx t dddg|d}ttj dd¡|d}dddg|d < t dddd gtd}|d
|
krft d d d d gtd}t |j|¡ d S )Nr% r( r* r9 rV r$ r) r+ FrË r r r ) r r r: r; r< Úobjectr- rÅ rW )r. r3 Úcolsr0 Z expected_colsr! r! r" Ú$test_setitem_bool_with_numeric_index s z9TestDataFrameSetItem.test_setitem_bool_with_numeric_indexÚindexerrE c C sb t ddgd}t dddgi}|||< t tjgd ddgd}|d d¡|d< t ||¡ d S )NrD rE rV r% r( ©rD rE rÎ )r r: Únanrn r- re )r. rÑ r0 Úotherr1 r! r! r" Ú#test_setitem_frame_length_0_str_key s z8TestDataFrameSetItem.test_setitem_frame_length_0_str_keyc C s¾ dddgd }t td|d}d|jd< d |jd d
df< d
|
|d< t d d ddd dgtjd dtjd dgtjd dtjd dggdd}|d tj¡|d< |d tj¡|d< ||_t ||¡ d S )NrD rE rt r( r* ri )r r* )r rD )r% rU ©r( rb r r% rU rb rÎ r9 )
|
r rc r r: rÓ rn r5 rW r- re )r. rÏ r0 r1 r! r! r" Ú$test_setitem_frame_duplicate_columns s
|
ýú
|
z9TestDataFrameSetItem.test_setitem_frame_duplicate_columnsc C s dddgd }t td|d}tjtdd d |dg< W 5 Q R X |jd d
d d
f }tjtdd d |dg< W 5 Q R X d S )
|
NrD rE rt r( r* ri ú"Columns must be same length as keyrS )r r* rb )r rc rX rY rZ rÀ )r. rÏ r0 rÁ r! r! r" Ú2test_setitem_frame_duplicate_columns_size_mismatch° s zGTestDataFrameSetItem.test_setitem_frame_duplicate_columns_size_mismatchrÏ r$ r) r+ c C sT t dddgg|d}t ddggddgd}d }tjt|d
|
||d < W 5 Q R X d S ) Nr% r( r* rV ra é rH rI rØ rS r$ ©r rX rY rZ )r. rÏ r0 Úrhsr_ r! r! r" Ú#test_setitem_df_wrong_column_number» s
|
z8TestDataFrameSetItem.test_setitem_df_wrong_column_numberc C s t dddggdddgd}t ddd ggdddgd}||ddg< t ddd ggdddgd}t ||¡ ||d
|
dg< t ddd dggdddd
|
gd}t ||¡ d S ) Nr% r( r* r$ r) rV ra rÚ é r+ r¢ )r. r0 rÜ r1 r! r! r" Ú/test_setitem_listlike_indexer_duplicate_columnsÄ s zDTestDataFrameSetItem.test_setitem_listlike_indexer_duplicate_columnsc C s^ t dddggdddgd}t ddggddgd}d }tjt|d
|
||ddg< W 5 Q R X d S ) Nr% r( r* r$ r) rV ra rÚ rØ rS rÛ )r. r0 rÜ r_ r! r! r" Ú@test_setitem_listlike_indexer_duplicate_columns_not_equal_lengthÐ s
|
zUTestDataFrameSetItem.test_setitem_listlike_indexer_duplicate_columns_not_equal_lengthc C sº t dtdi}t|d d}t|jjts0t||d< t |¡|d< |j
|
|d< t |j
|
¡|d< | t ¡|d< t |d jstt|d jjstt |d js¦tt|d jjsºtt|d sÊtt|d sÚtt|d sêtd d
|
}t ||j||j¡ tj||j||jd d tj||j||jd d tj||j||jd d t |d |d ¡ tj|d |d d d t |d |d ¡ tj|d |d d d d S ) NrD ra rb rE rt r ÚEÚFc S s t t | ¡S ©N)r r: r> ©rj r! r! r" Ú<lambda>ö ó z=TestDataFrameSetItem.test_setitem_intervals.<locals>.<lambda>FrÄ )r rc r r r Ú
|
categoriesr r@ r: r> rF rn rÎ r r3 r r r- rÅ rE rt r rp )r. r0 r^ r+ r! r! r" Útest_setitem_intervalsØ s0
|
z+TestDataFrameSetItem.test_setitem_intervalsc C s\ t dttd d¡i}|jj ddg¡|_t dtddg ddg¡i}t ||¡ d S )NrO ÚmnÚcategoryrµ rª )
|
r r Úlistrn rO r Zreorder_categoriesr r- re r r! r! r" Útest_setitem_categorical s ÿz-TestDataFrameSetItem.test_setitem_categoricalc C sD t g dd}tdg|d}g |d< tdg|d}t |j|j¡ d S )Nr rM rD ©rW r' )r r r- rÅ r' )r. r' rr r1 r! r! r" Ú test_setitem_with_empty_listlike s
|
z5TestDataFrameSetItem.test_setitem_with_empty_listlikezcols, values, expectedrt r% r( r* rU c C s8 t |g|d}|d |d< |d jd }||ks4td S )NrV r$ r )r rF r@ )r. rÏ rF r1 r0 rr r! r! r" Útest_setitem_same_column s z-TestDataFrameSetItem.test_setitem_same_columnc C s$ dddgddgdddgf}t |¡}td d
|
d }tj d d t|t|f¡}t|||d}|jj
|
¡ |d d d
}}tj |¡ |d j |d d d
f |d< t |d |d ¡ tj |¡ |d j ||f |d< t |d |d ¡ tj |¡ |d j ||f |d< t |d |d ¡ d S )NZjimÚjoeÚjolieÚfirstÚlastr Úcenterr Z20141006é )rà r% iè rí éÿÿÿÿ)rñ ró )rð rò )rñ rò )rð ró )r Z from_productr r: r; r[ r= r r' rF r¿ Úshuffler r- re )r. ÚitrÏ r' r¶ r0 rP rQ r! r! r" Útest_setitem_multi_index' s
|
z-TestDataFrameSetItem.test_setitem_multi_indexzcolumns,box,expectedrD é rV é é é r9 ra rÚ rÞ é é é c C s: t ddgddgddggddgd }|||< t ||¡ d S )
|
Nr% r( r* rU rb rü rD rE rV r¢ )r. rW Úboxr1 r0 r! r! r" Ú!test_setitem_list_missing_columns? s 3"z6TestDataFrameSetItem.test_setitem_list_missing_columnsc C sF t t|d |d }||d< |d }t||jdd}t ||¡ d S )NrD rE ÚtuplesrR )rë Úzipr r' r- rp )r. rA r rr r1 r! r! r" Útest_setitem_list_of_tuplesv s
|
z0TestDataFrameSetItem.test_setitem_list_of_tuplesc C s\ t dddgdddgd}dd d
|
D }d|j|< t dddgdddgd}t ||¡ d S ) Nr% r( r* rU rb rü rÈ c s s | ]
|
}|V qd S rã r! ©r rj r! r! r" Ú <genexpr> s zCTestDataFrameSetItem.test_setitem_iloc_generator.<locals>.<genexpr>rº ©r rÀ r- re ©r. r0 rÑ r1 r! r! r" Útest_setitem_iloc_generator~ s
|
|
z0TestDataFrameSetItem.test_setitem_iloc_generatorc C s` t dddgdddgd}dd d
|
D }d|j|df< t dddgdddgd}t ||¡ d S ) Nr% r( r* rU rb rü rÈ c s s | ]
|
}|V qd S rã r! r r! r! r" r s zSTestDataFrameSetItem.test_setitem_iloc_two_dimensional_generator.<locals>.<genexpr>rº r r r! r! r" Ú+test_setitem_iloc_two_dimensional_generator s
|
z@TestDataFrameSetItem.test_setitem_iloc_two_dimensional_generatorc C s t ddd}t|d}t d|tjd|d< t d|dd|d< t d |dd|d < t d
|
|tjd|d
|
< |j}t tjtttjgtd d}t ||¡ d S ) Nr¾ ZS24)r? r3 r& r$ )r? r' r3 r) ZS64r+ rH Zabcd)
|
r r r: r6 Zuint8ro rÎ rë r- rp )r. r' r0 rr r1 r! r! r" Ú(test_setitem_dtypes_bytes_type_to_object s
|
z=TestDataFrameSetItem.test_setitem_dtypes_bytes_type_to_objectc C s t ddgddgd ddd¡}td|jd }|d
|
|j|d
|
f< |d |j|d f< t ddgddgd ddd¡}t ||¡ d S ) Nr* rU rb rü rÈ r5 ÚInt64Fr& r$ r) )r rn r r' r r- re )r. rr Úmaskr1 r! r! r" Ú test_boolean_mask_nullable_int64 s ÿÿz5TestDataFrameSetItem.test_boolean_mask_nullable_int64c C sH t dddgi}tddgdd|d< t dddgidd}t ||¡ d S )Nr$ r% r( r r9 r r r! r! r" Ú test_setitem_ea_dtype_rhs_series¦ s z5TestDataFrameSetItem.test_setitem_ea_dtype_rhs_seriesc C sz t t d¡t d¡dt d¡d}t d¡}t t d¡d}||d< t t¡ t |¡|d< W 5 Q R X t ||¡ d S )Nra )únp-arrayú np-matrixr& )ra r% r r ) r r: Úonesrl r- Zassert_produces_warningÚPendingDeprecationWarningZmatrixre )r. r1 r$ r0 r! r! r" Útest_setitem_npmatrix_2d® s ÿ
|
z-TestDataFrameSetItem.test_setitem_npmatrix_2dr¶ rH c C s t ddgddgd|}ddd|jd d
d f< d
|
d d |jd d
d f< dd
|
i|jd d
df< t ddgddgd|d
|
d gtjd
|
gd}t ||¡ d S )Nr% r( r* rU rÈ r® éÈ )r% r r$ rb rü ©r r% r+ rI )r+ rI )r r r: rÓ r- re )r. r¶ r0 r1 r! r! r" Ú%test_setitem_aligning_dict_with_indexÀ s &ÿz:TestDataFrameSetItem.test_setitem_aligning_dict_with_indexc C s t dddgi}t dddgiddgd|d< t dtjdgi}t ||¡ t dddgi}| dt dddgiddgd¡ t ||¡ d S )Nr$ r% r( ra rÚ r& r )r r: rÓ r- re Zisetitemr r! r! r" Útest_setitem_rhs_dataframeÌ s z/TestDataFrameSetItem.test_setitem_rhs_dataframec C sb t ddgddgddggd}t ddd gi|d
|
|d< t tdd g|d
|
ddgd }t ||¡ d S ) Nr$ r) r% r( r* rU ©rW r/ ra rÚ r9 rÈ r ©r. Zany_numeric_ea_dtyper0 r1 r! r! r" Ú*test_setitem_frame_overwrite_with_ea_dtype× s þÿz?TestDataFrameSetItem.test_setitem_frame_overwrite_with_ea_dtypec C s8 t ddgi}| ¡ }d}||g ||< t ||¡ d S )NrÈ ra ©r r¿ r- re )r. r0 r1 Zcol_namer! r! r" Ú test_setitem_frame_midx_columnsã s
|
z4TestDataFrameSetItem.test_setitem_frame_midx_columnsN)Br r r r2 rX ÚmarkÚ parametrizerC rG r` rh rs ry r} r r
r r r r r r
|
r r r r r Ú ea_registryro r£ r¨ r· r¹ r rÇ rÊ rÐ rÕ r× rÙ rÝ rß rà rè rì rî rï rù r r: r> r5 r r r
|
r r r r ÚtdZ&skip_array_manager_not_yet_implementedr r r r r r! r! r! r" r . sþ ÿ
|
#
|
|
|
þýþ
|
þúþ
|
|
|
|
|
|
|
, úþ
|
|
|
þý
|
þý (ý
|
þý &
|
þý ÿýÚþ1
|
|
|
r c @ s@ e Zd Zejdd Zejdd Zdd Zdd Zd d
|
Z d S ) ÚTestSetitemTZAwareValuesc C s t ddgdd}| d¡}|S )Nz2013-1-1 13:00z2013-1-2 14:00rE rM z
|
US/Pacific)r Ú tz_localize)r. Znaiver r! r! r" r í s
|
z TestSetitemTZAwareValues.idxc C s. t tj| ¡ dddd}|j|jks*t|S )NrÎ r9 rE rM )r r: r> Útolistr3 r@ )r. r r1 r! r! r" r1 ó s z!TestSetitemTZAwareValues.expectedc C sl t tj dd¡dgd}||d< |jddgdj d ¡|d< |d }t| d¡ d ¡dd }t
|
||¡ d S )
|
Nr( r% rD rV rE r r& r~ rM ) r r: r; r< Z to_seriesÚdtZ
|
tz_convertr r$ r- rp )r. r r1 r0 rr Úcompr! r! r" Útest_setitem_dt64seriesù s z0TestSetitemTZAwareValues.test_setitem_dt64seriesc C s8 t tj dd¡dgd}||d< |d }t ||¡ d S ©Nr( r% rD rV rE )r r: r; r< r- rp ©r. r r1 r0 rr r! r! r" Útest_setitem_datetimeindex s z3TestSetitemTZAwareValues.test_setitem_datetimeindexc C s< t tj dd¡dgd}| ¡ |d< |d }t ||¡ d S r) )r r: r; r< Z to_pydatetimer- rp r* r! r! r" Ú.test_setitem_object_array_of_tzaware_datetimes s zGTestSetitemTZAwareValues.test_setitem_object_array_of_tzaware_datetimesN)
|
r r r rX Zfixturer r1 r( r+ r, r! r! r! r" r# ì s
|
|
|
r# c @ sT e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dS )Ú!TestDataFrameSetItemWithExpansionc C s t dddgdddgd}|d }t dd gd
|
d gd d gg¡|d dg< d|jd< |rhtdddgdd}ntdddgdd}t ||¡ d S )Nr% r( r* rU rü rÈ r$ g¹?gÉ?g333333Ó?gÙ?ç à?r+ rH r® )r r rM )r r: r> rÀ r r- rp )r. Úusing_copy_on_writer0 r^ r1 r! r! r" Útest_setitem_listlike_views s $
|
z=TestDataFrameSetItemWithExpansion.test_setitem_listlike_viewsc C sR t ddgddgg}ddg|d< t dddgdddggdddgd }t ||¡ d S )
|
Nr% r( r* rU rb rü z0 - Namer rV r¢ r r! r! r" Ú.test_setitem_string_column_numpy_dtype_raising* s "zPTestDataFrameSetItemWithExpansion.test_setitem_string_column_numpy_dtype_raisingc C sf t dddgdd}ttd|jd d
df< t dtjtjgdtjtjggdddgd}t ||¡ d S ) Nr$ r) r8 ©rW r3 r( r r% rV )r rë rc r r: rÓ r- re )r. r/ r0 r1 r! r! r" Ú'test_setitem_empty_df_duplicate_columns1 s ÿzITestDataFrameSetItemWithExpansion.test_setitem_empty_df_duplicate_columnsc C s> t dtjtj ddd¡ddi}tdd tddd D }|jdgd
|
d }t|j tdd d d |d}|j
|
}||d< t | |j }t t d¡t|d dgddgd}t ||¡ ||d< t | |j }t t d¡t|d dt|d dgdddgd}t ||¡ |d }|d }t |jj|¡ d|_t | ¡ | ¡ ¡ d S )NÚvaluer i' r® r4 r9 c S s g | ]}| d |d qS )z - ió r! ©r rP r! r! r" r¡ ? s zcTestDataFrameSetItemWithExpansion.test_setitem_with_expansion_categorical_dtype.<locals>.<listcomp>iô T)ZbyZ ascendingi) F)r Úlabelsr )rç Zorderedr& rá )r r: r> r; r[ r rc Ú sort_valuesr r4 rF r, ro r r3 r r- rp Zassert_categorical_equalr¼ r? Z
|
sort_index) r. r0 r6 r^ r rr r1 Zresult1Zresult2r! r! r" Ú-test_setitem_with_expansion_categorical_dtype: s> ÿþ
|
|
ýú zOTestDataFrameSetItemWithExpansion.test_setitem_with_expansion_categorical_dtypec C s2 t }d|d< t dgd tj¡}t ||¡ d S )Nr% rL rV )r rn r: r5 r- re r r! r! r" Ú test_setitem_scalars_no_indexg s z?TestDataFrameSetItemWithExpansion.test_setitem_scalars_no_indexc C sL d|j kst|d |d< d|j ks(t|d }|d }tj||dd d S )NrÒ rD FrÄ )rW r@ r- rp )r. rA rr r1 r! r! r" Ú test_setitem_newcol_tuple_keyn s ý z?TestDataFrameSetItemWithExpansion.test_setitem_newcol_tuple_keyc C sX t ddt d}t|tdd}tddd}t|}tj||< t || ¡ ¡ sTt
|
d S ) Nz1/1/2012z2/1/2012)ÚstartÚendrz ra rí iÜ rÚ r% ) r r r rc r r r: rÓ Úisnanr³ r@ )r. rW r/ ÚtÚtsr! r! r" Ú#test_frame_setitem_newcol_timestampz s
|
zETestDataFrameSetItemWithExpansion.test_frame_setitem_newcol_timestampc C sh t dddgi}|j|d< d|jddgdf< |jdgd d
f }t dgdgddgd}t ||¡ d S ) Nr$ r) r® FTr% rÈ r& )r r' r r- re rq r! r! r" Ú*test_frame_setitem_rangeindex_into_new_col s
|
zLTestDataFrameSetItemWithExpansion.test_frame_setitem_rangeindex_into_new_colc C sh t ddgddgddggd}t ddd gi|d
|
|d < t ddgddgtdd g|d
|
d }t ||¡ d S ) Nr$ r) r% r( r* rU r ra rÚ r9 r+ ©r$ r) r+ r r r! r! r" Ú test_setitem_frame_keep_ea_dtype s ýÿzBTestDataFrameSetItemWithExpansion.test_setitem_frame_keep_ea_dtypeN) r r r r0 r1 r3 r8 r9 r: r@ rA rC r! r! r! r" r- s - r- c @ sâ e Zd Zdd Zej dejej g¡ej de
|
e j e ej g¡ej ddddg¡d d
|
Zej de
|
e j e ej g¡ej ddddg¡d d Zej dejej g¡ej de
|
e j e ej g¡ej ddddg¡d d ZdS )ÚTestDataFrameSetItemSlicingc C sH t t d¡}d|dd
< t d¡}d|dd
< t |}t ||¡ d S )N)r® r% r% éüÿÿÿ)r r: Úzerosr- re rÉ r! r! r" Útest_setitem_slice_position s
|
z7TestDataFrameSetItemSlicing.test_setitem_slice_positionrÑ r rµ r% r( r* c C s~ t dddggdddgg| ddd gd
|
}|d d d g||dd
< t dddggd d d gg| ddd gd
|
}t ||¡ d S ©Nr% r* rb r( rU rü r$ r) r+ rV ra rÚ rÞ r¢ ©r. rµ r rÑ r0 r1 r! r! r" Ú+test_setitem_slice_indexer_broadcasting_rhs¤ s **zGTestDataFrameSetItemSlicing.test_setitem_slice_indexer_broadcasting_rhsc C s t dddggdddgg| ddd gd
|
}|d d d g|jttd|d < t dddggd d d gg| ddd gd
|
}t ||¡ d S rH )r rÀ rë rc r- re )r. rµ r r0 r1 r! r! r" Ú*test_setitem_list_indexer_broadcasting_rhs® s *"*zFTestDataFrameSetItemSlicing.test_setitem_list_indexer_broadcasting_rhsc C s t dddgdddggddd gg| d
|
d d gd }|dddg||dd
< t dddggdddgg|d d
|
d d gdd}t ||¡ d S )Nr% r* rb rj rk rv r( rU rü r$ r) r+ rV ra rÚ rÞ rÎ r2 r¢ rI r! r! r" Ú0test_setitem_slice_broadcasting_rhs_mixed_dtypes· s " ÿ ýzLTestDataFrameSetItemSlicing.test_setitem_slice_broadcasting_rhs_mixed_dtypesN)r r r rG rX r r r- ÚsetitemrÀ r r: r> rë ÚpdrJ rK rL r! r! r! r" rD s rD c @ s e Zd Zdd Zdd ZdS )Ú TestDataFrameSetItemCallablec C s` t ddddgddddgd }d
|
d d d g|dd < t d
|
d d d gddddgd }t ||¡ d S )Nr% r( r* rU rb rü rú rû rÒ rÚ rÞ rþ rÿ c S s dS )NrD r! rä r! r! r" rå Ì ræ zDTestDataFrameSetItemCallable.test_setitem_callable.<locals>.<lambda>r¢ rw r! r! r" Útest_setitem_callableÉ s z2TestDataFrameSetItemCallable.test_setitem_callablec C sL dd }t ddgddgg}|||dk< t d|g|dgg}t ||¡ d S )Nc S s | d S )Nr% r! rä r! r! r" ÚincÓ s zETestDataFrameSetItemCallable.test_setitem_other_callable.<locals>.incrö r% r r¢ )r. rQ r0 r1 r! r! r" Útest_setitem_other_callableÑ s
|
z8TestDataFrameSetItemCallable.test_setitem_other_callableN)r r r rP rR r! r! r! r" rO È s rO c @ sþ e Zd Zejejjddd dd gddgddd Zejj d
|
d ej d e
|
e j e g¡d d Zej d e
|
e j e g¡dd Zej dejejg¡dd Zdd Zej dddg¡ej di ddgid d gig¡d d Zd d! Zd"d# Zd$d% Zd&S )'Ú TestDataFrameSetItemBooleanMaskÚ mask_typec C s | t | ¡d kS ©Nr( )r: Úabs©r0 r! r! r" rå á ræ z(TestDataFrameSetItemBooleanMask.<lambda>c C s | t | ¡d kjS rU )r: rV rF rW r! r! r" rå á ræ Z dataframer> )Úidsc C s^ | ¡ }||}| ¡ }tj||< |j ¡ }tj|t |¡< t||j|jd}t ||¡ d S )Nri )
|
r¿ r: rÓ rF r> r r' rW r- re )r. rT rA r0 r rr r1 r! r! r" Útest_setitem_boolean_maskÞ s
|
|
z9TestDataFrameSetItemBooleanMask.test_setitem_boolean_maskz1Currently empty indexers are treated as all False)Úreasonr c C sh t dgdgdgd}|tkr,|g dd}n|g }d}tjt|d dg|j|dgf< W 5 Q R X d S ) Nr$ r% rB rÎ r9 z@Must have equal len keys and value when setting with an iterablerS r) )r r rX rY rZ r )r. r r0 rÑ r_ r! r! r" Ú:test_setitem_loc_empty_indexer_raises_with_non_empty_valueò s zZTestDataFrameSetItemBooleanMask.test_setitem_loc_empty_indexer_raises_with_non_empty_valuec C sz t dgdgdgd}|dg}d|d |j|dgf< t dgdgdgd}t ||¡ d|j|dgf< t ||¡ d S ) Nr$ r% rB Fra r+ r) rý )r r r- re )r. r r0 rÑ r1 r! r! r" Ú1test_setitem_loc_only_false_indexer_dtype_changedÿ s
|
zQTestDataFrameSetItemBooleanMask.test_setitem_loc_only_false_indexer_dtype_changedrÑ c C s\ t ddddgddddgd }| ¡ }|d
|
dk}||| d
|
¡|||< t ||¡ d S ) Nr% rU r( r* rb rü rú rû rÈ r$ )r r¿ r7 r- re )r. rÑ r0 r1 r r! r! r" Ú"test_setitem_boolean_mask_aligning s
|
zBTestDataFrameSetItemBooleanMask.test_setitem_boolean_mask_aligningc C sü t dddddddgddgd}tdddddd d
|
g}d d d d d d d g}t||d |d}t dddddddgdddgd}tdddddd d
|
g}d d ddd d d g}t||d |d}| ¡ } | d j dddg¡| d< |d dk}
|
dd g||
|
< t || ¡ d S )Nr$ r) )rç rO rP rQ ÚkÚlrª rµ r% r( )ÚcatsrF r& r+ r* r` )r r r r¿ r Zset_categoriesr- re ) r. Zcats2Zidx2Zvalues2Z exp_multi_rowZcatsfZidxfZvaluesfr0 Z exp_fancyr r! r! r" Ú test_setitem_mask_categorical s ÿ z=TestDataFrameSetItemBooleanMask.test_setitem_mask_categoricalr3 r r5 Úkwargsr' r% rW rD c C s6 ||d< t f |}| ¡ }d|||k< t ||¡ d S )Nr3 é/ r )r. r3 rb r0 rÁ r! r! r" Ú%test_setitem_empty_frame_with_boolean- s
|
|
zETestDataFrameSetItemBooleanMask.test_setitem_empty_frame_with_booleanc C sæ t td}dddg}t||tjdddgdd d
|
gd d d ggtdd}t||t t|t|f¡d}t||tjdddgdd dgdddggtdd}d||d | k< t ||¡ t
|
j t dd d||j d d
dk< W 5 Q R X d S )Nr* rD rE rt ç r. r g ø?r g @r g @g @r9 )r' rW r/ rö zItem wrong lengthrS r( )rë rc r r: r> r r r= r- re rX rY rZ r' )r. r rÏ Zdf1rÁ r1 r! r! r" Ú test_setitem_boolean_indexing8 s(
|
ÿý $ý z=TestDataFrameSetItemBooleanMask.test_setitem_boolean_indexingc C sd t ddgddgdd}| ¡ }tddgdd }t d
|
d d gid dgd|j|d
|
gf< t ||¡ d S )Nr% r( r* rU r$ rB Fr+ rM r) rb rü r r& )r r¿ r r r- re )r. r0 r1 rÑ r! r! r" Ú-test_loc_setitem_all_false_boolean_two_blocksO s
|
$zMTestDataFrameSetItemBooleanMask.test_loc_setitem_all_false_boolean_two_blocksc C sh t ddgddgg}t ddgddgg}t tddgdd ttjdgdd d
|
}d||< t ||¡ d S ) Nrö r( r* rE r TFÚbooleanr9 r )r r rN ZNAr- re )r. r0 r1 Zboolean_indexerr! r! r" Ú test_setitem_ea_boolean_maskW s þÿz<TestDataFrameSetItemBooleanMask.test_setitem_ea_boolean_maskN)r r r r" Ú skip_array_manager_invalid_testrX r r rY Úxfailrë r: r> r r[ r\ r- rM r r] ra rd rf rg ri r! r! r! r" rS Ý s* ý
|
|
rS c @ s e Zd Zdd Zej dddg¡dd Zdd Zej d
|
d e j
|
d gd ggd d d gd ggg¡dd Z ej d
|
de
|
dgdgg¡dgdggg¡dd Z ej dddgej ddgejjddddg¡ej dddeded fd g¡d d Zejd d! Zd"S )#Ú%TestDataFrameSetitemCopyViewSemanticsc C sZ d|j kst|d ¡ }||d< tj|jdd
|j d¡f< t|dd
¡ sVtd S )Nrá rD rb ra ) rW r@ r¿ r: rÓ rÀ Zget_locr r³ )r. rA r« r! r! r" Útest_setitem_always_copyf s
|
z>TestDataFrameSetitemCopyViewSemantics.test_setitem_always_copyÚ consolidateTFc C sè t ddddgddddgd d
|
d d d gd}| d dtj¡ |st|r`| ¡ t|jjd ksttnt|jjd kstt|d j }d|j
|
d d
df< t tjtjddg|j dd}t |d |¡ |sät ||j¡ t ||d j ¡sätd S )Ngñ?gÍÌÌÌÌÌ @gÍÌÌÌÌÌ@gffffff@gffffff@gffffff@gffffff @g333333 @)rj rk r r% r( r* r& rv é* rR )r rm r: rÓ Z_consolidate_inplacer= r¼ Úblocksr@ Z_valuesr r r' r- rp r§ rF Z shares_memory)r. rn Úusing_array_managerr/ r0 Zzvalsr1 r! r! r" Ú#test_setitem_partial_column_inplacen s"
|
ÿ
|
zITestDataFrameSetitemCopyViewSemantics.test_setitem_partial_column_inplacec C sj ddgd }t ddg|d}| ¡ }|d d
}d|d< t ddddgg|d }t ||¡ t ||¡ d S )
|
NrD rE r( re r ri rÖ rb rV r )r. rÏ r0 Údf_copyÚdf_viewr1 r! r! r" Ú*test_setitem_duplicate_columns_not_inplace s zPTestDataFrameSetitemCopyViewSemantics.test_setitem_duplicate_columns_not_inplacer4 r% r5 r9 c C sl ddg}t dddg|d}| ¡ }|d d
}||dg< t ddgddgg|d}t ||¡ t ||¡ d S )NrD rE r r% ri rV r )r. r4 rq rÏ r0 rs rt r1 r! r! r" Ú#test_setitem_same_dtype_not_inplace s
|
zITestDataFrameSetitemCopyViewSemantics.test_setitem_same_dtype_not_inplacer c C sl ddg}t dddg|d}| ¡ }|d d
}||dg< t ddgddgg|d}t ||¡ t ||¡ d S )NrD rE r r% ri r rV r )r. r4 rÏ r0 rs rt r1 r! r! r" Ú2test_setitem_listlike_key_scalar_value_not_inplace© s
|
zXTestDataFrameSetitemCopyViewSemantics.test_setitem_listlike_key_scalar_value_not_inplacerÑ r$ z+Boolean indexer incorrectly setting inplace)rZ Ústrict)Zmarkszvalue, set_value)r% rb )r g @z
|
2020-12-31z
|
2021-12-31rÈ c C s@ t d|iddgd}| ¡ }|d d
}|||< t ||¡ d S )Nr$ r r% r& r )r. r4 Ú set_valuerÑ r0 r1 r² r! r! r" Ú"test_setitem_not_operating_inplace¶ s
|
zHTestDataFrameSetitemCopyViewSemantics.test_setitem_not_operating_inplacec sÔ dd t dD t fdd D d}|jjd j}|s~|jD ]}d|| |< qDt |t d¡t d¡f dk¡sÐt nRt
|
¡ |jD ]}d|| |< qW 5 Q R X t |t d¡t d¡f dk¡sÐt d S ) Nc S s g | ]}d | qS )r+ r! r5 r! r! r" r¡ Ù s z\TestDataFrameSetitemCopyViewSemantics.test_setitem_column_update_inplace.<locals>.<listcomp>ra c s i | ]}|t t ¡qS r! )r: rF r= )r Úcol©r6 r! r" Ú
|
<dictcomp>Ú s z\TestDataFrameSetitemCopyViewSemantics.test_setitem_column_update_inplace.<locals>.<dictcomp>r& r r% ) rc r r¼ rp rF rW r: r³ rl r@ r- Z raises_chained_assignment_error)r. r/ r0 rF Úlabelr! r| r" Ú"test_setitem_column_update_inplaceÕ s
|
(
|
|
zHTestDataFrameSetitemCopyViewSemantics.test_setitem_column_update_inplaceN)r r r rm rX r r rr ru r: r> rv rw Úparamrk r rz r" rj r r! r! r! r" rl e sF
|
"ÿ
|
(
|
þþýþüþ rl )4r Únumpyr: rX Z pandas.util._test_decoratorsÚutilZ_test_decoratorsr" Zpandas.core.dtypes.baser r! Zpandas.core.dtypes.commonr r r Zpandas.core.dtypes.dtypesr r r r
|
ZpandasrN r r r r r r r r r r r r r r r r Zpandas._testingZ_testingr- Zpandas.core.arraysr Zpandas.tseries.offsetsr r r# r- rD rO rS rl r! r! r! r" Ú<module> s2 H C* .
|
