U
|
ý°deÆ ã @ s d Z ddlmZ ddlmZmZmZmZ ddlZddlm Z ddl
|
Z ddl Z ddl mZ ddlm mZ ddlZddlmZmZmZmZmZmZmZm Z m Z m Z m Z m Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z& ddl'm(Z) ddl*m+Z+m,Z, dd l-m.Z. dd
|
l/m0Z0 e j1 2d e j3e j3d gd e j3e j3gdddgge j3d gd e j3gddggg¡dd Z4G dd dZ5G dd dZ6G dd dZ7G dd dZ8G dd dZ9G d d d Z:G d d d Z;G d d! d!Z<G d"d# d#Z=G d$d% d%Z>G d&d' d'Z?e j1 2d(d)d*d+gd)d*gdd,gfd)d*d-gd)d*gdd,gfd)d*d-gd)d-gdd.gfg¡d/d0 Z@d1d2 ZAd3d4 ZBd5d6 ZCd7d8 ZDd9d: ZEd;d< ZFd=d> ZGd?d@ ZHdAdB ZIdCdD ZJdEdF ZKe j1 2dGdHe LdH¡e MdH¡g¡dIdJ ZNe j1 2dKe"dLdMfe"dNdOdPdQfg¡dRdS ZOdTdU ZPdVdW ZQG dXdY dYZRdS )Zz$ test label based indexing with loc é )Ú
|
namedtuple)ÚdateÚdatetimeÚtimeÚ timedeltaN)Úgettz)Ú IndexingError)Ú CategoricalÚCategoricalDtypeÚCategoricalIndexÚ DataFrameÚ DatetimeIndexÚIndexÚ
|
IndexSliceÚ
|
MultiIndexÚPeriodÚ PeriodIndexÚSeriesÚ SparseDtypeÚ TimedeltaÚ TimestampÚ
|
date_rangeÚtimedelta_rangeÚ to_datetimeÚ to_timedelta)Ú is_bool_dtypeÚ is_scalar)Ú_one_ellipsis_message)Ú"check_indexing_smoketest_or_raisesz series, new_series, expected_serÚbÚaFTc C sP t d| i}||jd d
df< t d|i}t | ¡ |¡ t | ¡ | ¡ d S ©NÚA)r ÚlocÚtmÚassert_frame_equalÚisnaZnotna)ÚseriesZ
|
new_seriesZ expected_serÚdfÚexpected© r* úUd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexing/test_loc.pyÚtest_not_change_nan_loc0 s
|
r, c @ sÚ e Zd Zej dddg¡dd Zej dddg¡dd Zej dd d
|
d d d dgdgd dgdgdd
|
d d gdgdd gdgddgdgddgdgg¡ej dddg¡dd Zej ddddgd
|
d dggdddgd
|
d dggg¡ej dddg¡dd Z ej ddddgd gdgddd gd
|
d dgdgd d d gd
|
d dgdgd!d"d#gd$gdgg¡ej dddg¡d%d& Z
|
ej d'd
|
d g¡ej dddg¡d(d) Z d*d+ Z ej dddg¡d,d- Z ej d.edd d d d ddgdeged/d0dgdegedd1d gdegedd1d gdegedd2dd gdegg¡ej dddg¡d3d4 Zd5d6 Zd7d8 Zej d9edgged:gd;edd
d:f edgd:d<fedged:gd=d:dgfedgged:gd=d:edgd:d<fg¡d>d? ZdS )@ÚTestLocÚkindr' Úframec C s$ | | d¡}t|ddtd d S )NZ_labelsr# é ©Úfails©Úgetfixturevaluer ÚKeyError©Úselfr. ÚrequestÚobjr* r* r+ Útest_loc_getitem_intA s z TestLoc.test_loc_getitem_intc C s$ | | d¡}t|ddtd d S )NÚ_emptyr# Úcr1 r3 r6 r* r* r+ Útest_loc_getitem_labelG s z TestLoc.test_loc_getitem_labelzkey, typs, axesÚfÚintsZuintsÚlabelsÚmixedÚtsNZfloatsé r c C s4 |D ]*}| | d| ¡}t|d||td qd S ©NÚ_r# ©Úaxesr2 r3 ©r7 ÚkeyÚtypsrG r. r8 Útypr9 r* r* r+ Ú#test_loc_getitem_label_out_of_rangeM s ÿz+TestLoc.test_loc_getitem_label_out_of_rangez key, typsé r0 ç @r" c C s2 |D ](}| | d| ¡}t|d|td qd S )NrE r# r1 r3 )r7 rI rJ r. r8 rK r9 r* r* r+ Útest_loc_getitem_label_lista s z#TestLoc.test_loc_getitem_label_listÚemptyé
|
é é é )rM rR )rM é )r0 é Zmultic C s4 |D ]*}| | d| ¡}t|d||td qd S rD r3 rH r* r* r+ Ú(test_loc_getitem_label_list_with_missingo s ÿz0TestLoc.test_loc_getitem_label_list_with_missingrJ c C s0 | | d| ¡}t|ddddgdtd d S )NrE r# rC é é( rM rF r3 )r7 rJ r. r8 r9 r* r* r+ Ú!test_loc_getitem_label_list_fails s ÿz)TestLoc.test_loc_getitem_label_list_failsc C s d S ©Nr* ©r7 r* r* r+ Ú!test_loc_getitem_label_array_like s z)TestLoc.test_loc_getitem_label_array_likec C s0 | | d¡}ddddg}t|d|td d S )Nr; TFr# r1 )r4 r Ú
|
IndexError)r7 r. r8 r9 r r* r* r+ Útest_loc_getitem_bool s z TestLoc.test_loc_getitem_boolzslc, typs, axes, failsZ20130102Z20130104é rU c C s4 |D ]*}| | d| ¡}t|d|||d qd S rD )r4 r ) r7 ZslcrJ rG r2 r. r8 rK r9 r* r* r+ Ú test_loc_getitem_label_slice s ûz$TestLoc.test_loc_getitem_label_slicec C st t ddgddgddggddgdddgd}t d ¡|jd
|
< t ddgd dgddggddgdddgd}t ||¡ d S ) NrC r éÈ Úcol1Úcol2rQ rM ©ÚcolumnsÚindexr0 )rM rc r )r ÚnpÚaranger# r$ r% ©r7 r( r) r* r* r+ Ú test_setitem_from_duplicate_axis· s ý ÿz(TestLoc.test_setitem_from_duplicate_axisc C s¤ t dddgdtjdgtdtjtdgdd }t d
|
gtdgd d }|j|j|d ¡ d d gf< t dddgdd
|
dgtdtdtdgdd }t
|
||¡ d S )NrM r0 rR zString 1zString 2z2019-06-11 11:00:00z2019-06-11 12:00:00)Zchannelr" ÚB)ÚdatazString 3©r" rl r" rl ) r rh ÚNaNr ÚpdÚNaTÚvaluesr# r& r$ r% )r7 r( Údf2r) r* r* r+ Ú test_column_types_consistentÄ s.
|
ýýÿ ÿ ýýÿ z$TestLoc.test_column_types_consistentz obj, key, expF©rf ©Úname©rg c C s6 |j | }t|ttfr&t ||¡ n ||ks2td S r[ )r# Ú
|
isinstancer r r$ Ú assert_equalÚAssertionError)r7 r9 rI ÚexpÚresr* r* r+ Ú#test_loc_getitem_single_boolean_argã s
|
z+TestLoc.test_loc_getitem_single_boolean_arg)Ú__name__Ú
|
__module__Ú __qualname__ÚpytestÚmarkÚ parametrizer: r= rL rO rW rZ r] r_ ÚsliceÚ TypeErrorr5 ra rk rt r r r r r~ r* r* r* r+ r- @ s
|
|
|
|
|
|
úþ þþ ûþ
|
|
üöþ ý$ùþ r- c @ sP e Zd Zdd Zej ddedddffdedd dffd
|
eddd ffd ed eddffded dd ffdeddedffdedd dffg¡dd Zdd Z dd Z
|
dd Z d d Z d d Z d d! Zd"d# Zej d$d%d&gd%d&d%d&gg¡d'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zd3d4 Zd5d6 Zejd7d8 Zej d9d:ejd:ej d;ejd:d:d:d:d:gej d;g¡d<d= Z d>d? Z d@dA Z dBdC Z dDdE Z!dFdG Z"ej dHdIdJdKdLdMdNdOdPg¡dQdR Z#dSdT Z$dUdV Z%dWdX Z&dYdZ Z'd[d\ Z(d]d^ Z)d_d` Z*dadb Z+dcdd Z,ej dedfge-dgdfdge dfg¡g¡ej dhdige dig¡g¡djdk Z.ej dld:dmgdfdndodKgfdpe/dpdpdpdpgdqdrej0ej0gdpdpdpdpggdfdndodKgdsfdtdodKgfdpduge/dtdmej0ej0gdqdrdpdugdvdwej0ej0ggdfdndodKgdsfdtdfdndogfejdpdudxgej d;e/dtdmej0gdpdudxgdvdwej0ggdfdndogdsfe-dtdqdgdndodKgfdpdudxgdydzd{gge/dtdmej0ej0gdqdpdudxgdvdydzd{ggdfdndodKgdsfe-dtdqdgdodfdKgfejdpdudxgdydzd{ggej d;e/dtdmej0ej0gdudrdpdxgdzdwdyd{ggdfdndodKgdsfe-dgdgdgdfdogfe/dpdugdxdygdzd{ggdfdogdse/dpdmdugdxdrdygdzdwd{ggdfdndogdsfg¡d|d} Z1d~d Z2dd Z3dd Z4dd
Z5dd Z6ejj7ej dddgddgg¡dd Z8dd Z9dd Z:dd Z;dd Z<dd Z=dd Z>dd Z?dd Z@dd ZAd¡d¢ ZBd£d¤ ZCd¥d¦ ZDd§d¨ ZEd©dª ZFd«d¬ ZGej dd®d¯d°g¡ej d±ej ejHeIg¡eJjKd²d³ ZLd´dµ ZMeJjKd¶d· ZNd¸d¹ ZOej dedºd»g¡d¼d½ ZPej d¾eQejeReSg¡d¿dÀ ZTdÁd ZUej dÃdÄeRdtgfdÅeRdmgfg¡dÆdÇ ZVdÈdÉ ZWej dÊd:dËdtdmdqdrdvdwdpdudxg
|
fe-drdud:dtdmdqdËdËdËdËdudxg
|
fdqdvgd:dtdmdËdrdËdwdpdudxg
|
fg¡dÌdÍ ZXdÎdÏ ZYdÐdÑ ZZdÒdÓ Z[ej dÔdÕdÕgg¡dÖd× Z\dØdÙ Z]ej dÚdfdfgdfe-dgfg¡dÛdÜ Z^dÝdÞ Z_dßdà Z`dádâ Zadãdä Zbdådæ Zcej dçdèdég¡dêdë Zddìdí Zedîdï Zfdðdñ Zgdòdó Zhdôdõ ZidgS )öÚTestLocBaseIndependentc C sH t tddd}|jt dg¡d d
}|jdd
}t ||¡ d S )NZ2021Ú2022rx z2021/6/1r é )r r r# rh ÚarrayÚilocr$ r% ©r7 r( Úresultr) r* r* r+ Útest_loc_npstrú s z%TestLocBaseIndependent.test_loc_npstrzmsg, keyz'Period\('2019', 'A-DEC'\), 'foo', 'bar'éã ÚfooÚbarz&Period\('2019', 'A-DEC'\), 'y1', 'bar'Úy1z&Period\('2019', 'A-DEC'\), 'foo', 'z1'Úz1z;Period\('2018', 'A-DEC'\), Period\('2016', 'A-DEC'\), 'bar'éâ éà z&Period\('2018', 'A-DEC'\), 'foo', 'y1'z;Period\('2017', 'A-DEC'\), 'foo', Period\('2015', 'A-DEC'\)éá éß z&Period\('2017', 'A-DEC'\), 'z1', 'bar'c C sz t tdddgtdtddgtddtd gd
|
d d gd ddgd dddg¡}tjt|d |j| W 5 Q R X dS )a(
|
parse_datetime_string_with_reso return parameter if type not matched.
|
PeriodIndex.get_loc takes returned value from parse_datetime_string_with_reso
|
as a tuple.
|
If first argument is Period and a tuple has 3 items,
|
process go on not raise exception
|
r Úx1Zx2r r r r r r rM r0 rR rQ rC rX )r" rl ÚCZV1ZV2r" rl r ©ÚmatchN)r r Ú set_indexr Úraisesr5 r# )r7 ÚmsgrI r( r* r* r+ Ú;test_contains_raise_error_if_period_index_is_in_multi_index s ûÿø zRTestLocBaseIndependent.test_contains_raise_error_if_period_index_is_in_multi_indexc C s> t ddgi}tjtdd |jd d
df W 5 Q R X d S )Nr rM u ×r )r r r r5 r# ©r7 r( r* r* r+ Ú$test_loc_getitem_missing_unicode_key( s z;TestLocBaseIndependent.test_loc_getitem_missing_unicode_keyc C sT t tj d¡dd tdD d}|jd }|jd d
df jd }t ||¡ d S ) N)rC rV c S s g | ]}d |d qS )ZABCDErV r* ©Ú.0Úxr* r* r+ Ú
|
<listcomp>1 s z@TestLocBaseIndependent.test_loc_getitem_dups.<locals>.<listcomp>rC rx )r" r r r" )r rh ÚrandomZ random_sampleÚranger# r$ Úassert_series_equal©r7 r( r) r r* r* r+ Útest_loc_getitem_dups- s
|
ÿ
|
z,TestLocBaseIndependent.test_loc_getitem_dupsc C s t ddddtdggdddddgdgd}tddddtdgdddddgdd}|jd }t ||¡ |jd }t ||¡ d S )
|
NrM r0 r r Ú20130101r re ©rg rw r )r r r r r$ r¨ r# r© r* r* r+ Útest_loc_getitem_dups27 s ý ý
|
|
z-TestLocBaseIndependent.test_loc_getitem_dups2c C s t tdtdtjdddd d tjdddd d d d
|
¡}d d d gf}| ¡ }|j| d9 < t |j| d|j| ¡ d}| ¡ }|j| d9 < |j| d|j| ksÀt dd d gf}| ¡ }|j| d9 < t
|
|j| d|j| ¡ d S )NZrtttiZaaaderV Úfloat64©Údtypegq=
|
×£põ?r0 gÃõ(\ÂÕ¿)Úmer r Úbar2r± Úrr r² ç @)r³ r Út) r Úlistrh ri r Úcopyr# r$ r¨ r{ r% )r7 Údf_origÚindexerr( r* r* r+ Útest_loc_setitem_dupsK s2 üÿù
|
þ þz,TestLocBaseIndependent.test_loc_setitem_dupsc C sð t dddgtdddgddd}|d dk}|j|d
|
f d }||j|d
|
f< t dddgtdd d gddd}t ||¡ t dddgdddgdd d}|d dk}|j|d
|
f }||j|d
|
f< t dddgdddgdd d}t ||¡ d S )Nr rM éd rb é, Zuint32r¯ ©r r r r éÉ i- Úuint64)r r r# r$ r% )r7 Zdf1ÚixZnewb1r) rs Znewb2r* r* r+ Útest_loc_setitem_slicen s " ÿ z-TestLocBaseIndependent.test_loc_setitem_slicec C s t dgdgdgdgd}dddg}|jd d
|f d ¡|jd d
|f< t dgtjdgd
|
d tjdgd
|
d tjdgd
|
d d}t ||¡ d S ) Nr" ç333333ó?ç g À)Úidr r r< r r r< Úfloat32r® r¯ )r r# Úastyperh r r$ r% )r7 r( Úcolsr) r* r* r+ Útest_loc_setitem_dtype s
|
&üÿ z-TestLocBaseIndependent.test_loc_setitem_dtypec C sr t tddddgd}tjtdd |ddg W 5 Q R X t td}tjtdd |d dg W 5 Q R X d S )
|
NrR r r r< rx ú not in indexr Údr )r r§ r r r5 )r7 Úsr* r* r+ Ú$test_getitem_label_list_with_missing s z;TestLocBaseIndependent.test_getitem_label_list_with_missingrg TFc C sN t dddg}dt| dt| }tjt|d |j| W 5 Q R X d S )NrM r0 rR z Boolean index has wrong length: z instead of r )r Úlenr r r^ r# )r7 rg rË r r* r* r+ Ú test_loc_getitem_bool_diff_len¡ s z5TestLocBaseIndependent.test_loc_getitem_bool_diff_lenc C s d S r[ r* r\ r* r* r+ Útest_loc_getitem_int_slice© s z1TestLocBaseIndependent.test_loc_getitem_int_slicec C sh t tj d¡dddgdddgd}d t ¡ j d
|
}tjt|d |jd d gd d gf W 5 Q R X d S )N)rR rR r r r< Úer> Úg©rg rf z$\"None of \[Index\(\[1, 2\], dtype='ú '\)\] are in the \[index\]\"r rM r0 ) r rh r¦ Úint_r° r r r5 r# ©r7 r( r r* r* r+ Útest_loc_to_fail s
|
ÿÿz'TestLocBaseIndependent.test_loc_to_failc C sH t td}d|jd< d|jd< tjtdd |jd W 5 Q R X dt ¡ j d }tjt|d |jdd
|
g W 5 Q R X d }tjt|d |jd g W 5 Q R X d |jd< tjtdd |jdd
|
g W 5 Q R X d|d< dt ¡ j d }tjt|d |jd
|
g W 5 Q R X |d= tjt|d d|jd
|
g< W 5 Q R X d S )Nr¯ rM r0 r z^-1$r éÿÿÿÿz&\"None of \[Index\(\[-1, -2\], dtype='rÓ éþÿÿÿzE\"None of \[Index\(\['4'\], dtype='object'\)\] are in the \[index\]\"Ú4rR rÉ z"\"None of \[Index\(\[-2\], dtype='r ) r Úobjectr# r r r5 rh rÔ r° ©r7 rË r r* r* r+ Útest_loc_to_fail2º s.
|
|
|
ÿ
|
ÿz(TestLocBaseIndependent.test_loc_to_fail2c C s t dgdggddgdgd}dt ¡ j d}tjt|d |jd
|
gd d
f W 5 Q R X tjt|d |jd
|
g W 5 Q R X d S ) Nr r rM r0 ÚvaluerÒ z!\"None of \[Index\(\[3\], dtype='rÓ r rR )r rh rÔ r° r r r5 r# rÕ r* r* r+ Útest_loc_to_fail3á s ÿ z(TestLocBaseIndependent.test_loc_to_fail3c C s t dddg}|jdg dt ¡ j d}tjtt |¡d |jdg W 5 Q R X tjtdd |jddg W 5 Q R X d S )NrM r0 rR z "None of [Index([3], dtype='z')] are in the [index]r rÉ )
|
r r# rh rÔ r° r r r5 ÚreÚescaperÛ r* r* r+ Ú test_loc_getitem_list_with_failð s z6TestLocBaseIndependent.test_loc_getitem_list_with_failc C s t tjjdddddddgd}|j d d
|
¡}|jt |¡ }|j| }t ||¡ |j|j }t ||¡ |jt
|
j|d d }t ||¡ d S ) N)rV rQ ©ÚsizeZalpha_0Zalpha_1Zalpha_2Zbeta_0Zbeta_1rx c S s d| kS )NÚalphar* ©r¤ r* r* r+ Ú<lambda> ó z7TestLocBaseIndependent.test_loc_index.<locals>.<lambda>Úbooleanr¯ ) r rh r¦ rg Úmapr# r r$ r% rr rp )r7 r( Úmaskr) r r* r* r+ Útest_loc_index s þ
|
z%TestLocBaseIndependent.test_loc_indexc C sÐ t tj dd¡ddddgddddgd}|jd d
dd
f jdd
d d
f }|jddgk ¡ sft|j ddgk ¡ s|tt t
|
d gd
|
gd jd }t t
|
d d
|
gd d gdd}t ||¡ |jtksÌtd S )NrU r" rl r ÚDre r r0 r« rM r½ r r r¬ )r rh r¦ Úrandr# r rf Úallr{ rg r r r$ r¨ r° rÚ r r* r* r+ Útest_loc_general s
|
|
ý( z'TestLocBaseIndependent.test_loc_generalc C s" t tddttdtjddS )Nú
|
2000-01-01ú 2000-01-5rV r¯ ©r Úval)r r r r§ rh Úint64r\ r* r* r+ Úframe_for_consistency' s
|
þÿz,TestLocBaseIndependent.frame_for_consistencyró r r¯ c C sV t tdtdtjdttdtjdd}| ¡ }||jd d
df< t ||¡ d S )Nr rV ©rg r° r¯ rò r © r r r§ rh rô r· r# r$ r% )r7 rõ ró r) r( r* r* r+ Ú test_loc_setitem_consistency0 s þÿz3TestLocBaseIndependent.test_loc_setitem_consistencyc C sR t tdtddttdtjdd}| ¡ }d|jd d
df< t ||¡ d S )Nr rV rx r¯ rò r r÷ ©r7 rõ r) r( r* r* r+ Ú(test_loc_setitem_consistency_dt64_to_strA s þÿz?TestLocBaseIndependent.test_loc_setitem_consistency_dt64_to_strc C sR t tdtddttdtjdd}| ¡ }d|jd d
df< t ||¡ d S )Nç ð?rV rx r¯ rò r r÷ rù r* r* r+ Ú*test_loc_setitem_consistency_dt64_to_floatO s þÿzATestLocBaseIndependent.test_loc_setitem_consistency_dt64_to_floatc C sJ t dttdgi}d|jd d
df< t dtdgi}t ||¡ d S )Nr Z20180101Ústring)r r r r# r$ r% rj r* r* r+ Ú'test_loc_setitem_consistency_single_row\ s z>TestLocBaseIndependent.test_loc_setitem_consistency_single_rowc C s t ddgd}t ddgd}t d ¡ d|jd d
df< W 5 Q R X t ||¡ t ddgd}d|d< |d tj¡|d< t ||¡ d S )Nr¤ Úyru rM )r r$ Zassert_produces_warningr# r% rÆ rh rô ©r7 r) r( r* r* r+ Ú"test_loc_setitem_consistency_emptyd s z9TestLocBaseIndependent.test_loc_setitem_consistency_emptyc C s dgd ddddgddddgg}t j|d d
|
d gd }d d d ddgdddddgg}t j|ddgd }ddddtjgdddddgdd d dtjgdd d dd gg}t|||d!}t|jd d
d"f |jd d
d"f< t|jd d
d#f |jd d
d#f< |jd$d%}|jd d
d#f |jd d
d"f |jd d
d&f< |jd d
d&f td' |jd d
d&f< t d(d)d*d+g|j
|
d&d,}t |d& |¡ d S )-NZRegion_1rU ZSite_1ZSite_2l ðBP[ l _Z l q=.Z l Ñ<.Z ZRegionZSiteZ RespondentID©ÚnamesÚ
|
RespondentZOtherCatZ SomethingÚ StartDateÚEndDatezYes/NoZ SomethingElseZLevel_0ZLevel_1r" z5/25/2015 10:59z5/25/2015 11:22ZYesz5/21/2015 9:40z5/21/2015 9:52z5/20/2015 8:27z5/20/2015 8:41z5/20/2015 8:33z5/20/2015 9:09ZNorÒ )r r )r r F)r· )r ZDurationl Xp7 g 7@ç (@g ,@g B@r¬ ) r Ú from_arraysrh Únanr r r# Z infer_objectsr r rg r$ r¨ )r7 ÚlevelsÚmiZclevelsrÇ rr r( r) r* r* r+ Ú-test_loc_setitem_consistency_slice_column_lent sJ
|
|
ý þ üÿÿ ÿÿ
|
ÿþ
|
ÿzDTestLocBaseIndependent.test_loc_setitem_consistency_slice_column_lenÚunitÚYÚMrì ÚhÚmrË ÚmsÚusc C sª t dt d¡t d¡gi}|jd d
df j d| d¡|jd d
|f< |jd d
df j d| d¡|d< t|jd d
df |d}t |jd d
|f |¡ d S )NÚ timestampz2017-02-11 12:41:29z1991-11-07 04:22:37z datetime64[ú]r) rv ) r rh Ú
|
datetime64r# rr rÆ r r$ r¨ )r7 r r( r) r* r* r+ Ú test_loc_assign_non_ns_datetime¥ s þÿÿ 0&z6TestLocBaseIndependent.test_loc_assign_non_ns_datetimec C sÈ t dddddgi¡}t|d ddd|d < |jd d
d f |jd d
d
|
f< |jd d gd f |jd d gd
|
f< t dd d gdddgdddgdddggdd d
|
gd}d d
|
g}|| t¡||< t ||¡ d S )Nr l çCO!gl ös~!gl 4 8l A{ï?r T)r ÚcacheZdate_dtZ
|
date_dt_cpr0 rR z2017-01-24 13:26:12.711z2017-01-24 13:52:05.110z2018-10-22 13:44:05.888z2018-10-23 08:08:41.025ru )r Ú from_dictr r# Úapplyr$ r% )r7 r( r) rf r* r* r+ Útest_loc_modify_datetime· s" ÿ üù
|
z/TestLocBaseIndependent.test_loc_modify_datetimec C sp t dddgdgtd}tjdddgdd |jdddgdf< tdddgdddgtd
|
}t d|i}t ||¡ d S ) NrR rV rU r" ©rg rf r° rM r0 rô r¯ rö )r Úfloatrh r r# r r$ r% ©r7 r( Úserr) r* r* r+ Ú#test_loc_setitem_frame_with_reindexÑ s
|
$ z:TestLocBaseIndependent.test_loc_setitem_frame_with_reindexc C s t dddgddgtd}d|d< tjdd dgd
|
d |jdddgdf< td ddgdddgd
|
d }t d| t¡i}d|d< t ||¡ d S ) NrR rV rU r" rl r rý rM r0 rô r¯ rö ) r r rh r r# r rÆ r$ r% r r* r* r+ Ú)test_loc_setitem_frame_with_reindex_mixedÜ s $z@TestLocBaseIndependent.test_loc_setitem_frame_with_reindex_mixedc C st t dddgddgtd}d|d< tjdddgdd |jtdd
|
d df< t d d dgdddddgd}t ||¡ d S )NrM r0 rR r" rl r rý rô r¯ r r× rN r´ rû rn rx )r r rh r r# r
r$ r% rj r* r* r+ Ú*test_loc_setitem_frame_with_inverted_sliceè s
|
& zATestLocBaseIndependent.test_loc_setitem_frame_with_inverted_slicec
|
C sÎ dd t dD }tjddd}dd t dD }tjddd}tt| |¡}t|d}tj|d < ||j|d f< tj|d
|
< ||j|d
|
f< t ||tj
|
d }t ||d}t||d j |d} t || ¡ d S ) Nc S s g | ]}d t | qS ©ú@©Ústr©r£ Úir* r* r+ r¥ ô s zGTestLocBaseIndependent.test_loc_setitem_empty_frame.<locals>.<listcomp>rV rô r¯ c S s g | ]}d t | qS r# r% r' r* r* r+ r¥ ÷ s rU rx r" rl rö rn )r§ rh ri r¶ ÚsetÚunionr r r# r r® Úreindexr$ r% )
|
r7 Zkeys1Zval1Úkeys2Zval2rg r( ZseraZserbr) r* r* r+ Ú test_loc_setitem_empty_frameò s
|
|
|
z3TestLocBaseIndependent.test_loc_setitem_empty_framec C s² t tj dd¡tdtdd}|jd }d|jd< |jd }|dksJt|jd }|dks`td|jd d
d d
|
f< |jd d
d d
|
f }|jd d
dd
f }t ||¡ d S ) NrU ZabcdÚABCDrÒ ©r r rM )r r" r rl rì )
|
r rh r¦ Úrandnr¶ r r# r{ r$ r% r r* r* r+ Útest_loc_setitem_frame s
|
|
|
|
z-TestLocBaseIndependent.test_loc_setitem_framec C sT t dddgtjd}|j|j|j|jkdf< t dddgtjd}t ||¡ d S )NrM r0 rR rn rl )r rh r r" r# rl r$ r% rj r* r* r+ Ú/test_loc_setitem_frame_nan_int_coercion_invalid s zFTestLocBaseIndependent.test_loc_setitem_frame_nan_int_coercion_invalidc C s t ddgddgddgd}|jdddgf }tddgtddgtd tdd
|
}t ||¡ t d dgd dgddgd}d d g|jdddgf< t ||¡ d S ) NrM r0 rR rU r r )rM r0 r r r¯ ©rg r° rw rV rS )r r# r r rÚ r$ r¨ r% r r* r* r+ Ú#test_loc_setitem_frame_mixed_labels" s ÿ z:TestLocBaseIndependent.test_loc_setitem_frame_mixed_labelsc C s t dddgttdtjdd}|jdd
}|jd d
|_||jd d
< t dddgtdddgtjdd}t ||¡ t t d
|
d ttd tjdd }t t
|
dt
|
dt
|
dt
|
dt
|
dgtd dd ddgtjdd }|jd d
}|jdd
|_||jdd
< t ||¡ d S )Nr r ÚbazrR r¯ rn rM r0 r rð rñ rV rò Z20000101Z20000102Z20000103rU ) r r r§ rh rô r# rg r$ r% r r )r7 r( Úrhsr) r* r* r+ Ú test_loc_setitem_frame_multiples1 s: ÿ ÿ þÿûøÿ z7TestLocBaseIndependent.test_loc_setitem_frame_multiplesr¹ r" NrÝ ÚZc C sL t ddgddggddgd}||jd|f< |jd }t|rD|d
|
ksHtd S ) NrM r0 rR rU r" rl ru r )r r" r8 )r r# r r{ )r7 r¹ rÝ r( r r* r* r+ Ú"test_loc_setitem_with_scalar_indexV s
|
z9TestLocBaseIndependent.test_loc_setitem_with_scalar_indexzindex,box,expectedr0 rl r rT rR rU ru rM r` rV rS é rQ é é c C s< t ddgddgddggddgd }||j|< t ||¡ d S )
|
NrM r0 rR rU rV rS r" rl ru ©r r# r$ r% )r7 rg Úboxr) r( r* r* r+ Ú test_loc_setitem_missing_columnse s 5"
|
z7TestLocBaseIndependent.test_loc_setitem_missing_columnsc C sZ t dtd d¡tjgi}|j}|jdg }t |j|¡ |jdg }t |j|¡ d S )Nr r« ÚUTCr rM ) r r Ú tz_localizerp rq Údtypesr r$ r¨ r© r* r* r+ Útest_loc_coercion s z(TestLocBaseIndependent.test_loc_coercionc C s^ t dtdddtdddgi}|j}|jdg }t |j|¡ |jdg }t |j|¡ d S )Nr éÜ rM iô r0 r )r r rB r r$ r¨ r© r* r* r+ Útest_loc_coercion2ª s z)TestLocBaseIndependent.test_loc_coercion2c C sZ t ddgd gd i}|j}|jdd
}t |j|¡ |jdd
}t |j|¡ d S )NÚtextz
|
some wordsr: r r0 rR )r rB r r$ r¨ r© r* r* r+ Útest_loc_coercion3µ s z)TestLocBaseIndependent.test_loc_coercion3c C sf t d d¡t d d¡g}t|ddgd}ttd}|d ||d< |d ||d< t ||¡ d S )
|
Né* r@ i r r rx r¯ r rM )r rA r rÚ r$ r¨ )r7 Z
|
indexer_slÚvalsr) r r* r* r+ Útest_setitem_new_key_tzÀ s þ
|
z.TestLocBaseIndependent.test_setitem_new_key_tzc C s t ddddddgddddddgd d
|
dd
|
dddgd }d }tjt|d |jdd
W 5 Q R X d}tjt|d |jd
|
d
W 5 Q R X d }tjt|d |jdd
W 5 Q R X t ddddddgddddddgd d
|
dd
|
dddgd jd
|
d}|jdd
}t ddddgddddgd ddddgd }t ||¡ |jd
|
d
}t ||¡ |jdd
}t dddgdddgd dddgd }t ||¡ d S )NrM r0 rR rU rV rS rT r` rn r rx z5'Cannot get left slice bound for non-unique label: 1'r z5'Cannot get left slice bound for non-unique label: 0'©Úaxis)r r r r5 r# Ú
|
sort_indexr$ r% )r7 r( r r r) r* r* r+ Útest_loc_non_uniqueÎ s6 ÿ ÿþ* $z*TestLocBaseIndependent.test_loc_non_uniquez
|
length, l2é r» i » é c C sÒ t d}t ttj |t|¡t |¡|dtt |t|f¡dg| |dg¡}|j
|
j dksbt t |¡}|j | }t | dg¡tt t|t|f¡dgt| |d| |dd
¡g¡}t ||¡ d S )NZABCDEFGrÒ r FrM )r¶ rp Úconcatr rh r¦ r0 rÍ ri Úonesrg Z is_uniquer{ r# Útaker$ r% )r7 ÚlengthÚl2rf r( rê r r) r* r* r+ Ú test_loc_non_unique_memory_errorð s0 ý úÿ
|
|
|
ýùÿ z7TestLocBaseIndependent.test_loc_non_unique_memory_errorc C s\ t ddgddgg}d|j_|jddg jj}|dks:t|jddg jj}|dksXtd S )NrM Z
|
index_namer )r rg rw r r{ r# ©r7 r( r r* r* r+ Ú test_loc_name s z$TestLocBaseIndependent.test_loc_namec C s t dd¡}t j|jd d
g f |jd d
d d
f ddd t j|jg d d
f |jd d
d d
f ddd t j|jg |jd d
d d
f ddd d S )NrV r0 r T)Úcheck_index_typeÚcheck_column_type)r$ ZmakeCustomDataframer% r# r r r* r* r+ Ú!test_loc_empty_list_indexer_is_ok s& ÿ ÿ ÿz8TestLocBaseIndependent.test_loc_empty_list_indexer_is_okc C sÀ t ddddgi}|jd d
}||k s,t|d d
|k s@t|jd d
d d
f |k s^tt |d j|d j¡sztdddg|jd d
df< |r²|d dddgk ¡ sÆtn|d dk ¡ sÆtt tj dd¡}|rü|d |jd d
df k sútn |d |jd d
df ks tt dddddd g}|jd d
}||k sLt|d d
|k sbtd
|
d d g|d d
< |rt|d d
dddgks¼tn t|d d
d
|
d d gks¼td S ) Nr rM r0 rR rU rQ r rV rS rT r` r: )
|
r r# r{ rh Z shares_memoryÚ_valuesrî r¦ r0 r )r7 Úusing_copy_on_writeZ original_dfZ sliced_dfr( Zoriginal_seriesZ sliced_seriesr* r* r+ Ú&test_identity_slice_returns_new_object- s, "z=TestLocBaseIndependent.test_identity_slice_returns_new_objectc C s |s t jjdd}|j |¡ tttdtdddgd}| ¡ }|j d d
df }|d7 }t
|
||¡ | ¡ }|j |j df }|d7 }t
|
||¡ d S )Nz!accidental fix reverted - GH37497)ÚreasonrR r r ru r0 ) r r ZxfailÚnodeZ
|
add_markerr Úzipr§ r· r# r$ r% rg )r7 r8 r] r r¤ rÿ ÚqÚzr* r* r+ Útest_loc_copy_vs_viewT s z,TestLocBaseIndependent.test_loc_copy_vs_viewc C s t d¡j}tddg|d |gd}|j|d }|jd }||ksHt|j|d g }|jdg }t ||¡ |j|d |g }t ||¡ d S )Nr¿ rM r0 rx r ) rh ÚiinfoÚmaxr r# r r{ r$ r¨ )r7 Úumaxr r r) r* r* r+ Útest_loc_uint64h s
|
z&TestLocBaseIndependent.test_loc_uint64c C sr t d¡j}tddg|d |gd}tjtdd |jd W 5 Q R X tjtdd |jdg W 5 Q R X d S )Nr¿ rM r0 rx z-1r r× )rh re rf r r r r5 r# )r7 rg r r* r* r+ Ú!test_loc_uint64_disallow_negativey s z8TestLocBaseIndependent.test_loc_uint64_disallow_negativec C sR dddg}t |d gt| d}t ddgd}||jd d
df< t ||¡ d S ©NrM r0 rR ©r¤ rÿ r¤ rÿ ru )r rÍ r# r$ r% ©r7 rm r) r( r* r* r+ Ú*test_loc_setitem_empty_append_expands_rows s
|
|
zATestLocBaseIndependent.test_loc_setitem_empty_append_expands_rowsc C sf dddg}t |d gt| d}t ddgd}|d tj¡|d< ||jd d
df< t ||¡ d S rj )r rÍ rÆ rh rô r# r$ r% rl r* r* r+ Ú6test_loc_setitem_empty_append_expands_rows_mixed_dtype s
|
zMTestLocBaseIndependent.test_loc_setitem_empty_append_expands_rows_mixed_dtypec C sD t dgtjgd}t ddgtd}|jd |jd< t ||¡ d S )Nrû rk r¤ rÿ ©rf r° )r r¤ )r rh r r r# r$ r% r r* r* r+ Ú*test_loc_setitem_empty_append_single_value s zATestLocBaseIndependent.test_loc_setitem_empty_append_single_valuec C s¤ ddg}t ddgd}|j tj¡|_dt ¡ j d}tjt |d ||j
|
d dgdf< W 5 Q R X d
|
d d d g¡}tjt |d ||j
|
d d
df< W 5 Q R X d S )NrM r0 r¤ rÿ ru z"None of \[Index\(\[0, 1\], dtype='z'\)\] are in the \[index\]r r ú|z?cannot copy sequence with size 2 to array axis with dimension 0zCcould not broadcast input array from shape \(2,\) into shape \(0,\)z@Must have equal len keys and value when setting with an iterable) r rg rÆ rh rô rÔ r° r r r5 r# ÚjoinÚ
|
ValueError)r7 rm r( r r* r* r+ Ú$test_loc_setitem_empty_append_raises£ s ÿ ýÿz;TestLocBaseIndependent.test_loc_setitem_empty_append_raisesc C sD t ddgddgg}|jt d¡ }tddgdd}t ||¡ d S )NrM r0 rR rU r rv )r r# rh r r r$ r¨ )r7 r( r rË r* r* r+ Ú test_indexing_zerodim_np_arrayº s z5TestLocBaseIndependent.test_indexing_zerodim_np_arrayc C s, t ddg}|jt d¡ }|dks(td S )NrM r0 r )r r# rh r r{ )r7 rË r r* r* r+ Ú%test_series_indexing_zerodim_np_arrayÁ s z<TestLocBaseIndependent.test_series_indexing_zerodim_np_arrayc C sl ddddddgd gd }t |tddd }t tddtjd
|
}ddddddg|jd dd
< t ||¡ d S ) NrM r0 rR rU rV rS iÚ iä rx rö r r× )r r§ rh r® r# r$ r¨ )r7 rm r) r r* r* r+ Útest_loc_reverse_assignmentÇ s
|
z2TestLocBaseIndependent.test_loc_reverse_assignmentc C s¶ t j d¡ dd tdD }t|dgd}t|dgtd}t ||¡ dd |D |j|j df< t|dgt
|
d t¡}t ||¡ d d |D |d< t|dgt
|
d}t ||¡ d S )
|
Né c S s g | ]}t tj ¡ d qS )gê-q=)r& rh r¦ )r£ rE r* r* r+ r¥ Ô s z^TestLocBaseIndependent.test_loc_setitem_str_to_small_float_conversion_type.<locals>.<listcomp>rV r" ru ro c S s g | ] }t |qS r* ©r r¢ r* r* r+ r¥ Û s c S s g | ] }t |qS r* ry r¢ r* r* r+ r¥ á s ) rh r¦ Úseedr§ r rÚ r$ r% r# rg r rÆ )r7 Zcol_datar r) r* r* r+ Ú3test_loc_setitem_str_to_small_float_conversion_typeÑ s zJTestLocBaseIndependent.test_loc_setitem_str_to_small_float_conversion_typec C sÄ t dddd}|jdk|jdk@ }ttj t|d¡|d}t ||¡}|j
|
t dd }|j
|
| }t ||¡ |j
|
d d
}|j
|
t dd }|d
|
d
}|j d ¡|_ |j d ¡|_ t ||¡ d S ) Nú1/1/2000z1/5/2000Z5min©Úfreqr: rX rR rx z1/4/2000r× )r ÚhourÚminuter rh r¦ r0 rÍ r$ Úget_objr# r rz rg Ú
|
_with_freq) r7 Úframe_or_seriesÚrngrê r9 r r| Úchunkr) r* r* r+ Ú test_loc_getitem_time_objectå s
|
z3TestLocBaseIndependent.test_loc_getitem_time_objectÚ
|
spmatrix_tZ
|
coo_matrixZ
|
csc_matrixZ
|
csr_matrixr° c C s dd l }t|j|}d\}}|tj|||d|d}tj |¡}td|}|j| j } |
|
¡ | }
|
t | |
|
¡ |j| j j } t |t|dd¡}
|
t | |
|
¡ d S )Nr )rV rT r¯ r0 )Z
|
fill_value)Ú scipy.sparseÚgetattrÚsparserh Úeyer Ú from_spmatrixr§ r# rr Ztoarrayr$ Zassert_numpy_array_equalrB Úfullr ) r7 r r° ZscipyÚrowsrÇ Zspmatrixr( Zitr_idxr r) r* r* r+ Ú$test_loc_getitem_range_from_spmatrixù s
|
z;TestLocBaseIndependent.test_loc_getitem_range_from_spmatrixc C s< t dtjddgtddi}|jddg }t ||¡ d S )Nr" r rô r¯ rM )r rp r r r# r$ r% rW r* r* r+ Ú,test_loc_getitem_listlike_all_retains_sparse s zCTestLocBaseIndependent.test_loc_getitem_listlike_all_retains_sparsec C s¦ ddl m} tj |d¡}|jtd }tdddddgdddddggtddd}t ||¡ |jtd jtd }tdddddggtddd}t ||¡ d S )
|
Nr )r rV r0 rû rà r® r¯ rM )
|
r r r r r r# r§ r r$ r% )r7 r r( r r) r* r* r+ Ú test_loc_getitem_sparse_frame s þ ÿz4TestLocBaseIndependent.test_loc_getitem_sparse_framec C s t dddddgtddd}|jtd }t ddgtddd}t ||¡ |jtd jtd }t ddgtddd}t ||¡ d S )Nrû rà r® r¯ r0 rR )r r r# r§ r$ r¨ )r7 rË r r) r* r* r+ Ú test_loc_getitem_sparse_series+ s z5TestLocBaseIndependent.test_loc_getitem_sparse_seriesr# r c C sV t dddgdddggttd}t||d }tdddgttdd}t ||¡ d S )Nrû rà ç ø?r´ r¯ r )r° rw )r r r r r r$ r¨ )r7 r¹ r( r r) r* r* r+ Ú!test_getitem_single_row_sparse_df7 s z8TestLocBaseIndependent.test_getitem_single_row_sparse_dfÚkey_typec C sH |dddg}|j d d
|f }|j d d
dddgf }t ||¡ d S )Nr" rl r )r# r$ r% )r7 Ú float_framer Úidxr r) r* r* r+ Útest_loc_getitem_iterable? s z0TestLocBaseIndependent.test_loc_getitem_iterablec C sb t tjjdd}tdddd|_|jtdd
d d
f }|jdd
d d
f }t ||¡ d S )N©rQ rU râ Z0srQ rË ©ÚstartÚperiodsr~ )
|
r rh r¦ Únormalr rg r# r r$ r% r© r* r* r+ Ú#test_loc_getitem_timedelta_0secondsF s
|
z:TestLocBaseIndependent.test_loc_getitem_timedelta_0secondsú val,expectedì ÿÿÿÿ ì c C s4 t ddgddgd}|j| }||_t ||¡ d S )NrM r0 r r¡ rx )r r# rw r$ r¨ )r7 ró r) r( r r* r* r+ Ú test_loc_getitem_uint64_scalarN s
|
z5TestLocBaseIndependent.test_loc_getitem_uint64_scalarc C sP |}t dddgtdddg|dd}| ¡ }d |jd< d |jd
|
< t ||¡ d S ) Nr r r< r ç à?rM r¯ rx Zzoor0 )r r r· r# r r$ r¨ )r7 Úfloat_numpy_dtyper° r r) r* r* r+ Ú+test_loc_setitem_int_label_with_float_indexY s
|
|
zBTestLocBaseIndependent.test_loc_setitem_int_label_with_float_indexzindexer, expectedrC c C s\ t tddd}tdtdid|d}d|j|j| df< t||dgdd}t ||¡ d S ) NrQ rË ©r r¤ rô ©r° rg rC r )r r§ r r# rg r$ r% )r7 r¹ r) Útdir( r* r* r+ Ú/test_loc_setitem_listlike_with_timedelta64indexd s üzFTestLocBaseIndependent.test_loc_setitem_listlike_with_timedelta64indexc C s t dddddgtdd}t dddddgtdd}tddgddgd|jdd
df< tddgddgd|jdd
df< t ||¡ d S )
|
NrM Zaaaaar½ r Zaabbar ©Ú
|
categoriesr0 rR )r r¶ r r# r$ r% )r7 r( r| r* r* r+ Ú8test_loc_setitem_categorical_values_partial_column_slice~ s
|
""zOTestLocBaseIndependent.test_loc_setitem_categorical_values_partial_column_slicec C s t dgdgd}t|d dddgd}||jd d
df< |d }t||jdd t¡}t ||¡ ||d< t |d t|dd ¡ d S )
|
Nr r )ÚAlphaZNumericr r r< rª r¬ rv ) r r r# r rg rÆ rÚ r$ r¨ )r7 r( r« r r) r* r* r+ Ú'test_loc_setitem_single_row_categorical s z>TestLocBaseIndependent.test_loc_setitem_single_row_categoricalc C s t dtdgd i}t d¡|jdd
df< td|jd ksDttd|jd ksZttd d
|
d
|
|jd < td ¡ |jd kstd S ) Nr< z
|
2010-10-01rR z
|
2008-08-08r rM )r r< )rM r< iÕ rV )r0 r< z
|
2005-05-05)r r rh r r# r{ r r r* r* r+ Ú"test_loc_setitem_datetime_coercion s z9TestLocBaseIndependent.test_loc_setitem_datetime_coercionÚidxerÚvarc C sX |}t ddd|d}td|dgd}t|dgtjd}||jd d
|f< t ||¡ d S ) Nz
|
2015-07-12rR ÚH)r r r~ Útzr r± rÒ r )r r rh r® r# r$ r% )r7 r° Ztz_naive_fixturer³ r r) r r* r* r+ Ú!test_loc_setitem_datetimeindex_tz¡ s z8TestLocBaseIndependent.test_loc_setitem_datetimeindex_tzc
|
C s| t dddd}ttj t|d¡|d}tddd}ttd ddtd
|
dd}d d d dg}ddddddddddddg }| ¡ }d|j |< |j | }|j | ¡ } d| j d d
< |r¼|
|
t ¡} t || ¡ | ¡ }d|j |< |j| |j |< t ||¡ | ¡ }d|j |< |j | }|j | ¡ } d| j d d
< |r>|
|
t ¡} t || ¡ | ¡ }d|j |< |j| |j |< t ||¡ d S )Nz
|
2012-01-01z
|
2012-01-05Z30minr} rV rx r< r rx é é éH éx é¨ é é é éJ éK éL éz é{ é| éª é« é¬ )r r rh r¦ r0 rÍ r r
r· r# rÆ r r$ r% r )
|
r7 Zusing_array_managerrg r( ZakeyZbkeyZaindsZbindsr r) r* r* r+ Útest_loc_setitem_time_key s<
|
|
|
|
|
|
|
|
z0TestLocBaseIndependent.test_loc_setitem_time_keyrI c C s¼ t dddg¡}tdddgddd gg|d
|
}| ¡ }tjd d d |jd d
|f< tdddgdddgg|d
|
}t ||¡ |j dd}tjd d d |jd d
|f< |j dd}t ||¡ d S )N)r" rU )rl Ú3)r" Ú2rM r0 rR rU rV rS ru )r0 r0 rô r¯ r rK )
|
r Ú from_tuplesr r· rh Úzerosr# r$ r% rM )r7 rI r r( r9 r) r* r* r+ Ú,test_loc_setitem_unsorted_multiindex_columnsÓ s zCTestLocBaseIndependent.test_loc_setitem_unsorted_multiindex_columnsc C s> t dddg|d}d|jd< t dddg|d}t ||¡ d S )NrM r0 rR r¯ rU r ©r r# r$ r¨ )r7 Zany_int_numpy_dtyper' r) r* r* r+ Útest_loc_setitem_uint_dropâ s
|
z1TestLocBaseIndependent.test_loc_setitem_uint_dropc C sp t dt dd¡g }t dd¡|jdddg< t dt dd¡g }tt dd¡|jdddg< t ||¡ d S )NrQ r rC rM r0 rR )r rh Ú timedelta64r# r r$ r¨ )r7 r r) r* r* r+ Ú test_loc_setitem_td64_non_nanoê s
|
z5TestLocBaseIndependent.test_loc_setitem_td64_non_nanoc C s t j dd¡}ttd}d ddg¡}tjt|d ||j td< W 5 Q R X d}tjt|d ||j d d
< W 5 Q R X d S )Nr0 rq z,shape mismatch: value array of shape \(2,2\)z0cannot reshape array of size 4 into shape \(2,\)r zEcould not broadcast input array from shape \(2,2\) into shape \(2,?\))
|
rh r¦ r0 r r§ rr r r rs r# )r7 rm r r r* r* r+ Ú test_loc_setitem_2d_to_1d_raisesò s þÿz7TestLocBaseIndependent.test_loc_setitem_2d_to_1d_raisesc C sX t jddd}tdddgdddgd d
|
d gg|d d dgd}d}|jd }t ||¡ d S )Nr rR )r r rM r0 rU rV rS rT r` r: r" rl r rÒ )r£ r" )rp Úinterval_ranger r# r$ Úassert_almost_equal)r7 rg r( r) r r* r* r+ Ú test_loc_getitem_interval_index s ÿ
|
z6TestLocBaseIndependent.test_loc_getitem_interval_indexc C sz t jdddd}tdddgddd gd
|
d d gg|d ddgd}t jddddd}tddg|d d}|jd }t ||¡ d S )Nr rR Zboth)r r ÚclosedrM r0 rU rV rS rT r` r: r" rl r rÒ )r r r~ rÔ r¬ ©rM r" )rp rÑ r r r# r$ r¨ )r7 rg r( Z index_expr) r r* r* r+ Ú test_loc_getitem_interval_index2 s ÿ
|
z7TestLocBaseIndependent.test_loc_getitem_interval_index2Útpl©rM ©rM r0 c C sR t ddgddd}t|d}|j|g }t |gddd}t|d}t ||¡ d S )NrØ rÙ r" F©rw Z tupleize_colsrx )r r r# r$ r% )r7 r× r r( r r) r* r* r+ Ú+test_loc_getitem_index_single_double_tuples s ý
|
|
zBTestLocBaseIndependent.test_loc_getitem_index_single_double_tuplesc C sn t dddg}|dd}|dd}t||gdd d
|
}td d g|d dgd}|j|dd d }|dksjtd S )NÚ IndexTyper r r r r5 ZbofZcomposite_indexFrÚ rÙ )rR rU r" rl rÒ rM )r r r r# r{ )r7 rÜ Zidx1Zidx2rg r( r r* r* r+ Ú!test_loc_getitem_index_namedtuple) s
|
|
z8TestLocBaseIndependent.test_loc_getitem_index_namedtuplec C s t tj dd¡dddddgdd d
|
gd }d |d < tj|j|jd d d
d f< tjtjd tjd tjgtd}t
|
|d j |¡ d S )NrV rR r r r< rÊ rÐ r r r5 rÒ Úquxr& r0 r¯ ) r rh r¦ r0 r r# rg r rÚ r$ rÒ rr rj r* r* r+ Ú$test_loc_setitem_single_column_mixed3 s ý z;TestLocBaseIndependent.test_loc_setitem_single_column_mixedc C sp t tj dd¡tdd}tj|d< d|jd< |j}tt d¡gd t d ¡g d
|
d d dgd }t
|
||¡ d S )NrX rR ÚABCru Úeventr )rQ rá r® rÚ r" rl r rx ) r rh r¦ rí Útupler r# rB r r° r$ r¨ r r* r* r+ Útest_loc_setitem_cast2> s
|
|
|
þz-TestLocBaseIndependent.test_loc_setitem_cast2c C sb t dtjdtjdi}d|jd< |jjt tj¡ks:tt d¡|_|jjt tj¡ks^td S )NÚonerS r¯ )rM rä rT ) r rh ri Zint8r# rB rä r° r{ r r* r* r+ Útest_loc_setitem_cast3K s
|
|
z-TestLocBaseIndependent.test_loc_setitem_cast3c C st |t ddddddgd}dd d
|
g}|jdkr>dgd gd
|
gg}||jt d< |ddd dd
|
g|jd}t ||¡ d S ) NrV rR rU rM r r0 rx r: rQ r; )r§ Úndimr# rg r$ rz )r7 r r9 rr r) r* r* r+ Útest_loc_setitem_range_keyS s
|
|
z1TestLocBaseIndependent.test_loc_setitem_range_key)jr r r r r r r r r r¡ rª r rº rÁ rÈ rÌ rÎ rÏ rÖ rÜ rÞ rá rë rï Úfixturerõ rh r rô rø rú rü rþ r r r r r r! r" r- r1 r2 r4 r7 r
r9 r r r? rC rE rG rJ rN Zarm_slowrV rX r[ r^ rd rh ri rm rn rp rt ru rv rw r{ r r® ÚcomplexÚtdZskip_if_no_scipyr r r r r Úiterr 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 ø sf þþóþ
|
|
#
|
'
|
*þ
|
1
|
|
% ÿ $
|
þý
|
(
|
þý ÿý$
|
þý $
|
þý ÿýØþ3
|
" "
|
'
|
|
|
|
ÿ
|
ûþ
|
|
|
&
|
|
|
r c @ sN e Zd Zejejejgddd Zejdd Z dd Z
|
dd Z d
|
d Z d S ) ÚTestLocWithEllipsis)Úparamsc C s |j S r[ )Úparam)r7 r8 r* r* r+ r¹ b s zTestLocWithEllipsis.indexerc C s |}|t k r| ¡ }|S r[ )r Úto_frame)r7 Úseries_with_simple_indexr r9 r* r* r+ r9 g s zTestLocWithEllipsis.objc C s ||d }t ||¡ d S )N.)r$ rz )r7 r9 r¹ r r* r* r+ Ú test_loc_iloc_getitem_ellipsisn s z2TestLocWithEllipsis.test_loc_iloc_getitem_ellipsisc
|
C s |}|t jkst|dkr dn|jd }|t jkr@t|jr@d S |t jkrt|jtrd}tj t
|
|d ||d|gf }W 5 Q R X n8t|dkrÀ||d|gf }|||g }t ||¡ |t jkrÎdn|j }| ¡ } || d|gf }|| d d
|gf }t ||¡ d S )Nr z2MultiIndex does not support indexing with Ellipsisr .)r$ r rÍ rg r# r ry r r r ÚNotImplementedErrorr¨ rw rï r% )
|
r7 rð r¹ r9 rI r r r) Úkey2r( r* r* r+ Ú&test_loc_iloc_getitem_leading_ellipsesr s" $ z:TestLocWithEllipsis.test_loc_iloc_getitem_leading_ellipsesc C sÔ |t jkst|dkrdn|jd }tjttd ||d W 5 Q R X tjttd ||d|gdf W 5 Q R X tjttd ||dd|f W 5 Q R X tjtdd |||ddf W 5 Q R X d S )Nr r )...úToo many indexers)r$ r rÍ rg r r r r )r7 r9 r¹ rI r* r* r+ Ú0test_loc_iloc_getitem_ellipses_only_one_ellipsis s $ zDTestLocWithEllipsis.test_loc_iloc_getitem_ellipses_only_one_ellipsisN) r r r r rè r$ r# r r¹ r9 rñ rô rö r* r* r* r+ rì a s
|
|
rì c @ sÐ e Zd Zej dddgddddgddddggfddgddddgddddggfddgddgfddddgddddggfddgddgfddddgddddggfddgddgfddddgddddggfddgddgfddddgddddggfddgddgfdddgdddggfg¡ej ddd g¡d
|
d Zd d Zdd Zdd Z dd Z
|
dd Z dd Z dd Z dd Zd d Zd d Zej d ejejd!ejf¡d"d# Zd$d% Zej d&d'd(g¡d)d* Zd+d, Zd-d. Zd!S )/ÚTestLocWithMultiIndexzkeys, expectedr r rM r0 r< Údimrg rf c C s |dddddgdddddggi}t t d¡ dd¡f|}t |¡}|dkrn|j|d d
f }t |j |¡ n(|d kr|jd d
|f }t |j
|
|¡ d S )
|
Nr< r r rM r0 é rV rg rf ) r rh ri Úreshaper r r# r$ Úassert_index_equalrg rf )r7 rø Úkeysr) Úkwargsr( Ú exp_indexr} r* r* r+ Ú'test_loc_getitem_multilevel_index_order s
|
z=TestLocWithMultiIndex.test_loc_getitem_multilevel_index_orderc C s |}|j d }|d j d }|jj|jjdd
ks8t|jj|jjdd
ksTt|j d }|d j d }|jj|jjd kst|jj|jjd kstd S )NéÐ r" rM )r r0 r0 )r# rg r r{ rw )r7 Z/multiindex_year_month_day_dataframe_random_dataZymdr Úresult2r* r* r+ Útest_loc_preserve_names¸ s
|
|
z-TestLocWithMultiIndex.test_loc_preserve_namesc C sf t dgddgg¡}td|d}|jg }|d d
}t ||¡ |j|jdd
}t ||¡ d S )Nr rM rx )r Ú from_productr r# r$ r¨ r )r7 r r r} r) Zres2r* r* r+ Ú.test_loc_getitem_multiindex_nonunique_len_zeroÅ s
|
zDTestLocWithMultiIndex.test_loc_getitem_multiindex_nonunique_len_zeroc C s¤ t d gt dgdgg¡}|jd }|d ks0tt dgddgg¡}t d gt| t|d}|jd }|d ksrtt dgt| t|d}|jd }|dks td S ) NÚLevel1ÚLevel2)r r ÚLevel2_aZLevel2_br§ )r r rM )r r r r# r{ r rÍ rÚ )r7 r r Zmidxr* r* r+ Ú0test_loc_getitem_access_none_value_in_multiindexÒ s
|
|
|
zFTestLocWithMultiIndex.test_loc_getitem_access_none_value_in_multiindexc C s t jtddddddddgddddddddgddgd }td
|
d
|
d
|
d
|
d
|
d
|
d
|
d
|
g|d }d |jd d
< td
|
d
|
d d d d d
|
d
|
g|d }t ||¡ d S )Nr r5 r rÞ rä ÚtwoÚfirstÚsecondr rM rx r» )r5 rä )r r )r rÉ ra r r# r$ r¨ )r7 rg r r) r* r* r+ Ú!test_loc_setitem_multiindex_sliceã s þû z7TestLocWithMultiIndex.test_loc_setitem_multiindex_slicec C sJ t dddd}ttd|}|jtdddtddd
}t ||¡ d S )Nrð Z10minrP ©r~ r il rM i4 )r r r§ r# r r$ r¨ )r7 Útimesr r r* r* r+ Ú7test_loc_getitem_slice_datetime_objs_with_datetimeindexõ s zMTestLocWithMultiIndex.test_loc_getitem_slice_datetime_objs_with_datetimeindexc C sh t tdtddtddd}|jddgddgf }t d d
|
gd d
|
gdtddgd}t ||¡ d S ) NrQ r½ z
|
2010-01-01z
|
2010-01-10rx z
|
2010-01-05r r r rU )r r§ r r# r r$ r% r r* r* r+ Ú3test_loc_getitem_datetime_string_with_datetimeindexû s þ
|
þzITestLocWithMultiIndex.test_loc_getitem_datetime_string_with_datetimeindexc C sª t jddddddgddgd}tddgdd gd
|
d gd d gddgddgg|ddgd}|jdd}tddgdd gddggddg| ddd g¡d}|jd }t ||¡ d S )N)r r )ZbahZbamr" rl r rû rM r´ r0 rN rR ç @rU ç @rV ç @rS r rì rÒ r )Úlevelre )r rÉ r rM rS r# r$ r% )r7 r r( r) r r* r* r+ Ú3test_loc_getitem_sorted_index_level_with_duplicates s: ú÷ úö ÿ
|
zITestLocWithMultiIndex.test_loc_getitem_sorted_index_level_with_duplicatesc C s@ t dddgdd}d|jd< t ddddgdd}t ||¡ d S ) Nr r r< Úcategoryr¯ r rR rÚ rÌ ©r7 r r) r* r* r+ Ú1test_additional_element_to_categorical_series_loc* s
|
zGTestLocWithMultiIndex.test_additional_element_to_categorical_series_locc C s@ t dddgdd}d|jd< t ddddgdd}t ||¡ d S ©Nr r r< r r¯ rR rÌ r r* r* r+ Ú'test_additional_categorical_element_loc1 s
|
z=TestLocWithMultiIndex.test_additional_categorical_element_locc C sª t dddgttdddg|dd}tj|jd< t dddtjgttdddg|dd}t ||¡ tj|jd< t dtjdtjgttdddg|dd}t ||¡ d S )NrM r0 rR r¯ )r r
|
r rh r r# r$ r¨ )r7 Zany_numeric_ea_dtypeZsrsr) r* r* r+ Ú&test_loc_set_nan_in_categorical_series8 s þ þ þz<TestLocWithMultiIndex.test_loc_set_nan_in_categorical_seriesÚnaNc C sj t dddgdd}||jd< t ddddgdd}||jd< t ||¡ t ddd|gdd}t ||¡ d S r rÌ )r7 r Z srs_enlargeZ srs_setintor) r* r* r+ Ú,test_loc_consistency_series_enlarge_set_intoM s
|
|
zBTestLocWithMultiIndex.test_loc_consistency_series_enlarge_set_intoc C s¢ t t ddd¡ttddgtdgdgd dgd tdgd d
|
gd d }tddgddgd d dd}|jjd }t ||¡ |j
|
dg jjd }t ||¡ d S )Nr0 é r r rQ r rV rM ZIndex1ZIndex2)r
|
Úcodesr )rm rg Fr )r« Úorderedrw r° ) r rh ri r r r§ rg r
|
r$ rû r# r© r* r* r+ Ú5test_loc_getitem_preserves_index_level_category_dtypeZ s& ýþ û zKTestLocWithMultiIndex.test_loc_getitem_preserves_index_level_category_dtypeÚlt_valuerX rQ c C sp t dddddgitdddd d
|
ggd }tjtd d 0 |j|d |k d d
f jdgd d
f W 5 Q R X d S )Nr r< é é" é- Úaabbr rM r0 rR rx z\['b'\] not in indexr r )r r¶ r r r5 r# )r7 r" r( r* r* r+ Ú>test_loc_multiindex_levels_contain_values_not_in_index_anymores s (zTTestLocWithMultiIndex.test_loc_multiindex_levels_contain_values_not_in_index_anymorec C sl t t jt jg¡}ddg}t ||g¡}tddg|d}|jd d
df }tdgt jgd}t ||¡ d S )Nr r5 r rM rx ) rh r r r r r r# r$ r¨ )r7 Úlev1Úlev2r r r r) r* r* r+ Ú'test_loc_multiindex_null_slice_na_levelz s z=TestLocWithMultiIndex.test_loc_multiindex_null_slice_na_levelc C st t jtdtdddggdddgd}ttd|d}|jd d d
d d
f }|j d
|
¡d d
}t |j|¡ d S ) NÚabZxyrM r0 Únumr r` rx r r rU )
|
r r r¶ r r§ r# rg Z droplevelr$ rû )r7 r r Z
|
loc_resultr) r* r* r+ Útest_loc_drops_level s ÿz*TestLocWithMultiIndex.test_loc_drops_level)r r r r r r rÿ r r r r r r r r r r rh r rp ZNArq r r! r' r* r- r* r* r* r+ r÷ s< &&&&"ùþ "
|
|
r÷ c @ sÌ e Zd Zejjdd Zdd Zdd Zdd Z d d
|
Z
|
ejj d d d dd dd dd gddddgddd Z dd Z dd Zd d Zd d Zej d ¡d!d" Zej d#d$d%d&d'd(d)g¡d*d+ Zd,S )-ÚTestLocSetitemWithExpansionc C sN t dtdidd}t|d |jt|< t dtdidd}t ||¡ d S )Nr¤ i@B rô r¯ rM iAB )r r§ rÍ r# r$ r% r r* r* r+ Ú/test_loc_setitem_with_expansion_large_dataframe s zKTestLocSetitemWithExpansion.test_loc_setitem_with_expansion_large_dataframec C sV t td}d|jd< t |t dgdgd¡ d|jd< t |t ddgddgd¡ d S )Nr¯ rM rx rR ©r rÚ r# r$ r¨ ©r7 r r* r* r+ Ú test_loc_setitem_empty_series s
|
|
|
|
z9TestLocSetitemWithExpansion.test_loc_setitem_empty_seriesc C sV t td}d|jd< t |t dgdgd¡ d|jd< t |t ddgddgd¡ d S )Nr¯ rû rM rx rN rR r0 r1 r* r* r+ Ú#test_loc_setitem_empty_series_float¦ s
|
|
|
|
z?TestLocSetitemWithExpansion.test_loc_setitem_empty_series_floatc C s t td}d|jd< t |t dgdgd¡ d|jd< t |t ddgddgd¡ d|jd< t |t dddgdddgd¡ d S )Nr¯ rM r rx rR r rU r0 r1 r* r* r+ Ú%test_loc_setitem_empty_series_str_idx° s
|
|
|
|
zATestLocSetitemWithExpansion.test_loc_setitem_empty_series_str_idxc sn t dddtdd fddtdD }td g|d gd
|
}|D ]}d|j|< qBtd|d
|
}t ||¡ d S ) Nr r; rM ú
|
US/Pacific)Útzinfoc s g | ]} t |d d qS )rO )Úseconds)r r' ©Úbaser* r+ r¥ ¿ s zUTestLocSetitemWithExpansion.test_loc_setitem_incremental_with_dst.<locals>.<listcomp>é r rx )r r r§ r r# r$ r¨ )r7 Zidxsr rB r) r* r8 r+ Ú%test_loc_setitem_incremental_with_dst¼ s zATestLocSetitemWithExpansion.test_loc_setitem_incremental_with_dstÚconvc C s | S r[ r* rå r* r* r+ ræ É rç z$TestLocSetitemWithExpansion.<lambda>c C s | ¡ S r[ )Ú to_datetime64rå r* r* r+ ræ Ê rç c C s | ¡ S r[ )Ú to_pydatetimerå r* r* r+ ræ Ë rç c C s
|
t | ¡S r[ )rh r rå r* r* r+ ræ Ì rç r7 r= r> z np.datetime64)Úidsc C sb t d}t d}t }d|j||df< d|j||df< tdddgi||gd}t ||¡ d S ) Nz20130101 09:00:00z20130101 10:00:00r» rä rb g Y@g i@rx )r r r# r$ r% )r7 r< Zdt1Zdt2r( r) r* r* r+ Ú#test_loc_setitem_datetime_keys_castÆ s z?TestLocSetitemWithExpansion.test_loc_setitem_datetime_keys_castc C sT t ddgi}tdg|d|jd d
df< t dgtdg|dd}t ||¡ d S )Nr" rM r )r rl rn )r r r# r$ r% )r7 r r r) r* r* r+ Ú1test_loc_setitem_categorical_column_retains_dtypeÛ s zMTestLocSetitemWithExpansion.test_loc_setitem_categorical_column_retains_dtypec C s t ddd}t ddd}t ddd}t||ddd}||ks@tt|d gd
|
}d |j|d f< ttjgt| d g | t |g¡d gt
|
d }t ||¡ d S ) Nz2017-10-29 00:00:00+0200z Europe/Madrid©r³ z2017-10-29 03:00:00+0100z2016-10-10 03:00:00Úleftr² )Z inclusiver~ rÝ rÒ r< r ) r r r{ r r# rh r rÍ Úappendr rÚ r$ r% )r7 r ÚendrB r r r) r* r* r+ Ú0test_loc_setitem_with_expansion_and_existing_dstâ s üzLTestLocSetitemWithExpansion.test_loc_setitem_with_expansion_and_existing_dstc C sì t tddgdgd}ddg|d< | d¡j d¡|_||jdk d¡j d ¡}| ¡ }||j |jdkdf< t
|
|d
|
d¡|j d gdd }t |j|¡ |j |jdkdf t d }||j |jdkdf< t |j |jdkdf |¡ d S )Nz2015-03-30 20:12:32z2015-03-12 00:11:11r )rm rf ÚnewÚoldÚnew_colr@ r5 r )rM r rv Z1s)r r r rg rA r rI Z
|
tz_convertr· r# r r$ r¨ r )r7 r( Úvrs r) r* r* r+ Útest_setitem_with_expansionô s
|
þ z7TestLocSetitemWithExpansion.test_setitem_with_expansionc C sV t }d|jd< d|jd< d|jdtjf< |j}tddtjgtjd}t ||¡ d S )NrM r/ r0 )rM rM rR r r¯ ) r r# rh Úinfrf r r® r$ rû r r* r* r+ Ú0test_loc_setitem_with_expansion_inf_upcast_empty
|
s
|
|
zLTestLocSetitemWithExpansion.test_loc_setitem_with_expansion_inf_upcast_emptyz"ignore:indexing past lexsort depthc C s` t |s d S | d¡}t |}t |¡ tj¡}t||dgd}d}||ksPt| t ||¡}t |t
|
r|d d |kstn|d |kstt |d ¡ tj ¡}t||dgd}| ¡ } || j |df< t | |¡ | ¡ d }
|
||
|
j |< |d tj¡}t |
|
|¡ | ¡ } || j |df< tt|tjg tjg| t|g d|d}t | |¡ d S ) Nr0 r rÒ Zkapowr× rM ©r rM rx )rÍ Úrepeatrh ri rÆ rô r r{ Úinsertry r r® r· r# r$ r% r¨ r¶ r r ) r7 rg ÚNÚarrÚorigrI rþ Zexp_datar) r( r r* r* r+ Ú/test_loc_setitem_with_expansion_nonunique_index s8
|
|
|
&þzKTestLocSetitemWithExpansion.test_loc_setitem_with_expansion_nonunique_indexr° ZInt32ÚInt64ZUInt32ZUInt64ZFloat32ZFloat64c C sv t ddddg|d}td|i}t|jd}||j|jdf< t ||¡ t|jd}|j|j|jdf< t ||¡ d S )Nr rM r0 rR r¯ rm rx )r r rg r# r$ r% r\ )r7 r° r r( r r* r* r+ Ú6test_loc_setitem_with_expansion_preserves_nullable_intC s zRTestLocSetitemWithExpansion.test_loc_setitem_with_expansion_preserves_nullable_intN)r r r r r Zslowr/ r2 r3 r4 r; r r@ rA rF rK rM ÚfilterwarningsrT rV r* r* r* r+ r. s6
|
|
|
|
ü
|
ø
|
|
|
|
- ÿr. c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
ÚTestLocCallablec C s t ddddgtdddddgd}|jdd }t ||j|jdk ¡ |jd d d d
f }t ||j|jd
|
kd d
f ¡ |jd d d d f }t ||j|jdkd dd gf ¡ |jdd dd f }t ||j|jdkdf ¡ |jdd dd f }t ||j|jdkddgf ¡ |jdd dd f }t ||j|jdkddgf ¡ |jdd dd f }||jd kstd S )NrM r0 rR rU r& ©r" rl r c S s
|
| j dkS ©Nr0 ©r" rå r* r* r+ ræ \ rç zATestLocCallable.test_frame_loc_getitem_callable.<locals>.<lambda>c S s
|
| j dkS )Nr )rl rå r* r* r+ ræ _ rç r c S s
|
| j dkS rZ r[ rå r* r* r+ ræ b rç c S s
|
| j dkS ©Nrl ru rå r* r* r+ ræ b rç FTc S s
|
| j dkS rZ r[ rå r* r* r+ ræ e rç c S s dS r\ r* rå r* r* r+ ræ e rç rl c S s
|
| j dkS rZ r[ rå r* r* r+ ræ h rç c S s ddgS ©Nr" rl r* rå r* r* r+ ræ h rç r" c S s
|
| j dkS rZ r[ rå r* r* r+ ræ k rç c S s ddgS r] r* rå r* r* r+ ræ k rç c S s dS )NrM r* rå r* r* r+ ræ o rç c S s dS r! r* rå r* r* r+ ræ o rç rÕ ) r r¶ r# r$ r% r" rl r¨ r{ ©r7 r( r} r* r* r+ Ú test_frame_loc_getitem_callableV s $ " z/TestLocCallable.test_frame_loc_getitem_callablec C s¾ t ddddgtdddddgd}|jdd d d
|
gf }t ||j|jdkd d
|
gf ¡ |jddgd d f }t ||jddgd d
|
gf ¡ |jdd d f }t ||jdd d
|
gf ¡ d S ) NrM r0 rR rU r& rY c S s
|
| j dkS rZ r[ rå r* r* r+ ræ v rç zITestLocCallable.test_frame_loc_getitem_callable_mixture.<locals>.<lambda>r" rl c S s ddgS r] r* rå r* r* r+ ræ y rç c S s ddgS r] r* rå r* r* r+ ræ | rç )r r¶ r# r$ r% r" r¨ r^ r* r* r+ Ú'test_frame_loc_getitem_callable_mixturer s $ z7TestLocCallable.test_frame_loc_getitem_callable_mixturec C s t ddddgtddtdd}|jd d
|
}t ||jd d g ¡ |jd d
|
d d
f }t ||jd d gd d
f ¡ |jdd
|
dd
|
f }t ||jd d gdf ¡ |jdd
|
dd
|
f }t ||jd d gdgf ¡ |jd d gdd
|
f }t ||jd d gdf ¡ |jd d gdd
|
f }t ||jd d gdgf ¡ |jdd
|
df }t ||jd d gdf ¡ |jdd
|
dgf }t ||jd d gdgf ¡ d S )NrM r0 rR rU r& ©ÚXr r. rx c S s ddgS ©Nr" r r* rå r* r* r+ ræ rç zHTestLocCallable.test_frame_loc_getitem_callable_labels.<locals>.<lambda>r" r c S s ddgS rc r* rå r* r* r+ ræ rç c S s ddgS rc r* rå r* r* r+ ræ rç c S s dS ©Nrb r* rå r* r* r+ ræ rç rb c S s ddgS rc r* rå r* r* r+ ræ rç c S s dgS rd r* rå r* r* r+ ræ rç c S s dS rd r* rå r* r* r+ ræ rç c S s dgS rd r* rå r* r* r+ ræ rç c S s ddgS rc r* rå r* r* r+ ræ rç c S s ddgS rc r* rå r* r* r+ ræ rç )r r¶ r# r$ r% r¨ r^ r* r* r+ Ú&test_frame_loc_getitem_callable_labels s" " z6TestLocCallable.test_frame_loc_getitem_callable_labelsc C sZ t ddddgtddtdd}| ¡ }d |jd
|
d < | ¡ }d |jd d g< t ||¡ | ¡ }d|jdd d d
f< | ¡ }d|jd d gd d
f< t ||¡ | ¡ }d|jdd dd f< | ¡ }d|jd d gdf< t ||¡ | ¡ }ddg|jdd dd f< | ¡ }ddg|jd d gdgf< t ||¡ | ¡ }t ddg¡|jd d gdd f< | ¡ }t ddg¡|jd d gdf< t ||¡ | ¡ }d|jd d gdd f< | ¡ }d|jd d gdgf< t ||¡ | ¡ }d|jdd df< | ¡ }d|jd d gdf< t ||¡ | ¡ }d |jd d dgf< | ¡ }d |jd d gdgf< t ||¡ d S ) NrM r0 rR rU r& ra r. rx iìÿÿÿc S s ddgS rc r* rå r* r* r+ ræ £ rç zATestLocCallable.test_frame_loc_setitem_callable.<locals>.<lambda>r" r rC c S s ddgS rc r* rå r* r* r+ ræ © rç r× c S s ddgS rc r* rå r* r* r+ ræ ¯ rç c S s dS rd r* rå r* r* r+ ræ ¯ rç rb rV rQ c S s ddgS rc r* rå r* r* r+ ræ µ rç c S s dgS rd r* rå r* r* r+ ræ µ rç rØ c S s dS rd r* rå r* r* r+ ræ ¼ rç c S s dgS rd r* rå r* r* r+ ræ Â rç c S s ddgS rc r* rå r* r* r+ ræ È rç éüÿÿÿc S s ddgS rc r* rå r* r* r+ ræ Î rç )r r¶ r· r# r$ r% rh r )r7 r( r} r| r* r* r+ Ú test_frame_loc_setitem_callable sR " z/TestLocCallable.test_frame_loc_setitem_callableN)r r r r_ r` re rg r* r* r* r+ rX U s rX c @ sF e Zd Zdd Zdd Zdd Zdd Zej d d
|
d g¡d d Z
|
d
|
S )ÚTestPartialStringSlicingc C sd t dddgdddgdtdd d
|
gd }t ddgddgdtdd
|
gd }|jd }t ||¡ d S ) Nr r r< rM r0 rR )rc rd z
|
2020-08-01z
|
2020-07-02z
|
2020-08-05rx z2020-08)r r r# r$ r% r© r* r* r+ Ú5test_loc_getitem_partial_string_slicing_datetimeindexÕ s þ
|
þ
|
zNTestPartialStringSlicing.test_loc_getitem_partial_string_slicing_datetimeindexc C sD t jdddd}| ¡ }|jd d
}|jd d
}t ||¡ d S )Nú
|
2017-01-01ú
|
2018-01-01r ©r rE r~ z2017-12r× )rp Ú period_rangeÚ to_seriesr# r r$ r¨ )r7 Úpir r r) r* r* r+ Ú8test_loc_getitem_partial_string_slicing_with_periodindexâ s
|
zQTestPartialStringSlicing.test_loc_getitem_partial_string_slicing_with_periodindexc C sB t dddd}| ¡ }|jd d
}|jd d
}t ||¡ d S )Nú1 dayú2 daysÚ1Hrl z1 daysr× )r rn r# r r$ r¨ )r7 rÀ r r r) r* r* r+ Ú;test_loc_getitem_partial_string_slicing_with_timedeltaindexê s
|
zTTestPartialStringSlicing.test_loc_getitem_partial_string_slicing_with_timedeltaindexc C sD t dtdittdddd}|jd }|jd }t ||¡ d S )Nr¤ rR Údaysr¦ rx r ú0 days)r r§ r r r# r$ r¨ )r7 r( r) Zslicedr* r* r+ Ú#test_loc_getitem_str_timedeltaindexò s
|
|
z<TestPartialStringSlicing.test_loc_getitem_str_timedeltaindexÚ indexer_endNú 2020-01-02 23:59:59.999999999c C s |dgd t tdtdtdtdtdg|dd }|dgd
|
t tdtdg|dd }td|}|| }t ||¡ |j| }t ||¡ d S ) NrM rV z
|
2019-12-30z
|
2020-01-01z
|
2019-12-25ry z
|
2019-12-19rB rx r0 )r r r
r$ rz r# )r7 Ztz_aware_fixturerx r r9 r) r¹ r r* r* r+ Ú/test_loc_getitem_partial_slice_non_monotonicityù s4 ûøþ þûþ
|
|
|
zHTestPartialStringSlicing.test_loc_getitem_partial_slice_non_monotonicity) r r r ri rp rt rw r r r rz r* r* r* r+ rh Ô s rh c
|
@ sø e Zd Zdd Zdd Zej dej dddd e
|
d
|
d d d g¡d d Z dd Z dd Z ej de dd¡deddge dd¡e dd¡eddgde d d¡edd gg¡d d Zej d d!d"g¡d#d$ Zej d%dd&g¡d'd( Zd)d* ZdS )+ÚTestLabelSlicingc C s t tjddddd dD d}t tjdddd d d
|
D d}||fD ]Æ}|jtd d d d d
}t ||¡ |jd td d dd
}t ||¡ |jtd d d d td d dd
}t ||¡ |jtd d d dd
}|jd d
}t ||¡ |jdd
}t ||¡ qHd S )Nr r® r¯ c S s g | ]}t d d|ddqS ©éÑ rM rQ r ©r r' r* r* r+ r¥ % s zMTestLabelSlicing.test_loc_getitem_slicing_datetimes_frame.<locals>.<listcomp>)rM r0 rR rU rx r c S s g | ]}t d d|ddqS r| r~ r' r* r* r+ r¥ + s )rM r0 r0 rR rU r} rM rQ rU r; z 20010101 11)r rh ri r# r r$ r% r )r7 Z df_uniqueZdf_dupsr( r r) r* r* r+ Ú(test_loc_getitem_slicing_datetimes_frame s( þ þ " z9TestLabelSlicing.test_loc_getitem_slicing_datetimes_framec C s t ddddd}tdddd d
|
g|d }td dd }tddd }|j||
}tdd g|dd
|
d }t ||¡ || }d}||kstd S )Nz2017-10-29 01:30:00z Europe/BerlinrV z30 min)r³ r r~ r rM r0 rR rU rx z2017-10-29 02:30:00+02:00rB z2017-10-29 02:00:00+01:00)r r r r# r$ r¨ r{ )r7 r Zseries2Zt_1Zt_2r r) r* r* r+ Ú'test_loc_getitem_label_slice_across_dst< s ÿ z8TestLabelSlicing.test_loc_getitem_label_slice_across_dstrg rj rk r rl rq rr rs c C s8 | ¡ }|jd |d
}|jd d
}t ||¡ d S )NrØ r× )rn r# r r$ r¨ )r7 rg r r r) r* r* r+ Ú-test_loc_getitem_label_slice_period_timedeltaM s z>TestLabelSlicing.test_loc_getitem_label_slice_period_timedeltac C s~ dddg}t tj dd¡|d}|jdd
}t|dks>t|jdd
}t|dks\t|jdd
|
}t|dksztd S ) Ngæx"p|é@g\Va³|é@gßÖË|é@rR r0 rx g3333c|é@g |é@g3333|é@gÍÌÌÌÜ|é@)r rh r¦ rí r# rÍ r{ )r7 rg r( Ús1r* r* r+ Ú%test_loc_getitem_slice_floats_inexact[ s
|
z6TestLabelSlicing.test_loc_getitem_slice_floats_inexactc C s® |}t tj d¡tjdd|dd}t|jdd
dks>tt|jdd
dksXttjdd|d}d |d
|
< ||_t|jdd
dkstt|jdd
dksªtd S ) NrQ rC r¯ rx r r` g )@rT gffffff(@r0 ) r rh r¦ rí ri rÍ r# r{ rg )r7 r¤ r° r r r* r* r+ Ú'test_loc_getitem_float_slice_floatindexh s z8TestLabelSlicing.test_loc_getitem_float_slice_floatindexzstart,stop, expected_slicer ÚnsNr; rM rì rS rT rU rV c C s> t tdtdd}|jt|| }|j| }t ||¡ d S )Nr; rv z10 days)r r§ r r# r
r r$ r¨ )r7 r ÚstopZexpected_slicer r r) r* r* r+ Ú$test_loc_getitem_slice_label_td64obju s
|
|
z5TestLabelSlicing.test_loc_getitem_slice_label_td64objr Z2018Z2020c C sP |dddgt dt dt dgd}tjtdd |j|d
|
W 5 Q R X d S ) NrM r0 rR Z2016Z2019Z2017rx z,Value based partial slicing on non-monotonicr r )r r r r5 r# )r7 r r r9 r* r* r+ Ú)test_loc_getitem_slice_unordered_dt_index s þ ÿz:TestLabelSlicing.test_loc_getitem_slice_unordered_dt_indexrÝ r c C sN |t d|dddgd}|j|d
}|t d|dddgd}t ||¡ d S )NrU r
|
r0 Úthirdrx )r§ r# r$ rz )r7 r rÝ r9 r r) r* r* r+ Ú1test_loc_getitem_slice_labels_int_in_object_index s zBTestLabelSlicing.test_loc_getitem_slice_labels_int_in_object_indexc C sZ t dddddgd}t ddggdgtddgtdd}t |jd d
dd
f |¡ d S ) NrM r0 rR )ÚtestrM r0 r rx r¯ )rm rg rf )r r rÚ r$ r% r# rj r* r* r+ Ú*test_loc_getitem_slice_columns_mixed_dtype s ÿz;TestLabelSlicing.test_loc_getitem_slice_columns_mixed_dtype)r r r r r r r r rp rm r r r r rh rÎ r
r r r r r* r* r* r+ r{ s0 þþ
|
ýþ
|
|
|
|
r{ c @ sH e Zd Zej dddg¡dd Zej dddg¡dd Zdd Zd
|
S ) Ú TestLocBooleanLabelsAndSlicesÚ
|
bool_valueTFc C sH | d}|j dkrD||dd}tjt|d |j| W 5 Q R X d S )Nz7: boolean label can not be used without a boolean indexrè rÚ rö r )Z inferred_typer r r5 r# )r7 rg r r Úmessager9 r* r* r+ Ú'test_loc_bool_incompatible_index_raises¡ s
|
|
|
zETestLocBooleanLabelsAndSlices.test_loc_bool_incompatible_index_raisesc C s& |t ddgdddd}|j| d S )NTFrè r¯ rÚ rö )r r# )r7 r r r9 r* r* r+ Ú test_loc_bool_should_not_raise¬ s
|
ÿz<TestLocBooleanLabelsAndSlices.test_loc_bool_should_not_raisec C s< d}||dd}t jt|d |jdd
W 5 Q R X d S )NzEslice\(True, False, None\): boolean values can not be used in a slicerÚ rö r TF)r r r r# )r7 rg r r r9 r* r* r+ Útest_loc_bool_slice_raises³ s
|
ÿ z8TestLocBooleanLabelsAndSlices.test_loc_bool_slice_raisesN) r r r r r r r r r r* r* r* r+ r s
|
|
|
|
r c @ sd e Zd Zdd Zej dddg¡dd Zdd Zd
|
d Z e
|
j d d Z dd Z dd Zdd ZdS )ÚTestLocBooleanMaskc C sâ t dtdi}ttddd|_|d dk|d dk|d dk g}ddddddddddg
|
ddddd d
|
d d d dg
|
ddddd d
|
d d d dg
|
g}t||D ]H\}}| ¡ }d|j|df< t |ttddddgdd}t ||¡ qd S )Nr¤ rQ rË r¦ rR r rM r0 rU rV rS rT r` r: rô r ) r r§ r rg ra r· r# r$ r% )r7 r( Z
|
conditionsÚ expected_dataZcondrm r r) r* r* r+ Ú)test_loc_setitem_bool_mask_timedeltaindex¾ s" "ýüz<TestLocBooleanMask.test_loc_setitem_bool_mask_timedeltaindexr³ Nr@ c C s t ddddg¡}tdd|d}tdt d¡i|d d¡}| ¡ }|j|d d
f |j|d d
f< t ||¡ | ¡ }|j| |j|< t ||¡ d S ) NTFZ20010101rU ©r r³ r rx r® )
|
rh r r r ri rÆ r· r# r$ r% )r7 r³ rê r r( r r* r* r+ Ú+test_loc_setitem_mask_with_datetimeindex_tzÔ s z>TestLocBooleanMask.test_loc_setitem_mask_with_datetimeindex_tzc C s~ t t d¡ dd¡tdtddddd}| ¡ }|jd gtj tj g |d
|
< |j
|
d k }|j | j|j |d
|
f< t ||¡ d S ) Nr rR r0 ZABr| rs )r r~ re r r rM )r rh ri rú r¶ r r· rg rp rq r" r# r$ r% )r7 r( r) rê r* r* r+ Ú2test_loc_setitem_mask_and_label_with_datetimeindexå s ý
|
zETestLocBooleanMask.test_loc_setitem_mask_and_label_with_datetimeindexc C s t d}t d}tdt||gi}| ¡ }t|g}|d jd j}||jddg< |d jd j}||ksptt ||¡ d S )Nr l ,GsO1ÚcolrM TF)
|
r r r r· r Ú_valuer# r{ r$ r% )r7 Ztd1Ztd2r( Zdf_copyr r) r r* r* r+ Ú'test_loc_setitem_mask_td64_series_valueô s
|
z:TestLocBooleanMask.test_loc_setitem_mask_td64_series_valuec C s\ | ¡ }|d dk}d|j|df< |j ¡ }d||jdf< t||j|jd}t ||¡ d S )Nr" r rl rM rÒ )r· r# rr r rg rf r$ r% )r7 r r) rê rr r* r* r+ Ú#test_loc_setitem_boolean_and_column
|
s
|
z6TestLocBooleanMask.test_loc_setitem_boolean_and_columnc C s~ t dddgdddgd}t dd d
|
d giddd gd |jdddgdgf< t ddd gdddgd}t ||¡ t dddgdddgd}td ddgddd gd |jdddgdgf< t ||¡ t dddgdddgd}td ddgddd gd |jdddgdf< t ||¡ t dddgdddgd}| ¡ }|d }td ddgddd gd |jdddg< |rnt ||¡ n t ||¡ d S )NrM r0 rR rU rV rS r½ r rQ rC rX r rx FTr; r< )r r# r$ r% r r· )r7 r] r( r) r¸ r r* r* r+ Ú)test_loc_setitem_ndframe_values_alignment
|
s( ÿ * ( $z<TestLocBooleanMask.test_loc_setitem_ndframe_values_alignmentc C sJ t g g dtd}| ¡ }|d |jtjg tjddgf< t ||¡ d S )Nr½ r¯ r © r rÚ r· r# rh r Zbool_r$ r% rj r* r* r+ Ú test_loc_indexer_empty_broadcast,
|
s z3TestLocBooleanMask.test_loc_indexer_empty_broadcastc C sP t dgdgdtd}| ¡ }|d |jtjdgtjddgf< t ||¡ d S )Nr¤ rÿ r½ r¯ r Fr r rj r* r* r+ Ú$test_loc_indexer_all_false_broadcast3
|
s "z7TestLocBooleanMask.test_loc_indexer_all_false_broadcast)r r r r r r r r r r rê Ú skip_array_manager_invalid_testr r r r r* r* r* r+ r ½ s
|
|
r c @ s` e Zd Zej ddd ejeg¡dd Z dd Z
|
dd Z ej d
|
d d g¡d d Z dd Z dS )ÚTestLocListliker> c C s | S r[ r* rå r* r* r+ ræ <
|
rç zTestLocListlike.<lambda>c C s t ddtjg}ttd|d}|j|| }t ||¡ ||| }t ||¡ | ¡ j|| }t || ¡ ¡ |d d
}|dd
}d}t
|
j t |d |j|| W 5 Q R X t
|
j t |d ||| W 5 Q R X t
|
j t |d | ¡ j|| W 5 Q R X d S ) Nr" rl rR rx r× rM rÉ r ) r rh r r r§ r# r$ r¨ rï r% r r r5 )r7 r> Úcir r Zser2Zci2r r* r* r+ Ú8test_loc_getitem_list_of_labels_categoricalindex_with_na<
|
s" zHTestLocListlike.test_loc_getitem_list_of_labels_categoricalindex_with_nac C sZ t jddddgdd}tdddd gtd
|
d d d }tjtdd |j| W 5 Q R X d S )Nz
|
2001-01-04z
|
2001-01-02z
|
2001-01-14r r¯ r0 rV r` r; z
|
2001-01-01rì rU r rÉ r )rh r r r r r r5 r# )r7 rI r r* r* r+ Ú1test_loc_getitem_series_label_list_missing_valuesX
|
s
|
ÿ zATestLocListlike.test_loc_getitem_series_label_list_missing_valuesc C sR t t ddg¡t ddg¡d}tjtdd |jt ddg¡ W 5 Q R X d S )Nl PF
|
AHl UBÿeN l P2$Þ$én©rg rm rÉ r l NZ4FçG )r rh r r r r5 r# r1 r* r* r+ Ú9test_loc_getitem_series_label_list_missing_integer_valuesa
|
s þzITestLocListlike.test_loc_getitem_series_label_list_missing_integer_valuesÚ to_periodTFc C sD t ddddd}|r | d¡}tddg|dd }tdtdg}|rRd
|
d |D }|j| }tddg|dd }|s|j d ¡|_tj||d d tdtdtdg}|r¸dd |D }tdddgt |d|j
|
ddd }|j| }tj||d d tdtdtdg}|r dd |D }t j t dd |j| W 5 Q R X d S )Nú
|
2011-01-01z
|
2011-01-02rì r )r~ rw g¹?gÉ?rË r¬ c S s g | ]}| d ¡qS ©rì ©r¨ r¢ r* r* r+ r¥ u
|
s zRTestLocListlike.test_loc_getitem_listlike_of_datetimelike_keys.<locals>.<listcomp>T)rY c S s g | ]}| d ¡qS rª r« r¢ r* r* r+ r¥
|
s )rw r° ú
|
2011-01-03c S s g | ]}| d ¡qS rª r« r¢ r* r* r+ r¥
|
s rÉ r )r r¨ r r r# rg r r$ r¨ r r° r r r5 )r7 r¨ r r rü r r| r* r* r+ Ú.test_loc_getitem_listlike_of_datetimelike_keysj
|
sB
|
|
ý ÿ
|
ýz>TestLocListlike.test_loc_getitem_listlike_of_datetimelike_keysc C sf t ddgddgddggddd gd
|
d gd }|jd d
}d |j_|jtddgd d }t ||¡ d S )NrM r0 rU rV rT r` ZcobraZviperZ
|
sidewinderZ max_speedZshieldrÒ r rv )r r rg rw r# r r$ r% r© r* r* r+ Útest_loc_named_index
|
s ýz$TestLocListlike.test_loc_named_indexN)r r r r r r rh Zasarrayr¶ r¤ r¥ r§ r r® r* r* r* r+ r¢ ;
|
s
|
|
)r¢ z%columns, column_key, expected_columnsiÛ rD iÝ rM ZAllr0 c C sV t tj dd¡| tdd}|jd d
|f }|jdddg|f }tj||dd d S ) NrR rà re r" rl r T)rZ ) r rh r¦ rí r¶ r r# r$ r% )rf Z
|
column_keyZexpected_columnsr( r) r r* r* r+ Ú*test_loc_getitem_label_list_integer_labels¡
|
s
|
r¯ c C s t j dd¡} t| }t j|jd d
df< t | t t jgd ¡ dd¡f¡}t|dddddgd }t
|
||¡ t| }t j|jd d
df< t
|
||¡ d S )
|
Nr` rU r£ rM rà rû r´ rN ru ) rh r¦ r0 r r r# Zhstackr rú r$ r% )Z rand_datar r r) r* r* r+ Ú test_loc_setitem_float_intindex²
|
s $ r° c C sv dd dD } t t d¡tdt | ¡d}|jddd d
|
}t t d ¡tdt d d d d
|
g¡d}t ||¡ d S )Nc S s g | ]}d D ] }||fq qS ))rT r` r: rQ r* )r£ Úyrr r* r* r+ r¥ Â
|
s z)test_loc_axis_1_slice.<locals>.<listcomp>)éÞ r )rQ r` Z
|
ABCDEFGHIJrÒ rM rK )r² r: )r r` r )r² rQ )r rT ) r rh rR râ r rÉ r# r$ r% )rÇ r( r r) r* r* r+ Útest_loc_axis_1_sliceÀ
|
s ýýr³ c C sX t dtdt tdtdg¡d} | ¡ }|jddgf |jddgf< t || ¡ d S )Nr r0 rÒ r rN )r r§ r r r· r# r$ r% )r) r r* r* r+ Ú!test_loc_set_dataframe_multiindexÑ
|
s ÿ r´ c C s4 t tdtddgtd} | jd }|dks0td S )Nr0 rM r´ r¯ r )r r§ r rÚ r# r{ )r r r* r* r+ Útest_loc_mixed_int_floatÛ
|
s
|
rµ c C sD t tdddddgd} tjtdd d | jd d
|
< W 5 Q R X d S ) NrU r" rl r rì rx ú$Slicing a positional slice with .locr r0 rR )r r§ r r r r# )r r* r* r+ Ú%test_loc_with_positional_slice_raisesã
|
s r· c
|
C sæ t ddd} ttj d¡| d}|d }d}||fD ]T}tjt|d |jd d
W 5 Q R X tjtd
|
d d |jd d
< W 5 Q R X q4tjt|d |jd d
d f W 5 Q R X tjtd
|
d d |jd d
d f< W 5 Q R X d S ) Nz
|
2016-01-01rR ©r )rR r0 rx r zOcannot do slice indexing on DatetimeIndex with these indexers \[1\] of type intr rM r¶ r0 )r r rh r¦ r r r r# )Údtir( r r r9 r* r* r+ Ú#test_loc_slice_disallows_positionalë
|
s ÿ rº c C s¤ t tj dd¡dddgtddddd } | jd
|
d d dg ¡ } | j}t |j
|
¡}d }t j t |d | j| W 5 Q R X t j t |d | d j| W 5 Q R X d S )NrV rR r r r< Z2012r² r re r r0 z/None of \[TimedeltaIndex.* are in the \[index\]r )r rh r¦ r0 r r r· rg rp ZTimedeltaIndexZasi8r r r5 r# )r( r¹ r¨ r r* r* r+ Ú'test_loc_datetimelike_mismatched_dtypes s ý r» c C s¢ t jdddd} ttj dd¡| d}t ||j| ¡ t ||jt | ¡ t ||jt | ¡ t |j
|
dd
|j| dd
¡ t ||jt | ¡ d S )
|
Nz2002-01z2003-12r r} r¶ rQ rx r rV ) rp rm r rh r¦ r0 r$ r% r# r¶ r )r r( r* r* r+ Ú"test_loc_with_period_index_indexer s $r¼ c C s~ t j dd¡} tddd}ddddd d
|
g}t| ||d }t j|j|jd d f< t j| d dd
< t| ||d }t ||¡ d S )Nr` rS r| r¸ r" rl r rì ÚEÚFrÒ rM rn r r0 )
|
rh r¦ r0 r r r r# rg r$ r% )rI r rÇ r| r} r* r* r+ Ú%test_loc_setitem_multiindex_timestamp( s r¿ c C s: dddg} ddg}ddg}t j| ||gdd d
|
gd }td td |d}|jd d
| d |d |d ff }|jd d
d d
f }t ||¡ |j| d |d |d fdddgdd}t ||¡ |jd }|jd d
} |j| d |d |d fdddgdd}
|
t |
|
| ¡ |j| d |d |d f } | d ks6t
|
d S )Nr r r< rN )rM r r rM r¤ rÿ rc r rS rV rÒ r0 )r rL ) r r r r§ r# r r$ r% Úxsr¨ r{ ) r( r) Zlev3rÇ r( r r) Zaltr Z expected2Zalt2r r* r* r+ Ú'test_loc_getitem_multiindex_tuple_level4 s
|
$ (
|
( rÁ c C s¬ t t ddddgddddgddtjtjgg¡jd d
|
d gd d } | d ¡}|jjd ksXt|j d }t
|
ddg|j d dd}t ||¡ |j t¡|_|j d }t ||¡ d S )NrM r0 rR rU rV rS rT r` r r r< rU )rm rf r° r3 )r rh r r ÚTr rg r° r{ r# r rf r$ r¨ rÆ rÚ )r( rs r} r) r* r* r+ Ú/test_loc_getitem_nullable_index_with_duplicatesR s ,ý
|
|
|
rà rÝ r¼ c C sJ t ddddgdgdd}| |jd< t ddd dgdgd
|
d}t ||¡ d S ) NrM r0 rR rU rc Zuint8ro )r0 rc r¼ Úuint16r= )rÝ r( r) r* r* r+ Ú test_loc_setitem_uint8_upcastf s
|
rÅ zfill_val,exp_dtypez
|
2022-01-06zdatetime64[ns]z
|
2022-01-07ú
|
US/EasternrB zdatetime64[ns, US/Eastern]c C sn ddd| ¡ g}t|| j|d}tddddgd g|d
|
}d |jd < | d ¡ t||d}tj|j|dd d S )Nz
|
2022-01-02z
|
2022-01-03z
|
2022-01-04)r³ r° rQ r; r< rµ r re rx )ú
|
2022-01-08r rÇ r¯ T)Úexact) r r r³ r r# rD r$ rû rg )Zfill_valZ exp_dtyperm rg r( Zexpected_indexr* r* r+ Ú0test_loc_setitem_using_datetimelike_str_as_indexq s
|
|
rÉ c C sJ t tdg} d| jd d
df< t dgdgdgdgd}t | |¡ d S )NÚabcrV rc r r r< )r rM r0 rc )r r¶ r# r$ r% )r( r) r* r* r+ Útest_loc_set_int_dtype s rË c C s t ddddgddd} tddd gdd
|
d gd d dgdddgg| dddgd}|jddgddd }|j| d ddf d ks|td S )Nz 20181101 1100z 20181101 1200z 20181102 1300z 20181102 1400r rl )rw r~ r" rû r r´ r8 ÚQrN ÚWr¾ r ZONEZTWOZVALUESrÒ T)rD r )r r r r# r{ )Zp_indexZ mi_seriesr* r* r+ Ú test_loc_periodindex_3_levels s
|
ý"ýrÎ c @ s e Zd Zej dddg¡dd Zdd Zdd Zd
|
d Z d d Z
|
dd Z dd Z dd Z dd Zdd Zej dejejeeg¡ej dddd d g¡d d Zej d ed!d"d"
f d!ed"fg¡d#d$ Zd%d& Zej d'd(d)g¡d*d+ Zej d,d- Z d.d/ Z d0d1 Z d"S )2Ú TestLocSeriesr )r rR )r¡ rU c C s$ t ddd}|j| |ks td S )NrR rU )r r¡ )r r# r{ )r7 ró r) r r* r* r+ rh s z TestLocSeries.test_loc_uint64c C sÔ |j dddg }t |j| | |¡¡ t |jdd d
|dd d
¡ |j ddg \}}|j||
}| ||¡}t ||¡ || ¡ k}t |j| || ¡ |j| || ksºt|j| || ksÐtd S )NrR rU rT rV r0 é ) rg r$ r¨ r# r+ r ÚtruncateÚmedianr{ ) r7 Ú string_seriesÚdatetime_seriesÚindsÚd1Úd2r r) rê r* r* r+ Útest_loc_getitem¤ s " z TestLocSeries.test_loc_getitemc C s |j ddg \}}|d d d
dddg }d}tjt|d |j||
W 5 Q R X tjt|d d|j||
< W 5 Q R X d S )NrV rÐ r0 rM r z"Timestamp\('2000-01-10 00:00:00'\)r )rg r r r5 r# )r7 rÔ rÖ r× Zts2r r* r* r+ Ú test_loc_getitem_not_monotonic· s z,TestLocSeries.test_loc_getitem_not_monotonicc C s> t tj d¡ttdddd}| ¡ }d|jdd
< |jdd
dk ¡ sPt | ¡ }d|jdd
< |jdd
dkj
|
¡ st |jdd
}|j dd
}| dd d
|
dg¡}t ||¡ t ||¡ |jttd ttd dd }tjtdd |j dd
W 5 Q R X tjtdd d|j dd
< W 5 Q R X d S )NrQ r rC r0 rx rU rR r; rS r` rV r: r× z^3$r )r rh r¦ r0 r¶ r§ r· r rî r{ rr r# r+ r$ r¨ r r r5 )r7 r Úcpr r r) Ús2r* r* r+ Ú0test_loc_getitem_setitem_integer_slice_keyerrors s" "z>TestLocSeries.test_loc_getitem_setitem_integer_slice_keyerrorsc C s4 t |jd d
}|j| }t ||d d
¡ d S )NrQ )rë rg r# r$ r¨ )r7 rÓ r r r* r* r+ Útest_loc_getitem_iteratorÝ s
|
z'TestLocSeries.test_loc_getitem_iteratorc C s: || ¡ k}| ¡ }d|j|< |}d||< t ||¡ d S )Nr )rÒ r· r# r$ r¨ )r7 rÓ rê r r) r* r* r+ Útest_loc_setitem_booleanâ s
|
z&TestLocSeries.test_loc_setitem_booleanc C sP t |jdddg }d|j|< d}tjt|d d|j|dg < W 5 Q R X d S )NrV r` r< z\['foo'\] not in indexr r )r¶ rg r# r r r5 )r7 rÓ rÕ r r* r* r+ Útest_loc_setitem_cornerë s
|
|
z%TestLocSeries.test_loc_setitem_cornerc C s |j dddg }| ¡ }| ¡ }d||< d|j|< t ||¡ | ¡ }| ¡ }d||d |d
< d|j|d |d
< t ||¡ d S )NrV rQ rÐ r r0 )rg r· r# r$ r¨ )r7 rÔ ÚindicesrÚ r| r* r* r+ Ú test_basic_setitem_with_labelsò s
|
z,TestLocSeries.test_basic_setitem_with_labelsc C s t tj d¡ttdddd}dddg}t dddg¡}| ¡ }| ¡ }d||< d|j|< t
|
||¡ | ¡ }| ¡ }d||< d|j|< t
|
||¡ ddddg}t ddddg¡}d }t j t |d
|
d||< W 5 Q R X t j t|d
|
d||< W 5 Q R X d S ) NrQ r rC r0 rx rU rS rV z\[5\] not in indexr )r rh r¦ r0 r¶ r§ r r· r# r$ r¨ r r r5 Ú Exception) r7 r rÕ Zarr_indsrÚ r| Z inds_notfoundZarr_inds_notfoundr r* r* r+ Ú!test_loc_setitem_listlike_of_ints s(
|
|
|
z/TestLocSeries.test_loc_setitem_listlike_of_intsc C s t tdddddddgd}| ¡ }td dd
|
}||jd< |jd }||ksPt| ¡ }||jd < |jd }||ksxt| ¡ }||d< |d }||kstd S ) Nr© rR rÆ r r r r< rx r¬ rB r )r r r· r r# r{ r )r7 r rÛ r) r r* r* r+ Ú test_loc_setitem_dt64tz_values s" þ
|
|
|
|
z,TestLocSeries.test_loc_setitem_dt64tz_valuesÚarray_fnrã r rU rV rS c C s~ |dg| }t |ddddgtdtd}t dtdtd}||jd< t ||¡ t dtdtd}||jd< t ||¡ d S )Nr Zabcderö r )r r¶ rÚ r# r$ r¨ r )r7 rã rå rR r) r r* r* r+ Ú#test_loc_iloc_setitem_with_listlike1 s
|
|
z1TestLocSeries.test_loc_iloc_setitem_with_listliker¹ r" Nc C s~ t t dddg¡dddgd}d}tjt|d |j|d d
f W 5 Q R X tjt|d d
|
|j|d d
f< W 5 Q R X d S ) N)r" Ú0)r" Ú1)rl rç é r r# r¦ rõ r rM )r r rÉ r r r r# )r7 r¹ r r r* r* r+ Ú+test_loc_series_getitem_too_many_dimensionsC s þ z9TestLocSeries.test_loc_series_getitem_too_many_dimensionsc C sÚ |j dddg }| ¡ }d|j|< | ¡ }d|dddg< t ||¡ d|jdd
< d|dd
< t ||¡ |j ddg \}}d|j||
< d|dd
< t ||¡ d|j|< d|j|< || dksÆt|| dksÖtd S ) NrR rU rT rV rQ rÐ rS r: )rg r· r# r$ r¨ r r{ )r7 rÓ rÕ r r) rÖ r× r* r* r+ Útest_loc_setitemQ s"
|
|
|
z TestLocSeries.test_loc_setitemr° rÚ rý c C s\ t ddgddgd|d}ddd|jd d d
f< t ddgddgd|d}t ||¡ d S )
|
NrÊ ÚdefZghiZjklrn r¯ ZnewAZnewBr r= )r7 r° r( r) r* r* r+ Útest_loc_assign_dict_to_rowk s z)TestLocSeries.test_loc_assign_dict_to_rowc C sp t ddgd}tddddd|jd < tddddd|jd < t tddddggddgd gd
|
}t ||¡ d S ) Nr rÝ ru rS rË r¦ r )r rÝ rM re )r r r# r$ r% r r* r* r+ Ú,test_loc_setitem_dict_timedelta_multiple_setu s ÿz:TestLocSeries.test_loc_setitem_dict_timedelta_multiple_setc C sl t ddgdgd}ddg|jdddgf< t ttjtjgd d
|
dtjgdtjgd ddgd }t ||¡ d S ) NrM r0 r rÒ rS rT r r< rÚ r¯ )r r r< rx )r r# r rh r r$ r% rj r* r* r+ Ú3test_loc_set_multiple_items_in_multiple_new_columns s ýú zATestLocSeries.test_loc_set_multiple_items_in_multiple_new_columnsc C s6 t jdddd}ttd|d}|jd dks2td S )NÚ2000rC rl r rx z
|
2000-01-14r: )rp rm r r§ r# r{ )r7 rg r' r* r* r+ Ú test_getitem_loc_str_periodindex s z.TestLocSeries.test_getitem_loc_str_periodindex) r r r r r r rh rØ rÙ rÜ rÝ rÞ rß rá rã rä rh r rp r¶ râ ræ r r
rê rë rí rê r¡ rî rï rñ r* r* r* r+ rÏ s.
|
$
|
|
|
|
rÏ )SÚ__doc__Ú collectionsr r r r r rß Z dateutil.tzr Únumpyrh r Z pandas.errorsr Z pandas.util._test_decoratorsÚutilZ_test_decoratorsrê Zpandasrp r r
|
r r r r r r r r r r r r r r r r Zpandas._testingZ_testingr$ Zpandas.api.typesr r Zpandas.core.indexingr Z pandas.tests.indexing.commonr r r r r, r- r rì r÷ r. rX rh r{ r r r¢ r¯ r° r³ r´ rµ r· rº r» r¼ r¿ rÁ rà rÄ Úint16rÅ rÉ rË rÎ rÏ r* r* r* r+ Ú<module> s P "þþ
|
9 s< w CJ ~fýþ
|
|
|
|
|
þþ
|
|
