Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/groupby/aggregate/__pycache__/test_numba.cpython-37.pyc
Назад
B �5�g� � @ s� d dl Zd dlZd dlmZ d dlm mZ d dl m Z m Z m Z m Z mZ d dlmZ d dlmZ e�dd�dd� �Ze�dd�d d � �Ze�dd�ej�d �ej�d d dg�ej�dddg�dd� ����Ze�dd�ej�d �ej�d d dg�ej�dddg�dd� ����Ze�dd�dd� �Z e�dd�ej�dddgdddgdd �e d dd �g�d d � ��Z e�dd�d!d"� �Z e�dd�d#d$� �Z dS )%� N)�NumbaUtilError)� DataFrame�Index�NamedAgg�Series�option_context)�NUMBA_FUNC_CACHE�numbaz0.46.0c C s� dd� } t dddddgddddd gd �d d gd �}tjtdd�� |�d �j| dd� W d Q R X tjtdd�� |�d �d j| dd� W d Q R X d S )Nc S s t | �d S )Ng������@)�sum)�x� r �_/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/groupby/aggregate/test_numba.py�incorrect_function s z;test_correct_function_signature.<locals>.incorrect_function�a�bg �?g @g @g @g @)�key�datar r )�columnsz The first 2)�matchr )�engine)r �pytest�raisesr �groupby�agg)r r r r r � test_correct_function_signature s r c C s� dd� } t dddddgddddd gd �d d gd �}tjtdd�� |�d �j| ddd� W d Q R X tjtdd�� |�d �d j| ddd� W d Q R X d S )Nc [ s t | �d S )Ng������@)r )r �kwargsr r r r $ s z6test_check_nopython_kwargs.<locals>.incorrect_functionr r g �?g @g @g @g @)r r r r )r znumba does not support)r r � )r r )r r r r r r )r r r r r �test_check_nopython_kwargs" s r z ignore:\nThe keyword argument�jitTF� pandas_objr r c C s� dd� }| r dd l }|�|�}tdddddgdddd d gd �dd gd �}|||d�}|�d�} |dkrn| d } | j|d|d�} | jdd� dd�} t�| | � d S )Nc S s t �| �d S )Ng������@)�np�mean)�values�indexr r r � func_numba8 s z(test_numba_vs_cython.<locals>.func_numbar r r g �?g @g @g @g @)r r r )r )�nogil�parallel�nopythonr r )r � engine_kwargsc S s t �| �d S )Ng������@)r r! )r r r r �<lambda>J � z&test_numba_vs_cython.<locals>.<lambda>�cython)r )r r r r r �tm� assert_equal) r r r% r& r' r$ r r r( �grouped�result�expectedr r r �test_numba_vs_cython2 s ( r1 c C s. dd� }dd� }| r0dd l }|�|�}|�|�}tdddddgdd d d d gd �ddgd�}|||d�} |�d�} |dkr�| d } | j|d| d�} | jdd� dd�} t�| | � |dftks�t�| j|d| d�} | jdd� dd�} t�| | � |dftks�t�| j|d| d�} | jdd� dd�} t�| | � d S )Nc S s t �| �d S )Ng333333 @)r r! )r" r# r r r �func_1V s ztest_cache.<locals>.func_1c S s t �| �d S )Ng������@)r r! )r" r# r r r �func_2Y s ztest_cache.<locals>.func_2r r r g �?g @g @g @g @)r r r )r )r% r& r' r r )r r( c S s t �| �d S )Ng333333 @)r r! )r r r r r) k r* z test_cache.<locals>.<lambda>r+ )r Z groupby_aggc S s t �| �d S )Ng������@)r r! )r r r r r) r r* c S s t �| �d S )Ng333333 @)r r! )r r r r r) x r* ) r r r r r r, r- r �AssertionError) r r r% r& r' r2 r3 r r r( r. r/ r0 r r r � test_cacheO s. ( r5 c C s~ dd� } t dddddgddddd gd �d d gd �}|�d �}|j| dd�}tdd�� |j| d d�}W d Q R X t�||� d S )Nc S s t �| �d S )Ng333333 @)r r! )r" r# r r r r2 ~ s z&test_use_global_config.<locals>.func_1r r g �?g @g @g @g @)r r r r )r r )r zcompute.use_numbaT)r r r r r, �assert_frame_equal)r2 r r. r0 r/ r r r �test_use_global_config| s ( r7 �agg_func�min�maxr )�B�Cr; )�columnZaggfuncc C s� t dddddgdddddgd�d d gd �}|�d �}tjtd d �� |j| dd� W d Q R X tjtd d �� |d j| dd� W d Q R X d S )Nr r g �?g @g @g @g @)r r r r )r zNumba engine can)r r )r )r r r r �NotImplementedErrorr )r8 r r. r r r � test_multifunc_notimplimented� s ( r? c C s� dd� } t ddddgddddgd��}|�d�d }|j| ddd �}td gd dtddgdd �d �}t�||� |j| d dd �}tdgd dtddgdd �d �}t�||� d S )Nc S s | | d � � � S )N)r )r" r# �nr r r �sum_last� s z&test_args_not_cached.<locals>.sum_lastr r )�idr rB r r )r g �?� )�name)rD r# g @)r r r r r r, Zassert_series_equal)rA �dfZ grouped_xr/ r0 r r r �test_args_not_cached� s rF c C sp dd� } t dddgdddgd�d d d gd �}|�d �j| dd�}t ddgdgtddgd d�d�}t�||� d S )Nc S s t �|�S )N)r r! )r" r# r r r �f� s z+test_index_data_correctly_passed.<locals>.f�Ar; � � � )�group�v���������)r# rL r )r g ��g �rM )rD )r r# )r r Z aggregater r, r6 )rG rE r/ r0 r r r � test_index_data_correctly_passed� s $ rQ ) �numpyr r Z pandas.errorsr Z pandas.util._test_decorators�utilZ_test_decorators�tdZpandasr r r r r Zpandas._testingZ_testingr, Zpandas.core.util.numba_r Z skip_if_nor r �mark�filterwarningsZ parametrizer1 r5 r7 r? rF rQ r r r r �<module> s4 )
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка