Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/reductions/__pycache__/test_stat_reductions.cpython-37.pyc
Назад
B �5�g�% � @ s� d Z ddlZddlZddlZddlm mZ ddl Z ddl m Z m Z ddl mZ ddlmZmZmZ G dd� d�ZG dd� d�ZdS ) zP Tests for statistical reductions of 2nd moment or higher: var, skew, kurt, ... � N)� DataFrame�Series)� DatetimeArray� PeriodArray�TimedeltaArrayc @ sf e Zd Zej�deeje g�dd� �Z ej�deeje g�dd� �Z ej�deeje g�dd� �ZdS ) � TestDatetimeLikeStatReductions�boxc C s� |}t jdd|d�}|�dddddd d d d d dg �}|j}||�}|�� t jd|d�ks\t�|jdd�t jd|d�kszt�t j|d< ||�}|�� t jd|d�ks�t�|jdd�t jks�t�d S )Nz 2001-01-01� )�periods�tz� � � � � � � � r � � z 2001-01-06)r F)�skipna���z2001-01-06 07:12:00)�pd� date_range�take�_data�mean� Timestamp�AssertionError�NaT)�selfZtz_naive_fixturer r �dtiZdtarr�obj� r# �b/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/reductions/test_stat_reductions.py�test_dt64_mean s z-TestDatetimeLikeStatReductions.test_dt64_meanc C s� t jddd�}|�dddddd d d d d dg �}|j�d�}||�}tjtdd�� |�� W d Q R X tjtdd�� |jdd� W d Q R X t j |d< tjtdd�� |�� W d Q R X tjtdd�� |jdd� W d Q R X d S )Nz 2001-01-01r )r r r r r r r r r r r r �HZ ambiguous)�matchT)r r ) r r r r Z to_period�pytest�raises� TypeErrorr r )r r r! Zparrr" r# r# r$ �test_period_mean- s z/TestDatetimeLikeStatReductions.test_period_meanc C s� t jddddddddddd g d d �}|j}||�}|�� }t�|��� }||ksTt�t j|d< |jd d �t jkstt�|jdd �}||dd � �� ks�t�|�d�|d d �d�ks�t�d S )Nr r r i����r r ���r r �D)�unitF)r T�usg &@r ) r ZTimedeltaIndexr r �np�arrayr r �round)r r ZtdiZtdarrr" �result�expectedZresult2r# r# r$ �test_td64_meanE s $ z-TestDatetimeLikeStatReductions.test_td64_meanN)�__name__� __module__� __qualname__r( �markZ parametrizer r �Indexr r% r r+ r r5 r# r# r# r$ r s r c @ sr e Zd Zddd�Zdd� Zdd� Zdd � Zd d � Zd d � Zdd� Z dd� Z dd� Z e j dd� �Ze j dd� �ZdS )�TestSeriesStatReductionsFc C s t �dd��� tt|�}tj|dd�< |dkrptt jddd��}d |� d �}tjt |d �� ||� W d Q R X t � ||��s�t �t � ||dd ��s�t �|� � } t�|| �|| j�� t�||�|| j�� |tj } |r�t�|| ��s�t �td ddd dg�} || � dg} | �tdd�� t| dd�} t�t|| ��t|| j��� |�r|tt jddd��} || �} || �}| |k�s|t �|dk�r�tjt d d �� |ttd��� W d Q R X d}tjt|d �� ||d d� W d Q R X dt�|�jk�rtjt |d �� ||dd � W d Q R X W d Q R X d S ) NZuse_bottleneckFr � )�max�minr �median�stdz1/1/2001r )r z.'DatetimeArray' does not implement reduction '�')r' )r r r r r l l � Zint64)�dtypez1/1/2000)�sumr> r= �abcz&No axis named 1 for object type Series)Zaxis� numeric_onlyT)rE ) r Zoption_context�getattrr r0 �NaNr r( r) r* Znotnar �isnaZdropna�tm�assert_almost_equal�values�nan�isnan�extend�range�floatZ bdate_range�list� ValueError�inspect�getfullargspec�args�NotImplementedError)r �nameZ alternateZstring_series_� check_objects� check_allna�fZds�msgZnonaZallna�s�items�res�expr# r# r$ �_check_stat_op_ sH z'TestSeriesStatReductions._check_stat_opc C s& t �� �d�}| jdtj|dd� d S )N�seriesrC F)rY )rI �makeStringSeries�renamer` r0 rC )r � string_seriesr# r# r$ �test_sum� s z!TestSeriesStatReductions.test_sumc C s" t �� �d�}| �dtj|� d S )Nra r )rI rb rc r` r0 r )r rd r# r# r$ � test_mean� s z"TestSeriesStatReductions.test_meanc C sR t �� �d�}| �dtj|� ttjdtd�t d�d�}t � t�|�|�� � d S )Nra r? r )rB )�index) rI rb rc r` r0 r? r �ones�intrO rJ )r rd Zint_tsr# r# r$ � test_median� s z$TestSeriesStatReductions.test_medianc C s" t �� �d�}| �dtj|� d S )Nra �prod)rI rb rc r` r0 rk )r rd r# r# r$ � test_prod� s z"TestSeriesStatReductions.test_prodc C s& t �� �d�}| jdtj|dd� d S )Nra r> T)rX )rI rb rc r` r0 r> )r rd r# r# r$ �test_min� s z!TestSeriesStatReductions.test_minc C s& t �� �d�}| jdtj|dd� d S )Nra r= T)rX )rI rb rc r` r0 r= )r rd r# r# r$ �test_max� s z!TestSeriesStatReductions.test_maxc C s� t �� �d�}t �� �d�}dd� }| �d||� dd� }| �d||� |jdd �}tj|jdd �}t �||� |j dd �}tj |jdd �}t �||� |j d g }|j d d �}t � |�s�t �|jd d �}t � |�s�t �d S ) Nra �tsc S s t j| dd�S )Nr )�ddof)r0 r@ )�xr# r# r$ �<lambda>� � z7TestSeriesStatReductions.test_var_std.<locals>.<lambda>r@ c S s t j| dd�S )Nr )rp )r0 �var)rq r# r# r$ rr � rs rt r )rp r r )rI rb rc �makeTimeSeriesr` r@ r0 rK rJ rt �ilocr rH r )r rd �datetime_series�altr3 r4 r\ r# r# r$ � test_var_std� s" z%TestSeriesStatReductions.test_var_stdc C s� t �� �d�}t �� �d�}dd� }| �d||� |jdd�}tj|jdd�t� t |j�� }t � ||� |j dg }|jd d�}t �|�s�t�d S ) Nra ro c S s t j| dd�t �t| �� S )Nr )rp )r0 r@ �sqrt�len)rq r# r# r$ rr � rs z3TestSeriesStatReductions.test_sem.<locals>.<lambda>�semr )rp r r )rI rb rc ru r` r| r0 r@ rK rz r{ rJ rv r rH r )r rd rw rx r3 r4 r\ r# r# r$ �test_sem� s z!TestSeriesStatReductions.test_semc s� ddl m� t�� �d�}� fdd�}| �d||� d}x�td|d �D ]z}tt� |��}t t� ||f��}||k r�t� |�� �s�t �t� |�� �� � s�t �qHd|�� ks�t �|�� dk� � sHt �qHW d S ) Nr )�skewra c s � | dd�S )NF)�biasr# )rq )r~ r# r$ rr � rs z4TestSeriesStatReductions.test_skew.<locals>.<lambda>r~ r r )� scipy.statsr~ rI rb rc r` rO r r0 rh r rM r �all)r rd rx �min_N�ir\ �dfr# )r~ r$ � test_skew� s z"TestSeriesStatReductions.test_skewc sb ddl m� t�� �d�}� fdd�}| �d||� tjdgdd d gdd ggddddddgdd d dd d gdd dd dd ggd �}tt j � d�|d�}t� t ��" t�|�� |jdd�d � W d Q R X d}x�td |d �D ]�}tt �|��}tt �||f��}||k �r4t �|�� ��st�t �|�� ��� �sZt�q�d|�� k�sFt�|�� dk�� s�t�q�W d S )Nr )�kurtosisra c s � | dd�S )NF)r r# )rq )r� r# r$ rr rs z4TestSeriesStatReductions.test_kurt.<locals>.<lambda>�kurt�bar�one�two�threer r )Zlevels�codesr )rg )�levelr )r� r� rI rb rc r` r Z MultiIndexr r0 �randomZrandnZassert_produces_warning� FutureWarningrJ r� rO rh r rM r r� )r rd rx rg r\ r� r� r� r# )r� r$ � test_kurt� s& 2 & z"TestSeriesStatReductions.test_kurtN)FF)r6 r7 r8 r` re rf rj rl rm rn ry r} �tdZskip_if_no_scipyr� r� r# r# r# r$ r; Z s >r; )�__doc__rS �numpyr0 r( Z pandas.util._test_decorators�utilZ_test_decoratorsr� Zpandasr r r Zpandas._testingZ_testingrI Zpandas.core.arraysr r r r r; r# r# r# r$ �<module> s B
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка