Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/window/__pycache__/test_ewm.cpython-37.pyc
Назад
B �5�g� � @ s> d dl Zd dlZd dlmZ d dlmZmZmZm Z d dl m Z d dl mZ dd� Zdd� Zej�d d d d g�d d� �Zdd� Zdd� Zdd� Zdd� Zej�de�d��d��d�e dd dd �e dd dd ��d �d g�ej�d d d!g�d"d#� ��Zd$d%� Z d&d'� Z d(d)� Z ej�d*d+d,d-d.g�d/d0� �Z d1d2� Z d3d4� Z!dS )5� N)�UnsupportedFunctionCall)� DataFrame� DatetimeIndex�Series� date_range)�ExponentialMovingWindowc C s0 t ddddtjdgi�} | | jdd��� d S )N�Br � � � g �?)�com)r �np�nan�ewm�mean)�df� r �R/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/window/test_ewm.py�test_doc_string s r c C s� | t d��j}|dd� |dd� |dd� |dd� |dd d � |dd d � |dd d � d }tjt|d �� |ddd� W d Q R X tjt|d �� |ddd� W d Q R X tjt|d �� |ddd� W d Q R X d}tjt|d �� |dd� W d Q R X d}tjt|d �� |dd� W d Q R X d}tjt|d �� |dd� W d Q R X d}x2dD ]*}tjt|d �� ||d� W d Q R X �qRW d S )N� g �?)r g �?)�span)�alphag �?)�halflife)r r )r r )r r z8comass, span, halflife, and alpha are mutually exclusive)�match)r r )r r )r r z comass must satisfy: comass >= 0g �z span must satisfy: span >= 1z#halflife must satisfy: halflife > 0r z"alpha must satisfy: 0 < alpha <= 1)g �g �?)�ranger �pytest�raises� ValueError)Zframe_or_series�c�msgr r r r �test_constructor s8 r �method�stdr �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 � g �?)r z2numpy operations are not valid with window objects)r r � )Zdtype)r r r r r �getattrr Zfloat64)r! �er r r r �test_numpy_compatC s r( c C s< d} t jt| d�� ttd��jt�d�d� W d Q R X d S )Nz%times must be datetime64\[ns\] dtype.)r r )�times)r r r r r r r �arange)r r r r �!test_ewma_times_not_datetime_typeP s r+ c C sB d} t jt| d��& ttd��jt�d��d�d� W d Q R X d S )Nz,times must be the same length as the object.)r r r zdatetime64[ns])r) ) r r r r r r r r* �astype)r r r r � test_ewma_times_not_same_lengthV s r- c C sD d} t jt| d��( ttd��jdt�d��d�d� W d Q R X d S )Nz6halflife must be a string or datetime.timedelta object)r r r zdatetime64[ns])r r) ) r r r r r r r r* r, )r r r r �#test_ewma_halflife_not_correct_type\ s r. c C s6 d}t jt|d�� ttd��j| d� W d Q R X d S )NzKhalflife can only be a timedelta convertible argument if times is not None.)r r )r )r r r r r r )�halflife_with_timesr r r r � test_ewma_halflife_without_timesb s r0 r) � z datetime64[D]zdatetime64[ns]�2000�D)�freq�periods�UTC�time_col� min_periodsr c C sl | }t �d�}t j|d d d�< t|tdddd�d��}|j|||d��� }|jd |d ��� }t�||� d S ) Ng $@r r2 r3 r1 )r4 r5 )�Ar7 )r r8 r) g �?)r r8 ) r r* r r r r r �tm�assert_frame_equal)r/ r) r8 r �datar �result�expectedr r r �"test_ewma_with_times_equal_spacingh s r? c C s^ | }d}t dddg��|�}t�d�}t|�}|j||d��� }tddd g�}t�||� d S ) Nz23 daysz 2020-01-01z2020-01-10T00:04:05z2020-02-23T05:00:23r% )r r) g g��E(�?g� *�M��?) r � tz_localizer r* r r r r: r; )Ztz_aware_fixture�tzr r) r< r r= r> r r r �%test_ewma_with_times_variable_spacing| s rB c C sD t td��}tdg�}tjtdd�� |jd| |d� W d Q R X d S )Nr ZNaTz$Cannot convert NaT values to integer)r g�������?)r r r) )r r r r r r r )r/ �serr) r r r �test_ewm_with_nat_raises� s rD c C st | }t �d�}t j|d d d�< tdddd�}t||d��}|j||d�d �� }|jd d �d �� }t�||� d S ) Ng $@r r2 r3 r1 )r4 r5 )r9 r )r r) r9 g �?)r ) r r* r r r r r r: �assert_series_equal)r/ r r< r) r r= r> r r r �test_ewm_with_times_getitem� s rF �argr r r r c sn | dd|d|i}t td�td�d��jf |�� � fdd�� jD �}� d }� fdd�|jD �}||ksjt�d S ) Nr �adjust� ignore_na)r9 r c s i | ]}t � |�|�qS r )r& )�.0�attr)r r r � <dictcomp>� s z8test_ewm_getitem_attributes_retained.<locals>.<dictcomp>r9 c s i | ]}t � |�|�qS r )r& )rJ rK )r r r rL � s )r r r � _attributes�AssertionError)rG rH rI �kwargsr> Z ewm_slicer= r )r r �$test_ewm_getitem_attributes_retained� s rP c C sR t td��} t�t�� | jdd��� }W d Q R X | jdd��� }t�||� d S )Nr g�������?)r ) r r r: Zassert_produces_warning� FutureWarningr Zvolr" rE )rC r= r> r r r �test_ewm_vol_deprecated� s rR c C s@ t jtdd��( ttd��jddtdddd�d � W d Q R X d S ) Nz)times is not supported with adjust=False.)r r g�������?Fr2 r3 )r4 r5 )rH r) )r r �NotImplementedErrorr r r r r r r r �#test_ewma_times_adjust_false_raises� s rT )"�numpyr r Z pandas.errorsr Zpandasr r r r Zpandas._testingZ_testingr: Zpandas.core.windowr r r �markZ parametrizer( r+ r- r. r0 r* r, r@ r? rB rD rF rP rR rT r r r r �<module> s2 ,
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка