Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/resample/__pycache__/test_time_grouper.cpython-37.pyc
Назад
B �5�gj+ � @ s< d dl m Z d dlmZ d dlZd dlZd dlZd dlmZm Z m Z d dl m Z d dlmZ d dlmZ e ej�d�eddd �d �Zd d � Zd d� Zdd� Zdd� Zej�dde jfde j fde j fddd� fg�dd� �Z d d � Z d d � Z ej�d d!i d fd!d"d id fd!d"d#iej!fd$i d#fd$d"d id#fd$d"d#iej!fg�d%d&� �Z"ej�d'd(ej!fd)ej!fd*d+d,g�d-d.� �Z#d/d0� Z$d1d2� Z%ej�d3d!i d#d d#gfd!d"d id#d d#gfd!d"d#id#ej!d#gfd!d"d4iej!ej!ej!gfd$i d#d#d#gfd$d"d id#d#d#gfd$d"d#id#ej!d#gfd$d"d4iej!ej!ej!gfg�d5d6� �Z&d7d8� Z'dS )9� )�datetime)� methodcallerN)� DataFrame�Series� Timestamp)�Grouper)� date_rangei� z1/1/2000)�periods)�indexc C sj t dddd�} t�| �}dd� }|�|�}t�dd� ��|�}|j�d�|_|j�d�|_t�||� d S ) N�A�right)�freq�label�closedc S s | � � dd � S )N���)Z sort_values)�x� r �]/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/resample/test_time_grouper.py�f s ztest_apply.<locals>.fc S s | j S )N)�year)r r r r �<lambda> � z test_apply.<locals>.<lambda>r )r � test_series�groupby�applyr Z droplevel�tm�assert_series_equal)�grouper�groupedr Zapplied�expectedr r r � test_apply s r c C sx t jtd d d�< t�dd� ��� } tdddd�}t�|��� }|j| _t�|| � t� d��� }|j| _t�|| � d S )N� c S s | j S )N)r )r r r r r ' r z test_count.<locals>.<lambda>r r )r r r ) �np�nanr r �countr r r r �resample)r r �resultr r r � test_count$ s r' c C s@ t jddd��� } t �dd� ��tj�}| j|_t�| |� d S )Nr r )r c S s | j S )N)r )r r r r r 6 r z&test_numpy_reduction.<locals>.<lambda>) r r% �prodr Zaggr" r r r )r& r r r r �test_numpy_reduction3 s r) c C sr d} t dd| d�}tddd�|d�}td d �}|�|�\}}}|j|d d �}d d� }|�|�}t�|j|j� d S )Ni� z 2000-01-01�D)�startr r � � )�open�close)r �M)r F)Z group_keysc S s | d | d S )Nr/ r. r )�dfr r r r H s z test_apply_iteration.<locals>.f) r r r Z _get_grouperr r r Zassert_index_equalr ) �N�indr1 �tg�_r r r r& r r r �test_apply_iteration< s r6 z name, funcZ Int64Index�IndexZ Float64Index� MultiIndexc C s t �| d�S )Nr- )r ZmakeCustomIndex)�mr r r r V r r c C s^ d}||�}t dtj�|�i|d�}d| � d�}tjt|d�� |�tdd�� W d Q R X d S ) Nr- �a)r zVOnly valid with DatetimeIndex, TimedeltaIndex or PeriodIndex, but got an instance of '�')�matchr* )r ) r r" �random�randn�pytestZraises� TypeErrorr r )�name�func�nr r1 �msgr r r � test_fails_on_no_datetime_indexP s rE c C s. d} t j�| d�}t|ddddgd�}tdd d �tdd d �tdd d �tdd d�tdd d �gd |d < |�td dd��}t�|� tdd d ��|d d d � � t�|� tdd d ��|d d d � � t�|� tdd d ��|d d d � � t�|� tdd d��|d d d � � t�|� tdd d ��|dd d � � d S )N� � r �B�Cr* )�columnsi� r, r- r! � �key)rL r ) r" r= r> r r r r r �assert_frame_equalZ get_group)rC �datar1 r r r r �test_aaa_group_orderf s $$$$rO c C s� t j�dd�}t|ddddgd�}dd d dd gd |d < t|ddddgd�}td dd�td dd �td dd �td dd�td dd �gd |d < |�d �}|�td dd��}t|| �� }t|| �� }tddd d d�|_ t � ||� dS )z?Check TimeGrouper's aggregation is identical as normal groupby.rF rG r rH rI r* )rJ r, r- r! rK rL i� )rL r z 2013-01-01)r+ r r rA N) r" r= r> r r r r �getattrr r r Z assert_equal)Zresample_methodrN � normal_df�dt_df�normal_grouped� dt_groupedr � dt_resultr r r �test_aggregate_normal| s" rV zmethod, method_args, unit�sumZ min_countr, r( c C sj t dgd tjgd tddd�d�}t| f|�|�d��}t d|gtjd d gd d �d�}t� ||� d S ) Nr r- �2017rG )r )r Z2dg z 2017-01-01z 2017-01-03Z2D)r ) r r" r# r r r% �pd� DatetimeIndexr r )�method� method_args�unit�sr& r r r r �!test_resample_entirely_nat_window� s & r_ zfunc, fill_value�min�max)rW r )r( r, )r$ r c C s4 d}t j�|d��d�}t|ddddgd�}d d t jdd gd |d < t|ddddgd�}td d d �td d d �tjtd d d�td d d �gd |d < |� d �}|� t d dd��}t || �� }t || �� } t|gd gdgddddgd�} |� | �} | � � } tddd d d�} | �d �| _t�| | � | jjd k�s0t�d S )NrF rG �int64r rH rI r* )rJ r, r- rK rL i� )rL r r! )r rJ z 2013-01-01)r+ r r rA )r" r= r> �astyper r# r rY �NaTr r rP �append� sort_indexr � _with_freqr r rM rA �AssertionError) rB Z fill_valuerC rN rQ rR rS rT � normal_resultrU �padr Zdtir r r �test_aggregate_with_nat� s, rk c C s d} t j�| d��d�}t|ddddgd�}d d t jdd gd |d < t|ddddgd�}td d d �td d d �tjtd d d�td d d �gd |d < |� d �}|� t d dd��}|� � }|� � }t dgdgd�}|� |�} | �� } tddd d d��d �| _t�| |� |jjd k�st�d S )NrF rG rb r rH rI r* )rJ r, r- rK rL i� )rL r r r! )r z 2013-01-01)r+ r r rA )r" r= r> rc r r# r rY rd r r �sizer re rf r rg r r r rA rh ) rC rN rQ rR rS rT ri rU rj r r r r � test_aggregate_with_nat_size� s. rm c C sF t tddd��} d}| |ks t�t tdddd��} d}| |ksBt�d S )Nr �H)rL r z�TimeGrouper(key='A', freq=<Hour>, axis=0, sort=True, closed='left', label='left', how='mean', convention='e', origin='start_day')z 2000-01-01)rL r �originz�TimeGrouper(key='A', freq=<Hour>, axis=0, sort=True, closed='left', label='left', how='mean', convention='e', origin=Timestamp('2000-01-01 00:00:00')))�reprr rh )r& r r r r � test_repr� s rq z$method, method_args, expected_valuesr- c C s` t dtdddd�d�}|�d�}tjdd d gdd �}t| f|�|�}t ||d�}t�||� d S ) Nr, rX r- rn )r r )r Z30Tz2017-01-01T00:00:00z2017-01-01T00:30:00z2017-01-01T01:00:00)r )r r r% rY rZ r r r )r[ r\ Zexpected_valuesr^ Z resampledr r& r r r r �test_upsample_sum s rr c C sf dddgdddgd�} t | �}tddd d �|d < |�d ��d ��d �jdd�}d}tjt|d��� t j j ddt d�fdt d�fdt d�fdt d�fdt d�fdt d�fdt d�fdt d�fdt d�fdt d �fdt d �fdt d �fdt d �fdt d �fdt d�fgd d gd!�}W d Q R X t d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1gd2gd3 dg d�|d4�}t� ||� d S )5N� � � �2 �< )Zprice�volumez 01/01/2018r! �W)r r Z week_startingrx Z1DZlinear)r[ z containing strings is deprecated)r< )rv z 2018-01-07z 2018-01-08z 2018-01-09z 2018-01-10z 2018-01-11z 2018-01-12z 2018-01-13z 2018-01-14z 2018-01-15z 2018-01-16z 2018-01-17z 2018-01-18z 2018-01-19z 2018-01-20z 2018-01-21)�namesg $@g�m۶m�#@gn۶m۶#@g%I�$I�#@g۶m۶m#@g�$I�$I#@gI�$I�$#@g #@g�m۶m�"@gn۶m۶"@g%I�$I�"@g۶m۶m"@g�$I�$I"@gI�$I�$"@g "@g &@g I@� )rN r )r r Z set_indexr r% Z interpolater Zassert_produces_warning� FutureWarningrY r8 � from_tuplesr rM )�dr1 r& rD Z expected_indr r r r �!test_groupby_resample_interpolate& s^ r )(r �operatorr �numpyr" r? ZpandasrY r r r Zpandas._testingZ_testingr Zpandas.core.groupby.grouperr Z pandas.core.indexes.datetimesr r= r> r r r' r) r6 �markZ parametrizeZ makeIntIndexZmakeUnicodeIndexZmakeFloatIndexrE rO rV r# r_ rk rm rq rr r r r r r �<module> sZ $ $$ &
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка