Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/__pycache__/test_interpolate.cpython-37.pyc
Назад
B �5�g�0 � @ sT d dl Zd dlZd dlm mZ d dlmZm Z m Z d dl m Z G dd� d�ZdS )� N)� DataFrame�Series� date_rangec @ s: e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zej d d � �Z ej d d� �Z dd� Z e j�de jdddd�e jdddd�e jdddd�g�dd � �Zd d � Ze j�d d e jd!ej d"�g�d#d$� �Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Ze j�dd1d2g�d3d4� �Zeje j�d5d6d7d8g�d9d:� ��Z d;d<� Z d=S )>�TestDataFrameInterpolatec C s� t ddtjdgdddtjgddddgtd�d��}t d d d d gd d d d gddddgtd�d��}|�� }t�||� |�d��� }|�d�}d|jd< d|jd< t�||� d S )N� � � � � � �abcd)�A�B�C�Dg �?g @g @g @g "@r )r r )r r ) r �np�nan�list� interpolate�tm�assert_frame_equal� set_index�loc)�self�df�expected�result� r �a/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/test_interpolate.py�test_interp_basic s" z*TestDataFrameInterpolate.test_interp_basicc C s. t � }|�� }||k st�|}t�||� d S )N)r r �AssertionErrorr r )r r r r r r r �test_interp_empty) s z*TestDataFrameInterpolate.test_interp_emptyc C s` t ddtjdgdddtjgddddgtd�d��}d }tjt|d �� |jd d � W d Q R X d S ) Nr r r r r r r )r r r r a method must be one of \['linear', 'time', 'index', 'values', 'nearest', 'zero', 'slinear', 'quadratic', 'cubic', 'barycentric', 'krogh', 'spline', 'polynomial', 'from_derivatives', 'piecewise_polynomial', 'pchip', 'akima', 'cubicspline'\]. Got 'not_a_method' instead.)�matchZ not_a_method)�method)r r r r �pytest�raises� ValueErrorr )r r �msgr r r �test_interp_bad_method1 s z/TestDataFrameInterpolate.test_interp_bad_methodc C s� t ddtjdgdddtjgdddd gtd �d ��}|d �� }tddd dgd d�}t�||� |d jdd�}tddddgd d�}t�||� d S )Ng �?g @g @r r r r r r r )r r r r r g @)�name�infer)�downcast)r r r r r r r Zassert_series_equal)r r r r r r r �test_interp_comboD s z*TestDataFrameInterpolate.test_interp_comboc C sZ t ddtjdgtjdddgd��}|�d�}d}tjt|d�� |jd d � W d Q R X d S ) Nr r r r )r r r zkInterpolation with NaNs in the index has not been implemented. Try filling those NaNs before interpolating.)r" �values)r# )r r r r r$ r% �NotImplementedErrorr )r r r' r r r �test_interp_nan_idxV s " z,TestDataFrameInterpolate.test_interp_nan_idxc C sn t ddtjddtjdgddddddd gd ��}|�d �}|�� }|jd dd �}d|jjd< d|jjd< t� ||� |jdd�}d|jjd< d|jjd< t� ||� |jdd�}d|jjd< d|jjd< tj ||dd� |jdd�}d|jjd< d|jjd< t� ||� |jdd�}d|jjd< d|jjd< t� ||� |jd d�}d |jjd< d|jjd< tj ||dd� d S ) Nr r r r � r � � � )r r r � polynomial)r# �orderg���UUU@g�u:�@Zcubic)r# gv��:�@g�b+hZ @ZnearestF)Z check_dtypeZ quadraticg�^�r�@g!�g���@ZslineargX� ;�@�zerog @) r r r r �copyr r r r r )r r r r r r r �test_interp_various` s8 , z,TestDataFrameInterpolate.test_interp_variousc C s� t ddtjddtjdgddddddd gd ��}|jd d �}|�� }d|jd < d|jd< t�||� |jd dd�}t�||�tj �� |jdd �}|�� }|d |d< t�||� |jdd �}d|jd < d|jd< t�||� d S )Nr r r r r0 r r1 r2 r3 )r r Z barycentric)r# )r r � )r r r* )r# r+ Zkroghr Zpchipg @) r r r r r7 r r r �astype�int64)r r r r Z expectedkr r r �test_interp_alt_scipy� s" , z.TestDataFrameInterpolate.test_interp_alt_scipyc C s� t ddtjdgdddtjgtjdddgdtjddgddddgd��}|jdd �}|�� }d|jd < d|jd < d|jd < |d �tj�|d< t� ||� |jdd d�}t� ||� |jdd �}|�� }t� ||� d S )Nr r r r r r9 r0 )r r r r r )�axis)r r )r r )r r r- )r= r# r ) r r r r r7 r r: �float64r r )r r r r r r r �test_interp_rowwise� s$ z,TestDataFrameInterpolate.test_interp_rowwisezaxis_name, axis_number�rowsr Zrows_0)�id�indexZindex_0�columnsr Z columns_1c C s\ dt jdgdt jdgdddgd�}t|t jd �}|j|d d �}|j|d d �}t�||� d S ) Nr r9 r r0 r r r1 )r r r )�dtype�linear)r= r# )r r r r> r r r )r � axis_name� axis_number�datar r r r r r �test_interp_axis_names� s "z/TestDataFrameInterpolate.test_interp_axis_namesc C sH t dddtjddtjtjdg ddd dd dddd g d ��}|jdd � d S ) Nr g �?g �?r r1 �@ r r r ���)r r )r= )r r r r )r r r r r �test_rowwise_alt� s z)TestDataFrameInterpolate.test_rowwise_alt� check_scipyFT)Zmarksc C sv t tjtjdddgtjddtjdgd��}|�� }|�� }d|d jd < t�||� |rr|jd d d �}t�||� d S )Ng �?g �?r ���g ����)r r g �r r r4 r )r# r5 )r r r r r7 r r r )r rM r r r r r r �test_interp_leading_nans� s ( z1TestDataFrameInterpolate.test_interp_leading_nansc C sx t ddtjdgddddgtjddd gtjtjd d gddd dgd ��}d }tjt|d�� |�d�j|d� W d Q R X d S )Nr r r �a�b�c�dr r0 r r )r r r r �EzvCannot interpolate with all object-dtype columns in the DataFrame. Try setting at least one column to a numeric dtype.)r" �object)r= )r r r r$ r% � TypeErrorr: r )r r= r r' r r r � test_interp_raise_on_only_mixed� s z8TestDataFrameInterpolate.test_interp_raise_on_only_mixedc C sH t dddgdddgd�dd �}d }tjt|d �� |�� W d Q R X d S ) Nr r r r r r9 )r r rV )rD zvCannot interpolate with all object-dtype columns in the DataFrame. Try setting at least one column to a numeric dtype.)r" )r r$ r% rW r )r r r' r r r �%test_interp_raise_on_all_object_dtype� s z>TestDataFrameInterpolate.test_interp_raise_on_all_object_dtypec C s� t dddtjdgi�}t dddddgi�}|�� }|d jdd�}|d ksNt�t�||� |�� }|d jddd �}|d ks�t�t�||�d �� d S ) NrQ g �?g @g @g @T)�inplacer* )rZ r+ r; ) r r r r7 r r r r r: )r r r r � return_valuer r r �test_interp_inplace s z,TestDataFrameInterpolate.test_interp_inplacec C sf t ddddgtjdddgddddgd��}|jdd d d �}|jdd d d �}|d ksVt�t�||� d S ) Ng �?g @g @g @r r )rQ rR rS rE r F)r# r= rZ T)r r r r r r r )r r r r[ r r r �test_interp_inplace_row s ( z0TestDataFrameInterpolate.test_interp_inplace_rowc C s� t ddtjdgddddgddtjdgddddgd ��}t tjddddgd d �tjddddgd d �tjddddgd d �tjddddgd d �d ��}|jd d �}t�||� |ddg jd d �}t�||ddg � d S )Nr r r r g �?g @g @g @)r r r r r> )rD r; )r+ r r )r r r �arrayr r r )r r r r r r r �test_interp_ignore_all_good s z4TestDataFrameInterpolate.test_interp_ignore_all_goodc C st d}t d|d�}tj�||�}tj||dk < t|||d�}|jddd�}|jddd d �}|d ksdt�t� ||� d S ) Nr z 2014-01-01)�start�periodsg �?)rB rC rH r �time)r= r# T)r= r# rZ ) r r �randomZrandr r r r r r )r r= ra �idxrH r r r[ r r r � test_interp_time_inplace_axis2 s z6TestDataFrameInterpolate.test_interp_time_inplace_axis)rB r )rC r c C sl t �ddd�}t �|�}tt �|d�t �d�|d�j|d d�}|jd |d �}|jd |d �}t� ||� d S ) Nr �d i� )� r rg )rH rB rC g�G�z�?)rC rE )r# r= ) r Zlinspace�sinr ZtileZarangeZreindexr r r )r rF rG �x�yr r r r r r �test_interp_string_axis? s z0TestDataFrameInterpolate.test_interp_string_axisr# ZffillZbfill�padc C sh t ddddtjdgdddtjddgddd tjtjd gd ��}|j||d �}|j||d �}t�||� d S )Ng �?g @g @g @g @g @g @g $@g "@g >@)r r r )r= r# )r# r= )r r r Zfillnar r r )r r= r# r r r r r r �test_interp_fillna_methodsK s z3TestDataFrameInterpolate.test_interp_fillna_methodsc C s^ t ddddgi�}d}tjt|d�� |�dd�}W d Q R X t ddddgi�}t�||� d S ) NrQ r r r zzIn a future version of pandas all arguments of DataFrame.interpolate except for the argument 'method' will be keyword-only)r" rl r )r r Zassert_produces_warning� FutureWarningr r )r r r' r r r r r �%test_interpolate_pos_args_deprecationZ s z>TestDataFrameInterpolate.test_interpolate_pos_args_deprecationN) �__name__� __module__� __qualname__r r! r( r, r/ �tdZskip_if_no_scipyr8 r< r? r$ �markZ parametrize�paramrI rL rP rX rY r\ r] r_ re rk Z&skip_array_manager_not_yet_implementedrm ro r r r r r s4 ' r )�numpyr r$ Z pandas.util._test_decorators�utilZ_test_decoratorsrs Zpandasr r r Zpandas._testingZ_testingr r r r r r �<module> s
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка