HOME


sh-3ll 1.0
DIR:/usr/local/lib64/python3.6/site-packages/pandas/tests/indexing/__pycache__/
Upload File :
Current File : //usr/local/lib64/python3.6/site-packages/pandas/tests/indexing/__pycache__/test_loc.cpython-36.pyc
3

���hT��@s>dZddlmZddlZddlZddlZddlmZddl	Z
ddl	mZmZm
Z
mZddljZddlmZddlmZGdd	�d	e�ZGd
d�d�Zdd
�Zejjddddgddgddgdfdddgddgddgdfdddgddgddgdfg�dd��Zdd�Zdd�Zdd�Zdd �Zd!d"�Z d#d$�Z!d%d&�Z"d'd(�Z#dS))z$ test label based indexing with loc �)�StringION)�
_is_numpy_dev)�	DataFrame�Series�	Timestamp�
date_range)�	is_scalar)�Basec@s\eZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�Zdd�ZdS)�TestLoccCs|jdddgtd�dS)N�loc��labels)�typs�fails)�check_result�KeyError)�self�r�@/tmp/pip-build-5_djhm0z/pandas/pandas/tests/indexing/test_loc.py�test_loc_getitem_intszTestLoc.test_loc_getitem_intcCs|jdddgtd�dS)Nr�c�empty)rr)rr)rrrr�test_loc_getitem_labelszTestLoc.test_loc_getitem_labelcCs�|jdddddddgtd�|jddd	gtd�|jddd	gtd�|jdd
dddgtd�|jdd
dgtd�|jdd
dgdtd�|jdd
d	gdtd�dS)
Nr�f�ints�uintsr
�mixed�ts)rr�floats�r)r�axesr)rr)rrrr�#test_loc_getitem_label_out_of_rangesz+TestLoc.test_loc_getitem_label_out_of_rangecCsdS)Nr)rrrr�test_loc_getitem_label_list+sz#TestLoc.test_loc_getitem_label_listcCsz|jddddgdgtd�|jddddgdd	d
gdtd�|jddd
dgdd	d
gdtd�|jddddgdgdtd�dS)Nrr�rr)rr�
rrr)rr r�����Zmulti)r#r%)r#r()rr))rr)rrrr�(test_loc_getitem_label_list_with_missing0sz0TestLoc.test_loc_getitem_label_list_with_missingcCs"|jddddgddgdtd�dS)	Nrr��(rrr#)rr r)rr)rrrr�!test_loc_getitem_label_list_failsAsz)TestLoc.test_loc_getitem_label_list_failscCsdS)Nr)rrrr�!test_loc_getitem_label_array_likeGsz)TestLoc.test_loc_getitem_label_array_likecCs$ddddg}|jd|dgtd�dS)NTFrr)rr)r�
IndexError)r�brrr�test_loc_getitem_boolLszTestLoc.test_loc_getitem_boolcCs�|jdtdd�dddddgtd	�|jdtd
d�dgdtd�|jdtd
d�dgdtd�|jdtd
d�dgdtd�|jdtd
dd
�dgdtd�dS)Nrr#r%r
rrrr)rrZ20130102Z20130104)rr rr�rr()r�slice�	TypeErrorr)rrrr�test_loc_getitem_label_sliceRsz$TestLoc.test_loc_getitem_label_slicecCsttddgddgddggddgdddgd�}tjd	�|jd<tddgd
dgddggddgdddgd�}tj||�dS)Nr�a���col1Zcol2r$r#)�columns�indexrr)r#r8)r�np�aranger�tm�assert_frame_equal)r�df�expectedrrr� test_setitem_from_duplicate_axisps(z(TestLoc.test_setitem_from_duplicate_axisN)
�__name__�
__module__�__qualname__rrr!r"r*r-r.r1r5rArrrrr
sr
c@s�eZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Ze	j
jd
ddgddddgg�dd��Zdd�Z
dd�Zdd�Zdd�Zdd�Zdd�Zdd�Zd d!�Ze	j
jd"d#d$d%d&d'd(d)d*g�d+d,��Zd-d.�Zd/d0�Zd1d2�Ze	j
jd3d4ged5d4d5�ejd4g�g�e	j
jd6d7gejd7g�g�d8d9���Ze	j
jd:d;d<gd4d=d>d%gfd?ejd?d?d?d?gd@dAejejgd?d?d?d?ggd4d=d>d%gdB�fdCd>d%gfd?dDgejdCd<ejejgd@dAd?dDgdEdFejejggd4d=d>d%gdB�fdCd4d=d>gfejd?dDdGgej dH�ejdCd<ejgd?dDdGgdEdFejggd4d=d>gdB�fedCd@d5�d=d>d%gfd?dDdGgdIdJdKggejdCd<ejejgd@d?dDdGgdEdIdJdKggd4d=d>d%gdB�fedCd@d5�d>d4d%gfejd?dDdGgdIdJdKggej dH�ejdCd<ejejgdDdAd?dGgdJdFdIdKggd4d=d>d%gdB�fed5d5d5�d4d>gfejd?dDgdGdIgdJdKggd4d>gdB�ejd?d<dDgdGdAdIgdJdFdKggd4d=d>gdB�fg�dLdM��Z!dNdO�Z"dPdQ�Z#dRdS�Z$dTdU�Z%dVdW�Z&dXdY�Z'dZd[�Z(e	j
j)d\d]�d^d_��Z*d`da�Z+dbdc�Z,e	j
j)e-ddd]�dedf��Z.dgdh�Z/didj�Z0dkdl�Z1d5S)m�TestLoc2cCsTttjjd�dd�td�D�d�}|jd	}|jdd�dfjd}tj||�dS)
Nrr)cSsg|]}d|d�qS)ZABCDEr)r)�.0�xrrr�
<listcomp>�sz2TestLoc2.test_loc_getitem_dups.<locals>.<listcomp>)r:�Ar)rr))rIr)rr;�randomZ
random_sample�rangerr=�assert_series_equal)rr?r@�resultrrr�test_loc_getitem_dups�s
 
zTestLoc2.test_loc_getitem_dupscCs�tddddtd�ggdddddgdgd�}tddddtd�gdddddgdd�}|jd	}tj||�|jd}tj||�dS)
Nr#r�foo�bar�20130101r6)r9r:)r:�namer)rrr�ilocr=rLr)rr?r@rMrrr�test_loc_getitem_dups2�s


zTestLoc2.test_loc_getitem_dups2cCsttd�td�tjddd�ddtjddd�ddd	��jd
�}tddd
gg�}|j�}|j|d9<tj	|j|d|j|�tddg�}|j�}|j|d9<|j|d|j|ks�t
�tddd
gg�}|j�}|j|d9<tj|j|d|j|�dS)NZrtttiZaaader)�float64)�dtypegq=
ףp�?rg�(\���?)�merOrP�bar2rW�rrPrXg@�tg�(\��տ)r�listr;r<Z	set_index�tuple�copyrr=rL�AssertionErrorr>)rZdf_orig�indexerr?rrr�test_loc_setitem_dups�s$zTestLoc2.test_loc_setitem_dupscCs�tdddgtdddgdd�d��}|d	dk}|j|d
fd}||j|d
f<tdddgtdddgdd�d��}tj||�tdddgdddgd�d
d�}|d	dk}|j|d
f}||j|d
f<tdddgdddgd�d
d�}tj||�dS)Nrr#�dr7i,Zuint32)rV)r6r0r6r0��i-�uint64)rrrr=r>)rZdf1�ixZnewb1r@Zdf2Znewb2rrr�test_loc_setitem_slice�s" zTestLoc2.test_loc_setitem_slicecCsxtdgdgdgdgd��}dddg}|jdd�|fjd	�|jdd�|f<tdgdgdgdgd�d	d
�}tj||�dS)
NrIg333333�?gg@)�idr6r0rr6r0rZfloat32)rVg�g�)rr�astyper=r>)rr?�colsr@rrr�test_loc_setitem_dtype�s
&zTestLoc2.test_loc_setitem_dtypecCsrttd�dddgd�}tjtdd��|ddgWdQRXttd��}tjtdd��|d	dgWdQRXdS)
Nr%r6r0r)r:zwith any missing labels)�match�dr)rrK�pytest�raisesr)r�srrr�$test_getitem_label_list_with_missing�sz-TestLoc2.test_getitem_label_list_with_missingr:TFc	CsNtdddg�}dt|��dt|���}tjt|d��|j|}WdQRXdS)Nr#rr%z Boolean index has wrong length: z instead of )rj)r�lenrlrmr/r)rr:rn�msg�_rrr�test_loc_getitem_bool_diff_len�sz'TestLoc2.test_loc_getitem_bool_diff_lencCsdS)Nr)rrrr�test_loc_getitem_int_slice�sz#TestLoc2.test_loc_getitem_int_slicecACs�ttjjd�dddgdddgd�}d	}tjt|d
��|jddgddgfWdQRXttd
�}d|jd<d|jd<tjtdd
��|jdWdQRXd}tjt|d
��|jddgWdQRXd}tjt|d
��|jdgWdQRXd|jd<tjtdd
��|jddgWdQRXd|d<d}tjt|d
��|jdgWdQRX|d=tjt|d
��d|jdg<WdQRXtdgdggddgdgd�}d}tjt|d
��|jdgdd�fWdQRXtjt|d
��|jdgWdQRXdS) Nr%r6r0r�er�g)r:r9zJ\"None of \[Int64Index\(\[1, 2\], dtype='int64'\)\] are in the \[index\]\")rjr#r)rVz^-1$zL\"None of \[Int64Index\(\[-1, -2\], dtype='int64'\)\] are in the \[index\]\"zE\"None of \[Index\(\['4'\], dtype='object'\)\] are in the \[index\]\"�4zwith any missing labelszH\"None of \[Int64Index\(\[-2\], dtype='int64'\)\] are in the \[index\]\"r�valuezG\"None of \[Int64Index\(\[3\], dtype='int64'\)\] are in the \[index\]\")r%r%���ry���ryryrzrzrz)	rr;rJrlrmrrr�object)rr?rqrnrrr�test_loc_to_fail�s@  



zTestLoc2.test_loc_to_failcCsrtdddg�}|jdgtjttjd�d��|jdgWdQRXtjtdd��|jddgWdQRXdS)Nr#rr%z="None of [Int64Index([3], dtype='int64')] are in the [index]")rjzwith any missing labels)rrrlrmr�re�escape)rrnrrr�test_loc_getitem_list_with_fail6sz(TestLoc2.test_loc_getitem_list_with_failcCs�ttjjdd�dddddgd	�}|jjd
d��}|jtj|�}|j|}tj||�|j|j	}tj||�|jt
j|dd
�}tj||�dS)Nr)r$)�sizeZalpha_0Zalpha_1Zalpha_2Zbeta_0Zbeta_1)r:cSsd|kS)N�alphar)rGrrr�<lambda>Ssz)TestLoc2.test_loc_index.<locals>.<lambda>�boolean)rV)r)r$)rr;rJr:�mapr�arrayr=r>�values�pd)rr?�maskr@rMrrr�test_loc_indexJs
zTestLoc2.test_loc_indexcCs�ttjjdd�ddddgddddgd�}|jdd�dd�fjdd�dd�f}|jddgkj�sft�|j	ddgkj�s|t�tt
d	�gd
gd��jd}tt
d	�d
gdd
gdd�}tj
||�|jtks�t�dS)Nr(rI�B�C�D)r9r:rrrQr#)r6r0r6r0)r:rR)rr;rJ�randrrSr9�allr^r:rrr=rLrVr{)rr?rMr@rrr�test_loc_general_s
(zTestLoc2.test_loc_generalcCs(ttdtd�tjd�ttd�tjd�d��}ttdd�ttd�tjd�d��}d|jdd�df<tj||�ttdd�ttd�tjd�d��}tj	dtjd�|jdd�df<tj||�ttdd�ttd�tjd�d��}tj	dddddgtjd�|jdd�df<tj||�ttd	td�d
�ttd�tjd�d��}ttdd�ttd�tjd�d��}d	|jdd�df<tj||�ttdtd�d
�ttd�tjd�d��}ttdd�ttd�tjd�d��}d|jdd�df<tj||�tdtt
d�g�i�}d
|jdd�df<tdtd
g�i�}tj||�dS)Nrr))r:rV)rV)�date�valz
2000-01-01z	2000-01-5r�rO)r:g�?Z20180101�string)rrrKr;�int64rrr=r>r�r)rr@r?rrr�test_loc_setitem_consistencyrsL(z%TestLoc2.test_loc_setitem_consistencycCsttddgd�}|djtj�|d<tddgd�}d|jdd�df<tj||�tddgd�}d|d<tj||�dS)NrG�y)r9r#)rrgr;r�rr=r>)rr@r?rrr�"test_loc_setitem_consistency_empty�sz+TestLoc2.test_loc_setitem_consistency_emptycCs�d}tjt|�ddgdddgd�}tj|jdd�df�|jdd�df<tj|jdd�df�|jdd�df<|jdd�df|jdd�df|jdd�df<|jdd�dfjd
�|jdd�df<tddd
dg|jdd�}tj	|d|�dS)Na�Level_0,,,Respondent,Respondent,Respondent,OtherCat,OtherCat
Level_1,,,Something,StartDate,EndDate,Yes/No,SomethingElse
Region,Site,RespondentID,,,,,
Region_1,Site_1,3987227376,A,5/25/2015 10:59,5/25/2015 11:22,Yes,
Region_1,Site_1,3980680971,A,5/21/2015 9:40,5/21/2015 9:52,Yes,Yes
Region_1,Site_2,3977723249,A,5/20/2015 8:27,5/20/2015 8:41,Yes,
Region_1,Site_2,3977723089,A,5/20/2015 8:33,5/20/2015 9:09,Yes,Norr#r)�headerZ	index_col�
Respondent�	StartDate�EndDate�Durationztimedelta64[s]idi�iHg�@)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�Zread_csvr�to_datetimerrgrr:r=rL)r�datar?r@rrr�-test_loc_setitem_consistency_slice_column_len�s	"""
z6TestLoc2.test_loc_setitem_consistency_slice_column_len�unit�Y�Mr��h�mrn�ms�uscCs�tdtjd�tjd�gi�}|jdd�dfjjd|�d��|jdd�|f<|jdd�dfjjd|�d��|d<t|jdd�df|d�}tj|jdd�|f|�dS)N�	timestampz2017-02-11 12:41:29z1991-11-07 04:22:37zdatetime64[�]r@)rR)	rr;�
datetime64rr�rgrr=rL)rr�r?r@rrr�test_loc_assign_non_ns_datetime�s0&z(TestLoc2.test_loc_assign_non_ns_datetimecCs�tjdddddgi�}tj|dddd�|d	<|jdd�d	f|jdd�d
f<|jddgd	f|jddgd
f<tdd
d
gdddgdddgdddggdd	d
gd�}d	d
g}||jtj�||<tj||�dS)Nr�l�CO!gl�s~!gl4	8�lA{�?�r�T)r��cacheZdate_dtZ
date_dt_cprr%z2017-01-24 13:26:12.711z2017-01-24 13:52:05.110z2018-10-22 13:44:05.888z2018-10-23 08:08:41.025)r9)rZ	from_dictr�r�r�applyr=r>)rr?r@r9rrr�test_loc_modify_datetime�s  
z!TestLoc2.test_loc_modify_datetimec	Cs�ttjjdd�td�td�d�}|jd}d|jd<|jd}|dksJt�|jd}|dks`t�d|jdd�d	d
�f<|jdd�d	d
�f}|jdd�dd�f}tj	||�tdddgdgd�}tj
dd
dgdd�|jdddgdf<tttdd
dgdddgd�d��j
dddgd�}tj	||�dd�td�D�}tjddd�}dd�td�D�}tjddd�}tt|�j|��}t|d�}tj|d<||j|df<tj|d	<||j|d	f<ttt||d�t||d�d��j
|d�}tj	||�tdd
dgtjd��}|j|j|j|jkd	f<tdd
dgtjd��}tj	||�tdd
gddgddgd��}|jddd
gf}tddgdd
gtdd�}tj||�tdd
gddgddgd��}ddg|jddd
gf<tj	||�dS)Nr(Zabcd�ABCD)r:r9rr#r6rIr�r�r%r)rr�)rV)r:)rIcSsg|]}dt|��qS)�@)�str)rF�irrrrH/sz3TestLoc2.test_loc_setitem_frame.<locals>.<listcomp>cSsg|]}dt|��qS)r�)r�)rFr�rrrrH2s)rIr�r0)r#rr6)r:rVrRr&)rr)r6rI)r6rI)rr)rr;rJ�randnr[rSrr^r=r>r��dictrZreindexrKr<�set�union�nanrIr�r{rL)	rr?rMr@Zkeys1Zval1�keys2Zval2r:rrr�test_loc_setitem_framesR 



$"


zTestLoc2.test_loc_setitem_framecCstdddgttd�tjd�d��}|jdd�}|jd	d�|_||jd	d�<tdddgtdddgtjd�d��}tj||�tt	d
d�ttd�tjd�d
��}tt
d�t
d�t
d�t
d�t
d�gtd	dd	ddgtjd�d
��}|jd	d�}|jdd�|_||jdd�<tj||�dS)NrOrPZbazr%)rV)rIr�r#rrz
2000-01-01z	2000-01-5r))r�r�Z20000101Z20000102Z20000103r()rrrKr;r�rr:r=r>rr)rr?�rhsr@rrr� test_loc_setitem_frame_multiplesUs, "z)TestLoc2.test_loc_setitem_frame_multiplesr_rINrx�ZcCsNtjddgddggddgd�}||jd|f<|jd
}t|�rF|d	ksJt�dS)Nr#rr%r(rIr�)r9rr�)rrI)r�rrrr^)rr_rxr?rMrrr�"test_loc_setitem_with_scalar_indexzs	
z+TestLoc2.test_loc_setitem_with_scalar_indexzindex,box,expectedrrr�r�r'r%r()r9r#r2r)r&�	)rVr$��cCs>tjddgddgddggddgd	�}||j|<tj||�dS)
Nr#rr%r(r)r&rIr�)r9)r�rrr=r>)rr:Zboxr@r?rrr� test_loc_setitem_missing_columns�s6$
z)TestLoc2.test_loc_setitem_missing_columnscCstdtd�jd�tjgi�}|j}|jdg}tj|j|�|jdg}tj|j|�ddl	}td|j	ddd�|j	ddd�gi�}|j}|jdg}tj|j|�|jdg}tj|j|�td	d
gdgdi�}|j}|jdd�}tj|j|�|jdd�}tj|j|�dS)
Nr�rQ�UTCrr#i�i�r�textz
some wordsr�r%)
rr�tz_localizer�ZNaTZdtypesrSr=rL�datetime)rr?r@rMr�rrr�test_loc_coercion�s("zTestLoc2.test_loc_coercioncCs�tjd�jd�tjd�jd�g}tj|ddgd�}tjtd�}|d|d<|d	|d<tj||�tjtd�}|d|jd<|d	|jd<tj||�dS)
N�*r�i�rOrP)r:)rVrr#)r�r�r�rr{r=rLr)r�valsr@�serrrr�test_setitem_new_key_tz�sz TestLoc2.test_setitem_new_key_tzcCs�tddddddgddddddgd	�d
dd
dddgd�}d}tjt|d
��|jdd�WdQRXd}tjt|d
��|jd
d�WdQRXd}tjt|d
��|jdd�WdQRXtddddddgddddddgd	�d
dd
dddgd�jd
d�}|jdd�}tddddgddddgd	�ddddgd�}tj||�|jd
d�}tj||�|jdd�}tdddgdddgd	�dddgd�}tj||�dS)Nr#rr%r(r)r&r'r2)rIr�r)r:z5'Cannot get left slice bound for non-unique label: 1')rjz5'Cannot get left slice bound for non-unique label: 0')�axis)rrlrmrrZ
sort_indexr=r>)rr?rqrMr@rrr�test_loc_non_unique�s,44*$zTestLoc2.test_loc_non_uniquecs�td���fdd�}�fdd�}|dd�}|jjdks:t�tjd�}|j|}|||�}tj||�|d	d
�}|jjdks~t�tjd
�}|j|}|||�}tj||�dS)NZABCDEFGcsJtjttjj|t���tj|��d�ttj|t��f�dg|�d�g�S)N)r:r9r)	r��concatrr;rJr�rpr<�ones)�l�l2)r9rr�gen_test$sz;TestLoc2.test_loc_non_unique_memory_error.<locals>.gen_testcsLt|�}tj|jdg�ttj|t��f�dg|�d�|j|dd��g�S)Nr)r:r9r#)rpr�r�Ztakerr;r�)r?r�Zlen_mask)r9rr�gen_expected2s
z?TestLoc2.test_loc_non_unique_memory_error.<locals>.gen_expectedi�raFi��
i��)	r[r:Z	is_uniquer^r;r<rr=r>)rr�r�r?r�rMr@r)r9r� test_loc_non_unique_memory_errors







z)TestLoc2.test_loc_non_unique_memory_errorcCs\tddgddgg�}d|j_|jddgjj}|dks:t�|jddgjj}|dksXt�dS)Nr#Z
index_namer)rr:rRrSr^r)rr?rMrrr�
test_loc_namePszTestLoc2.test_loc_namecCs�tjdd�}tj|jdd�gf|jdd�dd�fddd�tj|jgdd�f|jdd�dd�fddd�tj|jg|jdd�dd�fddd�dS)Nr)rrT)Zcheck_index_type�check_column_type)r=ZmakeCustomDataframer>rrS)rr?rrr�!test_loc_empty_list_indexer_is_okZs..z*TestLoc2.test_loc_empty_list_indexer_is_okcCs(tddddgi�}|jdd�}||k	s,t�|dd�|k	s@t�dddg|d<|ddkj�sbt�||jdd�dd�fks�t�ttjjdd��}|d|jdd�dfks�t�tdddddd	g�}|jdd�}||k	s�t�|dd�|k	s�t�d
ddg|dd�<t|dd�d
ddgk��s$t�dS)
Nr6r#rr%r(r$rr)r&r'r2r�)rrr^r�r;rJr�r)rZoriginal_dfZ	sliced_dfr?Zoriginal_seriesZ
sliced_seriesrrr�&test_identity_slice_returns_new_objectjsz/TestLoc2.test_identity_slice_returns_new_objectz!accidental fix reverted - GH37497)�reasoncCs|tttd�td��ddgd�}|j�}|jdd�df}|d7}tj||�|j�}|j|jdf}|d7}tj||�dS)Nr%r6r0)r9r)r�ziprKr]rr=r>r:)rrGr��q�zrrr�test_loc_copy_vs_view�szTestLoc2.test_loc_copy_vs_viewcCs�tjddgtjd�jdtjd�jgd�}|jtjd�jd}|jd}||ksVt�|jtjd�jdg}|jdg}tj	||�|jtjd�jdtjd�jg}tj	||�dS)Nr#rrc)r:r)
r�rr;Ziinfo�maxrrSr^r=rL)rrnrMr@rrr�test_loc_uint64�s&
"zTestLoc2.test_loc_uint64cCs�dddg}t|dgt|�d��}tddgd�}||jdd�df<tj||�tdgtjgd��}tddgtd	�}|jd|jd<tj||�dS)
Nr#rr%)rGr�rGr�)r9g�?)r9rVr)rrG)rrG)rrprr=r>r;r��float)rr�r@r?rrr�test_loc_setitem_empty_append�s
z&TestLoc2.test_loc_setitem_empty_appendzgh-35481cCs�ddg}tddgd�}|jjtj�|_d}tjt|d��||jddgdf<WdQRXd	}tjt	|d��||jdd�df<WdQRXdS)
Nr#rrGr�)r9zFNone of \[Int64Index\(\[0, 1\], dtype='int64'\)\] are in the \[index\])rjrz?cannot copy sequence with size 2 to array axis with dimension 0)
rr:rgr;r�rlrmrr�
ValueError)rr�r?rqrrr�$test_loc_setitem_empty_append_raises�sz-TestLoc2.test_loc_setitem_empty_append_raisescCsFtddgddgg�}|jtjd�}tjddgdd�}tj||�dS)Nr#rr%r(r)rR)rrr;r�r�rr=rL)rr?rMrnrrr�test_indexing_zerodim_np_array�sz'TestLoc2.test_indexing_zerodim_np_arraycCs,tddg�}|jtjd�}|dks(t�dS)Nr#rr)rrr;r�r^)rrnrMrrr�%test_series_indexing_zerodim_np_array�sz.TestLoc2.test_series_indexing_zerodim_np_arraycCsnddddddgdgd}t|tdd�d	�}tjtdd�tjd
�}ddddddg|jddd�<tj||�dS)
Nr#rr%r(r)r&i�i�)r:)r:rVi�ry)rrKr�r;rUrr=rL)rr�r@rMrrr�test_loc_reverse_assignment�s
z$TestLoc2.test_loc_reverse_assignment)2rBrCrDrNrTr`rerirorl�mark�parametrizersrtr|rr�r�r�r�r�r�r�r�r�r3r;r�r�r�rr�r�r�r�r�r�r�r�r�r�Zxfailr�r�r�rr�r�r�r�rrrrrE~s�
&CJ&B%&$
($$"6$"3
rEc	CsZtjddddgdd�}tdddd	gtd
ddd
��}tjtdd��|j|WdQRXdS)Nz
2001-01-04z
2001-01-02z
2001-01-14r�)rVrr)r2r�z
2001-01-01r�r()�freq�periodszwith any missing labels)rj)r;r�rrrlrmrr)�keyrnrrr�1test_series_loc_getitem_label_list_missing_values�s
r�z8columns, column_key, expected_columns, check_column_typei�i�i�r#TZAllFrcCsVttjjdd�|td�d�}|jdd�|f}|jdddg|f}tj|||d�dS)Nr%�ABC)r9r:rIr�r�)r�)	rr;rJr�r[rSrr=r>)r9Z
column_keyZexpected_columnsr�r?r@rMrrr�*test_loc_getitem_label_list_integer_labels�sr�cCs�tjjdd�}tj|�}tj|jdd�df<tj|tjtjgd�j	dd�f�}tj|dddddgd	�}t
j||�tj|�}tj|jdd�df<t
j||�dS)
Nr2r(g�?r#gg�?g@g@)r9)r;rJr�r�rr�rZhstackr�Zreshaper=r>)Z	rand_datarMZ
expected_datar@rrr�test_loc_setitem_float_intindex�s
$
r�c	Cs~dd�dD�}tjtjd�td�tjj|�d�}|jd	d
�dd�}tjtjd�td�tjjddddg�d�}tj	||�dS)NcSsg|]}dD]}||f�qqS)r'r2r�r$)r'r2r�r$r)rF�yrr�rrrrH
sz)test_loc_axis_1_slice.<locals>.<listcomp>����r$r2Z
ABCDEFGHIJ)r:r9r#)r�r�r(r')r�r�)r$r2)r�r�)r�r2)r$r()r�r�)r�r$)r�r')r�r2)
r�rr;r�r\�
MultiIndex�from_tuplesrr=r>)rhr?rMr@rrr�test_loc_axis_1_slicesr�cCs\tjdtd�tjjtd�td�g�d�}|j�}|jddgf|jddgf<tj||�dS)Nr6r)r:r9rr#)rr#)rr#)	r�rrKr�Zfrom_productr]rr=r>)r@rMrrr�!test_loc_set_dataframe_multiindexs
$r�cCs8tjtd�tjddgtd��}|jd}|dks4t�dS)Nrr#g@)rVr)r�rrK�Indexr{rr^)r�rMrrr�test_loc_mixed_int_float(s
r�c
Csptjtd�ddddgd�}tjtdd��d	|jdd
�<WdQRXtjd	d	d	d
gddddgd�}tj||�dS)Nr(rIr�r�r�)r:F)�check_stacklevelrr%)r�rrKr=�assert_produces_warning�
FutureWarningrrL)r�r@rrr�*test_loc_with_positional_slice_deprecation0s
r�cCs�tjddd�}tjtjjd�|d�}|d}d}x`||gD]T}tjt|d��|jd	d�WdQRXt	j
td
d��d	|jd	d�<WdQRXq:Wtjt|d��|jd	d�d	fWdQRXt	j
t��d|jd	d�d	f<WdQRXdS)
Nz
2016-01-01r%)r�r)r:rzOcannot do slice indexing on DatetimeIndex with these indexers \[1\] of type int)rjr#F)r�)r%r)r�rrr;rJrlrmr4rr=r�r�)�dtir?r�rq�objrrr�#test_loc_slice_disallows_positional;sr�cCs�tjtjjdd�dddgtjdddd�d	�}|jd
dddgj�}|j}tj	|j
�}d}tjt
|d
��|j|WdQRXtjt
|d
��|dj|WdQRXdS)Nr)r%r6r0rZ2012�H)r�r�)r9r:rrz/None of \[TimedeltaIndex.* are in the \[index\])rj)r�rr;rJr�rrSr]r:ZTimedeltaIndexZasi8rlrmrr)r?r�Ztdirqrrr�'test_loc_datetimelike_mismatched_dtypesWsr�cCs�tjdddd�}tjtjjdd�|d�}tj||j|�tj||jt	|��tj||jt	|��tj|j
dd	�|j|dd	��tj||jt	|��dS)
Nz2002-01z2003-12r�)r��r$)r:rr))r�Zperiod_rangerr;rJr�r=r>rr[rS)�idxr?rrr�"test_loc_with_period_index_indexerms$r)$�__doc__�iorr}Znumpyr;rlZpandas.compat.numpyrZpandasr�rrrrZpandas._testingZ_testingr=Zpandas.api.typesrZpandas.tests.indexing.commonr	r
rEr�r�r�r�r�r�r�r�r�r�r�rrrrr�<module>s@
mi
&