Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/window/__pycache__/test_timeseries_window.cpython-37.pyc
Назад
B �5�g�` � @ sd d dl Zd dlZd dlmZmZmZmZmZm Z m Z d dl m Z d dlm mZ G dd� d�ZdS )� N)� DataFrame�Index� MultiIndex�Series� Timestamp� date_range� to_datetimec @ s� e Zd Zdd� Zdd� Zdd� Zej�dde � d �g�d d � �Z ej�dd e � d �d g�dd� �Z ej�ddde�dd dg�g�dd� �Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Z d/d0� Z d1d2� Z d3d4� Z d5d6� Z d7d8� Z!d9d:� Z"d;d<� Z#d=d>� Z$d?d@� Z%ej�dAdBdCdDgdE fdBdFdDgdG dHg fdBdIdDgdG dJg fdKdCdDgdE fdKdFdDgdG dHg fdKdIdDgdG dJg fdCdCdDgdE fdCdFdDgdG dHg fdCdIdDgdG dJg fdLdCdDgdE fdLdFdDgdG dHg fdLdIdDgdG dJg fdMdCdDgdE fdMdFdDgdG dHg fdMdIdDgdG dJg fg�dNdO� �Z&ej�dPdQdFej'dRej�(dS�dT�dUdVdWdXdYdCdIg �dZd[� �Z)d\d]� Z*d^d_� Z+d`da� Z,dbdc� Z-ddde� Z.dfdg� Z/dhdi� Z0djS )k� TestRollingTSc C s` t tdddd�td�d���d�| _t dtd�i�| _td�td �td �td �td �g| j_d S ) N�20130101� �s)�periods�freq)�A�Br r z20130101 09:00:00z20130101 09:00:02z20130101 09:00:03z20130101 09:00:05z20130101 09:00:06)r r �range� set_index�regular�raggedr �index)�self�method� r �`/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/window/test_timeseries_window.py� setup_method s zTestRollingTS.setup_methodc C sP t ddddtjdgitd�td�td�td �td �gd �}| |�d ��� d S ) Nr r � � � z20130101 09:00:00z20130101 09:00:02z20130101 09:00:03z20130101 09:00:05z20130101 09:00:06)r �2s)r �np�nanr �rolling�sum)r �dfr r r �test_doc_string& s z TestRollingTS.test_doc_stringc C s` d}t jt|d�� | jjdd� W d Q R X d}t jt|d�� | j�� jdd� W d Q R X d S )Nz@passed window foobar is not compatible with a datetimelike index)�match�foobar)�windowzwindow must be an integer)�pytest�raises� ValueErrorr r! � reset_index)r �msgr r r �test_invalid_window_non_int5 s z)TestRollingTS.test_invalid_window_non_intr Z2MSr c C s0 d}t jt|d�� | jj|d� W d Q R X d S )Nz-\<2 \* MonthBegins\> is a non-fixed frequency)r% )r' )r( r) r* r r! )r r r, r r r � test_invalid_window_nonfixed@ s z*TestRollingTS.test_invalid_window_nonfixed�1DZ2msc C s | j j|d� d S )N)r' )r r! )r r r r r �test_valid_windowH s z TestRollingTS.test_valid_window�minpg �?Zfoor � c C s2 d}t jt|d�� | jjd|d� W d Q R X d S )NzQlocal variable 'minp' referenced before assignment|min_periods must be an integer)r% r/ )r' � min_periods)r( r) r* r r! )r r1 r, r r r �test_invalid_minpL s z TestRollingTS.test_invalid_minpc C s� | j }d}tjt|d�� |jddd� W d Q R X |�� }tdt|�d�|d< |jd dd��� d }tjt|d�� |jd d d� W d Q R X |jd dd�j �� d S ) NzSinvalid on specified as foobar, must be a column \(of DataFrame\), an Index or None)r% r r& )r' �onr )r �CZ2dzwindow must be an integerr ) r r( r) r* r! �copyr �lenr" r )r r# r, r r r �test_onV s zTestRollingTS.test_onc C sb t tdddd�td�d��}|jjs(t�|jddd��� |�d�}|j jsPt�|�d��� d S ) Nr r r )r r )r r r r )r5 ) r r r r � is_monotonic�AssertionErrorr! r" r r )r r# r r r �test_monotonic_ono s z TestRollingTS.test_monotonic_onc C s� t tdddd�td�d��}|�d�}|j�� }|d |d< ||_|jjrNt�d }tj t |d �� |� d �� � W d Q R X |� � }d }tj t |d �� |j d dd �� � W d Q R X d S )Nr r r )r r )r r r r2 r zindex must be monotonic)r% r zNinvalid on specified as A, must be a column \(of DataFrame\), an Index or None)r5 )r r r r r Zto_listr: r; r( r) r* r! r" r+ )r r# Znon_monotonic_indexr, r r r �test_non_monotonic_on} s z#TestRollingTS.test_non_monotonic_onc C s� t td�tdddd�d��}td�td�td�td�td �g|d < |�d ��d �j�� jd d �}|jd d d�j�� }t � ||� |�d ��d �dg �� �� dd g }|jd d d�dg �� }t � ||� d S )Nr z20130101 09:00:00�3s)r r )r r6 z20130101 09:00:02z20130101 09:00:03z20130101 09:00:05z20130101 09:00:06r r T)�drop)r5 r ) r r r r r r! r r" r+ �tm�assert_series_equal�assert_frame_equal)r r# �expected�resultr r r � test_frame_on� s &zTestRollingTS.test_frame_onc C s t dddddgdddtjdgttd�td�td�td �td �g�d �d d dgd�}t dddddgdddtjdg|d d �d d dgd�}|jdd d��� }|}t�||� t dddtjdgdd�}|jdd d�j �� }t� ||� |d dd g }|jdd d�d dd g �� }t�||� d S )Nr r r r2 r z20130101 09:00:00z20130101 09:00:02z20130101 09:00:03z20130101 09:00:05z20130101 09:00:06)r r r6 r r6 r )�columnsg � r )r5 )�name) r r r r r r! r" r@ rB r r rA )r r# Z expected1rD rC r r r �test_frame_on2� s, $ z TestRollingTS.test_frame_on2c C s� | j �� }tdddd�|_|jddd��� }|jdd��� }t�||� tddd d�|_|jddd��� }|jd dd��� }t�||� |jddd��� }|jd dd��� }t�||� |jdd��� }|jd d��� }t�||� d S ) Nr r �D)r r r )r' r3 r/ )r' Z2D)r r7 r r r! r" r@ rB )r r# rC rD r r r �test_basic_regular� s z TestRollingTS.test_basic_regularc C sf | j }|jddd��� }|�d��� }t�||� |jddd��� }|jddd��� }t�||� d S )Nr r )r3 r )r r! r" r@ rB )r r# rC rD r r r �test_min_periods� s z TestRollingTS.test_min_periodsc C sf t ddgd itd�td�td�td�td�gd �}d }tjt|d �� | jjd d d� W d Q R X |�� }dddddg|d< |jd dd��� }t � ||� |�d ��� }t � ||� |�� }dddddg|d< |jd dd��� }t � ||� |�� }t j ddddg|d< |jd dd��� }t � ||� |�� }t j dddt j g|d< |jd dd��� }t � ||� d S )Nr r r z20130101 09:00:01z20130101 09:00:02z20130101 09:00:03z20130101 09:00:04z20130101 09:00:06)r z3closed must be 'right', 'left', 'both' or 'neither')r% r Zblabla)r' �closedg �?r �right)rM r2 Zboth�leftZneither) r r r( r) r* r r! r7 r" r@ rB r r )r r# r, rC rD r r r � test_closed s8 zTestRollingTS.test_closedc C s� | j }|jddd��� }|�� }dddddg|d< t�||� |jd dd��� }|�� }ddddd g|d< t�||� |jd dd��� }|�� }tjtjdtjd g|d< t�||� |jd dd��� }|�� }dddd d g|d< t�||� |jd d ��� }|�� }dddd d g|d< t�||� |jddd��� }|�� }dddddg|d< t�||� |jddd��� }|�� }tjtjdddg|d< t�||� |jddd��� }|�� }dddddg|d< t�||� d S )N�1sr )r' r3 g r r2 r r r rG r> r )r' �4s� � �5s� )r r! r" r7 r@ rB r r )r r# rD rC r r r �test_ragged_sum, sB z TestRollingTS.test_ragged_sumc C sz | j }|jddd��� }|�� }dddddg|d< t�||� |jd dd��� }|�� }ddd d d g|d< t�||� d S ) NrQ r )r' r3 g r r2 r r r g �?g @g @)r r! �meanr7 r@ rB )r r# rD rC r r r �test_ragged_meanW s z TestRollingTS.test_ragged_meanc C sz | j }|jddd��� }|�� }dddddg|d< t�||� |jd dd��� }|�� }ddd d d g|d< t�||� d S ) NrQ r )r' r3 g r r2 r r r g �?g @g @)r r! �medianr7 r@ rB )r r# rD rC r r r �test_ragged_mediand s z TestRollingTS.test_ragged_medianc C s~ | j }|jddd��d�}|�� }dddddg|d < t�||� |jd dd��d�}|�� }ddd d d g|d < t�||� d S )NrQ r )r' r3 g �?g r r2 r r r g �?g @g @)r r! �quantiler7 r@ rB )r r# rD rC r r r �test_ragged_quantileq s z"TestRollingTS.test_ragged_quantilec C s� | j }|jddd�jdd�}|�� }dgd |d< t�||� |jddd�jdd�}|�� }tjgd |d< t�||� |jd dd�jdd�}|�� }dgd gd |d< t�||� |jd dd�jdd�}|�� }tjd dddg|d< t�||� d S )NrQ r )r' r3 r )�ddofg r r r> g �?r rU g���۞��?g �?g�PS��?)r r! �stdr7 r@ rB r r )r r# rD rC r r r �test_ragged_std~ s" z TestRollingTS.test_ragged_stdc C s� | j }|jddd�jdd�}|�� }dgd |d< t�||� |jddd�jdd�}|�� }tjgd |d< t�||� |jd dd�jdd�}|�� }dgd gd |d< t�||� |jd dd�jdd�}|�� }tjd dddg|d< t�||� d S )NrQ r )r' r3 r )r^ g r r r> g �?r rU g �?g �?g�������?)r r! �varr7 r@ rB r r )r r# rD rC r r r �test_ragged_var� s" z TestRollingTS.test_ragged_varc C s� | j }|jddd��� }|�� }tjgd |d< t�||� |jddd��� }|�� }tjgd dddg |d< t�||� d S ) Nr> r )r' r3 r r rU r g )r r! �skewr7 r r r@ rB )r r# rD rC r r r �test_ragged_skew� s z TestRollingTS.test_ragged_skewc C s| | j }|jddd��� }|�� }tjgd |d< t�||� |jddd��� }|�� }tjgd dg |d< t�||� d S ) Nr> r )r' r3 r r rU r g333333�)r r! �kurtr7 r r r@ rB )r r# rD rC r r r �test_ragged_kurt� s z TestRollingTS.test_ragged_kurtc C s� | j }|jddd��� }|�� }dddddg|d< t�||� | j }|jdd��� }t�||� |jddd��� }|�� }dddddg|d< t�||� |jddd��� }|�� }tjtjdtjdg|d< t�||� d S ) NrQ r )r' r3 g �?r )r' r r )r r! �countr7 r@ rB r r )r r# rD rC r r r �test_ragged_count� s z TestRollingTS.test_ragged_countc C s� t tdddd�ddddd gd ���d �}|�d ��� }|�� }ddddd g|d < t�||� t tdddd�dd dd dgd ���d �}t�||� |�d��� }|�� }dd ddd g|d < t�||� |�d��� }|�� }dd dddg|d < t�||� d S )Nr r r )r r g r r r2 r )r r r rQ r r g @rU )r r r r! �minr7 r@ rB )r r# rD rC r r r �test_regular_min� s& z TestRollingTS.test_regular_minc C s� | j }|jddd��� }|�� }dddddg|d< t�||� |jd dd��� }|�� }dddddg|d< t�||� |jd dd��� }|�� }dd d ddg|d< t�||� d S ) NrQ r )r' r3 g r r2 r r r rU r )r r! ri r7 r@ rB )r r# rD rC r r r �test_ragged_min� s z TestRollingTS.test_ragged_minc C s� d}t dtj�|�itd|dd�d�}|jddd ��� }|�d ��� }|| d k �� �� s^t �|jd dd ��� }|�d ��� }|| d k �� �� s�t �d S )Ni' r r r )r r )r r r )r3 r g{�G�z�?�� Z200s) r r �randomZrandnr r! ri �all�boolr; )r �NZdfprC rD r r r � test_perf_min s zTestRollingTS.test_perf_minc C s� | j }|jddd��� }|�� }dddddg|d< t�||� |jd dd��� }|�� }dddddg|d< t�||� |jd dd��� }|�� }dddddg|d< t�||� d S ) NrQ r )r' r3 g r r2 r r r rU )r r! �maxr7 r@ rB )r r# rD rC r r r �test_ragged_max s z TestRollingTS.test_ragged_maxzfreq, op, result_data�msri g rV rX rT g �q �q�?rr g @r �hrJ c C sl t dd|� �dd�}td|d�}tj|jd< d|jd < t|jd |� �d �|�� }t||d�}t�||� d S ) Nz2018-1-1 01:00:00�1rV )�startr r r )�datar r r ����10)r' ) r r r r Ziloc�getattrr! r@ rA )r r �opZ result_datar r rD rC r r r �test_freqs_ops* s z TestRollingTS.test_freqs_ops�fr" rg z ignore:min_periods:FutureWarning)ZmarksrZ r_ ra re rc c C sj | j d }|jdd�}|jdd�}t||�� }t||�� }t�||� |�d�}|�d�}t�||� d S )Nr r )r' rQ g �?)r r! r{ r@ rB r\ )r r~ r# Zer�rrD rC r r r �test_allI s zTestRollingTS.test_allc s� |� t dt�d�itdddd�d�}|�dd�}|jd d �}t|� �� }� fd d �}|�|jj �� |�j d dd�}t � ||� d S )Nr �2 r �H)r r )r z09:00z16:00Z5H)r' c s" | � dd�} t| jddd�� �� S )Nz09:00z16:00r r )r3 )� between_timer{ r! )�x)r~ r r � agg_by_day| s z+TestRollingTS.test_all2.<locals>.agg_by_dayr T)�levelr? )r r �aranger r� r! r{ �groupbyr �day�applyr+ r@ rB )r Zarithmetic_win_operatorsr# Zdftr rD r� rC r )r~ r � test_all2j s zTestRollingTS.test_all2c C s� dddgdddgdddgdddgd d dgd d dgd d dgd ddgd ddgd ddgg }t |dddgd�}t|d �|d< |�d��d��dd� �}|�d�jddd�d �� }t�||� d S )NZDavidz1/1/2015�d z1/5/2015i� z 5/30/2015r� z 7/25/2015ZRyanz1/4/2014z 1/19/2015z 3/31/2016ZJoez7/1/2015z9/9/2015z 10/15/2015rH �date�amount)rx rF c S s | � d�d �� S )N�180Dr� )r! r" )r� r r r �<lambda>� � z6TestRollingTS.test_groupby_monotonic.<locals>.<lambda>r� )r5 ) r r r r� r� r! r" r@ rA )r rx r# rC rD r r r �test_groupby_monotonic� s" z$TestRollingTS.test_groupby_monotonicc C s� t dddd�}tdgd dgd dgd t�||f�t�d �d ��}|�d �jd dd�j�� }|� d��d �� dd� �}t � ||� |� d�}|�d �jd dd�j�� }t � ||� d S )Nz2016-01-01 09:30:00� r )rw r r r r � r2 � �( )r r r6 r rR r )r5 c S s | � d�d �� S )NrR r6 )r! rX )r� r r r r� � r� z2TestRollingTS.test_non_monotonic.<locals>.<lambda>)r r r Z concatenater� r� r! r6 rX r r� r@ rA Z sort_values)r �datesr# rD rC Zdf2r r r �test_non_monotonic� s z TestRollingTS.test_non_monotonicc C s� t dddd�}tt�t|��|d�}|�d��� }ttjgdgt|�d |d�}t� ||� |jd dd ��� }t� ||� |�d ��� }ttjdgd gt|�d |d�}t� ||� |jd dd ��� }t� ||� d S )Nz 2017-01-01� Z1h)r r )r Z2hg �?r r )r3 Z3hg �?r2 ) r r r r� r8 r! Zcovr r@ rA )r �idx�ssrD rC Z expected2r r r �test_rolling_cov_offset� s " $ z%TestRollingTS.test_rolling_cov_offsetc C st t d�t d�t d�t d�t d�g}tddddd d gi|d �}|�d ��� }tdd d dddgi|d �}t�||� d S )Nz20190101 09:00:30z20190101 09:00:27z20190101 09:00:20z20190101 09:00:18z20190101 09:00:10�columnr2 r r rS )r rU g @g @g @)r r r! ri r@ rB )r r r# rD rC r r r � test_rolling_on_decreasing_index� s z.TestRollingTS.test_rolling_on_decreasing_indexc C s> t dg ig d�}|�d��� }t dg ig d�}t�||� d S )Nr� )r rU )r r! ri r@ rB )r r# rD rC r r r �test_rolling_on_empty� s z#TestRollingTS.test_rolling_on_emptyc C sz t dtd�itjtddd�td�gddgd �d �}|jd |j�d�d ��� }t dd dddddgi|jd �}t � ||� d S )Nr� rS Z20190101r2 )r r r� �seq)�names)r Z10d)r5 g g �?g @g @g $@g .@) r r r Z from_productr r! r Zget_level_valuesr" r@ rB )r r# rD rC r r r �!test_rolling_on_multi_index_level� s " z/TestRollingTS.test_rolling_on_multi_index_levelN)1�__name__� __module__� __qualname__r r$ r- r( �markZ parametrize�offsetsZ MonthBeginr. ZDayr0 r �arrayr4 r9 r< r= rE rI rK rL rP rW rY r[ r] r` rb rd rf rh rj rk rq rs r} �param�filterwarningsr� r� r� r� r� r� r� r� r r r r r s� "( !%,+ " r )�numpyr r( Zpandasr r r r r r r Zpandas._testingZ_testingr@ Zpandas.tseries.offsetsZtseriesr� r r r r r �<module> s $
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка