Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/window/__pycache__/test_groupby.cpython-37.pyc
Назад
B �5�gP� � @ s� d dl Zd dlZd dlmZmZmZmZmZm Z m Z d dl m Z d dlmZ d dlmZ G dd� d�ZG dd� d�ZG d d � d �ZdS ) � N)� DataFrame�Index� MultiIndex�Series� Timestamp� date_range� to_datetime)� BaseIndexer)� get_groupbyc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zej� d d d d d ej dej� d�d�ddg�dd� �Z ej� d ddg�dd� �Z ej� dddd d d g�d d � �Zej� d!d"d#gd$d%gg�d&d'� �Zej� d d"d$g�d(d)� �Zej� d d"d$g�d*d+� �Zej� d,d$d-d-gd-d.ggfd"d-d%gd%d-ggfg�d/d0� �Zd1d2� Zd3d4� Zej� d5d-d6gd7d8gg�d9d:� �Zd;d<� Zd=d>� Zej� d?d@dAdBg�dCdD� �ZdEdF� ZdGdH� ZdIdJ� ZdKdL� Z ej� dMd d g�dNdO� �Z dPdQ� Z dRdS� Z dTdU� Z dVdW� Z!ej� dXdYdZd#d[�fd\i fg�d]d^� �Z"ej� d_d`daidafd`dbidcfg�ddde� �Z#ej� dfd6d8g�dgdh� �Z$didj� Z%dkdl� Z&dmdn� Z'ej� doe(�)dpdqg�drdsgg�dtdu� �Z*dvdw� Z+ej� dxdygdzd{d|d|d}gigdydcge,d~�e,d�e,d~�e,d�gd{d}d|d�gd��gg�d�d�� �Z-dbS )�� TestRollingc C s4 t dgd dgd dgd t�d�d��| _d S ) N� � � � � � �( )�A�B)r �np�arange�frame)�self� r �V/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/window/test_groupby.py� setup_method s zTestRolling.setup_methodc C sb d}t jt|d�� | jjddd� W d Q R X | j�d�}|jrDt�t| jddd�}|js^t�d S )Nz4groupby\(\) got an unexpected keyword argument 'foo')�matchr r )�fooT)�by�mutated)�pytest�raises� TypeErrorr �groupbyr �AssertionErrorr )r �msg�gr r r � test_mutated s zTestRolling.test_mutatedc C s� | j �d�}t| j ddd�}|j�dd� �}|�d��� j}t�||� |�d�j�� }t�||� |j�d��� }t�||� | j j�| j j ��d��� }t�||� d S )Nr T)r r c S s | � d��� S )Nr )�rolling�mean)�xr r r �<lambda>% � z*TestRolling.test_getitem.<locals>.<lambda>r ) r r# r r �applyr( r) �tm�assert_series_equalr )r r& � g_mutated�expected�resultr r r � test_getitem! s zTestRolling.test_getitemc C sj | j �d�}|jddd�}t| j ddd�}|j�dd� �}|j�� }t�||� |j�� }t�||� d S ) Nr r r )� min_periodsT)r r c S s | j ddd��� S )Nr r )r4 )r( �count)r* r r r r+ 9 r, z3TestRolling.test_getitem_multiple.<locals>.<lambda>) r r# r( r r r- r5 r. r/ )r r& �rr0 r1 r2 r r r �test_getitem_multiple3 s z!TestRolling.test_getitem_multiple�f�sumr) �min�maxr5 z ignore:min_periods:FutureWarning)Zmarks�kurt�skewc sr | j �d�}|jdd�}t|� �� }|�� fdd��}|jddd�}t�| j d td�g�}||_ t � ||� d S ) Nr � )�windowc s t | �d�� �� S )Nr> )�getattrr( )r* )r8 r r r+ U r, z*TestRolling.test_rolling.<locals>.<lambda>r )�axisr ) r r# r( r@ r- �dropr � from_arrays�range�indexr. �assert_frame_equal)r r8 r& r6 r2 r1 �expected_indexr )r8 r � test_rollingA s zTestRolling.test_rolling�std�varc sv | j �d�}|jdd�}t|� �dd�}|�� fdd��}|jddd�}t�| j d td �g�}||_ t � ||� d S ) Nr r> )r? r )�ddofc s t | �d�� �dd�S )Nr> r )rK )r@ r( )r* )r8 r r r+ c r, z/TestRolling.test_rolling_ddof.<locals>.<lambda>)rA r ) r r# r( r@ r- rB r rC rD rE r. rF )r r8 r& r6 r2 r1 rG r )r8 r �test_rolling_ddof] s z TestRolling.test_rolling_ddof� interpolation�linear�lower�higher�midpoint�nearestc st | j �d�}|jdd�}|jd� d�}|�� fdd��}|jddd �}t�| j d td �g�}||_ t � ||� d S ) Nr r> )r? g�������?)rM c s | � d�jd� d�S )Nr> g�������?)rM )r( �quantile)r* )rM r r r+ t r, z3TestRolling.test_rolling_quantile.<locals>.<lambda>r )rA r ) r r# r( rS r- rB r rC rD rE r. rF )r rM r& r6 r2 r1 rG r )rM r �test_rolling_quantilek s z!TestRolling.test_rolling_quantilezf, expected_val�corrr �covg �?c C s� t td�dgd dgd dddddgd d���dd g�}t td�dddddgd ���d �}t|jd d ��d�|�|�}tjg|gd d }t |d gtj dgd dgd dgd dgd t tdd��d gddd gd�d�}t � ||� d S )N� r � r r r> )�value�idx1�idx2rZ r[ )rY r[ r )�levelrY � )�names)�columnsrE ) r rD � set_indexr@ r# r( r �nanr rC �listr. rF )r r8 Z expected_val�df�otherr2 � expected_datar1 r r r �/test_rolling_corr_cov_other_same_size_as_groups} s 0 z;TestRolling.test_rolling_corr_cov_other_same_size_as_groupsc sZ �j �d�}|jdd�}t|� ��j �}� �fdd�}|�|�}tj|d< t�||� d S )Nr r> )r? c s t | �d�� ��j�S )Nr> )r@ r( r )r* )r8 r r r �func� s zITestRolling.test_rolling_corr_cov_other_diff_size_as_groups.<locals>.func) r r# r( r@ r- r ra r. rF )r r8 r& r6 r2 rg r1 r )r8 r r �/test_rolling_corr_cov_other_diff_size_as_groups� s z;TestRolling.test_rolling_corr_cov_other_diff_size_as_groupsc sP | j �d�}|jdd�}t|j� �dd�}� fdd�}|�|�}t�||� d S )Nr r> )r? T)�pairwisec s t | j�d�� �dd�S )Nr> T)ri )r@ r r( )r* )r8 r r rg � s z8TestRolling.test_rolling_corr_cov_pairwise.<locals>.func)r r# r( r@ r r- r. r/ )r r8 r& r6 r2 rg r1 r )r8 r � test_rolling_corr_cov_pairwise� s z*TestRolling.test_rolling_corr_cov_pairwisezfunc, expected_valuesg �?g @c C s� t ddddgddddgddddgd��}|�d ��d �}t||�� }t dtjg |d dtjg dtjg |d dtjg d �tjd d ddddddgd d d gd�d�}t� ||� d S )N�g1�g2r r r r] r> )�a�b�crm r )rn ro )rk r rn )rk r ro )rk r rn )rk r ro )rk r rn )rk r ro )rl r rn )rl r ro )r^ )rE ) r r# r( r@ r ra r � from_tuplesr. rF )r rg Zexpected_valuesrc Zrolr2 r1 r r r � test_rolling_corr_cov_unordered� s( "z+TestRolling.test_rolling_corr_cov_unorderedc sx | j �d�}|jdd�}|jdd� � d�}|�� fdd��}|jddd �}t�| j d td �g�}||_t � ||� d S ) Nr r> )r? c S s | � � S )N)r9 )r* r r r r+ � r, z0TestRolling.test_rolling_apply.<locals>.<lambda>)�rawc s | � d�jdd� � d�S )Nr> c S s | � � S )N)r9 )�yr r r r+ � r, zBTestRolling.test_rolling_apply.<locals>.<lambda>.<locals>.<lambda>)rr )r( r- )r* )rr r r r+ � r, r )rA r ) r r# r( r- rB r rC rD rE r. rF )r rr r& r6 r2 r1 rG r )rr r �test_rolling_apply� s z TestRolling.test_rolling_applyc C s� t dgd dgd dgd d��}|�d�}t�dd d d d d g�}dd g|_t tjddgd dg|d�}|jdd��� }t � ||� |�� |jdd��� }t � ||� d S )Nr r �barr r] )r r r )ru r )ru r> )ru rX )r r )r r )r r g @r r )r_ rE )r? ) r r# r rp r^ r ra r( r9 r. rF )r rc r& �mir1 r2 r r r � test_rolling_apply_mutability� s $ z)TestRolling.test_rolling_apply_mutabilityzexpected_value,raw_valueTg Fc C sp dd� }t dddgdddgd��}|�d�j�d�j||d�}t|gd tjd dd gd �d d �}t� ||� d S ) Nc S s t t| tj��S )N)�int� isinstancer Zndarray)r* r r r r s z-TestRolling.test_groupby_rolling.<locals>.foor r r )�idrY rz )rr ))r r )r r )r r )r^ rY )rE �name) r r# rY r( r- r r rp r. r/ )r Zexpected_valueZ raw_valuer rc r2 r1 r r r �test_groupby_rolling� s z TestRolling.test_groupby_rollingc C s� t tdd��}|�|�jddd��� }t tjgd t�d�d�}t � ||� t tdd��}|�|�jddd��� }t tjgd t�d �d�}t � ||� t d gd d gd td �d��}|�d �jddd��� }t tjdddtjtjddddtjg tjdd d gd�d gd�}t � ||� t d gd d gd td�d��}|�d �jddd��� }t tjdddtjtjdddtjg tjdd d gd�d gd�}t � ||� d S )Nr r] Tr )�centerr? rX ))r r )r r )r r )r> r )rX r> )rE r> ))r r )r r )r r )r> r rm rn � )rm rn r � r � ) )rm r )rm r )rm r )rm r )rm r> )rn rX )rn r] )rn r )rn r )rn r� )rn rW )r^ )rE r_ rW ) )rm r )rm r )rm r )rm r )rm r> )rn rX )rn r] )rn r )rn r )rn r� ) r rD r# r( r) r ra r rp r. r/ r rF )r Zseriesr2 r1 rc r r r �"test_groupby_rolling_center_center s< " " z.TestRolling.test_groupby_rolling_center_centerc C s� t tdd�dgd dgd td�d�d �}|�d �jdd d d d�j�� }tddddddddddg dtj dt d�fdt d�fdt d�fdt d�fdt d�fdt d �fdt d �fdt d �fdt d �fdt d�ff d d gd �d!�}t � ||� d S )"Nz 2020-01-01z 2020-01-10�group_1r] �group_2r> rW )�Date�gbrY )�datar� r� Tr )�onr} r4 g �?g �?g @g @g @g @g @g @rY z 2020-01-02z 2020-01-03z 2020-01-04z 2020-01-05z 2020-01-06z 2020-01-07z 2020-01-08z 2020-01-09)r^ )r{ rE ) r r rD r# r( rY r) r r rp r r. r/ )r rc r2 r1 r r r � test_groupby_rolling_center_onM s, z*TestRolling.test_groupby_rolling_center_onr4 rX r> r c C s� t dgd dgd td�d��}d}|�d�j|d|d ��� }|�� dd g }d d d dddddddg }dd� |D �}td|d �}tjg| }|||d| � | } |||d| � | } t dgd dgd | | d��} t � || � d S )Nr rW r r )�groupr� rX r� T)r} r4 r� g �?g �?g @g @g @g @g @g @g @g @c S s g | ] }|d �qS )g $@r )�.0r* r r r � <listcomp>~ s zGTestRolling.test_groupby_rolling_center_min_periods.<locals>.<listcomp>r r ) r rD r# r( r) � reset_indexr; r ra r. rF ) r r4 rc � window_sizer2 Z grp_A_meanZ grp_B_meanZnum_nansZnansZgrp_A_expectedZgrp_B_expectedr1 r r r �'test_groupby_rolling_center_min_periodsp s " z3TestRolling.test_groupby_rolling_center_min_periodsc C s� t ddddgddddgdd d d gd ��}|�d �d g �d��� }t tjtjdtjgd gtjdd d gd�d�}t� ||� |�d �d �d��� }t tjtjdtjgtjdd d gd�d d�}t� ||� d S )Nr r r g @g @g @g �?rW r � )rm rn ro rm rn ))r r )r r )r r )r r )r^ )r_ rE )rE r{ ) r r# r( r; r ra r rp r. rF r r/ )r rc r2 r1 r r r � test_groupby_subselect_rolling� s & z*TestRolling.test_groupby_subselect_rollingc C s� G dd� dt �}tddddddgd id gd d gd d gd d�}|�|j�j|d d�d d��� }|�|j�jd d d��� }t�||� d S )Nc @ s e Zd Zddd�ZdS )zFTestRolling.test_groupby_rolling_custom_indexer.<locals>.SimpleIndexerr Nc S sH |d kr| j nd}tj|tjd�d }|�� | j }|||dk < ||fS )Nr )�dtyper )r� r r �int64�copy)r Z num_valuesr4 r} �closed�end�startr r r �get_window_bounds� s zXTestRolling.test_groupby_rolling_custom_indexer.<locals>.SimpleIndexer.get_window_bounds)r NNN)�__name__� __module__� __qualname__r� r r r r � SimpleIndexer� s r� rm g �?g @g @g @g @r r rX r r )rE )r� )r4 )r? r4 )r r r# rE r( r9 r. rF )r r� rc r2 r1 r r r �#test_groupby_rolling_custom_indexer� s 6 z/TestRolling.test_groupby_rolling_custom_indexerc C s� t td�td�dddg td�gd d��}|�d�jdd d d �d �� }ttjd dtjddgt j dtd�fgd dtd�fgd dd gd�d d�}t � ||� d S )Nr] r r r z 2019-01-01)�column1�column2r� �dater� �1Dr� �left)r� r� r� g g @g �?g @)r^ )rE r{ ) r rD r r# r( r9 r r ra r rp r. r/ )r rc r2 r1 r r r �'test_groupby_rolling_subset_with_closed� s z3TestRolling.test_groupby_rolling_subset_with_closedc C s� t td�td�dddg td�gd d��}|�d�dd g jd d d d �d �� }ttjd dtjddgt j dtd�fgd dtd�fgd dd gd�dd�}t � ||� d S )Nr] r r r z 2019-01-01)r� r� r� r� r� r� r� r� r� )r� r� g g @g �?g @)r^ )rE r{ ) r rD r r# r( r9 r r ra r rp r. r/ )r rc r2 r1 r r r �.test_groupby_subset_rolling_subset_with_closed� s z:TestRolling.test_groupby_subset_rolling_subset_with_closedrg c C s~ t dddgtjdddgddgd�d d �}t|�|��d�|�� }t tjtjd gtjd d dgd ddgd�d d �}t� ||� d S )Nr r )rm r* )rm rs )ro �z�1�2)r^ rm )rE r{ g @)r rm r* )r rm rs )r ro r� ) r r rp r@ r# r( r ra r. r/ )r rg Zdsr2 r1 r r r �"test_groupby_rolling_index_changed� s z.TestRolling.test_groupby_rolling_index_changedc C s� t dg i�}|�d�jdd��� }|jdd�}tjtg dd�tg dd�gdd gd�|_t � ||� t g g d ��}|�dd g�jdd��� }|jdd gd�}tjtg dd�tg dd�tg dd�gdd d gd�|_t � ||� d S ) N�s1r )r? )r_ Zfloat64)r� r� )r^ )r� �s2r� ) r r# r( r9 rB r Z from_productr rE r. rF )r r1 r2 r r r � test_groupby_rolling_empty_frame s $ z,TestRolling.test_groupby_rolling_empty_framec C s6 t ddtdddd�gddtdddd�gdd tddd d�gd dtddd d�gd d tdddd�ggdddgd��d�}|�d�}|�� |d< |jddd�}|�dd� �}t ddtdddd�dgddtdddd�dgd dtddd d�dgdd tddd d�dgd d tdddd�dggddddgd��ddg�}t�||� d S )Nr r� i� r r� r r �Zr� r �Hr] �Er rE r� Z eventTime)r_ Z count_to_dateZ10d)r� c S s | j d S )Nr )�shape)rc r r r r+ . r, z?TestRolling.test_groupby_rolling_string_index.<locals>.<lambda>g �?g @g @) r r r` r# Zcumcountr( r- r. rF )r rc �groupsZrolling_groupsr2 r1 r r r �!test_groupby_rolling_string_index s* z-TestRolling.test_groupby_rolling_string_indexc C s| t ddgddgd��jddd��d��� }t t�ddgddgg�dd gtjd d gdd gd �d �}|jdd�}t � ||� d S )Nr r )r ru r F)�sortg @g �?ru )r r )r r )r^ )r_ rE )r_ ) r r# r( r: r �arrayr rp rB r. rF )r r2 r1 r r r � test_groupby_rolling_no_sort; s z(TestRolling.test_groupby_rolling_no_sortc C s� t td�td�dddg tddd�d��}|�d�jd d d d �d �� }ttjddtjddgd t j dt d�fdt d�fdt d�fdt d�fdt d�fdt d�fgdd gd�d�}t � ||� d S )Nr] r r r Z20190101)r� Zperiods)r� r� r� r� r� Z3dr� r� )r� r� r� g �?z 2018-12-27z 2018-12-29z 2018-12-31z 2018-12-28z 2018-12-30z 2019-01-01)r^ )r{ rE )r rD r r# r( r5 r r ra r rp r r. r/ )r rc r2 r1 r r r �$test_groupby_rolling_count_closed_onL s& z0TestRolling.test_groupby_rolling_count_closed_on)rg �kwargsr( r )r? r4 � expandingc C s� t ddgddgddgddgddggddgd�}t|�d�|�f |��� }t tjgd tjdtjddgd �tjd d d d dgdd gd�d�}|jdd�}t � ||� d S )Nrm r r rn r )r_ rX g���&���?)rm rn )rm r )rm r )rn r )rn r )rn r> )r^ )rE ) r r@ r# Zsemr ra r rp rB r. rF )r rg r� rc r2 r1 r r r �test_groupby_rolling_seml s , z$TestRolling.test_groupby_rolling_sem)�rollings�keyr� rm NrE c C sp t tddtjg�dddgdddgd��}|dkr:|�d�}tjt|� d �d �� |�d �j d |� W d Q R X d S )Nz2020-06-01 12:00z2020-06-01 14:00r r r )rm rn ro rE rm z must be monotonic)r ro �60min)r� ) r r r ra r` r r! � ValueErrorr# r( )r r� r� rc r r r �"test_groupby_rolling_nans_in_index� s z.TestRolling.test_groupby_rolling_nans_in_index� group_keysc C s� dddgdddgg}t j|dd�}tdddg|d�}|jd d g|d ��d��� }td d dgt jdddgd d d d gd�d�}t�||� d S )N�val1�val2)rZ r[ )r^ r r r )rE rZ r[ )r� g �?g @g @)r� r� r� r� )r� r� r� r� ) r rC r r# r( r) rp r. r/ )r r� �arraysrE �sr2 r1 r r r � test_groupby_rolling_group_keys� s z+TestRolling.test_groupby_rolling_group_keysc C s� dddgdddgg}t j|dd�}tdddgtd�d�|d �}|�d d g��d��� }td d ddgit jdddgd d d dgd�d �}t� ||� d S )Nr� r� )rZ r[ )r^ r r r )r r )rE rZ r r g g �?g @)r� r r� r� )r� r r� r� r[ ) r rC r rD r# r( r) rp r. rF )r r� rE rc r2 r1 r r r �1test_groupby_rolling_index_level_and_column_label� s z=TestRolling.test_groupby_rolling_index_level_and_column_labelc C s^ t t�d�ddgd d��}|�d��d��� }tjdd d d d d ddgdd gd�}t� |j |� t t�d�ddgd ddddgd d��}|�ddg��d�� � }tjdddddddd d d d d g ddd gd�}t� |j |� t t�d�ddgd ddddgd d��}|j dd!d"�}|�d��d��� }tjd#d$d%d&d'd(d)d*gdd dgd�}t� |j |� d S )+Ng @r r r> )rm rn rn r )r r )r r )r r> )r r] )r r )r r )r rX )r r )r^ g (@r] )rm rn ro ro )r r r )r r r> )r r r )r r r )r r r] )r r rW )r r r )r r rX )r r r� )r r> r )r r> r )r r> r~ T)�append)r r r )r r r )r r> r )r r] r )r r r )r r r> )r rX r )r r r> ) r r r r# r( r) r rp r. �assert_index_equalrE r9 r` )r rc r2 rG r r r �)test_groupby_rolling_resulting_multiindex� sJ **z5TestRolling.test_groupby_rolling_resulting_multiindexc C s^ | j �d�}|�dd� �j}|jdd�}|�dd� �j}t�||� |jrNt�|j jrZt�d S )Nr c S s | � d��� S )Nr> )r( r9 )r* r r r r+ � r, zUTestRolling.test_groupby_rolling_object_doesnt_affect_groupby_apply.<locals>.<lambda>r> )r? c S s | � d��� S )Nr> )r( r9 )r* r r r r+ � r, ) r r# r- rE r( r. r� r r$ Zgrouper)r r& r1 �_r2 r r r �7test_groupby_rolling_object_doesnt_affect_groupby_apply� s zCTestRolling.test_groupby_rolling_object_doesnt_affect_groupby_applyr_ )r � )r �Cr r c C s| t ddggd ddggd ddggd |d�}|�d �}|jjd d �}|�d �}|�� }d |jksjt�t� |j|� d S ) Nr r r r r r r )r_ r T)�deepr> ) r r# �objr� r( r9 r_ r$ r. rF )r r_ rc r& Z original_objr6 r2 r r r � test_by_column_not_in_values� s 2 z(TestRolling.test_by_column_not_in_valuesc C s� ddddgddddgg}t j|dd�}tddd d d gi|d �}|jd d�d �d��� }ttjdtjdgt j ddddgdddgd�dd�}t � ||� d S )N�Falcon�Parrot�Captive�Wild)�Animal�Type)r^ z Max Speedg `x@g �u@g >@g 4@)rE r )r\ r g �@g I@)r� r� r� )r� r� r� )r� r� r� )r� r� r� r� r� )rE r{ ) r rC r r# r( r9 r r ra rp r. r/ )r r� rE rc r2 r1 r r r �test_groupby_level s z TestRolling.test_groupby_levelzby, expected_datarz �numg Y@g �b@g i@z 2018-01-01z 2018-01-02g @o@)r� r� c s� dddgdddgdddgdddgg}t |d d d gd �� t� d �� d < � �d g�� � fd d�|D �}� j|dd�jddd��� }d ddddgi}|�|� t |� jd�}t� ||� d S )Nr z 2018-01-01g Y@z 2018-01-02g i@r g �b@g @o@rz r� r� )r_ c s g | ]}t � |��qS r )r@ )r� �attr)rc r r r� ? s z3TestRolling.test_as_index_false.<locals>.<listcomp>F)Zas_indexr r )r? r4 )rE ) r r r` r# r( r) �updaterE r. rF )r r re r� Zgp_byr2 r1 r )rc r �test_as_index_false! s z TestRolling.test_as_index_false).r� r� r� r r' r3 r7 r �mark� parametrize�param�filterwarningsrH rL rT rf rh rj rq rt rw r| r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r rp r� r� r r� r r r r r sx " 2""A# $6 " r c @ s� e Zd Zdd� Zej�ddddddd d g�d d � �Zej�dd dg�dd� �Zej�ddddddg�dd� �Z ej�dddg�dd � �Z d d � Z d S ) � TestExpandingc C s4 t dgd dgd dgd t�d�d��| _d S ) Nr r r r r r r )r r )r r r r )r r r r r N s zTestExpanding.setup_methodr8 r9 r) r: r; r5 r< r= c sn | j �d�}|�� }t|� �� }|�� fdd��}|jddd�}t�| j d td�g�}||_ t � ||� d S )Nr c s t | �� � �� S )N)r@ r� )r* )r8 r r r+ Y r, z.TestExpanding.test_expanding.<locals>.<lambda>r )rA r ) r r# r� r@ r- rB r rC rD rE r. rF )r r8 r& r6 r2 r1 rG r )r8 r �test_expandingQ s z TestExpanding.test_expandingrI rJ c sr | j �d�}|�� }t|� �dd�}|�� fdd��}|jddd�}t�| j d td�g�}||_ t � ||� d S ) Nr r )rK c s t | �� � �dd�S )Nr )rK )r@ r� )r* )r8 r r r+ g r, z3TestExpanding.test_expanding_ddof.<locals>.<lambda>r )rA r ) r r# r� r@ r- rB r rC rD rE r. rF )r r8 r& r6 r2 r1 rG r )r8 r �test_expanding_ddofa s z!TestExpanding.test_expanding_ddofrM rN rO rP rQ rR c sp | j �d�}|�� }|jd� d�}|�� fdd��}|jddd�}t�| j d td�g�}||_ t � ||� d S ) Nr g�������?)rM c s | � � jd� d�S )Ng�������?)rM )r� rS )r* )rM r r r+ x r, z7TestExpanding.test_expanding_quantile.<locals>.<lambda>r )rA r ) r r# r� rS r- rB r rC rD rE r. rF )r rM r& r6 r2 r1 rG r )rM r �test_expanding_quantileo s z%TestExpanding.test_expanding_quantilerU rV c s� �j �d�}|�� }t|� ��j �}� �fdd�}|�|�}ttdd��ttdd�� }tj|j |df< tj|d< t � ||� t|j � �d d �}� fd d�}|�|�}t � ||� d S ) Nr c s t | �� � ��j�S )N)r@ r� r )r* )r8 r r r rg � s z3TestExpanding.test_expanding_corr_cov.<locals>.funcr �= �H �q r T)ri c s t | j�� � �dd�S )NT)ri )r@ r r� )r* )r8 r r rg � s )r r# r� r@ r- rb rD r ra �ilocr. rF r r/ )r r8 r& r6 r2 rg r1 Znull_idxr )r8 r r �test_expanding_corr_cov� s z%TestExpanding.test_expanding_corr_covc st | j �d�}|�� }|jdd� � d�}|�� fdd��}|jddd�}t�| j d td�g�}||_t � ||� d S ) Nr c S s | � � S )N)r9 )r* r r r r+ � r, z4TestExpanding.test_expanding_apply.<locals>.<lambda>)rr c s | � � jdd� � d�S )Nc S s | � � S )N)r9 )rs r r r r+ � r, zFTestExpanding.test_expanding_apply.<locals>.<lambda>.<locals>.<lambda>)rr )r� r- )r* )rr r r r+ � r, r )rA r ) r r# r� r- rB r rC rD rE r. rF )r rr r& r6 r2 r1 rG r )rr r �test_expanding_apply� s z"TestExpanding.test_expanding_applyN) r� r� r� r r r� r� r� r� r� r� r� r r r r r� M s r� c @ s� e Zd Zej�ddddddggdejdd d ggd ejd d dggg�dd� �Zej�ddejdddggdejd ddggg�dd� �Z dd� Z dd � Z d d � Z d S ) �TestEWMzmethod, expected_datar) g gUUUUUU�?g�m۶m�?g""""""@rI g���۞��?g������?g�t ���?rJ g �?gn۶m۶�?g��+��+�?c s� t dgd td�d��}t|�d�jdd�� �� }t d|itjdd d d gdd gd �d �}t�||� |�d�� � fdd��}t�|j dd�|� d S )Nrm r> )r r r g �?)�comr )rm r )rm r )rm r )rm r )r^ )rE c s t | jdd�� �� S )Ng �?)r� )r@ �ewm)r* )�methodr r r+ � r, z&TestEWM.test_methods.<locals>.<lambda>T)rB ) r rD r@ r# r� r rp r. rF r- r� )r r� re rc r2 r1 r )r� r � test_methods� s zTestEWM.test_methodsrU g �?r rV g�r��ڶ�?g߿yq�+�?c s� t dgd td�d��}t|�d�jdd�� �� }t d|itjdd d d gdd d gd �d �}t�||� |�d�� � fdd��}t�||� d S )Nrm r> )r r r g �?)r� r )rm r r )rm r r )rm r r )rm r r )r^ )rE c s t | jdd�� �� S )Ng �?)r� )r@ r� )r* )r� r r r+ � r, z/TestEWM.test_pairwise_methods.<locals>.<lambda>) r rD r@ r# r� r rp r. rF r- )r r� re rc r2 r1 r )r� r �test_pairwise_methods� s z TestEWM.test_pairwise_methodsc C sr d}|� d�j|dd��� }tddddd dd d dd d g itjddddddddddg dd gd�d�}t�||� d S )Nz23 daysr r� )�halflife�timesr g gCus�=�?g;�p�GR�?g�Z�kB��?g�> �(�?g8K�r��?gҧU���?gA�9w� �?)rm r )rm r )rm r] )rm r� )rn r )rn r> )rn r )ro r )ro rX )ro r )r^ )rE )r# r� r) r r rp r. rF )r � times_framer� r2 r1 r r r � test_times� s6 zTestEWM.test_timesc sn d� |� d�j� dd��� }|� d��� fdd��jddd d d d d dddg jdd�}t�|jdd�|� d S )Nz23 daysr r� )r� r� c s | j � dd��� S )Nr� )r� r� )r� r) )r* )r� r r r+ r, z-TestEWM.test_times_vs_apply.<locals>.<lambda>r r r] r� r r> r r rX r T)rB )r# r� r) r- r� r� r. rF )r r� r2 r1 r )r� r �test_times_vs_apply s zTestEWM.test_times_vs_applyc C sJ d}|� d�j|dd��� }|� d�j||d jd��� }t�||� d S )Nz23 daysr r� )r� r� )r# r� r) �valuesr. rF )r r� r� r2 r1 r r r �test_times_array s zTestEWM.test_times_arrayN) r� r� r� r r� r� r ra 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.api.indexersr Zpandas.core.groupby.groupbyr r r� r� r r r r �<module> s $ A_
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.06 |
proxy
|
phpinfo
|
Настройка