Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/__pycache__/test_shift.cpython-37.pyc
Назад
B �5�g0 � @ sh d dl Zd dlZd dlm mZ d dlZd dlm Z m Z m Z m Z m Z mZ d dlmZ G dd� d�ZdS )� N)�CategoricalIndex� DataFrame�Index�Series� date_range�offsetsc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z e j dd� �Z e j�d�dd� �Zdd� Zdd� Zdd� Zdd� Zd d � Ze j d d � �Zd d!� Zd"S )#�TestDataFrameShiftc C s� |� d�}t�|j|j� |d � d�}t�|d |� |� d�}t�|j|j� |d � d�}t�|d |� |� d�}t�||� |j dt�� d�}t|�t|�ks�t �|j ddd�}t�||� |jd }|t�d� }tj|� |�|� |�dd� |� d �} t� � } | � d �} | � d �}t�| j| j� t�|j| j� t� |j d d �df �� j| j d d �df j� | � d d�} | � d t�� �} t�| | � t�| | � d d�� d }tjt|d �� | j d d� W d Q R X ttj�dd��}tjttj|jdgd�|j d d �dd �f gdd d�}|j d d d�}t�||� ttj�dd��}tjttj|jdgd�|j d d �dd �f gdd d�}|j d dd�}t�||� d S )N� �A���r )�freq�BF)Z check_names� ���z does not match PeriodIndex freq)�match�D� )�index�columnsT)Z ignore_index�axis)r r )�shift�tmZassert_index_equalr �assert_series_equal�assert_frame_equalr �BDay�len�AssertionErrorZxs�makePeriodFrameZassert_numpy_array_equal�ilocZdropna�values�pytest�raises� ValueErrorr �np�randomZrand�pd�concat�nan)�self�datetime_frameZ int_frameZ shiftedFrameZ shiftedSeries� unshiftedZ shiftedFrame2�dZ shifted_dZ int_shifted�ps�shifted�shifted2�shifted3�msg�df�expected�result� r4 �[/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/test_shift.py� test_shift s^ , ( (z TestDataFrameShift.test_shiftc C sZ t ddgddgd��}|�d�}t tjtjtjgddggtd�ddgd�}t�||� d S ) NTF)�high�lowr )�dtyper7 r8 )r )r r r# �arrayr' �objectr r )r( r1 �rs�xpr4 r4 r5 �test_shift_bool] s z"TestDataFrameShift.test_shift_boolc C sf t dddgdd�}t dddgdd�}t||d ��}|�d �}t|�d �|�d �d ��}t�||� d S ) N�a�b�c�category)r9 r r �C)�one�twor )r r r r r )r( �s1�s2r1 r< r= r4 r4 r5 �test_shift_categoricalf s z)TestDataFrameShift.test_shift_categoricalc C s� t dddddgtdddd�d �}t d ddddgtdddd�d �}|jdd d �}t�||� t d d dddgtdddd�d �}|jdd d �}t�||� d S ) Nr � � � r z1/1/2000�H)Zperiodsr )r r )� fill_value)r r r r r )r( r1 �expr3 r4 r4 r5 �test_shift_fill_valueo s z(TestDataFrameShift.test_shift_fill_valuec C s& t dg i�}|�d�}t�||� d S )NZfoor )r r r r )r( r1 r< r4 r4 r5 �test_shift_empty� s z#TestDataFrameShift.test_shift_emptyc C s� t td��dgd dddddgg}tj�dd�}g }xl|D ]d}t|�� |d�}x:td�D ].}|jd d �|f �|d �|jd d �|f< qXW td�|_ |� |� q:W |d � � � � }t �|ttdd�dd �� t �|d |d � t �|d |d � d S ) Nr r rI � )r r � Zint64)r9 )�list�ranger# r$ Zrandnr �copyr r r �appendZisna�sumr r r r )r( Z column_lists�datar- r r1 �sZnullsr4 r4 r5 � test_shift_duplicate_columns� s " . z/TestDataFrameShift.test_shift_duplicate_columnsc C s$ t tjjddd��}t tjjddd��}tj||gdd�}|sRt|jj�dksRt �|j ddd�}|j ddd ddgdd�}tj |j d d �d d�f< |j|_t�||� tj||gdd�}|s�t|jj�dks�t �|j d dd�}|j dd d ddgdd�}tj |j d d �d d �f< |j|_t�||� d S ) Ni� )r rJ )�size)r rI r )r rI r r ���rJ rK )r r# r$ �randintr% r& r �_mgr�blocksr r �taker' r r r r )r( Zusing_array_manager�df1�df2�df3r3 r2 r4 r4 r5 � test_shift_axis1_multiple_blocks� s$ z3TestDataFrameShift.test_shift_axis1_multiple_blocksc C sz t tjjddd��}t tjjddd��}tj|jd d�dd�f |jd d�d d �f gdd�}|jd dt�d �d �}t |j j �d ks�t �|j d d d dgdd�}t�d �|jd d �d d �f< |j|_t�||� tj|jd d�dd�f |jd d�d d �f gdd�}|jd dt�d �d �}t |j j �d k�s0t �|j d dd d gdd�}t�d �|jd d �d d �f< |j|_t�||� d S )Ni� )r rJ )r[ )r rI rK r rJ )r rI r )r rM r r\ )r r# r$ r] r% r& r r �int_r r^ r_ r r` r r r )r( ra rb rc r3 r2 r4 r4 r5 �.test_shift_axis1_multiple_blocks_with_int_fill� s 6 6 zATestDataFrameShift.test_shift_axis1_multiple_blocks_with_int_fillz)ignore:tshift is deprecated:FutureWarningc C sf t �� }|�d�}|�d�}t �||� |jdd�}t �||� |jt�� d�}t �||� d}tjt|d�� |jdd� W d Q R X |�d�}|�d�}t �||� |j|j j d�}t �||� t |j t t�|j ��|jd�}|�d�}|�d�} | j �d �| _ t �|| � |�d�}t �||� |jd d d gd d �f } 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 �M)r r r � z6Freq was not set in the index hence cannot be inferred)r r �tshiftr r r r r! r" r r r r r r# �asarrayr � _with_freqr ) r( r) r, r- r* r. r/ r0 � inferred_tsr2 �no_freqr4 r4 r5 � test_tshift� s< z TestDataFrameShift.test_tshiftc C s" t �t�� |�� W d Q R X d S )N)r �assert_produces_warning� FutureWarningri )r( r) r4 r4 r5 �test_tshift_deprecated s z)TestDataFrameShift.test_tshift_deprecatedc C sh t �� }|jddd�}|jddd�}t �||� |jdd�}t �||� |jt�� d�}t �||� d S )Nr �infer)r r r )r r r r r r )r( r, r- r* r. r/ r4 r4 r5 �'test_period_index_frame_shift_with_freq s z:TestDataFrameShift.test_period_index_frame_shift_with_freqc C s� |j ddd�}|j ddd�}t�||� |j |jjd�}t�||� t|jtt� |j��|j d�}|j ddd�}|j ddd�}|j� d �|_t�||� |j ddd�}t�||� d S )Nr rr )r r )r ) r r r r r r r r r# rj r rk )r( r) r- r* r. rl r2 r4 r4 r5 �#test_datetime_frame_shift_with_freq s z6TestDataFrameShift.test_datetime_frame_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 rg )r )r r r r! r" r )r( r, r0 r4 r4 r5 �-test_period_index_frame_shift_with_freq_error, s z@TestDataFrameShift.test_period_index_frame_shift_with_freq_errorc C sF |j dddgd d �f }d}tjt|d�� |jdd� W d Q R X d S )Nr r rh z6Freq was not set in the index hence cannot be inferred)r rr )r )r r r! r" r )r( r) rm r0 r4 r4 r5 �)test_datetime_frame_shift_with_freq_error2 s z<TestDataFrameShift.test_datetime_frame_shift_with_freq_errorc C s� t t�d�t�d�g�}|�� }t�t�� |jddd�}W d Q R X t t�d�|d g��� }t�||� t ||d��}|� � t�t�� |jdddd�}W d Q R X t t�d�t�d�g|d d��}t�||� d S ) Nz 2020-01-01z 2020-01-02r r )rM )r r )r rM r ) r r% � TimestampZto_framer ro rp r r r Z_consolidate_inplace)r( Zserr1 r3 r2 rb r4 r4 r5 �)test_shift_dt64values_int_fill_deprecated8 s "z<TestDataFrameShift.test_shift_dt64values_int_fill_deprecatedc C s� t dddg�}tddgddgdd gd �|d d � |d �}|jdd �}ttjtjgddgddgd �|d d � |d �}t�||� |jddd �}ttjtjgtjtjgddgd �|d d � |d �}t�||� d S )Nr? r@ rA r rJ rI rK r rR )r? r@ rA r )r r )r )r r r r# r' r r )r( �cir1 r3 r2 r4 r4 r5 �$test_shift_axis1_categorical_columnsN s ( , z7TestDataFrameShift.test_shift_axis1_categorical_columnsN)�__name__� __module__� __qualname__r6 r> rH rO rP rZ rd �tdZ&skip_array_manager_not_yet_implementedrf r �mark�filterwarningsrn rq rs rt ru rv rx rz r4 r4 r4 r5 r s J 1 r )�numpyr# r Z pandas.util._test_decorators�utilZ_test_decoratorsr~ Zpandasr% r r r r r r Zpandas._testingZ_testingr r r4 r4 r4 r5 �<module> s
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.06 |
proxy
|
phpinfo
|
Настройка