Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/groupby/__pycache__/test_timegrouper.cpython-37.pyc
Назад
B �5�g8n � @ s� d Z ddlmZ ddlmZ ddlZddlZddlZddlZ ddlm Z m Z m Z m Z mZmZmZmZ ddlmZ ddlmZ ddlmZ G dd � d �ZdS ) z5 test with the TimeGrouper / grouping with datetimes � )�datetime)�StringION)� DataFrame� DatetimeIndex�Index� MultiIndex�Series� Timestamp� date_range�offsets)�Grouper)� BinGrouperc @ s� e Zd Zdd� Zej�dddg�dd� �Zdd � Zej�d d d d dg�dd� �Z dd� Z dd� Z dd� Z dd� Z dd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Zd5S )6� TestGroupByc C sF t d�� ddddddgtddddd �tddddd�tdd dd d �tdd dd d �tdd d d d �tddd dd �gd��}|jdd�}x�||gD ]�}|�dg�}t dd itdddddd�d�}tjdddgdd �|jd ddgd f< |� d�� � }t � ||� |� � }|�tdd ��� � }t � ||� |�tdd ��� � }t � ||� q�W d S ) Nz Carl Carl Carl Carl Joe Carl� � � � � i� � r � � � � � )�Buyer�Quantity�Dater )�byr Z20130901Z20131205Z5D�left)�freq�name�closed)�index� � �int64)�dtype)r )r �splitr � sort_values� set_indexr �np�array�iloc�resample�sum�tm�assert_frame_equal� sort_index�groupbyr ) �self� df_original� df_reordered�df�expected�result1� df_sorted�result2Zresult3� r; �[/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/groupby/test_timegrouper.py� test_groupby_with_timegrouper s0 $ z)TestGroupBy.test_groupby_with_timegrouper� should_sortTFc C s� t d�� d�� ddddddgtdddd d �tdddd d�tdd dd d �tdd d d d �tddd dd �tddd dd �gd��}|r�|jddd�}|jddd�}|�tdd��}|js�t�t |j t �s�t�|j }t |t �s�t�t|�dks�t�d S )Nz A A A A A BzCarl Mark Carl Joe Joe Carlr r r � r i� r r r r r r r )�Branchr r r r F)r � ascendingr )�dropZ6M)r )r r' r r( r) r2 r Z group_keys�AssertionError� isinstance�grouperr �groups�dict�len)r3 r> r6 �grF r; r; r< �%test_groupby_with_timegrouper_methodsF s& z1TestGroupBy.test_groupby_with_timegrouper_methodsc C s t d�� d�� ddddddddgtdddd d �tdddd d�tdd dd d �tdd d d d �tdd dd d �tdd d d d �tddd dd �tddd dd �gd���d�}|jddd�}�x||gD ]�}t d�� d ddgtdddd d �tdddd d �tdddd d �gd���ddg�}|�tdd�dg��� }t� ||� t d �� ddddgtdddd d �tdddd d �tdd dd d �tdd dd d �gd���ddg�}|�td d�dg��� }t� ||� q�W t d�� d�� ddddddddgtdd dd d �tdd dd d�tdd dd d �tdd d d d �tdd dd d �tdd d d d �tdd d dd �tdd d dd �gd���d�}|jddd�}�x�||gD �]�}t d �� d ddd!d gtdd dd d �tdd dd d �tdd dd d �tdd d d d �tdd d d d �gd���ddg�}|�td"d�dg��� }t� ||� |�td#d�dg��� }t d�� d ddgtdd dd d �tdd dd d �tdd dd d �gd���ddg�}t� ||� |� � }|�td#dd$�dg��� }t� ||� t j t d%d&�� |�td#d'd$�dg��� W d Q R X |�d�}|�td#dd(�dg��� }t� ||� |�td#d d(�dg��� }t� ||� t j td)d&�� |�td#d'd(�dg��� W d Q R X |�� }|jt�d � |d< |�td#dd$�dg��� }t d�� d ddgtdd*d+d d �tdd*d+d d �tdd*d+d d �gd���ddg�}t� ||� d,}t j t|d&��" |�td#ddd-�dg��� W d Q R X t dggdgttdd dd d �gt�� dd.�d/�}|�td#d���� }t� ||� |�td#d�g��� }t� ||� |j�d�|_|jjt�� k�s�t�|�td#dd$���� }t� ||� |�td#dd$�g��� }t� ||� �qpW d S )0NzA A A A A A A Bz$Carl Mark Carl Carl Joe Joe Joe Carlr r r r? r i� r r r r r r r )r@ r r r r r F)r rA z Carl Joe Markr � )r r r r �A)r zCarl Mark Carl Joe� Z6MSzCarl Joe Mark Carl Joer$ � Z1DZ1M)r �keyz#'The grouper name foo is not found')�matchZfoo)r �levelzThe level foo is not valid� � z2The Grouper cannot specify both a key and a level!)r rO rQ )r r )�columnsr" )r r' r r) r( r2 r r. r/ r0 � reset_index�pytestZraises�KeyError� ValueError�copyr" r ZMonthEndr �shiftr rC )r3 r4 r9 r6 r7 �result�msgr; r; r< � test_timegrouper_with_reg_groupsg s� $ $ "" z,TestGroupBy.test_timegrouper_with_reg_groupsr �D�MrL zQ-APRc C s t t�ddddddddddddg �ddddddddd d d d g d d d d ddddddddg dddddddddd ddg d ���d �}|�d �d �|�jdd!��� �d d g�� � � d"�}d |_ |� � �t |d#�d g�d �� }t �||� |�t |d#�d g�d �� }t �||� d S )$NZ20121002Z20121007Z20130130Z20130202Z20130305Z20121207r r r i� il i i �� io �Z i8 ig i- i! r � r r# �' r �- �" )�date�user_id� whole_costZcost1rf rg rh )Z min_countr% )r )r �pd� to_datetimer) r2 r- r. ZdropnaZreorder_levelsr1 Zastyper r r/ �assert_series_equal)r3 r r6 r7 r8 r: r; r; r< �%test_timegrouper_with_reg_groups_freq sN " z1TestGroupBy.test_timegrouper_with_reg_groups_freqc C s@ t d�� ddddddgtddddd �tddddd�tdd dd d �tdd dd d �tdd d d d �tddd dd �gd��}|jdd�}|jd ddg |jd dg |jdg g}dddg}xX||gD ]L}|�tddd��}x4t||�D ]&\}}t|�} |� | �} t � | |� q�W q�W |jdg |jdg |jdg g}dddg} xh||gD ]\}|�d tddd�g�}x>t| |�D ]0\\} }}t|�} |� | | f�} t � | |� �qhW �qBW |� d�}|jdd�}|jd ddg |jd dg |jdg g}xZ||gD ]N}|�tdd ��}x6t||�D ](\}}t|�} |� | �} t � | |� �q W �q�W d S ) NzCarl Joe Joe Carl Joe Carlr r r r r i� r r r r r r r )r r r r )r rN z 2013-09-30z 2013-10-31z 2013-12-31r_ r )r rO )�Joez 2013-09-30)ZCarlz 2013-10-31)rm z 2013-12-31r )r ) r r' r r( r, r2 r �zipr � get_groupr/ r0 r) ) r3 r4 r5 � expected_listZdt_listr6 �grouped�tr7 �dtr[ Zg_list�br; r; r< �test_timegrouper_get_groupN sR z&TestGroupBy.test_timegrouper_get_groupc C s� t ddgddgd��}|�� }t�|d �|d< dd� }|�tdd ���|�}|�td dd ���|�}t�|j d d �|j d d �� d S )Nz 10/10/2000z 11/10/2000r r )rf �valuerf c S s t | d �� gd�S )Nrv )r. )r r. )�xr; r; r< �sumfunc_series� s zMTestGroupBy.test_timegrouper_apply_return_type_series.<locals>.sumfunc_series)rO r_ )r rO T)rB ) r rY ri rj r2 r �applyr/ r0 rU )r3 r6 �df_dtrx r7 r[ r; r; r< �)test_timegrouper_apply_return_type_series� s z5TestGroupBy.test_timegrouper_apply_return_type_seriesc C s� t ddgddgd��}|�� }t�|d �|d< dd� }|�tdd ���|�}|�td dd ���|�}t�|j d d �|j d d �� d S )Nz 10/10/2000z 11/10/2000r r )rf rv rf c S s | j �� S )N)rv r. )rw r; r; r< � sumfunc_value� s zKTestGroupBy.test_timegrouper_apply_return_type_value.<locals>.sumfunc_value)rO r_ )r rO T)rB ) r rY ri rj r2 r ry r/ rk rU )r3 r6 rz r| r7 r[ r; r; r< �(test_timegrouper_apply_return_type_value� s z4TestGroupBy.test_timegrouper_apply_return_type_valuec C s< d}t dd|d�}tt�|�t�|�d�|d�}|�dd� �}|j}tt|�� �d t �s^t �t d d d d �}td ddddgddddd gd�|d�}|jd d�j}ddddd g}d d � |D �} t � || � |jd d�}xb|D ]Z} |� | �}|j| d f |j| d f gg} t| gd d!d"�} t| td#�| d$�} t �|| � q�W d S )%Ni� z2012/1/1Z5min)�startr �periods)�high�low)r" c S s t | j| j| j�S )N)r �year�month�day)rw r; r; r< �<lambda>� � z?TestGroupBy.test_groupby_groups_datetimeindex.<locals>.<lambda>r z 2015/01/01r rf )r r r$ rM r? r r r r rN )rL �B)rQ z 2015-01-05z 2015-01-04z 2015-01-03z 2015-01-02z 2015-01-01c S s i | ]}t |gd d�t|��qS )rf )r )r r )�.0rf r; r; r< � <dictcomp>� s zATestGroupBy.test_groupby_groups_datetimeindex.<locals>.<dictcomp>rL r� r^ )r r ZAB)rT r" )r r r* �aranger2 rF rD �list�keysr rC r/ Zassert_dict_equalro �locr r0 ) r3 r �indr6 rq rF r" r[ �datesr7 rf �dataZexpected_indexr; r; r< �!test_groupby_groups_datetimeindex� s* & z-TestGroupBy.test_groupby_groups_datetimeindexc C sj ddddddg}t ddddddg|tjddd�d d gd d ��}|d �dd� �|d < tddddddgdd d�}tddgd dd�}t�||g�}t dd d dd dgd d d d d d gd�|ddgd�}|�d dg�� � }t � ||� t|dd �}t tjddd�d d d d d d gd�|d �}tdddgdd �}t d dd gd ddgd�|ddgd�}|jdd �� � }t � ||� d S ) Nz2011-07-19 07:00:00z2011-07-19 08:00:00z2011-07-19 09:00:00�art r$ r% )r& r r r )�labelr �value1�value2r c S s t | dd�S )Nz US/Pacific)�tz)r )�dr; r; r< r� � r� zBTestGroupBy.test_groupby_groups_datetimeindex_tz.<locals>.<lambda>z US/Pacific)r� r r� )r r rN r )r� r� r� r� )r" rT z Asia/Tokyo)r� )r" rM )rQ ) r r* r� ry r r r � from_arraysr2 r. r/ r0 ) r3 r� r6 �exp_idx1�exp_idx2�exp_idxr7 r[ �didxr; r; r< �$test_groupby_groups_datetimeindex_tz� sT z0TestGroupBy.test_groupby_groups_datetimeindex_tzc C sR t dt�d�fdt�d�fgddgd�}|�d��� }|d d td�ksNt�d S )Nr z 2012-07-03z 2012-07-04r� rf )rT )r r* � datetime64r2 �firstr rC )r3 r6 r[ r; r; r< �&test_frame_datetime64_handling_groupby s z2TestGroupBy.test_frame_datetime64_handling_groupbyc C s� d}t jt|�d dddgd�}|�d�j�dd� �}ttdd d �td d d �td d d �tdd d �tddd �gdtd�}t � ||� d }|�d�j� |�}t � |�j �|�}tdddgdddgdd�}t � |�j �|�}t � ||� d S )Nz�0,2000-01-28 16:47:00,America/Chicago 1,2000-01-29 16:48:00,America/Chicago 2,2000-01-30 16:49:00,America/Los_Angeles 3,2000-01-31 16:50:00,America/Chicago 4,2000-01-01 16:50:00,America/New_Yorkrv rf r� )�header�namesc S s t �| �j�| j�S )N)ri rj rs � tz_localizer )rw r; r; r< r� % r� z9TestGroupBy.test_groupby_multi_timezone.<locals>.<lambda>z2000-01-28 16:47:00-0600zAmerica/Chicago)r� z2000-01-29 16:48:00-0600z2000-01-30 16:49:00-0800zAmerica/Los_Angelesz2000-01-31 16:50:00-0600z2000-01-01 16:50:00-0500zAmerica/New_York)r r& z2000-01-28 16:47:00z2000-01-29 16:48:00z2000-01-31 16:50:00r r r )r" r )ri Zread_csvr r2 rf ry r r �objectr/ rk ro rj rs r� )r3 r� r6 r[ r7 r� Z res_valuesZ exp_valuesr; r; r< �test_groupby_multi_timezone s, z'TestGroupBy.test_groupby_multi_timezonec C sd ddddddg}t ddddddgdd� |D �tjdd d �d d gd d��}tjddddddgddd�}tddgd dd�}t�||g�}t dd d dd dgd d d d d d gd�|ddgd�}|�ddg�� � }t � ||� tj|dd�}t tjdd d �d d d d d d gd�|d �}tjdddgdd�}t d dd gd ddgd�|ddgd�}|jdd �� � }t � ||� d S ) Nz2011-07-19 07:00:00z2011-07-19 08:00:00z2011-07-19 09:00:00r� rt c S s g | ]}t j|d d��qS )�H)r )ri ZPeriod)r� r� r; r; r< � <listcomp>L s z;TestGroupBy.test_groupby_groups_periods.<locals>.<listcomp>r$ r% )r& r r r )r� �periodr� r� r� r� )r r r� )r r rN r )r� r� r� r� )r" rT )r )r" rM )rQ ) r r* r� ri Z PeriodIndexr r r� r2 r. r/ r0 ) r3 r� r6 r� r� r� r7 r[ r� r; r; r< �test_groupby_groups_periods@ sR z'TestGroupBy.test_groupby_groups_periodsc C s� t ddg�}|d �d�|d< t|d jjtj�s6t�|jdd�� � }|d j}t|jtj�sbt�|d jdd�� � }|j}t|jtj�s�t�d S )N)r l �|>c� )r l �|>c� r zM8[ns]r )rQ ) r �view� issubclassr& �typer* r� rC r2 r� )r3 r6 r[ Zgot_dtr; r; r< � test_groupby_first_datetime64} s z)TestGroupBy.test_groupby_first_datetime64c C sR t td�t�d�d��}|�d�d �dd� �}|�d�d �� }t�||� d S )NZ20130101r )rL r� rL c S s | � � S )N)�max)rw r; r; r< r� � r� z9TestGroupBy.test_groupby_max_datetime64.<locals>.<lambda>) r r r* r� r2 ry r� r/ rk )r3 r6 r7 r[ r; r; r< �test_groupby_max_datetime64� s z'TestGroupBy.test_groupby_max_datetime64c C sV t td�td�gd d��}|�d�d �t�}ttd�gd dd�}t�||� d S )Nr z 2000-01-1)rL r� rL r� )r ) r �ranger r2 Z transform�minr r/ rk )r3 r6 r[ r7 r; r; r< � test_groupby_datetime64_32_bit� s z*TestGroupBy.test_groupby_datetime64_32_bitc C sh t j�d� tt jjdddd�tddddd �d ��}|�d ��� d }|�d �d �� }t� ||� d S ) N�* r r �< )�sizez01/01/2000 00:00�s�UTC)r r r� )�factor�timer� r� ) r* �random�seedr �randintr r2 r� r/ rk )r3 r6 Zdf1Zdf2r; r; r< �$test_groupby_with_timezone_selection� s z0TestGroupBy.test_groupby_with_timezone_selectionc C sp t dgt�tj�gd��}|d d jtjks2t�t ddddgi�}t�tj�|d< |d d jtjkslt�d S )Nr )r� rt rt r r� r r )r r �now�pytz�utc�tzinforC )r3 r6 r; r; r< �test_timezone_info� s z TestGroupBy.test_timezone_infoc C sd t dddgd tdddd�d��}|�d �j�� }tdddgtdddgd d �d d �}t�||� d S ) Nr r r r� r$ �T)r r )r� r� r� )r r� )r" r ) r r r2 r� �countr r r/ rk )r3 r6 r[ r7 r; r; r< �test_datetime_count� s "z TestGroupBy.test_datetime_countc C s� ddl m} ttjdddtjgtj|dd�|dd�|d d�tjgd ��}t�|j�|_d |d < ||j�� }|� d �}|� d �}t � |� � |� � � t � |� � |� � � t � |�� |�� � t � |�� |�� � d S ) Nr )� timedeltaz2015-07-24 10:10z2015-07-25 11:11z2015-07-23 12:12r )�daysr r )rs �tdrL �group)r r� r r* �nanri rj rs Znotnar2 r/ r0 r� r� r� �last)r3 r� Zdf_testZdf_refZ grouped_testZ grouped_refr; r; r< �$test_first_last_max_min_on_time_data� s" * z0TestGroupBy.test_first_last_max_min_on_time_datac C s| t td�tjtd�gdddgd��}tddd �}|�|�d �� }||j�� �|�d �� }|j � d �|_ t � ||� d S ) Nz2016-06-28 09:35:35z2016-06-28 16:46:28�1�2�3)r� r� r� �h)rO r r� ) r r ri ZNaTr r2 Znuniquer� Znotnullr" Z _with_freqr/ rk )r3 �testrE r[ r7 r; r; r< �%test_nunique_with_timegrouper_and_nat� s z1TestGroupBy.test_nunique_with_timegrouper_and_natc C st dddgt dddgdd�dd d gd �}t|��d �}td d�}|�|�}|�� }|�|g�}|�� }t�||� d S )NZshanghaiZbeijingz2017-08-09 13:32:23z2017-08-11 23:23:15z2017-08-11 22:23:15zdatetime64[ns])r& r r r )�locationr� rv r� r^ )r )r r r) r r2 r� r/ r0 )r3 Z data_framerE rq r[ r7 r; r; r< �!test_scalar_call_versus_list_call� s z-TestGroupBy.test_scalar_call_versus_list_callc C sn d}t jd|ddd�}tt|�|d�}|�|jj��� }ttd|�ttd|d �|j d �d�}t � ||� d S ) Nr z2018-01r_ ZMonth)r~ r r r )r" r r )r ) ri Z period_ranger r� r2 r" r� r. r r r/ rk )r3 r r" Z period_seriesr[ r7 r; r; r< �test_grouper_period_index s $z%TestGroupBy.test_grouper_period_indexN) �__name__� __module__� __qualname__r= rV �markZ parametrizerJ r] rl ru r{ r} r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r; r; r; r< r s. +! - ;B ? '= r )�__doc__r �ior �numpyr* rV r� Zpandasri r r r r r r r r Zpandas._testingZ_testingr/ Zpandas.core.groupby.grouperr Zpandas.core.groupby.opsr r r; r; r; r< �<module> s (
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка