Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/groupby/transform/__pycache__/test_numba.cpython-37.pyc
Назад
B �5�g� � @ sx d dl Z d dlmZ d dlm mZ d dlmZm Z m Z d dl m Z d dlmZ e�dd�dd� �Ze�dd�d d � �Ze�dd�e j�d �e j�d d dg�e j�dddg�dd� ����Ze�dd�e j�d �e j�d d dg�e j�dddg�dd� ����Ze�dd�dd� �Ze�dd�e j�dddgdddgdd �g�d d � ��Ze�dd�d d � �Ze�dd�d!d"� �ZdS )#� N)�NumbaUtilError)� DataFrame�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 | d S )N� � )�xr r �_/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/groupby/transform/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� transform)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 | d S )Nr 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 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 | d S )Nr r )�values�indexr r r �func5 s z"test_numba_vs_cython.<locals>.funcr r r g �?g @g @g @g @)r r r )r )�nogil�parallel�nopythonr r )r � engine_kwargsc S s | d S )Nr r )r r r r �<lambda>G � 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_cython/ s ( r, 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 | d S )Nr r )r r r r r �func_1S s ztest_cache.<locals>.func_1c S s | d S )N� r )r r r r r �func_2V 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 | d S )Nr r )r r r r r$ h r% z test_cache.<locals>.<lambda>r& )r Zgroupby_transformc S s | d S )Nr. r )r r r r r$ o r% c S s | d S )Nr r )r r r r r$ u r% ) r r r r r r' r( r �AssertionError) r r r r! r" r- r/ r r r# r) r* r+ r r r � test_cacheL s. ( r1 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 | d S )Nr r )r r r r r r- { 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)r- r r) r+ r* r r r �test_use_global_configy s ( r3 �agg_func�min�max�sum)�B�Cc 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 )r4 r r) r r r � test_multifunc_notimplimented� s ( r; c C s� dd� } t ddddgddddgd��}|�d�d }|j| ddd �}td gd dd �}t�||� |j| d dd �}tdgd dd �}t�||� d S )Nc S s | | d � � � S )N)r7 )r r �nr r r �sum_last� s z&test_args_not_cached.<locals>.sum_lastr r )�idr r> r r )r g �?� )�name� g @)r r r r r' Zassert_series_equal)r= �dfZ grouped_xr* r+ r r r �test_args_not_cached� s rC c C sl dd� } t dddgdddgd�d d d gd �}|�d �j| dd�}t dddgdgd d d gd�}t�||� d S )Nc S s |d S )Nr r )r r r r r �f� s z+test_index_data_correctly_passed.<locals>.f�Ar8 r? r. � )�group�v���������)r rG r )r g �g �g �rH )r r )r r r r' r2 )rD rB r* r+ r r r � test_index_data_correctly_passed� s $ rL ) r Z pandas.errorsr Z pandas.util._test_decorators�utilZ_test_decorators�tdZpandasr r r Zpandas._testingZ_testingr' Zpandas.core.util.numba_r Z skip_if_nor r �mark�filterwarningsZ parametrizer, r1 r3 r; rC rL r r r r �<module> s* ) &
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка