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/groupby/__pycache__/

Upload File :
current_dir [ Writeable ] document_root [ Writeable ]

 

Command :


[ Back ]     

Current File : C:/Python312/Lib/site-packages/pandas/tests/groupby/__pycache__/test_timegrouper.cpython-312.pyc
�

/	�gۇ����dZddlmZmZddlZddlZddlZddlZddlm	Z	m
Z
mZmZm
Z
mZmZmZddlmZddlmZddlmZej0d��Zej0d��ZGd	�d
�Zy)z5
test with the TimeGrouper / grouping with datetimes
�)�datetime�	timedeltaN)�	DataFrame�
DatetimeIndex�Index�
MultiIndex�Series�	Timestamp�
date_range�offsets)�Grouper)�
BinGrouperc���tgd�tddddd�tddddd�tdddd	d�tddd
dd�tjtddddd�gd
��}|S)z�
    DataFrame used by groupby_with_truncated_bingrouper, made into
    a separate fixture for easier reuse in
    test_groupby_apply_timegrouper_with_nat_apply_squeeze
    ������	r��rr�
rr�
�r��)�Quantity�Date)rr
�pd�NaT)�dfs �GC:\Python312\Lib\site-packages\pandas/tests/groupby/test_timegrouper.py�frame_for_truncated_bingrouperr"sy��
�+��$��1�b�!�,��$��1�b�!�,��$��A�r�1�-��$��A�r�1�-�����$��1�b�!�,�
�
	
�
�B��I�c���|}tdd��}|j|�}t|jj�t|jj
�k7sJ�|S)a.
    GroupBy object such that gb._grouper is a BinGrouper and
    len(gb._grouper.result_index) < len(gb._grouper.group_keys_seq)

    Aggregations on this groupby should have

        dti = date_range("2013-09-01", "2013-10-01", freq="5D", name="Date")

    As either the index or an index level.
    r�5D��key�freq)r
�groupby�len�_grouper�result_index�group_keys_seq)r"r �tdg�gbs    r!�!groupby_with_truncated_bingrouperr04sU��
(�B�
�f�4�
(�C�	���C��B��r�{�{�'�'�(�C����0J�0J�,K�K�K�K�
�Ir#c�d�eZdZd�Zej
j
dddg�d��Zd�Zej
j
dgd��d	��Z	d
�Z
d�Zd�Zd
�Z
d�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Zej
j@d ��Z!y!)"�TestGroupByc��tdj�gd�tddddd�tddddd�tdd	dd
d�tdd	dd	d�tddd
dd�tddd
dd�gd��}|jd��}||fD�]{}|j	dg�}tddddd|jj��}tddd�|��}|jdti�}d|jd<d|jd<d|jd <tjgd!�d"�#�|jgd$�df<|jd�j�}tj ||�|j#�}|j%t'd�%��j�}tj ||�|j%t'd�%��j�}	tj |	|���~y)&NzCarl Carl Carl Carl Joe Carlrrrrrrrrrr�rr��Buyerrrr��byr�20130901�20131205r%�left)r(�name�	inclusive�unit)r6r��indexr6�CarlCarlCarl)rr�CarlCarl)�r�Joe)rr)�rCr�int64��dtype)rrCr�r()r�splitr�sort_values�	set_indexrr@r>�astype�object�iloc�np�array�resample�sum�tm�assert_frame_equal�
sort_indexr)r
)
�self�df_original�df_reorderedr �exp_dti�expected�result1�	df_sorted�result2�result3s
          r!�test_groupby_with_timegrouperz)TestGroupBy.test_groupby_with_timegrouperLs��� �7�=�=�?�/��T�1�a��Q�/��T�1�a��Q�/��T�2�q�"�a�0��T�2�q�"�a�0��T�2�q�"�a�0��T�1�a��Q�/�
�
�

�� #�.�.�*�.�=����-�	5�B����v�h�'�B� ����� ��X�X�]�]�
�G�!���+���H�
 �����'8�9�H�"0�H�M�M�$��",�H�M�M�$��#(�H�M�M�%� �+-�8�8�J�g�+N�H�M�M�*�a�-�(��k�k�$�'�+�+�-�G��!�!�'�8�4��
�
��I��'�'��T�(:�;�?�?�A�G��!�!�'�8�4��j�j��d�!3�4�8�8�:�G��!�!�'�8�4�=	5r#�should_sortTFc�>�tdj�dj�gd�tddddd�tddddd�tdd	dd
d�tdd	dd	d�tddddd�tdddd
d�gd��}|r|jdd��}|j	dd��}|jt
d���}|jsJ�t|jt�sJ�|j}t|t�sJ�t|�dk(sJ�y)NzA A A A A BzCarl Mark Carl Joe Joe Carl)rrr�rrrrrrrrrrr4r��Branchr6rrrF�r8�	ascendingr��drop�6MErIr)rrJrrKrLr)r
�
group_keys�
isinstancer+r�groups�dictr*)rWrar �grms     r!�%test_groupby_with_timegrouper_methodsz1TestGroupBy.test_groupby_with_timegrouper_methods�s��
�'�-�-�/�6�<�<�>�.��T�1�a��Q�/��T�1�a��Q�/��T�2�q�"�a�0��T�2�q�"�a�0��T�2�q�"�a�0��T�2�q�"�a�0�
�	
�
�� ����:���?�B�
�\�\�&�u�\�
-���J�J�w�E�*�+���|�|��|��!�*�*�j�1�1�1������&�$�'�'�'��6�{�a���r#c�H�tdj�dj�gd�tddddd�tddddd�tdd	dd
d�tdd	dd	d�tdd	dd
d�tdd	dd	d�tddddd�tdddd
d�gd��jd�}|j	dd��}||fD�]M}tdj�gd�tddddd�tddddd�tddddd�gd��jddg�}d}|jt
d��dg�jd��}tj||�tdj�gd�tddddd�tddddd�tddddd�tddddd�gd��jddg�}|jt
d ��dg�jd��}tj||���Ptdj�dj�gd�tdd	ddd�tdd	ddd�tdd	dd
d�tdd	dd	d�tdd	dd
d�tdd	dd	d�tdd	ddd�tdd	dd
d�gd��jd�}|j	dd��}||fD�]�}td!j�gd"�tdd	ddd�tdd	ddd�tdd	ddd�tdd	ddd�tdd	ddd�gd��jddg�}|jt
d#��dg�jd��}tj||�|jt
d$��dg�jd��}tdj�gd�tdd	ddd�tdd	ddd�tdd	ddd�gd��jddg�}tj||�|j�}|jt
d$d�%�dg�jd��}tj||�tjtd&�'�5|jt
d$d(�%�dg�j�ddd�|jd�}|jt
d$d�)�dg�jd��}tj||�|jt
d$d�)�dg�jd��}tj||�tjtd*�'�5|jt
d$d(�)�dg�j�ddd�|j�}|j t#j$d�z|d<|jt
d$d�%�dg�jd��}tdj�gd�tdd+d,dd�tdd+d,dd�tdd+d,dd�gd��jddg�}tj||�d-}tjt|�'�5|jt
d$dd�.�dg�j�ddd�tdggdgt'tdd	ddd�gt#j$�d�/��0�}|jt
d$���jd��}tj||�|jt
d$��g�jd��}tj||�|j j)d�|_|j j*t#j$�k(sJ�|jt
d$d�%��jd��}tj||�|jt
d$d�%�g�jd��}tj||����y#1swY���xYw#1swY���xYw#1swY���xYw)1NzA A A A A A A Bz$Carl Mark Carl Carl Joe Joe Joe Carl)rrrrrcrrrrrrrrrrrr4rrdrrFrfz
Carl Joe Mark)rrr�r5r6z!The default value of numeric_only�YErIT)�numeric_onlyzCarl Mark Carl Joe)rrrr��6MSzCarl Joe Mark Carl Joe)rCrcr�r�1D�1ME�r(r'z#'The grouper name foo is not found'��match�foo)r(�levelzThe level foo is not valid��z2The Grouper cannot specify both a key and a level!)r(r'r~�r(r<��columnsr@)rrJrrLrKr)r
rSrTrU�reset_index�pytest�raises�KeyError�
ValueError�copyr@r�MonthEndr�shiftr()rWrXr]r r[�msg�results       r!� test_timegrouper_with_reg_groupsz,TestGroupBy.test_timegrouper_with_reg_groups�su�� �+�1�1�3�?�E�E�G�4��T�1�a��Q�/��T�1�a��Q�/��T�2�q�"�a�0��T�2�q�"�a�0��T�2�q�"�a�0��T�2�q�"�a�0��T�2�q�"�a�0��T�2�q�"�a�0�	�	
�
� �)�F�
�!	�$ �+�+�z�U�+�K�	��	�*�	4�B� �,�2�2�4� +� ��r�2�q�!�4� ��r�2�q�!�4� ��r�2�q�!�4���
��i���)�*�
�6�C��Z�Z��d�!3�W� =�>�B�B�PT�B�U�F��!�!�&�(�3� �1�7�7�9� -� ��q�!�Q��2� ��q�!�Q��2� ��q�!�Q��2� ��q�!�Q��2�	�	���i���)�*�
��Z�Z��e�!4�g� >�?�C�C�QU�C�V�F��!�!�&�(�3�=	4�@ �+�1�1�3�?�E�E�G�4��T�2�q�"�a�0��T�2�q�"�a�0��T�2�q�"�a�0��T�2�q�"�a�0��T�2�q�"�a�0��T�2�q�"�a�0��T�2�q�"�a�0��T�2�q�"�a�0�	�	
�
� �)�F�
�!	�$ �+�+�z�U�+�K�	��	�*�h	4�B� �5�;�;�=� 0� ��r�1�a��3� ��r�1�a��3� ��r�1�a��3� ��r�1�a��3� ��r�1�a��3��
���i���)�*�
��Z�Z��d�!3�W� =�>�B�B�PT�B�U�F��!�!�&�(�3��Z�Z��e�!4�g� >�?�C�C�QU�C�V�F� �,�2�2�4� +� ��r�2�q�!�4� ��r�2�q�!�4� ��r�2�q�!�4���
��i���)�*�
�
�!�!�&�(�3����!�B��Z�Z��e��!@�'� J�K�O�O�!�P��F�
�!�!�&�(�3����x�/T�U�
L��
�
�G��E�:�G�D�E�I�I�K�
L����f�%�B��Z�Z��e�6�!B�G� L�M�Q�Q�!�R��F�
�!�!�&�(�3��Z�Z��e�1�!=�w� G�H�L�L�!�M��F�
�!�!�&�(�3����z�1M�N�
N��
�
�G��e�<�g�F�G�K�K�M�
N�����B����G�$4�$4�Q�$7�7�B�v�J��Z�Z��e��!@�'� J�K�O�O�!�P��F�!�,�2�2�4� +� ��r�2�q�!�4� ��r�2�q�!�4� ��r�2�q�!�4���
��i���)�*�
�
�!�!�&�(�3�G�C����z��5�
��
�
��%�V�6�B�G�L���#�%�
�!����#��#��d�B��A�q�1�2��9I�9I�9K�RX���H��Z�Z��U� 3�4�8�8�d�8�K�F��!�!�&�(�3��Z�Z��e�!4� 5�6�:�:��:�M�F��!�!�&�(�3�%�^�^�1�1�!�4�H�N��>�>�&�&�'�*:�*:�*<�<�<�<��Z�Z��U�� ?�@�D�D�RV�D�W�F��!�!�&�(�3��Z�Z��e��!@� A�B�F�F�!�G��F�
�!�!�&�(�3�Qh	4�N
L�
L��
N�
N��0
�
�s$�#-a=�-b
�1.b�=b	�
b	�b!	r()�D�MErszQE-APRc��ttjgd��gd�gd�gd�d��jd�}|j	d�dj|�j
d	�
�j�jddg�j�jd�}d|_|j�j	t|��dg�dj
�}tj||�|j	t|��dg�dj
�}tj||�y)
N)�20121002�20121007�20130130�20130202�20130305r��20121207r�r�r�r�r�)rrrrrrrrrrrr)i�ilii��io�Zi8�gi-r�i!)r4�rrE�'rrr��-�"rr4)�date�user_id�
whole_cost�cost1r�r�r�r)�	min_countrFrI)rr�to_datetimerLr)rRrS�dropna�reorder_levelsrVrMr<r
rT�assert_series_equal)rWr(r r[r\r^s      r!�%test_timegrouper_with_reg_groups_freqz1TestGroupBy.test_timegrouper_with_reg_groups_freqXs������
�� @�
�G�A!
�#
�F�)�F�
�G	�L
�J�J�y�!�,�/�
�X�d�^�
�S�1�S�
�
�V�X�
�^�V�Y�/�
0�
�Z�\�
�V�G�_�	�%��
�
�M�M�O�#�#�W�$�%7��$C�D�\�R�V�V�X�	�	���w��1��*�*�g�4�0�)�<�=�l�K�O�O�Q��
���w��1r#c��tdj�gd�tddddd�tddddd�tdd	dd
d�tdd	dd	d�tddd
dd�tddd
dd�gd��}|jd��}|jgd�|jd
dg|jdgg}gd�}||fD]d}|jt
dd���}t||�D]7\}}t|�}	|j|	�}
tj|
|��9�f|jdg|jdg|jdgg}gd�}||fD]k}|jdt
dd��g�}t||�D]<\\}}}t|�}	|j||	f�}
tj|
|��>�m|jd�}|jd��}|jgd�|jd
dg|jdgg}||fD]c}|jt
d���}t||�D]7\}}t|�}	|j|	�}
tj|
|��9�ey)NzCarl Joe Joe Carl Joe Carlrrrrrrrrrrr4rrr5rr7)rrrrw)�
2013-09-30�
2013-10-31�
2013-12-31r�rrz))rDr�)�Carlr�)rDr�r6rI)
rrJrrKrOr)r
�zipr
�	get_grouprTrUrL)
rWrXrY�
expected_list�dt_listr �grouped�tr[�dtr��g_list�bs
             r!�test_timegrouper_get_groupz&TestGroupBy.test_timegrouper_get_group�s��� �5�;�;�=�/��T�1�a��Q�/��T�1�a��Q�/��T�2�q�"�a�0��T�2�q�"�a�0��T�2�q�"�a�0��T�1�a��Q�/�
�
�

��#�.�.�*�.�=��
���Y�'����a��V�$����a�S�!�
�
�
=����-�	8�B��j�j��d��!?�@�G�"�7�M�:�
8���8��q�\�� �*�*�2�.���%�%�f�h�7�
8�	8�
���a�S�!����a�S�!����a�S�!�
�
�
X����-�	8�B��j�j�'�7��&�+I�!J�K�G�$'��
�$>�
8� ���A���q�\�� �*�*�A�r�7�3���%�%�f�h�7�
8�	8�"�+�+�F�3��"�.�.�*�.�=��
���Y�'����a��V�$����a�S�!�
�
���-�	8�B��j�j��d�!3�4�G�"�7�M�:�
8���8��q�\�� �*�*�2�.���%�%�f�h�7�
8�	8r#c�b�tddgddgd��}|j�}tj|d�|d<d�}d}t	j
t|�	�5|jtd�
��j|�}ddd�d}t	j
t|�	�5|jtdd���j|�}ddd�t	jjd
��jd
���y#1swY��xYw#1swY�LxYw)N�
10/10/2000�
11/10/2000rr�r��valuer�c�>�t|dj�gd�S)Nr�)rS)r	rS��xs r!�sumfunc_serieszMTestGroupBy.test_timegrouper_apply_return_type_series.<locals>.sumfunc_series�s���1�W�:�>�>�+�,�h�7�7r#�7DataFrameGroupBy.apply operated on the grouping columnsr{�r'r�rzTrh)rr�rr�rT�assert_produces_warning�DeprecationWarningr)r
�applyrUr�)rWr �df_dtr�r�r[r�s       r!�)test_timegrouper_apply_return_type_seriesz5TestGroupBy.test_timegrouper_apply_return_type_series�s����|� <��B�x�P�
Q�����	�����u�V�}�5��f�
�	8�H��
�
'�
'�(:�#�
F�	M��z�z�'�f�"5�6�<�<�^�L�H�	M�G��
�
'�
'�(:�#�
F�	Y��]�]�7��&�#A�B�H�H��X�F�	Y�
������D��)�8�+?�+?�T�+?�+J�	
�	M�	M��	Y�	Y�s�+D�.,D%�D"�%D.c�^�tddgddgd��}|j�}tj|d�|d<d�}d}t	j
t|�	�5|jtd�
��j|�}ddd�t	j
t|�	�5|jtdd���j|�}ddd�t	jjd
��jd
���y#1swY��xYw#1swY�LxYw)Nr�r�rrr�r�c�6�|jj�S�N)r�rSr�s r!�
sumfunc_valuezKTestGroupBy.test_timegrouper_apply_return_type_value.<locals>.sumfunc_value�s���7�7�;�;�=� r#r�r{r�r�rzTrh)rr�rr�rTr�r�r)r
r�r�r�)rWr r�r�r�r[r�s       r!�(test_timegrouper_apply_return_type_valuez4TestGroupBy.test_timegrouper_apply_return_type_value�s����|� <��B�x�P�
Q�����	�����u�V�}�5��f�
�	!�H��
�
'�
'�(:�#�
F�	L��z�z�'�f�"5�6�<�<�]�K�H�	L�
�
'�
'�(:�#�
F�	X��]�]�7��&�#A�B�H�H��W�F�	X�
������D��)�8�+?�+?�T�+?�+J�	
�		L�	L��	X�	X�s�+D�,,D#�D �#D,c�0�d}tdd|��}ttj|�tj|�d�|��}|j	d��}|j
}t
tt|j���t�sJ�y)Ni�z2012/1/1�5min)�startr(�periods)�high�lowr?c�X�t|j|j|j�Sr�)r�year�month�dayr�s r!�<lambda>z?TestGroupBy.test_groupby_groups_datetimeindex.<locals>.<lambda>s��x����������'G�r#)rrrP�aranger)rmrl�next�iter�keysr)rWr��indr r�rms      r!�!test_groupby_groups_datetimeindexz-TestGroupBy.test_groupby_groups_datetimeindex�sw�����z���H��
��Y�Y�w�'��	�	�'�0B�C�3�
���*�*�G�H�������$�t�F�K�K�M�2�3�X�>�>�>r#c�0�tddd��}tgd�gd�d�|��}|jd�	�j}gd
�}|D�cic]}t	|�t|gd����}}t
j||�|jd�	�}|D]}}|j|�}|j|df|j|d
fgg}t|gdd|j��}	t|td�|	��}t
j||��ycc}w)Nz
2015/01/01rr�)r�r<)rrCrurcr)rrrrwr��A�Br?�r~)z
2015-01-05z
2015-01-04z
2015-01-03z
2015-01-02z
2015-01-01�r<r�r�r�)r<r(rH�ABr�)
rrr)rmr
rrT�assert_dict_equalr��locrH�listrU)
rWr@r r��datesr�r[r��data�expected_indexs
          r!�"test_groupby_groups_datetimeindex2z.TestGroupBy.test_groupby_groups_datetimeindex2s���<���@��
�_�?�C�5�
Q�����&��)�0�0��V��LQ�
�DH�I�d�O�]�D�6��?�?�
��
�	���V�X�.��*�*�6�*�*���	4�D��&�&�t�,�F��V�V�D�#�I�&����t�S�y�(9�:�;�D�*���V�#�U�[�[��N�!��t�D�z��P�H��!�!�&�(�3�	4��

s�Dc���gd�}tgd�|tjdd��ddgdzd	��}|d
jd��|d
<t	gd�d
d
��}tddgdzd��}t
j||g�}tgd�gd�d�|ddg��}|jd
dg�j�}tj||�t	|d��}ttjdd��gd�d�|��}t	gd�d��}tgd�gd�d�|ddg��}|jd �!�j�}tj||�y)"N��2011-07-19 07:00:00�2011-07-19 08:00:00�2011-07-19 09:00:00r�r�r���ar�r�r�r�r�rCrFrGrrr)�labelr�value1�value2rc��t|d��S)N�
US/Pacific��tz)r
)�ds r!r�zBTestGroupBy.test_groupby_groups_datetimeindex_tz.<locals>.<lambda>2s
��	�!��8U�r#�r�r�r�r�r�r�r�)r�r<r�r�r�r��rrrrwrr�rrrrrr�r�r�r�r��r@r�z
Asia/Tokyor��rrrrrrr?�r�r�r��rrru�rrwrCrr�)rrPr�r�rrr�from_arraysr)rSrTrU)	rWr�r �exp_idx1�exp_idx2�exp_idxr[r��didxs	         r!�$test_groupby_groups_datetimeindex_tzz0TestGroupBy.test_groupby_groups_datetimeindex_tz sm��
���7�!��)�)�A�W�5��a�&�1�*�	
�
���J��-�-�.U�V��:�� �
���
���#�s��a��g�6���(�(�(�H�)=�>���)�5G�H���x�(�
�����Z��1�2�6�6�8��
���f�h�/��U�|�4��
��y�y��'�2�>P�Q��
��
 �Q��
��� �I�6���x�(�
�����!��$�(�(�*��
���f�h�/r#c���tdtjd�fdtjd�fgddg��}|jd�j	�}|ddtd�k(sJ�y)Nrz
2012-07-03z
2012-07-04r�r�)r�)rrP�
datetime64r)�firstr
)rWr r�s   r!�&test_frame_datetime64_handling_groupbyz2TestGroupBy.test_frame_datetime64_handling_groupby_sk��
�����|�,�
-��2�=�=��3N�/O�P��&�M�
�����C��&�&�(���f�~�a� �I�l�$;�;�;�;r#c
��ttd�gd�gd�d��}|jdd��jj	d��}tt
d	d
��t
dd
��t
d
d��t
dd
��t
dd��gdt��}tj||�d
}|jd�jj|�}tj|�jj|�}tgd�gd�d��}tj|�jj|�}tj||�y)Nr)�2000-01-28 16:47:00�2000-01-29 16:48:00z2000-01-30 16:49:00�2000-01-31 16:50:00z2000-01-01 16:50:00)�America/Chicagor�America/Los_Angelesr�America/New_York)r�r�r�r�F)rkc�r�tj|�jj|j�Sr�)rr�r��tz_localizer<r�s r!r�z9TestGroupBy.test_groupby_multi_timezone.<locals>.<lambda>s$��b�n�n�Q�'�*�*�6�6�q�v�v�>�r#z2000-01-28 16:47:00-0600rr�z2000-01-29 16:48:00-0600z2000-01-30 16:49:00-0800rz2000-01-31 16:50:00-0600z2000-01-01 16:50:00-0500rr�)r<rH)rrr)rrr�r@r<)r�ranger)r�r�r	r
rNrTr�r�rr�r�r)rWr r�r[r��
res_values�
exp_valuess       r!�test_groupby_multi_timezonez'TestGroupBy.test_groupby_multi_timezonehs8��
��q����
�
��(���D�U��3�8�8�>�>�>�
����4�9J�K��4�9J�K��4�9N�O��4�9J�K��4�9K�L�
���

��	���v�x�0�
���Z�Z��%�*�*�4�4�R�8�
����
�+�.�.�:�:�2�>���Q���
�
�
�>�>�*�-�0�0�<�<�R�@��
���v�x�0r#c
��gd�}tgd�|D�cgc]}tj|d����c}tjdd��dd	gd
zd��}tj
gd�dd
��}t
ddgd
zd��}tj||g�}tgd�gd�d�|ddg��}|jd
dg�j�}tj||�tj
|d��}	ttjdd��gd�d�|	��}tj
gd�d��}tgd�gd�d�|ddg��}|jd��j�}tj||�ycc}w) Nr�r��hrIrCrFrGrrr)r��periodr�r�r�rr�r�r�r�r�r�r�r�r�r�r�rr?rrrrr�)
rr�PeriodrPr��PeriodIndexrrrr)rSrTrU)
rWr�r�r rrrr[r�rs
          r!�test_groupby_groups_periodsz'TestGroupBy.test_groupby_groups_periods�s}��
���7�;@�A�a�2�9�9�Q�S�1�A��)�)�A�W�5��a�&�1�*�	
�
���>�>�
���
���#�s��a��g�6���(�(�(�H�)=�>���)�5G�H���x�(�
�����X�w�/�0�4�4�6��
���f�h�/��~�~�e�#�.��
��y�y��'�2�>P�Q��
��
�.�.�Q��
��� �I�6���x�(�
�����!��$�(�(�*��
���f�h�/��_Bs�F

c��tddg�}|djd�|d<t|djjt
j�sJ�|jd��j�}|dj}t|jt
j�sJ�|djd��j�}|j}t|jt
j�sJ�y)N)r��|>c�)rr#rzM8[ns]rr�)	rrM�
issubclassrH�typerPrr)r)rWr r��got_dts    r!�test_groupby_first_datetime64z)TestGroupBy.test_groupby_first_datetime64�s���
�0�2J�K�
L���1����X�&��1���"�Q�%�+�+�*�*�B�M�M�:�:�:����!��$�*�*�,���������&�+�+�r�}�}�5�5�5��A����Q��'�-�-�/�������&�+�+�r�}�}�5�5�5r#c�.�ttd�tjd�d��}|j	d�djd��j
d�}|j	d�dj�}tj||�y)N�20130101rr�r�c�"�|j�Sr�)�maxr�s r!r�z9TestGroupBy.test_groupby_max_datetime64.<locals>.<lambda>�s������r#zM8[s])
rr
rPr�r)r�rMr+rTr�)rWr r[r�s    r!�test_groupby_max_datetime64z'TestGroupBy.test_groupby_max_datetime64�st���Y�z�2����1��F�
G���:�:�c�?�3�'�-�-�.?�@�G�G��P�����C���%�)�)�+��
���v�x�0r#c���ttd�td�gdzd��}|jd�dj	d�}ttd�gdzd��}t
j||�y)Nrz	2000-01-1r�r�r��minr�)rrr
r)�	transformr	rTr��rWr r�r[s    r!�test_groupby_datetime64_32_bitz*TestGroupBy.test_groupby_datetime64_32_bit�sj���U�1�X�Y�{�-C�,D�q�,H�I�
J�����C���%�/�/��6���9�[�1�2�Q�6�S�A��
���v�x�0r#c	�L�ttjjd�j	ddd��tdddd�	�d
��}|j
d�j�d}|j
d�dj�}tj||�y)
Nrrr�<)�sizez01/01/2000 00:00�s�UTC)r�r(r�)�factor�timer7r8)
rrP�random�default_rng�integersrr)r+rTr�)rWr �df1�df2s    r!�$test_groupby_with_timezone_selectionz0TestGroupBy.test_groupby_with_timezone_selection�s�����)�)�/�/��2�;�;�A�q�r�;�J�"�#5�r��PU�V�
�
���j�j��"�&�&�(��0���j�j��"�6�*�.�.�0��
���s�C�(r#c�d�tdgtjtj�gd��}|ddj
tjk(sJ�tdgd�i�}tjtj�|d<|ddj
tjk(sJ�y)Nr)r�r�r�rr��rrr)rr�now�pytz�utc�tzinfo)rWr s  r!�test_timezone_infozTestGroupBy.test_timezone_infos����a�S����T�X�X�(>�'?�@�
A���#�w�q�z� � �D�H�H�,�,�,�
��Y�'�
(���,�,�t�x�x�(��3���#�w�q�z� � �D�H�H�,�,�,r#c��tgd�dztddd��d��}|jd�jj	�}tgd	�t
gd�d�
�d��}tj||�y)
Nr@rrArCr.)r�r()r�r�r�)rrrr�r�r)	rrr)r��countr	rrTr�r0s    r!�test_datetime_countzTestGroupBy.test_datetime_countsd��
��a�-�*�U�A�E�*R�S�
�����C��&�&�,�,�.���)�5���+E�G�T��
���v�x�0r#c	�@�ttjdddtjgtjtd��td��td��tjgd��}t	j
|j�|_d	|d
<||jj�}|jd
�}|jd
�}tj|j�|j��tj|j�|j��tj|j�|j��tj|j�|j��y)Nz2015-07-24 10:10z2015-07-25 11:11z2015-07-23 12:12r)�daysrr)r��tdr��group)rrP�nanrrr�r��notnar)rTrUr+r.r�last)rW�df_test�df_ref�grouped_test�grouped_refs     r!�$test_first_last_max_min_on_time_dataz0TestGroupBy.test_first_last_max_min_on_time_datas5����F�F�&�&�&��F�F���F�F��1�%��1�%��1�%��F�F��
�
��$�^�^�G�J�J�/��
���������)�)�+�,�����w�/���n�n�W�-��
���k�o�o�/��1A�1A�1C�D�
���k�o�o�/��1A�1A�1C�D�
���k�/�/�1�<�3E�3E�3G�H�
���k�.�.�0�,�2C�2C�2E�Fr#c��ttd�tjtd�ggd�d��}t	dd��}|j|�dj
�}||jj�j|�dj
�}|jjd�|_	tj||�y)	Nz2016-06-28 09:35:35z2016-06-28 16:46:28)�1�2�3)r8r�r8rr&r�)
rr
rrr
r)�nuniquer8�notnullr@�
_with_freqrTr�)rW�test�grouperr�r[s     r!�%test_nunique_with_timegrouper_and_natz1TestGroupBy.test_nunique_with_timegrouper_and_nat9s�����3�4��F�F��3�4��
(�

�	
���f�3�/�����g�&�v�.�6�6�8����	�	�)�)�+�,�4�4�W�=�f�E�M�M�O��!���2�2�4�8���
���v�x�0r#c�0�gd�tgd�d��gd�d�}t|�jd�}td�	�}|j	|�}|j�}|j	|g�}|j�}t
j||�y)
N)�shanghai�beijingr`)z2017-08-09 13:32:23z2017-08-11 23:23:15z2017-08-11 22:23:15zdatetime64[ns]rGr@)�locationr8r�r8r�rI)r	rrLr
r)rGrTrU)rW�
data_framer]r�r�r[s      r!�!test_scalar_call_versus_list_callz-TestGroupBy.test_scalar_call_versus_list_callLs���<��U�&���

�
��z�*�4�4�V�<�
��s�#���$�$�W�-��������$�$�g�Y�/���=�=�?��
���f�h�/r#c
�h�d}tjd|dd��}tt|�|��}|j	|j
j�j�}tt|�ttd|dz�|j����}tj||�y)	Nrz2018-01�M�Month)r�r�r(r<r?rr�)r�period_ranger	rr)r@r�rSrr<rTr�)rWr�r@�
period_seriesr�r[s      r!�test_grouper_period_indexz%TestGroupBy.test_grouper_period_index`s���������W�3�W�
���u�W�~�U�;�
��&�&�}�':�':�'@�'@�A�E�E�G����'�N�%��a��1��(=�E�J�J�"O�
��	���v�x�0r#c�0�|}|djd��}|j}|djj}t	dddd|��}tj|dgt|�zg�}tgd	�|d�
�}tj||�y)Nrc��dt|�iS)Nr})r*r�s r!r�zRTestGroupBy.test_groupby_apply_timegrouper_with_nat_dict_returns.<locals>.<lambda>vs��e�S��V�_�r#r�
2013-09-01�
2013-10-01r%�r(r<r>r})rrrrrrrr)r��obj�_valuesr>rrrr*r	rTr�)	rWr0r/�resr r>�dti�mir[s	         r!�4test_groupby_apply_timegrouper_with_nat_dict_returnsz@TestGroupBy.test_groupby_apply_timegrouper_with_nat_dict_returnsns���/����n�"�"�#<�=��
�V�V���&�z�!�!�&�&����|�$�V�RV�W��
�
#�
#�S�5�'�C��H�*<�$=�
>���/�r�
�K��
���s�H�-r#c	��|}|djd��}|j}|djj}t	dddd|��}tdtjtjtjtjtjd	g|jd�d�
�}tj||�y)Nrc�V�t|�r|jdStjS)Nr)r*rOrPrMr�s r!r�zTTestGroupBy.test_groupby_apply_timegrouper_with_nat_scalar_returns.<locals>.<lambda>�s��#�a�&�Q�V�V�A�Y��b�f�f�r#rrmrnr%rorrr)r�rprqr>rr	rPrMr[rTr�)rWr0r/rrr r>rsr[s        r!�6test_groupby_apply_timegrouper_with_nat_scalar_returnszBTestGroupBy.test_groupby_apply_timegrouper_with_nat_scalar_returnss���
/����n�"�"�#L�M��
�V�V���&�z�!�!�&�&����|�$�V�RV�W���
�����������������;��.�.��&��
��	���s�H�-r#c��|}tdd��}|j|�}|jdk(sJ�|jj	|j
�jdk(sJ�d}tjt|��5|jd��}ddd�ttd�g|djd�	�}tgd
�g|tgd�d�
���}tj|�y#1swY�cxYw)Nr�100YEr&rr�r{c��|ddzS)Nrr�r�s r!r�zSTestGroupBy.test_groupby_apply_timegrouper_with_nat_apply_squeeze.<locals>.<lambda>�s��Q�z�]�Q�%6�r#r�)rHr<)�$rCrCrr)rrrrrrr�r�)r
r)�ngroups�
_selected_obj�	_get_axis�axis�nlevelsrTr�r�r�rr
rHrrU)	rWr"r r.r/r�rrrsr[s	         r!�5test_groupby_apply_timegrouper_with_nat_apply_squeezezATestGroupBy.test_groupby_apply_timegrouper_with_nat_apply_squeeze�s���,���&�w�/��
�Z�Z��_���z�z�Q�������)�)�"�'�'�2�:�:�a�?�?�?�H��
�
'�
'�(:�#�
F�	8��(�(�6�7�C�	8��Y�|�,�-�R��Z�5E�5E�F�S���
����/�
�;�
��
	���c�8�,�	8�	8�s�C8�8Dc�:�tjd�|}|djd�d��}|djd�}tj||�|dgjd�d��}|dgjd�}tj
||�y)N�numbarc�,�tj|�Sr��rP�nanmean��valuesr@s  r!r�zITestGroupBy.test_groupby_agg_numba_timegrouper_with_nat.<locals>.<lambda>����"�*�*�V�"4�r#)�engine�meanc�,�tj|�Sr�r�r�s  r!r�zITestGroupBy.test_groupby_agg_numba_timegrouper_with_nat.<locals>.<lambda>�r�r#)r��importorskip�	aggregaterTr�rU)rWr0r/r�r[�	result_df�expected_dfs       r!�+test_groupby_agg_numba_timegrouper_with_natz7TestGroupBy.test_groupby_agg_numba_timegrouper_with_nat�s���	���G�$�/���J��)�)�4�W�*�
���j�>�+�+�F�3��
���v�x�0��
�|�$�.�.�4�W�/�
�	��*��&�0�0��8��
���i��5r#N)"�__name__�
__module__�__qualname__r`r��mark�parametrizerpr�r�r�r�r�r�r�r	r
rr!r'r,r1r>rErHrTr^rdrjrurxr��
single_cpur�r|r#r!r2r2Ks���45�l�[�[���]�T�5�M�:� �;� �@s4�j�[�[���V�%@�A�82�B�82�t@8�D
�*
�(?�4�*=0�~<�01�d;0�z6�1�1�)�-�1� G�D1�&0�(1�.�".�(-�8�[�[���6��6r#r2)�__doc__rr�numpyrPr�rB�pandasrrrrrr	r
rr�pandas._testing�_testingrT�pandas.core.groupby.grouperr
�pandas.core.groupby.opsr�fixturer"r0r2r|r#r!�<module>r�ss����
�
���	�	�	��/�.�������,������,x
6�x
6r#

Youez - 2016 - github.com/yon3zu
LinuXploit