U
    sVcG                     @   s"  d Z ddlmZ ddlZddlmZmZ ddlZddlZddl	m
Z
 ddlm  mZ ddlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZ ddlm Z! ddl"m#Z# ddl$m%Z% e&d	Z'G d
d de%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-dS )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)Basezuonly integers, slices (`:`), ellipsis (`...`), numpy.newaxis (`None`) and integer or boolean arrays are valid indicesc                   @   s.   e Zd Zejddddddggdd ZdS )	TestiLockey   r      c                 C   s    | j d|dddddgtd d S )NiloclabelsmixedtsZfloatsempty)ZtypsZfails)Zcheck_result
IndexError)selfr    r#   C/tmp/pip-unpacked-wheel-xj8nt62q/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   r   c                   @   s  e Zd ZdZejdedededdddge	ede
dddggejdejejgd	d
 Zejdeegdd Zdd Zdd Zejde
dedfgejdedddggddgedfedddgedfgdd Zdd Zdd  Zd!d" Zd#d$ Zd%d& Zd'd( Zejd)d*d+gd*d+d*d+ggd,d- Zd.d/ Z d0d1 Z!d2d3 Z"d4d5 Z#d6d7 Z$d8d9 Z%d:d; Z&d<d= Z'd>d? Z(d@dA Z)dBdC Z*dDdE Z+dFdG Z,ejddgeddde
dggejdHdIge
dIggdJdK Z-dLdM Z.dNdO Z/dPdQ Z0dRdS Z1dTdU Z2dVdW Z3e4j5dXdY Z6dZd[ Z7d\d] Z8d^d_ Z9ejdHde:e
j;gd`da Z<ejdbe=dddcddgdedf Z>dgdh Z?ejdejejgdidj Z@dkdl ZAdmdn ZBejdoee
jgdpdq ZCejddgeddgdrds ZDdtdu ZEdvdw ZFdxdy ZGdzd{ ZHd|d} ZId~d ZJejdee
jgejdee
jgdd ZKdd ZLdd ZMe4j5dd ZNdd ZOdS )TestiLocBaseIndependentzTests Independent Of Base Classr   N   r   r   r   indexerc              	   C   s  t dtditd}tdddg}|s<|jjd |s<t| }|j	}t
|to^|td k}d }	|rlt}	d}
tj|	|
d ||||df< W 5 Q R X |rt d|i}t|j	|rtn,t d|it}|st|d j	|stt|| d|jd	< |d dkstt tjdd
dgtdtdd}| }|j	}tjt|
d ||||df< W 5 Q R X t |tdd}t|| d S )Nr   r-   dtypealphabetagamma.will attempt to set the values inplace insteadmatchr   r   r   r   r   r   )r	   rangeobjectr   _mgrblocksZ_can_hold_elementAssertionErrorcopyvalues
isinstancesliceDeprecationWarningtmassert_produces_warningnpshares_memoryastypeassert_frame_equalr   r   )r"   r.   r   using_array_managerframecatdfZ	orig_vals	overwritewarnmsgexpectedr#   r#   r$   %test_iloc_setitem_fullcol_categorical=   s:    
"z=TestiLocBaseIndependent.test_iloc_setitem_fullcol_categoricalboxc                 C   s   t ddddg}||d}|tkr.|j}n|jjd }|tkr^||dd  |jd d< n||dd  |jd ddf< |tj ddddgdd}t	|| |tkr|j|kst
nt|d j|st
d S )Nr   r   r-      i8r   r/   )r   Zto_numpyr   r?   r;   Zarraysr   rE   rC   Zassert_equalr=   rF   )r"   frame_or_seriesrR   arrobjr?   rP   r#   r#   r$   test_iloc_setitem_ea_inplacet   s    z4TestiLocBaseIndependent.test_iloc_setitem_ea_inplacec                 C   sJ   t dddg}ttd|d}|jds.t| }|jdsFtd S )Nr   r   r-   index)r   r   r   )r
   r   r9   r   Z_is_scalar_accessr=   to_frame)r"   rZ   serrL   r#   r#   r$   test_is_scalar_access   s
    z-TestiLocBaseIndependent.test_is_scalar_accessc              	   C   s  t tjdtdd}d}tjt|d$ |jd d dddd	d
dgf  W 5 Q R X tjt|d |jddg  W 5 Q R X tjt|d |jddg  W 5 Q R X tjt|d |jdg  W 5 Q R X |d }tjt|d |jdg  W 5 Q R X tjt|d |jdg  W 5 Q R X d}tjt|d |jd  W 5 Q R X tjt|d |jd  W 5 Q R X tjt|d |jd  W 5 Q R X tjt|d |jd  W 5 Q R X |jd d d
df }|jd d d
d f }t	
|| |jd d ddf }|jd d d df }t	
|| |jd d dd
df }|jd d d d
df }t	
|| |jd d d
ddf }|jd d d
d df }t	
|| |jd d dd
f }|jd d d d
f }t	
|| |jd d dd
f }|jd d d df }t	
|| |jd d dddf }|jd d d df }t	
|| |jd d ddf }|jd d d df }t	
|| |jdd }|jdd  }t	|| |jdd  }|jd d }t	|| |jdd d }|jd d d }t	|| dd }t tjddtdd}||jd d dd	f t |jd ||jd d dd	f |jd d dgf  ||jd
d |jd
g  d}tjt|d |jd
ddg  W 5 Q R X d}tjt|d |jd d d
f  W 5 Q R X d S )N)      ABCDEcolumnsz%positional indexers are out-of-boundsr5   r   r   r   r-   rS   r`      id   Aiz*single positional indexer is out-of-bounds
   ir   i      c                 S   s   t |  | j t| | d S N)strdtypesrC   rH   )resultrP   r#   r#   r$   check   s    z?TestiLocBaseIndependent.test_iloc_exceeds_bounds.<locals>.checkZABrY      )r	   rE   randomZrandom_samplelistr)   raisesr!   r   rC   rH   assert_series_equalrandnrZ   )r"   rL   rO   srn   rP   ro   Zdflr#   r#   r$   test_iloc_exceeds_bounds   s    ($,z0TestiLocBaseIndependent.test_iloc_exceeds_boundszindex,columnsr_   ra   zindex_vals,column_valsrf   D12i  c              	   C   sP   t tjt|t|||d}d}tjt|d |j||f  W 5 Q R X d S )NrZ   rc   z$.iloc requires numeric indexers, gotr5   )	r	   rE   rq   ru   lenr)   rs   r!   r   )r"   rZ   rc   Z
index_valsZcolumn_valsrL   rO   r#   r#   r$   test_iloc_non_integer_raises   s      z4TestiLocBaseIndependent.test_iloc_non_integer_raisesc              	   C   sJ   t tddd}t||}tjtdd |j	d  W 5 Q R X d S )Nre   rg   zCannot index by location indexr5   a)
r	   rE   arangereshaperC   get_objr)   rs   	TypeErrorr   )r"   rU   rW   r#   r#   r$    test_iloc_getitem_invalid_scalar  s    z8TestiLocBaseIndependent.test_iloc_getitem_invalid_scalarc                 C   s|   t dddg}| }tdddgddd	gd
ddgddddgd}|j|  t|| |jd d |f  t|| d S )Nr   r   r   re   e   f   g   h   i   j   k   l   rf   BCr-   rY   )rE   r   r>   r	   r   rC   Zassert_numpy_array_equal)r"   Zarray_with_neg_numbersZ
array_copyrL   r#   r#   r$   -test_iloc_array_not_mutating_negative_indices  s    
zETestiLocBaseIndependent.test_iloc_array_not_mutating_negative_indicesc                 C   s   t dddgdddgd}|d }|jd	 }|jd
 }t|| |jd	g }|jd
g }t|| |jd	 }|jd
 }||kst|jd	g }|jd
g }t|| tdgdgd}|jdg }t|| d S )Nr   r-   r`      ri      rf   r   rf   r   r~   rY   r   )r	   r   rC   rt   rH   r=   r   )r"   rL   rv   rP   rn   r#   r#   r$   /test_iloc_getitem_neg_int_can_reach_first_index'  s"    



zGTestiLocBaseIndependent.test_iloc_getitem_neg_int_can_reach_first_indexc                 C   s   t d dddddg}t ddddddg}t||gdd}|jd }t|sVt|jdd d f }ttjdddgd	d
d	d
gdd}t	|| d S )Nr   r   r   r-   rS   Zaxisr7   r   rf   r   )rZ   name)
r	   r   r   r   r=   r   rE   nanrC   rt   )r"   df1df2rL   rn   rP   r#   r#   r$   test_iloc_getitem_dupsB  s    
 z.TestiLocBaseIndependent.test_iloc_getitem_dupsc                 C   s   t dddddddddd	d
dg}t ddddg}t|jdg | t ddddddddg}t|jddg | t dddd	d
dgddgd}|jddgddgf }t|| d S )Nr   r   r-   r   re      ,        r   )r   r   rY   )r	   rC   rH   r   r"   rL   rP   rn   r#   r#   r$   test_iloc_getitem_arrayP  s    


 z/TestiLocBaseIndependent.test_iloc_getitem_arrayc                 C   s   t dddddddddd	d
dg}t ddddddddg}|jdddg }t|| t dddddd	d
dgddgd}|jdd  }t|| d S )Nr   r   r-   r   re   r   r   r   r   r   TFr   rY   c                 S   s   | j d dkS )Nr   r   rY   xr#   r#   r$   <lambda>s      z@TestiLocBaseIndependent.test_iloc_getitem_bool.<locals>.<lambda>r	   r   rC   rH   r   r#   r#   r$   test_iloc_getitem_boolc  s    


 z.TestiLocBaseIndependent.test_iloc_getitem_boolrZ   TFc              	   C   sN   t dddg}dt| dt| }tjt|d |j|  W 5 Q R X d S )Nr   r   r-   z Boolean index has wrong length: z instead of r5   )r   r|   r)   rs   r!   r   )r"   rZ   rv   rO   r#   r#   r$   test_iloc_getitem_bool_diff_lenv  s    z7TestiLocBaseIndependent.test_iloc_getitem_bool_diff_lenc                 C   s   t dddddddddd	d
dg}t ddddddddg}|jd d }t|| t dddgdgd}|jddddf }t|| t dddddddd
dg}|jd d dd f }t|| d S )Nr   r   r-   r   re   r   r   r   r   r   r   rY   r   )rf   r   c                 S   s   ddgS )Nr   r   r#   )rL   r#   r#   r$   r     r   zATestiLocBaseIndependent.test_iloc_getitem_slice.<locals>.<lambda>r   r   r#   r#   r$   test_iloc_getitem_slice~  s"    


z/TestiLocBaseIndependent.test_iloc_getitem_slicec                 C   s  t tjddddddgd}t tjjdddddd	dd
gd}t||gdd}t|j	d d d df | t|j	d d dd f | t||gdd}t|j	d d d d	f | t|j	d d d	d f | t||j	d d dgf gdd}t|j	d d ddf | t||gdd}t|j	ddd d	f | t|j	ddd	d f | t|j	dd d d	f | t|j	dd d	d f | d S )Nrg   rS   rf   r   rb   r   r_   sizer   r   r   r   r-   )
r	   rE   rq   ru   randintr   r   rC   rH   r   )r"   r   r   rL   expr#   r#   r$   test_iloc_getitem_slice_dups  s$      z4TestiLocBaseIndependent.test_iloc_getitem_slice_dupsc                 C   s   t tjddtdddtdddd}d|jd	< |jd	 }|dksLtd|jd d ddf< |jd d ddf }|jd d ddf }t|| t	ddd
dgd}|jdd  d7  < t	dddgdd
dgd}t
|| d S )NrS   r      r      r-   r{   r   r   r   r`   rp   rY   )r	   rE   rq   ru   r   r   r=   rC   rH   r   rt   r"   rL   rn   rP   rv   r#   r#   r$   test_iloc_setitem  s       

z)TestiLocBaseIndependent.test_iloc_setitemc                 C   s   t dddgdddgddd	gg}t dddgdddgd
d
d
gg}d
|jddd< t|| t dddgdddgddd	gg}t ddd
gddd
gddd
gg}d
|jddd< t|| d S )Nrp   crg   r   dri   r   er   r`   r   r   r   r   r   r"   rL   rP   r#   r#   r$   test_iloc_setitem_axis_argument  s        z7TestiLocBaseIndependent.test_iloc_setitem_axis_argumentc                 C   s   t tdddddgdddgd}|jddgdd	gf  |jddgdd	gf  d
7  < t tdddddddddg	ddddgdddgd}t|| d S )N	   )r-   r-   rf   r   r   r{   r   r   r   re   r   r   r-   r   r   rp   r   r   )r	   rE   r   r   r   r   rC   rH   r   r#   r#   r$   test_iloc_setitem_list  s       z.TestiLocBaseIndependent.test_iloc_setitem_listc                 C   sx   t ddddg}t ddddg}| }ddg|jt ddg< t|| | }ddg|jtddg< t|| d S )Nr   r   r   r-   r   )r   r>   r   rC   rt   r
   )r"   Zs_origrP   rv   r#   r#   r$   test_iloc_setitem_pandas_object  s    z7TestiLocBaseIndependent.test_iloc_setitem_pandas_objectc                 C   sf  t d dddddg}t ddddddg}t||gdd}|d}t|jd d df }|| j}|j|df |j|df< t|| t ddgddgd}d	d	g|_	|d
= t|| |jddgddgf |jddgddgf< t|| |jddgddgf j
dd|jddgddgf< |jddgddgf j
dd|jddgddgf< t|| d S )Nr   r   r   r-   rS   r   r   r8   r   rf   T)Zdrop)r	   r   ZfillnarE   isnanr   rZ   rC   rH   rc   Zreset_index)r"   r   r   rL   rP   Zindsmaskr#   r#   r$   test_iloc_setitem_dups  s"    


(00z.TestiLocBaseIndependent.test_iloc_setitem_dupsc                 C   s   t ddgddggddgd}|jd d df d|jd d df< |sZt|jjdksZt|jd d df d |jd d df< |st|jjdkst| }|jddgddgf |jddgddgf< t	|| d S )	Nr   r   r   r-   r   rb   Zf8g      ?)
r	   r   rG   r|   r;   r<   r=   r>   rC   rH   r"   rI   rL   rP   r#   r#   r$   9test_iloc_setitem_frame_duplicate_columns_multiple_blocks	  s    &$(zQTestiLocBaseIndependent.test_iloc_setitem_frame_duplicate_columns_multiple_blocksc                 C   s  t tjddtdddtdddd}|jd }|jd }t|| |jd }|jd	 }||ksht	|jdd }|jdd
 }t
|| |jd d ddf }|jd d ddf }t
|| |jdddg }|jdddg }t
|| |jdddgddgf }|jdddgddgf }t
|| |jdddgddgf }|jdddgddgf }t
|| |jddddgddgf }|jddddgddgf }t
|| ttddtd}|j|j }|jddddg }t
|| d S )Nrg   rS   r   r_   r   r   r{   )r   r   )rS   rS      r-   r`   r   rp   r   rj   )rZ   r0   )r	   rE   rq   ru   r9   r   locrC   rt   r=   rH   r   r:   rZ   )r"   rL   rn   r   rP   rv   r#   r#   r$   test_iloc_getitem_frame%  sB     
 




z/TestiLocBaseIndependent.test_iloc_getitem_framec              	   C   s   t tjddtdtdd}|jd }|jd }||ks@t|jd d dd	f }|jd d d
gf }t	|| |jd }|jd }||kstd}t
jt|d |jd  W 5 Q R X d}t
jt|d |jd  W 5 Q R X d S )Nrg   rS   
abcdefghijABCDr{   r   )br   r   r-   r   )r   r   )jrx   z/index 5 is out of bounds for axis 0 with size 4r5   )rg   r`   zLocation based indexing can only have \[integer, integer slice \(START point is INCLUDED, END point is EXCLUDED\), listlike of integers, boolean array\] types)r	   rE   rq   ru   rr   r   r   r=   rC   rH   r)   rs   r!   
ValueError)r"   rL   rn   r   rP   rO   r#   r#   r$    test_iloc_getitem_labelled_frameT  s*      



z8TestiLocBaseIndependent.test_iloc_getitem_labelled_framec                 C   s  t jdd}tddd}td}t|||d}|  |jddd	d
f }t| |j	 t|ddd	d
f |dd |d	d
 d}t
|| td|_|jddd	d
f }t| |j	 t|ddd	d
f |dd tdd}t
|| t jdd}ttd	dd
}ttd	dd
}t|||d}|sJ|jjd	 j |jddd
df }t| |j	 t|ddd
df |dd |d
d d}t
|| d S )Nrp   rS   Z20130101)Zperiodsr   r{   r-   r`   r   r   ZaaaaZaar   r   r   )rE   rq   ru   r   rr   r	   Zdescriber   rl   rm   rC   rH   rc   r9   r;   r<   Zmgr_locs)r"   rI   rV   rZ   rc   rL   rn   rP   r#   r#   r$   test_iloc_getitem_doc_issueu  s6    .
*.z3TestiLocBaseIndependent.test_iloc_getitem_doc_issuec                 C   sb  t tjddtdtdd}d|jd< |jd }|dks@td|jd d d	d
f< |jd d d	d
f }|jd d d	d
f }t|| t	tjdt
ddd	d}d|jd< |jd }|dkstd|jd d< |jd d }|jd d }t|| t	dgd }dd	dg|jdd d	< dd
dg|jdd d	< |}t	ddd	d
ddg}t|| d S )Nrg   rS   r   r   r{   r   r   r   r   r-   r_   rY   r   rp   r`   )r	   rE   rq   ru   rr   r   r=   rC   rH   r   r9   rt   r   r#   r#   r$   test_iloc_setitem_series  s4      



z0TestiLocBaseIndependent.test_iloc_setitem_seriesc                 C   s   t tjdddtjddddd}ddgddgg|jd	d
< t ddddd
gdddddgd}t|| t dddddgtjddddd}ddgddgg|jd	d
< t dddddgdddddgd}t|| d S )Nr`   int64r/   rg   r   ri   r   r   r   rS   r   r   rp   r   r~   r   r   r   r   r   y)r	   rE   r   r   rC   rH   r   r#   r#   r$   test_iloc_setitem_list_of_lists  s    ""z7TestiLocBaseIndependent.test_iloc_setitem_list_of_listsvalueZc                 C   sL   t ddgddggddgd}||jd|f< |jd	 }t|rD|d
ksHtd S )Nr   r   r-   rS   rf   r   rb   r   r7   r   )r	   r   r   r=   )r"   r.   r   rL   rn   r#   r#   r$   #test_iloc_setitem_with_scalar_index  s    
z;TestiLocBaseIndependent.test_iloc_setitem_with_scalar_indexc                 C   s  t ttdtddgd}|jd dk}d}tjt|d |j|  W 5 Q R X tt||_	d	}tjt
|d |j|  W 5 Q R X |jtjd
gt| td }t|| td}d| }dd |D }t ||d|}ddddddddd	d	}td
d tdt dD ]}	|jdkj}|	rBt|ttt||	}dD ]}
z:|
rft||
dd  }n|}tt|| d  }W n2 ttt
fk
r } zt|}W 5 d }~X Y nX |	|
f}||}||krFtd| d| d| dqFqW 5 Q R X d S )Nr`   ra   r~   r{   r   r   zBiLocation based boolean indexing cannot use an indexable as a maskr5   zDiLocation based boolean indexing on an integer type is not availableTr/   rS   c                 S   s   g | ]}t |qS r#   )bin).0numr#   r#   r$   
<listcomp>  s     z:TestiLocBaseIndependent.test_iloc_mask.<locals>.<listcomp>)locsnumsZ0b1100Z0b11ztUnalignable boolean Series provided as indexer (index of the boolean Series and of the indexed object do not match).)	)N )N.loc)N.iloc)rZ   r   )rZ   r   )rZ   r   )r   r   )r   r   )r   r   )recordignore)NrZ   r   )r   r   r   r   r   [z] does not match [z], received [])r	   rr   r9   r~   r)   rs   r   r   r|   rZ   NotImplementedErrorrE   r   boolrC   rH   r   r   r   UserWarningr   r?   r   reversedgetattrrl   r   sumr   getr=   )r"   rL   r   rO   rn   r   r   ZrepsrP   idxmethodaccessorZanswerr   r   rr#   r#   r$   test_iloc_mask  s^    



z&TestiLocBaseIndependent.test_iloc_maskc              	   C   s   t dgd dgd d}tdd }|j| }t|d| d| g}|j| }t|| t dgd	 dgd	 d}t|d| d| g}tjt	d
d |j
|  W 5 Q R X d S )N皙?r   r   r   rd   c   r   r-   r   znot in indexr5   )r	   rE   r   r   r   rC   rH   r)   rs   KeyErrorr   )r"   rL   r   rP   Zdf3rn   r   r#   r#   r$   test_iloc_non_unique_indexing&  s    

z5TestiLocBaseIndependent.test_iloc_non_unique_indexingc                 C   s   t dd}t j|jd d g f |jd d d df ddd t j|jg d d f |jd dd d f ddd t j|jg  |jd dd d f ddd d S )Nr`   r   r   T)Zcheck_index_typeZcheck_column_type)rC   ZmakeCustomDataframerH   r   r"   rL   r#   r#   r$   "test_iloc_empty_list_indexer_is_ok8  s&       z:TestiLocBaseIndependent.test_iloc_empty_list_indexer_is_okc           	      C   s:  |rt jjdd}|j| tddddgi}|jd d  }||k	sJtt	|d |d sbtdddg|j
d d df< |r|d dddgk stn|d dk sttdddddd	g}|jd d  }||k	std
ddg|d d< |rt|d d dddgks6tn t|d d d
ddgks6td S )Nz0setting with .loc[:, 'a'] does not alter inplace)reasonr~   r   r   r-   rS   r`   rp   r   r   r   )r)   r*   ZxfailnodeZ
add_markerr	   r   r=   rE   rF   r   allr   )	r"   rI   using_copy_on_writerequestr*   Zoriginal_dfZ	sliced_dfZoriginal_seriesZsliced_seriesr#   r#   r$   &test_identity_slice_returns_new_objectN  s(    "z>TestiLocBaseIndependent.test_identity_slice_returns_new_objectc                 C   sD   t ddgddgg}|jtd }tddgdd}t|| d S )Nr   r   r-   rS   r   r   )r	   r   rE   r   r   rC   rt   )r"   rL   rn   rv   r#   r#   r$   test_indexing_zerodim_np_arrays  s    z6TestiLocBaseIndependent.test_indexing_zerodim_np_arrayc                 C   s,   t ddg}|jtd }|dks(td S )Nr   r   r   )r   r   rE   r   r=   )r"   rv   rn   r#   r#   r$   %test_series_indexing_zerodim_np_arrayz  s    z=TestiLocBaseIndependent.test_series_indexing_zerodim_np_arrayc              	   C   s   t dddg}| }t|dddgddd	}t|d |sBtd
}tjt|d" |d d d |jd d df< W 5 Q R X |st|d |stt dddgdddgd}n|}t	|| d S )Nrf   r   r   r   r   r-   )r   r   F)r>   r4   r5   r   r   
categories)
r   r>   r	   rC   rF   r=   rD   rB   r   Zassert_categorical_equal)r"   r   rK   Zcat_originalrL   rO   rP   r#   r#   r$   -test_iloc_setitem_categorical_updates_inplace  s    &zETestiLocBaseIndependent.test_iloc_setitem_categorical_updates_inplacec                 C   sB  t ddgddgddgdtjgg}|j|jdk  d9  < t ddgddgdd	gdtjgg}t|| |j|jdk  d9  < t ddgddgdd	gd
tjgg}t|| |jddddg  d9  < t ddgdd
gdd	gd
tjgg}t|| |jddddg  d  < t ddgddgddgdtjgg}t|| d S )Nr   r   r   r-   rS   r`   rp   r   rg   r   TFg      @g      (@g      @)r	   rE   r   r   rZ   rC   rH   r"   rn   rP   r#   r#   r$    test_iloc_with_boolean_operation  s    """""z8TestiLocBaseIndependent.test_iloc_with_boolean_operationc                 C   sX   t dtd i}|jd }tdgdddddgd	}t|dgdd
d}t|| d S )Nr   z	a b c d er   r~   r   r   r   r   r   category)rZ   r   r0   )r	   r   splitr   r   rC   rt   )r"   rL   rn   Zraw_catrP   r#   r#   r$   ?test_iloc_getitem_singlerow_slice_categoricaldtype_gives_series  s
    
zWTestiLocBaseIndependent.test_iloc_getitem_singlerow_slice_categoricaldtype_gives_seriesc                 C   s   t dddgd}|jdd }t ddgtdddg}t|| |jddg }t ddgtdddg}t|| |jdddg }t dgtdddg}t|| d S )Nr   r   r-   r   r   TF)r   rG   r   r   rC   rt   )r"   r]   rn   rP   r#   r#   r$   $test_iloc_getitem_categorical_values  s    z<TestiLocBaseIndependent.test_iloc_getitem_categorical_valuesc                 C   s>   t dddgdd}||jd< t tddgdd}t|| d S )Nr   r   r   ztimedelta64[ns]r/   )r   r   r   rC   rt   )r"   r   ZseriesrP   r#   r#   r$   %test_iloc_setitem_td64_values_cast_na  s    
z=TestiLocBaseIndependent.test_iloc_setitem_td64_values_cast_nanot_nar~   g      ?c                 C   sX   t |gd}t||||g|d}|||g|jd d< t||||g|d}t|| d S )Nr   r/   r-   )r   r   r   rC   rt   )r"   r  Znulls_fixturer0   r]   r   r#   r#   r$   $test_setitem_mix_of_nan_and_interval  s    
 z<TestiLocBaseIndependent.test_setitem_mix_of_nan_and_intervalc              	   C   sl   t g }ttjt|t|||d}tjjddd}d|j }tj	t
|d d|j|< W 5 Q R X d S )Nr{   r`   )r   r   r   r   zCannot set values with ndim > r5   r   )r
   r	   rE   rq   ru   r|   r   ndimr)   rs   r   r   )r"   r   rW   Znd3rO   r#   r#   r$   4test_iloc_setitem_empty_frame_raises_with_3d_ndarray  s     zLTestiLocBaseIndependent.test_iloc_setitem_empty_frame_raises_with_3d_ndarrayc                 C   s   t d}t|}t d}|jdd t|}t||dddg ||dddg  t||dg ||dg  t||d ||d  t||dd ||dd  d S )Nrg   F)writer   r   r-   )rE   Zeyer	   ZsetflagsrC   rH   rt   )r"   r.   Zrw_arrayZrw_dfZro_arrayZro_dfr#   r#   r$   "test_iloc_getitem_read_only_values  s    

( z:TestiLocBaseIndependent.test_iloc_getitem_read_only_valuesc                 C   s   t dtjdddi}tdddg}d|j_|j| }|jdddg }t	|| |d j| }|d jdddg }t
|| d S )	Ndatare   float64r/   r   r-   rp   F)r	   rE   Zonesr   flagsZ	writeabler   r   rC   rH   rt   )r"   rL   indicesrn   rP   r#   r#   r$   test_iloc_getitem_readonly_key  s    
z6TestiLocBaseIndependent.test_iloc_getitem_readonly_keyc                 C   sV   t dgdgd}tdddg|jd< t dtdddggidgdgd}t|| d S )Nr~   r   rc   rZ   r   r   r-   r7   )r	   r   r   rC   rH   r   r#   r#   r$   "test_iloc_assign_series_to_df_cell  s    "z:TestiLocBaseIndependent.test_iloc_assign_series_to_df_cellklassc                 C   sn   t dddgdddgd}|dd	d	g}|j|df d
 |j|df< t dddgd
ddgd}t|| d S )Nr   r   zr   r-   rS   )flagr   TFr   r   )r"   r  rL   r.   rP   r#   r#   r$   test_iloc_setitem_bool_indexer  s
    z6TestiLocBaseIndependent.test_iloc_setitem_bool_indexerc                 C   s   t dddgdddgd}t dd	d
gdddgdddgd}|jd d dgf |jd d |f< t dd	d
gdddgdddgd}t|| d S )Nri   r   r   r         )Za2Zb2r   r   r-   rS   r`   rp   r   r   r   )r~   r   r   r   r   )r"   r.   r   r   rP   r#   r#   r$   %test_iloc_setitem_pure_position_based  s
    """z=TestiLocBaseIndependent.test_iloc_setitem_pure_position_basedc                 C   s   t ddgddgd}ddd}||jd< t ddgddgd}t|| t ddgddgd}||jd< t ddgddgd}t|| d S )Nr   r   )r   r   r   r   g       @g     X@r   )r"   rL   rhsrP   r#   r#   r$   "test_iloc_setitem_dictionary_value  s    


z:TestiLocBaseIndependent.test_iloc_setitem_dictionary_valuec                 C   st  t tjdddddgtdd}|jdd  }t|jd | |jdd df }t	|jd | dddg|_
|jdd  }t|jd | |jdd df }t	|jd | t tjd	dddddgtdd}|jdd
 }t|jd | |jdd
df }t	|jd | ddddg|_
|jdd
g }t|jd | |jdd
gdf }t	|jd | d S )Nr-   r   皙?abcr{   r   r   )r  r~   rS   r   r   )r	   rE   rq   ru   rr   r   rC   rH   r   rt   rZ   )r"   rL   expectr#   r#   r$   "test_iloc_getitem_float_duplicates(  s8       
 z:TestiLocBaseIndependent.test_iloc_getitem_float_duplicatesc                 C   s   G dd d}t ddgdgd}|d|jd< |d|jd< t ddgdgd}|d|jd< t|| t ddgdgd}|d|jd< tj|jd< t ddgdgd}t|| d S )Nc                   @   sB   e Zd ZddddZedddZeZedddZd	d
 Z	dS )zCTestiLocBaseIndependent.test_iloc_setitem_custom_object.<locals>.TON)returnc                 S   s
   || _ d S rk   r   )r"   r   r#   r#   r$   __init__L  s    zLTestiLocBaseIndependent.test_iloc_setitem_custom_object.<locals>.TO.__init__c                 S   s   d| j  dS )Nr   r   r   r"   r#   r#   r$   __str__O  s    zKTestiLocBaseIndependent.test_iloc_setitem_custom_object.<locals>.TO.__str__c                 S   s   | j |j kS rk   r   )r"   otherr#   r#   r$   __eq__T  s    zJTestiLocBaseIndependent.test_iloc_setitem_custom_object.<locals>.TO.__eq__c                 S   s   | S rk   r#   r"  r#   r#   r$   viewW  s    zHTestiLocBaseIndependent.test_iloc_setitem_custom_object.<locals>.TO.view)
r&   r'   r(   r!  rl   r#  __repr__r   r%  r&  r#   r#   r#   r$   TOK  s
   r(  r   r   r{   r[   r   )r	   r   rC   rH   rE   r   )r"   r(  rL   rn   r#   r#   r$   test_iloc_setitem_custom_objectI  s    z7TestiLocBaseIndependent.test_iloc_setitem_custom_objectc                 C   s   t tjddtdtdd}|jd }t|ts8tt	
|j|jd  |jjd d df }t|tsntt	
|j|jd  d S )Nr-   ABCZaabr  r   )r	   rE   rq   Zrandrr   r   r@   r   r=   rC   assert_almost_equalr?   T)r"   rL   rn   r#   r#   r$   !test_iloc_getitem_with_duplicatesk  s     
z9TestiLocBaseIndependent.test_iloc_getitem_with_duplicatesc                 C   sV   t dddgdddggdddgd}|jd d dgf }|jdgdd	}t|| d S )
Nr   r   r-   rS   r`   rp   rb   r   r   )r	   r   ZtakerC   rH   r"   rL   rn   rP   r#   r#   r$   "test_iloc_getitem_with_duplicates2w  s    "z:TestiLocBaseIndependent.test_iloc_getitem_with_duplicates2c                 C   s   t tddddgi}|jd }ttdddidd}t|| |jd d df }tddgtddd}t|| | }|jd d df  d7  < t tddddgi}t|| d S )Nr   r   r   r   r-   )r	   r   r   r   rC   rt   r>   rH   r.  r#   r#   r$   test_iloc_interval~  s    
z*TestiLocBaseIndependent.test_iloc_intervalindexing_funcrhs_funcc                 C   s   t dddg}|ddg|j|dddg< t dddg}t|| tddddgi}|dgdgg|j|dddg< tddddgi}t|| d S )	Nr   r   r   r`   rg   TFr~   )r   r   rC   rt   r	   rH   )r"   r2  r1  r]   rP   rL   r#   r#   r$   test_loc_setitem_boolean_list  s     z5TestiLocBaseIndependent.test_loc_setitem_boolean_listc                 C   s   t ddddgidd}|jd d d d df }t|| d|d	< |jd d d d df }t |d	 |d d
}t|| d S )Nrf   r   r   r-   Int64r/   r   Zfoor   )r   rf   r   )r"   rL   resrP   r#   r#   r$   .test_iloc_getitem_slice_negative_step_ea_block  s    zFTestiLocBaseIndependent.test_iloc_getitem_slice_negative_step_ea_blockc                 C   sp   t ddddgidd}tdgdgg|jtddgtdgf< t ddddgi|d jd}t|| d S )	Nstatusr~   r   r   r   r/   r   r   )r	   rE   r   r   r0   rC   rH   r   r#   r#   r$   *test_iloc_setitem_2d_ndarray_into_ea_block  s    .zBTestiLocBaseIndependent.test_iloc_setitem_2d_ndarray_into_ea_blockc                 C   sV   t ddj}t|}|jd }|d |d ks2t|d |d< |d |d ksRtd S )Nr   g      $@r   r   r   )r   Z_valuesr	   r   r=   )r"   rV   rL   r]   r#   r#   r$   *test_iloc_getitem_int_single_ea_block_view  s    
zBTestiLocBaseIndependent.test_iloc_getitem_int_single_ea_block_viewc                 C   sl   t ddgddgd}t dtddgi|jd d dgf< t tdtd	gddgd}tj|||d
 d S )Nz
2022-01-01z
2022-01-02Z2021Z2022r   rf   r   z2021-01-01 00:00:00z2022-01-01 00:00:00)Zcheck_dtype)r	   r   r   r   rC   rH   r   r#   r#   r$   )test_iloc_setitem_multicolumn_to_datetime  s    $	zATestiLocBaseIndependent.test_iloc_setitem_multicolumn_to_datetime)Pr&   r'   r(   __doc__r)   r*   r+   rA   r9   r
   rE   ZasarrayrC   r   r   rQ   r   r   rX   r^   rw   r   rr   r   r}   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   tdZ&skip_array_manager_not_yet_implementedr   r   r  r  r   r   r  r   r  r  r
  r  r  r  r  r  r  r)  r-  r/  r0  r3  r6  r8  r9  r:  r#   r#   r#   r$   r,   :   s   
+

f		
/!+ $G%
	


	


!"
r,   c                   @   s$   e Zd Zdd Zdd Zdd ZdS )TestILocErrorsc              	   C   sd   |}|t kr| }d}tjt|d |jd  W 5 Q R X tjttd d|jd< W 5 Q R X d S )Nz5Cannot index by location index with a non-integer keyr5   g      @r   )r	   r\   r)   rs   r   r   r!   _slice_iloc_msg)r"   Zseries_with_simple_indexrU   rW   rO   r#   r#   r$   test_iloc_float_raises  s    z%TestILocErrors.test_iloc_float_raisesc              	   C   sp   t jtdd" |jd d d d d d f  W 5 Q R X t jtdd" d|jd d d d d d f< W 5 Q R X d S )NzToo many indexersr5   ztoo many indices for arrayr   )r)   rs   r   r   r!   )r"   Zfloat_framer#   r#   r$   *test_iloc_getitem_setitem_fancy_exceptions  s    &z9TestILocErrors.test_iloc_getitem_setitem_fancy_exceptionsc              	   C   sp   t ddddgi}t ddddgi}tt d|j|< W 5 Q R X d}tjt|d |j|  W 5 Q R X d S )	Nr~   r   r   r-   TFzWDataFrame indexer is not allowed for .iloc
Consider using .loc for automatic alignment.r5   )r	   rC   rD   FutureWarningr   r)   rs   r!   )r"   rL   r.   rO   r#   r#   r$   test_iloc_frame_indexer  s    z&TestILocErrors.test_iloc_frame_indexerN)r&   r'   r(   r?  r@  rB  r#   r#   r#   r$   r=    s   r=  c                   @   sD   e Zd Zdd Zdd Zdd Zejddd	gd
dgdd Z	dS )TestILocSetItemDuplicateColumnsc                 C   s   t d dddddg}t ddddddg}t||gdd}d|jd< |jd dks\t|jd	 dksnt|jjd tjkstd S )
Nr   r   r   r-   rS   r   r   r7   )r   r   )r	   r   r   r=   rm   rE   r   )r"   r   r   rL   r#   r#   r$   *test_iloc_setitem_scalar_duplicate_columns   s    
zJTestILocSetItemDuplicateColumns.test_iloc_setitem_scalar_duplicate_columnsc                 C   sX   t dddggdddgd}dg|jd d df< t dddggdddgd}t|| d S )	Nr   rl   str2r~   r   rb   Zstr3r   r   r   r#   r#   r$   (test_iloc_setitem_list_duplicate_columns  s    zHTestILocSetItemDuplicateColumns.test_iloc_setitem_list_duplicate_columnsc                 C   sj   t tjdtjdddddddgd}|jd d df tj|jd d df< |jjd tjksft	d S )	Nr   r/   r   rS   rf   r   rb   r   )
r	   rE   r   r   r   r   rG   r  rm   r=   r   r#   r#   r$   *test_iloc_setitem_series_duplicate_columns  s     
(zJTestILocSetItemDuplicateColumns.test_iloc_setitem_series_duplicate_columnsrm   
init_valueexpected_value)r   0r   )floatz1.2g333333?c              	   C   s   t |ddggdddgd}d}tjt|d, |jd d df ||jd d df< W 5 Q R X t |ddggdddgd}t|| d S )	Nrl   rE  r~   r   rb   r4   r5   r   )r	   rC   rD   rB   r   rG   rH   )r"   rm   rH  rI  rL   rO   Zexpected_dfr#   r#   r$   *test_iloc_setitem_dtypes_duplicate_columns  s    0
 zJTestILocSetItemDuplicateColumns.test_iloc_setitem_dtypes_duplicate_columnsN)
r&   r'   r(   rD  rF  rG  r)   r*   r+   rL  r#   r#   r#   r$   rC    s   	rC  c                   @   s   e Zd Zdd Zdd ZdS )TestILocCallablec                 C   s  t ddddgtddtdd}|jd	d
  }t||jddg  |jdd
 d d f }t||jddgd d f  |jdd
 dd
 f }t||jddgdf  |jdd
 dd
 f }t||jddgdgf  |jddgdd
 f }t||jddgdf  |jddgdd
 f }t||jddgdgf  |jdd
 df }t||jddgdf  |jdd
 dgf }t||jddgdgf  d S )Nr   r   r-   rS   aabbXYr   rY   c                 S   s   ddgS Nr   r-   r#   r   r#   r#   r$   r   4  r   zCTestILocCallable.test_frame_iloc_getitem_callable.<locals>.<lambda>c                 S   s   ddgS rR  r#   r   r#   r#   r$   r   7  r   c                 S   s   ddgS rR  r#   r   r#   r#   r$   r   :  r   c                 S   s   dS Nr   r#   r   r#   r#   r$   r   :  r   r   c                 S   s   ddgS rR  r#   r   r#   r#   r$   r   =  r   c                 S   s   dgS rS  r#   r   r#   r#   r$   r   =  r   c                 S   s   dS rS  r#   r   r#   r#   r$   r   A  r   c                 S   s   dgS rS  r#   r   r#   r#   r$   r   D  r   c                 S   s   ddgS rR  r#   r   r#   r#   r$   r   G  r   c                 S   s   ddgS rR  r#   r   r#   r#   r$   r   J  r   )r	   rr   r   rC   rH   rt   )r"   rL   r5  r#   r#   r$    test_frame_iloc_getitem_callable/  s"    "z1TestILocCallable.test_frame_iloc_getitem_callablec                 C   sF  t ddddgtddtdd}| }d	|jd
d < | }d	|jddg< t|| | }d|jdd d d f< | }d|jddgd d f< t|| | }d|jdd dd f< | }d|jddgd	f< t|| | }d|jdd dd f< | }d|jddgd	gf< t|| | }d|jddgdd f< | }d|jddgd	f< t|| | }d|jddgdd f< | }d|jddgd	gf< t|| | }d|jdd d	f< | }d|jddgd	f< t|| | }ddg|jdd d	gf< | }ddg|jddgd	gf< t|| d S )Nr   r   r-   rS   rN  rO  r   rY   r   c                 S   s   ddgS rR  r#   r   r#   r#   r$   r   S  r   zCTestILocCallable.test_frame_iloc_setitem_callable.<locals>.<lambda>r   c                 S   s   ddgS rR  r#   r   r#   r#   r$   r   Y  r   r`   c                 S   s   ddgS rR  r#   r   r#   r#   r$   r   _  r   c                 S   s   dS rS  r#   r   r#   r#   r$   r   _  r      c                 S   s   ddgS rR  r#   r   r#   r#   r$   r   e  r   c                 S   s   dgS rS  r#   r   r#   r#   r$   r   e  r   r   c                 S   s   dS rS  r#   r   r#   r#   r$   r   l  r   c                 S   s   dgS rS  r#   r   r#   r#   r$   r   r  r   rg   c                 S   s   ddgS rR  r#   r   r#   r#   r$   r   x  r   c                 S   s   ddgS rR  r#   r   r#   r#   r$   r   ~  r   )r	   rr   r>   r   rC   rH   )r"   rL   r5  r   r#   r#   r$    test_frame_iloc_setitem_callableM  sR    "z1TestILocCallable.test_frame_iloc_setitem_callableN)r&   r'   r(   rT  rW  r#   r#   r#   r$   rM  .  s   rM  c                   @   s,   e Zd Zdd Zdd Zdd Zdd Zd	S )
TestILocSeriesc              	   C   s  t tjdttdddd}| }tt|D ](}|j| }||j	|  }t
|| q4|jtdd }|jdd }t
|| t
d  d|d d < W 5 Q R X |rt
|| n|jdd dk st|jddddd	g }||j	ddddd	g }t
|| d S )
Nrg   r   r_   r   rY   r   r-   rS   r`   )r   rE   rq   ru   rr   r9   r>   r|   r   rZ   rC   r+  rA   r   rt   rD   r   r=   Zreindex)r"   r   r]   Zser_originalirn   r   rP   r#   r#   r$   	test_iloc  s"     
zTestILocSeries.test_ilocc                 C   s.   t dddgdddgd}|jd dks*td S )Nr   r   r   rY   )r   r   r=   )r"   r]   r#   r#   r$   test_iloc_getitem_nonunique  s    z*TestILocSeries.test_iloc_getitem_nonuniquec                 C   s\   t dddg}t dddgdddgd}|jdd |jdd< t dddg}t|| d S )	Nr   r   r-   rS   r`   rp   r   rY   )r   r   rC   rt   )r"   Zser1Zser2rP   r#   r#   r$   r    s
    z4TestILocSeries.test_iloc_setitem_pure_position_basedc                 C   sd   t dgtjgd}|jd d df d|jd d df< t dgttgddd}t|| d S )Ntest)r~   r   r   r4  r/   )	r	   rE   r   r   rG   r   r   rC   rH   r   r#   r#   r$   #test_iloc_nullable_int64_size_1_nan  s    &z2TestILocSeries.test_iloc_nullable_int64_size_1_nanN)r&   r'   r(   rZ  r[  r  r]  r#   r#   r#   r$   rX    s   rX  ).r;  r   rewarningsr   r   ZnumpyrE   r)   Zpandas.errorsr   Zpandas.util._test_decoratorsutilZ_test_decoratorsr<  Zpandasr   r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   Zpandas._testingZ_testingrC   Zpandas.api.typesr   Zpandas.tests.indexing.commonr   escaper>  r   r,   r=  rC  rM  rX  r#   r#   r#   r$   <module>   s:   D         !-/V