Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/__pycache__/test_fillna.cpython-37.pyc
Назад
B �5�g�O � @ s� d dl Zd dlZd dlm mZ d dlmZm Z m Z m Z m Z m Z mZmZmZ d dlmZ d dlmZ G dd� d�Zdd� ZdS )� N) � Categorical� DataFrame� DatetimeIndex�NaT� PeriodIndex�Series�TimedeltaIndex� Timestamp� date_range)�_check_mixed_floatc @ sp e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z ejdd� �Zejdd� �Zejej�dd d ejd g�d d � ��Zejd!d"� �Zd#d$� Zd%d&� Zd'd(� Z d)d*� Z d+d,� Z d-d.� Z d/d0� Z ej�d1e!e"g�d2d3� �Z#ej�d1e!e"g�d4d5� �Z$d6d7� Z%d8d9� Z&d:d;� Z'd<d=� Z(d>d?� Z)d@dA� Z*dBdC� Z+dDdE� Z,dFdG� Z-dHdI� Z.dJS )K� TestFillNAc C s" |}t j|j|jd d� df< t j|j|jdd � df< |�d�}|j|jd d� df dk�� sft�|jdd�}t �|j|jd d� df ��� s�t�|j|jdd � df |j|jd df k�� s�t�d}tj t |d�� |�� W d Q R X d }tj t |d�� |jdd d� W d Q R X d S ) N� �A���r �pad)�methodz'Must specify a fill 'value' or 'method')�matchz(Cannot specify both 'value' and 'method'�ffill) �np�nan�loc�index�fillna�all�AssertionError�isnan�pytest�raises� ValueError)�self�datetime_frame�tfZ zero_filledZpadded�msg� r# �\/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/test_fillna.py�test_fillna_datetime s $ &4z TestFillNA.test_fillna_datetimec C sT |}t j|j|jdd� df< t j|j|jdd � df< |jdd� |jdd � d S ) Nr � �fooi����r r )�valuer )r )r r r r r )r �float_string_frame�mfr# r# r$ �test_fillna_mixed_type+ s z!TestFillNA.test_fillna_mixed_typec C sh |j dddgd�}tj|j|jdd � df< |jdd�}t|dd id � |jd d �}t|dd id � d S ) Nr �B�D)�columnsi����r )r( �C)�dtyper )r )�reindexr r r r r r )r Zmixed_float_framer* �resultr# r# r$ �test_fillna_mixed_float4 s z"TestFillNA.test_fillna_mixed_floatc C s< t dgd�}x*dD ]"}|jj|dd� |jj|d� qW d S )N�x)r. )r �backfillT)r �inplace)r )r r4 r )r �df�mr# r# r$ �test_fillna_empty? s z TestFillNA.test_fillna_emptyc C s� t ddtjdgddtjdgddtjdgg�}|�ddi�}t ddddgddddgddddgg�}t�||� |jddidd�}t�||� |d ks�t�d S )N�a�b�c� r' T)r6 )r r r r �tm�assert_frame_equalr )r r7 r2 �expectedZ return_valuer# r# r$ �test_fillna_different_dtypeF s *$ z&TestFillNA.test_fillna_different_dtypec C st t tj�dd��}tj|jdd�df< tj|jdd�df< |�� }d|jd< d|jd < |jdd d �}t� ||� d S ) N� � r= � r r i� )r= r )rC r= � )�limit) r r �random�randnr �iloc�copyr r> r? )r r7 r@ r2 r# r# r$ �test_fillna_limit_and_valueV s z&TestFillNA.test_fillna_limit_and_valuec C sl t ttd�gtd�tgd��}|�� }|d �|j|jd df �|d< |jd|d id�}t�||� d S )Nz2014-1-1z2013-1-1)�Date�Date2rL r rM )r( ) r r r rJ r r r r> r? )r r7 r@ r2 r# r# r$ �test_fillna_datelikeb s "z TestFillNA.test_fillna_datelikec C s� t dtd�tgi�}t dtd�td�gi�}t�|jdd�|� t dttd�gi�}t dtd�td�gi�}t�|jdd�|� d S )Nr z2012-11-11 00:00:00+01:00r )r �bfill)r r r r> r? r )r r7 �expr# r# r$ �test_fillna_tzawareq s z TestFillNA.test_fillna_tzawarec C s` t tdddd�ddtjtjgd��}|jdd �}t tdddd�d d d d gd��}t�||� d S ) N�20130101� z US/Eastern)�periods�tzrE r= )r r, r )r g �?g @)r r r r r r> r? )r r7 r2 r@ r# r# r$ �$test_fillna_tzaware_different_column� s z/TestFillNA.test_fillna_tzaware_different_columnc C s� t dddtjgdddgd�}ddtjdg}t||d��}t ddddgdddgd�}ddddg}t||d��}t dddgdddgd�}ddtjg}t||d��} t ddgdddgd�} ddg} t| | d��} |jddd�d �} t�| |� d }tjt |d �� |jd d d�d � W d Q R X |jdd�} t�| |� |j dgd�} t�| | � |� � } t�| | � t tjdtjgddgd�}t|dddgd��}t dddgddgd�}t|dddgd��}|�d�} t�| |� d S )NrE r= rC )� categoriesr: r; �d)�cats�vals)r( z3Cannot setitem on a Categorical with a new category)r rS r< r )r rY )Zsubset) r r r r r r> r? r r r Zdropna)r �catrZ r7 �cat2Zvals2Z df_exp_fillZcat3Zvals3Zdf_exp_drop_catsZcat4Zvals4Zdf_exp_drop_all�resr" r< Zcat_exp�df_expr# r# r$ �test_na_actions_categorical� s: z&TestFillNA.test_na_actions_categoricalc C sd t tjdtjg�}t tjtjtjg�}t||d��}tdtjd��}|�|�}tjtjtjg}tdddg|d�dd�}t�||� |j�tj�}t� ||j� |j �tj�}t� ||j � t dddt t g�} tdt | �i�}t�|jt d �|� t d d d t t gd d �} tdt | �i�}t�|jt d �|� td dd t t g�} tdt | �i�}t�|jt d �|� d S )Nr= )rY rZ g @�category)r0 z2011-01-01 09:00z2016-01-01 23:45r: )r( z2011-01�M)�freqz1 daysz2 days)r r r r r r r> r? rY �assert_series_equalrZ r r r r ) r r[ �valr7 Zmedianr] Zv_expr^ r2 �idxr# r# r$ �test_fillna_categorical_nan� s, z&TestFillNA.test_fillna_categorical_nanc C s� t ddtjgi�}|jddd�}t dddgi�}t�||� t ddtjgi�}|jddidd�}t dddgi�}t�||� d S )Nr: g �?r Zinfer)�downcastrE )r r r r r> r? )r r7 r2 r@ r# r# r$ �test_fillna_downcast� s z TestFillNA.test_fillna_downcastc C s� t dddgdddddgd �}|j}tt�d �gd dddddgd �}t�||� |�d�}t ddddgdddddgd �}t�||� t t d�ddgd d �}|�d�}t dt d�ddgd �}t�||� d S )Nr r, r/ rE r= rC rS r )r r. �object)r Zfloat64)r r. r0 r ) r Zdtypesr r r0 r> rc r r? �range)r r7 r2 r@ r# r# r$ � test_fillna_dtype_conversion� s " z'TestFillNA.test_fillna_dtype_conversionrd � rE g �?c C s@ t dtjgddgd��}|�tj|�}|�|�}t�||� d S )NrE g �?g @)r r, )r r r �replacer r> r? )r rd r7 r@ r2 r# r# r$ �*test_fillna_dtype_conversion_equiv_replace s z5TestFillNA.test_fillna_dtype_conversion_equiv_replacec C s t ddtjgtddd�ddd gdd d gd �td dd�d �}|�d �}t ddd gtddd�ddd gdd d gd �td dd�d �}t�||� t ddtjgtd�td�tgddd gdd d gd �td dd�d �}|�d �}t ddd gtd�td�d gddd gdd d gd �td dd�d �}t�||� d S )N������rR rC )rT r' �barZfoo2Zbar2)r r, r/ r- Z20130110)r �?z 2013-01-01z 2013-01-02) r r r r r r> r? r r )r r7 r2 r@ r# r# r$ � test_fillna_datetime_columns s8 z'TestFillNA.test_fillna_datetime_columnsc C s@ t j|d d d�< t j|d dd �< t�|�� |jdd�� d S )Nr r r r )r )r r r> r? r r )r r r# r# r$ � test_ffillA s zTestFillNA.test_ffillc C s\ t ddddgi�}d}tjt|d�� |�d�}W d Q R X t ddddgi�}t�||� d S )Nr: rE r= rC zSIn a future version of pandas all arguments of DataFrame.ffill will be keyword-only)r r )r r> �assert_produces_warning� FutureWarningr r? )r r7 r" r2 r@ r# r# r$ � test_ffill_pos_args_deprecationI s z*TestFillNA.test_ffill_pos_args_deprecationc C s@ t j|d d d�< t j|d dd �< t�|�� |jdd�� d S )Nr r r rO )r )r r r> r? rO r )r r r# r# r$ � test_bfillU s zTestFillNA.test_bfillc C s\ t ddddgi�}d}tjt|d�� |�d�}W d Q R X t ddddgi�}t�||� d S )Nr: rE r= rC zSIn a future version of pandas all arguments of DataFrame.bfill will be keyword-only)r r )r r> ru rv rO r? )r r7 r" r2 r@ r# r# r$ � test_bfill_pos_args_deprecation] s z*TestFillNA.test_bfill_pos_args_deprecationc C s� t �d�}tt j�dd�|d�}|d d� j|ddd�}|d d� �|�jdd�}t j|jd d �< t � ||� |d d � j|d dd�}|d d � �|�jd d�}t j|jd d �< t � ||� d S ) NrB rS )r r= r r )r rF )r ���rp r5 rC ) r �aranger rG rH r1 r r rI r> r? )r r r7 r2 r@ r# r# r$ � test_frame_pad_backfill_limiti s z(TestFillNA.test_frame_pad_backfill_limitc C s� t �d�}tt j�dd�|d�}|d d� �|�}|jddd�}|d d� �|�jdd�}t j|jd d �< t � ||� |d d � �|�}|jd dd�}|d d � �|�jd d�}t j|jd d �< t � ||� d S ) NrB rS )r r= r r )r rF )r rz rp r5 rC ) r r{ r rG rH r1 r r rI r> r? )r r r7 r2 r@ r# r# r$ �test_frame_fillna_limity s z"TestFillNA.test_frame_fillna_limitc C s( t tj�dd��t��}|�tj� d S )NrB rS )r r rG rH �astype�intr r )r r7 r# r# r$ � test_fillna_skip_certain_blocks� s z*TestFillNA.test_fillna_skip_certain_blocks�typec C sH t tj�dd���|�}d}tjt|d�� |jddd� W d Q R X d S )NrB rS z Limit must be greater than 0)r r r )rF ) r r rG rH r~ r r r r )r r� r7 r" r# r# r$ �test_fillna_positive_limit� s z%TestFillNA.test_fillna_positive_limitc C sH t tj�dd���|�}d}tjt|d�� |jddd� W d Q R X d S )NrB rS zLimit must be an integer)r r g �?)rF ) r r rG rH r~ r r r r )r r� r7 r" r# r# r$ �test_fillna_integer_limit� s z$TestFillNA.test_fillna_integer_limitc C s� t tj�dd��}tj|d d d�< tj|d dd �< |jdd�}||k sNt�|jddd � t�||� |jddidd �}|d ks�t�tj|d d d�< tj|d dd �< |jd d �}||k s�t�|jd dd � t�||� d S ) NrB rS rE rC ���r )r( T)r( r6 r )r )r r6 ) r r rG rH r r r r> r? )r r7 r@ r# r# r$ �test_fillna_inplace� s z TestFillNA.test_fillna_inplacec C s� t tjddtjtjgdddtjtjgtjddddgd��}|�ddd��}|�� }|d �d�|d < |d �d�|d < t�||� |�ddd d ��}|�|�� �}|�|�� �� �}t�||� t j t d d�� |j|�d�dd� W d Q R X d S )NrE r= rC rS )r: r; r< r r )r: r; r: r; rD )r: r; rX zcolumn by column)r )�axis) r r r r rJ r> r? �max�to_dictr r �NotImplementedError)r r7 r2 r@ r# r# r$ �test_fillna_dict_series� s z"TestFillNA.test_fillna_dict_seriesc C s� t tjddtjtjgdddtjtjgtjddddgd�td�d�}t tjdd d d gd d dddgdgd d�td�d�}|�|�}t tjddtjd gdddtjdgtjddddgd�td�d�}t�||� d S )NrE r= rC rS )r: r; r< ZVWXYZ)r rB r& � �( �2 �< �F �P �Z rq r )r: r; r' ZVWXuZ)r r r �listr r> r? )r r7 Zdf2r2 r@ r# r# r$ �test_fillna_dataframe� s" z TestFillNA.test_fillna_dataframec C s� t tj�dd��}tj|jd d �d d d�f< |jddd�}|jjdd�j}t� ||� |� dd d � |jddd�}|� t �jddd�}t� ||� d S ) NrB r= r rE )r r� r )r � r' r ) r r rG rH r �valuesr �Tr> r? �insertr~ �float)r r7 r2 r@ r# r# r$ �test_fillna_columns� s z TestFillNA.test_fillna_columnsc C s* t jtdd�� |jdd� W d Q R X d S )NZffil)r )r )r r r r )r � float_framer# r# r$ �test_fillna_invalid_method s z%TestFillNA.test_fillna_invalid_methodc C s� d}t jt|�d�d�� |�ddg� W d Q R X t jt|�d�d�� |�d� W d Q R X d}t jt|d�� |jd d �d f �|� W d Q R X d S ) NzA"value" parameter must be a scalar or dict, but you passed a "{}"r� )r rE r= �tuple)rE r= zP"value" parameter must be a scalar, dict or Series, but you passed a "DataFrame"r )r r � TypeError�formatr rI )r r� r" r# r# r$ �test_fillna_invalid_value s z$TestFillNA.test_fillna_invalid_valuec C s^ dd� t ddd�D �}tj�dd�}tt d�||d�}|jdd �}|j�� |j�� ksZt�d S ) Nc S s g | ]}d t |� �qS )zCOL.)�str)�.0�ir# r# r$ � <listcomp> s z9TestFillNA.test_fillna_col_reordering.<locals>.<listcomp>r r ro r& )r r. �datar )r ) rj r rG Zrandr r r. �tolistr )r �colsr� r7 �filledr# r# r$ �test_fillna_col_reordering s z%TestFillNA.test_fillna_col_reorderingc C s� |}t j|j|jdd� df< t j|j|jdd � df< |jdd�}|j|jdd� df dk�� sht�|d= |jg d�}|jdd�}d S ) Nr r& r' i����r r )r( )r. )r r r r r r r r1 )r r� r) r* r� Z empty_floatr2 r# r# r$ �test_fill_corner s $ zTestFillNA.test_fill_cornerc C sH t ddtjgi�}|jddiddid�}t dddgi�}t�||� d S )NZcol1rE r= Zint64)rg )r r r r r> r? )r r7 r2 r@ r# r# r$ �test_fillna_downcast_dict. s z$TestFillNA.test_fillna_downcast_dictc C sn t ddddtjgitd�}d}tjt|d�� |�dd d �}W d Q R X t dddddgitd�}t�||� d S ) Nr: rE r= rC )r0 ztIn a future version of pandas all arguments of DataFrame.fillna except for the argument 'value' will be keyword-only)r r ) r r r r� r> ru rv r r? )r r7 r" r2 r@ r# r# r$ � test_fillna_pos_args_deprecation5 s z+TestFillNA.test_fillna_pos_args_deprecationN)/�__name__� __module__� __qualname__r% r+ r3 r9 rA rK rN rQ rV r_ rf �tdZ&skip_array_manager_not_yet_implementedrh rk r �markZ parametrizer r rn Z skip_array_manager_invalid_testrs rt rw rx ry r| r} r� r r� r� r� r� r� r� r� r� r� r� r� r� r� r# r# r# r$ r sF +$"- # r c C sd t ddddgddddgddddggddd d gd �} | �dd�}|�d �}|�� �� �� d ks`t�d S ) NrE g �?r= g @rC g @�i1�i2Zi3�f1)r. r )r Zpivotr Zisna�sumr )r7 Z df_nonconsolr2 r# r# r$ �!test_fillna_nonconsolidated_frameB s r� )�numpyr r Z pandas.util._test_decorators�utilZ_test_decoratorsr� Zpandasr r r r r r r r r Zpandas._testingZ_testingr> Zpandas.tests.frame.commonr r r� r# r# r# r$ �<module> s , 1
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.07 |
proxy
|
phpinfo
|
Настройка