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/indexing/multiindex/__pycache__/

Upload File :
current_dir [ Writeable ] document_root [ Writeable ]

 

Command :


[ Back ]     

Current File : /Python312/Lib/site-packages/pandas/tests/indexing/multiindex/__pycache__/test_loc.cpython-312.pyc
�

0	�g����ddlZddlZddlmZmZddlZddlmZm	Z	m
Z
mZddlm
Zejd��Zejd��ZGd�d�Zej&j)dggfd	ged
�fd	dggfddggfdggfej,dd�d
gfeddd
�fej,dd�d
dgfeddd
�fg�d��Zej&j)dged�fd
ggfg�d��Zd�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Z d�Z!d�Z"d�Z#d�Z$d�Z%d �Z&Gd!�d"�Z'd#�Z(d$�Z)d%�Z*d&�Z+d'�Z,d(�Z-ej&j)d)gd*�gd+�g�d,��Z.d-�Z/d.�Z0d/�Z1d0�Z2d1�Z3d2�Z4d3�Z5d4�Z6d5�Z7y)6�N)�
IndexingError�PerformanceWarning)�	DataFrame�Index�
MultiIndex�Seriesc�,�tgd�ggd�gdg��S)zsingle level MultiIndex)�foo�bar�baz�qux)r����first)�levels�codes�names)r���KC:\Python312\Lib\site-packages\pandas/tests/indexing/multiindex/test_loc.py�single_level_multiindexrs���,�-�l�^�G�9��rc��ddggd�g}gd�gd�g}t||��}ttjj	d�jd�|�	�S)
Nrr�rrr)rrrrrr)rrrrrr)rrr)�r��index)rr�np�random�default_rng�standard_normal)rrrs   r�%frame_random_data_integer_multi_indexr"sN���!�f�i�
 �F�
�!3�4�E��f�E�2�E��R�Y�Y�*�*�1�-�=�=�f�E�U�S�Src�Z�eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
d	�Zejjd
ddgd
dgfdggfddggfg�d��Zd�Zd�Zd�Zd�Zd�Zejjdeeeeej4eef�ejjdeeeeej4eef�d���Zd�Zejjdddg�d��Zejjded�d fd!ej@fg�d"��Z!d#�Z"d$�Z#d%�Z$y)&�TestMultiIndexLocc���|}d|jd<|jddk(sJ�|j�}ttd��|_d|jd<|jddk(sJ�y)N�)�r�two�Br�)r'r)�loc�copy�list�range�columns)�self� multiindex_dataframe_random_data�frame�dfs    r�&test_loc_setitem_frame_with_multiindexz8TestMultiIndexLoc.test_loc_setitem_frame_with_multiindex$sq��0��)*��	�	�%�&��y�y�,�-��2�2�2��Z�Z�\���%��(�^��
�$%���� �!��v�v�'�(�A�-�-�-rc��|}dddddd�dddddd�d	d
d	d	d	d�d�}t|�}|j||d��}|jd
dg��}d}tjt
�5tj|j||jdd�ddd�|jd��}|�J�|j|}tjdgdz|��}tjd	gdz|��}	tj||	gd
dg��}
tdgd�i|
��}tj||�y#1swY��xYw)Ni�iX���M�)rrrr�g@�@��)�amount�col�num)r>r?r>r?)�keys)r:r;rT)�inplacer��dtype�rr=)r6r7r8r)r�astype�	set_index�tm�assert_produces_warningr�assert_frame_equalr+�iloc�
sort_indexr�arrayr�from_arrays)r0�any_real_numpy_dtyperC�datar3�key�return_value�res�col_arr�year_arrr�expecteds            r�test_loc_getitem_generalz*TestMultiIndexLoc.test_loc_getitem_general/sL��$���#�#�#�#�>��s�s�s�s�;��b�R�B�2�6�
��
�t�_��
�Y�Y�u�U�3�
4��
�\�\��u�~�\�
.�����
'�
'�(:�
;�	<��!�!�"�&�&��+�r�w�w�q�r�{�;�	<��}�}�T�}�2���#�#�#��f�f�S�k���(�(�C�5�1�9�E�2���8�8�R�D�1�H�E�2���&�&���':�5�%�.�Q���h��8��F��
���c�8�,�	<�	<�s�)1E�E	c��ttjjd�j	d�gd�gd�ggd�gd�g��}tjtd�	�5|jdddd�y#1swYyxYw�
Nr�rr�rrr9�r��
�r9r9r\�r\r]r;�r/rz^2$��match�	rrrr r!�pytest�raises�KeyErrorr+�r0r3s  r�0test_loc_getitem_multiindex_missing_label_raiseszBTestMultiIndexLoc.test_loc_getitem_multiindex_missing_label_raisesLsi��
��I�I�!�!�!�$�4�4�V�<��
�+��k�*�
���]�]�8�6�
2�	��F�F�1�I�	�	�	���"A;�;Bc��|d}|j|jdd�}|jddg}tj||�y)N�A�1�3)��rr])rnr�
)�reindexrr+rG�assert_series_equal)r0�/multiindex_year_month_day_dataframe_random_data�serrU�results     r�/test_loc_getitem_list_of_tuples_with_multiindexzATestMultiIndexLoc.test_loc_getitem_list_of_tuples_with_multiindexWsJ��>�c�B���;�;�s�y�y��B�/�0�����-��7�8��
���v�x�0rc��tjgd�gd�g�}t|td�tj
��}tddg�}tgd�tjddggd�g�tj
��}|j|}tj||�|jddg}tj||�tddgdd	g�
�}|j|}tj||�tgtj
��}tgt|jgggtj
��tj
�
�}|j|}tj||�y)N�rrr�rkr)�C�	�rrOrCrr�rrrrr*r\�rOrrCrr)rOrC�rrrC�rrC)
r�from_productrr.r�float64r+rGrqr)r0r�x�yrUrt�y1�emptys        r�test_loc_getitem_seriesz)TestMultiIndexLoc.test_loc_getitem_series_s5���'�'��O�(D�E����U�1�X�R�Z�Z�@���A�q�6�N���#��)�)�A�q�6�?�*C�D��*�*�
��
���q���
���v�x�0�����1�v���
���v�x�0��Q��F�1�a�&�
)�����r���
���v�x�0��B�b�j�j�1�����E�L�L��R���
�
�S��*�*�
��
���u���
���v�x�0rc��tjgd�gd�g�}t|td�tj
��}t	jddg�}tgd�tjddggd�g�tj
��}|j|}tj||�t	jg�}tgt|jgggtj
�	�d
��}|j|}tj||�t	jd�}tgd�gd�tj
��}|j|}tj||�y)
Nrwrxrzr{rrr|r}r~r�rr)rr�rr.rr�rLr+rGrqr�int64)r0rr�r�rUrtr��scalars        r�test_loc_getitem_arrayz(TestMultiIndexLoc.test_loc_getitem_array~s���'�'��O�(D�E����U�1�X�R�Z�Z�@���H�H�a��V����#��)�)�A�q�6�?�*C�D��*�*�
��
���q���
���v�x�0����������E�L�L��R���
�
�S��
��
���u���
���v�x�0����!����y��r�z�z�R�����v���
���v�x�0rc���ttjjd�j	d�gd�gd�ggd�gd�g��}|j
ddgj
d�}|jd	}tj||�|j
dd�dgfj
dd�
�}|jdd�df}tj||�|j
dgdgfj
d�j
dd�
�}|jdjdd�df}tj||�|j
ddg}|jd}tj||�y)
NrrY)�ir��j)rkrkr))�Xr��Yr`rrr���axisr�)r�r��
rrrr r!rJ�	droplevelr+rGrI�r0r3rUrts    r�test_loc_multiindex_labelsz,TestMultiIndexLoc.test_loc_multiindex_labels�sH��
��I�I�!�!�!�$�4�4�V�<�$�o�6�"�O�4�
���7�7�A�q�6�?�,�,�Q�/��������
���f�h�/��7�7�1�q�c�6�?�,�,�Q�Q�,�7������3����
���f�h�/��7�7�A�3���8�$�.�.�q�1�;�;�A�A�;�F����������C��(��
���f�h�/��7�7�A�q�6�?�����
�#��
���f�h�/rc��ttjjd�j	d�gd�gd�ggd�gd�g��}|j
dd	gj
d�}|jd
}tj||�y)NrrYrZr[r^r_r`rrr9r�r�s    r�test_loc_multiindex_intsz*TestMultiIndexLoc.test_loc_multiindex_ints�st��
��I�I�!�!�!�$�4�4�V�<��
�+��k�*�
��
�7�7�A�q�6�?�,�,�Q�/��������
���f�h�/rc��ttjjd�j	d�gd�gd�ggd�gd�g��}tjtd�	�5|jdddd�y#1swYyxYwrXrcrgs  r�(test_loc_multiindex_missing_label_raisesz:TestMultiIndexLoc.test_loc_multiindex_missing_label_raises�si��
��I�I�!�!�!�$�4�4�V�<��
�+��k�*�
���]�]�8�6�
2�	��F�F�1�I�	�	�	�rizkey, posrr9rrrc��ttjjd�j	d�gd�gd�ggd�gd�g��}tjtd�	�5|j|ddd�y#1swYyxYw)
NrrYrZr[r^r_r`znot in indexrarc)r0rP�posr3s    r�&test_loc_multiindex_list_missing_labelz8TestMultiIndexLoc.test_loc_multiindex_list_missing_label�sk����I�I�!�!�!�$�4�4�V�<��
�+��k�*�
���]�]�8�>�
:�	��F�F�3�K�	�	�	�ric	���ttd�tjddgddgddgg���}t	j
td	�
�5|jdddd�t	j
td�
�5|jd
ddd�t	j
td�
�5|jdddd�y#1swY�pxYw#1swY�IxYw#1swYyxYw)Nr\�a�b�c�d�e�frz^\('a', 'b'\)$ra�r�r�z^\('a', 'd', 'g'\)$)r�r��gzToo many indexers)r�r�r�r�)	rr.rr�rdrerfr+r)r0�ss  r�(test_loc_multiindex_too_many_dims_raisesz:TestMultiIndexLoc.test_loc_multiindex_too_many_dims_raises�s�����!�H��)�)�C��:��S�z�C��:�*N�O�
��
�]�]�8�+<�
=�	�
�E�E�(�O�	�
�]�]�8�+A�
B�	!�
�E�E�-� �	!�
�]�]�=�0C�
D�	&�
�E�E�$�%�	&�	&�		�	��	!�	!��	&�	&�s$�C
�C�4C%�
C�C"�%C.c��td�D�cgc]}dt|�z��}}td�D�cgc]}dt|�z��}}tj||g�}dtj
j
d�jd�zdz}t||�	�}||}tj||�tt	jd
�jdd�tjgd�gd
�g���}|jddgfdd�f}|jddg}tj||�ycc}wcc}w)Nr�	Attributer&�Valueg�������?r)r]r&g�?�r/r;����rrrr9rwr)r.�strrr�rrr r!rrGrI�arange�reshaper+)r0r��
attributes�attribute_valuesrr3rtrUs        r� test_loc_multiindex_indexer_nonez2TestMultiIndexLoc.test_loc_multiindex_indexer_none�s#��5:�!�H�=�q�k�C��F�*�=�
�=�6;�A�h�?��G�c�!�f�,�?��?��'�'��5E�(F�G��
�2�9�9�(�(��+�;�;�K�H�
H�3�
N��
�r�5�
)���J���
���f�b�)���I�I�b�M�!�!�"�a�(��)�)�<��*C�D�
��
�6�6�A�q�6�)�Q�,�'������A����
���f�h�/��%>��?s
�E�Ec��ttjdd��tjtd�gd�g��}|jdd�dd�f}|jdd	�dd�f}tj||�|jdd	�dd�f}tj||�|jdd�dd�f}tj||�ttjdd��tjtd�gd�g��}|jgd
�}|jdd	d�dd�f}tj||�y)N�r�rBr&�r�r�r�r�r�rr9)rr*r\r;ro�r)
rrr�rr�r.r+rGrqrJ)r0r�rUrts    r�test_loc_multiindex_incompletez0TestMultiIndexLoc.test_loc_multiindex_incomplete�s5��
��I�I�b��(��#�#�U�1�X��$?�@�
���5�5��C��G��$�����q��s�C��G�|�$��
���v�x�0����r��r�3�s�7�{�#��
���v�x�0����q�r�3�s�7�{�#��
���v�x�0�
��I�I�b��(��#�#�U�1�X��$?�@�
���6�6�/�0�����q��1�u�c�#�g�~�&��
���v�x�0rc��|}ttjjd�j	t|��|��}|jD]}||�	y)Nrr)rrrr r!�len�values)r0r�single_levelr��ks     r�test_get_loc_single_levelz+TestMultiIndexLoc.test_get_loc_single_levelsT��.����I�I�!�!�!�$�4�4�S��5F�G��
���$�$�	�A�
�a�D�	rc��tjgd�ddgg�}ttjjd�j
d�||�}|jdd�dd�f}|}tj||�tjgd�ddgg�}ttjjd�j
d�||�}|jd	d
�dd�f}|jdd}tj||�|jddd�f}|jdd}ddg|_tj||�|jdd�df}|d}tj||�y)
N)rr*r\r�r�r)rrrr\�r]��r�r�r]r)rr�rrrr r!r+rGrIrJr)r0rr3rtrUs     r�test_loc_getitem_int_slicez,TestMultiIndexLoc.test_loc_getitem_int_slice!sN���'�'��S�#�J�(?�@��
�r�y�y�,�,�Q�/�?�?��G��PU�
V������!��Q������
���f�h�/��'�'���S�z�(B�C��
�r�y�y�,�,�Q�/�?�?��G��PU�
V������2��q��!���7�7�1�2�;��
���f�h�/�����A�����7�7�1�Q�<���s����
���f�h�/�����2�����b�6��
���f�h�/r�indexer_type_1�indexer_type_2c�0��d��gd�}gd�}tj||g�}ttjt|�d��|dg��}dd	gd
dgf}||f}t
�fd�t||�D��}	|tus|tur6tjtd
��5|j|	dfddd�y|j|	df}
tgd�dtj|���}tj|
|�y#1swYyxYw)Nc��|tjk(rtj|�S|tk(rt|�S||�S�N)r�ndarrayrL�slice)�indexer_typer@s  r�convert_nested_indexerzQTestMultiIndexLoc.test_loc_getitem_nested_indexer.<locals>.convert_nested_indexerFs9���r�z�z�)��x�x��~�%��u�$��d�|�#���%�%rr�rwr�rB�Data�rr/r]r�rrc3�6�K�|]\}}�||����y�wr�r)�.0r�r�r�s   �r�	<genexpr>zDTestMultiIndexLoc.test_loc_getitem_nested_indexer.<locals>.<genexpr>Ys$�����
���a�
#�<��3�
�s�zas an indexer is not supportedra)rrr9r&��namer)rr�rrr�r��tuple�zip�setrdre�	TypeErrorr+rrGrq)
r0r�r�r�r�rr3r@�types�indexerrtrUr�s
            @r�test_loc_getitem_nested_indexerz1TestMultiIndexLoc.test_loc_getitem_nested_indexer;s���	&�
�����'�'��A��/��
��I�I�c�%�j��0���x�
���R��1�a�&�!����0���
�#&�u�d�#3�
�
���S� �N�c�$9����y�0P�Q�
(����w���'�
(�
��V�V�G�V�O�,�F���v�Z�-D�-D�T�-J�
��	���v�x�0�
(�
�s�&D�Dc��tjddg�}|ddg|��}d|jd<|ddg|��}tj||�y)N�r�rk�r�rkrrrr�r�)r�from_tuplesr+rG�assert_equal)r0�frame_or_series�mi�objrUs     r�)test_multiindex_loc_one_dimensional_tuplez;TestMultiIndexLoc.test_multiindex_loc_one_dimensional_tuplejsR��
�
#�
#�Z��$<�
=���q�!�f�B�/��������"�A�q�6��4��
����X�&rr�r�r�c��tjddg�}tddg|��}d|j|dd�f<tddg|��}t	j
||�y)Nr�r�rrrr�rr�rr+rGrI)r0r�r�r�rUs     r�-test_multiindex_one_dimensional_tuple_columnsz?TestMultiIndexLoc.test_multiindex_one_dimensional_tuple_columnsrs[���
#�
#�Z��$<�
=����A��b�)��������
���a��V�2�.��
���c�8�,rzindexer, exp_valueN��?�rrc���tjddg�}tddgddgg|ddg�	�}d
|j|ddgf<tgd
�dd||gg|gd��	�}t	j
||�y)Nr��rr9rrrr9r�r�r�r�r�r�)rrr�r��r�r�r�r�r�)r0r��	exp_valuer�r3rUs      r�)test_multiindex_setitem_columns_enlargingz;TestMultiIndexLoc.test_multiindex_setitem_columns_enlarging{s���
�
#�
#�V�V�$4�
5��
��A���A��'�r�C��:�
F��&)����w��c�
�"�#��
��1�i��;�<��(�
��
	���b�(�+rc�\�tjtjdd��t	ddg�g�}td|��}td|dd��}tj||gd��}|j�}|jd	d
}tj||�t||d��}|jd	d
}tj||�tj||j|j�gd��}|jd	d
}tj||�y)N�20110101r��periodsr�r�rrr�z
2011-01-01z
2011-01-02�rr)rr��pd�
date_rangerr�concatr,r+rGrIrrpr)r0�midx�ser1�ser2r3rUrts       r�"test_sorted_multiindex_after_unionz4TestMultiIndexLoc.test_sorted_multiindex_after_union�s����&�&�
�]�]�:�q�
1�5�#�s��3D�E�
���a�t�$���a�t�B�Q�x�(��
�Y�Y��d�|�!�
,���7�7�9������\�2��
���f�h�/�
�4�D�)�
*������\�2��
���f�h�/�
�Y�Y��d�l�l�4�:�:�6�7�a�
@������\�2��
���f�h�/rc	�f�ttjtd�td�td�g�dg��}|jt
jdd�ddd�f}ttjtd�td�g�dg��}tj||�y)N�ab�cdr��Valr�r�)	rrr�r-r+r�s_rGrI)r0r3rRrUs    r�test_loc_no_second_level_indexz0TestMultiIndexLoc.test_loc_no_second_level_index�s���
��)�)�4��:�t�D�z�4��9�*M�N��G�
���f�f�R�U�U�1�c�1�9�%�&����)�)�4��:�t�C�y�*A�B�U�G�
��	���c�8�,rc��tgd�gd�gd�gd�d��}tjtd��5|jdddd�y#1swYyxYw)	Nr�)r�r�r�)r��hr�)r�r��l)r�)rr�rr)rr9z(1, 4)ra)r�rr9)rrdrerfr+rgs  r�test_loc_multi_index_key_errorz0TestMultiIndexLoc.test_loc_multi_index_key_error�sN��
�'�'�'�'�	
�
���]�]�8�9�
5�	��F�F�9��	�	�	�s�A
�
A)%�__name__�
__module__�__qualname__r4rVrhrur�r�r�r�r�rd�mark�parametrizer�r�r�r�r�r�r-r�r�r�rr�rrr�r�r��nanr�r�rrrrrr$r$#sz��	.�-�:	�1�1�>1�<0�60���[�[���Z�A�q�6�A�q�6�*:�a�S�"�I��A��PR�|�)T�U�	�V�	�&�0�.1�8�0�4�[�[����4���U�B�J�J���N���[�[����4���U�B�J�J���N��'1���'1�R'��[�[���Y��#��8�-�9�-��[�[�����d��S�1�F�B�F�F�3C�D��
,��
,�0�(
-�rr$zindexer, posrkr�D�Er
r�bahc�h�tjgd�gd�gddg��}ttjdd��|�	�j�}|j|}|jd
k(rB|gk7r=tjtt|���5|j|ddd�y|td�dd
gfk(r4tjtd��5|j|ddd�y|j|}tj||�y#1swYyxYw#1swYyxYw)Nrx�r
rr�oner(rDrzr�rBrrrar
rz'bah')rr�rrr�rKrJ�sizerdrerfr�r+r�rGrq)r�r��idxrsrUrts      r�7test_loc_getitem_duplicates_multiindex_missing_indexersr�s���
!�
!�	�/�0�����C�����1�G�,�C�
8�
C�
C�
E�C��x�x��}�H��}�}���g��m�
�]�]�8�3�w�<�
8�	��G�G�G��	�	�	�U�4�[�5�%�.�1�	1�
�]�]�8�7�
3�	��G�G�G��	�	�����!��
���v�x�0�	�	��	�	�s�D�D(�D%�(D1�columns_indexerc��tjgd�ddgf�}ttjjd�j
d�td�|��}|jdd	�
�}ttd�|jg�d��}|jdd�|f}tj||�y)Nr�alpha�betar�r&rr&r�rr)�levelr�)
rr�rrrr r!r.rKrpr+rGrI)r�multi_indexr3rUrts     r�4test_loc_getitem_duplicates_multiindex_empty_indexerr�s����)�)�+@�7�F�BS�*T�U�K�	�
�	�	���a� �0�0��8��A�h��
�B�

���Q�Q��	'�B��u�Q�x��1D�1D�R�1H��1K�L�H�
�V�V�A��&�
'�F����&�(�+rc���ttjtjgddggt	j
ddg�ddg��}|jd}tj}||k(sJ�y)	N�mean�median��functsr)r"r �functionr�r`)r#r!)rrrr rr�r+�r3rtrUs   r�=test_loc_getitem_duplicates_multiindex_non_scalar_type_objectr%�sm��
�
�'�'�2�9�9�	���1�2��&�&�(:�<P�'Q�R��6�"�
�B�
�V�V�2�
3�F��w�w�H��X���rc��ttjd�tjd�tjj	d�jd�tjj	d�jd�d��j
ddg�}|jd}|jddd�f}tj||�y)Nr]rr�r�r�)rr)
rrr�rr r!rFr+rGrq)r3rUrts   r�!test_loc_getitem_tuple_plus_slicer'�s���	����2�����2�����&�&�q�)�9�9�"�=����&�&�q�)�9�9�"�=�		
�
��i��c�
����v�v�d�|�H�
�V�V�F�A�I�
�F����6�8�,rc��|}|jd}|dd}|jjd�|_tj||�y)Nr���r)r+rr�rGrI)r"r3rtrUs    r�test_loc_getitem_intr*sE��	.�B�
�V�V�A�Y�F��"�#�w�H��^�^�-�-�a�0�H�N����&�(�+rc��|}tjtd��5|jdddd�y#1swYyxYw)Nz^3$rar)rdrerfr+)r"r3s  r�%test_loc_getitem_int_raises_exceptionr,s5��	.�B�	���x�v�	.��
���q�	����s	�7�Ac���|}tjtd��5|jdddd�d|jd<d}|j	�jd}||k(sJ�y#1swY�?xYw)Nz^\('bar', 'three'\)$ra))r�threer)r)rdrerfr+rK)r1r3rUrts    r� test_loc_getitem_lowerdim_cornerr/su��	)�B�
���x�'>�	?�&�
���$�%�&�%&�B�F�F� �!��H�
�]�]�_�
 �
 �!6�
7�F��X����&�&�s�A,�,A5c��tdtd�tjdgdg���}d|d<|j|j
dd�df<t
jtj|d�tj|j��ttjtd�tjgd	����}|j�}tjd�|j
dd�d
f<tjd�|jdd�df<t
j ||�y)N�string�abcd�Main)�anotherrr�r��labelsr9))rk�1)rk�2)r)r6r)r)rr-rr�rr+rG�assert_numpy_array_equalr�asarrayrr.r�r,r�rJrI)r3rUs  r�$test_loc_setitem_single_column_slicer:"s���	���6�l��'�'�&��3E�(F�G�
�B�
�B�x�L��(�(�B�F�F�1�h�;������
�
�2�h�<� 8�"�*�*�R�X�X�:N�O�
�
����A�h��&�&�'K�L�
�B�
�w�w�y�H��Y�Y�q�\�B�F�F�1�c�6�N��)�)�A�,�H�M�M�!�Q�$�����"�h�'rc
�d�ddtjfddtjfddtjfddtjfdddd	tjfdd
tjfg}ttjd�t	gd��tj|gd
�����}|jdjd}ttjd�t	tjg|sdndd��t	gd����}tj||�y)NzGood Thingsry�Rz
Bad Things�T)�Okay Things�Nr))r>r?r
r>r)r
)r\r9)�d1�d2�d3�d4)�u1�u2�u3rDr`r�objectzstring[pyarrow_numpy]rF)rCr�r�)
rrr�onesrrr�r+rGrI)�using_infer_string�tupsr3rtrUs     r�test_loc_nan_multiindexrK9s���
��R�V�V�$�	��R�V�V�$�	�s�B�F�F�#�	�s�B�F�F�#�!�!�	��R�V�V�$�	��R�V�V�$�	�D�
�
������.�/��$�$�T�1C�D�
�B�
�V�V�M�
"�
&�
&�s�
+�F��
������
�V�V�H�"4�(�:Q��
�
�.�/��H����&�(�+rc�v�tjddd��}d}tj||fd��}t	|d�	�}|j
d
}|j
|ddfd
f}t
j|�sJ�t
j|�sJ�|j
d
}t
j|�sJ�y)N�2013Q1�2013Q4�Q)�freq)�Wi�i
)�Period�CVRrD)	�OMS�OMK�RES�	DRIFT_IND�
OEVRIG_IND�FIN_IND�VARE_UD�LOEN_UD�FIN_UDr�))rMrQrTrrQrT)r��period_rangerr�rr+r�isnan)r�r�rr3rt�alts      r�test_loc_period_string_indexingr`Ws���
����(��5�A��A�
�
!�
!�1�a�&�0A�
B�C�	��

�
�B��V�V�+�
,�F�
�&�&�!�A�$���u�$�
%�C�
�8�8�C�=��=��8�8�F����
�&�&�(�
)�C�
�8�8�C�=��=rc
���tjddg�}tj||gddg��}t	ddgdd	gd
dgddgg|d
dg��}|j
|d|jjd�dkDfd
f}tdgd
tjtjd�tjd�fgddg����}tj||�y)Nz
2017-05-04z
2017-05-05�Idx1�Idx2rDrrrr9r&rr*�C1�C2�rOrr/rr�)
r��to_datetimerr�rr+r�get_level_valuesrr��	TimestamprGrq)�dt_idx�m_idxr3rtrUs     r�test_loc_datetime_mask_slicingrlws���
�^�^�\�<�8�
9�F��#�#�V�V�$4�V�V�<L�M�E�	��!�f�q�!�f�q�!�f�q�!�f�
-�U�T�4�L�
�B��V�V�V�A�Y����!:�!:�1�!=��!L�N�PT�T�
U�F��	
��
��$�$��l�l�<�(�"�,�,�|�*D�
E�F��6�"�
��H����6�8�,rc���tjd�}tdtjd|fgddg��d��}|j
dd�|gf}t
j||�y)N�2000rr�r�rDr�)rr�)r�rirrr�r+rGrq)�datersrts   r�&test_loc_datetime_series_tuple_slicingrp�s`��
�<�<���D�
�	��$�$�s�D�k�]�3��*�E�
��C�
�W�W�Q���Y�
�F����6�3�'rc�Z�tddgddgddgddggtjgd�dd	g�
�ddg�
�}tjddgdd	g�
�}|j|dd�f}tddgddgddggtjgd�dd	g�
�ddg��}t	j
||�y)Nr�rrr�r�r)r�)rr�rrrrrrorD�author�pricerfr�rr)r�rrrrr�)rrr�r+rGrI)r3rrtrUs    r�test_loc_with_mi_indexerru�s���	��A�h��a��3��(�S�!�H�
5��$�$�,�W�f�4E�
��7�#�
�B��
 �
 �&�&�!1�'�6�9J�
K�C�
�V�V�C��F�^�F��

�q��C��8�c�1�X�&��$�$�%=�g�v�EV�W��7�#��H�
���&�(�+rc�~�tjddd��}ttd�|��}|j	�}||d<|jdd�	�}|jjd
k(sJ�|jj|d�|j|d}|jdgjd�}tj||�|d}|jjd
k(sJ�|j|d}|jdgjd�}tj||�y)Nz
2016-01-01rz
US/Pacific)r��tzrrrT)�append)Nr)r�r�rr.�to_framerFrr�get_locr+rJr�rGrIrq)�dtirsr3�df2rtrUr�s       r�test_loc_mi_with_level1_named_0r}�s��
�-�-��a�L�
A�C�
��q���
%�C�	����B��B�q�E�
�,�,�q��,�
&�C��9�9�?�?�i�'�'�'��I�I���c�!�f��
�W�W�S��V�_�F��x�x���}�&�&�t�,�H����&�(�+��q�6�D��:�:���y�(�(�(�
�X�X�c�!�f�
�F��y�y�!��~�'�'��-�H����6�8�,rc	�z�tgd�gd�gd�gd�gd�gd�gdjd��	�}|jd
dg�j�}|jdtd
�fdd�fj
d�}|jdjtd
�dd�f}tj||�y)N)z20160525 13:30:00.023�MSFT�51.95r�)z20160525 13:30:00.048�GOOGz720.50z720.93)z20160525 13:30:00.076�AAPLz98.55z98.56)z20160525 13:30:00.131r��98.61�98.62)�20160525 13:30:00.135rz51.92r�)r�r�r�r�ztime,ticker,bid,ask�,r��ticker�timer�z2016-05-25 13:30:00r)	r�splitrFrKr+r�r�rGrI)r3r|rRrUs    r�test_getitem_str_slicer��s���	�?�A�?�?�?�?�
	
�&�+�+�C�0�

�B�
�,�,��&�)�
*�
5�
5�
7�C�

�'�'�6�5�!6�7�8�!�;�
<�
F�
F�q�
I�C��w�w�v��"�"�5�)>�#?��#B�C�H����#�x�(rc��tjgd�dd��}gd�}gd�}tj|||g�}t	td�|tj��}|j|d	d
df}|dk(sJ�y)
N)z
20181101 1100z
20181101 1200z
20181102 1300z
20181102 1400�datetimer
)r�rP)rkrk�Z�W)r)ryrO�Fr9rrrkr)g)	r��PeriodIndexrrMrr.rr�r+)�pi�lev2�lev3r�rsrts      r�!test_3levels_leading_period_indexr��sw��	���L�
�
�
�B�
 �D��D�	�	�	��T�4� 0�	1�B�
��q���2�:�:�
6�C�
�W�W�b��e�S�#�&�
'�F��S�=��=rc��eZdZd�Zd�Zd�Zy)�TestKeyErrorsWithMultiIndexc��ttjd�jdd�gd���}|j	ddg�}tjtd�	�5|jd
ddd�y#1swYyxYw)Nr;r9rrxr�rkr)r6ra�rr)	rrr�r�rFrdrerfr+)r0r3r|s   r�!test_missing_keys_raises_keyerrorz=TestKeyErrorsWithMultiIndex.test_missing_keys_raises_keyerror�sc��
�r�y�y��}�,�,�Q��2�O�
L���l�l�C��:�&��
�]�]�8�3�
/�	��G�G�F�O�	�	�	�s�!A:�:Bc���tdtjddggdz���}tjt
d��5|jdddd�y#1swYyxYw)	Nr�rrrrz\(0, 3\)ra)rr)rrr�rdrerfr+)r0rss  r�!test_missing_key_raises_keyerror2z=TestKeyErrorsWithMultiIndex.test_missing_key_raises_keyerror2�sQ���R�z�6�6��A��x�!�|�D�E��
�]�]�8�;�
7�	��G�G�D�M�	�	�	�s�A�A$c��tjtjgd��tjgd��tjgd��ggd���}t	tj
j
d�jd�|��}d	}tjt|�
�5|jddtd�fdd�fddd�tjt|�
�5|jjddtd�f�ddd�tjtd
�
�5|jddd�fddd�y#1swY��xYw#1swY�MxYw#1swYyxYw)N�r�r�r�r�)r6r7r7�3)r�r�r�r�)rr(r.rDr�r9rrz'\('b', '1', slice\(None, None, None\)\)rar�r6z\('b', '1'\))r�r6)rrMrrLrrr rdrerfr+r�r�get_locs)r0r�r3�msgs    r�test_missing_key_combinationz8TestKeyErrorsWithMultiIndex.test_missing_key_combination�s'��
�
#�
#����-�.����-�.����-�.�
�
*�

���r�y�y�,�,�Q�/�6�6�v�>�b�
I��8��
�]�]�8�3�
/�	/��F�F�C��e�D�k�*�A�-�.�	/�
�]�]�8�3�
/�	7��H�H���s�C��t��5�6�	7�
�]�]�8�?�
;�	"��F�F�:�q�=�!�	"�	"�		/�	/��	7�	7��	"�	"�s$�2 E�5(E)�E5�E&�)E2�5E>N)rrr	r�r�r�rrrr�r��s����"rr�c�l�|}|d}|d}|jdd}tj||�y)Nrk)rnr&)r+rGrq)rrr3rsrtrUs     r�test_getitem_loc_commutabilityr�	s;��	8�B�
�S�'�C�
��\�F��v�v�g��s�#�H����6�8�,rc��tgd�ddtjggd�d��jddg�}|jdg}tdd	gitjd
gddg����}tj||�|jd}tdd	gitd	gd�
���}tj||�y)N)rrr&r�r�)rr9r&)r>�ind1�ind2r�r�r>r)r�rrDr�r�)
rrrrFr+rr�rGrIrr$s   r�test_loc_with_nanr�s���	��C��b�f�f�#5�y�I�
��i��� �!���V�V�S�E�]�F��	����J�2�2�H�:�f�f�EU�V��H����&�(�+�
�V�V�C�[�F��%�!���U�A�3�V�-D�E�H����&�(�+rc���tgd�ggd���jgd��}tjtd��5|j
dddd�y#1swYyxYw)Nr�r�r�r�z\(2\.0, 2\.0, 3\.0\)ra)�@r��@)rrFrdrerfr+�r3s r�test_getitem_non_found_tupler�!sQ��	�L�>�+?�	@�	J�	J��
�B�
���x�'>�	?� �
����� � � �s�A�A!c��tjdd��}tj|g�}|j	d�tddd�k(sJ�|j	d�tddd�k(sJ�|ddd�j	d�}|ddd�j	d�}||k(sJ�|j
d�j	d�}|j
d�j	d�}||k(sJ�|j|�j	d�}|j|�j	d�}tj|j�d|�y)Nz
2001-01-01�dr�z2001-01r�r)r�r�rrMrzr��repeatrxrGr8�nonzero)rr�r+rUs    r�test_get_loc_datetime_indexr�*s&���M�M�,��4�E�	�	�	���	(�B�
�:�:�i� �E�!�R��$6�6�6�6��=�=��#�u�Q��D�'9�9�9�9�
�S�q�S�'�/�/�)�
$�C��S�q�S�z�!�!�)�,�H��(�?��?�
�)�)�A�,�
�
�y�
)�C��|�|�A��&�&�y�1�H��(�?��?�
�)�)�B�-�
�
�	�
*�C��|�|�E�"�*�*�9�5�H�������
�a� 0�(�;rc�.�tjddgddgg�}tddgddgdd	gd
dgg|��}dddgf}tjd
dgddgg�|j
|dd�f<tddgd
dgdd	gd
dgg|��}t
j||�y)Nr�r�rrrrr9r&rr*r\rrzr]r<r;)rr�rrrLr+rGrI)r�r3r�rUs    r�,test_loc_setitem_indexer_differently_orderedr�@s���	�	 �	 �3��*�q�!�f�!5�	6�B�	�Q��F�Q��F�Q��F�Q��F�3�2�	>�B��Q��F�m�G����A�r�7�R��H�"5�6�B�F�F�7�A�:���2�r�(�Q��G�a��V�a��V�<�B�G�H����"�h�'rc��tddgddgddgddgggd	�gd
�gddg�
�}|jtd�ddgfdd�f}tddgddgddgddgggd�gd�gddg�
�}tj||�y)Nrrrr9r&rr*r\r��rrrrr�r�r�)r�r�r�r�)rrrr)rr+r�rGrIr$s   r�5test_loc_getitem_index_differently_ordered_slice_noner�Ks���	�
�Q��!�Q��!�Q��!�Q��(�#�\�2��c�
�
�B�
�V�V�U�4�[�1�a�&�)�1�,�
-�F��
�Q��!�Q��!�Q��!�Q��(�#�\�2��c�
��H�
���&�(�+rr�)rrr*rrrr\r*)rrr*rrr\c�v�tdgdztjgd��dg��}|jt	d�|fdd�f}tdgdzgd�gd�gdg��}tj||�|j|jj|d��dd�f}tj||�y)	Nrr\)rrr�)rr*r�r)rr)rr\)rr*r�r�)rrrrrrrr)rrrr*r*rrr\)r)	rrr�r+r�rGrIr�isin)r�r3rtrUs    r�@test_loc_getitem_index_differently_ordered_slice_none_duplicatesr�[s���
�	
��a���$�$�L�
���
�B��V�V�U�4�[�'�*�A�-�
.�F��	
��a��'�)A�B����H�
���&�(�+�
�V�V�B�H�H�M�M�'��M�3�Q�6�
7�F����&�"�%rc��tjdgdgdgggd���}tddgi|��}|jdd	g�}|jddd�df}tj||�tdg|��}|jddd�df}tdgtdgd
����}tj||�y)Nr�r��zr�rDr�rrrr�r�)
rrMrr�r+rGrIrrrq)r�r3rUrtrss     r�3test_loc_getitem_drops_levels_for_one_row_dataframer�qs���	�	�	�#�����u� 5�_�	M�B�	�C�!��:�R�	(�B��|�|�Q��F�#�H�
�V�V�C��C�K�
 �F����&�(�+�
�!��B�
�C�
�W�W�S�!�S�[�
!�F��q�c���u�3�!7�8�H����6�8�,rc�:�tjgd�ddgg�}ttjd�|��}|j
dd�ddgf}ttjd�tjgd	����}tj||�y)
Nrxrrrr�r)rk)r&r9))r)r)r)r)rkr)rkr)	rr�rr�zerosr+r�rGrI)�colsr3rtrUs    r�$test_mi_columns_loc_list_label_orderr�s}���"�"�O�a��V�#<�=�D�	�2�8�8�F�#�T�	2�B�
�V�V�A��S�z�M�
"�F��
������&�&�'O�P��H����&�(�+rc�N�ttjd�jd�gd�gd�ggd�gd�g��}dd	g|j_d
dg|j_tjtd�
�5|jddgdfddd�y#1swYyxYw)Nr;r�r�r�)�Ohior��Colorado)�Green�Redr�r��key1�key2�state�colorz\[2\] not in indexrar�rr�)rrr�r�rrr/rdrerfr+)r2s r�$test_mi_partial_indexing_list_raisesr��s����
�	�	�"�
���f�%�#�\�2�-�/H�I�
�E�
 ��(�E�K�K��"�G�,�E�M�M��	���x�'=�	>�(�
�	�	�3��(�J�&�'�(�(�(�s�>B�B$c���ttd�tjddgddgg���}t	j
td��5|jd	d
gddd�y#1swYyxYw)Nr9rrr�r�rz\['not' 'found'\] not in indexra�not�found)rr.rr�rdrerfr+)r�s r�(test_mi_indexing_list_nonexistent_raisesr��s_���u�Q�x�z�6�6��A���c�
�7K�L�M�A�	���x�'I�	J� �	���u�g��� � � �s�A&�&A/c	��tgd�gd�gddgtjddgddgg���}d	|jd
<d|jd<d|jd
<tgd�gd�gd�dtj
tj
tj
gggd�tjddgddgg���}t
j||�y)N)rrr&r)rr9r*r\r�rrrkr)r�r�r�r)r��rrk)r�r�)r�r�g@g@)r�r:g@g @)r�g�r�r�r�)r�r�r�r�)rrr�r+rrrGrI)rtrUs  r�'test_mi_add_cell_missing_row_non_uniquer��s���
�	�|�$��C�j��'�'�!�Q��#�s��(<�=��F�
�F�J�J�s�O� !�F�J�J�}�� !�F�J�J�}��� � ��
�"�&�&�"�&�&�"�&�&�)�		
�#��'�'�!�Q��#�s��(<�=�	�H����&�(�+rc�B�tddd�tjdgdggddg���	�}|jd
}|dk(sJ�t	|t
j�sJ�|jdgdfjd
}|dk(sJ�t	|t
j�sJ�y)Nr�rr�rr9r�r�rDr)r�r�r�r�r)rrrMr+�
isinstancerr�rJ)r3rts  r�$test_loc_get_scalar_casting_to_floatr��s���	����*�"8�"8�1�#��s��C�QT�:�"V�
�B��V�V�K�
 �F��Q�;��;��f�b�h�h�'�'�'�
�V�V�V�H�c�M�
"�
'�
'��
*�F��Q�;��;��f�b�h�h�'�'�'rc�2�tjddgddggddg��}t|tj��}|j
d}ttjtjgtddgd�	��
�}tj||�y)Nr�r�rkr)rrrDrr�r)
rr�rrr�r+rrrGrq)r�s2rtrUs    r�)test_loc_empty_single_selector_with_namesr��sv��
�
!�
!�C��:��S�z�":�1�a�&�
I�C�	�c����	,�B�
�V�V�C�[�F��r�v�v�r�v�v�&�e�S�#�J�Q�.G�H�H����6�8�,rc���tgd�gd�td�d��}|jddg�}tjt
d��5|jd	ddd�y#1swYyxYw)
N)r�r���r��,r�)r]r]r��r��!rrxrkr)z^1$ra)r�r)rr.rFrdrerfr+r�s r�'test_loc_keyerror_rightmost_key_missingr��sc��
�/�)��q��	
�
�B�
���s�C�j�	!�B�	���x�u�	-��
���x�����s�A$�$A-c�|�tjddg�}tddg|��}|jd}|dk(sJ�y)Nr�)r)r9r&rrr)rr�rr+)r�rsrts   r�*test_multindex_series_loc_with_tuple_labelr��sA��	�	�	��� 5�	6�B�
�!�Q��r�
"�C�
�W�W�[�
!�F��Q�;��;r)8�numpyrrd�
pandas.errorsrr�pandasr�rrrr�pandas._testing�_testingrG�fixturerr"r$r
rr��
IndexSlicerrr%r'r*r,r/r:rKr`rlrprur}r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�rrr�<module>r�s���
��
�������������T��T�R�R�j������	�R��

���a���

�s��R��

�s��R��

����	���q�5�'�z�	"�E�!�T�1�$5�6�	���q�5�%�.�(�	)�5��D�!�+<�=���1��1�*�����*�b�%��+�->�%��"�
�,N�O�
,�P�
,� 
�-�,���(�.,�<�@-�&	(�,�&-�0)�(
� !"�!"�H-�
,�  �<�,(�
,� ������%=�?Q�$R�S�&�T�&�*-�	,�
(� �,�.
(�-��r

Youez - 2016 - github.com/yon3zu
LinuXploit