3
���h~ � @ s� d dl Zd dlZd dljjZd dlmZm Z d dl
j Z d dl
mZ ejdd�ejjd�G dd� d���Zejdd�d d
� �ZdS ) � N)�Series�option_context)�NUMBA_FUNC_CACHE�numbaz0.46.0z ignore:\nThe keyword argumentc @ s@ e Zd Zejjdddg�dd� �Zejjdddg�dd� �ZdS ) � TestApply�jitTFc
C s� dd� }|r dd l }|j|�}|||d�}d
} ttd��}
|
jd|d�j|| d|d d
�} |
jd|d�j|d | d d �} tj| | � d S )Nc W s( d}x|D ] }||7 }q
W t j| �| S )Nr )�np�mean)�x�argsZarg_sum�arg� r
�@/tmp/pip-build-5_djhm0z/pandas/pandas/tests/window/test_numba.py�f s
z)TestApply.test_numba_vs_cython.<locals>.fr )�nogil�parallel�nopython� �
)�centerr T)r �engine�
engine_kwargs�raw�cython)r r r )r )r r r �range�rolling�apply�tm�assert_series_equal)
�selfr r r r r r r r r �s�result�expectedr
r
r �test_numba_vs_cython s
z TestApply.test_numba_vs_cythonc C s� dd� }dd� }|r0dd l }|j|�}|j|�}|||d�}ttd��jd�} | j|d |d
d �}
| j|d d
d
�} tj|
| � |dftks�t �| j|d |d
d �}
| j|d d
d
�} tj|
| � | j|d |d
d �}
| j|d d
d
�} tj|
| � d S )Nc S s t j| �d S )N� )r r )r
r
r
r �func_1+ s z$TestApply.test_cache.<locals>.func_1c S s t j| �d S )N� )r Zstd)r
r
r
r �func_2. s z$TestApply.test_cache.<locals>.func_2r )r r r r r r T)r r r r )r r Z
rolling_apply)
r r r r r r r r r �AssertionError) r r r r r r% r' r r Zrollr! r" r
r
r �
test_cache( s*
zTestApply.test_cacheN)�__name__�
__module__� __qualname__�pytest�markZ parametrizer# r) r
r
r
r r s r c C sf dd� } t td��}tdd�� |jd�j| d dd�}W d Q R X |jd�j| ddd�}tj||� d S ) Nc S s t j| �d S )Nr )r r )r
r
r
r r R s z!test_use_global_config.<locals>.fr zcompute.use_numbaTr )r r r )r r r r r r r )r r r! r" r
r
r �test_use_global_configP s r/ )Znumpyr r- Z pandas.util._test_decorators�utilZ_test_decorators�tdZpandasr r Zpandas._testingZ_testingr Zpandas.core.util.numba_r Z
skip_if_nor. �filterwarningsr r/ r
r
r
r �<module> s
D |