Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/window/moments/__pycache__/test_moments_ewm.cpython-37.pyc
Назад
B �5�g�( � @ s| d dl Zd dlZd dlmZmZ d dlmZ ej � ddddg�dd� �Z ej � ddddg�d d � �Z d d � Z ej � d ddg�ej � dddg�dd� ��Zdd� Zej � deejddg�ddejddgfeejddg�ddejddgfeejddg�ddejddgfeejddg�ddejddgfedejdg�dddejdgfedejdg�dddejdgfedejdg�dddejdgfedejdg�dddejdgfeejdejejdejg�ddejdejejdejgfeejdejejdejg�ddejdejejdejgfeejdejejdejg�ddejdejejdejgfeejdejejdejg�ddejdejejdejgfedejdd g�dddejddgfedejdd g�dddejddgfedejdd g�dddejd d gfedejdd g�dddejd dgfg�d d � �Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zej � d+dddg�d,d-� �Zej � d.d d/g�ej � ddddg�d0d1� ��ZdS )2� N)� DataFrame�Series�name�varZstd�meanc C s& t | jdd�|�� }t|t�s"t�d S )N� )�com)�getattr�ewm� isinstancer �AssertionError)�seriesr Z series_result� r �b/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/window/moments/test_moments_ewm.py�test_ewma_series s r c C s& t | jdd�|�� }t|t�s"t�d S )Nr )r )r r r r r )�framer Z frame_resultr r r �test_ewma_frame s r c C sF t t�d��} d| d< | jddd��� �� }t�|d �dk sBt�d S )Ni� � � �d F)�span�adjustg{�G�z�?)r �npZzerosr r �sum�absr )�vals�resultr r r �test_ewma_adjust s r r TF� ignore_nac C sZ t ddddg�}| r&t ddddg�}nt ddd d g�}|jd| |d ��� }t�||� d S ) Ng �?g @g @g @g�������?gʍ"k �@gY��;�@gR��TU�?g��S �@g ���@)r r r )r r r �tm�assert_series_equal)r r �s�expectedr r r r �test_ewma_cases s r# c C s� t dgtjgd dg �} | jdd��� }t�|t dgt| � �� t tjgd dg tjgd dg �} | jdd��� }t�|t tjgd dgd �� d S )Ng �?r )r � � )r r �nanr r r r �len)r! r r r r �test_ewma_nan_handling. s (r( zs, adjust, ignore_na, wg �?g @Y@gVUUUUU�?gUUUUUU�?g �q �q�?gj/�����?g I@g �q �q�?g| �%���?c C sl | � |��� t|��� jdd�}| jd||d��� }t�||� |dkrh| jd|d��� }t�||� d S )NZffill)�methodg @)r r r F)r r )�multiplyZcumsumr Zfillnar r r r )r! r r �wr" r r r r � test_ewma_nan_handling_cases8 s " r, c C s� | j dd��� }| j dd��� }t�||� d}tjt|d�� | j ddd� W d Q R X d}tjt|d�� | � � �� W d Q R X d S ) Ng #@)r � )r z8comass, span, halflife, and alpha are mutually exclusive)�match)r r z1Must pass one of comass, span, halflife, or alpha)r r r �assert_almost_equal�pytest�raises� ValueError)r �A�B�msgr r r �test_ewma_span_com_args� s r6 c C s� | j dd��� }| j dd��� }t�||� d}tjt|d�� | j ddd � W d Q R X tjt|d�� | j d dd � W d Q R X tjt|d�� | j d ddd � W d Q R X d }tjt|d�� | � � W d Q R X d S )Ng���H��+@)r g $@)�halflifez8comass, span, halflife, and alpha are mutually exclusive)r. r- �2 )r r7 g #@)r r7 )r r r7 z1Must pass one of comass, span, halflife, or alpha)r r r r/ r0 r1 r2 )r r3 r4 r5 r r r �test_ewma_halflife_arg� s r9 c C s� t j�d�} t �dd�}t j| |< t| �}|jdd��� }|jdd��� }|jdd ��� }|jd d ��� }t� ||� t� ||� t� ||� d S ) Nr r- �( g�V��R��?)�alphag�ƷC��?)r g�_��!�@)r gXlWY��?)r7 ) r �random�randn�arange�NaNr r r r r )�arr�locsr! �a�b�c�dr r r �test_ewm_alpha� s rF c C s� | }d}t jt|d�� |�� W d Q R X d}t jt|d�� |jddd� W d Q R X t jt|d�� |jddd� W d Q R X t jt|d�� |jddd� W d Q R X d S ) Nz1Must pass one of comass, span, halflife, or alpha)r. z8comass, span, halflife, and alpha are mutually exclusiveg $@g �?)r r; )r r; )r7 r; )r0 r1 r2 r )r r! r5 r r r �test_ewm_alpha_arg� s rG c C s� t j�d�} t �dd�}t j| |< t| �}d}tjt|d�� |j dd� W d Q R X |j dd� |j d d� d }tjt|d�� |j dd � W d Q R X tjt|d�� |j dd � W d Q R X tjt|d�� |j d d � W d Q R X |j d d � |j dd � d}tjt|d�� |j dd� W d Q R X tjt|d�� |j dd� W d Q R X |j d d� d}tjt|d�� |j dd� W d Q R X tjt|d�� |j dd� W d Q R X |j d d� |j d d� tjt|d�� |j dd� W d Q R X d S )Nr r- r: z comass must satisfy: comass >= 0)r. g��������)r g g�������?z span must satisfy: span >= 1)r g�������?g �?g�������?z#halflife must satisfy: halflife > 0)r7 z"alpha must satisfy: 0 < alpha <= 1)r; ) r r< r= r> r? r r0 r1 r2 r )r@ rA r! r5 r r r �test_ewm_domain_checks sB rH r) c C s4 t g tjd�}|�d�}t|| �� }t�||� d S )N)�dtype� )r r �float64r r r r/ )r) r r r r r r �test_ew_empty_series2 s rL � min_periodsr c C s� t j�d�}t j|d d�< t j|dd �< t|�}t|jddd�|�� }|d d� �� �� s^t �|dd � �� � � rvt �t|jd| d�|�� }|dkr�|d d� �� �� s�t �|dd � �� � � r�t �n0|d d� �� �� s�t �|dd � �� � � r�t �ttt d�jd| d�|�� }t � |td d�� ttd g�jd| d �|�� }|dk�r^t � |td g�� nt � |tt jg�� ttt �d��jdd �|�� }|jt jk�s�t �d S ) Nr8 r i����r$ )r rM � r )rI rK g �?)rM )r )r r< r= r? r r r Zisna�allr �any�objectr r r> rI Zfloat_)rM r r@ r! r Zresult2r r r �test_ew_min_periods; s* rR )�numpyr r0 Zpandasr r Zpandas._testingZ_testingr �markZ parametrizer r r r# r( r& r, r6 r9 rF rG rH rL rR r r r r �<module> s� )
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка