Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/arrays/timedeltas/__pycache__/test_reductions.cpython-37.pyc
Назад
B �5�g � @ sZ d dl Zd dlZd dlZd dlmZ d dlmZ d dl m Z d dlmZ G dd� d�Z dS )� N)� Timedelta)�nanops)�TimedeltaArrayc @ s� e Zd Zej�ddddddg�ej�ddd g�d d� ��Zej�ddd g�dd � �Zdd� Zdd� Z dd� Z dd� Zdd� Zej�de d�ej�� ej�d�ej�d�g�dd� �Zdd� Zd d!� Zd"d#� Zd$S )%�TestReductions�name�std�min�max�median�mean�skipnaTFc C sP t �g �}|j}t||�|d�}|t jks.t�t||�|d�}|t jksLt�d S )N)r )�pd�TimedeltaIndex�array�getattr�NaT�AssertionError)�selfr r �tdi�arr�result� r �d/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/arrays/timedeltas/test_reductions.py�test_reductions_empty s z$TestReductions.test_reductions_emptyc C sh t �g �}|j}|j|d�}t|t�s*t�|td�ks:t�|j|d�}t|t�sTt�|td�ksdt�d S )N)r r )r r r �sum� isinstancer r )r r r r r r r r �test_sum_empty s zTestReductions.test_sum_emptyc C s� t �ddddddg�}|�� }td�}||ks2t�|�� }td�}||ksNt�|jdd�}|tjksht�|jdd�}|tjks�t�d S )N�3Hr �2H�5H�4HF)r )r Z_from_sequencer r r r r r )r r r �expectedr r r �test_min_max% s zTestReductions.test_min_maxc C s0 t �ddddddg�}|j}|jdd�}tdd �}t|t�s@t�||ksLt�|jdd�}t|t�sft�||ksrt�|jd d�}|t jks�t�|jd d�}|t jks�t�|jdd�}|t jks�t�|jdd�}|t jks�t�|jd d�}t|t�s�t�||k�st�|jd d�}t|t��st�||k�s,t�d S )Nr r r r r T)r � )�hoursF� )Z min_count� )r r r r r r r r )r r r r r! r r r �test_sum6 s. zTestReductions.test_sumc C sb t jdt jd��d��d�}t|�}t �|�}||�� ks>t�t �t� |��}||�� ks^t�d S )N� )Zdtypezm8[h]zm8[ns]) �np�arange�int64�view�astyper r r r r )r �valsr r r r r �test_np_sumX s zTestReductions.test_np_sumc C sp t �dddddg�}|j}t�|�}tdd�}t|t�s<t�||ksHt�t�|�}t|t�s`t�||kslt�d S )Nr r r r r# )r$ )r r r r) r r r r )r r r r r! r r r � test_npsumb s zTestReductions.test_npsumc C s� t �d��t j��d��d��dd�}d|d<