Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/resample/__pycache__/test_resample_api.cpython-37.pyc
Назад
B �5�gR � @ sz d dl m Z d dlZd dlZd dlZd dlmZmZ d dlm Z d dl m Z e e ddd�e ddd�dd �Z eej�ee ��e �Zeeee�ee ��d ��Zejd d � �Zd d� Zdd� Zdd� Zdd� Zdd� Zdd� Zej � ddgddgg�d d � �Z d d � Z d d!� Z d"d#� Z!d$d%� Z"d&d'� Z#d(d)� Z$d*d+� Z%d,d-� Z&d.d/� Z'd0d1� Z(d2d3� Z)d4d5� Z*d6d7� Z+d8d9� Z,d:d;� Z-ej � d<d=d>d?d@dAdBdCdDg�dEdF� �Z.dGdH� Z/ej � dIdJdKdLd dMdNdOdPdQdRdSdTg dUdVdd dRdWdXgdYdZf d[d\d]dgd^ dLdVd_ddNd`dNd`dNd`dNd`dNg dadf dJdKdLd dMdNdOdPdQdRdSdTg dUdVdbd dRdWdcdTgddd`f dJdKdLd dMdNdOdPdQdRdSdTg dUdeddMdQdfdfgdgdZf g�dhdi� �Z0dS )j� )�datetimeN)� DataFrame�Series)� date_rangei� � � ZMin)�start�end�freq)�A�B�Cc C s t �� S )N)� _test_frame�copy� r r �]/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/resample/test_resample_api.py� test_frame s r c C s< t �d�} dt| �kst�t jddd�} dt| �ks8t�d S )N�HziDatetimeIndexResampler [freq=<Hour>, axis=0, closed=left, label=left, convention=start, origin=start_day]z 2000-01-01)�originzsDatetimeIndexResampler [freq=<Hour>, axis=0, closed=left, label=left, convention=start, origin=2000-01-01 00:00:00])� test_series�resample�str�AssertionError)�rr r r �test_str s r c C sh t �d�} | �� }t|t�s t�t|�dks0t�t �� �d�} | �� }t|t�sTt�t|�dksdt�d S )Nr �� ) r r �mean� isinstancer r �lenZto_framer )r �resultr r r �test_api( s r c C s� t tdddd�ddddgddd d gd ���d �} td d d��� tdd d��� }tjjdgd dgd |gdd gd�}t ddgd dg d gd d g i|d�}| �d��dd� �dg }t � ||� d S )Nz 2016-01-01� �W)r �periodsr r � � � � � )�date�group�valr) z 2016-01-03)r# z 2016-01-17r* )�namesr+ )�indexc S s | � d��� S )N�1D)r �ffill)�xr r r �<lambda>I � z+test_groupby_resample_api.<locals>.<lambda>) r r � set_index�tolist�pd� MultiIndex� from_arrays�groupby�apply�tm�assert_frame_equal)�df�ir- �expectedr r r r �test_groupby_resample_api5 s (,r? c C sj t ddgd tddd�tj�d�d��} | �d��d ��d ��� }| �d �jd dd ��� }t � ||� d S ) Nr r r% z 2016-01-01r )r# )�key�dates�valuesrA r@ �D)�on) r r �np�random�randnr3 r8 r r r: r; )r< r> r r r r � test_groupby_resample_on_apiM s rH c C sl t �d�}|�� |�� }|�dd� �}t�||� | �d�}|�� |�� }|�dd� �}t�||� d S )Nr c S s | � � | �� S )N)�maxr )r0 r r r r1 f r2 ztest_pipe.<locals>.<lambda>c S s | � � | �� S )N)rI r )r0 r r r r1 l r2 )r r rI r �piper: �assert_series_equalr; )r r r> r r r r � test_pipe` s rL c C s� | � d�}t�|jj| j� | � d�d }|jj| jd ks@t�| � d�d }t�|jj| jddg � | � d�d }t�|jj| jddg � d S )Nr r r )r r r )r r: Zassert_index_equalZ _selected_obj�columns�namer )r r r r r � test_getitemp s rO r@ rC r c C s4 |� d�}d}tjt|d�� || W d Q R X d S )Nr z ^\"Columns not found: 'D'\"$)�match)r �pytest�raises�KeyError)r@ r �g�msgr r r �test_select_bad_cols� s rV c C s( | � d�}t�|j�� |d �� � d S )Nr r )r r: rK r �sum)r r r r r �test_attribute_access� s rX c C s\ xVdD ]N} t dddd�}tt�t|��|d�}|�d�}t|| � |�� t|| � qW d S )N)�groupsZngroups�indicesz1/1/2012�d �S)r# r )r- Z30s)r r rE �aranger r �getattrr )�attr�rng�ts�rsr r r �test_api_compat_before_use� s rc c C sz | }d|d< |� d�}|ddg �� }tj|j�� |j�� gdd�}t�||� |dddg �� }|�� }t�||� d S ) NZfoorC r r r r )�axisr )r rW r5 �concatr r r: r; )r r< r r r> r r r �tests_skip_nuisance� s rf c C sf t dddd�} ttjt| �dd�| d�}|�d��� }td d d d d gt dddd�d�}t�||� d S )Nz1/1/2012r[ r\ )r# r �int64)�dtype)r- Z20sr � �( �< �P z2012-01-01 00:00:00r% )r r# ) r r rE r] r r �asfreqr: rK )r` ra r r> r r r �'test_downsample_but_actually_upsampling� s rn c C s� t dddd�} tt�t| ��| d�}|jdddd d d d d dddg }|�d��� �� }tddddddddddddddddgt j dd d d d d d!d"d#d$d%d&d'd(d)d*gd+d,d-�d�}t � ||� d S ).Nz1/1/2012r[ r\ )r# r )r- r r r$ � r% r' � � � � � Z2sg �?g @g @g @g &@g .@g 0@g 9@g >@z2012-01-01 00:00:00z2012-01-01 00:00:02z2012-01-01 00:00:04z2012-01-01 00:00:06z2012-01-01 00:00:08z2012-01-01 00:00:10z2012-01-01 00:00:12z2012-01-01 00:00:14z2012-01-01 00:00:16z2012-01-01 00:00:18z2012-01-01 00:00:20z2012-01-01 00:00:22z2012-01-01 00:00:24z2012-01-01 00:00:26z2012-01-01 00:00:28z2012-01-01 00:00:30zdatetime64[ns]�2S)rh r ) r r rE r] r Zilocr r r/ r5 Z DatetimeIndexr: rK )r` ra Zts2r r> r r r �*test_combined_up_downsampling_of_irregular� sR rv c C s< t �d�} t �tjdd���d�}| �d�}t�||� d S )N�20min)r r )r r r8 r5 �GrouperZ transformr: rK )r r> r r r r �test_transform� s ry c C s� t dddd�} ttjt| �dd�| d�}|�d�}|�� }|jd d �}t� ||� |� � }|jd d �}t� ||� d }t j t |d �� |�d� W d Q R X d S )Nz1/1/2012r ru )r# r rg )rh )r- �sr/ )�method�bfillzTInvalid fill method\. Expecting pad \(ffill\), backfill \(bfill\) or nearest\. Got 0)rP r )r r rE r] r r r/ Zfillnar: rK r| rQ rR � ValueError)r` ra r r> r rU r r r � test_fillna� s r~ c C sL t �d�} t �tjdd��}x*|| gD ] }|�dd� �}t�|t � q&W d S )Nrw )r c S s | S )Nr )r0 r r r r1 r2 z0test_apply_without_aggregation.<locals>.<lambda>)r r r8 r5 rx r9 r: rK )r rT �tr r r r � test_apply_without_aggregation s r� c C sd t tj�dd�tdddd�dddgd �} | �d �}d }tjt|d �� |� d dd�� W d Q R X d S )Ni� ro z1/1/2012r\ )r r# r r r )r- rM �3Tz'Column\(s\) \['r1', 'r2'\] do not exist)rP r rW )�r1�r2) r rE rF rG r r rQ rR rS �agg)r< r rU r r r �test_agg_consistency s r� c C sb t tj�dd�tdddd�ddgd�} | �d �}d }tjt|d �� |� d d d�� W d Q R X d S )Ni� r$ z1/1/2012r\ )r r# r �a)r- rM r� z#Column\(s\) \[2, 'b'\] do not exist)rP r rW )r$ �b) r rE rF rG r r rQ rR rS r� )r< r rU r r r �'test_agg_consistency_int_str_column_mix/ s r� c C s� t j�d� ttddd�tddd�dd�} d| _tt j�dd�td �| d �}|� � }|� � }t j j td�|jgd dgd �|_|�d �}||jd dd�|jd dd�|�t jd d��g}|d �� }|d �� }|d �� }|d �� } |d �� } |d �� } t j||| | gdd�} t j �ddgddgg�| _x,|D ]$} | �t jt jg�}t�|| � �q2W t j|| gdd�} x2|D ]*} | �t jt jd��}tj|| dd� �qrW t j||gdd�} t j � ddg�| _x,|D ]$} | �dddgi�}t�|| � �q�W t j||gdd�} ddg| _x |D ]} | d �ddg�}�qW t�|| � d}xB|D ]:} t j t j j j!|d �� | �dddd �i� W d Q R X �qFW t j||| | gdd�} t j � dd d d g�| _xJ|D ]B} t j t j j j!|d ��" | �ddd �ddd!�d�� W d Q R X �q�W t j||| | gdd�} t j � ddd"d#g�| _x6|D ].} | �ddgddgd��}tj|| dd� �q.W t j||| | gdd�} t j � d$d%d&d'g�| _d S )(Ni� i� r r rC )r r) r$ �AB)rM r- r- )r, �2D)rD )�levelr r )rd r �std)r r T)� check_like)r r )r r� rW z nested renamer is not supported)rP )r rW )r rW )r �mean2)r �sum2)r� r� )r r )r r� )r� r r )r� r rW )r� r r )r� r rW )"rE rF �seedr r rN r �rand�list� reset_indexr r5 r6 r7 �ranger- r r8 rx r r� rW re Z from_productrM � aggregater: r; � from_tuplesrQ rR �core�base�SpecificationError)r- r< �df_col�df_multr �casesZa_meanZa_stdZa_sumZb_meanZb_stdZb_sumr> r r rU r r r �test_aggB s| $ r� c C sD t j�d� ttddd�tddd�dd�} d| _tt j�dd�td �| d �}|� � }|� � }t j j td�|jgd dgd �|_|�d �}||jd dd�|jd dd�|�t jd d��g}x^|D ]V}|�t jdd� d��}|d �dd� �}t j|d �� |gdd�} tj|| dd� q�W t j|d �� |d �� |d �� |d �� gdd�} t j �dddd g�| _d } xD|D ]<}t j t | d ��" |ddg �t jt jd �� W d Q R X �qjW t j|d �� |d � � |d �� |d � � gdd�} t j �d d!d"d#g�| _x6|D ].}|�d$d%gd&d%gd��}tj|| dd� �q�W x>|D ]6}|ddg �d$d%gd&d%gd��}tj|| dd� �q4W d'} x�|D ]x}t j t j j!j"| d �� |d �dd$d%gi� W d Q R X t j t j j!j"| d ��" |d �d$d%gd&d%gd�� W d Q R X �qxW d(} xF|D ]>}t j t | d ��$ |dg �d$d%gd&d%gd�� W d Q R X �q�W d S ))Ni� i� r r rC )r r) r$ r� )rM r- r- )r, r� )rD )r� c S s t j| dd�S )Nr )�ddof)rE r� )r0 r r r r1 � r2 z test_agg_misc.<locals>.<lambda>)r r r c S s t j| dd�S )Nr )r� )rE r� )r0 r r r r1 � r2 r )rd T)r� )�result1r )r� r )�result2r )r� r z1Column\(s\) \['result1', 'result2'\] do not exist)rP )r� r� )r rW )r r� )r r )r r� rW r� r z nested renamer is not supportedz Column\(s\) \['B'\] do not exist)#rE rF r� r r rN r r� r� r� r r5 r6 r7 r� r- r r8 rx r� rW r9 re r: r; r r� rM rQ rR rS r� r� r� r� ) r- r< r� r� r r� r r Zrcustomr> rU r r r � test_agg_misc� s\ 2 ,2 , r� c C s� t j�d� ttddd�tddd�dd�} d| _tt j�dd�td �| d �}|� � }|� � }t j j td�|jgd dgd �|_|�d �}||jd dd�|jd dd�|�t jd d��g}d}xL|D ]D}tjt jjj|d��& |�dddgidddgid�� W d Q R X q�W x�|D ]�}tjt jjj|d��. |ddg �dddgidddgid�� W d Q R X tjt jjj|d��& |�dddgidddgid�� W d Q R X �q W d S )Ni� i� r r rC )r r) r$ r� )rM r- r- )r, r� )rD )r� z nested renamer is not supported)rP r r rW r )r� r� �rar� �rb)r r )rE rF r� r r rN r r� r� r� r r5 r6 r7 r� r- r r8 rx rQ rR r� r� r� r� r� )r- r< r� r� r r� rU r r r r �test_agg_nested_dicts� s. . &r� c C s� t dddd�ddd�t dddd�ddd�t dddd�d d d�g} t| ��d �}d }tjt|d ��$ |�d��dgdgdgd�� W d Q R X d S )Ni� r& r r g �?g @)�dtr0 �yr$ g @g �?r� z Column\(s\) \['z'\] do not exist)rP Z30Tr ZmedianrW )r0 r� �z)r r r3 rQ rR rS r r� )�datar< rU r r r �&test_try_aggregate_non_existing_column s r� c C s� t tddd�tddd�dd�} tjt| �tjd�}t| |d�tjj || gdd gd �d �}td |i| d �}d }t j t |d�� |j ddd� W d Q R X d}t j t|d�� |j ddd d� W d Q R X d}t j t |d�� |j dd dgd� W d Q R X d}t j t|d�� |j dd dgd� W d Q R X d}t j t|d�� |j dd d��� W d Q R X t j t|d�� |j ddd��� W d Q R X |� d��� }d|j_t�||j ddd��� � d |j_t�||j dd d��� � d S )Ni� r r rC )r )rh )r) r� �v�d)r, )r- r� zaOnly valid with DatetimeIndex, TimedeltaIndex or PeriodIndex, but got an instance of 'Int64Index')rP r� )r� z2The Grouper cannot specify both a key and a level!r) )rD r� zunhashable type: 'list')rD z#\"Level \['a', 'date'\] not found\"z|Upsampling from level= or on= selection is not supported, use \.set_index\(\.\.\.\) to explicitly set index to datetime-like)r r rE r] r rg r r5 r6 r7 rQ rR � TypeErrorr r} rS rm rW r- rN r: r; )r- r` r< Zdf_exprU �expr r r � test_selection_api_validation s8 r� �col_name�t2Zt2xZt2qZT_2MZt2pZt2mZt2m1ZT2Mc C s� t ttd��tddddd�| gd�}|�d��dg�}t d d d gtdd d dd�tj| gdggdgdggd�d�}t� ||� d S )N�� z 2017-01-01Z15minz Europe/Berlin)r r r# �tz)r- rM Z1dr g �G@g �a@g ph@rC ro r )Zlevels�codes) r r� r� r r r� r5 r6 r: r; )r� r< r r> r r r �*test_agg_with_datetime_index_list_agg_funcG s "r� c C s� t dddd�} t�| �}|jdd� t|| d�}|�d�}tt�d �t�d �g| d d d � d�}|�d �}t � ||� |�d �}t � ||� |�d �}t � ||� |�d�}t � ||� d S )Nz 2020-01-01z 2020-01-02Z1h)r F)�write)r- r. r � �last�firstrI �min) r rE Z zeros_likeZsetflagsr r r5 � Timestampr� r: rK )r- ZarrZserrb r> r r r r �test_resample_agg_readonly` s & r� zLstart,end,freq,data,resample_freq,origin,closed,exp_data,exp_end,exp_periodsz2000-10-01 23:30:00z2000-10-02 00:26:00Z7minro r& � � rq � � r� Z17minr � �? z20001002 00:26:00r! z20200101 8:26:35z20200101 9:31:58Z77s�3 �rightr% z2020-01-01 09:30:45�left�' z20001002 00:43:00Zend_day�- z2000-10-02 00:29:00c C sT t | ||d�} t|| d�} | j|||d��� } t|t ||| d�d�} t�| | � d S )N)r )r- )r �closed)r r r# )r r r rW r: rK )r r r r� Z resample_freqr r� Zexp_dataZexp_endZ exp_periodsr` ra �resr> r r r �test_end_and_end_day_originx s A r� )1r �numpyrE rQ Zpandasr5 r r Zpandas._testingZ_testingr: Z pandas.core.indexes.datetimesr ZdtirF r� r r r] r Zfixturer r r r? rH rL rO �markZ parametrizerV rX rc rf rn rv ry r~ r� r� r� r� r� r� r� r� r� r� r� r r r r �<module> s� 8 WK#1
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.06 |
proxy
|
phpinfo
|
Настройка