Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/window/__pycache__/test_apply.cpython-37.pyc
Назад
B �5�g- � @ s� d dl Zd dlZd dlmZmZmZmZmZm Z d dl m Z ej �dddd g�dd� �Zdd� Zej �d d d g�d d � �Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zej �dddd igd dgg�d d � �ZdS ) � N)� DataFrame�Index� MultiIndex�Series� Timestamp� date_range�bad_raw� c C s: t jtdd��" ttd���d�jt| d� W d Q R X d S )Nz'raw parameter must be `True` or `False`)�match� r )�raw)�pytest�raises� ValueErrorr �range�rolling�apply�len)r � r �T/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/window/test_apply.py� test_rolling_apply_invalid_raw s r c C s� | \}}t ddddg�}|�d�jtj||d�}|�� �� s@t�|jddd�jtj||d�}t ddddgtd �}t � ||� d S ) Nr � r � � )�enginer )� min_periods� )�dtype) r r r �np�sumZisna�all�AssertionError�float�tmZassert_almost_equal)�engine_and_rawr r �vals�result�expectedr r r � test_rolling_apply_out_of_bounds s r( �windowr �2sc s� t tj�d�tjjdddd�d�tdddd�d �� � fd d �}� �| �j|d d �}� jdd � � � �}t � ||� t � t �� � �| �j|dd � W d Q R X d S )N� r r )�size)�A�B�20130101�s)�periods�freq)�indexc s$ | j d � j d krtjS | jd S )Nr ���)r3 r �nan�iloc)�x)�dfr r �f- s z1test_rolling_apply_with_pandas_objects.<locals>.fF)r r T)r r �randomZrandn�randintr r r r6 Z reindex_liker# �assert_frame_equalZexternal_error_raised�AttributeError)r) r9 r& r' r )r8 r �&test_rolling_apply_with_pandas_objects# s r> c C s� | \}}t g dd�}|�d�jdd� ||d�}t�||� t d d d g�}|jddd �jd d� ||d�}t d d d g�}t�||� |jddd �jt||d�}t�||� d S ) NZfloat64)r r c S s | � � S )N)Zmean)r7 r r r �<lambda>> � z$test_rolling_apply.<locals>.<lambda>)r r r r )r c S s t | �S )N)r )r7 r r r r? C r@ g �?g @)r r r r# �assert_series_equalr )r$ r r r' r& r0 r r r �test_rolling_apply: s rB c C s~ | \}}t tdddd�td�d���d�d }|jdd �}|jd d �}|jd d � ||d �}|jdd � ||d �}t�||� d S )Nr/ r+ r0 )r1 r2 )r- r. r- r r )r) �1sc S s dS )Nr r )r7 r r r r? W r@ z test_all_apply.<locals>.<lambda>)r r c S s dS )Nr r )r7 r r r r? X r@ )r r r Z set_indexr r r# r< )r$ r r r8 Zer�rr& r' r r r �test_all_applyK s rE c C s� | \}}t dtd�i�}td�td�td�td�td�g|_dd � }|jd d d �j|||d �}|�� }d|d< t�||� |jdd d �j|||d �}|�� }d|d< t�||� |jdd d �j|||d �}|�� }d|d< t�||� d S )Nr. r+ z20130101 09:00:00z20130101 09:00:02z20130101 09:00:03z20130101 09:00:05z20130101 09:00:06c S s dS )Nr r )r7 r r r r? h r@ z#test_ragged_apply.<locals>.<lambda>rC r )r) r )r r g �?r* Z5s) r r r r3 r r �copyr# r< )r$ r r r8 r9 r& r' r r r �test_ragged_apply\ s( rG c C s> t jtdd��& ttd���d�jdd� dd� W d Q R X d S )Nz)engine must be either 'numba' or 'cython')r r c S s | S )Nr )r7 r r r r? { r@ z%test_invalid_engine.<locals>.<lambda>�foo)r )r r r r r r r r r r r �test_invalid_enginey s rI c C sD t jtdd��, ttd���d�jdd� dddid � W d Q R X d S ) Nz+cython engine does not accept engine_kwargs)r r c S s | S )Nr )r7 r r r r? � r@ z3test_invalid_engine_kwargs_cython.<locals>.<lambda>�cythonZnopythonF)r Z engine_kwargs)r r r r r r r r r r r �!test_invalid_engine_kwargs_cython~ s rK c C s@ t jtdd��( ttd���d�jdd� ddd� W d Q R X d S ) Nz.raw must be `True` when using the numba engine)r r c S s | S )Nr )r7 r r r r? � r@ z(test_invalid_raw_numba.<locals>.<lambda>FZnumba)r r )r r r r r r r r r r r �test_invalid_raw_numba� s rL � args_kwargs�parr )r c C s� dd� }t ddgddgd��}tddg�}t ddgdd gg|d �}|�d�j|| d | d d �}t�||� tjd dgdd gd�}tdd g|dd�}|� d�d �d�}|j|| d | d d �}t� ||� d S )Nc S s t �| | �S )N)r r )r7 rN r r r rH � s z+test_rolling_apply_args_kwargs.<locals>.foor r )�gr�arO rP g &@g (@)�columnsr )�args�kwargs)r r )r r )�names)r3 �name) r r r r r# r< r � from_tuplesr �groupbyrA )rM rH r8 �idxr' r& ZmidxZ gb_rollingr r r � test_rolling_apply_args_kwargs� s rY )�numpyr r Zpandasr r r r r r Zpandas._testingZ_testingr# �markZ parametrizer r( r> rB rE rG rI rK rL rY r r r r �<module> s
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка