U
|
ý°díÿ ã @ s d dl mZ d dlmZmZ d dlmZ d dlZd dlZd dl Z d dl
|
m Z d dl m Z mZmZ d dlm mZ d dlmZ d dlZd dlmZmZmZmZm Z m Z m Z m Z m Z m!Z!m"Z" d dl#m$Z% d Z&G d
|
d d Z'G d d d Z(dd Z)dZ*dZ+G dd dZ,G dd dZ-dS )é )Ú
|
namedtuple)ÚdatetimeÚ timedelta©ÚDecimalN)ÚiNaT)ÚInvalidIndexErrorÚPerformanceWarningÚSettingWithCopyError)Ú
|
is_integer) Ú CategoricalÚ DataFrameÚ DatetimeIndexÚIndexÚ
|
MultiIndexÚSeriesÚ TimestampÚ
|
date_rangeÚisnaÚnotnaÚ to_datetimezBslice indices must be integers or None or have an __index__ methodc @ s e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
e j ddddgdddgdddgg¡dd Zdd Zdd Zdd Zd d Zd d Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- Zd.d/ Zd0d1 Z d2d3 Z d4d5 Z d6d7 Z e j!d8d9 Z"d:d; Z#d<d= Z$d>d? Z%e j!d@dA Z&dBdC Z'e j!dDdE Z(dFdG Z)dHdI Z*dJdK Z+dLdM Z,dNdO Z-dPdQ Z.dRdS Z/dTdU Z0dVdW Z1dXdY Z2dZd[ Z3d\d] Z4d^d_ Z5d`da Z6dbdc Z7e j ddddg¡dedf Z8dgdh Z9didj Z:dkdl Z;dmdn Z<dodp Z=dqdr Z>dsdt Z?dudv Z@dwdx ZAdydz ZBd{d| ZCd}d~ ZDdd ZEdd ZFe j deGjHeGjH Id¡eGjH Id
¡g¡dd ZJdd ZKdd ZLe j dddg¡e j dddgg¡dd ZMe j!dd ZNdd ZOe j dddgeGjPeGjPgg¡dd ZQe j ddeGjPg¡dd ZRe j deSeTeUjVg¡dd ZWdd ZXe j d¡dgdg¡e j d¢i d£dig¡d¤d¥ ZYe j d¢i d¦dig¡d§d¨ ZZd©dª Z[d«d¬ Z\dd® Z]e j d¡dd¯g¡e j d°e^d±g¡d²d³ Z_e j d´ddg¡e j d¡dµdµgg¡d¶d· Z`e j d¸eajbdfeajcd¹fg¡dºd» ZddS )¼ÚTestDataFrameIndexingc C s¬ |d d
}t |jdks t| ¡ D ].\}}t |jdks@tt |j|j¡s&tq&|j ¡ D ]\}}|| d k s`tq`d|ksttjt dd |d W 5 Q R X d S )Né Úrandom©Úmatch)
|
ÚlenÚindexÚAssertionErrorÚitemsÚtmZ equalContentsZ_seriesÚpytestÚraisesÚKeyError)ÚselfÚ float_frameÚslÚ_ÚseriesÚkey© r* ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/indexing/test_indexing.pyÚ test_getitem+ s z"TestDataFrameIndexing.test_getitemc C sr |}t dddg|d}tdddgdddgg|d }|d }tddgddggt ddg|dd }tj||d
|
d d S ) Né r ©Údtypeé é é é é ©ÚcolumnsT©Z check_exact©r r r Úassert_frame_equal)r$ Úany_numeric_dtyper/ ÚidxÚdfÚresultÚexpectedr* r* r+ Ú6test_getitem_numeric_should_not_fallback_to_positional< s $zLTestDataFrameIndexing.test_getitem_numeric_should_not_fallback_to_positionalc C sx | ¡ }tj t|¡|d< tj t|¡}||d< tjtt d¡d |
|
d¡ W 5 Q R X |d }t ||j ¡ d S )Nz$10z@awesome_domainz 'df["$10"]'r z df["$10"])ÚcopyÚnpr Úrandnr r! r" r# ÚreÚescapeÚ __getitem__r Zassert_numpy_array_equalÚvalues)r$ r% r< ÚadÚresr* r* r+ Ú test_getitem2E s z#TestDataFrameIndexing.test_getitem2c C sj |}t dddg|d}tdddgdddgg|d }d
|
|d< td
|
dd
|
gd
|
dd
|
gg|d }tj||d d d S ) Nr- r r. r0 r1 r2 r3 r4 r5 é
|
Tr7 r8 )r$ r: r/ r; r< r> r* r* r+ Ú6test_setitem_numeric_should_not_fallback_to_positionalR s zLTestDataFrameIndexing.test_setitem_numeric_should_not_fallback_to_positionalc C sÒ d|d< |ddg }||ddg< t j|d |d dd t j|d |d dd d}tjt|d |ddg |dg< W 5 Q R X tt|jd }d
|
t| d t| d }tjt|d ||d< W 5 Q R X d S ) NÚfooÚEÚAÚBF©Z check_namesz"Columns must be same length as keyr r- zLength of values \(z$\) does not match length of index \(z\))r Úassert_series_equalr! r" Ú
|
ValueErrorÚranger r )r$ r% ÚdataÚmsgZ newcolumndatar* r* r+ Útest_setitem_list[ s ÿz'TestDataFrameIndexing.test_setitem_listc C sÔ t dtdddgtjd}ddg|jdddgf< |j|jd ddgf }tddg|jtjdd}t ||¡ d |d< |d< d
|
d g|j|jd ddgf< |j|jd ddgf }td
|
d g|jdd }t ||¡ d S ) Nr r1 Ztt1Ztt2)r r6 r/ r- r0 )r/ ÚnameÚ0Ú1Ú2©rW )
|
r rS rA Úint_Úlocr r r6 r rQ ©r$ r< r= r> r* r* r+ Útest_setitem_list2n s z(TestDataFrameIndexing.test_setitem_list2c sÊ |j d }|j |k}| t¡}|j | }|| }t ||j ¡ tjtdd ||d d
W 5 Q R X || } t | |¡ tjtdd || W 5 Q R X t ||j }|| } t | |¡ t
|
t ¡0 | |j d d d
¡}|| } t | |¡ W 5 Q R X |||fD ]¼}
|
|
|
¡ |
|
|
|
dk } t fdd jD j jd } |
|
jD ] } | | krH|
|
| | | < qH| j |
|
jd
|
} tj| | d d |
|
jD ]6} | | j| | jkr| | j|
|
| jkstqqd S ) NrJ zItem wrong lengthr éÿÿÿÿzBoolean array expectedr c s* i | ]"}|t | d k | t j¡qS )r )rA ÚwhereÚnan)Ú.0Úc©rT r* r+ Ú
|
<dictcomp>§ s z>TestDataFrameIndexing.test_getitem_boolean.<locals>.<dictcomp>©r r6 r5 F)Z check_dtype)r ÚastypeÚobjectr Zassert_index_equalr! r" rR r9 r Úassert_produces_warningÚ UserWarningÚreindexZ_get_numeric_datar r6 r/ r )r$ Zmixed_float_frameZmixed_int_frameÚdatetime_frameÚdÚindexerZ indexer_objZsubindexZsubframeZ subframe_objr< ZbifZbifwrd r* re r+ Útest_getitem_boolean| sL
|
|
|
|
ý ý
|
|
|
z*TestDataFrameIndexing.test_getitem_booleanc
|
C sX | ¡ }d|d< |d d¡|d< |d ¡ |d< d|d< |d d¡|d< |d ¡ |d< ||dk }|j}tt d ¡gd
|
t d¡gd t d¡gd d d ddddddgd}t ||¡ d|j|j dd
ddgf< ||dk }|j}tt d ¡gd
|
t d¡g t d ¡g t d¡g t d ¡g d d ddddddgd}t ||¡ d S )Nr- rM Úint32ZE1ÚFÚint64ZF1r Úfloat64r2 r0 rN rO ÚCÚD©r r1 )
|
r@ rh Údtypesr rA r/ r rQ r] r )r$ rm r< Zcastedr= r> r* r* r+ Ú test_getitem_boolean_casting· sD ÿþü
|
ÿ
|
þ
|
ý
|
üúz2TestDataFrameIndexing.test_getitem_boolean_castingÚlstTFc C s> t t d¡ dd¡}|| }|j|j| }t ||¡ d S )Né r1 r2 )r rA ÚarangeZreshaper] r r r9 )r$ rz r< r= r> r* r* r+ Útest_getitem_boolean_listÙ s z/TestDataFrameIndexing.test_getitem_boolean_listc C sb t j dd¡}t| ¡ dddddgd}||dk d 7 < ||dk d 7 < t |j|¡ d S )
|
Nr3 rN rO ru rv rM r5 r r- )rA r rB r r@ r Úassert_almost_equalrF )r$ Úarrr< r* r* r+ Útest_getitem_boolean_iaddâ s
|
z/TestDataFrameIndexing.test_getitem_boolean_iaddc C s> t t ddg¡dgtg d}t g t¡}|| d||< d S )Nr r- rN ©r6 r )r rA Úemptyr ÚarrayÚbool)r$ ZblahÚkr* r* r+ Ú test_boolean_index_empty_cornerì s z5TestDataFrameIndexing.test_boolean_index_empty_cornerc C s t tj dd¡ddddgdddgd}|jd d
}|j|jd d
}t ||¡ |jddg }|jt ddg }t ||¡ d S )
|
Nr2 r1 r- rJ ru rM r0 rg r` )
|
r rA r rB Úilocr] r r r9 r r^ r* r* r+ Ú test_getitem_ix_mixed_integerö s
|
ÿ z3TestDataFrameIndexing.test_getitem_ix_mixed_integerc C s t dddddgdddd gd
|
dd d gd ddd
|
dgd}|dg }|jd d
dgf }t ||¡ |dg }|jd d
dgf }t ||¡ d S )N)ç ø?g@g @ç @é é é$ é( rJ é é+ é" r r é )Úrnaéüÿÿr éè r r r r5 r1 r- ©r r r r9 r^ r* r* r+ Ú test_getitem_ix_mixed_integer2 s
|
|
|
ü
|
ù
|
|
z4TestDataFrameIndexing.test_getitem_ix_mixed_integer2c C s: t |j|d ¡ d}tjt|d |j W 5 Q R X d S )NrN z6'DataFrame' object has no attribute 'NONEXISTENT_NAME'r )r rQ rN r! r" ÚAttributeErrorZNONEXISTENT_NAME)r$ r% rU r* r* r+ Ú test_getattr s z"TestDataFrameIndexing.test_getattrc C s0 t dditdd}d|_|jdk ¡ s,td S )NÚfoobarr- rJ rw r3 )r rS r Úallr ©r$ r< r* r* r+ Útest_setattr_column s z)TestDataFrameIndexing.test_setattr_columnc C sª |d d d d
}||d< d|ks&t t|dks6t t|dksFt t t |jtjgd f¡¡}t||jdd}t
|
|d |¡ |d }||d< t j
|
||d dd tj t|¡}||d
|
< |d
|
|k ¡ sÔt d |d < |d d k ¡ sðt d |d< |d d k ¡ st d|d< |d dk ¡ s,t |d d
}d}|rPddg|d< n&tjt|d ddg|d< W 5 Q R X |d jtjkst |d ddgk ¡ s¦t d S )NrN r0 Zcol5é r ©r rW Zcol6FrP Zcol9r3 Zcol7g
ëQ¸ @Zcol0rL Zcol8zC\nA value is trying to be set on a copy of a slice from a DataFramerY rZ Zcol10r )r r rA ZravelZ column_stackrF rb r r r rQ r rB r r! r" r
|
r/ Úobject_)r$ r% Úusing_copy_on_writer( Úexpr ZsmallerrU r* r* r+ Ú test_setitem! s8 z"TestDataFrameIndexing.test_setitemc C sb t ddgg}tj|jd< t tjtjgg}t ||¡ t ddgg}tj|jd< t ||¡ d S )Nr )r rA rb r r r9 r] ©r$ r< r> r* r* r+ Ú test_setitem2O s z#TestDataFrameIndexing.test_setitem2c C s. | ¡ }|j ¡ }d||d dk< d||d d
df dk< t |j|¡ |d dk}| |jd d d
¡}d||< d||d d
df dk< t |j|¡ d||dk< d||dk< t |j|¡ d||dk< d||dk< t |j|¡ d||d d
dk < t |d d
|d d
dk d¡ t |j|¡ d||d d d
dk< d||dk< t |j|¡ d }tj t
|
|d
|
d||d < W 5 Q R X | ¡ }|t |¡k}tj ||t |¡k< |j ¡ }tj ||j< t ||j|jd }t ||¡ |d ||t |¡k< t ||j|jd ¡ t ||j|jd }t ||¡ d S ) Nr2 rN r r` r- r3 r0 r1 z;Must pass DataFrame or 2-d ndarray with boolean values onlyr rg )r@ rF r r~ rl r rA Zputmaskr! r" Ú TypeErrorÚabsrb r r6 r9 ) r$ r% r< rF r( rU Zdf_origÚmaskr> r* r* r+ Útest_setitem_booleanZ sH
|
"
|
z*TestDataFrameIndexing.test_setitem_booleanc C s |d d¡|d< |d jtjks&td|d< |d jtjksBtt t|¡|d< t|d jjtj sltd|d< d|d< |d jtjkstd|d< d|d< |d jtj
|
ks´td|d< |d jtjksÐtd |d< |d jtjksìtd|d< |d jtj
|
ks
|
td S )
|
Nrv Úi8r rO ÚbarrL ç @Z somethingr0 ) rh r/ rA rs r r| r Ú
|
issubclassÚtypeÚintegerrt ©r$ r% r* r* r+ Útest_setitem_cast s$ z'TestDataFrameIndexing.test_setitem_castc C s t dddgdddgdt d¡d }|d
|
= dddg|d
|
< d
|
|ksDtt|jd ksVtd |d < d|d< d|d< d|t ¡ < d|t ¡ < t |jd }d|d < d|d
|
< t|jd ks´t|j j
|
tj ksÆtd|d< |d j
|
tj ksâtd|d< |d j
|
tj ksþtd|d < d|d d ks tt t d¡d }d|d < d|d< |d= d|d< |d j
|
tj ks`tdddg|d< |d j
|
tj kstd S )Nç ð?ç @ç @ÚaÚbrd )rO ru r1 rw rO r0 Z beginningrN rL rM r« rv Údateç @r- ru r rY rZ Ú3Z coercible)r rA r| r r r6 r Únowr rF r/ r rs rt )r$ r% r< Údmr* r* r+ Útest_setitem_corner« s: $ z)TestDataFrameIndexing.test_setitem_cornerc C s dddgdgd t j d¡d}t|}||d dk j}d|j|dgf< d|j|dgf< |jd dksnt|jd
|
dkstd S ) Nr r« é r )ÚtitleÚcruftr¾ r r¿ )r- r¾ )r- r¿ )rA r r r r] r )r$ rT r< Úixr* r* r+ Útest_setitem_corner2Õ s
|
þz*TestDataFrameIndexing.test_setitem_corner2c C s´ t tdtdd}tdd tdD tdd}tdddgtdd}t d¡|d < t|jdksht||d
|
< t|jdkst||d < t|jdkst|d jtj ks°td S ) Nr1 rg c S s g | ] }t d qS )r- r )rc r' r* r* r+ Ú
|
<listcomp>é s z<TestDataFrameIndexing.test_setitem_ambig.<locals>.<listcomp>rw rL ÚbzrÚbazr r- r0 )
|
r rS r rA Úonesr r6 r r/ r )r$ r» Zcoercable_seriesZuncoercable_seriesr* r* r+ Útest_setitem_ambigä s z(TestDataFrameIndexing.test_setitem_ambigc C st |d |d < t j|jd d
df |d dd t j|jd d
d f |d dd t j|d |d dd t| d S )NrN r` FrP )r rQ r r] Úreprr° r* r* r+ Útest_setitem_Noneö s ÿ ÿz'TestDataFrameIndexing.test_setitem_Nonec C sP t dddgdddgddd gd
|
}| ¡ }|j|j|j ¡ d f< t ||¡ d S ) NrY rZ r¹ Z11Z22Z33Z111Z222Z333)rµ r¶ rd rµ )r r@ rµ r] r¶ r r r9 ©r$ r< r= r* r* r+ Ú&test_loc_setitem_boolean_mask_allfalse s ÿz<TestDataFrameIndexing.test_loc_setitem_boolean_mask_allfalsec C sV t tj dd¡}|jd dd
}tj|jd dd
< t|jd dd
j ¡ sRt d S )NrJ r3 é r0 )
|
r rA r rB r rb r rF r r rÉ r* r* r+ Ú&test_getitem_fancy_slice_integers_step s z<TestDataFrameIndexing.test_getitem_fancy_slice_integers_stepc C sF t tj dd¡tdddd}| ¡ }d|jdd
< |jdd
dkj ¡ sPt | ¡ }d|jdd
< |jdd
dkj ¡ st |jdd
|
}|j
|
dd
}| dd
|
d dg¡}t ||¡ t ||¡ |jttdttddd d d
}tjtd d |j
|
dd
W 5 Q R X tjtd d d|j
|
dd
< W 5 Q R X d S )NrJ r3 r r r0 rw r2 r1 r r4 rË r` z^3$r )r rA r rB rS r@ r rF r r r] rl r r9 Úlistr! r" r# )r$ r< Úcpr= Úresult2r> Údf2r* r* r+ Ú,test_getitem_setitem_integer_slice_keyerrors s" *zBTestDataFrameIndexing.test_getitem_setitem_integer_slice_keyerrorsc C s¸ |j d d
dd
f }|d jtjks*t| ¡ }|j d d
dd
f }t |d j|d j¡sdtd|jd d
df< |s¨|d dk ¡ stt |d j|d j¡ n t
|
||¡ d S )Néýÿÿÿrv ru ç @r2 ) r r/ rA rt r r@ Ú shares_memoryÚ_valuesr] r r r9 )r$ r% Úfloat_string_framer¡ ZslicedÚoriginalr* r* r+ Ú test_fancy_getitem_slice_mixed/ s z4TestDataFrameIndexing.test_fancy_getitem_slice_mixedc C s² t ¡ }|jddg \}}|j||
}|||
}|dd
}t ||¡ t ||¡ | ¡ }d|j||
< | ¡ }d|||
< | ¡ }d|dd
< t ||¡ t ||¡ d S )Nr3 rJ r r )r ZmakeTimeDataFramer r] r9 r@ )r$ r< ÚstartÚendr= rÏ r> r* r* r+ Ú"test_getitem_setitem_non_ix_labelsF s z8TestDataFrameIndexing.test_getitem_setitem_non_ix_labelsc C s t tj dd¡}|j|jdkd d
f }| dg¡}t ||¡ t tj dd¡}|j|jdk|j dkf }|jdgdgd}t ||¡ d S )Nr1 r0 r r- rg )
|
r rA r rB r] r rl r r9 r6 )r$ r< ÚrsZxpr* r* r+ Útest_ix_multi_takeZ s z(TestDataFrameIndexing.test_ix_multi_takec C sP |}|j }|jD ]:}|| }|jd d d
D ] }|||f || ks,tq,qd S ©Nr3 )r] r6 r r )r$ r% ÚfrÀ ÚcolÚtsr; r* r* r+ Útest_getitem_fancy_scalarf s
|
z/TestDataFrameIndexing.test_getitem_fancy_scalarc C s |}| ¡ }|j}t|jD ]b\}}|| }|jd d d
D ]@}|j |¡} tj ¡ }
|
|
|
|j | |f< |
|
|||f< t
|
||¡ q<q d S rÞ ) r@ r] Ú enumerater6 r Zget_locrA r rB r r r9 ) r$ r% rß r> rÀ Újrà rá r; ÚiÚvalr* r* r+ Útest_setitem_fancy_scalarp s
|
z/TestDataFrameIndexing.test_setitem_fancy_scalarc C s |}|j }|jddgd}|d d
ddddgf }t ||¡ |j|jdd
ddgd}||jdd
ddddgf }t ||¡ |j|jd k}|j|j| d
|
}|| }t ||¡ ||d d
f }t ||¡ |||jd d
f }|j|j| d dgd}t ||¡ d S ) NrO rv r5 FTr3 rJ rg é rw r0 ru )r] rl r r9 r r6 )r$ r% rß rÀ r> r= Zboolvecr* r* r+ Útest_getitem_fancy_boolean s" z0TestDataFrameIndexing.test_getitem_fancy_booleanc C s¸ | ¡ }| ¡ }|j ¡ }|d dk}d|j|< d||j< t||j|jd}t ||¡ | ¡ }| ¡ }|j ¡ }d|j|ddgf< d||jd d
f< t||j|jd}t ||¡ d S )NrN r ç rg rO r0 )r@ rF r] r r r6 r r9 )r$ r% Úframer> rF r¨ r* r* r+ Útest_setitem_fancy_boolean s
|
|
|
|
z0TestDataFrameIndexing.test_setitem_fancy_booleanc C sx |j dddg }|j|jdddg }t ||¡ |j d d
dddgf }|jd d
|jdddg f }t ||¡ d S )Nr- r2 rè r0 r )r r] r r r9 r6 )r$ r% r= r> r* r* r+ Útest_getitem_fancy_ints s z-TestDataFrameIndexing.test_getitem_fancy_intsc C st |d d d d
dk}|j | }|j |d d d
}t ||¡ | ¡ }| ¡ }d|j |< d|j |< t ||¡ d S )NrN r` r- r )r] r r9 r@ )r$ r% r¨ r= r> rÎ r* r* r+ Ú'test_getitem_setitem_boolean_misaligned¶ s
|
|
|
z=TestDataFrameIndexing.test_getitem_setitem_boolean_misalignedc C s® t tj dd¡}t dddg¡}t ddg¡}|j||f }|jddgdgf }t ||¡ | ¡ }d|jt dddg¡t ddg¡f< d|jddgdgf< t ||¡ d S )Nr1 r0 TFr r- r3 ) r rA r rB r r] r r9 r@ )r$ r< Zk1Zk2r= r> r* r* r+ Ú"test_getitem_setitem_boolean_multiÄ s $z8TestDataFrameIndexing.test_getitem_setitem_boolean_multic C sÎ t dddddg}ttj dd¡|d}|jdd
}| ddddg¡}t ||¡ t |dksdt
|
|jdd
}| ddg¡}tj||dd t |dks t
|
|jdd
}| d d
|
g¡}t ||¡ t |dksØt
|
|jd d
}|j d d
}t ||¡ d |jd d
< |d d
}|d k ¡ ¡ s0t
|
t d dddd
|
g}ttj dd¡|d}d}t jt|d |j d d
W 5 Q R X |j dd
}| d
|
g¡}t ||¡ t |d ks¼t
|
| ¡ }t jttd d |j d d
< W 5 Q R X t jt|d |j d d
d k}W 5 Q R X |j ¡ s(t
|
|j d d
|j d d
kj ¡ sPt
|
| ¡ }d |j dd
< |j dd
d kj ¡ st
|
|j d d
|j d d
kj ¡ s¬t
|
|jd d
}|}t ||¡ t |dksÜt
|
|jdd
}| dddd
|
g¡}t ||¡ t |dkst
|
|jdd
}| d
|
g¡}t ||¡ t |d ksRt
|
|jd d
|
}| d dddd
|
g¡}t ||¡ t |dkst
|
| ¡ }d |jd d
|
< |jd d
|
}|d kj ¡ sÊt
|
d S )Nr r0 r1 r2 r3 rw F)Zcheck_index_typerÓ r¸ r- r r² r¬ g @r zPcannot do positional indexing on Index with these indexers \[1.0\] of type floatr gñ?g
|
×£p=
|
@)r r rA r rB r] rl r r9 r r r r r! r" r¦ r@ Ú
|
_slice_msgrF )r$ Úusing_array_managerr r< r= r> rU rÎ r* r* r+ Ú!test_getitem_setitem_float_labelsÓ sz ÿ ( ( z7TestDataFrameIndexing.test_getitem_setitem_float_labelsc C s
|
t tj dd¡dddddgdd d
|
gd }td |d < |j}tt d¡gd t d¡g dd d
|
d gd}t ||¡ t
|
|j d< t |j d rt |d jtjks¢t |j d t
|
ks´t tj|j d< t |j d sÒt tj|j dd d
f< t |j dd d
f ¡ rt d S )Nr3 r1 rµ r¶ rd rn ÚerL r« rÄ rg Z20010102Ú timestamprt údatetime64[ns]rw )r¶ rô )rd rô )r rA r rB r rx r r/ r rQ r r] r r r rb r r^ r* r* r+ Ú)test_setitem_single_column_mixed_datetime* s( ý
|
þ
|
z?TestDataFrameIndexing.test_setitem_single_column_mixed_datetimec C sÔ t ddddddgtddddddtdddtd ddgd
|
}t dtd td d }tj|d< tddd|jd< d|jd< d|jddgdf< t dt d¡gdt d¡gg¡}||jddgddgf< t
|
||¡ d S )Nr é é iÜ r- ÚxÚyiÝ iÞ ©rµ r¶ Úabr4 r r¶ ©r r¶ )r- r¶ ©rù rú r0 r1 z2013-01-01T00:00:00z2014-01-01T00:00:00r2 r3 rµ ) r r rÍ rS ÚpdÚNaTr] rA r Ú
|
datetime64r r9 )r$ r> r< rN r* r* r+ Útest_setitem_mixed_datetimeI s.
|
|
|
úþÿ
|
|
þÿz1TestDataFrameIndexing.test_setitem_mixed_datetimec C sn |j |jd d
ddgf }|j|j |jd d
ddgf< |j |jdd
ddgf j}|j}t ||¡ d S ©Nr0 rN rO éþÿÿÿ)r] r rF r r~ ©r$ r% Úpiecer= r> r* r* r+ Útest_setitem_frame_floatf s
|
z.TestDataFrameIndexing.test_setitem_frame_floatc C s~ | ¡ }tddgddgg|jdd
ddgd }|jtd d ddgf}||j|< t |j|jdd
ddgf j|j¡ d S )
|
Nr² r³ r´ rÓ r r0 rN rO rg )r@ r r Úslicer] r r~ rF ©r$ rÖ rß r r) r* r* r+ Útest_setitem_frame_mixedm s ÿ
|
z.TestDataFrameIndexing.test_setitem_frame_mixedc C s | ¡ }tddgddgddgddggt|jd d
|
d d g d dgd}|jtd d
|
d dgf}||j|< t |j|jd d
|
d dgf j|jd d
|
¡ d S )Nr² r³ r´ rÓ r¸ g @g @g @r r0 rL r« rN rO rg ) r@ r rÍ r r r] r r~ rF r r* r* r+ Ú'test_setitem_frame_mixed_rows_unalignedy s ý
|
ÿz=TestDataFrameIndexing.test_setitem_frame_mixed_rows_unalignedc C s | ¡ }|j|jd d
dgf }|jdd
|_|jtdd ddgf}||j|< tj|d< t |j|jdd
ddgf j|j¡ d S )Nr0 rN r rO ) r@ r] r r rA rb r r~ rF r r* r* r+ Ú&test_setitem_frame_mixed_key_unaligned s
|
|
z<TestDataFrameIndexing.test_setitem_frame_mixed_key_unalignedc C st | ¡ }|j|jd d
ddgf }|jtdd ddgf}|j|j|< t |j|jdd
ddgf j|j¡ d S r )r@ r] r r rF r r~ r r* r* r+ Ú test_setitem_frame_mixed_ndarray s
|
z6TestDataFrameIndexing.test_setitem_frame_mixed_ndarrayc C s t dddgdddggddd gd
|
}| ¡ }|jd d
ddgf d |jd d
ddgf< |jddgd
|
}|d 7 }|d |d < t ||¡ d S ) Nr- r0 rL r1 r2 r« rN rO ru r5 ç à?)r r@ r] rl r r9 )r$ r< rÐ r> r* r* r+ Útest_setitem_frame_upcast s ", z/TestDataFrameIndexing.test_setitem_frame_upcastc C s |j |jd d
ddgf }|jdd
|_ddg|_||j |jdd
ddgf< |j |jdd
ddgf j}|j}t ||¡ d S r )r] r r6 rF r r~ r r* r* r+ Útest_setitem_frame_align£ s
|
z.TestDataFrameIndexing.test_setitem_frame_alignc C s t tj dd¡dddddgd}|jd }|d d
}t ||¡ |jd }|jddg }t ||¡ |jd }|jd }t ||¡ d S ) Nr3 r1 rL r« rÄ rw r0 r2 ) r rA r rB r] r r9 r rQ r^ r* r* r+ Ú"test_getitem_setitem_ix_duplicates¬ s
|
|
|
|
z8TestDataFrameIndexing.test_getitem_setitem_ix_duplicatesc C s¦ t tj dd¡dddddgd}|jdg }|jddg }t ||¡ |j|d d
|
k }||d d
|
k }t ||¡ |j|d
|
d
|
k }||d
|
d
|
k }t ||¡ d S ) Nr3 r1 rL r« rÄ rw r0 r2 r- r )r rA r rB r] r r r9 )r$ r< r= r¢ r* r* r+ Ú+test_getitem_ix_boolean_duplicates_multiple¼ s zATestDataFrameIndexing.test_getitem_ix_boolean_duplicates_multipleÚ
|
bool_valuec C sl t ddddgi}| d}tjt|d |j| W 5 Q R X d}tjt|d d|j|< W 5 Q R X d S ) Nrµ r- r0 r1 z7: boolean label can not be used without a boolean indexr z.cannot use a single bool to index into setitemr )r r! r" r# r] )r$ r r< ÚmessagerU r* r* r+ Ú%test_getitem_setitem_ix_bool_keyerrorÌ s
|
z;TestDataFrameIndexing.test_getitem_setitem_ix_bool_keyerrorc C s¢ d|d< t |d jjttjfs$t|j|jd df }t |sDtt
|
ddgi}d|d< |jd }t |sptt dgd gdd
|
}|jd gdf }t ||¡ d S ) Nr- rM r3 rµ g®Gáz®ó?i r¶ rý r r[ )r r/ r® ÚintrA r¯ r r] r r r r r rQ )r$ r% r= r< r> r* r* r+ Ú"test_single_element_ix_dont_upcastÙ s
|
z8TestDataFrameIndexing.test_single_element_ix_dont_upcastc C s t tj dd¡tdddd}|jd }|jd }t ||¡ |jd }|jd }t ||¡ |jt dd }|jdd
}t
|
||¡ |jdddd
|
g }| |j dddd
|
g ¡}t
|
||¡ d S ) NrJ r2 r r r0 rw r- rË rø r4 ) r rA r rB rS r r] r rQ r r9 rl r ©r$ r< r= r¢ r> r* r* r+ Ú test_iloc_rowë s
|
|
|
|
z#TestDataFrameIndexing.test_iloc_rowc C s¼ t tj dd¡tdddd}| ¡ }|jtdd }t |d |d ¡sNt |d ¡ }d|j
|
d d
df< |s¨d|j
|
d d
df< d|j dd
< t |d |d ¡s¨t t |d |¡ d S ) NrJ r2 r r r0 rw rË rê )r rA r rB rS r@ r r rÔ r r] rÕ r rQ )r$ r¡ Úrequestr< r× ÚsubsetZexp_colr* r* r+ Útest_iloc_row_slice_view s z.TestDataFrameIndexing.test_iloc_row_slice_viewc C sü t tj dd¡tdddd}|jd d
df }|jd d
df }t ||¡ |jd d
df }|jd d
df }t ||¡ |jd d
t ddf }|jd d
dd
f }t
|
||¡ |jd d
dddd
|
gf }|j |j dddd
|
g d}t
|
||¡ d S ) Nr2 rJ r r r0 r5 r- rË rø r4 ) r rA r rB rS r r] r rQ r r9 rl r6 r r* r* r+ Ú test_iloc_col s z#TestDataFrameIndexing.test_iloc_colc C sò t tj dd¡tdddd}| ¡ }|jd d
tddf }|s¦|s¦t |d j |d j ¡sbt
|
d|j d d
df< |d dk ¡ st
|
t |d j |d j ¡sît
|
nH|rÆt |d j |d j ¡sÆt
|
d|d< |d dk ¡ sât
|
t ||¡ d S ) Nr2 rJ r r r0 r5 rË rê )r rA r rB rS r@ r r rÔ rÕ r r] r r r9 )r$ rñ r¡ r< r× r r* r* r+ Útest_iloc_col_slice_view) s z.TestDataFrameIndexing.test_iloc_col_slice_viewc C sä t tddddtddddd}|jdtddddd}td|dd gd
|
}t d d d d d d g¡}d |j|| df< tddddd d gddddddgd|d}t ||¡ td|dd gd
|
}|j|| df d 7 < t ||¡ d S )Niá r- )ÚyearÚmonthÚdayr3 )rÙ rÚ )r] Úitemr rN rO rg FTr4 )rN rO rw ) r r Úinsertr rA r r] r r9 )r$ Ztranger< Zbool_idxr> r* r* r+ Útest_loc_duplicatesB s þ ÿ z)TestDataFrameIndexing.test_loc_duplicatesc C s t tdddddd}td|i}|ddd g |d< t |d |¡ td|i}|ddd g |jddd gdf< t |d |¡ d S )
|
Nz
|
2015-01-01r1 Úutc)ÚperiodsÚtzÚdatesr[ r- r r0 )r r r r rQ r] )r$ Úcolumnr< r* r* r+ Ú4test_setitem_with_unaligned_tz_aware_datetime_column] s zJTestDataFrameIndexing.test_setitem_with_unaligned_tz_aware_datetime_columnc C s t dd}ttdddd}tjd| gd dd|d < tjd
|
| gd dd|jd d
d f< tjd | gd dd|j|jd d
d f< tjd| gd dd|jd d
df< tjd| gd dd|j|jd d
df< t d¡|d< tjd| gd dd|j|jd d
df< tdd d|j|jd d
df< t d¡|d< |j }t
|
t d¡gd t d¡gd
|
t dd}t ||¡ d S )Nr- )ÚhoursZ20130101r2 ©r& rw úm8[ns]r. rN r0 rO r1 ru rv r3 rM r rr r4 r` rÒ ÚGÚHztimedelta64[ns]rõ ZABCDEFGH)r r r rA r r] r Z timedelta64r rx r r/ rÍ r rQ )r$ Zone_hourr< r= r> r* r* r+ Ú,test_loc_setitem_datetimelike_with_inferencej s"
|
&,&,, þzBTestDataFrameIndexing.test_loc_setitem_datetimelike_with_inferencec C s t tjtjtjtjtjdtjdtjdddtjtjdtjtjdtjtjtjtjtjddtjtjtjtjdddddddd}| ¡ }d||d k< | ¡ }d|jd
|
< d|jd < d|jd < d|jd < t ||¡ d|d< d}tjt |d d||d k< W 5 Q R X d S )N)é# r r é1 é2 gjBzâÔ?g% tÙ?gÂØkÒ?gøãÖåÕ?r r- )r r- r0 r1 r2 rú g333333Ó?)r r- )r2 r- )r3 r- )r1 r2 ÚtestrL z1not supported between instances|unorderable typesr )
|
r rA rb r@ r] r r9 r! r" r¦ )r$ r< rÐ r> rU r* r* r+ Ú#test_getitem_boolean_indexing_mixed sH ûûûèÿ
|
|
|
|
z9TestDataFrameIndexing.test_getitem_boolean_indexing_mixedc C s. t jddgddggd dgd}tddddgddd d gg|tddgd
|
d d }tjtd d |d d
df W 5 Q R X ttdd
|
d }t ddgddggddgddggd dgd}tddgdd gg||d }|jd d
t d dff }t
|
||¡ d}ttdd
|
d }t ddg||d}|d }t
|
||¡ d S )Nrù rú r r- rd ©Únamesr0 r1 r2 rå r[ r r r )ÚlevelsÚcodesr7 )rù r r )r Ú from_productr r r! r" r rS r] r r r9 r rQ )r$ ÚmiZdgr r6 r> r= rW r* r* r+ Útest_type_error_multiindex® s, ÿ ÿ z0TestDataFrameIndexing.test_type_error_multiindexc C sh t t d¡tj t d¡¡d}|jd d
df }|d }t ||¡ |j
|
d d
df }t ||¡ d S )N©r1 r2 r3 r5 r r ) r rA rÅ rÿ Z IntervalIndexZ from_breaksr| r r rQ r] )r$ r< r> rH r* r* r+ Ú,test_getitem_interval_index_partial_indexingÇ s ÿ zBTestDataFrameIndexing.test_getitem_interval_index_partial_indexingc C sZ t ddgtd}t d¡t d¡d|jd< t t d¡gt d¡gd}t ||¡ d S )Nrµ r¶ )r6 r/ )r0 )r0 r0 rû r )r ri rA Úzerosr] r r9 r¤ r* r* r+ Ú test_setitem_array_as_cell_valueÕ s z6TestDataFrameIndexing.test_setitem_array_as_cell_valuec C s´ t ddddgidd}| ¡ }|jd d
d d d
f |jd d
< t ||¡ tjj |jd d
d d d
f ¡|jd d
< t ||¡ |j
|
d d
d d
f |j
|
d d
< t ||¡ d S )NrN r- r0 r1 ÚInt64r. r` ) r r@ rF r] r r9 rÿ ÚcoreZarraysZ PandasArrayr )r$ r< Úorigr* r* r+ Ú$test_iloc_setitem_nullable_2d_valuesÜ s " , z:TestDataFrameIndexing.test_iloc_setitem_nullable_2d_valuesc C s2 t tjdtd}t |d ¡|d< |dg d S )N©r- r- r. r )r rA r ri Z
|
empty_liker r* r* r+ Ú,test_getitem_segfault_with_empty_like_objecté s zBTestDataFrameIndexing.test_getitem_segfault_with_empty_like_objectÚnullúM8[ns]r- c C sÞ t ddddgi|d}|d }|j}d ddd d
|
g¡}tjt|d ||d < W 5 Q R X tjt|d ||g|d d
< W 5 Q R X tjt|d ||d < W 5 Q R X tjt|d ||g|d d
< W 5 Q R X tjt|d ||jd < W 5 Q R X tjt|d ||g|jd d
< W 5 Q R X tjt|d ||jd < W 5 Q R X tjt|d ||g|jd d
d f< W 5 Q R X | ¡ }| ¡ |d< tjt|d ||jd < W 5 Q R X tjt|d ||g|jd d
d f< W 5 Q R X d S )NrN r- r0 r1 r. ú|z@timedelta64\[ns\] cannot be converted to (Floating|Integer)Dtypez?datetime64\[ns\] cannot be converted to (Floating|Integer)Dtypez 'values' contains non-numeric NAz1Invalid value '.*' for dtype (U?Int|Float)\d{1,2}r r ©r r rO )r rÕ Újoinr! r" r¦ r r@ )r$ rG Zany_numeric_ea_dtyper< Úserr rU rÐ r* r* r+ Ú.test_setting_mismatched_na_into_nullable_failsð s@ üÿ zDTestDataFrameIndexing.test_setting_mismatched_na_into_nullable_failsc C sL t dgtg ddd}d|jd< t ddgitdgddd}t ||¡ d S )Nr¶ rµ r[ r r- r rw )r r r] r r9 r¤ r* r* r+ Ú+test_loc_expand_empty_frame_keep_index_name$ s
|
zATestDataFrameIndexing.test_loc_expand_empty_frame_keep_index_namec C s\ t dgtjg dddgdd}d|jd< t ddgitjdgdddgdd }t ||¡ d S )
|
Nrn rµ r¶ rd r6 r rL )r- r0 r1 rw ©r r Ú from_tuplesr] r r9 r¤ r* r* r+ Ú+test_loc_expand_empty_frame_keep_midx_names+ s ÿ
|
þzATestDataFrameIndexing.test_loc_expand_empty_frame_keep_midx_namesræ rù r- Úidxrrµ c C sn t dddgi}t d ¡, t d|dgiddgd|jd d
|f< W 5 Q R X t dtj|gi}t ||¡ d S )Nrµ r- r0 r rw )r r rj r] rA rb r9 )r$ rR ræ r< r> r* r* r+ Útest_loc_setitem_rhs_frame7 s
|
0z0TestDataFrameIndexing.test_loc_setitem_rhs_framec C sj t ddgd}| ¡ }|d d
}t d ¡( tjddgtjd|jd d
df< W 5 Q R X t ||¡ d S )Nrµ r¶ r5 r- r0 r. r ) r r@ r rj rA r rt r r9 )r$ r< r> Úviewr* r* r+ Ú$test_iloc_setitem_enlarge_no_warningA s ,z:TestDataFrameIndexing.test_iloc_setitem_enlarge_no_warningc C s¾ t ddddtjddgddggddgdd }d
|
g}d |d < d |j|d f< |j|d f |j|ddgf d|j|d f< |j|d f }tdtjdgdggddgdd d }}t ||¡ d S )NTr- r¬ )Zbool_colrµ r¶ r0 Úidx1Úidx2r6 rw rE r1 rd r rµ r¶ r r )r r Ú from_arraysr] r r rQ )r$ r< r; r= r> r* r* r+ Ú(test_loc_internals_not_updated_correctlyK s"
|
þý
|
z>TestDataFrameIndexing.test_loc_internals_not_updated_correctlyNc C sR t ddddgidd}||jdgd d
f< t dtjddgidd}t ||¡ d S ©Nrµ r¶ rd Ústringr. r ©r r rÿ ÚNAr r9 ©r$ ræ r< r> r* r* r+ Ú test_iloc_setitem_string_list_nab s z6TestDataFrameIndexing.test_iloc_setitem_string_list_nac C sP t ddddgidd}||jdd d
f< t dtjddgidd}t ||¡ d S rZ r\ r^ r* r* r+ Útest_iloc_setitem_string_naj s z1TestDataFrameIndexing.test_iloc_setitem_string_naÚfuncc C sT t ddddgidd}d|jd d
|dgf< t ddddgidd}t ||¡ d S ) Nrµ r- r0 r1 rA r. r3 r r )r$ ra r< r> r* r* r+ Ú/test_iloc_setitem_ea_null_slice_length_one_listr s zETestDataFrameIndexing.test_iloc_setitem_ea_null_slice_length_one_listc C s t tjddgdddggddgdd }td
|
|jj}|dddgd }|j|d d
f }t tjd d gddgdd }t ||¡ d S )NrN rO rµ r¶ rd ÚfirstÚsecondr6 rw ZIndexer)rc rd )rN rµ )rN r¶ )
|
r r r: r r r7 r] rP r r9 )r$ r< Z indexer_tuplerR r= r> r* r* r+ Ú test_loc_named_tuple_for_midxz s ÿÿ ÿÿz3TestDataFrameIndexing.test_loc_named_tuple_for_midxro rà r¶ c C sh t dtdtdgi|}|d tjdd |jdg|f< t dtdtdgi|}t ||¡ d S )Nrµ z
|
2022-12-29z
|
2022-12-30r- )Údaysz
|
2022-12-31)r r rÿ Ú Timedeltar] r r9 )r$ rà ro r< r> r* r* r+ Ú test_set_2d_casting_date_to_int s ÿ ÿz5TestDataFrameIndexing.test_set_2d_casting_date_to_intrW c C sV d}t |t|t|d}| ¡ }|ddg |j|dg ddgf< t ||¡ d S )Nr½ rþ rù rú T)r rS r@ r] r r9 )r$ rà Únr< r> r* r* r+ Ú1test_loc_setitem_reordering_with_all_true_indexer s
|
zGTestDataFrameIndexing.test_loc_setitem_reordering_with_all_true_indexerc C sZ t ddgd}| ¡ }t dgd}t d ¡ ||jd d
df< W 5 Q R X t ||¡ d S )Nrµ r¶ r5 )r r@ r rj r] r9 )r$ r< r> Úrhsr* r* r+ Útest_loc_rhs_empty_warning¡ s z0TestDataFrameIndexing.test_loc_rhs_empty_warningc C s t dddddgdddd d
|
gg}tddgd d }tdgd d }|j||f }t ddggdgd }t ||¡ |j|j|jf }t ||¡ d S )Nr r- r0 r1 r2 r3 r4 rè rË é rA r. rw )r r r r r9 rF )r$ r< ro Z row_indexerr= r> r* r* r+ Útest_iloc_ea_series_indexerª s z1TestDataFrameIndexing.test_iloc_ea_series_indexerc C s t dddddgdddd d
|
gg}tdtjgd d }d }tjt|d |jd d
|f W 5 Q R X tjt|d |jd d
|jf W 5 Q R X d S )Nr r- r0 r1 r2 r3 r4 rè rË rm rA r. zcannot convertr ) r r rÿ r] r! r" rR r rF )r$ r< ro rU r* r* r+ Ú#test_iloc_ea_series_indexer_with_na¶ s z9TestDataFrameIndexing.test_iloc_ea_series_indexer_with_na)Tr/ Úbooleanc C s¨ t jtddddg|dtddddg|dgddgd}tddd d
|
d gi|d }t tt|t¡ |j | }W 5 Q R X tddd git
|
ddgd|d d }t ||¡ d S )NTFr. rµ r¶ r6 rd r- r0 r1 r2 rw )rW r/ ) r rX r r r Zmaybe_produces_warningr Ú
|
isinstanceÚtupler] r r9 )r$ r/ ro Zmidxr< r= r> r* r* r+ Útest_loc_bool_multiindexÀ s þû
|
ÿz.TestDataFrameIndexing.test_loc_bool_multiindexr% r· c C sl t ttdddtdddg|dddgd}|jdd }||d
|
}|d |j|d
|
|f< t ||¡ d S ) Niæ r- r r )r% TF)r· Úupdate)Údeeprt r· )r r r r@ r] r r9 )r$ r% ro r< r> Z update_dfr* r* r+ Ú2test_loc_datetime_assignment_dtype_does_not_changeÓ s ÿüÿ zHTestDataFrameIndexing.test_loc_datetime_assignment_dtype_does_not_changez indexer, idxr0 c C s t dtjgdtjgdtjggtd}t dtjgdtjgg}|||d |
d d
f< t dtjgdtjgdtjggtd}t ||¡ d S )NrY rZ r¹ r. r- r0 )r rA rb ri r r9 )r$ ro r; r< rk r> r* r* r+ Ú"test_setitem_value_coercing_dtypesç s
|
$$z8TestDataFrameIndexing.test_setitem_value_coercing_dtypes)eÚ__name__Ú
|
__module__Ú __qualname__r, r? rI rK rV r_ rp ry r! ÚmarkÚ parametrizer} r r r r r r r£ r¥ r© r± r¼ rÁ rÆ rÈ rÊ rÌ rÑ ÚtdZ skip_array_manager_invalid_testrØ rÛ rÝ râ rç ré rì rí rî rï rò rö r r r
|
r r r r r r r r r r r r r r$ r* r0 r5 r< r> r@ rD rF rÿ r Zto_numpyrM rN rQ rS rU rY r] r_ r` rÍ r rA r rb re rh rj rl rn ro r rs rv r r] r rw r* r* r* r+ r * sÖ ;" ÿ
|
|
|
. 4 *
|
|
|
|
|
W
|
|
|
. ÿ
|
1
|
|
|
|
|
|
r c @ s e Zd Zdd ZdS )ÚTestDataFrameIndexingUInt64c C sú |}|d d¡}d|jks t||d< t |d t|dd¡ d|jksNtd|d< ||d< t |d t|dd¡ |d= | ¡ }tj|j d< tj|j d< |d }t t
|
|td d
|
d gdd¡ t |j tt d ¡t d ¡t d ¡gdddgd ¡ d S )NrN rL ru r[ rv rE ©r- r0 rO TFZuint64ÚOrw )Úrenamer6 r r rQ r r@ rÿ r r r rx rA r/ )r$ Z uint64_framer< r; rÐ r= r* r* r+ r£ ò s, þþz(TestDataFrameIndexingUInt64.test_setitemN)rx ry rz r£ r* r* r* r+ r~ ñ s r~ c C s t ddgtdddtjdddd}|jd }t|jd tsDtt|jd tj sZt|j
|
d }t|jd tsxtt|jd tj st|j ddd}t|jd ts°tt|jd tj sÆt| rÎd S |j }| ¡ | d¡j}t|d tsútt|d tj st|j|jd }|jd ks0t| d
|
¡}t|tsJt|j|jd }|jd ksjt| d
|
¡}t|tj std S ) Nr- r0 Ú2000r, z1 Day)rN rO ru r )ZaxisrH rJ r- )r r rÿ Ztimedelta_ranger] rq rF r r rg r ÚxsZ_mgrZ_rebuild_blknos_and_blklocsZfast_xsr ÚblocksZblknosr/ Úiget)rñ r< rL Zmgrr Zblkræ r* r* r+ Ú/test_object_casting_indexing_wraps_datetimelike s<
|
ýÿ
|
|
|
|
r zFCannot setitem on a Categorical with a new category( \(.*\))?, set thezCCannot set a Categorical with another, without identical categoriesc @ s8 e Zd Zejdd Zejdd Zejdd Zejdd Zejd d
|
Z ej
|
d e j e jg¡d d Zej
|
d e j e je je jg¡dd Zej
|
d e j e jg¡dd Zej
|
d e j e jg¡dd Zej
|
d e j e jg¡dd Zej
|
d e j e jg¡dd Zej
|
d e je j e jg¡dd Zdd Zd S ) Ú TestLocILocDataFrameCategoricalc C s\ t dddddddgddgd}tdddddd d
|
g}d d d d d d d g}t||d |d }|S ©Nrµ r¶ ©Ú
|
categoriesÚhrå rä r
ÚlÚmri r- ©ÚcatsrF rw ©r r r )r$ r r; rF rC r* r* r+ rC A s
|
z$TestLocILocDataFrameCategorical.origc C s\ t dddddddgddgd}tdddddd d
|
g}d d d d d d d g}t||d |d}|S ©Nrµ r¶ r r rå rä r
r r ri r- r0 r rw r )r$ Zcats1rV Zvalues1Úexp_single_rowr* r* r+ r I s
|
z.TestLocILocDataFrameCategorical.exp_single_rowc C s\ t dddddddgddgd}tdddddd d
|
g}d d d d d d d g}t||d |d}|S r r )r$ Zcats2rW Zvalues2Ú exp_multi_rowr* r* r+ r R s
|
z-TestLocILocDataFrameCategorical.exp_multi_rowc C s\ t dddddddgddgd}tdddddd d
|
g}d d d d d d d g}t||d |d }|S r r )r$ Zcats3Zidx3Zvalues3Úexp_parts_cats_colr* r* r+ r \ s
|
z2TestLocILocDataFrameCategorical.exp_parts_cats_colc C s\ t dddddddgddgd}tdddddd d
|
g}d d d d d d d g}t||d |d }|S r r )r$ Zcats4Zidx4Zvalues4Úexp_single_cats_valuer* r* r+ r e s ÿz5TestLocILocDataFrameCategorical.exp_single_cats_valuero c C s | ¡ }tdd}|tjkr&tdd}ddgddgg|||d d
f< t ||¡ | ¡ }tjttd& ddgddgg|||d d
f< W 5 Q R X d S )Nr0 r2 rä r
r¶ r rd ) r@ r r r] r9 r! r" r¦ Úmsg1)r$ rC r ro r< r) r* r* r+ Ú#test_loc_iloc_setitem_list_of_listsp s
|
|
|
zCTestLocILocDataFrameCategorical.test_loc_iloc_setitem_list_of_listsc C sr | ¡ }d}|tjtjfkr0|jd |jd f}d|||< t ||¡ tjt t
|
d d|||< W 5 Q R X d S )N)r0 r r0 r r¶ r rd ) r@ r r] Úatr r6 r9 r! r" r¦ r )r$ rC r ro r< r) r* r* r+ Ú7test_loc_iloc_at_iat_setitem_single_value_in_categories s zWTestLocILocDataFrameCategorical.test_loc_iloc_at_iat_setitem_single_value_in_categoriesc C sJ | ¡ }|jdk}d}|tjkr*|j| }d||||f< t ||¡ d S )Nrä r r¶ )r@ r r r] r6 r9 )r$ rC r ro r< r¨ r) r* r* r+ Ú5test_loc_iloc_setitem_mask_single_value_in_categories s
|
|
|
zUTestLocILocDataFrameCategorical.test_loc_iloc_setitem_mask_single_value_in_categoriesc C sz | ¡ }d}|tjkr |jd }ddg|||d d
f< t ||¡ tjttd ddg|||d d
f< W 5 Q R X d S )Nr0 r¶ r rd ) r@ r r] r r9 r! r" r¦ r )r$ rC r ro r< r) r* r* r+ Ú2test_loc_iloc_setitem_full_row_non_categorical_rhs¢ s
|
|
zRTestLocILocDataFrameCategorical.test_loc_iloc_setitem_full_row_non_categorical_rhsc C sØ | ¡ }tdddf}|tjkr4tdd|jd f}tddgddgd}||||< t ||¡ ttd td
|
d}tj t
|
t d ||||< W 5 Q R X ttd td
|
d}tj t
|
t d ||||< W 5 Q R X d S ) Nr0 r2 r rä r
r¶ rµ r ZbbÚabcr Úcc) r@ r r r] r6 r r9 rÍ r! r" r¦ Úmsg2) r$ rC r ro r< r) ÚcompatZ semi_compatZincompatr* r* r+ Ú1test_loc_iloc_setitem_partial_col_categorical_rhsµ s
|
zQTestLocILocDataFrameCategorical.test_loc_iloc_setitem_partial_col_categorical_rhsc C s~ | ¡ }tdddf}|tjkr4tdd|jd f}ddg|||< t ||¡ tjtt d ddg|||< W 5 Q R X d S ) Nr0 r2 r rä r
r¶ r rd )
|
r@ r r r] r6 r9 r! r" r¦ r )r$ rC r ro r< r) r* r* r+ Ú)test_loc_iloc_setitem_non_categorical_rhsÓ s
|
zITestLocILocDataFrameCategorical.test_loc_iloc_setitem_non_categorical_rhsc C sx t ddd t¡}tddddgi|d}|jjtks8t|tjkrP||d }n||d d
df }|jjtksttd S )NZ2012r1 r, r r- r0 rw ) r rh ri r r r/ r r Úgetitem)r$ ro r; r< rL r* r* r+ Ú-test_getitem_preserve_object_index_with_datesæ s
|
zMTestLocILocDataFrameCategorical.test_getitem_preserve_object_index_with_datesc C s\ t dgdggtjddgdgdd}t dggtjdgdgdd}|jd }t ||¡ d S ) Nr r- )rµ )r¶ rc r6 )rT r rµ rO )r$ r< r> r= r* r* r+ Ú test_loc_on_multiindex_one_levelõ s
|
þ ÿ
|
z@TestLocILocDataFrameCategorical.test_loc_on_multiindex_one_levelN)rx ry rz r! ZfixturerC r r r r r{ r| r r] r r r Ziatr r r r r¡ r¢ r£ r¤ r* r* r* r+ r @ s2
|
|
|
|
|
|
|
|
|
|
|
|
|
r c @ s, e Zd Zej ddhddidhdfddidfddhfdddifg¡dd Zej ddhddidhdfdfddidfdfddhfdddifg¡dd Zej ddhddidhdfddidfddhfdddifg¡d
|
d Zej ddhddidhdfdfddidfdfddhfdddifg¡d d Z dS )ÚTestDeprecatedIndexersr) r- rµ c C sD t ddgddggddgd}tjtdd |j| W 5 Q R X d S ©
|
Nr- r0 r1 r2 rµ r¶ r5 ú as an indexer is not supportedr ©r r! r" r¦ r] ©r$ r) r< r* r* r+ Ú$test_getitem_dict_and_set_deprecated s z;TestDeprecatedIndexers.test_getitem_dict_and_set_deprecatedr0 r c C sP t ddgddggddgt ddg¡d }tjtd
|
d |j| W 5 Q R X d S © Nr- r0 r1 r2 rµ r¶ r r= r r§ r ©r r rP r! r" r¦ r] r© r* r* r+ Ú/test_getitem_dict_and_set_deprecated_multiindex s ýzFTestDeprecatedIndexers.test_getitem_dict_and_set_deprecated_multiindexc C sD t ddgddggddgd}tjtdd d|j|< W 5 Q R X d S r¦ r¨ r© r* r* r+ Ú$test_setitem_dict_and_set_disallowed! s z;TestDeprecatedIndexers.test_setitem_dict_and_set_disallowedc C sP t ddgddggddgt ddg¡d }tjtd
|
d d|j|< W 5 Q R X d S r« r¬ r© r* r* r+ Ú/test_setitem_dict_and_set_disallowed_multiindex* s ýzFTestDeprecatedIndexers.test_setitem_dict_and_set_disallowed_multiindexN)
|
rx ry rz r! r{ r| rª r r® r¯ r* r* r* r+ r¥ s@ 0ÿ
|
|
úþ
|
|
0ÿ
|
|
úþ r¥ ).Ú collectionsr r r Údecimalr rC ÚnumpyrA r! Z pandas._libsr Z pandas.errorsr r r
|
Z pandas.util._test_decoratorsÚutilZ_test_decoratorsr} Zpandas.core.dtypes.commonr Zpandasrÿ r r r r r r r r r r r Zpandas._testingZ_testingr rð r r~ r r r r r¥ r* r* r* r+ Ú<module> s@ 4 R + C
|
