Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/series/methods/__pycache__/test_shift.cpython-37.pyc
Назад
B �5�g�3 � @ sr d dl Zd dlZd dlmZ d dlZd dlmZmZm Z m Z m Z m Z m Z d dlmZ d dlmZ G dd� d�ZdS )� N)�NullFrequencyError)� DatetimeIndex�Index�NaT�Series�TimedeltaIndex� date_range�offsets)�BDayc @ sf e Zd Zej�dee�d�g�e dddd�ee d�e ddd�d �g�ej�d d d d g�dd� ��Z ej�de � d�e � d�g�dd� �Zdd� Zdd� Zdd� Zdd � Zd d � Zej�d �d d!� �Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Z d0d1� Z d2d3� Z ej�d4d d d5d6g�d7d8� �Z ej�d9ej d:d;�g fej!d<d;�ej"d=gfg�d>d?� �Z#d@S )A� TestShift�ser� z1/1/2011� �H)�periods�freq�2017)r )�index� shift_sizer � � c C s |� |�|k st�d S )N)�shift�AssertionError)�selfr r � r �\/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/series/methods/test_shift.py�test_shift_always_copy s z TestShift.test_shift_always_copy� move_by_freqZ1DZ1minc C s0 t td�tddd�d�}|j|d�|k s,t�d S )Nr r )r )r )r )r �ranger r r )r r r r r r � test_datetime_shift_always_copy# s z)TestShift.test_datetime_shift_always_copyc C s� |� d�}|� d�}t�|j|j� t�|j|j� t�|�� j|jd d� � t� }|j d|d�}|j d|d�}t�||� |j d|d�}t�||� |j ddd�}|j ddd�}t�||� |� d�}t�||� t� � }|� d�}|� d�}t�|j|j� t�|j|j� t�|�� j|jd d� � |� dd�}|� dt� �}t�||� t�||� dd�� d}t j t |d�� |j dd� W d Q R X |j ddd�} t�|| � |j dt� d�} t�| | � t d d d �} x^d D ]V} ttjd | d �| d�} | jd }| j |d �}ttjddddg| d�}t�||� �q�W tt dd dd�dd�}||� � }ttdgdgd �dd�}t�||� tt dd dd�dd�}d}t j t|d�� || W d Q R X d S )Nr ���)r r �Bz.Given freq D does not match PeriodIndex freq B)�match�Dz 2000-01-01r )r )Zint32Zint64)�dtype)r r � z2000-01-01 09:00:00z US/Eastern)r �tzZfoo)�namer z1 days� ZCETzFDatetimeArray subtraction must have the same timezones or no timezones)r �tm�assert_index_equalr Zassert_numpy_array_equal�dropna�valuesr �assert_series_equal�makePeriodSeries�pytest�raises� ValueErrorr r �np�arange�iloc�nanr � TypeError)r �datetime_series�shifted� unshifted�offset�ps�shifted2�shifted3�msgZshifted4Zshifted5r r$ �s1�p�result�expected�s�exp�s2r r r � test_shift) sd zTestShift.test_shiftc C s� t tj�d�tdddd�d�}|jddd�}|jjddd�}t�|j|� |jdd d�}|jt � d � }t�|j|� t d d d g�}d}t j t|d�� |�d� W d Q R X d S )Nr z1/1/2000r )r r )r r Z5T)r Z4Hr( z 2000-01-01z 2000-01-02z 2000-01-04zCannot shift with no freq)r" )r r2 �randomZrandnr r r r) r* r ZHourr r/ r0 r )r �tsrA Z exp_index�idxr> r r r � test_shift2w s zTestShift.test_shift2c C s� t dddddgtdddd �d �}t d ddddgtdddd �d �}|jd d d �}t�||� t d d dddgtdddd �d �}|jdd d �}t�||� t d ddg�}|jddd �}|j|jks�t�d S )Ng �?g @g @g @g @z1/1/2000r r )r r )r g r )� fill_valuer r% r )r r r r) r- r$ r )r rH rD rA �resr r r �test_shift_fill_value� s z TestShift.test_shift_fill_valuec C s� t ddddgdd�}|jddd�}t tjddddgddddgd d ��}t�||� d }tjt|d �� |jdd d� W d Q R X d S )N�a�b�c�d�category)r$ r )rK F)� categoriesZorderedz>'fill_value=f' is not present in this Categorical's categories)r" �f) r r �pdZ Categoricalr) Z assert_equalr/ r0 r6 )r rH rL rB r> r r r �!test_shift_categorical_fill_value� s z+TestShift.test_shift_categorical_fill_valuec C s t ddddd�}t|�}|�d�}t�||� |jdks<t�|�d�}tg|�t �j � � d d � }t|�}t�||� |jdks�t�|�d �}|�t �j � � d d � ttg }t|�}t�||� |jdks�t�xBd D ]:}|�|�}ttgd dd �}t�||� |jdks�t�q�W d S )Nz 2016-11-06r � z US/Eastern)r r r& r zdatetime64[ns, US/Eastern]r � ���r )rW i����� i��)r$ ) r r r r) r- r$ r r �astype�objectr, �tolist)r �datesrC rL Zexp_valsrD �exr r r �test_shift_dst� s( zTestShift.test_shift_dstz)ignore:tshift is deprecated:FutureWarningc C sX t �� }|�d�}|�d�}t �||� |jdd�}t �||� |jt� d�}t �||� d}tjt|d�� |jdd� W d Q R X |�d�}|�d�}t �||� |j|jj d�}t �||� t |j t t �|j��dd �}|�d�}|�d�} | j�d �| _t �|| � |�d�}t �||� |d d d g } d }tjt|d�� | �� W d Q R X d S )Nr r r! )r z.Given freq M does not match PeriodIndex freq B)r" �MrH )r' r r � z6Freq was not set in the index hence cannot be inferred)r) r. �tshiftr- r r/ r0 r1 r r r r, r r2 �asarray� _with_freq) r r7 r; r8 r9 r< r= r> � inferred_tsrB �no_freqr r r � test_tshift� s8 zTestShift.test_tshiftc C s" t �t�� |�� W d Q R X d S )N)r) �assert_produces_warning� FutureWarningrc )r r7 r r r �test_tshift_deprecated� s z TestShift.test_tshift_deprecatedc C sf t �� }|jddd�}|jddd�}t �||� |jdd�}t �||� |jt� d�}t �||� d S )Nr �infer)r r r! )r) r. r r- r )r r; r8 r9 r< r= r r r �(test_period_index_series_shift_with_freq� s z2TestShift.test_period_index_series_shift_with_freqc C s� |j ddd�}|j ddd�}t�||� |j |jjd�}t�||� t|jtt� |j��dd�}|j ddd�}|j ddd�}|j� d �|_t�||� |j ddd�}t�||� d S )Nr rl )r r rH )r' ) r r) r- r r r r, r r2 rd re )r r7 r8 r9 r< rf rB r r r �$test_datetime_series_shift_with_freq s z.TestShift.test_datetime_series_shift_with_freqc C s6 t �� }d}tjt|d�� |jdd� W d Q R X d S )Nz.Given freq M does not match PeriodIndex freq B)r" ra )r )r) r. r/ r0 r1 r )r r; r> r r r �.test_period_index_series_shift_with_freq_error s z8TestShift.test_period_index_series_shift_with_freq_errorc C s< |dddg }d}t jt|d�� |jdd� W d Q R X d S )Nr r rb z6Freq was not set in the index hence cannot be inferred)r" rl )r )r/ r0 r1 r )r r7 rg r> r r r �*test_datetime_series_shift_with_freq_error$ s z4TestShift.test_datetime_series_shift_with_freq_errorc C s4 |� t�}|�d�}|� t��d�}t�||� d S )Nr )r[ �intr �floatr) r- )r r7 rH r8 rB r r r �test_shift_int* s zTestShift.test_shift_intc C s� t td��}tjtdd�� |jdg d� W d Q R X |�� }tjtdd�� |jdt�d�d� W d Q R X |� t �}|jdi d�}|d i ks�t �|�� }|jdi d�}|j d i ks�t �d S )Nr% zfill_value must be a scalar)r" r )rK r )r r ) r r r/ r0 r1 r Zto_framer2 r3 r[ r\ r r4 )r r ZdfZobj_serrA Zobj_dfr r r �!test_shift_object_non_scalar_fill0 s z+TestShift.test_shift_object_non_scalar_fillc C s* t ddddgdd�}t�|jd d� |�d��d��� � |�d�}t�|j|j� t� |j j d d� dk�srt �t� |j j d d� |j j dd � k�s�t �|�d �}t�|j|j� t� |j j d d � dk�s�t �t� |j j d d � |j j d d � k�s�t �t�|j j |j j � t�|j j |j j � d S ) NrN rO rP rQ rR )r$ r r rY r )r r) r- r4 r r+ r* r r2 �allr, �codesr rS )r rC Zsp1Zsn2r r r �test_shift_categoricalB s & * *z TestShift.test_shift_categoricalc C sb t t�d�t�d�g�}t�t�� |jddd�}W d Q R X t t�d�|d g�}t�||� d S )Nz 2020-01-01z 2020-01-02r r )rK )r rU � Timestampr) ri rj r r- )r r rA rB r r r �)test_shift_dt64values_int_fill_deprecatedU s z3TestShift.test_shift_dt64values_int_fill_deprecatedr r% r( c C sP t t|�td|dd�d�}|�dd�}t t|�td|dd�d�}t�||� d S )Nz2016-1-1 00:00:00r )r r )r r Z2Hz2016-1-1 02:00:00)r r r r r) r- )r r r rA rB r r r �test_shift_preserve_freqstr_ s z%TestShift.test_shift_preserve_freqstrzinput_data, output_data)r )�shape)r g �?c C s6 |j dd� t|��d�}t|dd�}t�||� d S )NF)�writer Zfloat64)r$ )Zsetflagsr r r) r- )r Z input_dataZ output_datarA rB r r r � test_shift_non_writable_arrayo s z'TestShift.test_shift_non_writable_arrayN)$�__name__� __module__� __qualname__r/ �markZ parametrizer r2 r3 r r r rU Z Timedeltar rF rJ rM rV r` �filterwarningsrh rk rm rn ro rp rs rt rw ry rz �emptyZonesr5 r} r r r r r s4 &N. r )�numpyr2 r/ Z pandas.errorsr ZpandasrU r r r r r r r Zpandas._testingZ_testingr) Zpandas.tseries.offsetsr r r r r r �<module> s $
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка