403Webshell
Server IP : 209.209.40.120  /  Your IP : 216.73.217.112
Web Server : Microsoft-IIS/10.0
System : Windows NT NEWWWW 10.0 build 17763 (Windows Server 2019) i586
User : NEWWWW$ ( 0)
PHP Version : 8.3.30
Disable Function : NONE
MySQL : OFF  |  cURL : ON  |  WGET : OFF  |  Perl : OFF  |  Python : OFF  |  Sudo : OFF  |  Pkexec : OFF
Directory :  C:/Python312/Lib/site-packages/pandas/tests/frame/indexing/__pycache__/

Upload File :
current_dir [ Writeable ] document_root [ Writeable ]

 

Command :


[ Back ]     

Current File : C:/Python312/Lib/site-packages/pandas/tests/frame/indexing/__pycache__/test_setitem.cpython-312.pyc
�

/	�g����n�ddlmZddlZddlZddlmcmZddlm	Z
ddlmZddl
mZmZmZmZddlZddlmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#ddl$m%Z&ddl'm(Z(ddl)m*Z*Gd	�d
�Z+Gd�d�Z,Gd
�d�Z-Gd�d�Z.Gd�d�Z/Gd�d�Z0Gd�d�Z1d�Z2d�Z3y)�)�datetimeN)�	_registry)�is_object_dtype)�CategoricalDtype�DatetimeTZDtype�
IntervalDtype�PeriodDtype)�Categorical�	DataFrame�
DatetimeIndex�Index�Interval�
IntervalIndex�
MultiIndex�NaT�Period�PeriodIndex�Series�	Timestamp�cut�
date_range�notna�period_range)�SparseArray)�BDayc�~�eZdZd�Zej
j
dgd��d��Zd�Zd�Z	d�Z
d�Zd	�Zd
�Z
d�Zd�Zd
�Zd�Zd�Zd�Zej
j
ded�ed�fedd��edd�fedd��edd��fg�d��Zej
j
dej8D����cgc](}t;|j<t>�s|j<��*c}}}}d d!gz�d"��Z d#�Z!ej
j
d$gd%��d&��Z"ej
j
d$gd%��d'��Z#d(�Z$d)�Z%d*�Z&d+�Z'ej
j
dgd,��d-��Z(ej
j
d.d/d/gg�d0��Z)d1�Z*d2�Z+ej
j
d3gd4�gd5�g�d6��Z,d7�Z-d8�Z.d9�Z/d:�Z0d;�Z1ej
j
d<gd=�gd>�d?fgd@�gd>�d?fgdA�gd>�d?fgdB�gdC�dDfgdE�gdF�dGfgdH�gdF�dIfg�dJ��Z2dK�Z3ej
j
dLgdM�dNe4gdO�gdO�gdO�ggdM��P�fdQdRgdNdSge4gdT�gdU�gdV�ggdM��P�fgdW�e5jlgdX�e5jn�Y�e4gdX�gdX�gdX�ggdW��P�fgdZ�gdX�gd[�gd\�ge4gd]�gd^�gd_�ggdM��P�fgd`�e5jlgdX�gd[�gd\�ge5jn�Y�e4gda�gdb�gdc�ggdM��P�fdddQge4dNdSgdedfgdgdhggdddQg�P�e4gdi�gdj�gdk�ggdW��P�fg�dl��Z8dm�Z9dn�Z:do�Z;dp�Z<dq�Z=dr�Z>e?j�ds��ZAej
j
dtidudvig�dw��ZBdx�ZCdy�ZDdz�ZEd{�ZFd|�ZGd}�ZHy~cc}}}}w)�TestDataFrameSetItemc	���Gd�dt�}dg}t|�}tddgi|��}d|d<d	||d
�<tddgddg|d
�d	gi|��}tj||�y)Nc��eZdZy)�@TestDataFrameSetItem.test_setitem_str_subclass.<locals>.mystringN)�__name__�
__module__�__qualname__���JC:\Python312\Lib\site-packages\pandas/tests/frame/indexing/test_setitem.py�mystringr -s��r%r'z2020-10-22 01:21:00+00:00�a���index��b��c)�strrr�tm�assert_equal)�selfr'�datar+�df�expecteds      r&�test_setitem_str_subclassz.TestDataFrameSetItem.test_setitem_str_subclass+s|��	�s�	�,�,���d�#��
��a�S�z��
/����3����8�C�=���c�A�3��a�S�(�3�-�!��E�U�S��
����H�%r%�dtype)�int32�int64�uint32�uint64�float32�float64c���tjjd�jddt	|��}tj
||��||<||jj|k(sJ�y)Nr,r)�
�r8)�np�random�default_rng�integers�len�arrayr8�name)r3r8�float_frame�arrs    r&�test_setitem_dtypez'TestDataFrameSetItem.test_setitem_dtype8s`��
�i�i�#�#�A�&�/�/��2�s�;�7G�H���X�X�c��7��E���5�!�'�'�,�,��5�5�5r%c���tjjd�jt	|�df�}||ddg<tj|ddgj|�y)Nr,�A�B)rBrCrD�standard_normalrFr1�assert_almost_equal�values)r3rIr4s   r&�test_setitem_list_not_dataframez4TestDataFrameSetItem.test_setitem_list_not_dataframeBsX���y�y�$�$�Q�'�7�7��[�9I�1�8M�N��"&��S�#�J��
���{�C��:�6�=�=�t�Dr%c��tgd�gd�d�tgd�d����}tgd�tgd	�d��d
��}d}tjt
|�
�5||d<ddd�ttjjd�jddd�gd���}d}tjt
|�
�5|jddg�j�|d<ddd�d}tjt
|�
�5t�|d<ddd�y#1swY��xYw#1swY�IxYw#1swYyxYw)N�r)r,r.)�d�e�f)�bar�baz�r(r-r/�foo�rHr*)�g�h�i�j)r(r-r/r(�fiz�r+rHz/cannot reindex on an axis with duplicate labels��match�newcolr,r)�rf�r(r-r/rU��columnszDCannot set a DataFrame with multiple columns to the single column grr-r/�grz7Cannot set a DataFrame without columns to the column gr)rr
r�pytest�raises�
ValueErrorrBrCrDrE�groupby�count)r3r5�ser�msgs    r&�test_setitem_error_msmgsz-TestDataFrameSetItem.test_setitem_error_msmgsGs$��
��o�6���e�4�
��� ��,�5�9��
��
@��
�]�]�:�S�
1�	��B�x�L�	���I�I�!�!�!�$�-�-�a��F�;�(�
��
U��
�]�]�:�S�
1�	6��z�z�3��*�-�3�3�5�B�t�H�	6�H��
�]�]�:�S�
1�	#� �{�B�t�H�	#�	#�	�	��	6�	6��
	#�	#�s$�D%�%D1�D=�%D.�1D:�=Ec�\�d}d}tt|���}tjj	d�j|�}t|�D]}|||<�	ttj||�j||�t|���}tj||�y)Nr@�r*r,)
r�rangerBrCrDrO�repeat�reshaper1�assert_frame_equal)r3�N�Kr5�new_colr_r6s       r&�test_setitem_benchmarkz+TestDataFrameSetItem.test_setitem_benchmarkes�����
��
�U�1�X�
&���)�)�'�'��*�:�:�1�=���q��	�A��B�q�E�	��R�Y�Y�w��2�:�:�1�a�@��a��Q��
���b�(�+r%c��ttjjd�j	d�tj
d�gd���}|j
dd|d�|j
dd	|d
�|djd�|d<|j}ttjd
�gdztjd�gzgd���}tj||�|djd�|d<|j}ttjd
�gdztjd�gdzzgd���}tj||�|djd�|d<|j}ttjd
�gdztjd�gdzztjd�gzgd���}tj||�y)Nr,�rtr.rt)r/r-r(�r+rirr[r(rXr/r=�xr>)r[r/rXr-r(r�r*rfr9�y)r[r/rXr-r(r�r�)
rrBrCrDrO�arange�insert�astype�dtypesrr8r1�assert_series_equal�r3r5�resultr6s    r&�test_setitem_different_dtypez1TestDataFrameSetItem.test_setitem_different_dtypeps���
��I�I�!�!�!�$�4�4�V�<��)�)�A�,�#�
��
	�	�	�!�U�B�s�G�$�
�	�	�!�U�B�s�G�$�
�S�'�.�.��+��3�������
�X�X�i�
 �!�A�%����)�)<�(=�=�4�
��	���v�x�0��S�'�.�.��+��3�������
�X�X�i�
 �!�A�%����)�)<�(=��(A�A�4�
��	���v�x�0��S�'�.�.��)��3�������
�X�X�i�
 �!�A�%����)�)<�(=��(A�A�R�X�X�g�EV�DW�W�9�
��	���v�x�0r%c��tgd���}|j|d<gd�|d<tdgd�igd���}tj||�y)N�rMrN�Cr*�X�r�r��z)r4r+)rr+r1rx�r3r5�exps   r&�test_setitem_empty_columnsz/TestDataFrameSetItem.test_setitem_empty_columns�sC��
�_�
-���(�(��3��!��3���c�?�3�?�K��
���b�#�&r%c���tddd��}ttjt	|����}||d<|dj
tj
d�k(sJ�y)Nz1/1/2000 00:00:00z1/1/2000 1:59:50�10s)�freqr*rM�M8[ns])rrrBr�rFr8)r3�rngr5s   r&�%test_setitem_dt64_index_empty_columnsz:TestDataFrameSetItem.test_setitem_dt64_index_empty_columns�sQ���,�.@�u�M��
�R�Y�Y�s�3�x�0�
1����3���#�w�}�}����� 2�2�2�2r%c���ttd���}tdd��jd�|d<ttdd��ggdzgd�dg�	�}t	j
||�y)
Nr.r*�20130101�UTC��tz�ns�now�rr)r,r)rrur�as_unitr1rx�r3r5r6s   r&�$test_setitem_timestamp_empty_columnsz9TestDataFrameSetItem.test_setitem_timestamp_empty_columns�sb��
�U�1�X�
&���j�U�3�;�;�D�A��5�	���
�u�-�
.�/�!�3�9�u�g�
��	���b�(�+r%c��tjgd�gd��}ttd�dg��}dt	|��dt	|��d�}tjt|�	�5||d
<ddd�y#1swYyxYw)N)rr)r)rr)r,rZr@rXrhzLength of values \(z$\) does not match length of index \(z\)rcr[)r
�
from_codesrrurFrkrlrm)r3�catr5rqs    r&�2test_setitem_wrong_length_categorical_dtype_raiseszGTestDataFrameSetItem.test_setitem_wrong_length_categorical_dtype_raises�s|���$�$�%7��I��
�u�R�y�5�'�
2��#�3�s�8�*�-1�14�R��	��
=�	��]�]�:�S�
1�	��B�u�I�	�	�	�s�(A7�7Bc��tgd�gd�d��}tgd��}||d<t|d��}tj|d|�y)NrZ���?�@�@��c_1�n_1�rrr)�
new_columnr\)rrrr1r�)r3r5�sp_arrayr6s    r&�test_setitem_with_sparse_valuez3TestDataFrameSetItem.test_setitem_with_sparse_value�sF��
���G�
H���y�)��#��<���(��6��
���r�,�/��:r%c���tgd�gd�d��}ttgd��gd���}||d<ttgd��d�	�}tj|d|�y)
NrZr�r�r�)r,r)rr*r�)r)rrr\)rrrr1r�)r3r5�	sp_seriesr6s    r&�(test_setitem_with_unaligned_sparse_valuez=TestDataFrameSetItem.test_setitem_with_unaligned_sparse_value�sR��
���G�
H���;�y�1��C�	�$��<���+�i�0�|�D��
���r�,�/��:r%c��tdd�g}tg�}||d<td|i�}tj||�y)Nz2003-12�Dr()rrr1rx)r3r4r�r6s    r&�#test_setitem_period_preserves_dtypez8TestDataFrameSetItem.test_setitem_period_preserves_dtype�sA���y�#�&�'���2�����s���c�4�[�)��
���f�h�/r%c	��ttgd�d��tgd�t��tgd�t��tgd�d��d��}ttgd��tgt��tgt��tgd��d��}tgd��D]T\}}t	|�t|�t|�tj|�d�|j|jd<�Vtj||�y)Nr�r:rArTr;rgr)rr�float�	enumerate�intrBr;�loc�shaper1rx)r3r6r5�idxr-s     r&�"test_setitem_dict_preserves_dtypesz7TestDataFrameSetItem.test_setitem_dict_preserves_dtypes�s�����I�W�5��I�U�3��I�U�3��I�X�6�	
�
����B�g�.��B�e�,��B�e�,��B�h�/�	
�
�� �	�*�	�F�C����X��1�X��1�X��Y�Y�q�\�	#�B�F�F�2�8�8�A�;��	�	���b�(�+r%z	obj,dtypez2020-01�Mrrt)�left�rightr:r�z
2011-01-01z
US/Easternr��s)�unitr�c��tgd�t|gdz|��d��}tdgd�i�}||d<tj||�y)NrTr.rA)r��objr�r��rrr1rx)r3r�r8r6r5s     r&�test_setitem_extension_typesz1TestDataFrameSetItem.test_setitem_extension_types�sH���Y�v�s�e�a�i�u�7U�V�W��
��y�)�
*����5�	�
���b�(�+r%�ea_namezdatetime64[ns, UTC]�	period[D]c�v�tdg�}dg||<tddg|dgi�}tj||�y)Nrr)�rr1rx)r3r�r�r6s    r&�test_setitem_with_ea_namez.TestDataFrameSetItem.test_setitem_with_ea_namesA���A�3����#��w���a�!��g��s�3�4��
���f�h�/r%c��tjddgd��}t|�j�}||d<t	ddgddgd�d��}tj||�tjddgd��}||d<tj|d|d�tj|dj�|�y)	Nr)�nat�datetime64[ns]rA�new)rr��
datetime64[s]r)
rBrGr�to_framerr1rxr��assert_numpy_array_equal�to_numpy)r3�data_nsr�r6�data_ss     r&�6test_setitem_dt64_ndarray_with_NaT_and_diff_time_unitszKTestDataFrameSetItem.test_setitem_dt64_ndarray_with_NaT_and_diff_time_unitss����(�(�A�u�:�-=�>�����)�)�+����u�
��!�T��A�t�9�=�EU�V��
���f�h�/����1�e�*�O�<����u�
�
���v�a�y�(�1�+�6�
�#�#�F�5�M�$:�$:�$<�f�Er%r�)r^�mr��msr�r��Yc���d}tjd|�d��}tj|tj��j	|�}|dvr|}n|jd�}t
dtj|�itj|���}|||<||j|jk(sJ�||j|k(j�sJ�y)	N�d�M8[�]rA)r�r�r��intsr*�	rBr8r�r:�viewr�rrQ�all�r3r��nr8�vals�ex_valsr5s       r&�-test_frame_setitem_datetime64_col_other_unitszBTestDataFrameSetItem.test_frame_setitem_datetime64_col_other_units#s���
�����3�t�f�A��'���y�y��"�(�(�+�0�0��7���;���G��k�k�/�2�G�
���	�	�!��-�R�Y�Y�q�\�
B����4���$�x�~�~����.�.�.��4����7�*�/�/�1�1�1r%c���d}tjd|�d��}tj|tj��j	|�}|jd�}t
dtj|�itj|���}tj|tj��j	d�|d	<||d	<|d	j|k(j�sJ�y)
Nr�r�r�rAr�r�r*r��datesr�r�s       r&�6test_frame_setitem_existing_datetime64_col_other_unitszKTestDataFrameSetItem.test_frame_setitem_existing_datetime64_col_other_units8s���
�����3�t�f�A��'���y�y��"�(�(�+�0�0��7���+�+�.�/��
���	�	�!��-�R�Y�Y�q�\�
B���i�i�����2�7�7��A��7����7���7��"�"�g�-�2�2�4�4�4r%c�Z�|}|djd�}||d<tj|dt|d���d|d<||d<tj|dt|d���|d=|jj
d}|jj
d}tj||�|jj}|jj}|s|�2t|�t|�k7sJ�t|�t|�k(sJ�|j�}	t|	jd<t|	jd	<|	d}
tjt|
�tgd
�d���tj|	j|j�y)NrNr[r�r\r�r)r,)r)r)�r)r,)TFT)�renamer1r�r�_mgr�arrays�assert_extension_array_equal�_ndarray�base�id�copyr�ilocrr�)r3�timezone_frame�using_copy_on_writer5r��v1�v2�v1base�v2base�df2r�s           r&�test_setitem_dt64tzz(TestDataFrameSetItem.test_setitem_dt64tzIs^��
����g�n�n�U�#����3��
���r�#�w��s��(=�>���3����3��
���r�#�w��s��(=�>��s�G�
�W�W�^�^�A�
��
�W�W�^�^�A�
��
�'�'��B�/����!�!�����!�!��"��>�b��j�B�v�J�&>�?�?��f�:��F��+�+�+��g�g�i���������������S���
���u�V�}�f�5H�s�.S�T�
���s�z�z�2�9�9�5r%c���tddd��}ttjj	d�jd�|��}||d<t
|d�}tj||d	�
�|jdk(sJ�|jdk(sJ�|j�jd�}t|jt�sJ�tj|j|�y)Nz1/1/2000rtr+)�periodsrHr,r~r*r
F��check_names)rrrBrCrDrOr
r1�assert_index_equalrH�reset_index�	set_index�
isinstancer+r)r3r�r5�rss    r&�test_setitem_periodindexz-TestDataFrameSetItem.test_setitem_periodindexks����:�q�w�?��
�r�y�y�,�,�Q�/�?�?��G�s�
S����7��
�2�g�;�
��
���b�#�5�9��w�w�'�!�!�!��x�x�7�"�"�"�
�^�^�
�
'�
'��
0���"�(�(�K�0�0�0�
���b�h�h��,r%c���tgd�gd�d��}tjddgddgddgg�}||ddg<tgd�gd�gd	�gd
�d��}|dj|j�|d<|dj|j�|d<|dj|jk(sJ�|dj|jk(sJ�tj||�y)N)�one�two�threerT�r(r-r)r.rtr/rU�r)r.rt�r)r)r)rg)rrBrGr�r8r1rx�r3r5rJr6s    r&�'test_setitem_complete_column_with_arrayz<TestDataFrameSetItem.test_setitem_complete_column_with_arrayys���
�4�9�E�
F���h�h��A���A���A��/�0����C��:���,����	
�
��!��
�,�,�S�Y�Y�7���
� ��
�,�,�S�Y�Y�7���
���}�"�"�c�i�i�/�/�/���}�"�"�c�i�i�/�/�/�
���b�(�+r%c��tdddd��}t|�}tdgd�id��}|j|jddt	j
||�y)	Nz
2016-01-01�	r�rM)rr�rH)	rrrrrrrrrr�rA)rr�	_na_valuer�r1rx)r3r�r�r6s    r&�test_setitem_period_d_dtypez0TestDataFrameSetItem.test_setitem_period_d_dtype�sT���<���3�G���3����
�Q�R��
��������A��
���f�h�/r%)�f8�i8�u8c�8�tgd�|��}ttjj	d�jd�|��}gd�|d<tgd�t��}|d	k(rtgd
�t��}tj|j|�y)NrTrAr,)r.r.rhrZF)r)r,r.Fr)r�r�r�F)
r
rrBrCrDrO�objectr1rri)r3r8�colsr5�
expected_colss     r&�$test_setitem_bool_with_numeric_indexz9TestDataFrameSetItem.test_setitem_bool_with_numeric_index�su���Y�e�,��
�r�y�y�,�,�Q�/�?�?��G�QU�
V��#��5�	��.�f�=�
��D�=�!�"8��G�M�
���b�j�j�-�8r%�indexerrNc���tddg��}tdddgi�}|||<ttjgdzddgd��}|djd�|d<t	j
||�y)NrMrNrhr)r,�rMrNr)rrB�nanr�r1rx)r3r r5�otherr6s     r&�#test_setitem_frame_length_0_str_keyz8TestDataFrameSetItem.test_setitem_frame_length_0_str_key�sr����S�z�
*���3��A��-�(����7���B�F�F�8�a�<�q�!�f�=�>�� ��
�,�,�X�6���
�
���b�(�+r%c
��gd�dz}ttd�|��}d|jd<d|jdd�df<d	|d
<tgd�tjddtjd
dgtjddtjd
dggd��}|djtj�|d<|djtj�|d<||_tj||�y)Nr�r,r.r)rr.)rrM)r)rfrN�r,rtr�)rr)r,r.rfrtr)rfrtrrA)
rrur�rBr#r�r:rir1rx)r3rr5r6s    r&�$test_setitem_frame_duplicate_columnsz9TestDataFrameSetItem.test_setitem_frame_duplicate_columns�s�����"��
�U�1�X�t�
4������v������q�#�v����3���"�����A�r�v�v�q�!�,�����A�r�v�v�q�!�,�
�
�

���q�k�(�(����2�����q�k�(�(����2�������
���b�(�+r%c�@�gd�dz}ttd�|��}tjtd��5d|dg<ddd�|j
dd�dd�f}tjtd��5d|dg<ddd�y#1swY�IxYw#1swYyxYw)	Nr�r,r.r�"Columns must be same length as keyrc)rr.rtrM)rrurkrlrmr�)r3rr5r�s    r&�2test_setitem_frame_duplicate_columns_size_mismatchzGTestDataFrameSetItem.test_setitem_frame_duplicate_columns_size_mismatch�s�����"��
�U�1�X�t�
4��
�]�]�:�-Q�
R�	"�!�B��u�I�	"��g�g�a��!��e�n��
�]�]�:�-Q�
R�	#�"�C���J�	#�	#�		"�	"��	#�	#�s�B�8B�B�BrrZ)r(r(r(c��tgd�g|��}tddggddg��}d}tjt|��5||d	<ddd�y#1swYyxYw)
NrTrhr@�rUrVr*rcr(�rrkrlrm)r3rr5�rhsrqs     r&�#test_setitem_df_wrong_column_numberz8TestDataFrameSetItem.test_setitem_df_wrong_column_number�sY���	�{�D�
1���"�b��
�S�#�J�7��2��
�]�]�:�S�
1�	��B�s�G�	�	�	�s�A�Ac��tgd�ggd���}tgd�ggd���}||ddg<tgd�ggd���}tj||�||ddg<tgd�ggd	���}tj||�y)
NrT�r(r-r-rh�r@r-�r(r-r/)r@r-r4r@)r(r-r-r/r��r3r5r/r6s    r&�/test_setitem_listlike_indexer_duplicate_columnszDTestDataFrameSetItem.test_setitem_listlike_indexer_duplicate_columns�sw��
�	�{�O�
<������@����C��:���l�^�_�E��
���b�(�+���C��:���.�/�9M�N��
���b�(�+r%c��tgd�ggd���}tddggddg��}d}tjt|�	�5||ddg<ddd�y#1swYyxYw)
NrTr2rhr@r-r(r-r*rcr.)r3r5r/rqs    r&�@test_setitem_listlike_indexer_duplicate_columns_not_equal_lengthzUTestDataFrameSetItem.test_setitem_listlike_indexer_duplicate_columns_not_equal_length�s[��
�	�{�O�
<���"�b��
�S�#�J�7��2��
�]�]�:�S�
1�	!� �B��S�z�N�	!�	!�	!�s�A�Ac���tdtd�i�}t|dd�}t|jj
t�sJ�||d<tj|�|d<|j|d<tj|j�|d<|jt�|d<t|djt�sJ�t|djj
jt�sJ�t|djt�sJ�t|djj
jt�sJ�t|djt�sJ�t|djt�sJ�t|d�sJ�d	�}t!j"||j$�||j$��t!j"||j$�||j&�d
��t!j"||j$�||j(�d
��t!j"||j&�||j(�d
��t!j*|d|d�t!j*|d|dd
��t!j*|d|d�t!j*|d|dd
��y)NrMr@rtrNr�r��E�Fc�>�ttj|��S�N)r
rBrG�r�s r&�<lambda>z=TestDataFrameSetItem.test_setitem_intervals.<locals>.<lambda>s��e�B�H�H�Q�K�(�r%Fr)rrurrr��
categoriesrrBrGrQr�rr8rrrr1rrNr�r�r�)r3r5rpr/s    r&�test_setitem_intervalsz+TestDataFrameSetItem.test_setitem_intervals�sD��
��U�2�Y�'�
(���"�S�'�1�o���#�'�'�,�,�m�<�<�<�
��3���(�(�3�-��3���*�*��3���(�(�3�:�:�&��3���*�*�V�$��3���"�S�'�-�-�)9�:�:�:��"�S�'�+�+�0�0�6�6�
�F�F�F��"�S�'�-�-�)9�:�:�:��"�S�'�+�+�0�0�6�6�
�F�F�F��"�S�'�-�-��7�7�7��"�S�'�-�-��7�7�7��r�#�w�'�'�'�
)��
���a����g�q����w�/�
���a����g�q����w�E�B�
���a����g�q����w�E�B�
���a����g�q����w�E�B�	���r�#�w��3��0�
���r�#�w��3��U�C�	���r�#�w��3��0�
���r�#�w��3��U�Cr%c�6�tdttd��jd�i�}|jj
j
ddg�|_tdtddg�j
ddg�i�}tj||�y)Nr^�mn�categoryr�r�)
rr�listr�r^r��reorder_categoriesr
r1rxr�s   r&�test_setitem_categoricalz-TestDataFrameSetItem.test_setitem_categoricals��
��V�D��J�/�6�6�z�B�C�
D���t�t�x�x�*�*�C��:�6����
�+�s�C�j�)�<�<�c�3�Z�H�I�
��	���b�(�+r%c��tgd��}tdg|��}g|d<tdg|��}tj|j|j�y)Nr�r\rM�rir+)r
rr1rr+)r3r+r�r6s    r&� test_setitem_with_empty_listlikez5TestDataFrameSetItem.test_setitem_with_empty_listlike#sK���b�u�%���C�5��6����s���c�U�%�8��
���f�l�l�H�N�N�;r%zcols, values, expected)r�r�r�r(�r)r,r.rfrf)r�r�r�r()r�rNrNr()r�rNr(rTr.)rNr�r()r.r,r)r))r�r(rNr,c�b�t|g|��}|d|d<|djd}||k(sJ�y)Nrhr(r)rrQ)r3rrQr6r5r�s      r&�test_setitem_same_columnz-TestDataFrameSetItem.test_setitem_same_column+s?����x��
.���S�'��3���C�����"����!�!�!r%c��gd�ddggd�f}tj|�}tdd��}tjjd�j
d	d
t|�t|�f�}t|||��}|jjj�|ddd}}tjjd�j|�|d
j|ddd�f|d<tj|d|d
�tjjd�j|�|dj||f|d<tj|d|d�tjjd�j|�|dj||f|d<tj|d|d�y)N)�jim�joe�jolie�first�last)r��centerr��20141006�)rr,r)i�rI���rQrO)rQrS)rPrR)rQrR)rPrS)r�from_productrrBrCrDrErFrr+rQr��shuffler�r1rx)r3�itrr+r�r5r_r`s        r&�test_setitem_multi_indexz-TestDataFrameSetItem.test_setitem_multi_index=s���%�w��&7�9T�
T���&�&�r�*���:�r�2���y�y�$�$�Q�'�0�0��D�3�u�:�s�4�y�:Q�R��
�t�T��
7���x�x���#�#�%�r�"�v�a�y�1��
�	�	���a� �(�(��+��w�K�O�O�A�t��t�G�,��5�	�
���b��i��G��5�
�	�	���a� �(�(��+�!�"3�4�8�8��A��>����
���b�!1�2�B�7H�4I�J�
�	�	���a� �(�(��+� �!3�4�8�8��A��>��?��
���b��1�2�6H�3I�Jr%zcolumns,box,expected)rMrNr�r��)r\r\r\r\rhr�r��)r)r,r\r])r.rfr\r]�rt�r\r]r�)r\r]rrA)rNr�r�r3)�
��)r)r\r]r)r.r@r-r4)rtr`rarb)r�rMr�)r]r,r\r)r-rfr@r4)rar_r`rbrMrr@r-r4)r\r,r])rrfr@)r-r_r4c�j�tddgddgddggddg�	�}|||<tj||�y)
Nr)r,r.rfrtr_rMrNrhr�)r3ri�boxr6r5s     r&�!test_setitem_list_missing_columnsz6TestDataFrameSetItem.test_setitem_list_missing_columnsUsA��f��A���A���A��/�#�s��
D����7��
���b�(�+r%c��tt|d|d��}||d<|d}t||jd��}t	j
||�y)NrMrN�tuplesrb)rE�ziprr+r1r�)r3rIrgr�r6s     r&�test_setitem_list_of_tuplesz0TestDataFrameSetItem.test_setitem_list_of_tuples�sU���c�+�c�*�K��,<�=�>�� &��H���X�&���&��(9�(9��I��
���v�x�0r%c��tgd�gd�d��}d�dD�}d|j|<tgd�gd�d��}tj||�y)	NrT�rfrtr_rc3� K�|]}|���y�wr=r$��.0r�s  r&�	<genexpr>zCTestDataFrameSetItem.test_setitem_iloc_generator.<locals>.<genexpr>������%��1�%���r�r)r�rfr)r)�rr�r1rx�r3r5r r6s    r&�test_setitem_iloc_generatorz0TestDataFrameSetItem.test_setitem_iloc_generator�sH��
�Y�Y�7�
8��%�f�%���������9�9�=�>��
���b�(�+r%c��tgd�gd�d��}d�dD�}d|j|df<tgd�gd�d��}tj||�y)NrTrkrc3� K�|]}|���y�wr=r$rms  r&rozSTestDataFrameSetItem.test_setitem_iloc_two_dimensional_generator.<locals>.<genexpr>�rprqr�r)rrrsrts    r&�+test_setitem_iloc_two_dimensional_generatorz@TestDataFrameSetItem.test_setitem_iloc_two_dimensional_generator�sL��
�Y�Y�7�
8��%�f�%��������
���9�9�=�>��
���b�(�+r%c��tdd��}t|��}td|tj��|d<td|d��|d<td	|d��|d	<td
|tj��|d
<|j
}ttjtttjgtd���}tj||�y)Nr��S24)rHr8r*r()rHr+r8r-�S64r/rU�abcd)
rrrBr;�uint8r�rrEr1r�)r3r+r5r�r6s     r&�(test_setitem_dtypes_bytes_type_to_objectz=TestDataFrameSetItem.test_setitem_dtypes_bytes_type_to_object�s����D��.��
�U�
#���c��b�i�i�@��3���c��e�<��3���c��e�<��3���c��b�h�h�?��3�������2�9�9�f�f�b�h�h�?�t�F�|�T��
���v�x�0r%c�>�tddgddgd��jddd��}td|j�	�}|d
|j|d
f<|d|j|df<tddgddgd��jddd��}tj||�y)Nr.rfrtr_rr:�Int64Fr*r(r-)rr�rr+r�r1rx)r3r��maskr6s    r&� test_boolean_mask_nullable_int64z5TestDataFrameSetItem.test_boolean_mask_nullable_int64�s����!�Q��q�!�f�5�6�=�=���(�
���e�6�<�<�0�� &�s���
�
�4��9�� &�s���
�
�4��9���A�q�6��A��7�8�?�?���(�
��	���f�h�/r%c��tdddgi�}tddgd��|d<tdddgid��}tj||�y)Nr(r)r,r�rAr�r�s   r&� test_setitem_ea_dtype_rhs_seriesz5TestDataFrameSetItem.test_setitem_ea_dtype_rhs_series�sL��
��a��V�}�
%���!�Q��w�/��3���c�A�q�6�]�'�:��
���b�(�+r%c��ttjd�tjd�d�tjd���}tjd�}ttjd���}||d<t	j
t�5tj|�|d<ddd�t	j||�y#1swY� xYw)Nr@)�np-array�	np-matrixr*)r@r)r�r�)	rrB�onesr�r1�assert_produces_warning�PendingDeprecationWarning�matrixrx)r3r6r(r5s    r&�test_setitem_npmatrix_2dz-TestDataFrameSetItem.test_setitem_npmatrix_2d�s���������2�7�7�2�;�?�r�y�y�QS�}�
��
�G�G�G���
�R�Y�Y�r�]�
+����:���
'�
'�(A�
B�	+� �i�i��l�B�{�O�	+�	���b�(�+�	+�	+�s�C�Cr�rUr(c�0�tddgddgd�|��}ddd�|jdd�d	f<d
dd�|jdd�d
f<dd
i|jdd�df<tddgddgd�|�d
dgtjd
gd���}t	j
||�y)Nr)r,r.rfrr���)r)rr(rtr_�rr)r/rV)r/rV)rr�rBr#r1rx)r3r�r5r6s    r&�%test_setitem_aligning_dict_with_indexz:TestDataFrameSetItem.test_setitem_aligning_dict_with_index�s����a��V�1�a�&�9�D�9�
:�� �S�)����q�#�v���1�����q�#�v���Q�����q�#�v�����*�A�q�6�Q�T�Q��A��b�f�f�a�[�Q�
��	���b�(�+r%c�D�tdddgi�}tdddgiddg��|d<tdtjdgi�}tj||�tdddgi�}|jdtdddgiddg���tj||�y)Nr(r)r,r@r-r*r)rrBr#r1rx�isetitemr�s   r&�test_setitem_rhs_dataframez/TestDataFrameSetItem.test_setitem_rhs_dataframe�s���
��a��V�}�
%���S�2�r�(�O�A�q�6�:��3���c�B�F�F�B�<�0�1��
���b�(�+�
��a��V�}�
%��
���A�y�#��B�x���A��?�@�
���b�(�+r%c��tddgddgddgg��}tddd	gi|�
�|d<ttdd	g|�
�ddgd��}tj||�y)Nr(r-r)r,r.rf�rir4r@r-rArr��r3�any_numeric_ea_dtyper5r6s    r&�*test_setitem_frame_overwrite_with_ea_dtypez?TestDataFrameSetItem.test_setitem_frame_overwrite_with_ea_dtype�so��
��S�z�!�Q��!�Q��0@�
A���S�2�r�(�O�3G�H��3����R��H�,@�A���V�
�
��	���b�(�+r%c�<�tjd�tdddgi�}tjdd�5tddgt��|d<ddd�tddgtddgt��d	��}tj||�y#1swY�AxYw)
N�pyarrowr(r)r,zfuture.infer_stringTr-rAr)
rk�importorskipr�pd�option_contextr
rrr1rxr�s   r&�'test_setitem_string_option_object_indexz<TestDataFrameSetItem.test_setitem_string_option_object_index�s������I�&�
��a��V�}�
%��
�
�
�4�d�
;�	6��S�#�J�f�5�B�s�G�	6��A�q�6���S�z��0P�Q�R��
���b�(�+�	6�	6�s�B�Bc��tddgi�}|j�}d}||g||<tj||�y)Nrr@�rr�r1rx)r3r5r6�col_names    r&�test_setitem_frame_midx_columnsz4TestDataFrameSetItem.test_setitem_frame_midx_columnssB��
�
�R�D�)�
*���7�7�9�����8�*�~��8��
���b�(�+r%c��tdtjdgd��i�}tdgd��|jdd�df<tdtjdgd��i�}tj||�tdtjdgd��i�}tdgd��|jdd�df<tj||�y)Nr(r@rrAr-r�r)rrBrGrr�r1rxr�r�s   r&�test_loc_setitem_ea_dtypez.TestDataFrameSetItem.test_loc_setitem_ea_dtype
s���
��R�X�X�r�d�$�7�8�
9�����G�4����q�#�v���c�2�8�8�R�D��#=�>�?��
���b�(�+�
��R�X�X�r�d�$�7�8�
9����t�7�3�����1��
�
���b�(�+r%c	���ttd�gt��}tddgi�}t	j
td��5||jdd�df<ddd�t	j
td��5||d<ddd�tdgttd�gd	��ttd�gd	��d
��}t	j||�y#1swY��xYw#1swY�fxYw)Nz
2019-12-31rAr(r)�inferrcr-r/r�rZ)
r
rrrr1r��
FutureWarningr�rrx)r3r�r5r6s    r&�test_setitem_object_inferringz2TestDataFrameSetItem.test_setitem_object_inferrings����Y�|�,�-�V�<��
��a�S�z�
"��
�
'�
'�
�W�
E�	!� �B�F�F�1�c�6�N�	!�
�
'�
'�
�W�
E�	��B�s�G�	���S��Y�|�4�5�=M�N��Y�|�4�5�=M�N�
�
��	���b�(�+�	!�	!��	�	�s�C�<C'�C$�'C0N)Ir!r"r#r7rk�mark�parametrizerKrRrrr|r�r�r�r�r�r�r�r�r�rr	rrrrr��ea_registryr�rrH�propertyr�r�r�r�rr
rrrr%r(r+r0r6r8rArGrJrMr[rrBrGr:rerirurxr~r�r��td�&skip_array_manager_not_yet_implementedr�r�r�r�r�r�r�r�)rnr8rr�s0000r&rr*s���&��[�[����M��6��6�E�
#�<	,�#1�J'�3�,�
�;�;�0�,�6�[�[����
�I�
��C� 0�1�
�1�A�
&�
�g�w�(G�H��,�<�8��S�\�:�
�	
�
�,�
�,��[�[����%�+�+�	
�	
���e�j�j�(�3�
�J�J�	
�!�+�
.�
	/�	�0�	�0�F��[�[���V�%I�J�2�K�2�(�[�[���V�%I�J�5�K�5�  6�D-�,�&	0��[�[���W�&8�9�9�:�9��[�[���Y��s�e��5�,�6�,�,�.	#��[�[���V�o��%G�H��I��
,�!�*D�X,�<��[�[��� �
!�<��3�
!�<��3�
!�<��3�
�i��+�
�i��+�
�i��+�
	
�
�"�
�"�K�0�[�[����%���!�<��>�0��
��c�
��A���!�<��>�0��
� ������"�(�(�3��9�i��;�_�U�
� ��L�,�7��!�?�O�D�0��
� �����)�\�<�@����Q��!�?�O�D�0��
��c�
��A�q�6�A�r�7�R��H�5��S�z�J���
�K�8�/��
�M-	
�0�b,�c0�b,�1�,�,�
1�0�,��.�.�,�/�,�"�[�[���V�b�3��*�%5�6�	,�7�	,�	,�
,�,�,�	,�,��_	
s�
-N7rc�f�eZdZejd��Zejd��Zd�Zd�Zd�Z	y)�TestSetitemTZAwareValuesc�F�tddgd��}|jd�}|S)Nz2013-1-1 13:00z2013-1-2 14:00rNr\z
US/Pacific)r�tz_localize)r3�naiver�s   r&r�zTestSetitemTZAwareValues.idx)s*���/�1A�B��M������-���
r%c��ttj|j�d��d��}|j|jk(sJ�|S)NrrArNr\)rrBrG�tolistr8)r3r�r6s   r&r6z!TestSetitemTZAwareValues.expected/s:���"�(�(�3�:�:�<�x�@�s�K���~�~����*�*�*��r%c�n�ttjjd�j	d�dg��}||d<|jddg��jjd�|d<|d}t|jd	�jd�d�
�}tj||�y)Nr,�r,r)rMrhrNrr)r*r�r\)rrBrCrDrO�	to_series�dt�
tz_convertrr�r1r�)r3r�r6r5r��comps      r&�test_setitem_dt64seriesz0TestSetitemTZAwareValues.test_setitem_dt64series5s���
�r�y�y�,�,�Q�/�?�?��G�RU�QV�
W����3���-�-�q�!�f�-�-�0�0�;�;�D�A��3���C����c�n�n�U�+�7�7��=�C�H��
���v�t�,r%c��ttjjd�j	d�dg��}||d<|d}tj||�y�Nr,r�rMrhrN)rrBrCrDrOr1r��r3r�r6r5r�s     r&�test_setitem_datetimeindexz3TestSetitemTZAwareValues.test_setitem_datetimeindex?sP��
�r�y�y�,�,�Q�/�?�?��G�RU�QV�
W����3���C���
���v�x�0r%c���ttjjd�j	d�dg��}|j�|d<|d}t
j||�yr�)rrBrCrDrO�
to_pydatetimer1r�r�s     r&�.test_setitem_object_array_of_tzaware_datetimeszGTestSetitemTZAwareValues.test_setitem_object_array_of_tzaware_datetimesHsY��
�r�y�y�,�,�Q�/�?�?��G�RU�QV�
W���#�#�%��3���C���
���v�x�0r%N)
r!r"r#rk�fixturer�r6r�r�r�r$r%r&r�r�(s?���^�^����
�^�^����
-�1�1r%r�c�H�eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
d	�Zd
�Zy)�!TestDataFrameSetItemWithExpansionc�X�tgd�gd�d��}|d}tjddgddgdd	gg�|d
dg<tj|�5d|j
d
<ddd�|rt
gd�d��}nt
gd�d��}tj||�y#1swY�AxYw)NrT)rfrfr_rr(g�������?g�������?g333333�?g�������?��?r/rUr�)rrr\)r�r,r.)rrBrGr1�assert_cow_warningr�rr�)r3r��warn_copy_on_writer5rpr6s      r&�test_setitem_listlike_viewsz=TestDataFrameSetItemWithExpansion.test_setitem_listlike_viewsSs���
�Y�Y�7�
8����g�����C��:��S�z�C��:�"F�G��C��:���
"�
"�#5�
6�	 ��B�G�G�D�M�	 ���i�c�2�H��k��4�H�
���s�H�-�	 �	 �s�B � B)c��tddgddgg�}ddg|d<tgd�gd	�ggd
���}tj||�y)Nr)r,r.rfrtr_�0 - Name)r)r,rt)r.rfr_)rr)r�rhr�r�s   r&�.test_setitem_string_column_numpy_dtype_raisingzPTestDataFrameSetItemWithExpansion.test_setitem_string_column_numpy_dtype_raisinggsH��
��A���A��'�
(���Q���:���i��3�=O�P��
���b�(�+r%c�6�tgd�d��}ttd��|jdd�df<tdtj
tj
gdtj
tj
gggd���}t
j||�y)	Nr2r>�rir8r,r(rr)rh)rrErur�rBr#r1rx)r3r�r5r6s    r&�'test_setitem_empty_df_duplicate_columnszITestDataFrameSetItemWithExpansion.test_setitem_empty_df_duplicate_columnsnsk��
��i�
@���e�A�h�����q�#�v����������
 �1�b�f�f�b�f�f�"5�6��
��	���b�(�+r%c
���tdtjtjj	d�jddd�d��i�}t
tddd�D�cgc]}|�d	|d
z����c}�}|jdgd��}t|jtdd
d�d|��}|j}||d<|j}ttjd�t|d��gddg��}t!j"||�||d<|j}ttjd�t|d��t|d��ggd���}t!j"||�|d}|d}	t!j$|j&j|�d|_t!j"|	j+�|j+��ycc}w)N�valuer,ri'r�r9rAi�z - i�T)�by�	ascendingi)F)r��labelsr�)r@�orderedr*r:)r�r�r:)rrBrGrCrDrEr
ru�sort_valuesrr�rQr�rr8rr1r��assert_categorical_equalr�rH�
sort_index)
r3r5r_r�rpr�r�r6�result1�result2s
          r&�-test_setitem_with_expansion_categorical_dtypezOTestDataFrameSetItemWithExpansion.test_setitem_with_expansion_categorical_dtypews���
������I�I�)�)�!�,�5�5�a���D�G��
�
���%��5�#�:N�O�Q���3�q�3�w�i�0�O�P��
�^�^��y�D�^�
9���"�(�(�E�!�U�C�0��f�M���j�j����3�������
�X�X�g�
� 0�F�E� R�S��C�.�
��	���v�x�0���3�����������!� �F�E�B� �F�E�B�
�
&�

��	���v�x�0��S�'���S�'��
�#�#�G�L�L�$6�$6��<����
���w�1�1�3�S�^�^�5E�F��EPs�%G'c��t�}d|d<tdg��jtj�}t	j
||�y)Nr)r[rh)rr�rBr:r1rxr�s   r&�test_setitem_scalars_no_indexz?TestDataFrameSetItemWithExpansion.test_setitem_scalars_no_index�s=��
�[����5�	��e�W�-�4�4�R�X�X�>��
���b�(�+r%c��d|jvsJ�|d|d<d|jvsJ�|d}|d}tj||d��y)Nr"rMFr)rir1r�)r3rIr�r6s    r&�test_setitem_newcol_tuple_keyz?TestDataFrameSetItemWithExpansion.test_setitem_newcol_tuple_key�sl��
��$�$�%�	%�%�!,�C� 0��H���[�0�0�0�0�0��X�&���s�#��
���v�x�U�Cr%c��tddt���}t|td���}t	ddd�}t|�}tj||<t
j||�j�sJ�y)	Nz1/1/2012z2/1/2012)�start�endr�r@rIi�r-r))
rrrrurrrBr#�isnanr�)r3rir4�t�tss     r&�#test_frame_setitem_newcol_timestampzETestDataFrameSetItemWithExpansion.test_frame_setitem_newcol_timestamp�se���:�:�D�F�K�����b�	�:���T�2�q�!��
�q�\���6�6��R���x�x��R��!�%�%�'�'�'r%c���tdddgi�}|j|d<d|jddgdf<|jdgdd�f}tdgdgd�dg��}tj||�y)	Nr(r-r�FTr)rr*)rr+r�r1rxr�s    r&�*test_frame_setitem_rangeindex_into_new_colzLTestDataFrameSetItemWithExpansion.test_frame_setitem_rangeindex_into_new_col�ss��
��c�3�Z�(�
)���(�(��3��%(�����t�}�c�!�"������Q�����C�5��u�5�a�S�A��
���f�h�/r%c���tddgddgddgg��}tddd	gi|�
�|d<tddgddgtdd	g|�
�d��}tj||�y)
Nr(r-r)r,r.rfr�r@r-rAr/rZr�r�s    r&� test_setitem_frame_keep_ea_dtypezBTestDataFrameSetItemWithExpansion.test_setitem_frame_keep_ea_dtype�sv��
��S�z�!�Q��!�Q��0@�
A���S�2�r�(�O�3G�H��3�����V���V��R��H�,@�A�
�
��	���b�(�+r%c�\�ttd���}tjdd��tjdd��dd�|jd	<ttj
d
�tj
d�dggt
d	g�gd���}tj||�y)
N�abcrhrtr�)r��H�23rZrz0 days 00:00:05z0 days 00:01:12r)	rrEr��to_timedeltar��	Timedeltar
r1rx)r3r�r6s   r&�&test_loc_expansion_with_timedelta_typezHTestDataFrameSetItemWithExpansion.test_loc_expansion_with_timedelta_type�s����4��;�/�������-�����#�.��
��
�
�1�
�
��l�l�,�-�r�|�|�<M�/N�PT�
U�V����*�$�
��
	���f�h�/r%N)
r!r"r#r�r�r�r�r�r�r�r�r�r�r$r%r&r�r�Rs7��.�(,�,�+G�Z,�
D�(�0�,�0r%r�c��eZdZd�Zej
j
dejejg�ej
j
de
eje
ejg�ej
j
dgd��d����Zej
j
de
eje
ejg�ej
j
dgd��d���Zej
j
dejejg�ej
j
de
eje
ejg�ej
j
dgd��d����Zy	)
�TestDataFrameSetItemSlicingc��ttjd��}d|ddtjd�}d|ddt|�}tj||�y)N)r�r)r)���)rrB�zerosr1rxrs    r&�test_setitem_slice_positionz7TestDataFrameSetItemSlicing.test_setitem_slice_position�sR��
�r�x�x��)�
*����2�3���h�h�x� ����B�C���S�>��
���b�(�+r%r rdr�rTc���tgd�ggd�g|zzgd���}|gd��||�ddtgd�ggd�g|zzgd���}tj||�y�Nr�r,rfr_rZrhr3r)r��r3r�rdr r5r6s      r&�+test_setitem_slice_indexer_broadcasting_rhszGTestDataFrameSetItemSlicing.test_setitem_slice_indexer_broadcasting_rhs�s[��
�	�{�i�[�1�_�4�o�
N���l�+����A�B���i�[�L�>�A�+=�=��W��
���b�(�+r%c	��tgd�ggd�g|zzgd���}|gd��|jttd|dz��<tgd�ggd�g|zzgd���}t	j
||�yr�)rr�rErur1rx)r3r�rdr5r6s     r&�*test_setitem_list_indexer_broadcasting_rhszFTestDataFrameSetItemSlicing.test_setitem_list_indexer_broadcasting_rhs�si���	�{�i�[�1�_�4�o�
N��),�\�):�����U�1�a�!�e�_�%�&��i�[�L�>�A�+=�=��W��
���b�(�+r%c���tgd�gd�ggd�g|zzgd���}|gd��||�ddtgd�ggd�g|dzzzgd�d�	�}tj||�y)
Nrr�r�rZrhr3r)rr�r�r�s      r&�0test_setitem_slice_broadcasting_rhs_mixed_dtypeszLTestDataFrameSetItemSlicing.test_setitem_slice_broadcasting_rhs_mixed_dtypessp��
�
��(�I�;��?�:�O�
���l�+����A�B���
�K�<�.�A��E�2�2�#��
��
	���b�(�+r%N)r!r"r#r�rkr�r�r1�setitemr�rrBrGrEr�r�rrr$r%r&r�r��sC��,��[�[���Y����R�W�W�(=�>��[�[���U�V�R�X�X�t�R�X�X�$F�G��[�[���S�)�,�,�-�H�?�,��[�[���U�V�R�X�X�t�R�X�X�$F�G��[�[���S�)�,�,�-�H�,��[�[���Y����R�W�W�(=�>��[�[���U�V�R�X�X�t�R�X�X�$F�G��[�[���S�)�,�,�-�H�?�,r%r�c��eZdZd�Zd�Zy)�TestDataFrameSetItemCallablec��tgd�gd�d��}gd�|d�<tgd�gd�d��}tj||�y)NrKr^r")r-r4r`rac��y)NrMr$r>s r&r?zDTestDataFrameSetItemCallable.test_setitem_callable.<locals>.<lambda>s�r%r�r�s   r&�test_setitem_callablez2TestDataFrameSetItemCallable.test_setitem_callables;��
�\��=�
>��,��=���.�\�B�C��
���b�#�&r%c��d�}tddgddggt��}|||dkD<td|g|dgg�}tj||�y)Nc��|dzS)Nr)r$r>s r&�inczETestDataFrameSetItemCallable.test_setitem_other_callable.<locals>.incs���q�5�Lr%rWr)rAr)rrr1rx)r3rr5r6s    r&�test_setitem_other_callablez8TestDataFrameSetItemCallable.test_setitem_other_callablesV��	���Q��!�R��)��
8����2��6�
��r�3�i�#�r��3�4��
���b�(�+r%N)r!r"r#rrr$r%r&rrs��'�
,r%rc��eZdZejejjdd�d�gddg��d���Zejjd�	�ejjd
e
eje
g�d���Zejjd
e
eje
g�d��Zejjd
ej"ej$g�d��Zd�Zejjdddg�ejjdiddgiddgig�d���Zd�Zd�Zd�Zy)�TestDataFrameSetItemBooleanMask�	mask_typec�8�|tj|�dzkDS�Nr,)rB�abs�r5s r&r?z(TestDataFrameSetItemBooleanMask.<lambda>-s��B������a��'�r%c�L�|tj|�dzkDjSr)rBrrQrs r&r?z(TestDataFrameSetItemBooleanMask.<lambda>-s��R�"�&�&��*�q�.�5H�4P�4P�r%�	dataframerG)�idsc�j�|j�}||�}|j�}tj||<|jj�}tj|tj|�<t||j|j��}tj||�y)Nr)
r�rBr#rQrGrr+rir1rx)r3rrIr5r�r�r6s       r&�test_setitem_boolean_maskz9TestDataFrameSetItemBooleanMask.test_setitem_boolean_mask*s����
�
�
����}��������v�v��t���9�9�>�>�#��#%�6�6�����$�� ��X�R�X�X�r�z�z�J��
���f�h�/r%z1Currently empty indexers are treated as all False)�reasonrdc���tdgdgdgd��}|tk(r|gd��}n|g�}d}tjt|��5dg|j
|dgf<ddd�y#1swYyxYw)	Nr(r)rZrrAz@Must have equal len keys and value when setting with an iterablercr-)rrrkrlrmr�)r3rdr5r rqs     r&�:test_setitem_loc_empty_indexer_raises_with_non_empty_valuezZTestDataFrameSetItemBooleanMask.test_setitem_loc_empty_indexer_raises_with_non_empty_value>su���c�U�!��A�3�7�
8���&�=��"�H�-�G��"�g�G�P��
�]�]�:�S�
1�	)�&'�S�B�F�F�7�S�E�>�"�	)�	)�	)�s�A)�)A2c�
�tdgdgdgd��}|dg�}d|dz
|j|dgf<tdgdgdgd��}tj||�d|j|dgf<tj||�y)	Nr(r)rZFr@r/r-r)rr�r1rx)r3rdr5r r6s     r&�1test_setitem_loc_only_false_indexer_dtype_changedzQTestDataFrameSetItemBooleanMask.test_setitem_loc_only_false_indexer_dtype_changedKs���
�c�U�!��A�3�7�
8���u�g�,��!#�b��g�����w���~���C�5��s�!��=�>��
���b�(�+�!"����w���~��
���b�(�+r%r c���tgd�gd�d��}|j�}|ddk\}||�|jd�||�|<tj||�y)N)r)rfr,r.r^rr(r.)rr�r�r1rx)r3r r5r6r�s     r&�"test_setitem_boolean_mask_aligningzBTestDataFrameSetItemBooleanMask.test_setitem_boolean_mask_aligningYs]���\��=�
>���7�7�9���#�w�!�|��#�B�K��-�9�9�#�>����D��
���b�(�+r%c�|�tgd�ddg��}tgd��}gd�}t||d�|��}tgd	�gd
���}tgd��}gd�}t||d�|��}|j�}	|	djjgd
��|	d<|dd
k(}
ddg||
<t
j||	�y)N)r(r(r-r-r(r(r(r(r-)r@)r^r_r`�k�lr�r�)r)r)r,r,r)r)r))�catsrQr*)r(r(r/r/r(r(r(rZ)r)r)r.r.r)r)r)r#r/r,)r
r
rr�r��set_categoriesr1rx)r3�cats2�idx2�values2�
exp_multi_row�catsf�idxf�valuesfr5�	exp_fancyr�s           r&�test_setitem_mask_categoricalz=TestDataFrameSetItemBooleanMask.test_setitem_mask_categoricalbs����?�S�RU�J�W���8�9��'��!�5�G�"D�D�Q�
��/�O�
���8�9��'��
���9��
F��!�&�&�(�	�%�f�-�1�1�@�@��Q�	�&���&�z�S� ����8��4��
���b�)�,r%r8r�r:�kwargsr+r)rirMc��||d<tdi|��}|j�}d|||kD<tj||�y)Nr8�/r$r�)r3r8r.r5r�s     r&�%test_setitem_empty_frame_with_booleanzETestDataFrameSetItemBooleanMask.test_setitem_empty_frame_with_booleanysB�� ��w��
�
 ��
 ���g�g�i����2��8��
���b�#�&r%c
�*�ttd��}gd�}t||tjgd�gd�gd�gt
����}t||tjt|�t|�f���}t||tjgd�gd�gd	�gt
����}d
||d|zkD<tj||�tjtd�
�5d
||jdd
dkD<ddd�y#1swYyxYw)Nr.r�)�r�r�)��?r��@)r��@�@rA)r+rir4)r4r�rW)rWrWrWrWr�zItem wrong lengthrcr,)rErurrBrGr�r�rFr1rxrkrlrmr+)r3r�r�df1r�r6s      r&�test_setitem_boolean_indexingz=TestDataFrameSetItemBooleanMask.test_setitem_boolean_indexing�s����5��8�n���������� �/�?�C�5��
���c�4�b�g�g�s�3�x��T��>S�6T�U��������?�N�L�I�QV�W�
�� "��C�#��)�O��
���c�8�,�
�]�]�:�-@�
A�	)�&(�C��	�	�#�2���"�#�	)�	)�	)�s�*D	�	Dc���tddgddgdd��}|j�}tddgd�	�}td
ddgid
dg��|j|d
gf<t	j
||�y)Nr)r,r.rfr(rZFr/r\r-rtr_rr*)rr�rr�r1rx)r3r5r6r s    r&�-test_loc_setitem_all_false_boolean_two_blockszMTestDataFrameSetItemBooleanMask.test_loc_setitem_all_false_boolean_two_blocks�sm��
�a��V�1�a�&�s�;�
<���7�7�9���%���c�2��!*�C�!�Q��=��A��!G����w���~��
���b�(�+r%c���tddgddgg�}tddgddgg�}ttddgd�	�ttjdgd�	�d
��}d||<t	j
||�y)NrWr,r.r�rTF�booleanrAr�)rrr��NAr1rx)r3r5r6�boolean_indexers    r&�test_setitem_ea_boolean_maskz<TestDataFrameSetItemBooleanMask.test_setitem_ea_boolean_mask�s{��
��Q��!�R��)�
*���q�!�f�q�!�f�-�.��#��4��-�y�9��2�5�5�$�-�y�9�
�
�� ��?��
���b�(�+r%N)r!r"r#r��skip_array_manager_invalid_testrkr�r�r�xfailrErBrGrrrr1rr�rr-r1r9r;r@r$r%r&rr)sd���'�'��[�[����	'�)P�Q�
�'�"���
0��(�0��[�[���Q��R��[�[���U�T�2�8�8�V�$<�=�	)�>�S�	)��[�[���U�T�2�8�8�V�$<�=�,�>�,��[�[���Y����R�V�V�(<�=�,�>�,�-�.�[�[���W�w��&8�9��[�[���X��W�q�c�N�Y���<N�'O�P�'�Q�:�'�)�.,�,r%rc���eZdZd�Zej
j
dddg�d��Zd�Zej
j
dde	jdgdggd	�
�dgdggg�d��Zej
j
dde	jdgdgg�dgdggg�d
��Zej
j
dddgejddgej
jdd����g�ej
j
ddded�ed�fdg�d���Zej$d��Zd�Zej
j
dd	dg�d��Zd�Zd �Zy!)"�%TestDataFrameSetitemCopyViewSemanticsc��d|jvsJ�|dj�}||d<tj|jdd�|jjd�f<t
|dd�j�sJ�y)Nr:rMrtr@)rir�rBr#r��get_locrr�)r3rIr�s   r&�test_setitem_always_copyz>TestDataFrameSetitemCopyViewSemantics.test_setitem_always_copy�s}���+�-�-�-�-�-����!�!�#����C��CE�6�6�����2��{�2�2�:�:�3�?�?�@��Q�q��W�~�!�!�#�#�#r%�consolidateTFc��tgd�gd�d�gd���}|jddtj�|sZ|r4|j	�t|jj�dk(s&J�t|jj�dk(sJ�|dj}d	|jdd�df<ttjtjd	d	g|jd�
�}tj|d|�|sFtj||j�tj ||dj�sJ�yy)N)g�������?g������@g������@gffffff@)gffffff@gffffff@gffffff@g333333 @)r�r�)rr)r,r.r*r,r�r)�*rb)rr�rBr#�_consolidate_inplacerFr��blocks�_valuesr�rr+r1r�r�rQ�
shares_memory)r3rH�using_array_managerr�r5�zvalsr6s       r&�#test_setitem_partial_column_inplacezITestDataFrameSetitemCopyViewSemantics.test_setitem_partial_column_inplace�s
���&�-A�B�,�
��	�	�	�!�S�"�&�&�!�"���'�'�)��2�7�7�>�>�*�a�/�/�/��2�7�7�>�>�*�a�/�/�/��3���������q�r�3�w���2�6�6�2�6�6�2�r�2�"�(�(��M��
���r�#�w��1�#��'�'��x���?��#�#�E�2�c�7�?�?�;�;�;�#r%c���ddgdz}tddg|��}|j�}|dd}d|d<tgd�g|�	�}tj||�tj||�y)
NrMrNr,r3rrr')r3r,r3rtrhr�)r3rr5�df_copy�df_viewr6s      r&�*test_setitem_duplicate_columns_not_inplacezPTestDataFrameSetitemCopyViewSemantics.test_setitem_duplicate_columns_not_inplace�sl���S�z�A�~��
�s�1�#�t�
4���'�'�)���Q�%����3���.�/��>��
���g�w�/�
���b�(�+r%r�r)r:rAc���ddg}tdddg|��}|j�}|dd}||dg<tddgddgg|��}tj||�tj||�y)NrMrNrr)rrhr�)r3r�rOrr5rSrTr6s        r&�#test_setitem_same_dtype_not_inplacezITestDataFrameSetitemCopyViewSemantics.test_setitem_same_dtype_not_inplace�sw��
�S�z��
�q��A���
5���'�'�)���Q�%����C�5�	��q�!�f�q�!�f�-�t�<��
���b�(�+�
���g�w�/r%r�c���ddg}tdddg|��}|j�}|dd}||dg<tddgddgg|��}tj||�tj||�y)NrMrNrr)rr�rhr�)r3r�rr5rSrTr6s       r&�2test_setitem_listlike_key_scalar_value_not_inplacezXTestDataFrameSetitemCopyViewSemantics.test_setitem_listlike_key_scalar_value_not_inplace�sw���S�z��
�q��A���
5���'�'�)���Q�%����C�5�	��q�#�h��C��1�4�@��
���g�w�/�
���b�(�+r%r r(z+Boolean indexer incorrectly setting inplace)r�strict)�markszvalue, set_value)r)rt)r�g@z
2020-12-31z
2021-12-31rc��td|iddg��}|j�}|dd}|||<tj||�y)Nr(rr)r*r�)r3r��	set_valuer r5r6r�s       r&�"test_setitem_not_operating_inplacezHTestDataFrameSetitemCopyViewSemantics.test_setitem_not_operating_inplacesE��2��U�|�A�q�6�
2���7�7�9���!�u����7��
���d�H�-r%c��td�D�cgc]}d|����	}}t|D�cic]!}|tjt	|����#c}|��}|j
jdj}tj�5|jD]
}d|||<�	ddd�|sFtj|tjd�tjd�fdk(�sJ�ytj|tjd�tjd�fdk(�sJ�ycc}wcc}w#1swY��xYw)Nr@r/r*rr))
rurrBr�rFr�rLrQr1�raises_chained_assignment_errorrir�r�)	r3r�r�r_r��colr5rQ�labels	         r&�"test_setitem_column_update_inplacezHTestDataFrameSetitemCopyViewSemantics.test_setitem_column_update_inplace!s��$)��9�-�a�A�a�S�'�-��-�
�f�E�s��R�X�X�c�&�k�2�2�E�V�
T��������"�)�)��
�
/�
/�
1�	%����
%��#$��5�	�%� �
%�	%�#��6�6�&����2���	�	�"�
�!=�>�!�C�D�D�D��6�6�&����2���	�	�"�
�!=�>�!�C�D�D�D��.��E�	%�	%�s�D<�&E�E�Ec���tgd��}tgd�d��|d<tgd�d��|d<tgd�gd�t��}tj|j
|�y)	NrTrDrA�col1�col2)r:rDrD)rrerf)r+r8)rrrr1r�r�)r3r5�expected_typess   r&�%test_setitem_column_frame_as_categoryzKTestDataFrameSetitemCopyViewSemantics.test_setitem_column_frame_as_category5sU��
�y�
!���y�
�;��6�
��I�Z�8��6�
��-�5H�PV�
��	���r�y�y�.�9r%r8r�c�2�tdtjd�i|��}tjdgg�|jtjdg�tjdg�f<tdgd�i|��}tj||�y)Nr(r.rAr,r)r,r)r))rrBr�rGr�r1rx)r3r8r5r6s    r&�"test_setitem_iloc_with_numpy_arrayzHTestDataFrameSetitemCopyViewSemantics.test_setitem_iloc_with_numpy_array@sq����R�W�W�Q�Z�(��
6��02���1�#���������!��
�r�x�x���}�,�-��c�9�-�U�;��
���b�(�+r%c�l�tgd�gd�ggd���}tddgddggd	d	g��}||d	<tgd
�gd�ggd���}tj||�tgd�gd
�ggd���}tddgddggd	d	g��}||d	<tgd�gd�ggd���}tj||�y)NrK)rfrtr_r\)r(r-r(r/rhrr4r,r5r()rr,r4rf)r,rtr5r\rTrk)r(r(r-)rr4r.)r,r5r_r�r5s    r&�!test_setitem_frame_dup_cols_dtypezGTestDataFrameSetitemCopyViewSemantics.test_setitem_frame_dup_cols_dtypeIs���
��l�3�=Q�
R���!�S��A�s�8�,�s�C�j�A����3���
�^�,�6J�
��	���b�(�+�
�	�9�-��
G���!�S��A�s�8�,�s�C�j�A����3���k�;�7��Q��
���b�(�+r%c���tdgdd��}td|i�jd�}|ddj�}d|d<d|d<tgddg|dd��}t	j
||�y)	Nz
2000-01-01r��date)r8rHr�3010�2010rI)rrrr�r1rx)r3�dtir5r6s    r&�"test_frame_setitem_empty_dataframezHTestDataFrameSetitemCopyViewSemantics.test_frame_setitem_empty_dataframeYs~���\�N�(��H��
���}�
%�
/�
/��
7��
��!�W�\�\�^����6�
���6�
����V�$��b�q�'�
��
	���b�(�+r%N)r!r"r#rGrkr�r�rQrUrBrGrWrY�paramrBrr^r�rArcrhrjrlrrr$r%r&rDrD�s���$��[�[���]�T�5�M�:�<�;�<�>
,��[�[����!�X�R�X�X��s�Q�C�j��8�A�3���*�E��
0��
0��[�[���W�s�H�B�H�H�s�e�c�U�^�,D��u�s�e�n�&U�V�
,�W�
,��[�[�����
�E��F�L�L��u�
��k�k�'�'�H� �(��
�
	
�
��[�[������
�|�
$�i��&=�>��		
��.��
�..��'�'�E�(�E�&	:��[�[���W�w��&8�9�,�:�,�,� ,r%rDc�z�tdddgi�}tjtd��5d|jdd�df<ddd�tdddgi�}tj
||�tdddgi�}tjtd��5ddd	�|jdd�df<ddd�tdddgi�}tj
||�tdddgi�}d
dd	�|jdd�df<tdd
dgi�}tj
||�y#1swY��xYw#1swY��xYw)Nr(r)r,zincompatible dtypercTr6g@r�r.rf)rr1r�r�r�rx�r5r6s  r&�'test_full_setter_loc_incompatible_dtypervjs(��	�C�!�Q��=�	!�B�	�	#�	#�M�9M�	N������q�#�v����#��d�|�,�-�H����"�h�'�	�C�!�Q��=�	!�B�	�	#�	#�M�9M�	N�*� �S�)����q�#�v��*��#��S�z�*�+�H����"�h�'�	�C�!�Q��=�	!�B��q�\�B�F�F�1�c�6�N��#��1�v��'�H����"�h�'����*�*�s�D%�D1�%D.�1D:c
���tgd�gd�d��}d|j|jdkddgf<tgd�gd�ddtd	�gd
d
td	�gd��}t	j
||�y)NrT)r7rtr_r")r)r�r)r;�Gr�r#r�)rMrNr;rx)rr�r+r�r1rxrus  r&�)test_setitem_partial_row_multiple_columnsry~ss��	���5�	6�B�(2�B�F�F�2�8�8�q�=�3��*�$�%�����q�%��,�'����e��-�		
��H����"�h�'r%)4r�numpyrBrk�pandas.util._test_decorators�util�_test_decoratorsr��pandas.core.dtypes.baserr��pandas.core.dtypes.commonr�pandas.core.dtypes.dtypesrrrr	�pandasr�r
rrr
rrrrrrrrrrrr�pandas._testing�_testingr1�pandas.core.arraysr�pandas.tseries.offsetsrrr�r�r�rrrDrvryr$r%r&�<module>r�s�����
�)�)�<�5���������$�*�'�{,�{,�|'1�'1�TP0�P0�f+,�+,�\,�,�,E,�E,�Pv,�v,�r(�(
(r%

Youez - 2016 - github.com/yon3zu
LinuXploit