Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/window/__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 d dlmZ d dlmZ e�dd�ej�d�G dd � d ���Ze�dd�G d d � d ��Ze�dd�d d � �Ze�dd�dd� �Ze�dd�ejjej�d�G dd� d����ZdS )� N)�NumbaUtilError)� DataFrame�Series�option_context� to_datetime)�NUMBA_FUNC_CACHE�numbaz0.46.0z ignore:\nThe keyword argumentc @ sP e Zd Zej�dddg�dd� �Zdd� Zdd � Zej�dddg�d d � �Z d S ) � TestEngine�jitTFc C s� dd� }|r dd l }|�|�}|||d�}d} ttd��} | jd|d�j|| d |d d �} | jd|d�j|d | d d �} t�| | � d S )Nc W s( d}x|D ] }||7 }q W t �| �| S )Nr )�np�mean)�x�argsZarg_sum�arg� r �T/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/window/test_numba.py�f s z0TestEngine.test_numba_vs_cython_apply.<locals>.fr )�nogil�parallel�nopython)� � r )�centerr T)r �engine� engine_kwargs�raw�cython)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_apply s z%TestEngine.test_numba_vs_cython_applyc C sv |}|||d�}t t�d��}|�d�}t||�d|d�} t||�dd�} ttd|� ��d ftksft�t�| | � d S ) N)r r r � r r )r r r )r �nan�Rolling_apply_single) r r �eyer �getattrr �AssertionErrorr �assert_frame_equal) r" r r r �$arithmetic_numba_supported_operators�methodr �df�rollr$ r% r r r �$test_numba_vs_cython_rolling_methods. s z/TestEngine.test_numba_vs_cython_rolling_methodsc C st |}|||d�}t t�d��}|�� }t||�d|d�} t||�dd�} ttd|� ��dftksdt�t�| | � d S ) N)r r r r' r )r r r )r r( ZExpanding_apply_single) r r r* � expandingr+ r r, r r- ) r" r r r r. r/ r r0 �expandr$ r% r r r �&test_numba_vs_cython_expanding_methods@ s z1TestEngine.test_numba_vs_cython_expanding_methodsc C s� dd� }dd� }|r0dd l }|�|�}|�|�}|||d�}ttd���d�} | j|d |d d �} | j|d d d �} t�| | � |dftks�t �| j|d |d d �} | j|d d d �} t�| | � | j|d |d d �} | j|d d d �} t�| | � d S )Nc S s t �| �d S )N� )r r )r r r r �func_1X s z+TestEngine.test_cache_apply.<locals>.func_1c S s t �| �d S )Nr' )r Zstd)r r r r �func_2[ s z+TestEngine.test_cache_apply.<locals>.func_2r )r r r r r r T)r r r r )r r r) ) r r r r r r r r! r r, ) r" r r r r r7 r8 r r r1 r$ r% r r r �test_cache_applyU s* zTestEngine.test_cache_applyN) �__name__� __module__� __qualname__�pytest�mark� parametrizer& r2 r5 r9 r r r r r s r c @ s� e Zd Zejjddd� dd� gddgd�dd � �Zejjdd d� d d� gddgd�d d � �Zejjddd� dd� gddgd�dd� �Zejjddd� dd� gddgd�dd� �Z dS )� TestEWMMean�grouperc C s | S )Nr )r r r r �<lambda>� � zTestEWMMean.<lambda>c C s | � d�S )N�A)�groupby)r r r r rB � rC �NonerE )Zidsc C sP t ddddgtd�d��}tjtdd�� ||�jdd�jd d � W d Q R X d S ) N�a�br6 )rD �Bzengine must be either)�matchg �?)�comZfoo)r )r r r= �raises� ValueError�ewmr )r" rA r0 r r r �test_invalid_engine s z TestEWMMean.test_invalid_enginec C s | S )Nr )r r r r rB � rC c C s | � d�S )NrD )rE )r r r r rB � rC c C sV t ddddgtd�d��}tjtdd��$ ||�jdd�jd d d id � W d Q R X d S ) NrG rH r6 )rD rI zcython engine does not)rJ g �?)rK r r T)r r )r r r= rL rM rN r )r" rA r0 r r r �test_invalid_engine_kwargs� s z&TestEWMMean.test_invalid_engine_kwargsc C s | S )Nr )r r r r rB � rC c C s | � d�S )NrD )rE )r r r r rB � rC c C sd t ddddgtd�d��}||�jd||d�}|||d�} |jd| d �} |jd d �} t�| | � d S ) NrG rH r6 )rD rI g �?)rK �adjust� ignore_na)r r r r )r r r )r )r r rN r r r- ) r" rA r r r rR rQ r0 rN r r$ r% r r r �test_cython_vs_numba� s z TestEWMMean.test_cython_vs_numbac C s | S )Nr )r r r r rB � rC c C s | � d�S )NrD )rE )r r r r rB � rC c C s� d}t ddddddg�}tddddddgd d d d d d gd ��}||�j|d ||d�} |||d�} | jd| d�} | jdd�} t�| | � d S )Nz23 daysz 2020-01-01z 2020-01-02z 2020-01-10z 2020-02-23z 2020-01-03rG rH r � r )rD rI T)�halfliferQ rR �times)r r r r )r r r )r )r r rN r r r- ) r" rA r r r rR rU rV r0 rN r r$ r% r r r �test_cython_vs_numba_times� s & z&TestEWMMean.test_cython_vs_numba_timesN) r: r; r< r= r> r? rO rP rS rW r r r r r@ } s $$ $r@ c C sf dd� } t td��}tdd�� |�d�j| d dd�}W d Q R X |�d�j| ddd�}t�||� d S ) Nc S s t �| �d S )Nr )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_config� s rX c C sF t jtdd��. ttd���d�jdd� ddiddd � W d Q R X d S ) Nz"numba does not support kwargs with)rJ rT c S s | S )Nr )r r r r rB � rC z.test_invalid_kwargs_nopython.<locals>.<lambda>rG r T)�kwargsr r )r= rL r r r r r r r r r � test_invalid_kwargs_nopython� s rZ c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )�TestTableMethodc C sH dd� }t jtdd��( ttd��jddd�j|dd d � W d Q R X d S ) Nc S s t j| dd�d S )Nr )�axisrT )r �sum)r r r r r � s z7TestTableMethod.test_table_series_valueerror.<locals>.fz1method='table' not applicable for Series objects.)rJ rT �table)r/ r T)r r )r= rL rM r r r r )r" r r r r � test_table_series_valueerror� s z,TestTableMethod.test_table_series_valueerrorc C sn |}|||d�}t t�d��}t|jdd|dd�|�|dd�} t|jdd |dd�|�|dd�} t�| | � d S ) N)r r r � r r^ r )r/ r\ � min_periodsr )r r �single)r r r* r+ r r r- ) r" r\ r r r r. r/ r r0 r$ r% r r r �!test_table_method_rolling_methods� s z1TestTableMethod.test_table_method_rolling_methodsc C sr |||d�}dd� }t t�d��}|jdd|dd�j|d |d d �}|jdd |dd�j|d |d d �} t�|| � d S ) N)r r r c S s t j| dd�d S )Nr )r\ rT )r r] )r r r r r � s z:TestTableMethod.test_table_method_rolling_apply.<locals>.fr` r r^ r )r/ r\ ra Tr )r r r rb )r r r* r r r r- ) r" r\ r r r r r r0 r$ r% r r r � test_table_method_rolling_apply� s z/TestTableMethod.test_table_method_rolling_applyc C s� dd� }t dddgdddgddd gdd d gg�}|jdd d d�j|ddd�}t dddgdddgdddgdddgg�}t�||� d S )Nc S sh t �d| jd f�}| d d �d d�f | d d �df jdd�| d d �df �� |d d �d d�f< |S )NrT r r )r\ )r Zones�shaper] )r Zarrr r r � weighted_mean s PzNTestTableMethod.test_table_method_rolling_weighted_mean.<locals>.weighted_meanrT r g333333�?r` g�������?r6 g�������?r' gffffff�?r^ r )r/ ra Tr )r r g �?g @g������?g�k�}�� @g�k�}��@g��1���?� )r r r r r- )r" rf r0 r$ r% r r r �'test_table_method_rolling_weighted_mean s ( z7TestTableMethod.test_table_method_rolling_weighted_meanc C sj |||d�}dd� }t t�d��}|jd|d�j|d|dd �}|jd |d�j|d|dd �} t�|| � d S ) N)r r r c S s t j| dd�d S )Nr )r\ rT )r r] )r r r r r s z<TestTableMethod.test_table_method_expanding_apply.<locals>.fr` r^ )r/ r\ Tr )r r r rb )r r r* r3 r r r- ) r" r\ r r r r r r0 r$ r% r r r �!test_table_method_expanding_apply s z1TestTableMethod.test_table_method_expanding_applyc C sf |}|||d�}t t�d��}t|jd|d�|�|dd�} t|jd|d�|�|dd�} t�| | � d S )N)r r r r` r^ )r/ r\ r )r r rb )r r r* r+ r3 r r- ) r" r\ r r r r. r/ r r0 r$ r% r r r �#test_table_method_expanding_methods# s z3TestTableMethod.test_table_method_expanding_methodsN) r: r; r< r_ rc rd rh ri rj r r r r r[ � s r[ ) �numpyr r= Z pandas.errorsr Z pandas.util._test_decorators�utilZ_test_decorators�tdZpandasr r r r Zpandas._testingZ_testingr Zpandas.core.util.numba_r Z skip_if_nor> �filterwarningsr r@ rX rZ Zslowr[ r r r r �<module> s k?
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка