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 :  /Python312/Lib/site-packages/pandas/tests/frame/methods/__pycache__/

Upload File :
current_dir [ Writeable ] document_root [ Writeable ]

 

Command :


[ Back ]     

Current File : /Python312/Lib/site-packages/pandas/tests/frame/methods/__pycache__/test_set_index.cpython-312.pyc
�

/	�g�g����dZddlmZmZddlZddlZddlmZmZm	Z	m
Z
mZmZm
Z
mZmZmZddlmZej(d��ZGd�d�ZGd�d	�ZGd
�d�Zy)z/
See also: test_reindex.py:TestReindexSetIndex
�)�datetime�	timedeltaN)
�Categorical�CategoricalIndex�	DataFrame�
DatetimeIndex�Index�
MultiIndex�Series�
date_range�period_range�to_datetimec
�>�tgd�gd�gd�tjjd�j	d�tjjd�j	d�tjjd�j	d�d��}|S)aM
    Fixture for DataFrame of columns that can be used for indexing

    Columns are ['A', 'B', 'C', 'D', 'E', ('tuple', 'as', 'label')];
    'A' & 'B' contain duplicates (but are jointly unique), the rest are unique.

         A      B  C         D         E  (tuple, as, label)
    0  foo    one  a  0.608477 -0.012500           -1.664297
    1  foo    two  b -0.633460  0.249614           -0.364411
    2  foo  three  c  0.615256  2.154968           -0.834666
    3  bar    one  d  0.234246  1.085675            0.718445
    4  bar    two  e  0.533841 -0.005702           -3.533912
    )�foorr�barr)�one�two�threerr)�a�b�c�d�e��)�A�B�C�D�E��tuple�as�label)r�np�random�default_rng�standard_normal)�dfs �KC:\Python312\Lib\site-packages\pandas/tests/frame/methods/test_set_index.py�frame_of_index_colsr+sz��
�4�6�*����&�&�q�)�9�9�!�<����&�&�q�)�9�9�!�<�&(�i�i�&;�&;�A�&>�&N�&N�q�&Q�
	
�	
�B��I�c��eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
d	�Zejjd
dddd
gdg�ejjdddg�ejjdddg�d����Zejjd
dddd
gdg�ejjdddg�d���Zejjd
dddd
gdg�ejjdddg�d���Zd�Zejjdeeej,ed�d�g�ejjdgd��ejjdddg�d����Zejjdeeej,ed�g�ejjdgd��ejjdddg�d����Zejjd eeej,eed!�d"�g�ejjd#eeej,eed$�d%�g�ejjdgd&��ejjdddg�d'�����Zejjd(ddg�ejjdddg�d)���Zd*�Zd+�Zd,�Zd-�Z y.)/�TestSetIndexc��gd�gd�d�}t|�}gd�}||d<tj|d�}|j|�y)N)rg@�)�r�)�t1�t2))r�)rr)r5r�tuples)rr
�from_tuples�	set_index)�selfrr)r6�indexs     r*�test_set_index_multiindexz&TestSetIndex.test_set_index_multiindex9sC���i�0��
�q�\��)����8���&�&�r�(�|�4��
���U�r,c	���tddd�ddd�ddd	d
�ddd
d
�ggd���}|jddg�}|ddg}tj|d|dgddg��|_tj||�y)Nr5r)r�pr�
)r�mr0��)rr?r=r1��)rr?r=�x��columnsrrDr?r=��names)rr8r
�from_arraysr:�tm�assert_frame_equal)r9r)�result�expecteds    r*�test_set_index_empty_columnz(TestSetIndex.test_set_index_empty_columnDs���
��a� ��b�!��b�r�*��b�r�*�	
�)�
�����s�C�j�)���s�C�j�>��#�/�/��C��"�S�'�0B�3�PS�*�U���
���f�h�/r,c��ttd��td��gd��}|jddg�}|jj	�j
}|ddgj
}t
j||�y)Nzdatetime64[ns]��dtype�int64�rrrrr)rrr8r:�to_frame�dtypesrJ�assert_series_equal)r9�df1�df2rLrMs     r*�test_set_index_empty_dataframez+TestSetIndex.test_set_index_empty_dataframeVsq����/�0�v�G�7L�SU�V�
���m�m�S�#�J�'�����#�#�%�,�,����S�z�?�)�)��
���v�x�0r,c��tjgd��}ttjjd�j
d�|��}|j|jd�}|jdd�dd�f}|jdd�dfj|_|jdg|j_tj||�y)N))rr5)rr)rr5r)r0r0rErr5)r
r7rr%r&r'r(r8rF�iloc�valuesr:rHrJrK)r9rFr)rLrMs     r*� test_set_index_multiindexcolumnsz-TestSetIndex.test_set_index_multiindexcolumnsas����(�(�)M�N��
��I�I�!�!�!�$�4�4�V�<�g�
�����b�j�j��m�,���7�7�1�a�b�5�>������A���-�-��� "�
�
�1�
������
���f�h�/r,c��tdgd��jd�}td|i�}|j|�jdj
dk(sJ�tt
|j��dj
dk(sJ�|j|j�jdj
dk(sJ�y)Nz2014-01-01 10:10:10�UTC��tzzEurope/Romerrr@)r�
tz_convertrr8r:�hourrr)r9�idxr)s   r*�test_set_index_timezonez$TestSetIndex.test_set_index_timezonens����2�3��>�I�I�-�X��
��S�z�
"���|�|�C� �&�&�q�)�.�.�"�4�4�4��V�B�D�D�\�*�1�-�2�2�b�8�8�8��|�|�B�D�D�!�'�'��*�/�/�2�5�5�5r,c
�8�ttd�D�cgc]}tddd�t|�z��c}tj
j
d�jd�d��}|jd�}t|jt�sJ�ycc}w)Ni�i�r5r�rrr)r�rangerrr%r&r'r(r8�
isinstancer:r)r9�ir)�idfs    r*�!test_set_index_cast_datetimeindexz.TestSetIndex.test_set_index_cast_datetimeindexws|��
�CH��;�O�a�h�t�Q��*�Y�q�\�9�O��Y�Y�*�*�1�-�=�=�d�C�
�
���l�l�3����#�)�)�]�3�3�3��
Ps� Bc���tdddd��}tgd�gd�d�|�	�j�}|jd
�}tgd�gd�d�t	|d
���	�}|j
j
d�|_tj||�|jd
dg�}tj|gd�gd
dg�
�}tdgd�i|��}tj||�y)Nz2006-10-29 00:00:00r0�h�
US/Pacific)�periods�freqra�rr5r)r0r1r)rr)�datar:r:��namerrGr�r:)rr�reset_indexr8r	r:�
_with_freqrJrKr
rI)r9�dir)�res�exp�	exp_indexs      r*�test_set_index_dstzTestSetIndex.test_set_index_dst�s���
�-�q�s�|�
T��
�)�)�<�B�
G�
S�
S�
U���l�l�7�#��� �y�1����)�
���I�I�(�(��.��	�
���c�3�'��l�l�G�S�>�*���*�*�B�	�?�7�C�.�Q�	���i�(�	�:��
���c�3�'r,c�R�|}ttjt|��ddd��}|j	|�}tj|j|�tjtd��5|j	|ddd��ddd�y#1swYyxYw)N���zLength mismatch��matchr)r	r%�arange�lenr8rJ�assert_index_equalr:�pytest�raises�
ValueError)r9�float_string_framer)rds    r*�test_set_indexzTestSetIndex.test_set_index�s���
���B�I�I�c�"�g�&�t��t�,�-��
�\�\�#�
��
���b�h�h��,�
�]�]�:�->�
?�	#��L�L��S�q�S��"�	#�	#�	#�s�<B�B&c��ttjd�tt	d�t
��tt
d�D�cgc]}d|����	c}t
����}d|j_|j|j�jjdgk(sJ�tj|dd	gjjdd	g�
�}tj|gd�jjgd��
�}|jdd	g�}|j|j�jjdd	gk(sJ�t|j|j�jt�sJ�t!j"|j|j�j|�|jj%d
dg�}t|j|j|g�jt�sJ�t!j"|j|j|g�j|�ycc}w)N)r>r1�ABCDrPr>zi-)rFr:rurrrG)rrrr)rrrrrr)rr%�onesr	�list�objectrhr:rur8rHr
rI�Tr\rirJr��rename)r9rjr)�mi�mi2�idx2s      r*�test_set_index_namesz!TestSetIndex.test_set_index_names�s���
��G�G�G���$�v�,�f�5��5��9�5�a�2�a�S��5�V�D�
��
����
��|�|�B�H�H�%�+�+�1�1�f�X�=�=�=�
�
#�
#�B��S�z�N�$4�$4�$;�$;�C��:�
N���$�$��#�$�&�&�-�-�5I�
���\�\�3��*�
%���|�|�B�H�H�%�+�+�1�1�c�3�Z�?�?�?��"�,�,�r�x�x�0�6�6�
�C�C�C�	���b�l�l�2�8�8�4�:�:�B�?��x�x����S�z�*���"�,�,����$�'7�8�>�>�
�K�K�K�	���b�l�l�B�H�H�d�+;�<�B�B�C�H��96s�I�keysrrrr!�inplaceTF�dropc��|}t|t�r)tj|D�cgc]}||��	c}|��}nt	|||��}|r|j|d��n|}||_|r(|j�}	|	j||d��}
|
�J�|j||��}	tj|	|�ycc}w)NrGrtr5��axisT)r�r�)r�)rir�r
rIr	r�r:�copyr8rJrK)r9r+r�r�r�r)rDrdrMrL�return_values           r*�test_set_index_drop_inplacez(TestSetIndex.test_set_index_drop_inplace�s���!���d�D�!��(�(��)>�A�"�Q�%�)>�d�K�C���4��t�,�C�,0�2�7�7�4�a�7�(�b�������W�W�Y�F�!�+�+�D�t�T�+�J�L��'�'�'��\�\�$�T�\�2�F�
���f�h�/��*?s�B>c�\�|}t|t�r|n|g}tj|jg|D�cgc]}||��	c}zdg|z��}|r|j|d��n|j
�}||_|j||d��}tj||�ycc}w)NrGr5r�T�r��append)
rir�r
rIr:r�r�r8rJrK)	r9r+r�r�r)rDrdrMrLs	         r*�test_set_index_appendz"TestSetIndex.test_set_index_append�s���!��!�$��-�t�D�6���$�$�
�X�X�J��.�A�"�Q�%�.�.�t�f�t�m�
��-1�2�7�7�4�a�7�(�b�g�g�i��������d��d��;��
���f�h�/��/s�B)
c���|jdg|d��}t|t�r|n|g}|jdg|z|d��}|j||d��}tj||�y)NrTr�)r8rir�rJrK)r9r+r�r�r)rMrLs       r*�#test_set_index_append_to_multiindexz0TestSetIndex.test_set_index_append_to_multiindex�sq��!�
*�
*�C�5�t�D�
*�
I��!�$��-�t�D�6��&�0�0�#����D�QU�0�V�����d��d��;��
���f�h�/r,c��tgd�gd�d��}tdddgitddgd	�
��}|j|jj	d��}|jd	�}t
j||�y)NrrrS)�val�keyr�r5rrrr�rtc��|dk\S)Nr5�)�indxs r*�<lambda>z<TestSetIndex.test_set_index_after_mutation.<locals>.<lambda>�s
��t�q�y�r,)rr	�locr:�mapr8rJrK)r9r)rMrXrLs     r*�test_set_index_after_mutationz*TestSetIndex.test_set_index_after_mutation�sj��
�y��A�
B���e�a��V�_�e�S�#�J�U�.K�L���f�f�R�X�X�\�\�"8�9�:�����u�%��
���f�h�/r,�boxc��t|�gS�N)r��rDs r*r�zTestSetIndex.<lambda>	s��t�A�w�i�r,c�.�tj|g�Sr��r
rIr�s r*r�zTestSetIndex.<lambda>
���j�,�,�a�S�1�r,zappend, index_name)�TN�Tr�T�test�FNc���|}||j_||d�}|tk(r;d}tjt
|��5|j
|||��ddd�yt|dd�}	|	�t|dd�gn|	}
|j
|||��}|j
dgd|��}|r|g|
zn|
|j_tj||�y#1swYyxYw)Nrz%['one', 'two', 'three', 'one', 'two']r�r�rHruF)r:rur�r�r��KeyErrorr8�getattrrHrJrK)
r9r+r�r��
index_namer�r)r��msg�name_mirurLrMs
             r*� test_set_index_pass_single_arrayz-TestSetIndex.test_set_index_pass_single_arrays���$!��"����
��"�S�'�l���$�;�9�C����x�s�3�
<����S�t�F��;�
<�
<��c�7�D�1�G�3:�?�G�C���.�/��D��\�\�#�D��\�@�F��|�|�S�E��f�|�E�H�:@�J�<�$�#6�d�H�N�N� ��!�!�&�(�3�
<�
<�s�C�C$c�.�tj|g�Sr�r�r�s r*r�zTestSetIndex.<lambda>.s���9O�9O�QR�PS�9T�r,)r��Trr�r�r�c�t�|}||j_d||d�g}d|tjtt
tfvrdndg}|j|||��}	|jddgd|��}
|r|
jdd��n|
}
|r|g|zn||
j_	tj|	|
�y)Nrrr�Fr5r�)r:rur%�arrayr�r"�iterr8r�rHrJrK)r9r+r�r�r�r�r)r�rHrLrMs           r*�test_set_index_pass_arraysz'TestSetIndex.test_set_index_pass_arrays-s���!��"����
��S��C��\�"���c�b�h�h��e�T�%B�B�d��L�����d��f��=���<�<��c�
��v�<�F��15�8�=�=��1�=�-�8��7=�
�|�e�3�5�����
���f�h�/r,�box2c�.�tj|g�Sr�r�r�s r*r�zTestSetIndex.<lambda>Tr�r,c��|jSr�rtr�s r*r�zTestSetIndex.<lambda>U�
��a�f�f�r,�box1c�.�tj|g�Sr�r�r�s r*r�zTestSetIndex.<lambda>`r�r,c��|jSr�rtr�s r*r�zTestSetIndex.<lambda>ar�r,)r�r�r�r�c��|}||j_||d�||d�g}|j|||��}	||d�||d�g}t|dt�r%|ddk(rt|dt�r
|ddk(rdn|}
|j|dg|
|��}|j|dg|d��}tj|	|�y)Nrr�rr5FT)r:rur8ri�strrJrK)r9r+r�r�r�r�r�r)r�rL�
first_droprMs            r*�$test_set_index_pass_arrays_duplicatez1TestSetIndex.test_set_index_pass_arrays_duplicateLs���>!��"����
��R��W�
�t�B�s�G�}�-�����d��f��=���R��W�
�t�B�s�G�}�-���4��7�C�(���G�s�N��t�A�w��,���G�s�N�
��	��<�<��a��	�
�6�<�J���%�%�t�A�w�i�d�4�%�H��
���f�h�/r,r�c���|}tj|d|dgddg��}|j|||��}|jddgd|��}tj||�y)NrrrGr�F)r
rIr8rJrK)r9r+r�r�r)r�rLrMs        r*�test_set_index_pass_multiindexz+TestSetIndex.test_set_index_pass_multiindex�sj��!���%�%�r�#�w��3��&8��c�
�K�����d��f��=���<�<��c�
��v�<�F��
���f�h�/r,c�t�ttd�dzd��}ttjjd�j
d�|jd��}|jd�}tj|j|�ttjjd�j
d�|d��}|jd�}tj|j|�|j�jd�}tj|j|�y)N�abrrrtrr>rg)
rr�rr%r&r'r(r\r8rJr�r:rw)r9�cir)rks    r*�(test_construction_with_categorical_indexz5TestSetIndex.test_construction_with_categorical_index�s���
�d�4�j�1�n�3�
7����)�)�'�'��*�:�:�2�>�R�Y�Y�O�
���l�l�3���
���c�i�i��,��R�Y�Y�2�2�1�5�E�E�b�I�PR�S�
T���l�l�3���
���c�i�i��,��o�o��)�)�#�.��
���c�i�i��,r,c�j�tgd�gd�ttd�td�d��ttd�td�d��d��}d	d
dd	gdd
gd	d
gfD]S}|j|�j	�}|j|j��}tj||��Uy)
N)r5rr5r5r)r>����"�abaab�bacF)�
categories�orderedT)rr�C1�C2r�r�rrE)	rrr�r8rw�reindexrFrJrK)r9r)�colsrLs    r*�)test_set_index_preserve_categorical_dtypez6TestSetIndex.test_set_index_preserve_categorical_dtype�s���
�$�)�!�$�w�-�D��K�QV�W�!�$�w�-�D��K�QU�V�	
�
���4�#�t��s�D�k�D�$�<�H�	.�D��\�\�$�'�3�3�5�F��^�^�B�J�J�^�7�F��!�!�&�"�-�	.r,c�0�tgd�gd�td�d��}t|jd�d��|_|jjd�|_t
gd	�d�
�}|jd�jd�}|jdd�
�}tj|jjd|�tj|jjdtddgd�
��|jjddgk(sJ�|jdd�}tj|jjdtddgd�
��tj|jjd|�|jjddgk(sJ�ttj j#d�j!d��}t
gd�d��}t
gd�d��}t%ddd��}|j'd�}|j|�}|j|d�
�}|j|d�
�}t
gd	�d��}t
ddgd��}tj|jjd|�tj|jjd|�tj|jjd|�tj|jj)d�|�tj|jj)d�|�tj|jj)d�|�y)N)rrrrrr)�2011-07-19 07:00:00�2011-07-19 08:00:00�2011-07-19 09:00:00r�r�r�r2)r$r�valuerT)�utcro)r�r�r�rtr_r$�r�rr5rrrz
US/Easternr`)�2012-04-01 09:00r�r��2012-04-02 09:00r�r�z2011-01-01 09:00z
Asia/Tokyo)rprar�r�)rrhr�popr:rbr�tz_localizer8rJr��levelsr	rH�	swaplevelr%r&r'rrx�get_level_values)r9r)rM�idx1r��idx3�	expected1�	expected2s        r*�test_set_index_datetimez$TestSetIndex.test_set_index_datetime�s���
�7���q��
�

���r�v�v�j�1�t�<����8�8�&�&�|�4��� �Q��
���'�'��.�9�9�,�G��
�\�\�'�$�\�
/��
���b�h�h�o�o�a�0�(�;�
���b�h�h�o�o�a�0�%��c�
��2Q�R��x�x�~�~�*�g�!6�6�6�6�
�\�\�!�Q�
��
���b�h�h�o�o�a�0�%��c�
��2Q�R�
���b�h�h�o�o�a�0�(�;��x�x�~�~�'�:�!6�6�6�6�
�r�y�y�,�,�Q�/�6�6�q�9�
:���
��

���
��

���,�a�L�I�����t�$��
�\�\�$�
��
�\�\�$�t�\�
,��
�\�\�$�t�\�
,��!�Q��
�	�"�
�!3�4��
�	�	���b�h�h�o�o�a�0�)�<�
���b�h�h�o�o�a�0�)�<�
���b�h�h�o�o�a�0�$�7�	���b�h�h�7�7��:�D�A�
���b�h�h�7�7��:�D�A�
���b�h�h�7�7��:�D�Ar,c��ttjjd�jd��}t	ddd��}|j|�}t	ddd��}|j|�j|�}t	d	dd
��}|j
|�}|j
|d��}|j
|d��}t	ddd��}t	ddd��}tj|jjd
|�tj|jjd|�tj|jjd|�tj|jjd
�|�tj|jjd�|�tj|jjd�|�y)Nrr2z
2011-01-01r0�M�rprqz2013-01-01 09:00rn�2005�YTr�rr5)rr%r&r'r
r�r8rJr�r:r�r�)r9r)r�r�r�r�r�s       r*�test_set_index_periodz"TestSetIndex.test_set_index_period	s{��
�r�y�y�,�,�Q�/�6�6�q�9�
:���L�!�#�>���{�{�4� ���.���D���{�{�4� �'�'��-���F�A�C�8��
�\�\�$�
��
�\�\�$�t�\�
,��
�\�\�$�t�\�
,�� ��q�s�C�	� �!3�Q�S�I�	�
���b�h�h�o�o�a�0�)�<�
���b�h�h�o�o�a�0�)�<�
���b�h�h�o�o�a�0�$�7�
���b�h�h�7�7��:�D�A�
���b�h�h�7�7��:�D�A�
���b�h�h�7�7��:�D�Ar,N)!�__name__�
__module__�__qualname__r;rNrYr]rerlr}r�r�r��mark�parametrizer�r�r�r�rr	r%r�r�r�r�r�r�r�r�r�r�r�r�r,r*r.r.8sV��	�0�$	1�0�6�	4�(�&#� I�F�[�[���V�c�3��c�
�<T�%U�V��[�[���Y��u�
�6��[�[���V�d�E�]�3�0�4�7�W�0�(�[�[���V�c�3��c�
�<T�%U�V��[�[���V�d�E�]�3�0�4�W�0��[�[���V�c�3��c�
�<T�%U�V��[�[���V�d�E�]�3�	0�4�W�	0�0��[�[���
����H�H���1�
	
�
��[�[����X���[�[���V�d�E�]�3�4�4��
�4�8�[�[���
���r�x�x��/T�U���[�[����O���[�[���V�d�E�]�3�0�4�	��0�.�[�[�������H�H���1��	
���[�[�������H�H���1��	
���[�[����X���[�[���V�d�E�]�3�0�4����80�@�[�[���X��e�}�5��[�[���V�d�E�]�3�	0�4�6�	0�-�&
.�PB�dBr,r.c��eZdZd�Zej
j
dddg�ej
j
dddg�d���Zej
j
dddg�ej
j
dddg�ej
j
degdg�	�d
����Z	ej
j
de
eejed�ggd��	�ej
j
d
ddgddg�	�ej
j
dddg�ej
j
dddg�d�����Zy)�TestSetIndexInvalidc�$�|}tjtd��5|jdd��ddd�tjtd��5|j|d|dgd��ddd�y#1swY�IxYw#1swYyxYw)NzIndex has duplicate keysr�rT)�verify_integrity)r�r�r�r8)r9r+r)s   r*�test_set_index_verify_integrityz3TestSetIndexInvalid.test_set_index_verify_integrity#s���
 ��
�]�]�:�-G�
H�	5��L�L��t�L�4�	5��]�]�:�-G�
H�	D��L�L�"�S�'�2�c�7�+�d�L�C�	D�	D�	5�	5��	D�	D�s�A:�B�:B�Br�TFr�c	�~�|}tjtd��5|jgd�||��ddd�tjtd��5|j|d|ddg||��ddd�d}tjt|��5|jt	|d�||��ddd�tjt|��5|jd|dt	|d�g||��ddd�y#1swY��xYw#1swY��xYw#1swY�lxYw#1swYyxYw)	Nz['foo', 'bar', 'baz']r�)rr�bazr��Xrrz%[('foo', 'foo', 'foo', 'bar', 'bar')])r�r�r�r8r")r9r+r�r�r)r�s      r*�test_set_index_raise_keysz-TestSetIndexInvalid.test_set_index_raise_keys,s*��!��
�]�]�8�+B�
C�	J��L�L�.�T�&�L�I�	J�
�]�]�8�3�
/�	L��L�L�"�S�'�2�c�7�C�0�t�F�L�K�	L�6��
�]�]�8�3�
/�	C��L�L��r�#�w��d�6�L�B�	C��]�]�8�3�
/�	S��L�L�#�r�#�w��b��g��7�d�6�L�R�	S�	S�	J�	J��
	L�	L��
	C�	C��	S�	S�s/�D�D�!D'�'D3�D�D$�'D0�3D<r��set)�idsc	�L�|}d}tjt|��5|j||d�||��ddd�tjt|��5|jd|d||d�g||��ddd�y#1swY�QxYw#1swYyxYw)N�,The parameter "keys" may be a column key, .*r�rr�)r�r��	TypeErrorr8)r9r+r�r�r�r)r�s       r*�test_set_index_raise_on_typez0TestSetIndexInvalid.test_set_index_raise_on_typeBs���!��<��
�]�]�9�C�
0�	A��L�L��R��W��D��L�@�	A��]�]�9�C�
0�	Q��L�L�#�r�#�w��B�s�G��5�D��L�P�	Q�	Q�		A�	A��	Q�	Q�s�B�!$B�B�B#c�.�tj|g�Sr�r�r�s r*r�zTestSetIndexInvalid.<lambda>Ts��*�2H�2H�!��2M�r,)rr	znp.arrayr�r
�lengthr1r2�	too_short�too_longc��|}tjjd�jdd|f�}d}t	j
t|��5|j||�||��ddd�t	j
t|��5|jd|j||�g||��ddd�y#1swY�UxYw#1swYyxYw)Nrrr>z<Length mismatch: Expected 5 rows, received array of length.*r�r�r)	r%r&r'�integersr�r�r�r8r)	r9r+r�rr�r�r)r\r�s	         r*�test_set_index_raise_on_lenz/TestSetIndexInvalid.test_set_index_raise_on_lenRs���!�����&�&�q�)�2�2�1�b�6�)�D��L���]�]�:�S�
1�	@��L�L��V��4��L�?�	@��]�]�:�S�
1�	M��L�L�#�r�t�t�S��[�1��V�L�L�	M�	M�		@�	@��	M�	M�s�C�(C�C	�CN)r�rrrr�rrrr
rrr	r%r�r�rr�r,r*rr"sa��D��[�[���X��e�}�5��[�[���V�d�E�]�3�S�4�6�S�(�[�[���X��e�}�5��[�[���V�d�E�]�3��[�[���U�S�E��w��7�
Q�8�4�6�
Q��[�[���
�	�����$�(M�N�A���
�[�[���X��1�v�K��3L��M��[�[���X��e�}�5��[�[���V�d�E�]�3�M�4�6�N��Mr,rc�$�eZdZd�Zd�Zd�Zd�Zy)�TestSetIndexCustomLabelTypec�X�Gd�d�}|dd�}|dd�}t|ddg|d	d
gi�}t|ddgitd	d
g|����}|j|�}tj||�|j|g�}tj||�|d
d�}d}tjt|��5|j|�ddd�tjt|��5|j|g�ddd�y#1swY�@xYw#1swYyxYw)Nc�$�eZdZdd�Zdefd�ZeZy)�KTestSetIndexCustomLabelType.test_set_index_custom_label_type.<locals>.Thing�returnNc� �||_||_yr��ru�color�r9rur"s   r*�__init__zTTestSetIndexCustomLabelType.test_set_index_custom_label_type.<locals>.Thing.__init__r��� ��	�"��
r,c�4�dt|j��d�S�Nz<Thing �>��reprru�r9s r*�__str__zSTestSetIndexCustomLabelType.test_set_index_custom_label_type.<locals>.Thing.__str__v��� ��d�i�i�� 1��3�3r,�rN)r�rrr$r�r,�__repr__r�r,r*�Thingrqs��
#�
4��
4��Hr,r0�One�red�Two�bluerr5rr0rtrv�Three�pinkz<Thing 'Three'>r�)rr	r8rJrKr�r�r��	r9r0�thing1�thing2r)rMrL�thing3r�s	         r*� test_set_index_custom_label_typez<TestSetIndexCustomLabelType.test_set_index_custom_label_typens��		�		��u�e�$���u�f�%��
���A����A��7�
8���f�q�!�f�-�U�A�q�6��5O�P�����f�%��
���f�h�/����v�h�'��
���f�h�/��w��'����
�]�]�8�3�
/�	!��L�L�� �	!��]�]�8�3�
/�	#��L�L�&��"�	#�	#�		!�	!��	#�	#�s�D�8D �D� D)c�h�Gd�dt�}|ddg�}|ddg�}t|ddg|d	d
gi�}t|ddgitd	d
g|����}|j|�}t	j
||�|j|g�}t	j
||�|d
dg�}d}t
jt|��5|j|�ddd�t
jt|��5|j|g�ddd�y#1swY�@xYw#1swYyxYw)Nc��eZdZdefd�Zy)�XTestSetIndexCustomLabelType.test_set_index_custom_label_hashable_iterable.<locals>.Thingrc�b�t|�}djtt|��}d|�d�S)Nz, zfrozenset({z}))�sorted�joinr�r*)r9�tmp�joined_reprss   r*r/zaTestSetIndexCustomLabelType.test_set_index_custom_label_hashable_iterable.<locals>.Thing.__repr__�s.���T�l��#�y�y��T�3��8��%�l�^�3�7�7r,N)r�rrr�r/r�r,r*r0r>�s��
8�#�
8r,r0r1r2r3r4rr5rr0rtrvr5r6z frozenset\(\{'Three', 'pink'\}\)r�)	�	frozensetrr	r8rJrKr�r�r�r7s	         r*�-test_set_index_custom_label_hashable_iterablezITestSetIndexCustomLabelType.test_set_index_custom_label_hashable_iterable�s&��	8�I�	8���u�~�&����v��'��
���A����A��7�
8���f�q�!�f�-�U�A�q�6��5O�P�����f�%��
���f�h�/����v�h�'��
���f�h�/����(�)��1��
�]�]�8�3�
/�	!��L�L�� �	!��]�]�8�3�
/�	#��L�L�&��"�	#�	#�		!�	!��	#�	#�s�D�D(�D%�(D1c�z�Gd�dt�}|dd�}|dd�}tddgd	d
gg||g��}d}tjt|�
�5|j|�ddd�tjt|�
�5|j|g�ddd�y#1swY�@xYw#1swYyxYw)Nc� �eZdZdd�Zdefd�Zy)�RTestSetIndexCustomLabelType.test_set_index_custom_label_type_raises.<locals>.ThingrNc� �||_||_yr�r!r#s   r*r$z[TestSetIndexCustomLabelType.test_set_index_custom_label_type_raises.<locals>.Thing.__init__�r%r,c�4�dt|j��d�Sr'r)r+s r*r,zZTestSetIndexCustomLabelType.test_set_index_custom_label_type_raises.<locals>.Thing.__str__�r-r,r.)r�rrr$r�r,r�r,r*r0rH�s��
#�
4��
4r,r0r1r2r3r4rrr5r0rErr�)r
rr�r�rr8)r9r0r8r9r)r�s      r*�'test_set_index_custom_label_type_raiseszCTestSetIndexCustomLabelType.test_set_index_custom_label_type_raises�s���	4�C�	4��u�e�$���u�f�%��
��A���A��'�&�&�1A�
B��<��
�]�]�9�C�
0�	!��L�L�� �	!��]�]�9�C�
0�	#��L�L�&��"�	#�	#�		!�	!��	#�	#�s�B%�	B1�%B.�1B:c�n�ttjjd�jd��}t	ddd��}t	ddd��}|j|�}t
j|j|�|j|�}t
j|j|�y)Nrr2z
2011/01/01r�r��2013r�)	rr%r&r'r
r8rJr�r:)r9r)r�r�s    r*�test_set_index_periodindexz6TestSetIndexCustomLabelType.test_set_index_periodindex�s���
�r�y�y�,�,�Q�/�6�6�q�9�
:���L�!�#�>���F�A�C�8��
�\�\�$�
��
���b�h�h��-�
�\�\�$�
��
���b�h�h��-r,N)r�rrr;rErKrNr�r,r*rrms��$#�L%#�N#�4	.r,r)�__doc__rr�numpyr%r��pandasrrrrr	r
rrr
r�pandas._testing�_testingrJ�fixturer+r.rrr�r,r*�<module>rUsm����
�
�����������6gB�gB�THM�HM�Vq.�q.r,

Youez - 2016 - github.com/yon3zu
LinuXploit