Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/window/__pycache__/test_rolling.cpython-37.pyc
Назад
B �5�g/� � @ s( d dl m Z mZ d dlZd dlZd dlmZmZ d dlm Z d dl m Z m Z m Z mZmZmZmZmZmZmZ d dlmZ d dlmZ d dlmZ dd � Z d d � Z ej � d d de� dg�g�dd� �Z!ej � dedd�edd�g�dd� �Z"ej � dedd�edd�dg�dd� �Z#ej � ddd d d d d g�d!d"� �Z$d#d$� Z%ej � d%d&d'd'd(d(d(gd'd'd'd(d(gd(d'd'd'd(gd(d(d'd'd'gd(d(d(d'd'ggfd)d'd(d(d(d(gd'd'd(d(d(gd(d'd'd(d(gd(d(d'd'd(gd(d(d(d'd'ggfd*d'd'd(d(d(gd(d'd'd(d(gd(d(d'd'd(gd(d(d(d'd'gd(d(d(d(d'ggfd+d'd(d(d(d(gd(d'd(d(d(gd(d(d'd(d(gd(d(d(d'd(gd(d(d(d(d'ggfg�d,d-� �Z&ej � d.d/d*d0d0d0gfd/d&d0d0d0gfd/d)d0d0d0gfd/d+d0d0d0gfd1d*d0d d gfd1d&d0d0d0gfd1d)d2d0d0gfd1d+d2d d gfg�d3d4� �Z'd5d6� Z(d7d8� Z)ej � d9d+d)g�d:d;� �Z*ej � d<d d g�d=d>� �Z+ej � d<d d g�d?d@� �Z,ej � dAdBdCg�ej � dDd d*dEd d dFdddGdHdIdJg fd d&dEd d d dFdddGdHdIg fd d+ej-d d dFdddGdHdIdJg fd d)ej-d d d dFdddGdHdIg fd d*dEdFdddGdHdIdJdKdLg fd d&dEdFdddGdHdIdJdKdLg fd d+ej-d dFdddGdHdIdJdKg fd d)ej-d dFdddGdHdIdJdKg fg�dMdN� ��Z.dOdP� Z/ej � dDd d*ej-d d dFdddGdHej-ej-g fd d&ej-d d d dFdddGdHej-g fd d+ej-ej-d dFdddGdHej-ej-g fd d)ej-ej-d d dFdddGdHej-g fd d*ej-dFdddGdHdIdIej-ej-g fd d&ej-dFdddGdHdIdIdIej-g fd d+ej-ej-dFdddGdHdIej-ej-g fd d)ej-ej-dFdddGdHdIdIej-g fg�dQdR� �Z0ej � dSd*d dTdFdddGdHdIdJdKg fd&d dTdFdUdVdWdXdYdZd[g fd+ej-d dTdUdVdWdXdYdZd[g fd)ej-d dTdFdddGdHdIdJg fg�d\d]� �Z1ej � d^d_dFg�d`da� �Z2dbdc� Z3ddde� Z4dfdg� Z5dhdi� Z6djdk� Z7dldm� Z8dndo� Z9dpdq� Z:ej � drd'dsgdJ dtgdL dug dvgdw dxgdy dzgdH d{gdH d|gdw d}gd~ dgd� dsgd dgdL d|gdw dxgdI d�g d�gdI d�gdw d�gd d�gd� d�gdH dugdw dtgd~ dzgdJ d�gdG d�gdG dxgdw d�gd~ d�gd d�gdH d�gd dugdw dxgd~ d�gdJ d�g dzgd dxgd� d�gd fd(ej-gd dsgdy dtgdL dug dvgdw dxgdy dzgdH d{gdH d|gdw d}gd~ dgd� dsgd dgdL d|gdw dxgdI d�g d�gdI d�gdw d�gd d�gd� d�gdH dugdw dtgd~ dzgdJ d�gdG d�gdG dxgdw d�gd~ d�gd d�gdH d�gd dugdw dxgd~ d�gdJ d�g dzgd dxgd� fg�d�d�� �Z;d�d�� Z<d�d�� Z=d�d�� Z>ej � d�e dFddgdGdHdIgd���dFgdGgd��d gfdFdgdGdHgd��d dFgfdFddgdGdHdIgd��d dFdgfgddfe dFddgdGdHdIgd���dFgdGgd��d gfdFdgdGdHgd��d dFgfddgdHdIgd��dFdgfgddFfe dFddgdGdHdIgd���dFgdGgd��d gfdFdgdGdHgd��d dFgfddgdHdIgd��dFdgfgddfe dFddgdGdHdIgd���dFgdGgd��d gfdgdHgd��dFgfdgdIgd��dgfgdFdFfe dFddgdGdHdIgd���dFgdGgd��d gfdgdHgd��dFgfdgdIgd��dgfgdFd fe dFgdGgd���g ddfe dFgdGgd���g ddFfe � i g fgddfe dFej-dgej-dHdIgd���d2gej-gd��d gfdFej-gej-dHgd��d dFgfdFej-dgej-dHdIgd��d dFdgfgddfg �d�d�� �Z?ej � d�dFgdGgd��d gfdFdgdGdHgd��d dFgfddgdHdIgd��dFdgfgd�fdFgdGgd��d gfdFdgdGdHgd��d dFgfdFddgdGdHdIgd��d dFdgfgdfdFgdGgd��d gfdgdHgd��dFgfdgdIgd��dgfgd�fg�d�d�� �Z@d�d�� ZAej � d�edFddg�dFgd gfdFdgd dFgfdFddgd dFdgfgddfedFddg�dFgd gfdFdgd dFgfdFddgd dFdgfgddFfedFddg�dFgd gfdFdgd dFgfddgdFdgfgddFfedFddg�dFgd gfdFdgd dFgfddgdFdgfgddfedFddg�dFgd gfdgdFgfdgdgfgdFd fedFddg�dFgd gfdgdFgfdgdgfgdFdFfedFdg�dFgd gfdFdgd dFgfgdd feg d�d��g ddFfg�d�d�� �ZBej � d�d gdFgdgdgdGgged�dFd�d��ed�dFd�d��ed�dFd�d��ed�dFd�d��ed�dFd�d��gd�fd gd dFgdFdgddgddGgged�dFd�d��ed�dd�d��ed�dd�d��ed�dd�d��ed�dd�d��gd�fd gd dFgd dFdgdFddgdddGgged�dFd�d��ed�dd�d��ed�dd�d��ed�dd�d��ed�dd�d��gdfg�d�d�� �ZCej � d�d�d ie jDd�d�d�d�d�gddgd��fd�d�ie jDd�d�d�d�d�gd�dgd��fg�d�d�� �ZEej � d�dEd g�d�d�� �ZFd�d�� ZGd�d ZHd�dĄ ZId�dƄ ZJej � d�d�d�d�d�d�g�d�d΄ �ZKd�dЄ ZLej � d�d�edӃedԃg�d�dք �ZMd�d ZNej � d�ed�d�d�dݍd�fed�d�d�dݍd�fg�ej � d�d ej-d d dFdddGdHdIg fd ej-d dFdddGdHdIdJg fd ej-d dFddHdJdLd�d�g fg�d�d� ��ZOd�d� ZPej jQe� �o^e� d�d�ej � d�d dFd�d dTdTdd gfddFd�d d�d�d�d gfd dd�dTd dTdd gfddd�d�d d�d�d gfg�d�d� ��ZRd�d� ZSej � dd d d d�d�d d g�d�d�� �ZTej � d�d eUd��d�eUd��d�d�d�d�d�gfd eUd��d[eUd��d��d �d�d�dgfd eUd���deUd��d��d�dd��dgfd�eUd���deUd���d�d �d �d �d gfd�eUd���d eUd���d�d�d�d�dgfg��d�d� �ZVej � d�d dFdddGgdfe�dd�dH�d�d�fg��d�d� �ZW�d�d� ZXej � dd�d�g��d�d � �ZYej � �d d�d �d dE�d gfd�d �d!�d"�d#gfg��d$�d%� �ZZ�d&�d'� Z[ej � ddF�d(g��d)�d*� �Z\�d+�d,� Z]�d-�d.� Z^ej � �d/dF�d0�d1�d1�d2�d3gfd�d4�d4�d5�d6gfg��d7�d8� �Z_�d9�d:� Z`�d;�d<� ZadS (= � )�datetime� timedeltaN)�is_platform_arm�is_platform_mac)�UnsupportedFunctionCall) � DataFrame� DatetimeIndex� MultiIndex�Series� Timedelta� Timestamp� date_range� period_range� to_datetime� to_timedelta)� BaseIndexer)�Rollingc C s@ t ddddtjdgi�} | | �d��� | jddd��� d S )N�Br � � � )� min_periods)r �np�nan�rolling�sum)�df� r �V/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/window/test_rolling.py�test_doc_string s r c C sr | t d��j}|d� |dd� |ddd� |dddd� |ddd d� d }tjt|d �� |d � W d Q R X d S ) N� r r )�windowr )r! r T)r! r �centerFz&window must be an integer 0 or greater)�match���)�ranger �pytest�raises� ValueError)�frame_or_series�c�msgr r r �test_constructor( s r, �wg @�foor c C s� | t d��j}d}tjt|d�� ||d� W d Q R X d}tjt|d�� |d|d� W d Q R X d}tjt|d�� |dd |d � W d Q R X d S ) Nr zWwindow must be an integer|passed window foo is not compatible with a datetimelike index)r# )r! z min_periods must be an integerr )r! r zcenter must be a booleanr )r! r r" )r% r r& r' r( )r) r- r* r+ r r r �test_invalid_constructor< s r/ r! � )�daysc C s� d}t dt�|�itd|dd�d�}t�ddgt�d d d ��}|j| d ��� }t d|itd|dd�d�}t�||� |�d ��� }t�||� d S )N� �valuez 2015-12-24�D)�periods�freq)�indexg g �?g @g ;@r0 )r! �3D) r r �aranger �appendr r �tm�assert_frame_equal)r! �nr � expected_data�result�expectedr r r �&test_constructor_with_timedelta_windowR s rA r8 c C s� d}t dt�|�itd|dd�d�}t dt�tjdgt�dd d ��itd|dd�d�}|j| d d ��� }|j| d d �jt|d �}t � ||� t � ||� d S )Nr2 r3 z 2017-08-08r4 )r5 r6 )r7 g �?g @g ;@r0 r )r! r )�raw) r r r9 r r: �NaNr r �applyr; r< )r! rB r= r r@ Zresult_roll_sumZresult_roll_genericr r r �0test_constructor_timedelta_window_and_minperiodsf s rE �method�std�meanr �max�min�varc C sv t tdddg�dd�}d}tjt|d�� t|| �ddd� W d Q R X tjt|d�� t|| �tjd � W d Q R X d S ) Nr r � )r! z2numpy operations are not valid with window objects)r# r r0 )�dtype)r r r& r' r �getattrr �float64)rF �rr+ r r r �test_numpy_compatx s rQ c C s� |}t ddddddgi�}t ddddddgitddd �d �}t|jd| dd �|�� }t|jd | dd �|�� jd d�}t�||� d S )N�Ar r r r0 r �2020r )r5 )r7 )�closedr �2DT)�drop)r r rN r Z reset_indexr; r< )rT �arithmetic_win_operators� func_nameZdf_fixed�df_timer? r@ r r r �test_closed_fixed� s "rZ zclosed, window_selectionsZbothTF�left�rightZneitherc s� |�t ddddddgitddd �d �� t d� �fd d �|D �itddd �d �}�d kr`ddi}ni }t� jd| ddd���f |�}tj||dd� d S )NrR g g �?g @g @g @rS r )r5 )r7 c s" g | ]}t � d j| ��� �qS )rR )rN �iloc)�.0�s)rY rX r r � <listcomp>� s z9test_datetimelike_centered_selections.<locals>.<listcomp>�semZddofr rU r T)rT r r" F)Z check_dtype)r r rN r r; r< )rT Zwindow_selectionsrW r@ �kwargsr? r )rY rX r �%test_datetimelike_centered_selections� s 1 rc zwindow,closed,expected�3sg @Z2sg �?c C sX t d�t d�t d�g}|dddg|d�}|j| |dd��� }|||d�}t�||� d S )Nz20130101 09:00:01z20130101 09:00:02r )r7 T)rT r" )r r r r; � assert_equal)r! rT r@ r) r7 r r? r r r �,test_datetimelike_centered_offset_covers_all� s rf c C sR t td�tdddd�d�} | jdddd ��� }t d d d g| jd�}t�||� d S ) Nr0 z 2020-01-01r4 )�startr6 r5 )r7 rU r T)r! r r" g �?g �?r )r r% r r rH r7 r; �assert_series_equal)r_ r? r@ r r r �!test_even_number_window_alignment� s ri c C s� ddddddddg}t d|itddt|�d�d�}| rLdd d d d d d d g}ntjdd dd d d d g}t |dgtddt|�d�d �}|jt|�dd| d�}|�� }t�||� d S )Nr r Z binary_colz 2020-01-01rJ )rg r6 r5 )r7 gUUUUUU�?g �?g�������?g��@�m�?g.�_x%I�?)�columnsr7 r[ )r! rT r r" ) r r �lenr r r rH r; r< )r" �datar r> r@ r r? r r r � test_closed_fixed_binary_col s rm rT c C s` |}t t�d�tdddd�d�}|jd| d�}t||�� }t tjgd |jd�}t� ||� d S ) Nr �2000rU )r5 r6 )rl r7 �1D)rT )r7 ) r r r9 r r rN r r7 r; rh )rT rW rX �ser�rollr? r@ r r r �test_closed_empty# s rr �funcc C sJ t dgtddd�d�}t|jddd�| �� }t�|t tjg|jd �� d S ) Nr rn r )r5 )rl r7 �10Dr[ )rT )r7 ) r r rN r r; rh r r r7 )rs rp r? r r r �test_closed_one_entry/ s ru c C s� t dddgdddgd�tddd�d�}t|jdd d �d jd d d�| �� }tjdddg|jgdd�}tt j dt j g|d d�}t � ||� d S )Nr r r0 )rR r rn )r5 )rl r7 rR F)�sortr rt r[ )rT )rR N)Zarrays�names)rl r7 �name) r r rN �groupbyr r Z from_arraysr7 r r r r; rh )rs rp r? Zexp_idxr@ r r r � test_closed_one_entry_groupby7 s "rz � input_dtype�int�floatzfunc,closed,expectedg r r r rL � � � c C sT t t�d��| �tddd�d�}t|jd|d�|�� }t ||jd�}t� ||� d S )Nr2 rn )r5 )rl r7 r8 )rT )r7 ) r r r9 Zastyper rN r r7 r; rh )r{ rs rT r@ rp r? r r r � test_closed_min_max_datetimeF s r� c C sr t t�d�tddd�d�} | j| jddg d�} | jdd d ��� }t tjd d d d dddg| jd�}t � ||� d S )Nr2 rn )r5 )rl r7 r r )r7 r8 r[ )rT r r r0 r rL ) r r r9 r rV r7 r rJ r r; rh )rp r? r@ r r r �test_closed_uneven` s r� c C sd t t�d�tddd�d�}tj||jdd � < t|jdd|d�| �� }t ||jd �}t� ||� d S ) Nr2 rn )r5 )rl r7 ���r8 r )r rT )r7 ) r r r9 r r r7 rN r r; rh )rs rT r@ rp r? r r r �test_closed_min_max_minpk s r� zclosed,expectedg �?g �?g @g @g @g @g @g @c C sd t t�d�tddd�d�}|jd| d�}t ||jd�}|�� }t�||� |� d�}t�||� d S ) Nr2 rn )r5 )rl r7 r8 )rT )r7 g �?) r r r9 r r r7 �medianr; rh �quantile)rT r@ rp rq r? r r r �test_closed_median_quantile� s r� �roller�1sc C sX t � }t � �| ��� }t�||� t tg �d�}t tg �d��| ��� }t�||� d S )N)r7 )r r r r; r< r )r� r@ r? r r r �tests_empty_df_rolling� s r� c C sV t tjtjtjg�} t t�d���d�}|�� }t�|| � |�d�}t�|| � d S )Nr0 r g�������?) r r r r9 r r� r; rh r� )r@ rq r? r r r �!test_empty_window_median_quantile� s r� c C sb t tjg�} | jddd��� }t dg�}t�||� | jddd��� }t tjg�}t�||� d S )Nr r )r g )r r r r r r; rh )�xr? r@ r r r �test_missing_minp_zero� s r� c C sT t tjgd tddddg�d�} | jtd�dd ��� }t d | jd�}t� ||� d S ) Nr z 2017-01-01z 2017-01-04z 2017-01-06z 2017-01-07)r7 Z2dr )r g ) r r r r r r r r7 r; rh )r� r? r@ r r r � test_missing_minp_zero_variable� s r� c C sj t jddgdddggddgd�} tt�d �| d �}|�d ��� }t�|j |j � |j j d ddgksft �d S ) NrR r �Cr4 �E�1�2)rw )r2 rL )rj r0 ) r Z from_productr r �onesr Zcovr; Zassert_index_equalrj r7 rw �AssertionError)�colsr r? r r r �test_multi_index_names� s r� c C sz t t�d��}|�| �}|dkr8t dd� td�D ��}n t tjgd dgd gd �}|jd | d ��� }t� ||� d S ) N)r2 � r c S s$ i | ] }t jgd dgd |�qS )r g @r )r r )r^ �ir r r � <dictcomp>� s z)test_rolling_axis_sum.<locals>.<dictcomp>r� r g @� r2 r0 )�axis) r r r� �_get_axis_numberr% r r r r; r< )� axis_framer r� r@ r? r r r �test_rolling_axis_sum� s r� c C s� t td�td�d��}|�| �}|dkrDt dddgdddgd��}nt dddgdddgd��}|jd| dd��� }t�||� d S ) Nr0 )r� �y)r r7 g �?g @r r )r� r )r r% r� r �countr; r< )r� r r� r@ r? r r r �test_rolling_axis_count� s r� c C s\ t �ddt jddg�} | jdd� t| ��d��� }tt jdt jt jdg�}t�||� d S )Nr r0 r F)�writer r ) r �arrayr Zsetflagsr r rH r; rh )Zarrr? r@ r r r �test_readonly_array� s r� c C s� |}t dd� tddd|d�D ��}| dkrB|jjd| d ��� j}n|jd| d ��� }t d d� tdd d|d �D �d d� tddd|d�D ���}t�||� d S )Nc S s i | ]}d gd |�qS )r r r )r^ r� r r r r� s z)test_rolling_datetime.<locals>.<dictcomp>z 2019-8-01z 2019-08-03r4 )r6 �tz)r r7 rU )r� c S s i | ]}d gd |�qS )g �?r r )r^ r� r r r r� s r )r5 r6 r� c S s i | ]}d gd |�qS )g @r r )r^ r� r r r r� s z 2019-8-02z 2019-8-03)r r �Tr r r; r< )r� Ztz_naive_fixturer� r r? r@ r r r �test_rolling_datetime� s r� zcenter, expected_datag V@g @X@g �X@g �X@� g �W@� g @W@g @V@g X@g �W@� g �V@� g �V@g �U@g W@g �T@g �U@r2 g @U@g S@g @T@� g @S@r� c C s� t �� }t||td� dd�}tjjdd�}|jddt|�d�}t ||d ��}|j d d d � |j d dd| d�d � d�}|� d �}|�d �}t||dd�} t�|| � d S )Nim r4 )r6 i� )�seedr �d )�high�size)�DateCol�metricr� T)ZinplaceZ21Dr r[ )r! r rT r" r� rI )r7 rx )r �nowr r r �randomZ RandomState�randintrk r Z set_indexr Zagg�renameZ _with_freqr r; rh ) r" r> Z date_todayr1 Znprrl r r? r7 r@ r r r � test_rolling_window_as_string s [ r� c C sV t dddddgdgd�} | d jdddd��� }td d d d d gdd �}t�||� d S ) Nr r r �a)rj r0 T)r" r g �?g @)rx )r r rI r r; rh )r r? r@ r r r �test_min_periods1� s r� c C s@ | t d��jddd��� }| tjtjdddg�}t�||� d S )Nr r0 )r g @)r% r r� r r r; re )r) r? r@ r r r �#test_rolling_count_with_min_periods� s r� c C sj dddt jdddg}ddd dddd g}tjtd d �� | |��d��� }W d Q R X | |�}t�||� d S ) Nr r r0 r r rL g �?g @g @F)Zcheck_stacklevel)r r r; Zassert_produces_warning� FutureWarningr r� re )r) �valuesZexpected_countsr? r@ r r r �7test_rolling_count_default_min_periods_with_null_values� s r� z df,expected,window,min_periods)rR r c C s@ dd� |D �}x,t || j||d��D ]\}}t�||� q$W d S )Nc S s g | ]\}}t ||d ��qS ))r7 )r )r^ r� r7 r r r r` � s z/test_iter_rolling_dataframe.<locals>.<listcomp>)r )�zipr r; r< )r r@ r! r �actualr r r �test_iter_rolling_dataframe� s Fr� zexpected,windowrU ro c sr t dddddgdddddgtd dd d �d ��� � fd d�| D �} x,t| � j|dd��D ]\} }t�|| � qVW d S )Nr r r0 r r rL r~ r z 2016-01-01r4 )rg r5 r6 )rR r r� c s& g | ] \}}t |� j|d f d��qS )r� )r7 )r �loc)r^ r� r7 )r r r r` s z2test_iter_rolling_on_dataframe.<locals>.<listcomp>r� )�on)r r r� r r; r< )r@ r! r� r )r r � test_iter_rolling_on_dataframe� s r� c sv t dddgdddgd��� t� �d��d��} � fdd �dgddgdggD �}x"t| |�D ]\}}t�||� qZW d S ) Nr� r� r r r )r� �br� c s g | ]}� j |d gf �qS )r )r] )r^ �idx)r r r r` s z<test_iter_rolling_on_dataframe_unordered.<locals>.<listcomp>)r �listry r r� r; r< )�results� expectedsr? r@ r )r r �(test_iter_rolling_on_dataframe_unordered s r� z ser,expected,window, min_periods�int64)rM c C s@ dd� |D �}x,t || j||d��D ]\}}t�||� q$W d S )Nc S s g | ]\}}t ||d ��qS ))r7 )r )r^ r� r7 r r r r` G s z,test_iter_rolling_series.<locals>.<listcomp>)r )r� r r; rh )rp r@ r! r r� r r r �test_iter_rolling_series$ s #r� z expected,expected_index,windowz 2020-01-01r4 )r5 r6 z 2020-01-02z 2020-01-03z 2020-01-04z 2020-01-05c C s\ t td�tdddd�d�}dd� t| |�D �} x(t| |�|��D ]\} }t�|| � q@W d S )Nr z 2020-01-01r4 )rg r5 r6 )r7 c S s g | ]\}}t ||d ��qS ))r7 )r )r^ r� r� r r r r` z s z.test_iter_rolling_datetime.<locals>.<listcomp>)r r% r r� r r; rh )r@ Zexpected_indexr! rp r� r r r �test_iter_rolling_datetimeO s (r� zgrouping,_index�level)r r )r r )rw �by�X)r r )r r )r0 r )r r c C s� dd� }t dtd�idddddgd�}t ddd d d d gi|d�}d | krV|jddd�}|jf | ��d�j||dd�}t�||� d S )Nc W s( t | �dk rtd��| \}}|�� | S )Nr z The function needs two arguments)rk r( r )�argsr� �scaler r r � scaled_sum� s z4test_rolling_positional_argument.<locals>.scaled_sumr� r r r )rl r7 g g �?g �?g �?g @r� �ignore)rj �errors)r )rB r� )r r% rV ry r rD r; r< )�grouping�_indexrB r� r r@ r? r r r � test_rolling_positional_argument� s r� �addc C s� t dd| ddgitd�td�td�gd�}|�d��� jd d d d ��� }tdd dd�}t dtjtjtjddddgi|d�}t � ||� d S )NrR gVUUUUU%Cg �z19700101 09:00:00z19700101 09:00:03z19700101 09:00:06)r7 r� rd r[ r0 )rT r r~ �S)r5 r6 gUUUUUU%Cg �q �q Cg �q �q Cg ) r r ZresampleZffillr rH r r r r; r< )r� r r? �datesr@ r r r �*test_rolling_numerical_accuracy_kahan_mean� s$ r� c C sX t ddddddgdgd�} | d �d��� }ttjtjddddgdd�}t�||� d S ) Ng�rh��|@g'1� Z��g r� )rj r0 gsh��|?�?)rx )r r r r r r r; rh )r r? r@ r r r �)test_rolling_numerical_accuracy_kahan_sum� s r� c C s^ t dddd��tdg��} tj�t| ��}td|i| d�}|�d�� � }t � ||dg � d S )Nz 2020-01-01z 2020-01-02Z60s)rg �endr6 z 2020-01-03rl )r7 ) r r: r r r� �randrk r r rH r; r< )r7 rl r r? r r r �$test_rolling_numerical_accuracy_jump� s r� c C s8 t ddddgtdd�d�} | �d��� }t�|| � d S )Ng�(�@�S ?ga2U0*�3?g �z 1999-02-03z 1999-02-06)rl r7 r )r r r rH r; rh )r_ r? r r r �,test_rolling_numerical_accuracy_small_values� s r� c C sp t dddd�} ttd�| tjd�}d|d< |�d��� }ttjtjtjtjd d d d d d g | d �}t� ||� d S )Nz 2015-01-01r2 r4 )r5 r6 )rl r7 rM gТ������r r g��k�/��g @g @g @)r7 ) r r r% r rO r rH r r; rh )r� �dsr? r@ r r r �(test_rolling_numerical_too_large_numbers� s" r� )rs r3 )r g @)rI g �?)rJ g �?)rH g �?)r� g �?c C sl t dddgdddgd�}d|d< t|jdddd�| �� }t ddg||g||gd �ddgd �}t�||� d S ) Nr r r� r� r* )r7 rj g �?)r! r r� )r� r� r* )r7 )r rN r r; r< )rs r3 r r? r@ r r r � test_rolling_mixed_dtypes_axis_1� s r� c C s� t ddddtjtjtjgdddtjtjtjtjgdddtjdtjdgg�} | jdddd��� }t dd d d d d d gdd d d d d d gdd d d ddd gg�}t�||� d S )Nr r r r r~ rj )r! r r� g g �?g @g @g @g @g @)r r r r r r; r< )r r? r@ r r r � test_rolling_axis_one_with_nan s r� r3 �testz 2019-12-31z1 days 06:05:01.00003c C sL t dddgi�}| |d< |jdddd��� }t dddgi�}t�||� d S )Nr� r r r� )r! r r� g �?g @)r r r r; r< )r3 r r? r@ r r r �&test_rolling_axis_1_non_numeric_dtypes s r� c C s� t dd gddgddgd��} t dtjgddgd d gd��}| jdd dd ��� }t�||� | jjdd d ��� j}t�||� d S )Nr r r r~ r )rR r r� g �?g @g &@g *@r )r r! r� )r r! )r r r r r r; r< r� )r r@ r? r r r � test_rolling_on_df_transposed, s r� )r7 r! z2020-01-01 08:00z2020-01-01 08:08r� )rg r� r6 Z2Tz2020-01-01 12:00Z30TZ1h)rs r� � c C sP t ddddddddd g | d �}t|j|d d �|�� }t || d �}t�||� d S ) Nr r r r0 r r rL r~ r )r7 r[ )rT )r rN r r; rh )r7 r! rs r� r� r? r@ r r r �test_rolling_period_index7 s r� c C s^ | dddg�}|j ddd��� }t|t�r8t|d j�}ttjgdgd �}t� ||� d S )Nr r r )r g�;f���?) r ra � isinstancer r r� r r r; rh )r) �objr? r@ r r r �test_rolling_semT s r� zGH 38921)�reason)rs � third_valuer� g|����Fg >g��foCgP�p���?g���۞��?c C sH t dd|ddddg�}t|�d�| �� }t tjg| �}t�||� d S )Nl �;� r r r0 )r rN r r r r; rh )rs r� r� r� r? r@ r r r �!test_rolling_var_numerical_issues^ s r� c C s t tjtjddddddddg tjtjddddddddg d�t�td�d ftd�d ftd �d ftd �d ftd �d ftd �d ftd �d ftd �d ftd�d ftd�d fg �d�} t dddddgdddddgd�td�td �td �td �td�gd�}|jdd��� }t� | |� d S )Ng�������?g �g �?g6�<A��Կg �?g�����?)r rR z20130101 09:00:00r rR z20130102 09:00:02z20130103 09:00:03z20130105 09:00:05z20130106 09:00:06)r7 r r r r r0 r~ rL r� Z3d)r! ) r r r r � from_tuplesr r �corrr; r< )�expr �resr r r �,test_timeoffset_as_window_parameter_for_corrp sR r� ZskewZkurtc C s� t dt�dd�d i�}t d|d ddd� i|jddd� d�}t|jdd �| �� }t|jdd �| �� }t�|jddd� dd � |jd d� ��� d k s�t �dS ) zf Make sure that decreasing indices give the same results as increasing indices. GH 36933 r� i�r2 r Nr$ )r7 r )r! ���r g�-���q=) r r r9 r7 rN r �absr� rI r� )rF r Z df_reverseZ increasingZ decreasingr r r � test_rolling_decreasing_indices� s *r� zmethod,expectedr g �E@g����� a@g �E@g��C�ݽW@g �f@g �s@g �5@g @g��`UU"@g *@g �1@g >@g �K@g �a@g/�e����?g8j��{ �?g��LM��?g�4`��i�?g +TT��?g�����?g %�b���g�Oq&� �g zCen@��gk�=��Cпg�~�~�gf`8g�ݿc C sx ddddddddg}t dt�t|��d i�}G dd� dt�}|d|d�}t|�|�| �� }t d|i�}t�||� d S ) z� Make sure the (rare) branch of non-monotonic indices is covered by a test. output from 1.1.3 is assumed to be the expected output. Output of sum/mean has manually been verified. GH 36933. TFr� r c @ s e Zd Zdd� ZdS )z1test_rolling_non_monotonic.<locals>.CustomIndexerc S sp t j|t jd�}t j|t jd�}xFt|�D ]:}| j| rNd||< |d ||< q*|||< || j ||< q*W ||fS )N)rM r r )r �emptyr� r% � use_expanding� window_size)�selfZ num_valuesr r" rT rg r� r� r r r �get_window_bounds� s zCtest_rolling_non_monotonic.<locals>.CustomIndexer.get_window_boundsN)�__name__� __module__� __qualname__r� r r r r � CustomIndexer� s r r )r� r� N) r r r9 rk r rN r r; r< )rF r@ r� r r Zindexerr? r r r �test_rolling_non_monotonic� s > r z 2001-01-01)r6 r5 c C sd t dddddg| d�}|�� }d|dd�< |�|��|�}t tjtjdddg| d�}t�||� d S ) Nr r r0 r r )r7 g r )r �copyr r� r r r; Zassert_almost_equal)r7 r! r� r� r? r@ r r r �!test_rolling_corr_timedelta_index s r c C s� dt jddt jgdddddgd�} t| �}|jd d d �jddd ��� }td dddddgitddt jgdddddggdddddgdddddggd d gd�d�}t�||� d S )N�g1�g2r r r r0 r )�groupr r F)Zdropna)r r g g @g @g �?g @)rw )r7 ) r r r ry r rH r r; r< )rl r r? r@ r r r �!test_groupby_rolling_nan_included# s " r c C s` t tj�d��}|�� }t|�d�| �� }t�||� |d }t|�d�| �� }t�||� d S )Nr2 r0 iP� ) r r r� r� r rN r r; rh )rF rp Zser_copyr@ r? r r r �*test_rolling_skew_kurt_numerical_stability9 s r )rF r� g��+��X�?gJ_9�?g @g��B�ʠ�g333333�g� >��@c C sL t dddddddg�}t|�d�| �� }t tjgd | �}t�||� d S )Ni��- r r r0 r i� )r rN r r r r; rh )rF r� r_ r? r@ r r r �(test_rolling_skew_kurt_large_value_rangeE s r c C s4 t jtdd�� ttd��jddd� W d Q R X d S )Nz!method must be 'table' or 'single)r# r r. )rF )r& r' r( r r% r r r r r �test_invalid_methodT s r �1dc C s� t dddd�}|tdd�|d�}|jddd ��� }|tjdd g|d�}t�||� |jd d d � jddd ��� }t ddd d�}|tjd d g|d�}t�||� d S )Nz 2020-01-01z 2020-01-03r )rg r� r6 r r )r7 r[ )rT r r$ z-1dr0 ) r r% r r r r r; re r] )r! r) r� r� r? r@ r r r �.test_rolling_descending_date_order_with_offsetY s r c C sH t ddddg�} | �d��� }t tjtjddg�}tj||ddd� d S )Nr~ r r0 gUUUUUU�?r gV瞯�<)�atol�rtol)r r rK r r r; rh )r_ r? r@ r r r �,test_rolling_var_floating_artifact_precisionh s r c C sB t dddg�} | �d��� }t tjddg�}tj||ddd� d S )NgQ�؎� �>g 4PS�ȡ>r g�[��^>>gV瞯�<)r r )r r rG r r r; rh )r_ r? r@ r r r � test_rolling_std_small_valuesp s r zstart, exp_valuesg� ��Q��?gX9��v��?g�I +��?gˡE��?g����MbP?g�~j�t�X?g�<#� n?c C s� t dddtjddtjtjtjtjtjtjddg�}|ddddtjtjddg }t |tt| t|�| ��d �}|j| d � jd d d ��� }t � ||� d S ) Ng� ��Q��?g����MbP?g����Mb`?g����Mb�?g{�G�zt?g�������?g�<#� n?g= ףp=�?)r7 r r )r ) r r rC r� r% rk r] r rH r; r< )rg Z exp_valuesr r� r@ r? r r r �3test_rolling_mean_all_nan_window_floating_artifacts~ s: r c C sP t dddtjtjtjg�} | jddd��� }t dddd dd g�}t�||� d S ) Ng����Mb`?g����Mb�?g{�G�zt?r0 r )r g{�G�z�?g� ��Q��?g9��v���?g )r r rC r r r; r< )r r? r@ r r r �2test_rolling_sum_all_nan_window_floating_artifacts� s r c C s6 t td��} | �d��� }t tjg�}t�||� d S )Nr r )r r% r rJ r r r; rh )r_ r? r@ r r r �test_rolling_zero_window� s r )br r �numpyr r& Z pandas.compatr r Z pandas.errorsr Zpandasr r r r r r r r r r Zpandas._testingZ_testingr; Zpandas.api.indexersr Zpandas.core.windowr r r, �markZ parametrizer� r/ rA rE rQ rZ rc rf ri rm rr ru rz r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� Zxfailr� r� r� r} r r r r r r r r r r r r r r r r �<module> sX 0 $&(" . " $"" $0 * &� g&� w $ , , ( (22* $ & * $ =$$$&& ( (*
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка