Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/window/__pycache__/test_base_indexer.cpython-37.pyc
Назад
B �5�g�6 � @ s0 d dl Zd dlZd dlmZmZmZmZmZ d dl m Z d dl m Z mZ d dlmZmZmZ d dlmZ dd� Zdd � Zd d � Zd d � Zej�deeg�ej�ddedddddddddej g i fdej dddddddddej g i fdej dddddddd d ej g i fd ej dddd d d ddd!ej g d"d#ifd$ej dddd%d&d'ddd(ej g d"d#ifd)ej!ddddddddd*ej g i fg�ej�"d+�d,d-� ���Z#ej�deeg�d.d/� �Z$ej�d0d1dddd2dd3ddej ej g fd4dddd5d6d7ddej ej g fg�d8d9� �Z%ej�d:d;dddddd<dddd g gd=ddddd>d d>dddg gg�d?d@� �Z&dAdB� Z'ej�dCd#dd#d#dDdEgfdFdd#d dDd#gfg�ej�dGd)g fdHd(gfg�dIdJ� ��Z(ej�dKeeeg�ej�dLd#dEdMg�ej�dNd#d#gd d#gdO�d#dEgd d#gdO�d#gdP ej d#dEej ge)e*dQdP�� dO�g�dRdS� ���Z+ej�dTd#d#d gd#gfd#dEd d#gd#dEgfdEd#d gd#gfdEdEd d#gdEdEgfdUdMe*dM�e)e*dUdM��dMgdU fdMdUe*dU�dUgdU fd d e�,g �e�,g �fd#d e�,g �e�,g �fd d#d gd gfg �dVdW� �Z-ej�dXed d#dEgd#dEdEgdY��dEed dZdgej.d#dEdEge*dD�gd[dgd\�d]ej/d^�feej d#dEej ge)e*dQd_�� d#gd` dEgda e*d_�db��dMedcdcddd>d>dedd<dfdgdhdidjdkdldmdndogej.d#gd` dEgda e*d_�gd[dgd\�d]ej/d^�fg�dpdq� �Z0ej�drdsdtdudvdwdxdydzd{g �ej�dLd#dEdDdQdUd|d}g�d~d� ��Z1dS )�� N)� DataFrame� MultiIndex�Series�concat� date_range)� BaseIndexer�FixedForwardWindowIndexer)�ExpandingIndexer�FixedWindowIndexer�VariableOffsetWindowIndexer)� BusinessDayc C sF G dd� dt �} | � }tjtdd�� ttd���|� W d Q R X d S )Nc @ s e Zd Zdd� ZdS )z8test_bad_get_window_bounds_signature.<locals>.BadIndexerc S s d S )N� )�selfr r �[/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/window/test_base_indexer.py�get_window_bounds s zJtest_bad_get_window_bounds_signature.<locals>.BadIndexer.get_window_boundsN)�__name__� __module__� __qualname__r r r r r � BadIndexer s r z BadIndexer does not implement)�match� )r �pytest�raises� ValueErrorr �range�rolling)r �indexerr r r �$test_bad_get_window_bounds_signature s r c C s< t td��} t� }| �|��� }| �� �� }t�||� d S )N� )r r r r �meanZ expanding�tm�assert_series_equal)�sr �result�expectedr r r �test_expanding_indexer# s r% c C sn dddddg} t dtd�i�}G dd� dt�}|d| d�}|�|��� }t dd d d d d gi�}t�||� d S ) NTF�valuesr c @ s e Zd Zdd� ZdS )z3test_indexer_constructor_arg.<locals>.CustomIndexerc S sp t j|t jd�}t j|t jd�}xFt|�D ]:}| j| rNd||< |d ||< q*|||< || j ||< q*W ||fS )N)�dtyper � )�np�empty�int64r � use_expanding� window_size)r � num_values� min_periods�center�closed�start�end�ir r r r 1 s zEtest_indexer_constructor_arg.<locals>.CustomIndexer.get_window_boundsN)r r r r r r r r � CustomIndexer0 s r5 r( )r- r, g g �?g @g $@)r r r r �sumr �assert_frame_equal)r, �dfr5 r r# r$ r r r � test_indexer_constructor_arg+ s r9 c C sf t dtd�i�} G dd� dt�}|dd�}| j|dddd ��� }t dd d d d dgi�}t�||� d S )Nr& r c @ s e Zd Zdd� ZdS )z8test_indexer_accepts_rolling_args.<locals>.CustomIndexerc S s~ t j|t jd�}t j|t jd�}xTt|�D ]H}|r\|dkr\|dkr\|dkr\d||< |||< q*|||< || j ||< q*W ||fS )N)r' r( �both� r )r) r* r+ r r- )r r. r/ r0 r1 r2 r3 r4 r r r r G s zJtest_indexer_accepts_rolling_args.<locals>.CustomIndexer.get_window_boundsN)r r r r r r r r r5 F s r5 r( )r- Tr: )r0 r/ r1 g g �?g $@g @g @)r r r r r6 r r7 )r8 r5 r r# r$ r r r �!test_indexer_accepts_rolling_argsC s r< � constructorz func,np_func,expected,np_kwargs�countg @g @�ming g �?g @g @g @g @�maxg Y@g "@�stdg�����K@g�D�mK@g�!�?�J@g�'�e���?Zddofr( �varg[{���@�@g[{�����@g[{���Ħ@g �?�mediang !@z ignore:min_periods:FutureWarningc s@ t �d�}d|d< tdd�}d}tjt|d��$ | |�j|dd �}t||�� W d Q R X d }tjt|d��$ | |�j|d d �}t||�� W d Q R X | |�j|d d�}t||�� } | |�}t� | |� | |� � �fdd���} t� | | � |dkr�dnd } | |�j|| d�} t| |�� } | | � � �fdd���}t� | |� d S )Ng $@g Y@r � )r- z.Forward-looking windows can't have center=True)r T)�windowr0 zAForward-looking windows don't support setting the closed argument�right)rE r1 r; )rE r/ c s � | f��S )Nr )�x)�np_func� np_kwargsr r �<lambda>� � z-test_rolling_forward_window.<locals>.<lambda>r> r c s � | f��S )Nr )rG )rH rI r r rJ � rK ) r) �aranger r r r r �getattrr � assert_equal�apply)r= �funcrH r$ rI r& r r r r# Z expected2r/ Zrolling3Zresult3Z expected3r )rH rI r �test_rolling_forward_windowY s, 9 rQ c C sf t �d�}d|d< tdd�}| |�j|dd�}|�� }| ddd d d d ddt jt jg �}t�||� d S ) Ng $@g Y@r )r- rD )rE r/ g g � o��@g�\QJ�@g����@g�v��-�@gA� �@)r) rL r r Zskew�nanr rN )r= r& r r r# r$ r r r � test_rolling_forward_skewness� s" rS z func,expectedZcovg @X@g @W�Zcorrg�����?g�h�"���?g��1=��c C s� t �d��dd�}|d }d|d< t j||gdd�}tdd �}t|�j|dd �}t|| �� jt d �dfd f }|j d d �}t |�}|j |_ t �||� d S )Nr ���r( r; �d )r r )ZaxisrD )r- )rE r/ r T)Zdrop)r) rL ZreshapeZ concatenater r r rM �loc�sliceZ reset_indexr �namer rN )rP r$ Zvalues1Zvalues2r& r r r# r r r � test_rolling_forward_cov_corr� s rY zclosed,expected_datarF g (@�leftg @c C s^ t ddd�}ttd�|d�}td�}t||d�}|j|| d��� }t||d�}t�||� d S )NZ2020r )Zperiods)�indexr( )r[ �offset)r1 ) r r r r r r r6 r r7 )r1 Z expected_datar[ r8 r\ r r# r$ r r r �&test_non_fixed_variable_window_indexer� s r] c C sT t dd d d dgi�} tdd�}| j|dd��� }t dddddgi�}t�||� d S ) N�b� r; )r- r )rE r/ g g �?)r r r r> r r7 )r8 r r# r$ r r r � test_fixed_forward_indexer_count s r` )� end_valuer& rD r; rT )rP �argsZquantilec sn G � fdd�dt �}dddddg}tdtd�i�}|d|d�}t|�|�|�|� }td|i�} t�|| � d S ) Nc s e Zd Z� fdd�ZdS )z0test_indexer_quantile_sum.<locals>.CustomIndexerc sv t j|t jd�}t j|t jd�}xLt|�D ]@}| j| rTd||< t|� d�||< q*|||< || j ||< q*W ||fS )N)r' r r( )r) r* r+ r r, r@ r- )r r. r/ r0 r1 r2 r3 r4 )ra r r r s zBtest_indexer_quantile_sum.<locals>.CustomIndexer.get_window_boundsN)r r r r r )ra r r r5 s r5 TFr& r r( )r- r, )r r r rM r r r7 ) ra r& rP rb r5 r, r8 r r# r$ r )ra r �test_indexer_quantile_sum s rc � indexer_classr- � �df_data)�ar^ � � c C s\ t |�}d}| |d�}|j}x:t|�D ].}|�d�d j|dd��� |j|ks&t�q&W d S )NrD )r- rg r^ r( )rE r/ )r r- r �groupbyr r �AssertionError)rd r- rf r8 Z num_trialsr Zoriginal_window_sizer4 r r r �0test_indexers_are_reusable_after_groupby_rolling- s rl z5window_size, num_values, expected_start, expected_endr c C s^ t | d�}|j|d�\}}tj|t�|�dd� tj|t�|�dd� t|�t|�ksZt�d S )N)r- )r. F)Z check_dtype)r r r Zassert_numpy_array_equalr) �array�lenrk )r- r. Zexpected_startZ expected_endr r2 r3 r r r �!test_fixed_forward_indexer_boundsF s ro zdf, window_size, expected)r^ rg g �?rg )�namesr^ )r[ rX r' � r_ � )r^ rg �cg������ @g @g @g )@g *@g +@g ,@g -@g .@g /@g 0@g �0@g 1@c C s6 t |d�}| �d�d j|dd��� }t�||� d S )N)r- rg r^ r( )rE r/ )r rj r r r r! )r8 r- r$ r r# r r r �0test_rolling_groupby_with_fixed_forward_specific` s 6 rt � group_keys)r( )r( r; )r; r( )r( r( r; )r( r; r( )r( r( r; r; )r( r; rD r; rD ) r( r( r; r( r( r; r( r( r; r( r( r; )r( r; rD r( r; rD r( r; rD r( r; rD r( r; rD � � c s� t t�t| ��tjt| �tjd�d tjt| �tjd�d��}t� d�}|� d�d j |dd�� � }dd g|j _ |� d�ddg }t� fd d �|D ��}|�dd g�d }t�||� d S ) N)r' � )rg r^ rs )r- rg r^ r( )rE r/ rs c s4 g | ],\}� � j � �fd d�tt� ��D �d��qS )c s* g | ]"}� d j ||� � jdd��qS )r^ r( )Z min_count)Zilocr6 )�.0r4 )�gr- r r � <listcomp>� s zKtest_rolling_groupby_with_fixed_forward_many.<locals>.<listcomp>.<listcomp>)r^ )Zassignr rn )ry �_)r- )rz r r{ � s z@test_rolling_groupby_with_fixed_forward_many.<locals>.<listcomp>)r r) rm �listrL rn �float64r+ r rj r r6 r[ rp r Z set_indexr r! )ru r- r8 r r# �groupsZmanualr )r- r �,test_rolling_groupby_with_fixed_forward_many� s r� )2�numpyr) r Zpandasr r r r r Zpandas._testingZ_testingr Zpandas.api.indexersr r Zpandas.core.window.indexersr r r Zpandas.tseries.offsetsr r r% r9 r< �markZ parametrizern rR r? r@ rA rB rC �filterwarningsrQ rS rY r] r` rc r} r rl rm ro Z from_arraysr~ rt r� r r r r �<module> s ") ( && :" $
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка