U
|
®ý°dCÐ ã @ s d dl mZm Z d dlZd dlZd dlmZ d dlmZ d dl m
|
Z
|
m Z m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z d dl m Z d dl m!Z! G dd dZ"G d d
|
d
|
Z#G d d d Z$G d d dZ%G dd dZ&G dd dZ'G dd dZ(dd Z)dd Z*dd Z+dd Z,G d d d Z-G d d d Z.ej/ 0d!ej1eed"d#eed"d$ej2ed%d&ed&d#gd'd(d"d)d*ej1ed$d%d&d#d+d,d-d.d/g eej2d%ej2d#ej2d,ej2d.ej2g e3ddd$d0d*ej1ed1d1d2d2geej2d1ej2d2ge4d(e3ddd$d3d*ej1ee 5d/¡eej2ej2ej2ej2ej2d#d+d,d-d.g
|
e3dd#d4d*ej1ed"d$d%geej2d$d%gd d5d*ej1ed2geej2ge4d(d d6d*ej1ed2d1geej2d1ge4d(d d7d*g¡G d8d9 d9e.Z6G d:d; d;e.Z7G d<d= d=e.Z8G d>d? d?e.Z9G d@dA dAe.Z:G dBdC dCe.Z;ej/ 0dDed"d$d%geej2d$d%gfedEdFdGgeej2dFdGgfee dHd"d"e dHd"d$e dHd"d%geee dHd"d$e dHd"d%gfedIdJdKgeddJdKgfg¡G dLdM dMe.Z<G dNdO dOe.Z=G dPdQ dQe.Z>ej/ 0dRe dFdGg¡e dSdTg¡ejd$dU d$dU d" gej?d(g¡G dVdW dWe.Z@ej/ 0dRdXe AdX¡g¡G dYdZ dZe.ZBej/ 0dRd[d\d]g¡G d^d_ d_e.ZCG d`da dae.ZDej/ 0dbe Ed"¡ejFfe AdX¡ejAfg¡G dcdd ddeDZGej/ 0dRd"dedfd1g¡ej/ 0dge4g¡G dhdi dieDZHej/ 0dbd"ejIfdeejIfdfejIfd1e4fg¡G djdk dkeDZJej/ 0dbd"e4fdle4fd%e4fdee4fdfe4fd1eKfg¡G dmdn dneDZLej/ 0dbd"ejMfdeej?fdfejIfd1e4fg¡G dodp dpeDZNej/ 0dbd"ej?fdeej?fdfejIfd1e4fg¡G dqdr dreDZOej/ 0dbd"ejPfej1deejPej/jQdsdtdudfejIfd1e4fe Rd$¡ejPfe Sd$¡ejPfe Se TejS¡jU¡ej?fe Vd$¡ejPfe Md$¡ejPfg ¡G dvdw dweDZWej/ 0dbedxdyfd"e4fdze4fg¡G d{d| d|eDZXej/ 0dbedxd}d~dfedxdd~dfedxe4fd"e4fg¡G dd deDZYej/ 0dbeddfd"e4fdze4fg¡G d
d deDZZej/ 0dRdIeddded"d$ddg¡ej/ 0dge4g¡G dd deDZ[dd Z\dd Z]dd Z^dd Z_dd Z`dd Zadd Zbdd Zcdd Zdd d¡ Zed¢d£ Zfej/ 0d¤egd$d+¡ej/ 0d¥d1d2d2d2d2gd1d2gd2gg¡ej/ 0d¦dFej2e hei¡jUe hei¡jjg¡ej/ 0d§d¨d© dªd© d«d© g¡d¬d Zkd®d¯ ZldS )°é )ÚdateÚdatetimeN)Ú IndexingError)Ú is_list_like)ÚNAÚ CategoricalÚ DataFrameÚ DatetimeIndexÚIndexÚIntervalÚ IntervalIndexÚ
|
MultiIndexÚNaTÚPeriodÚSeriesÚ TimedeltaÚ TimestampÚarrayÚconcatÚ
|
date_rangeÚinterval_rangeÚ period_rangeÚtimedelta_range)ÚBDayc @ sX e Zd Zdd Zdd Zdd Zdd Zej d d
|
d d g¡d d Z
|
dd Z dd Z dS )ÚTestSetitemDT64Valuesc C s t tddd}d |d< |d tks(td |dd
< |d tksDttj|d< |d tks^ttj|dd
< |d tks|td S ) Nz1/1/2000é
|
©Úperiodsé é é é é )r r r ÚAssertionErrorÚnpÚnan©ÚselfÚseries© r) ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/series/indexing/test_setitem.pyÚtest_setitem_none_nan( s
|
z+TestSetitemDT64Values.test_setitem_none_nanc C s@ t ddg¡}tddg|d}| ¡ }d|jg < t ||¡ d S )N)Úaé )Úbé r- r/ ©Úindexr )r Ú from_tuplesr ÚcopyÚlocÚtmÚassert_series_equal)r' ÚidxÚresultÚexpectedr) r) r* Ú#test_setitem_multiindex_empty_slice6 s
|
|
z9TestSetitemDT64Values.test_setitem_multiindex_empty_slicec C sN t dddgdddgd}t ¡ |d< |jt ¡ ks6t|d t ¡ ksJtd S )Nr- r/ r ÚDater. Úotherr0 )r r Útodayr; r# ©r' Úserr) r) r* Ú test_setitem_with_string_index> s z4TestSetitemDT64Values.test_setitem_with_string_indexc C sT t dddd}ddddg}t||d }| ¡ }tj|d< tj|jd
|
< t ||¡ d S ) NZ2017r ú
|
US/Eastern©r Útz)r r- )r r/ )r r )r r r0 r )r r r3 r$ r% Úilocr5 r6 )r' Úarrr1 r8 r9 r) r) r* Ú)test_setitem_tuple_with_datetimetz_valuesE s
|
z?TestSetitemDT64Values.test_setitem_tuple_with_datetimetz_valuesrC rA ÚUTCz
|
Asia/Tokyoc C s t tddd|d}|jd| dks*tt td|dtd |dtd
|
|dg}| ¡ }td |d||d < t ||¡ t td |dtd |dgd dgd}|jd| dks²tt td|dtd |dtd|dg}| ¡ }|||d dg< t ||¡ d S )Nú
|
2016-01-01ÚHr ©Úfreqr rC údatetime64[ns, ú]z2016-01-01 00:00©rC ú2011-01-01 00:00z2016-01-01 02:00ú
|
2011-01-01r- ú
|
2012-01-01r/ r0 ú2012-01-01 00:00©r r Údtyper# r r3 r5 r6 )r' rC Ú indexer_sliÚorigÚexpr? Úvalsr) r) r* Útest_setitem_with_tzO s2
|
|
|
ýÿ þ
|
|
|
ýÿz*TestSetitemDT64Values.test_setitem_with_tzc C s d}t tddd|d}|jd| dks.tt td|d td
|
|d td |d g}| ¡ }td |d ||d < t ||¡ t td |d td|d gd dgd}|jd| dks¶tt td|d td|d td|d g}| ¡ }|||d dg< t ||¡ d S )NrA z
|
2016-11-06rI r rJ rL rM z2016-11-06 00:00-04:00rN z2011-01-01 00:00-05:00z2016-11-06 01:00-05:00rP r- rQ r/ r0 z2016-11-06 00:00rO rR rS )r' rU rC rV rW r? rX r) r) r* Útest_setitem_with_tz_dstt s4
|
|
|
ýÿ þ
|
|
|
ýÿz.TestSetitemDT64Values.test_setitem_with_tz_dstc C sp t dtjitd}dg}tjdgdd}||j|< t |d gdgtd}tdd |jD s`tt
|
||¡ d S )
|
NÚX©rT Tr úM8[ns]r ©r1 rT c s s | ]}t |tjV qd S ©N)Ú
|
isinstancer$ Ú
|
datetime64©Ú.0Úxr) r) r* Ú <genexpr>¨ s zYTestSetitemDT64Values.test_object_series_setitem_dt64array_exact_match.<locals>.<genexpr>) r r$ r% Úobjectr rD ÚallÚvaluesr# r5 r6 )r' r? ÚindexerÚvaluer9 r) r) r* Ú0test_object_series_setitem_dt64array_exact_match s
|
zFTestSetitemDT64Values.test_object_series_setitem_dt64array_exact_matchN) Ú__name__Ú
|
__module__Ú __qualname__r+ r: r@ rF ÚpytestÚmarkÚ parametrizerY rZ rk r) r) r) r* r ' s
|
|
$&r c @ sd e Zd Zdd Zej dejej g¡ej dddg¡dd Z
|
ej d d
|
de j fg¡d d Z d S )ÚTestSetitemScalarIndexerc C sF t t dd¡t dd¡d}d}tjt|d d|d< W 5 Q R X d S )Nr r r0 z2index -11 is out of bounds for axis 0 with size 10©ÚmatchÚfooiõÿÿÿ)r r5 Z rands_arrayro ÚraisesÚ
|
IndexError)r' r? Úmsgr) r) r* Ú#test_setitem_negative_out_of_bounds® s z<TestSetitemScalarIndexer.test_setitem_negative_out_of_boundsri Ú ser_indexr r- c C sX t ddgdd}||}t dg|gd|d< t t dg|gddgdd}t ||¡ d S )Nr rf r\ é* r0 ©r r5 r6 )r' ri rz r? Zidxrr9 r) r) r* Ú test_setitem_series_object_dtypeµ s
|
z9TestSetitemScalarIndexer.test_setitem_series_object_dtypezindex, exp_value)r r{ c C s> t ddg}t dg|gd|jd< t |dg}t ||¡ d S )Nr r{ r0 )r r4 r5 r6 )r' r1 Z exp_valuer? r9 r) r) r* Útest_setitem_series¿ s z,TestSetitemScalarIndexer.test_setitem_seriesN)rl rm rn ry ro rp rq r5 r4 Úatr} r$ r% r~ r) r) r) r* rr s rr c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) ÚTestSetitemSlicesc C sd d}t jt|jddd d|dd
< W 5 Q R X t jt|jddd d|d d
< W 5 Q R X d S )
|
NzUcannot do slice indexing on DatetimeIndex with these indexers \[{key}\] of type floatz4\.0)Úkeyrs r g @g $@z4\.5g @)ro rv Ú TypeErrorÚformat)r' Údatetime_seriesrx r) r) r* Ú test_setitem_slice_float_raisesÉ s ÿz1TestSetitemSlices.test_setitem_slice_float_raisesc C sT t tdttdd}d|dd
< |dk ¡ s4td|d d
< |dk ¡ sPtd S )Nr r0 r iôÿÿÿr )r ÚrangeÚlistrg r# r> r) r) r* Útest_setitem_sliceÔ s
|
z$TestSetitemSlices.test_setitem_slicec
|
C sd t tj d¡ddddddddgd }d
|
|d d
< |d d
d
|
k ¡ sHt|dd
d
|
k ¡ r`td S ) Né r/ r r" r é é é r0 r )r r$ ÚrandomÚrandnrg r# Úanyr> r) r) r* Útest_setitem_slice_integersÜ s $ z-TestSetitemSlices.test_setitem_slice_integersc C sJ t t d¡ t¡t d¡d}d|d d d
< |d d d
dk ¡ sFtd S )Né r0 r r/ )r r5 Z makeIntIndexÚastypeÚfloatrg r# r& r) r) r* Útest_setitem_slicestepã s z(TestSetitemSlices.test_setitem_slicestepc C sl t ddgtdf¡}ttjdtjd|}d||d d d
< tdddd dd
|
d d dd g
|
|}t ||¡ d S )Nr r- Zabcder r\ éd r r/ r r r" r! é ) r Ú from_productr r r$ ÚarangeÚint64r5 r6 )r' rU Úmir8 r9 r) r) r* Ú test_setitem_multiindex_sliceê s
|
z/TestSetitemSlices.test_setitem_multiindex_sliceN)rl rm rn r
r r r r r) r) r) r* r È s
|
r c @ s e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z e
|
j de ejeg¡dd Zdd Zdd Zdd Zdd Zdd Zd S ) ÚTestSetitemBooleanMaskc C sV t ddgddgdd}t dgdgdd|ddg< t ddgddgdd}t ||¡ d S )Nr- r/ r r^ r TFr| ©r' r? r9 r) r) r* Útest_setitem_mask_castô s z-TestSetitemBooleanMask.test_setitem_mask_castc C sl t tj d¡t ddd¡d d¡}|dk}| ¡ }|| ¡ t¡}|||< | dd ¡}t
|
||¡ d S )Nr r éÿÿÿÿr0 r c S s | dkrt | S | S ©Nr )Ústr)Útr) r) r* Ú<lambda> ó zLTestSetitemBooleanMask.test_setitem_mask_align_and_promote.<locals>.<lambda>) r r$ r r r Úroundr3 Úmapr¡ r5 r6 )r' ÚtsÚmaskÚleftÚrightr9 r) r) r* Ú#test_setitem_mask_align_and_promoteý s $z:TestSetitemBooleanMask.test_setitem_mask_align_and_promotec C sN t ddddg}|dk}|| t¡}|||< t ddddg}t ||¡ d S )Nr r- r/ Ú1Ú2)r r¦ r¡ r5 r6 ©r' r? r¨ Úser2r9 r) r) r* Ú test_setitem_mask_promote_strs s z5TestSetitemBooleanMask.test_setitem_mask_promote_strsc C sP t ddddg}t ddddg}|| }|||< t ddddg}t ||¡ d S )Nr ru ÚbarFTr| r® r) r) r* Útest_setitem_mask_promote s z0TestSetitemBooleanMask.test_setitem_mask_promotec C s || ¡ k}| ¡ }|d ||< |d }t || || ¡ | ¡ }|d dd
||< |d dd
|¡}|| || < t || || ¡ d S )Nr/ r r )Úmedianr3 r5 r6 Z reindex_like)r' Ú string_seriesr¨ r8 r9 r) r) r* Útest_setitem_boolean s z+TestSetitemBooleanMask.test_setitem_booleanc C sj |}|j dt d| ¡ k}d}tjt|d d||< W 5 Q R X tjt|d d|j|< W 5 Q R X d S )Nr- ©rK zsUnalignable boolean Series provided as indexer \(index of the boolean Series and of the indexed object do not matchrs )Úshiftr r³ ro rv r r4 )r' r r§ Z mask_shiftedrx r) r) r* Útest_setitem_boolean_corner) s ÿz2TestSetitemBooleanMask.test_setitem_boolean_cornerc C s@ | ¡ }| ¡ }d||dk< | ¡ }d||dk< t ||¡ d S r )Z sort_valuesr3 r5 r6 )r' r´ Zorderedr3 r9 r) r) r* Ú$test_setitem_boolean_different_order7 s z;TestSetitemBooleanMask.test_setitem_boolean_different_orderÚfuncc C sF t d dd g}|dddg}ddg||< t dddg}t ||¡ d S )Nr. TFr, Úcr| )r' rº r? r¨ r9 r) r) r* Ú test_setitem_boolean_python_listB s
|
z7TestSetitemBooleanMask.test_setitem_boolean_python_listc C sÒ t ddddg|d}t td|d||dk< t ddddg|d}t ||¡ t ddddg|d}t td|d|j|dk< t ||¡ t ddddg|d}t td|d}|j|d k |j|dk< t ||¡ d S )
|
Nr r" r! r r\ r r/ r r- )r r r5 r6 r4 )r' Úany_numeric_ea_dtyper? r9 Zloc_serr) r) r* Ú'test_setitem_boolean_nullable_int_typesK s z>TestSetitemBooleanMask.test_setitem_boolean_nullable_int_typesc C sv t d gd }dgd dgd dgd }td||< |}t d gd ttd d gd d¡}t ||¡ d S )Nr Fr Tr r/ rf )r r r r r5 r6 )r' r? r¨ r8 r9 r) r) r* ÚAtest_setitem_with_bool_mask_and_values_matching_n_trues_in_length[ s *zXTestSetitemBooleanMask.test_setitem_with_bool_mask_and_values_matching_n_trues_in_lengthc C s< t dddg}tj|d< t tjddgtd}t ||¡ d S )NTFr r\ )r r$ r% rf r5 r6 )r' r8 r9 r) r) r* Útest_setitem_nan_with_boold s
|
z1TestSetitemBooleanMask.test_setitem_nan_with_boolc C s´ t dddgdd}tjdddgtjd}t d d
|
d g¡}| ¡ }t |||< t dddg}t ||¡ | ¡ }|j||d d t ||¡ | ¡ }| | t |¡}t ||¡ d S ) Nr- r/ r Úint8r\ iç éè ié TF©Zinplace) r r$ r r r3 r5 r6 r¨ Úwhere) r' rV Úaltr¨ r? r9 r¯ Úser3Úresr) r) r* Ú!test_setitem_mask_smallint_upcastk s z8TestSetitemBooleanMask.test_setitem_mask_smallint_upcastc C s t dddgdd}t dddgtjd}t d d
|
d g¡}| ¡ }|||< t dddgdd}t ||¡ | ¡ }|j||d d t ||¡ d S ) Nr- r/ r Úuint8r\ éõ r éö TFrà )r r$ r r r3 r5 r6 r¨ )r' rV rÅ r¨ r? r9 r¯ r) r) r* Ú$test_setitem_mask_smallint_no_upcast~ s z;TestSetitemBooleanMask.test_setitem_mask_smallint_no_upcastN)rl rm rn r r« r° r² rµ r¸ r¹ ro rp rq r r$ r r r¼ r¾ r¿ rÀ rÈ rÌ r) r) r) r* r ó s
|
|
r c @ s e Zd Zdd Zdd ZdS )Ú TestSetitemViewCopySemanticsc C sÒ t dddd}|d }t|}|j|k s,t|rJ|jjj|jjjksbtn|jjj|jjjk sbt|jdksptt|j d< |jjd kst|j|k st|jjj|jjjk s°t|d |ksÀt|jdksÎtd S )NZ20130101r rA rB r- ÚD)
|
r r Ú_valuesr# Ú_ndarrayÚbaseÚ_datarK r rD ©r' Zusing_copy_on_writeÚdtir§ r? r) r) r* Ú,test_setitem_invalidates_datetime_index_freq s
|
zITestSetitemViewCopySemantics.test_setitem_invalidates_datetime_index_freqc C sä t dddd}|d }t|}|j|k s,t|rh|jjj|jjjksHt|jjd jj|jjjkstn6|jjj|jjjk st|jjd jj|jjjk st|jjd |k s²tt |d d d
< |d t ksÐt|d |ksàtd S )NrH r ú
|
US/PacificrB r r )
|
r r rÏ r# rÐ rÑ rÒ Z_mgrZarraysr rÓ r) r) r* Ú'test_dt64tz_setitem_does_not_mutate_dti s zDTestSetitemViewCopySemantics.test_dt64tz_setitem_does_not_mutate_dtiN)rl rm rn rÕ r× r) r) r) r* rÍ s rÍ c @ s e Zd Zdd Zdd ZdS )ÚTestSetitemCallablec C sL t ddddgtdd}d|dd < t ddddgtdd}t ||¡ d S )
|
Nr- r/ r r ÚABCDr0 r c S s dS )NÚAr) ©rd r) r) r* r£ Å r¤ z?TestSetitemCallable.test_setitem_callable_key.<locals>.<lambda>)r r r5 r6 r r) r) r* Útest_setitem_callable_key s z-TestSetitemCallable.test_setitem_callable_keyc C sD dd }t ddddg}|||dk < t dd|dg}t ||¡ d S )Nc S s | d S ©Nr- r) rÛ r) r) r* r£ Ì r¤ zATestSetitemCallable.test_setitem_callable_other.<locals>.<lambda>r- r/ r r r r| )r' Úincr? r9 r) r) r* Útest_setitem_callable_otherÊ s
|
z/TestSetitemCallable.test_setitem_callable_otherN)rl rm rn rÜ rß r) r) r) r* rØ Á s rØ c @ sâ e Zd Zdd Zdd Zdd Zej de de d
|
¡ e d ¡ g¡d d
|
Z d d Z d d Zdd Zej dddg¡ej deeddfeejddfeeddfejeddfejeddfejejddfg¡dd Zd d Zd S ) ÚTestSetitemWithExpansionc C s6 t d}ttd}d||< td|g}t ||¡ d S )NrQ r\ é/ )r r rf r5 r6 )r' r r( r9 r) r) r* Útest_setitem_empty_seriesÖ s
|
|
z2TestSetitemWithExpansion.test_setitem_empty_seriesc C s^ t g tg ddtd}td}d||< t dt|gdd}t ||¡ |jj|jjksZtd S )NrÎ r¶ r\ rQ rá ) r r rf r r5 r6 r1 rK r# )r' r( r r9 r) r) r* Ú6test_setitem_empty_series_datetimeindex_preserves_freqÞ s zOTestSetitemWithExpansion.test_setitem_empty_series_datetimeindex_preserves_freqc C sV t d}t|gdgtd}|d }tg td}d|d< ||d< |d }||ksRtd S )Nl PiÕ5Ò Ú timestampr^ r\ g Àr@Zanything)r r rf r# )r' rä r( r9 r8 r) r) r* Ú3test_setitem_empty_series_timestamp_preserves_dtypeç s zLTestSetitemWithExpansion.test_setitem_empty_series_timestamp_preserves_dtypeÚtdú9 daysc C s t d|gddgtd}t dg}||d< t ||¡ t|d tsFtt dg}td|jd< t ||¡ t|d ts|td S )Nrd r ræ r^ rç )r rf r5 r6 r` r r# r4 )r' ræ r9 r? r) r) r* Ú#test_append_timedelta_does_not_castó s
|
|
|
z<TestSetitemWithExpansion.test_append_timedelta_does_not_castc C sR t td}td|d< d|d< d|d< t tdddgdddgd}t ||¡ d S ) Nr\ rH r, ç @r. ru r» r0 )r rf r r5 r6 r r) r) r* Ú*test_setitem_with_expansion_type_promotion s
|
zCTestSetitemWithExpansion.test_setitem_with_expansion_type_promotionc C sL | ¡ }d|jkstd|d< tdgdgdd}t||g}t ||¡ d S )NZfoobarr- r( )r1 Úname)r3 r1 r# r r r5 r6 )r' r´ r? Zappr9 r) r) r* Útest_setitem_not_contained s z3TestSetitemWithExpansion.test_setitem_not_containedc C s: t ddg|d}d|d< t dddg|d}t ||¡ d S )Nr- r/ r\ r r| )r' r½ r? r9 r) r) r* Útest_setitem_keep_precision s z4TestSetitemWithExpansion.test_setitem_keep_precisionri r- r/ z"na, target_na, dtype, target_dtypeZInt64r Úfloat64rf ZFloat64c C sN t ddg|d}|||< |dkr(d|gndd|g}t ||d}t ||¡ d S )Nr- r/ r\ r| ) r' ÚnaZ target_narT Z target_dtyperi r? Zexpected_valuesr9 r) r) r* Ú test_setitem_enlarge_with_na# s
|
z5TestSetitemWithExpansion.test_setitem_enlarge_with_nac C sL t ddg}||d< t dd|gdddgd}t ||¡ |d |ksHtd S )Nr, r. r r r- r0 )r r5 r6 r# )r' Z nulls_fixturer? r9 r) r) r* Ú$test_setitem_enlargement_object_none7 s
|
z=TestSetitemWithExpansion.test_setitem_enlargement_object_noneN)rl rm rn râ rã rå ro rp rq r Zto_timedelta64Zto_pytimedeltarè rê rì rí r r$ r% r rð rñ r) r) r) r* rà Õ s6
|
|
ýþ
|
|
|
|
úþ rà c
|
C sd t d¡} d| j_t| dd}|jD ]:}d}tjt|d d||< W 5 Q R X | | dks$t q$d S )Nr F©r3 ú#assignment destination is read-onlyrs r- r )
|
r$ ÚzerosÚflagsÚ writeabler r1 ro rv Ú
|
ValueErrorr# )r r( Únrx r) r) r* Ú.test_setitem_scalar_into_readonly_backing_data@ s
|
|
rù c C sX t d¡} d| j_t| dd}d}tjt|d d|dd
< W 5 Q R X | ¡ rTt d S )Nr Frò ró rs r- r )
|
r$ rô rõ rö r ro rv r÷ r r# )r r( rx r) r) r* Ú-test_setitem_slice_into_readonly_backing_dataO s
|
rú c C s@ t tddgddgd} | ¡ }d|d d
< t tddgddgd}t ||¡ | ¡ }d|d< t tddgddgd}t ||¡ | ¡ }d||jdk< t tddgddgd}t ||¡ | ¡ }d|ddg< t tddgddgd}t ||¡ | ¡ }dd g|_d|d < t tddgddgddd gd
|
}t ||¡ d S ) Nr. r, ©Ú
|
categoriesr- r FTrd Úyr0 )r r r3 r5 r6 r1 )rV r? rW r) r) r* Ú&test_setitem_categorical_assigning_ops] s,
|
rþ c C sJ t tdddg} t tdtjdgdddgd}tj| d< t | |¡ d S )Nr- r/ r rû )r r r$ r% r5 r6 )r? rW r) r) r* Ú!test_setitem_nan_into_categoricalz s
|
rÿ c @ sF e Zd Zej dddg¡ejjddddgedd d
|
Zd d Zd S )ÚTestSetitemCastingÚuniqueTFÚvalr ré Ú3)Úidsc C s t ddg}|sddg|_|||d< t|jd t|ks@tt d|gt|jd}|sx|tjk rxt ||gtddgd}t ||¡ d S )NTFr- )rT r1 )r r1 ÚtyperD r# rf r5 r6 )r' r rU r r? r9 r) r) r* Ú test_setitem_non_bool_into_bool s
|
z2TestSetitemCasting.test_setitem_non_bool_into_boolc C sp t dddg}|j}tddd g}|d d
|d d
< |j|ksDt|dd
|dd
< t |}t ||¡ d S )NTFr/ r- )r rÏ r r# r5 r6 )r' r? rh rE r9 r) r) r* Ú&test_setitem_boolean_array_into_npbool s z9TestSetitemCasting.test_setitem_boolean_array_into_npboolN) rl rm rn ro rp rq r r r r) r) r) r* r s r c @ s^ e Zd ZdZejdd Zdd Zdd Zdd Z d
|
d Z
|
d d Z dd Z dd Z dd ZdS )ÚSetitemCastingEquivalentszÓ
|
Check each of several methods that _should_ be equivalent to `obj[key] = val`
|
|
We assume that
|
- obj.index is the default Index(range(len(obj)))
|
- the setitem does not expand the obj
|
c C s* z|j |j kW S tk
|
r$ Y dS X dS )zF
|
Whether we expect the setting to be in-place or not.
|
FN)rT r )r' Úobjr9 r) r) r* Ú
|
is_inplaceª s z$SetitemCastingEquivalents.is_inplacec C s> |}| ¡ }|j}||||< t ||¡ | ||||¡ d S r_ )r3 rÏ r5 r6 Ú_check_inplace) r' r r r9 r ri r
|
rV rE r) r) r* Ú check_indexerµ s z'SetitemCastingEquivalents.check_indexerc C sP |d kr
|
nB|r>|j jdkr.|j|jjks<tqL|j|ksLtnt ||j¡ d S )N)ÚmÚM)rT ÚkindrÐ rÏ r# r5 Z assert_equal)r' r
|
rV rE r r) r) r* r ¿ s z(SetitemCastingEquivalents._check_inplacec C s t |tsd S | ||||||¡ |tjkrD| ||||tj|¡ n |tjkrd| ||||tj|¡ t||d }| ||||||¡ |tjk r²t ||d }| ||||||¡ |g} | || ||||¡ t
|
| ¡}
|
| ||
|
||||¡ dd |fD } | || ||||¡ d S )Nr- c s s | ]
|
}|V qd S r_ r) rb r) r) r* re ç s z9SetitemCastingEquivalents.test_int_key.<locals>.<genexpr>) r` Úintr r5 r4 r rD Úiatr Úslicer$ r ) r' r r r9 r rU r
|
ÚrngZslcÚilkeyÚindkeyÚgenkeyr) r) r* Ú test_int_keyÎ s$
|
|
|
|
|
z&SetitemCastingEquivalents.test_int_keyc
|
C s t |tsd S |tjk r,| ||||||¡ ttt|| }| ||||||¡ t |¡}| ||||||¡ dd |D } | || ||||¡ d S )Nc s s | ]
|
}|V qd S r_ r) rb r) r) r* re ø s z;SetitemCastingEquivalents.test_slice_key.<locals>.<genexpr>)
|
r` r r5 r4 r r r Úlenr$ r )
|
r' r r r9 r rU r
|
r r r r) r) r* Útest_slice_keyê s
|
|
|
z(SetitemCastingEquivalents.test_slice_keyc C s t j|jtd}d||< | ¡ }t|rft|| ¡ k rfd}tj t
|
|d ||||< W 5 Q R X d S ||||< t ||¡ d S )Nr\ Tz9boolean index did not match indexed array along dimensionrs ) r$ rô ÚshapeÚboolr3 r r Úsumro rv rw r5 r6 )r' r r r9 r rU r¨ rx r) r) r* Ú test_mask_keyû s z'SetitemCastingEquivalents.test_mask_keyc C s t j|jtd}d||< t|r`t|t|k r`d}tjt|d | | |¡ W 5 Q R X d S |}|
|
¡ }|j } | | |¡}
|
t |
|
|¡ | ||| |¡ d S )Nr\ Tz4operands could not be broadcast together with shapesrs )r$ rô r r r r ro rv r÷ rÄ r3 rÏ r5 r6 r ) r' r r r9 r r
|
r¨ rx rV rE rÇ r) r) r* Útest_series_where s z+SetitemCastingEquivalents.test_series_wherec C sH t j|jtd}d||< t| | |¡}t||jd}t ||¡ d S ©Nr\ T) r$ rô r r r
|
rÄ rT r5 Úassert_index_equal)r' r r r9 r r¨ rÇ Z expected_idxr) r) r* Útest_index_where s
|
z*SetitemCastingEquivalents.test_index_wherec C sB t j|jtd}d||< t| ||¡}t |t||jd¡ d S r ) r$ rô r r r
|
Zputmaskr5 r rT )r' r r r9 r r¨ rÇ r) r) r* Útest_index_putmask' s z,SetitemCastingEquivalents.test_index_putmaskN)rl rm rn Ú__doc__ro Úfixturer
|
r r r r r r r! r" r) r) r) r* r ¡ s
|
|
|
r zobj,expected,keyr- r r/ r r zinterval[float64]r\ Zinterval_int_na_value)Úidr" r! r r r Zint_series_slice_key_stepTFZbool_series_slice_key_stepZint_series_slice_keyZint_series_int_keyZ bool_series_int_key_change_allZbool_series_int_keyc @ s0 e Zd Zejeje d¡degddd Z dS )Ú TestSetitemCastingEquivalentsÚNaNN©Úparamsc C s |j S )z·
|
NA values that should generally be valid_na for *all* dtypes.
|
|
Include both python float NaN and np.float64; only np.float64 has a
|
`dtype` attribute.
|
©Úparam©r' Úrequestr) r) r* r j s z!TestSetitemCastingEquivalents.val)
|
rl rm rn ro r$ r$ r% rî r r r) r) r) r* r& / s ;r& c @ s^ e Zd Zejdd Zejeeegddd Z ejdd Z
|
ejdd Z ejd
|
d Z d S ) Ú!TestSetitemTimedelta64IntoNumericc C s t dd¡}|S )Nr Úns)r$ Ú timedelta64)r' ræ r) r) r* r y s z%TestSetitemTimedelta64IntoNumeric.valr( c C s |j S r_ r* r, r) r) r* rT s z'TestSetitemTimedelta64IntoNumeric.dtypec C s t d¡ |¡}t|}|S )Nr )r$ r r r ©r' rT rE r? r) r) r* r s z%TestSetitemTimedelta64IntoNumeric.objc C s8 t d¡ |¡}t|}| t¡}t dd¡|jd< |S )Nr r r/ r )r$ r r r rf r0 rD r1 r) r) r* r9 s
|
|
z*TestSetitemTimedelta64IntoNumeric.expectedc C s dS r r) ©r' r) r) r* r s z%TestSetitemTimedelta64IntoNumeric.keyN) rl rm rn ro r$ r Úcomplexr r rT r r9 r r) r) r) r* r. u s
|
|
|
|
r. c @ sx e Zd Zejddgddd Zejdd Zejdd Zejd
|
d Zejd d Z ejde
|
e j gddd Z dS )ÚTestSetitemDT64IntoIntr] úm8[ns]r( c C s |j S r_ r* r, r) r) r* rT s z TestSetitemDT64IntoInt.dtypec C s t dd¡}|dkr || }|S )Nz2021-01-18 13:25:00r/ r5 )r$ ra )r' rT r r) r) r* Úscalar s z TestSetitemDT64IntoInt.scalarc C s, t ||dgtd}t|d t|s(t|S )Nr r\ r )r rf r` r r# )r' r6 r9 r) r) r* r9 ¤ s z TestSetitemDT64IntoInt.expectedc C s t dddgS )Nr- r/ r ©r r2 r) r) r* r ª s zTestSetitemDT64IntoInt.objc C s
|
t d dS )Nr ©r r2 r) r) r* r ® s zTestSetitemDT64IntoInt.keyNc C s |j }|d kr|S |||gS r_ r* )r' r6 r- Úboxr) r) r* r ² s zTestSetitemDT64IntoInt.val)rl rm rn ro r$ rT r6 r9 r r r r$ r r r) r) r) r* r4 s
|
|
|
|
|
r4 c @ s` e Zd Zejdd Zejdd Zejdeddgddd Zejd
|
e j
|
gdd d Z d
|
S ) ÚTestSetitemNAPeriodDtypec C sN t tdddd}tj|j d¡|< || tksJtdd || D sJt|S )Nú
|
2000-01-01r rÎ ©r rK Úi8c s s | ]}|t kV qd S r_ )r rb r) r) r* re Á s z4TestSetitemNAPeriodDtype.expected.<locals>.<genexpr>)r r r Ú_valuerÏ Úviewrg r# )r' r rW r) r) r* r9 ½ s &z!TestSetitemNAPeriodDtype.expectedc C s t tddddS )Nr; r rÎ r< )r r r2 r) r) r* r Ä s z TestSetitemNAPeriodDtype.objr r r( c C s |j S r_ r* r, r) r) r* r È s z TestSetitemNAPeriodDtype.keyNc C s |j S r_ r* r, r) r) r* r Ì s z TestSetitemNAPeriodDtype.val) rl rm rn ro r$ r9 r r r r$ r% r r) r) r) r* r: º s
|
|
|
r: c @ s e Zd Zejddddgddd Zejdd Zejd
|
eje e
|
d d ¡e d d ¡gdd d Z ejdd Z ejdd Zejdd Zd
|
S )Ú TestSetitemNADatetimeLikeDtyper5 r] zdatetime64[ns, UTC]zdatetime64[ns, US/Central]r( c C s |j S r_ r* r, r) r) r* rT × s z$TestSetitemNADatetimeLikeDtype.dtypec C s0 t dddj}t||d}|j|ks(tt|S )NrH r r r\ )r Zasi8r
|
rT r# r )r' rT Zi8valsr7 r) r) r* r Ý s z"TestSetitemNADatetimeLikeDtype.objNr r/ c C s |j S r_ r* r, r) r) r* r ä s
|
z"TestSetitemNADatetimeLikeDtype.valc C s& |t kp$|d kp$|tjkp$|j|jkS r_ )r r$ r% rT )r' r r r) r) r* r
|
ð s z)TestSetitemNADatetimeLikeDtype.is_inplacec C s0 |r
|
|j nt}t|gt|dd
|d}|S )Nr- r\ )rT rf r r )r' r r r
|
rT r9 r) r) r* r9 ÷ s z'TestSetitemNADatetimeLikeDtype.expectedc C s dS r r) r2 r) r) r* r ý s z"TestSetitemNADatetimeLikeDtype.key)rl rm rn ro r$ rT r r$ r% r r0 ra r r
|
r9 r r) r) r) r* r@ Ñ s*
|
ÿ
|
|
|
|
ûÿ
|
|
|
r@ c @ sD e Zd Zejdd Zejdd Zejdd Zejdd Zd S )
|
Ú$TestSetitemMismatchedTZCastsToObjectc C s t tddddS )NÚ2000r/ ú
|
US/CentralrB ©r r r2 r) r) r* r s z(TestSetitemMismatchedTZCastsToObject.objc C s t dddS )NrB rA rN )r r2 r) r) r* r s z(TestSetitemMismatchedTZCastsToObject.valc C s dS r r) r2 r) r) r* r s z(TestSetitemMismatchedTZCastsToObject.keyc C s$ t | d¡tdddg|jd}|S )NrC z2000-01-02 00:00:00-06:00rN r\ )r Z
|
tz_convertr rT )r' r r r9 r) r) r* r9 s
|
þûz-TestSetitemMismatchedTZCastsToObject.expectedN© rl rm rn ro r$ r r r r9 r) r) r) r* rA s
|
|
|
rA z obj,expectedç ð?g @ré iÐ ru r± Zbazc @ s( e Zd Zejdd Zejdd ZdS )ÚTestSeriesNoneCoercionc C s dS r r) r2 r) r) r* r . s zTestSeriesNoneCoercion.keyc C s d S r_ r) r2 r) r) r* r 2 s zTestSeriesNoneCoercion.valN)rl rm rn ro r$ r r r) r) r) r* rG s
|
rG c @ sL e Zd Zdd Zejdd Zejdd Zejdd Zejd d
|
Z d S ) Ú-TestSetitemFloatIntervalWithIntIntervalValuesc C s: t td¡}t|}tdd}||d< |jdks6td S )Nr ç à?ç ø?r zInterval[float64, right])r Ú from_breaksr r r rT r# )r' r7 r r r) r) r* Útest_setitem_example: s
|
|
zBTestSetitemFloatIntervalWithIntIntervalValues.test_setitem_examplec C s t td¡}t|S )Nr )r rK r r )r' r7 r) r) r* r C s z1TestSetitemFloatIntervalWithIntIntervalValues.objc C s
|
t ddS )NrI rJ )r r2 r) r) r* r H s z1TestSetitemFloatIntervalWithIntIntervalValues.valc C s dS r r) r2 r) r) r* r L s z1TestSetitemFloatIntervalWithIntIntervalValues.keyc C s* |gt |dd
}t|dd}t|S )Nr- zInterval[float64]r\ )r r r )r' r r Údatar7 r) r) r* r9 P s z6TestSetitemFloatIntervalWithIntIntervalValues.expectedN)
|
rl rm rn rL ro r$ r r r r9 r) r) r) r* rH 7 s
|
|
|
rH c @ sD e Zd Zejdd Zejdd Zejdd Zejdd Zd S )
|
Ú!TestSetitemRangeIntoIntegerSeriesc C s t |¡}ttd|d}|S ©Nr r\ )r$ rT r r )r' Úany_int_numpy_dtyperT r? r) r) r* r [ s
|
z%TestSetitemRangeIntoIntegerSeries.objc C s
|
t ddS )Nr/ r )r r2 r) r) r* r a s z%TestSetitemRangeIntoIntegerSeries.valc C s
|
t ddS ©Nr r/ r8 r2 r) r) r* r e s z%TestSetitemRangeIntoIntegerSeries.keyc C s$ t |¡}tdddddg|d}|S )Nr/ r r r\ )r$ rT r )r' rP rT rW r) r) r* r9 i s
|
z*TestSetitemRangeIntoIntegerSeries.expectedNrE r) r) r) r* rN W s
|
|
|
rN r ç @g @éA c @ s6 e Zd Zejdd Zejdd Zejdd ZdS )Ú(TestSetitemFloatNDarrayIntoIntegerSeriesc C s t tdtjdS rO )r r r$ r r2 r) r) r* r y s z,TestSetitemFloatNDarrayIntoIntegerSeries.objc C s
|
t ddS rQ r8 r2 r) r) r* r } s z,TestSetitemFloatNDarrayIntoIntegerSeries.keyc C s@ |d dkrt j}nt j}t jtd|d}||d d
< t|S )Nr r/ r r\ )r$ r rî r r r )r' r rT Z
|
res_valuesr) r) r* r9 s z1TestSetitemFloatNDarrayIntoIntegerSeries.expectedN©rl rm rn ro r$ r r r9 r) r) r) r* rT p s
|
|
rT é c @ s6 e Zd Zejdd Zejdd Zejdd ZdS )Ú'TestSetitemIntoIntegerSeriesNeedsUpcastc C s t dddgtjdS )Nr- r/ r r\ ©r r$ rÁ r2 r) r) r* r s z+TestSetitemIntoIntegerSeriesNeedsUpcast.objc C s dS rÝ r) r2 r) r) r* r s z+TestSetitemIntoIntegerSeriesNeedsUpcast.keyc C s t dddgtjdS )Nr- rV r r\ )r r$ Úint16r2 r) r) r* r9 s z0TestSetitemIntoIntegerSeriesNeedsUpcast.expectedNrU r) r) r) r* rW s
|
|
rW g BgÍÌ Bl c @ s6 e Zd Zejdd Zejdd Zejdd ZdS )Ú TestSmallIntegerSetitemUpcastc C s t dddgddS )Nr- r/ r Úi4r\ r7 r2 r) r) r* r s z!TestSmallIntegerSetitemUpcast.objc C s dS r r) r2 r) r) r* r £ s z!TestSmallIntegerSetitemUpcast.keyc C s( |d dkrd}nd}t |ddg|dS )Nr- r Zf8r= r/ r r\ r7 )r' r rT r) r) r* r9 § s z&TestSmallIntegerSetitemUpcast.expectedNrU r) r) r) r* rZ s
|
|
rZ c @ s( e Zd Zejdd Zejdd ZdS )Ú CoercionTestc C s dS rÝ r) r2 r) r) r* r ³ s zCoercionTest.keyc C s t |}|||< t||dS )Nr\ )r r )r' r r r Ú exp_dtyperX r) r) r* r9 · s zCoercionTest.expectedN)rl rm rn ro r$ r r9 r) r) r) r* r\ ° s
|
r\ z val,exp_dtypec @ s e Zd Zejdd ZdS )ÚTestCoercionInt8c C s t ddddgtjdS )Nr- r/ r r r\ rX r2 r) r) r* r à s zTestCoercionInt8.objN©rl rm rn ro r$ r r) r) r) r* r^ ¾ s r^ çñ?ù ð? ð?r] c @ s e Zd Zejdd ZdS )ÚTestCoercionObjectc C s t ddddgtdS )Nr, r. r» Údr\ )r rf r2 r) r) r* r Ì s zTestCoercionObject.objNr_ r) r) r) r* rb È s rb c @ s e Zd Zejdd ZdS )ÚTestCoercionComplexc C s t ddddgS )Nra y @ @y @ @y @ @r7 r2 r) r) r* r × s zTestCoercionComplex.objNr_ r) r) r) r* rd Ñ s rd r c @ s e Zd Zejdd ZdS )ÚTestCoercionBoolc C s t ddddgtdS )NTFr\ )r r r2 r) r) r* r é s zTestCoercionBool.objNr_ r) r) r) r* re Ü s re c @ s e Zd Zejdd ZdS )ÚTestCoercionInt64c C s t ddddgS )Nr- r/ r r r7 r2 r) r) r* r ô s zTestCoercionInt64.objNr_ r) r) r) r* rf î s rf c @ s e Zd Zejdd ZdS )ÚTestCoercionFloat64c C s t ddddgS )Nr` ç@çffffff
|
@ç@r7 r2 r) r) r* r ÿ s zTestCoercionFloat64.objNr_ r) r) r) r* rg ù s rg zbnp.float32(1.1) ends up as 1.100000023841858, so np_can_hold_element raises and we cast to float64)Úreason)Zmarksc s* e Zd Zejdd Z fddZ ZS )ÚTestCoercionFloat32c C s t ddddgtjdS )Nr` rh ri rj r\ )r r$ Úfloat32r2 r) r) r* r s zTestCoercionFloat32.objc s. t ||||||¡ t|tkr*tdd S )Nz!xfail not relevant for this test.)Úsuperr r r r# )r' r r r9 r rU r
|
©Ú __class__r) r* r ! s z"TestCoercionFloat32.test_slice_key)rl rm rn ro r$ r r Ú __classcell__r) r) ro r* rl s
|
rl rQ zdatetime64[ns]rd c @ s e Zd Zejdd ZdS )ÚTestCoercionDatetime64c C s t tddddS )NrP rÎ r )rK r rD r2 r) r) r* r 0 s zTestCoercionDatetime64.objNr_ r) r) r) r* rr ) s rr rA rN zdatetime64[ns, US/Eastern]rÖ c @ s e Zd Zejdd ZdS )ÚTestCoercionDatetime64TZc C s d}t tddd|dS )NrA rP rÎ r rJ rD )r' rC r) r) r* r A s z TestCoercionDatetime64TZ.objNr_ r) r) r) r* rs 5 s rs z12 dayztimedelta64[ns]c @ s e Zd Zejdd ZdS )ÚTestCoercionTimedelta64c C s t tdddS )Nz1 dayr r )r r r2 r) r) r* r M s zTestCoercionTimedelta64.objNr_ r) r) r) r* rt G s rt Z2016ÚYr¶ Zboth)Úclosedc @ s4 e Zd Zejeddddeddgddd Zd
|
S ) ÚTestPeriodIntervalCoercionrH r rÎ r< r- r r( c C s
|
t |jS r_ )r r+ r, r) r) r* r X s z TestPeriodIntervalCoercion.objN)rl rm rn ro r$ r r r r) r) r) r* rw R s þÿrw c C sT t dddgdddgd} t dddgdddgd}| ¡ }d|jd< t ||¡ | ¡ }d|jd< t ||¡ | ¡ }d|d< t ||¡ | ¡ }d|jd< t ||¡ | ¡ }d|jd< t ||¡ | d ¡}| d ¡}| ¡ }d|jd
|
< t ||¡ | ¡ }d|jd
|
< t ||¡ | ¡ }d|jd < t ||¡ | ¡ }d|jd < t ||¡ d S ) Nr r- r/ r, r. r» r0 g@rÚ )r. rÚ )r- r )
|
r r3 r r5 r6 r4 r rD Zto_frameÚassert_frame_equal)rV r9 r? Zorig_dfZ expected_dfÚdfr) r) r* Ú
|
test_20643b s>
|
|
|
|
|
|
|
|
|
|
rz c C sr t dddgdddgd} t tjddgdddgd}| ¡ }d |jd< t ||¡ | ¡ }d |jd< t ||¡ d S )Nr r- r/ r, r. r» r0 )r r$ r% r3 r r5 r6 rD )rV r9 r? r) r) r* Útest_20643_comment s
|
|
r{ c C s t dddg} | | dk d7 < t dddg}t | |¡ t dddg} | d d7 < t | |¡ t dddg} | jd d7 < t | |¡ t dddg} | jd d7 < t | |¡ t dddg} | jd d7 < t | |¡ t dddg} | jd d7 < t | |¡ d S )Nr- r/ r rI rR )r r5 r6 r4 rD r r ©r? r9 r) r) r* Ú
|
test_15413 s& r} c C s\ t d¡ d¡} t| }t t t j¡j¡}||d< t|ddddgt jd}t ||¡ d S ) Nr r[ r r- r/ r r r\ ) r$ r r r r ÚiinfoÚmaxr5 r6 ©rE r? r r9 r) r) r* Útest_32878_int_itemsize¼ s r c C sn t d¡ d¡} t| }t t j¡j}| d¡}||d< |d |ksHtt|ddddgdd }t ||¡ d S )
|
Nr Zc8Zc16r r- r/ r r r\ )
|
r$ r r r Úfinforî r r# r5 r6 r r) r) r* Útest_32878_complex_itemsizeÆ s
|
r c C sN t dddgdddgd}d| |d< t dddgdddgtd }t ||¡ d S )
|
Nr- r/ r r, r. r» r0 Útestr^ ©r rf r5 r6 )Z
|
indexer_alr? r9 r) r) r* Ú
|
test_37692Ó s r c C sÊ t jt jfD ]\}tdtd|d}d| |d< |jtks<ttdtdtd}|d|d< |jtks tq tdtdt jd}t d¡| |d< |jt jks ttdtdt jd}t d¡| |d< d S )Nr r r^ Tr- rF ) r$ rî r r r rT rf r# r )rU rT r? r) r) r* Ú'test_setitem_bool_int_float_consistencyÛ s r c C sJ t dddgdddgd} d| d < t dddgdddgtd
|
}t | |¡ d S ) Nr- r/ r r, r. r» r0 r[ r r^ r
r| r) r) r* Ú$test_setitem_positional_with_castingó s r c C sH t dddgdddgd} d| d < t dddgdddgd}t | |¡ d S )
|
Nr- r/ r r, r. r» r0 rJ r r| r| r) r) r* Ú.test_setitem_positional_float_into_int_coercesý s r c C sF t ddddgddddgd } | jjr(td
|
| d< t ddd
|
dg| jd }t | |¡ d | d < t ddd
|
dd gddddd gd }t | |¡ t td
|
¡d d d
}t tt ||d }|
|
t ¡ t dg¡}t d dddddg|d }d|d< t ||¡ t | jddgg¡}t tt ||d }| ¡ }d|jd< d|d< t ||¡ d S )Nr- r/ r r r` gÍÌÌÌÌÌ @ré gffffff@r0 r r g @r r rÚ ÚBéc )r r1 Z _should_fallback_to_positionalr# r5 r6 r rK r r r rf Úappendr
|
r r r3 r4 ) r? r9 Úiir¯ Z exp_indexZ expected2r rÆ Z expected3r) r) r* Ú test_setitem_int_not_positional s(
|
r c C s t dddgdddgd} | d¡}d |d
|
d d g< td ddgdd }t ||¡ d | jd
|
d d gdf< t dd ddgi}t | |¡ d S )Nr- r/ r r r r" )r, r. r. r TF)rM rë r r, )r Úpopr r5 r6 r4 rx )ry r8 r9 r) r) r* Ú test_setitem_with_bool_indexer( s
|
r Úsizer¨ Úitemr9 c C s t | g¡S r_ )r$ r rÛ r) r) r* r£ @ r¤ r£ c C s | gS r_ r) rÛ r) r) r* r£ @ r¤ c C s | fS r_ r) rÛ r) r) r* r£ @ r¤ c C s t || ¡}t j| td}t|}| ¡ dkrZd}tjt|d ||||< W 5 Q R X n2||||< tt j| td}|||< t
|
||¡ d S )Nr\ r- zKcannot set using a list-like indexer with a different length than the valuers ) r$ Úresizer r r r ro rv r÷ r5 r6 ) r r¨ r r9 Z selectionrM r? rx r9 r) r) r* Ú7test_setitem_bool_indexer_dont_broadcast_length1_values6 s ÿ r c C s~ t ddd} t| }| ¡ }tjdtd}d||< |j| jksBtt ||¡ |j
|
|ddd |j| jksntt ||¡ d S )NrH r r r\ ru Trà ) r r r3 r$ rô r rT r# r5 r6 r¨ )rÔ r? rV r¨ r) r) r* Ú(test_setitem_empty_mask_dont_upcast_dt64^ s r )mr r Únumpyr$ ro Z pandas.errorsr Zpandas.core.dtypes.commonr Zpandasr r r r r
|
r r r r r r r r r r r r r r Zpandas._testingZ_testingr5 Zpandas.tseries.offsetsr r rr r r rÍ rØ rà rù rú rþ rÿ r r rp rq r+ r% r rf r r& r. r4 r: r@ rA rG rH rN rî rT rY rW rZ r\ Úint32rÁ r^ rb Z
|
complex128rd r re r rf rg rm ZxfailrÉ Zuint32r~ r Zuint64rl rr rs rt rw rz r{ r} r r r r r r r r r r r Úminr r r) r) r) r* Ú<module> s¢ T + $+k þ÷ "
|
û
|
ü $û ûú
|
ûÑþ: !$1 $ þöþ
|
ýþ ÿ þúþ þ þÿýíþ þ
|
ûþ
|
þ ÿ .
|
|
" ÿ ÿ ÿ
|
