Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/groupby/__pycache__/test_libgroupby.cpython-37.pyc
Назад
B �5�gf# � @ s� d dl Zd dlZd dlmZ d dlmZmZm Z m Z d dl m Z d dl mZ d dlmZ G dd� d�ZG dd � d e�ZG d d � d e�Zd d � Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Z dS ) � N)�groupby)�group_cumprod_float64� group_cumsum� group_mean� group_var)�ensure_platform_int)�isnac @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �GroupVarTestMixinc C s� t j�d�}t jt �d� �| j�}t jddd�}d|�dd��| j� }t � t � d�d ��d �}t � |�j d d d �j ddd�d d d �t jf }|d }| �||||� t �||| j�s�t�t�||� d S )Ni� )� � r �int64)�dtype� � r )� �intp)r r �F)�order)�axis�ddof� r )�np�random� RandomState�nan�ones�astyper �zeros�rand�tile�arangeZsqueeze�reshape�stdZnewaxis�algo�allclose�rtol�AssertionError�tm�assert_numpy_array_equal)�self�prng�out�counts�values�labels� expected_out�expected_counts� r1 �Z/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/groupby/test_libgroupby.py�test_group_var_generic_1d s & z+GroupVarTestMixin.test_group_var_generic_1dc C s� t j�d�}t jt �d� �| j�}t jddd�}d|�dd��| j� }t jddd�}t � |j dd �d gg�}|d }| � ||||� t � ||| j �s�t�t�||� d S ) Ni� )r r r r )r r r r )r r )r r r r r r r r r �arrayr" r# r$ r% r& r' r( )r) r* r+ r, r- r. r/ r0 r1 r1 r2 �%test_group_var_generic_1d_flat_labels$ s z7GroupVarTestMixin.test_group_var_generic_1d_flat_labelsc C s� t j�d�}t jt �d� �| j�}t jddd�}d|�dd��| j� }t � t � d�d��d �}t j |� ddd�d d d �d }|d }| � ||||� t �||| j�s�t�t�||� d S ) Ni� )r r r r )r r r )r r r r )r r )r r r r r r r r r r r r" r! r# r$ r% r& r' r( )r) r* r+ r, r- r. r/ r0 r1 r1 r2 �$test_group_var_generic_2d_all_finite4 s z6GroupVarTestMixin.test_group_var_generic_2d_all_finitec C s� t j�d�}t jt �d� �| j�}t jddd�}d|�dd��| j� }t j|d d �df< t � t � d�d ��d �}t � |d d �d f j ddd d �j ddd�d t jt �d� g�j�| j�}|d }| �||||� tj||dd� t�||� d S )Ni� )r r r r )r r r r )r r r r )r )r r g���Ơ>)r% )r r r r r r r r r r r Zvstackr! r" �Tr# r' �assert_almost_equalr( )r) r* r+ r, r- r. r/ r0 r1 r1 r2 �"test_group_var_generic_2d_some_nanC s (z4GroupVarTestMixin.test_group_var_generic_2d_some_nanc C s� t jt jgg| jd�}t jdgdd�}dt jd| jd� }t jddd�}| �||||� |d dksht�|d dksxt�t� |d d � d S ) N)r r r gUUUժ��?)r r r r )r r g ) r r4 r r r r r# r& r' r8 )r) r+ r, r- r. r1 r1 r2 �test_group_var_constantX s z)GroupVarTestMixin.test_group_var_constantN)�__name__� __module__� __qualname__r3 r5 r6 r9 r: r1 r1 r1 r2 r s r c @ s* e Zd ZdZee�ZejZ dZ dd� Z dS )�TestGroupVarFloat64Tg�h㈵��>c C s� t j�d�}t jt jgg| jd�}t jdgdd�}|�d�d �| j�}d|_t j ddd�}| � ||||� |d dks|t �t j |d d d d � d S ) Ni� )r r r i@B l J)�)i@B r r )r r gUUUUUU�?g����Mb@?)r% )r r r r4 r r r r �shaper r# r& r' r8 )r) r* r+ r, r- r. r1 r1 r2 �test_group_var_large_inputsn s z/TestGroupVarFloat64.test_group_var_large_inputsN) r; r< r= �__test__� staticmethodr r# r �float64r r% r@ r1 r1 r1 r2 r> g s r> c @ s" e Zd ZdZee�ZejZ dZ dS )�TestGroupVarFloat32Tg{�G�z�?N) r; r< r= rA rB r r# r �float32r r% r1 r1 r1 r2 rD } s rD c C s dd� } | d� | d� d S )Nc S s4 t jt j�d�| d�}t �dddg�}t �d| �}t jt|�t jd�}tt �t � d�t � t j d|f ���}t j }||||d d �d f |� dd � }t �||d d� �||dd� �||dd � �g�}t�||� t�|t jddd gt jd�� t j|d d�< ||||d d �d f |� t j|d< t�||� d S ) N� )r � � )r � r r c S s6 t | ��� rt�tjd�S | d | �� | �� | d gS )NrI r ���)r �allr �repeatr �max�min)�groupr1 r1 r2 �_ohlc� s z.test_group_ohlc.<locals>._check.<locals>._ohlc� )r r4 r Zrandnr �lenr r rL r �diffZr_� libgroupbyZ group_ohlcr' r8 r( r ) r �objZbinsr+ r, r. �funcrP �expectedr1 r1 r2 �_check� s &4 z test_group_ohlc.<locals>._checkrE rC r1 )rX r1 r1 r2 �test_group_ohlc� s rY c C s| d}t jdgdgdgdgg|d�}t �|�}t jddddgt jd�}d}| |||||� tj||�|dd�df dd � dS ) a Check a group transform that executes a cumulative function. Parameters ---------- pd_op : callable The pandas cumulative function. np_op : callable The analogous one in NumPy. dtype : type The specified dtype of the data. Fr r r rI )r r N)Z check_dtype)r r4 � zeros_liker r' r( )�pd_op�np_opr �is_datetimelike�dataZanswerr. �ngroupsr1 r1 r2 �(_check_cython_group_transform_cumulative� s r` c C s( t �| �j}tt j }}t|||� d S )N)r r �typer Zcumsumr` )Zany_real_dtyper r[ r\ r1 r1 r2 �"test_cython_group_transform_cumsum� s rb c C s" t j} tt j }}t||| � d S )N)r rC r Z cumproductr` )r r[ r\ r1 r1 r2 �#test_cython_group_transform_cumprod� s rc c C s� d} t jdddddgt jd�}d}t jdgdgdgt jgdggdd�}t �|�}|�t j� t||||| � t jddd t jd gdd�}t�|d d �df |� t �|�}|�t j� t ||||| � t jddd t jd gdd�}t�|d d �df |� d } t jt � dd �gd dd�d d �d f }t j|dd�}t ||� d�||| � t �t � dd �t � dd �t � dd �t � dd �t � dd �g�}t�|d d �df � d�|� d S )NFr )r r r r rI rC rG � r T�nsr zm8[ns]r ) r r4 r r rZ �fillr r' r( r � timedelta64�view)r] r. r_ r^ �actualrW r1 r1 r2 �!test_cython_group_transform_algos� s2 $ ( rj c C s� t jddd�} t jdgdd�}t jt �dd�t �d d�t �d �gd d�d d �d f �d��d�}t jt|�t jd�}t| |||d d � t � | d d �df t jdgdd�� d S )N)r r rC )r? r r r )r r re rI �NaTzm8[ns]T)r] r ) r r r4 rg rh r rR r r r' r( )ri r, r^ r. r1 r1 r2 �#test_cython_group_mean_datetimelike� s rl c C sx t jddd�} t jddd�}t jddd�d d �d f }t jdt jd�}tjtdd�� t| |||d d d � W d Q R X d S ) N)r r rC )r? r r r )r � min_count)�matchTr )r] rm )r r r �pytestZraisesr& r )ri r, r^ r. r1 r1 r2 �&test_cython_group_mean_wrong_min_count s rp c C s� t jddd�} t jdgdd�}t jt �d�t �d�gdd�d d �d f �d��d�}t jt|�t jd�}t| |||d d � t � | d d �df t jt � t � |d |d �d �dd�� d S ) N)r r rC )r? r r r )r rk zm8[ns]F)r] r r ) r r r4 rg rh r rR r r r' r( �divide�add)ri r, r^ r. r1 r1 r2 �:test_cython_group_mean_not_datetimelike_but_has_NaT_values s rs ) �numpyr ro Z pandas._libsr rT Zpandas._libs.groupbyr r r r Zpandas.core.dtypes.commonr Zpandasr Zpandas._testingZ_testingr' r r> rD rY r` rb rc rj rl rp rs r1 r1 r1 r2 �<module> s" U &
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка