fume-manage-python.git

U  
ý°dßÇã@s dZddlmZddlZddlmZmZddlZddlZddl    m
Z
ddlmm ZddlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZddlm Z!ddl"m#Z#ddl$m%Z%e &d    ¡Z'Gd
ddZ(Gdd d Z)GdddZ*GdddZ+GdddZ,GdddZ-dS)z* test positional based indexing with iloc é)ÚdatetimeN)Úcatch_warningsÚsimplefilter)Ú IndexingError)ÚNAÚCategoricalÚCategoricalDtypeÚ    DataFrameÚIndexÚIntervalÚNaTÚSeriesÚ    TimestampÚarrayÚconcatÚ
date_rangeÚinterval_rangeÚisnaÚto_datetime)Ú    is_scalar)Ú"check_indexing_smoketest_or_raiseszuonly integers, slices (`:`), ellipsis (`...`), numpy.newaxis (`None`) and integer or boolean arrays are valid indicesc
@sXeZdZej ddddddgg¡ej dddg¡ej d    d
ddd dg¡ddZdS)ÚTestiLocÚkeyééÿÿÿÿréÚkindÚseriesÚframeÚcolÚlabelsÚmixedÚtsZfloatsÚemptycCs(| |d|¡}t|d|tddS)NÚ_Úiloc)Zfails)ZgetfixturevaluerÚ
IndexError)ÚselfrrrÚrequestÚobj©r*úVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexing/test_iloc.pyÚ"test_iloc_getitem_int_and_list_int-süz+TestiLoc.test_iloc_getitem_int_and_list_intN)Ú__name__Ú
__module__Ú__qualname__ÚpytestÚmarkÚparametrizer,r*r*r*r+r,sþrc @seZdZdZej dedededdddge    ede
 dddg¡g¡ej dej ejg¡d    d
Zej deeg¡dd ZddZddZej de
 d¡edfg¡ej dedddggddgedfedddgedfg¡ddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zej d)d*d+gd*d+d*d+gg¡d,d-Zd.d/Z d0d1Z!d2d3Z"d4d5Z#d6d7Z$d8d9Z%d:d;Z&d<d=Z'd>d?Z(d@dAZ)dBdCZ*dDdEZ+dFdGZ,ej ddgeddde
 dg¡g¡ej dHdIge
 dIg¡g¡dJdKZ-dLdMZ.dNdOZ/dPdQZ0dRdSZ1dTdUZ2dVdWZ3dXdYZ4dZd[Z5d\d]Z6d^d_Z7ej dHde8e
j9g¡d`daZ:ej dbe;dddcddg¡dedfZ<dgdhZ=ej dej ejg¡didjZ>dkdlZ?dmdnZ@ej doee
jg¡dpdqZAej ddgeddg¡drdsZBdtduZCdvdwZDdxdyZEdzd{ZFd|d}ZGd~dZHej dee
jg¡ej dee
jg¡ddZIddZJddZKeLjMddZNddZOdS)ÚTestiLocBaseIndependentzTests Independent Of Base ClassrNérrrÚindexerc    Cstdtditd}tdddg}|s<|jjd |¡s<t| ¡}|j    }||||df<td|i 
t¡}|st |dj    |¡stt  ||¡d|jd<|ddks¬tttjddd    gtdtdd
}| ¡}|j    }||||df<t| 
t¡tdd
}t  ||¡dS)Nrr4©ÚdtypeÚalphaÚbetaÚgamma©rrrr©rr)r    ÚrangeÚobjectrÚ_mgrÚblocksZ_can_hold_elementÚAssertionErrorÚcopyÚvaluesÚastypeÚnpÚ shares_memoryÚtmÚassert_frame_equalr%r)    r'r5rÚusing_array_managerrÚcatÚdfZ    orig_valsÚexpectedr*r*r+Ú%test_iloc_setitem_fullcol_categoricalCs& 
"z=TestiLocBaseIndependent.test_iloc_setitem_fullcol_categoricalÚboxcCsútddddg}|| d¡}|tkr.|j}n|jjd}|tkr^||dd|jdd<n||dd|jdddf<|tjddddgdd}t     ||¡|tkrà|rÐ|j|k    s¼t
t |j|¡sÞt
qö|j|ksöt
nt |dj|¡söt
dS)Nrrr4éÚi8rr6)rZto_numpyr rCr?Zarraysr%rErGZassert_equalrArF)r'Úframe_or_seriesrNÚusing_copy_on_writeÚarrr)rCrLr*r*r+Útest_iloc_setitem_ea_inplaceps z4TestiLocBaseIndependent.test_iloc_setitem_ea_inplacecCsJtdddg}ttd|d}|j d¡s.t| ¡}|j d¡sFtdS)Nrrr4©Úindex)r©rr)r
r r=r%Z_is_scalar_accessrAÚto_frame)r'rVÚserrKr*r*r+Útest_is_scalar_accesss
z-TestiLocBaseIndependent.test_is_scalar_accessc    Csêttj d¡tdd}d}tjt|d$|jdddddd    d
dgfW5QRXtjt|d|jddgW5QRXtjt|d|jdd gW5QRXtjt|d|jdgW5QRX|d}tjt|d|jdgW5QRXtjt|d|jdgW5QRXd}tjt|d|jdW5QRXtjt|d|jd W5QRXtjt|d|jdW5QRXtjt|d|jd W5QRX|jddd
df}|jddd
df}t     
||¡|jddddf}|jddddf}t     
||¡|jdddd
df}|jdddd
df}t     
||¡|jddd
ddf}|jddd
ddf}t     
||¡|jdddd
f}|jdddd
f}t     
||¡|jdddd
f}|jddddf}t     
||¡|jdddddf}|jddddf}t     
||¡|jddddf}|jddddf}t     
||¡|jdd}|jdd}t     ||¡|jdd}|jdd}t     ||¡|jddd}|jddd}t     ||¡dd}ttj dd¡tdd}||jdddd    ft|j gd||jdddd    f|jdddgf||jd
d|jd
gd}tjt|d|jd
ddgW5QRXd}tjt|d|jddd
fW5QRXdS)N)ééÚABCDE©Úcolumnsz%positional indexers are out-of-bounds©Úmatchrrrr4rOr\éiâÿÿÿédÚAiÿÿÿz*single positional indexer is out-of-boundsé
éüÿÿÿiöÿÿÿriõÿÿÿéécSst||jt ||¡dS©N)ÚstrÚdtypesrGrH)ÚresultrLr*r*r+Úcheckìsz?TestiLocBaseIndependent.test_iloc_exceeds_bounds.<locals>.checkZAB©rVr_é)r    rEÚrandomZ random_sampleÚlistr0Úraisesr&r%rGrHÚassert_series_equalÚrandnrV)r'rKÚmsgÚsrlrLrmZdflr*r*r+Útest_iloc_exceeds_boundss(&,z0TestiLocBaseIndependent.test_iloc_exceeds_boundsz index,columnsr[r]zindex_vals,column_valsrdÚDÚ1Ú2iãc    CsPttj t|t|¡||d}d}tjt|d|j||fW5QRXdS)Nrnz$.iloc requires numeric indexers, gotr`)    r    rErprtÚlenr0rrr&r%)r'rVr_Z
index_valsZcolumn_valsrKrur*r*r+Útest_iloc_non_integer_raisesýsÿz4TestiLocBaseIndependent.test_iloc_non_integer_raisesc    CsJtt d¡ dd¡}t ||¡}tjtdd|j    dW5QRXdS)NrcrezCannot index by location indexr`Úa)
r    rEÚarangeÚreshaperGÚget_objr0rrÚ    TypeErrorr%)r'rQr)r*r*r+Ú test_iloc_getitem_invalid_scalarsz8TestiLocBaseIndependent.test_iloc_getitem_invalid_scalarcCs|t dddg¡}| ¡}tdddgddd    gd
ddgd dddgd}|j|t ||¡|jdd|ft ||¡dS)Nrrrrcéeéfégéhéiéjékél©rdÚBÚCr4rU)rErrBr    r%rGZassert_numpy_array_equal)r'Zarray_with_neg_numbersZ
array_copyrKr*r*r+Ú-test_iloc_array_not_mutating_negative_indicessþ
zETestiLocBaseIndependent.test_iloc_array_not_mutating_negative_indicescCsÖtdddgdddgd}|d}|jd    }|jd
}t ||¡|jd    g}|jd
g}t ||¡|jd    }|jd
}||kst|jd    g}|jd
g}t ||¡tdgdgd}|jd g}t ||¡dS)Nrr4r\érgé ©rdrrdréýÿÿÿr}rUr)r    r%rGrsrHrAr )r'rKrvrLrlr*r*r+Ú/test_iloc_getitem_neg_int_can_reach_first_index%s"
 
 
 
zGTestiLocBaseIndependent.test_iloc_getitem_neg_int_can_reach_first_indexcCstddddddg}tddddddg}t||gdd}|jd}t|sVt|jdddf}ttjdddgd    d
d    d
gdd}t     ||¡dS)Nrrrr4rO©Zaxisr;rrdr)rVÚname)
r    rr%rrAr rEÚnanrGrs)r'Údf1Údf2rKrlrLr*r*r+Útest_iloc_getitem_dups@s
 z.TestiLocBaseIndependent.test_iloc_getitem_dupscCsÄtdddddddddd    d
dg}tddddg}t |jdg|¡tddddddddg}t |jddg|¡tdddd    d
dgddgd }|jddgddgf}t ||¡dS)Nrrr4rrcéÈé,éèéÐé¸r)rrrU)r    rGrHr%©r'rKrLrlr*r*r+Útest_iloc_getitem_arrayNs
 
 
ýÿ z/TestiLocBaseIndependent.test_iloc_getitem_arraycCs tdddddddddd    d
dg}tddddddddg}|jdddg}t ||¡tdddddd    d
dgd dgd}|jdd}t ||¡dS)Nrrr4rrcrrrrrTFrrUcSs|jddkS)NrrrU©Úxr*r*r+Ú<lambda>qóz@TestiLocBaseIndependent.test_iloc_getitem_bool.<locals>.<lambda>©r    r%rGrHrr*r*r+Útest_iloc_getitem_boolas
 
 
ýÿÿz.TestiLocBaseIndependent.test_iloc_getitem_boolrVTFc    CsNtdddg}dt|dt|}tjt|d|j|W5QRXdS)Nrrr4z Boolean index has wrong length: z instead of r`)r r{r0rrr&r%)r'rVrvrur*r*r+Útest_iloc_getitem_bool_diff_lentsz7TestiLocBaseIndependent.test_iloc_getitem_bool_diff_lencCsÚtdddddddddd    d
dg}tddddddddg}|jdd}t ||¡tdddgdgd}|jddd df}t ||¡tdddddddd
dg}|jddddf}t ||¡dS)Nrrr4rrcrrrrrrrUr)rdrcSsddgS)Nrrr*)rKr*r*r+r£r¤zATestiLocBaseIndependent.test_iloc_getitem_slice.<locals>.<lambda>r¥rr*r*r+Útest_iloc_getitem_slice|s"
 
 
ýÿÿz/TestiLocBaseIndependent.test_iloc_getitem_slicecCs¦ttj dd¡ddddgd}ttjjdddd dd    ¡dd
gd}t||gdd}t |j    ddddf|¡t |j    ddddf|¡t||gdd}t |j    dddd    f|¡t |j    ddd    df|¡t||j    dddgfgdd}t |j    dddd f|¡t||gdd}t |j    dddd    f|¡t |j    ddd    df|¡t |j    dddd    f|¡t |j    ddd    df|¡dS)NrerOrdrr^rr[©Úsizerrrrr4)
r    rErprtÚrandintrrrGrHr%)r'rrrKÚexpr*r*r+Útest_iloc_getitem_slice_dupss$ÿ z4TestiLocBaseIndependent.test_iloc_getitem_slice_dupscCsêttj dd¡t ddd¡t ddd¡d}d|jd    <|jd    }|dksLtd|jddddf<|jddddf}|jddddf}t ||¡t    ddd
dgd}|jddd7<t    dddgdd
dgd}t 
||¡dS) NrOrérér4rnr©rrr\rorU)r    rErprtr~r%rArGrHr rs©r'rKrlrLrvr*r*r+Útest_iloc_setitem¬s ÿ
 
z)TestiLocBaseIndependent.test_iloc_setitemcCs¼tdddgdddgddd    gg}tdddgdddgd
d
d
gg}d
|jddd <t ||¡tdddgdddgddd    gg}tddd
gddd
gddd
gg}d
|jddd <t ||¡dS)NroÚcrerÚdrgr®Úer¯r\rrrrr¥©r'rKrLr*r*r+Útest_iloc_setitem_axis_argumentÀs    z7TestiLocBaseIndependent.test_iloc_setitem_axis_argumentcCs¢tt d¡ d¡dddgdddgd}|jddgdd    gf|jddgdd    gfd
7<tt dddd dddddg    ¡ d¡dddgdddgd}t ||¡dS)Né    )r4r4rdrrrnrrrrcrrr4rrrorr®)r    rEr~rr%rrGrHr¶r*r*r+Útest_iloc_setitem_listÌsÿ ýz.TestiLocBaseIndependent.test_iloc_setitem_listcCsxtddddg}tddddg}| ¡}ddg|jtddg<t ||¡| ¡}ddg|jtddg<t ||¡dS)Nrrrr4réþÿÿÿ)r rBr%rGrsr
)r'Zs_origrLrvr*r*r+Útest_iloc_setitem_pandas_objectÛsz7TestiLocBaseIndependent.test_iloc_setitem_pandas_objectcCsftddddddg}tddddddg}t||gdd}| d¡}t |jdddf¡}||j}|j|df|j|df<t ||¡tddgddgd}d    d    g|_    |d
=t ||¡|jddgddgf|jddgddgf<t ||¡|jddgddgfj
dd|jddgddgf<|jddgddgfj
dd|jddgddgf<t ||¡dS) Nrrrr4rOrrr<rrdT)Zdrop)r    rZfillnarEÚisnanr%rVrGrHr_Zreset_index)r'rrrKrLZindsÚmaskr*r*r+Útest_iloc_setitem_dupsès"
 
 
(00z.TestiLocBaseIndependent.test_iloc_setitem_dupscCsÖtddgddggddgd}|jdddf d¡|jdddf<|sZt|jjdksZt|jdddfd|jdddf<|st|jjdkst| ¡}|jddgddgf|jddgddgf<t     ||¡dS)    Nrrrr4rr^Zf8gà?)
r    r%rDr{r?r@rArBrGrH)r'rIrKrLr*r*r+Ú9test_iloc_setitem_frame_duplicate_columns_multiple_blockss&$(zQTestiLocBaseIndependent.test_iloc_setitem_frame_duplicate_columns_multiple_blockscCsìttj dd¡tdddtdddd}|jd}|jd}t ||¡|jd}|jd    }||ksht    |jdd}|jdd
}t 
||¡|jddddf}|jddddf}t 
||¡|jdd dg}|jdddg}t 
||¡|jdd dgdd gf}|jdddgddgf}t 
||¡|jdd dgdd gf}|jdddgddgf}t 
||¡|jddd dgdd gf}|jddddgddgf}t 
||¡ttd dtd}|j|j }|jddddg}t 
||¡dS)NrerOrr[rr®rn)rr)rOrOér4r\rrorrh)rVr7)r    rErprtr=r%ÚlocrGrsrArHr r>rV)r'rKrlr¬rLrvr*r*r+Útest_iloc_getitem_frame sB
 
ÿ
 
 
 
z/TestiLocBaseIndependent.test_iloc_getitem_framec    Csêttj dd¡tdtdd}|jd}|jd}||ks@t|jdddd    f}|jddd
gf}t     ||¡|jd}|jd}||kstd }t
jt|d|jdW5QRXd}t
jt |d|jdW5QRXdS)NrerOÚ
abcdefghijÚABCDrnr°)Úbrrr4r)rr)Újrxz/index 5 is out of bounds for axis 0 with size 4r`)rer\zLocation based indexing can only have \[integer, integer slice \(START point is INCLUDED, END point is EXCLUDED\), listlike of integers, boolean array\] types)r    rErprtrqr%rÁrArGrHr0rrr&Ú
ValueError)r'rKrlr¬rLrur*r*r+Ú test_iloc_getitem_labelled_frameOs*ÿ
 
 
 
ÿz8TestiLocBaseIndependent.test_iloc_getitem_labelled_framecCs¬tj dd¡}tddd}td}t|||d}| ¡|jddd    d
f}t||j    t|ddd    d
f|dd|d    d
d}t
 ||¡td|_|jddd    d
f}t||j    t|ddd    d
f|ddtdd}t
 ||¡tj dd¡}tt d    d d
}tt d    dd
}t|||d}|sJ|jjd    j|jddd
df}t||j    t|ddd
df|dd|d
dd}t
 ||¡dS)NrorOZ20130101)ZperiodsrÄrnr4r\rrZaaaaZaar¯r®r)rErprtrrqr    Zdescriber%rjrkrGrHr_r=r?r@Zmgr_locs)r'rIrSrVr_rKrlrLr*r*r+Útest_iloc_getitem_doc_issueps6.
*.z3TestiLocBaseIndependent.test_iloc_getitem_doc_issuecCsbttj dd¡tdtdd}d|jd<|jd}|dks@td|jddd    d
f<|jddd    d
f}|jddd    d
f}t ||¡t    tj d¡t
ddd    d}d|jd<|jd}|dksÊtd|jdd<|jdd}|jdd}t ||¡t    d gd}dd    dg|jddd    <dd
dg|jddd    <|}t    ddd    d
ddg}t ||¡dS)NrerOrÃrÄrnrr°rrr4r[rUrror\)r    rErprtrqr%rArGrHr r=rsr±r*r*r+Útest_iloc_setitem_seriess4ÿ
 
 
 
z0TestiLocBaseIndependent.test_iloc_setitem_seriescCsÜttjdddtjddddd}ddgddgg|jd    d
<tddddd
gdd dddgd}t ||¡tdddddgtjddddd}ddgddgg|jd    d
<tdddddgdd dddgd}t ||¡dS)Nr\Úint64r6rerrgr¯rrrOrrror¸r}rÅr³r´rµr¢Úy)r    rEr~r%rGrHr¶r*r*r+Útest_iloc_setitem_list_of_listsºsÿ"ÿ"z7TestiLocBaseIndependent.test_iloc_setitem_list_of_listsÚvalueÚZcCsLtddgddggddgd}||jd|f<|jd    }t|rD|d
ksHtdS)Nrrr4rOrdrr^rr;rÏ)r    r%rrA)r'r5rÎrKrlr*r*r+Ú#test_iloc_setitem_with_scalar_indexËs
z;TestiLocBaseIndependent.test_iloc_setitem_with_scalar_indexc Cstttdtddgd}|jddk}d}tjt|d|j|W5QRXtt||_    d    }tjt
|d|j|W5QRX|jtjd
gt|t d}t ||¡t d¡}d|}d d|D}t||d|}ddddddddd    d    }td
døtdtdD]â}    |jdkj}|    rBt|ttt||    }dD]¬}
z:|
rft||
dd}n|}tt||d ¡}Wn2ttt
fk
r¶} zt| }W5d} ~ XYnX|    |
f}| |¡}||krFtd|d|d|dqFqW5QRXdS)Nr\r]r}rnrrzBiLocation based boolean indexing cannot use an indexable as a maskr`zDiLocation based boolean indexing on an integer type is not availableTr6rOcSsg|]}t|qSr*)Úbin)Ú.0Únumr*r*r+Ú
<listcomp>ësz:TestiLocBaseIndependent.test_iloc_mask.<locals>.<listcomp>)ÚlocsÚnumsZ0b1100Z0b11ztUnalignable boolean Series provided as indexer (index of the boolean Series and of the indexed object do not match).)    )NÚ)Nú.loc)Nú.iloc)rVr×)rVrØ)rVrÙ)rÕr×)rÕrØ)rÕrÙ)ÚrecordÚignore)NrVrÕ)r×rØrÙrrÖú[z] does not match [z ], received [ú])r    rqr=r}r0rrrÇr%r{rVÚNotImplementedErrorrErÚboolrGrHr~rrÚUserWarningrÖrCr ÚreversedÚgetattrrjrÑÚsumrÚgetrA)r'rKr½rurlrÕrÖZrepsrLÚidxÚmethodÚaccessorZanswerrµrÚrr*r*r+Útest_iloc_maskØs^
ð
þ
 
ÿz&TestiLocBaseIndependent.test_iloc_maskc    Cs¶tdgddgdd}t d¡d}|j|}t|d|d|g}|j|}t ||¡tdgd    dgd    d}t|d|d|g}tjt    d
d|j
|W5QRXdS)Nç¹?rrrrbécrr4rznot in indexr`)r    rEr~r%rrGrHr0rrÚKeyErrorrÁ)r'rKrårLZdf3rlrr*r*r+Útest_iloc_non_unique_indexings
 
z5TestiLocBaseIndependent.test_iloc_non_unique_indexingcCst dd¡}tj|jddgf|jddddfdddtj|jgddf|jddddfdddtj|jg|jddddfddddS)Nr\rrT)Zcheck_index_typeZcheck_column_type)rGZmakeCustomDataframerHr%©r'rKr*r*r+Ú"test_iloc_empty_list_indexer_is_ok/s&üüÿz:TestiLocBaseIndependent.test_iloc_empty_list_indexer_is_okcCstddddgi}|jdd}||k    s,tt |d|d¡sDtdddg|jdddf<|r||ddddgk ¡stn|ddk ¡sttddddddg}|jdd}||k    s¾tdd    d
g|dd<|rôt|dddddgksòtn t|dddd    d
gkstdS)Nr}rrr4rOr\rorr®r¸)r    r%rArErFrÁÚallr )r'rRZoriginal_dfZ    sliced_dfZoriginal_seriesZ sliced_seriesr*r*r+Ú&test_identity_slice_returns_new_objectDs z>TestiLocBaseIndependent.test_identity_slice_returns_new_objectcCsDtddgddgg}|jt d¡}tddgdd}t ||¡dS)Nrrr4rOr©r)r    r%rErr rGrs)r'rKrlrvr*r*r+Útest_indexing_zerodim_np_arrayasz6TestiLocBaseIndependent.test_indexing_zerodim_np_arraycCs,tddg}|jt d¡}|dks(tdS)Nrrr)r r%rErrA)r'rvrlr*r*r+Ú%test_series_indexing_zerodim_np_arrayhsz=TestiLocBaseIndependent.test_series_indexing_zerodim_np_arraycCstdddg}t|dddgddd    }t |d|¡s:t|ddd
|jdddf<t |d|¡sjttdddgdddgd}t ||¡dS) Nrdrrrrr4)rrF)rBrr©Ú
categories)rr    rGrFrAr%Zassert_categorical_equal)r'rJrKrLr*r*r+Ú-test_iloc_setitem_categorical_updates_inplacenszETestiLocBaseIndependent.test_iloc_setitem_categorical_updates_inplacecCsBtddgddgddgdtjgg}|j|jdkd9<tddgddgdd    gdtjgg}t ||¡|j|jdkd9<tddgddgdd    gd
tjgg}t ||¡|jddddgd9<tddgdd
gdd    gd
tjgg}t ||¡|jddddgd<tdd gddgddgdtjgg}t ||¡dS)Nrrrr4rOr\ror®rer¯TFg@g(@g@)r    rErr%rVrGrH©r'rlrLr*r*r+Ú test_iloc_with_boolean_operation}s"""""z8TestiLocBaseIndependent.test_iloc_with_boolean_operationcCsXtdtd ¡i}|jd}tdgdddddgd    }t|dgdd
d}t ||¡dS)Nr¢z    a b c d err}rÅr³r´rµrõÚcategory)rVrr7)r    rÚsplitr%r rGrs)r'rKrlZraw_catrLr*r*r+Ú?test_iloc_getitem_singlerow_slice_categoricaldtype_gives_seriess
 
zWTestiLocBaseIndependent.test_iloc_getitem_singlerow_slice_categoricaldtype_gives_seriescCsºtdddg d¡}|jdd}tddg tdddg¡}t ||¡|jddg}tddg tdddg¡}t ||¡|jdddg}tdg tdddg¡}t ||¡dS)Nrrr4rúrTF)r rDr%rrGrs)r'rYrlrLr*r*r+Ú$test_iloc_getitem_categorical_valuessz<TestiLocBaseIndependent.test_iloc_getitem_categorical_valuescCs>tdddgdd}||jd<ttddgdd}t ||¡dS)Nrrrztimedelta64[ns]r6)r r%rrGrs)r'rÎrrLr*r*r+Ú%test_iloc_setitem_td64_values_cast_na®s
z=TestiLocBaseIndependent.test_iloc_setitem_td64_values_cast_naÚnot_nar}gð?cCsXt|gd}t||||g|d}|||g|jdd<t||||g|d}t ||¡dS)Nrõr6r4)rr r%rGrs)r'rÿZ nulls_fixturer7rYr¬r*r*r+Ú$test_setitem_mix_of_nan_and_interval¶s
ÿz<TestiLocBaseIndependent.test_setitem_mix_of_nan_and_intervalc    Csltg}ttj t|t|¡||d}tjjddd}d|j}tj    t
|dd|j|<W5QRXdS)Nrnr\)rrrr©zCannot set values with ndim > r`r)r
r    rErprtr{r«Úndimr0rrrÇr%)r'rår)Znd3rur*r*r+Ú4test_iloc_setitem_empty_frame_raises_with_3d_ndarrayÁs zLTestiLocBaseIndependent.test_iloc_setitem_empty_frame_raises_with_3d_ndarraycCs¼t d¡}t|}t d¡}|jddt|}t ||dddg||dddg¡t ||dg||dg¡t ||d||d¡t ||dd||dd¡dS)NreF)Úwriterrr4)rEZeyer    ZsetflagsrGrHrs)r'r5Zrw_arrayZrw_dfZro_arrayZro_dfr*r*r+Ú"test_iloc_getitem_read_only_valuesÊs
 
( z:TestiLocBaseIndependent.test_iloc_getitem_read_only_valuescCstdtjdddi}t dddg¡}d|j_|j|}|jdddg}t     ||¡|dj|}|djdddg}t 
||¡dS)    NÚdatarcÚfloat64r6rr4roF)r    rEZonesrÚflagsZ    writeabler%rÁrGrHrs)r'rKÚindicesrlrLr*r*r+Útest_iloc_getitem_readonly_keyÚs
z6TestiLocBaseIndependent.test_iloc_getitem_readonly_keycCsVtdgdgd}tdddg|jd<tdtdddggidgdgd}t ||¡dS)Nr}r©r_rVrrr4r;)r    r r%rGrHr¶r*r*r+Ú"test_iloc_assign_series_to_df_cellès"z:TestiLocBaseIndependent.test_iloc_assign_series_to_df_cellÚklasscCsntdddgdddgd}|dd    d    g}|j|dfd
|j|df<tdddgd
ddgd}t ||¡dS)Nr¢rÌÚzrr4rO)ÚflagrÎTFrr¥)r'rrKr5rLr*r*r+Útest_iloc_setitem_bool_indexerïs
z6TestiLocBaseIndependent.test_iloc_setitem_bool_indexercCstdddgdddgd}tdd    d
gddd gdddgd}|jdddgf|jdd|f<tdd    d
gdddgdddgd}t ||¡dS)Nrgr¯rrÀéé)Za2Zb2rrr4rOr\rorr®r¸)r}rÅr³rr¥)r'r5rrrLr*r*r+Ú%test_iloc_setitem_pure_position_basedøs
"""z=TestiLocBaseIndependent.test_iloc_setitem_pure_position_basedcCstddgddgd}ddd}||jd<tddgddgd}t ||¡tddgddgd}||jd<tddgddgd}t ||¡dS)Nrr)r¢rÌr¸rëg@gÀX@r¥)r'rKÚrhsrLr*r*r+Ú"test_iloc_setitem_dictionary_values
 
 
z:TestiLocBaseIndependent.test_iloc_setitem_dictionary_valuecCstttj dd¡dddgtdd}|jdd}t |jd|¡|jdddf}t     |jd|¡dddg|_
|jdd}t |jd|¡|jdddf}t     |jd|¡ttj d    d¡ddddgtdd}|jdd
}t |jd|¡|jdd
df}t     |jd|¡ddddg|_
|jdd
g}t |jd|¡|jdd
gdf}t     |jd|¡dS)Nr4rêçÉ?Úabcrnrr)rr}rOrr)r    rErprtrqr%rGrHrÁrsrV)r'rKÚexpectr*r*r+Ú"test_iloc_getitem_float_duplicatess8ÿ
ÿz:TestiLocBaseIndependent.test_iloc_getitem_float_duplicatescCs¶Gddd}tddgdgd}|d|jd<|d|jd<tddgdgd}|d|jd<t ||¡tddgdgd}|d|jd<tj|jd<tddgdgd}t ||¡dS)Nc@sBeZdZddddZedddZeZedddZd    d
Z    dS)zCTestiLocBaseIndependent.test_iloc_setitem_custom_object.<locals>.TON)ÚreturncSs
||_dSri©rÎ)r'rÎr*r*r+Ú__init__3szLTestiLocBaseIndependent.test_iloc_setitem_custom_object.<locals>.TO.__init__cSsd|jdS)NrÜrÝr©r'r*r*r+Ú__str__6szKTestiLocBaseIndependent.test_iloc_setitem_custom_object.<locals>.TO.__str__cSs|j|jkSrir)r'Úotherr*r*r+Ú__eq__;szJTestiLocBaseIndependent.test_iloc_setitem_custom_object.<locals>.TO.__eq__cSs|Srir*rr*r*r+Úview>szHTestiLocBaseIndependent.test_iloc_setitem_custom_object.<locals>.TO.view)
r-r.r/rrjrÚ__repr__rßrr r*r*r*r+ÚTO2s
r"rrrnrWr)r    r%rGrHrEr)r'r"rKrlr*r*r+Útest_iloc_setitem_custom_object0sz7TestiLocBaseIndependent.test_iloc_setitem_custom_objectcCsttj dd¡tdtdd}|jd}t|ts8tt     
|j|jd¡|jjdddf}t|tsntt     
|j|jd¡dS)Nr4ÚABCZaabr
r) r    rErpZrandrqr%Ú
isinstancer rArGÚassert_almost_equalrCÚT)r'rKrlr*r*r+Ú!test_iloc_getitem_with_duplicatesRs 
z9TestiLocBaseIndependent.test_iloc_getitem_with_duplicatescCsVtdddgdddggdddgd}|jdddgf}|jdgdd    }t ||¡dS)
Nrrr4rOr\ror^rr)r    r%ZtakerGrH©r'rKrlrLr*r*r+Ú"test_iloc_getitem_with_duplicates2]s"z:TestiLocBaseIndependent.test_iloc_getitem_with_duplicates2cCs¾ttddddgi}|jd}ttdddidd}t ||¡|jdddf}tddgtddd}t ||¡| ¡}|jdddfd7<ttddddgi}t ||¡dS)Nrrrròr4)r    rr%r rGrsrBrHr)r*r*r+Útest_iloc_intervalds
z*TestiLocBaseIndependent.test_iloc_intervalÚ indexing_funcÚrhs_funccCstdddg}|ddg|j|dddg<tdddg}t ||¡tddddgi}|dgdgg|j|dddg<tddddgi}t ||¡dS)    Nrrrr\reTFr})r r%rGrsr    rH)r'r-r,rYrLrKr*r*r+Útest_loc_setitem_boolean_listus z5TestiLocBaseIndependent.test_loc_setitem_boolean_listcCstddddgidd}|jdddddf}t ||¡d|d    <|jdddddf}t|d    |dd
}t ||¡dS)Nrdrrr4ÚInt64r6rZfoor)rrdr¥)r'rKÚresrLr*r*r+Ú.test_iloc_getitem_slice_negative_step_ea_blockszFTestiLocBaseIndependent.test_iloc_getitem_slice_negative_step_ea_blockcCsptddddgidd}t dgdgg¡|jt ddg¡t dg¡f<tddddgi|djd}t ||¡dS)    NÚstatusr}rÅr³rúr6rr)r    rErr%r7rGrHr¶r*r*r+Ú*test_iloc_setitem_2d_ndarray_into_ea_blocks.zBTestiLocBaseIndependent.test_iloc_setitem_2d_ndarray_into_ea_blockcCsVtddj}t|}|jd}|d|dks2t|d|d<|d|dksRtdS)Nrg$@rrr)rZ_valuesr    r%rA)r'rSrKrYr*r*r+Ú*test_iloc_getitem_int_single_ea_block_views
zBTestiLocBaseIndependent.test_iloc_getitem_int_single_ea_block_viewcCsltddgddgd}tdtddgi|jdddgf<ttdtd    gddgd}tj||d
ddS)Nz
2022-01-01z
2022-01-02Z2021Z2022rrdrz2021-01-01 00:00:00z2022-01-01 00:00:00F)Zcheck_dtype)r    rr%rrGrHr¶r*r*r+Ú)test_iloc_setitem_multicolumn_to_datetime¦s$þûÿ    zATestiLocBaseIndependent.test_iloc_setitem_multicolumn_to_datetime)Pr-r.r/Ú__doc__r0r1r2Úslicer=r
rEZasarrayrGrÁr%rMrr rTrZrwr~rqrr|rrrrr r¦r§r¨rr²r·r¹r»r¾r¿rÂrÈrÉrÊrÍrÐrérírïrñrórôr÷rùrürýrrrþrrrrr    rrrrrr#r(r*r+r.r1r3ÚtdZ&skip_array_manager_not_yet_implementedr4r5r*r*r*r+r3@s®
úþ!
 
eýþ         
 /!* $F    
 
 
    
 
 
!"
r3c@s$eZdZddZddZddZdS)ÚTestILocErrorsc    Csd|}|tkr| ¡}d}tjt|d|jdW5QRXtjttdd|jd<W5QRXdS)Nz5Cannot index by location index with a non-integer keyr`g@r)r    rXr0rrrr%r&Ú_slice_iloc_msg)r'Zseries_with_simple_indexrQr)rur*r*r+Útest_iloc_float_raisesºsz%TestILocErrors.test_iloc_float_raisesc    Csptjtdd"|jddddddfW5QRXtjtdd"d|jddddddf<W5QRXdS)NzToo many indexersr`ztoo many indices for arrayr)r0rrrr%r&)r'Zfloat_framer*r*r+Ú*test_iloc_getitem_setitem_fancy_exceptionsÍs&z9TestILocErrors.test_iloc_getitem_setitem_fancy_exceptionsc    Csxtddddgi}tddddgi}d}tjt|dd|j|<W5QRXd    }tjt|d|j|W5QRXdS)
Nr}rrr4TFzADataFrame indexer for .iloc is not supported. Consider using .locr`zWDataFrame indexer is not allowed for .iloc
Consider using .loc for automatic alignment.)r    r0rrrr%r&)r'rKr5rur*r*r+Útest_iloc_frame_indexerÕsÿz&TestILocErrors.test_iloc_frame_indexerN)r-r.r/r;r<r=r*r*r*r+r9·sr9c@sDeZdZddZddZddZej ddd    gd
dg¡dd Z    dS)ÚTestILocSetItemDuplicateColumnscCstddddddg}tddddddg}t||gdd}d|jd<|jddks\t|jd    dksnt|jjdtjkstdS)
Nrrrr4rOrrr;)rr)r    rr%rArkrErË)r'rrrKr*r*r+Ú*test_iloc_setitem_scalar_duplicate_columnsæs
zJTestILocSetItemDuplicateColumns.test_iloc_setitem_scalar_duplicate_columnscCsXtdddggdddgd}dg|jdddf<tdddggdddgd}t ||¡dS)    NrrjÚstr2r}rÅr^Zstr3rr¥r¶r*r*r+Ú(test_iloc_setitem_list_duplicate_columnsñszHTestILocSetItemDuplicateColumns.test_iloc_setitem_list_duplicate_columnscCsjttjdtjd dd¡ddddgd}|jdddf tj¡|jdddf<|jjdtjksft    dS)    Nr®r6rrOrdrr^r)
r    rEr~rËrr%rDrrkrArîr*r*r+Ú*test_iloc_setitem_series_duplicate_columnsús
ÿ(zJTestILocSetItemDuplicateColumns.test_iloc_setitem_series_duplicate_columnsrkÚ
init_valueÚexpected_value)rËÚ0r)Úfloatz1.2g333333ó?cCslt|ddggdddgd}|jdddf |¡|jdddf<t|ddggdddgtd}t ||¡dS)Nrjr@r}rÅr^r)r_r7)r    r%rDr>rGrH)r'rkrCrDrKZexpected_dfr*r*r+Ú*test_iloc_setitem_dtypes_duplicate_columnss&
ýzJTestILocSetItemDuplicateColumns.test_iloc_setitem_dtypes_duplicate_columnsN)
r-r.r/r?rArBr0r1r2rGr*r*r*r+r>ås    þr>c@seZdZddZddZdS)ÚTestILocCallablecCstddddgtddtdd}|jd    d
}t ||jddg¡|jdd
ddf}t ||jddgddf¡|jdd
d d
f}t ||jddgdf¡|jdd
dd
f}t ||jddgdgf¡|jddgdd
f}t ||jddgdf¡|jddgdd
f}t ||jddgdgf¡|jdd
df}t ||jddgdf¡|jdd
dgf}t ||jddgdgf¡dS)Nrrr4rOÚaabb©ÚXÚYrÄrUcSsddgS©Nrr4r*r¡r*r*r+r£r¤zCTestILocCallable.test_frame_iloc_getitem_callable.<locals>.<lambda>cSsddgSrMr*r¡r*r*r+r£ r¤cSsddgSrMr*r¡r*r*r+r£#r¤cSsdS©Nrr*r¡r*r*r+r£#r¤rcSsddgSrMr*r¡r*r*r+r£&r¤cSsdgSrNr*r¡r*r*r+r£&r¤cSsdSrNr*r¡r*r*r+r£*r¤cSsdgSrNr*r¡r*r*r+r£-r¤cSsddgSrMr*r¡r*r*r+r£0r¤cSsddgSrMr*r¡r*r*r+r£3r¤)r    rqr%rGrHrs)r'rKr0r*r*r+Ú test_frame_iloc_getitem_callables""z1TestILocCallable.test_frame_iloc_getitem_callablecCsFtddddgtddtdd}| ¡}d    |jd
d<| ¡}d    |jddg<t ||¡| ¡}d|jd dddf<| ¡}d|jddgddf<t ||¡| ¡}d|jddddf<| ¡}d|jddgd    f<t ||¡| ¡}d|jddddf<| ¡}d|jddgd    gf<t ||¡| ¡}d|jddgddf<| ¡}d|jddgd    f<t ||¡| ¡}d|jddgddf<| ¡}d|jddgd    gf<t ||¡| ¡}d|jddd    f<| ¡}d|jddgd    f<t ||¡| ¡}ddg|jddd    gf<| ¡}ddg|jddgd    gf<t ||¡dS)Nrrr4rOrIrJrÄrUrcSsddgSrMr*r¡r*r*r+r£<r¤zCTestILocCallable.test_frame_iloc_setitem_callable.<locals>.<lambda>rcSsddgSrMr*r¡r*r*r+r£Br¤r\cSsddgSrMr*r¡r*r*r+r£Hr¤cSsdSrNr*r¡r*r*r+r£Hr¤écSsddgSrMr*r¡r*r*r+r£Nr¤cSsdgSrNr*r¡r*r*r+r£Nr¤rcSsdSrNr*r¡r*r*r+r£Ur¤éûÿÿÿcSsdgSrNr*r¡r*r*r+r£[r¤recSsddgSrMr*r¡r*r*r+r£ar¤cSsddgSrMr*r¡r*r*r+r£gr¤)r    rqrBr%rGrH)r'rKr0r¬r*r*r+Ú test_frame_iloc_setitem_callable6sR"z1TestILocCallable.test_frame_iloc_setitem_callableN)r-r.r/rOrRr*r*r*r+rHsrHc@s,eZdZddZddZddZddZd    S)
ÚTestILocSeriesc    Csttj d¡ttdddd}| ¡}tt|D](}|j|}||j    |}t
 ||¡q4|jtdd}|j dd}t
 ||¡t
 d¡d|dd<W5QRX|r¼t
 ||¡n|jdddk ¡sÖt|jddddd    g}| |j    ddddd    g¡}t
 ||¡dS)
Nrerr[rrUrr4rOr\)r rErprtrqr=rBr{r%rVrGr&r7rÁrsZassert_produces_warningrðrAZreindex)r'rRrYZser_originalÚirlr¬rLr*r*r+Ú    test_ilocns" 
zTestILocSeries.test_iloccCs.tdddgdddgd}|jddks*tdS)NrrrrU)r r%rA)r'rYr*r*r+Útest_iloc_getitem_nonuniquesz*TestILocSeries.test_iloc_getitem_nonuniquecCs\tdddg}tdddgdddgd}|jdd|jdd<tdddg}t ||¡dS)    Nrrr4rOr\rorrU)r r%rGrs)r'Zser1Zser2rLr*r*r+rs
z4TestILocSeries.test_iloc_setitem_pure_position_basedcCsdtdgtjgd}|jdddf d¡|jdddf<tdgttgddd}t ||¡dS)NÚtest)r}rÅrÅr/r6)    r    rErrÁrDrrrGrHrør*r*r+Ú#test_iloc_nullable_int64_size_1_nans&z2TestILocSeries.test_iloc_nullable_int64_size_1_nanN)r-r.r/rUrVrrXr*r*r*r+rSmsrS).r6rÚreÚwarningsrrÚnumpyrEr0Z pandas.errorsrZpandas.util._test_decoratorsÚutilZ_test_decoratorsr8Zpandasrrrr    r
rrr rrrrrrrZpandas._testingZ_testingrGZpandas.api.typesrZpandas.tests.indexing.commonrÚescaper:rr3r9r>rHrSr*r*r*r+Ú<module>s:Dÿ.2V