Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/__pycache__/test_diff.cpython-37.pyc
Назад
B �5�g`% � @ sN d dl Zd dlZd dlZd dlmZmZmZmZ d dl m Z G dd� d�Z dS )� N)� DataFrame�Series� Timestamp� date_rangec @ s� e Zd Zdd� Zdd� Zdd� Zej�ddd g�d d � �Z ej�ddd g�d d � �Z ej�ddd g�dd� �Z ej�ddd g�dd� �Z dd� Z dd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zej�d&d'eejd'd(d'ejejejd'gejd(ejejd)d(ejejge�ejd*�ejd+d,d-d.d/d0d1gd2�d3d4�fd(ee�ejd*�d'd(ejd(ejejejd'ge�ejd*�e�ejd*�d2�d3d4�fg�d5d6� �Zd7d8� Z d9d:� Z dS );�TestDataFrameDiffc C s: t tj�dd��}tjtdd�� |�d� W d Q R X d S )N� zperiods must be an integer)�matchg �?)r �np�random�randn�pytestZraises� ValueError�diff)�self�df� r �Z/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/test_diff.py�test_diff_requires_integer s z,TestDataFrameDiff.test_diff_requires_integerc C s� |� d�}t�|d |d |d �d� � d}|d }t||g�}td|i�� � }|jd dksft�|�d�}|� d�}t�|d |d |d �d� � ttdg�tdg�d��}|� d d d� |j dd �} tt j td�td�d ��} t� | | � d S ) N� �Al �_� �s�float32r � )�y�zr �x)�axis)r r r ) r �tmZassert_series_equal�shiftr r r �AssertionError�astype�insertr �nan�assert_frame_equal) r �datetime_frameZthe_diff�a�br �rs�tfr �result�expectedr r r � test_diff s " zTestDataFrameDiff.test_diffc C s� t �d��dd��d�}t �dd�|d d �df< t|�}|jddd �}t|d tjt� d�t� d�gd ��}t � ||� |�d�}|| }|d � � � � s�t�t � ||� |jd dd �}|t j }t � ||� d S ) N� r r ztimedelta64[ns]�NaT�nsr r )r )r r ���)r �arangeZreshaper Z timedelta64r r �pdr- � Timedeltar Z assert_equal�isna�allr r" )r �arrr r) r* r r r � test_diff_timedelta64_with_nat/ s & z0TestDataFrameDiff.test_diff_timedelta64_with_nat�tzN�UTCc C sd t jdddg|d�}t|�}|�� }|�� }t �t jt jt jdd�g�}t|��� }t� ||� d S )Nr- z 2019-01-01z 2019-01-02)r7 r )�days) r1 Z DatetimeIndexr �to_framer �TimedeltaIndexr- r2 r r# )r r7 �dti�serr r) Zex_indexr* r r r �!test_diff_datetime_axis0_with_natC s z3TestDataFrameDiff.test_diff_datetime_axis0_with_natc C s� t dd|d�}t|�}|�� }|�� |d< tj|jd d �df< || }|d �� �� sZt �|j ddd�}t � ||� |j ddd�}t � ||� d S )Nz 2016-01-01� )�periodsr7 r r )r ) r r r: �copyr1 r- Zilocr3 r4 r r r r# )r r7 r< r= r r* r) r r r �(test_diff_datetime_with_nat_zero_periodsP s z:TestDataFrameDiff.test_diff_datetime_with_nat_zero_periodsc C sd t tddd|d�tddd|d�d��}|jdd�}t t�dd g�t�dd g�d��}t�||� d S ) N�2010�Dr )�freqr@ r7 )r r r )r r- z1 days)r r r r1 r; r r# )r r7 r r) r* r r r �test_diff_datetime_axis0c s z*TestDataFrameDiff.test_diff_datetime_axis0c C sd t tddd|d�tddd|d�d��}|jdd�}t t�ddg�t�d d g�d��}t�||� d S ) NrC rD r )rE r@ r7 )r r r )r r- z0 days)r r r r1 r; r r# )r r7 r r) r* r r r �test_diff_datetime_axis1v s z*TestDataFrameDiff.test_diff_datetime_axis1c C s\ t td�td�gddgd��}|�� }t tjtjgt�d�dggdd gd �}t� ||� d S ) Nz 20130101 9:01z 20130101 9:02g �?g @)�time�valuez00:01:00r rH rI )�columns) r r r r1 r- r r" r2 r r# )r r �res�expr r r �test_diff_timedelta� s $z%TestDataFrameDiff.test_diff_timedeltac C sN t tj�dd��}tjdddddgtd�|d< |�� }|d jtjksJt �d S ) N� r r r r? )�dtyper r ) r r r r �array�objectr rO �float64r )r r r) r r r �test_diff_mixed_dtype� s z'TestDataFrameDiff.test_diff_mixed_dtypec C s( |� d�}||�d� }t�||� d S )Nr/ )r r r r# )r r$ r' �xpr r r �test_diff_neg_n� s z!TestDataFrameDiff.test_diff_neg_nc C s$ |� d�}|� d�}t�||� d S )Ng �?r )r r r# )r r$ r' rT r r r �test_diff_float_n� s z#TestDataFrameDiff.test_diff_float_nc C sh t ddgddgg�}t�|jdd�t tjdgtjdgg�� t�|jdd�t tjtjgddgg�� d S )Ng �?g @g @g @r )r r )r r r# r r r" )r r r r r �test_diff_axis� s $z TestDataFrameDiff.test_diff_axisc C sL t ddd��d�}td|i�}|jddd�}|tj �t�}t� ||� d S )Nz 2016-01-01r )r@ rD r r )r ) r Z to_periodr r r1 r- r rQ r r# )r �pir r) r* r r r �test_diff_period� s z"TestDataFrameDiff.test_diff_periodc C s� t td�dtjdtjd� d��}t tjtjtjg|d d d��}|jdd�}t�||� t tjddd�tjdd d�d ��}|jdd�}t |d tj |d d d ��}t�||� d S )Nr r )rO )r �BrZ r )r r rR )r% r& r% r& r ) r �ranger r0 rR r" r r r# )r r r* r) r r r � test_diff_axis1_mixed_dtypes� s "" z.TestDataFrameDiff.test_diff_axis1_mixed_dtypesc C sJ t td�dtjdtjd� d��}|tj }|jddd�}t�||� d S )Nr r )rO )r rZ r )r r@ ) r r[ r r0 rR r" r r r# )r r r* r) r r r �*test_diff_axis1_mixed_dtypes_large_periods� s " z<TestDataFrameDiff.test_diff_axis1_mixed_dtypes_large_periodsc C s` t td�dtjdtjd� d��}t d|d |d tj d��}|jdd d �}t�||� d S ) Nr r )rO )r rZ g �r rZ r r/ )r r@ ) r r[ r r0 rR r" r r r# )r r r* r) r r r �-test_diff_axis1_mixed_dtypes_negative_periods� s " z?TestDataFrameDiff.test_diff_axis1_mixed_dtypes_negative_periodsc C sT t ddgddggdd�}|�� }t tjtjgddggt�dd�d�}t�||� d S ) Nr r z Sparse[int])rO g �?g ��floatg )r r r r" r1 Z SparseDtyper r# )r Z sparse_dfr) r* r r r �test_diff_sparse� s "z"TestDataFrameDiff.test_diff_sparsez axis,expectedr r ���� r rN � � � � � )r% r& �c�d�Int64)rO c C sl t t�ddtjdgd�t�ddtjdgd�t�tjd�t�dd�d d�dd�}|j|d �}t�||� d S ) Nr r r rb rd )r% r& rh ri rj )rO )r ) r r �repeatr" Ztiler0 r r r# )r r r* r r) r r r �test_diff_integer_na� s z&TestDataFrameDiff.test_diff_integer_nac C sH t j�dd�}d|j_t|�}|�� }tt �|���� }t� ||� d S )NrN r F) r r r �flagsZ writeabler r rP r r# )r r5 r r) r* r r r �test_diff_readonly s z$TestDataFrameDiff.test_diff_readonlyc C sV t td��}|�|�}|�� }|dkr*dnd}t tjddddg|d�}t�||� d S )NrN )Zint8Zint16r rR g �?)rO )r r[ r r r r" r r# )r Z any_int_dtyper r) Zexpected_dtyper* r r r �test_diff_all_int_dtype! s z)TestDataFrameDiff.test_diff_all_int_dtype) �__name__� __module__� __qualname__r r+ r6 r �markZ parametrizer> rB rF rG rM rS rU rV rW rY r\ r] r^ r` r r r" rk rl rn ro r r r r r sF r ) �numpyr r Zpandasr1 r r r r Zpandas._testingZ_testingr r r r r r �<module> s
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка