Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/series/methods/__pycache__/test_fillna.cpython-37.pyc
Назад
B �5�gry � @ s� d dl m Z mZmZ d dlZd dlZd dlZd dlmZm Z m Z m Z m Z m Z mZmZmZmZ d dlmZ G dd� d�ZG dd� d�ZdS )� )�datetime� timedelta�timezoneN) � Categorical� DataFrame� DatetimeIndex�NaT�Period�Series� Timedelta� Timestamp� date_range�isnac @ sP e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Ze j � d ddg�dd� �Z dd� Z dd� Zdd� Zdd� Zdd� Zd d � Ze j � d d d d d d d gfd d d d!�d d d d d gfd"d id d d ejejgfd d d#�d d d d ejgfed �d ejd ejejgfed d"gd$�d d d ejejgfed d d#��d d d d ejgfed d gd%d&gd$�d ejd d d gfg�d'd(� �Ze j � d ed d d)d*d+g�d d d d*d+gfed d*d d*d g�d d*d d*d gfeed d*d d*d gd d)d*d+d gd,��d d*d d*d gfg�d-d.� �Zd/d0� Ze j � d1ed2d3g�e j � d4ej �d5d6� ��Z d7d8� Z d9d:� Z d;d<� Z d=d>� Z!d?d@� Z"dAdB� Z#dCS )D�TestSeriesFillNAc C s` t dddtjgdd�}|jdd�}|j|jd d�}|�� }|jd |jd < t�||� t�||� td |i�}|jdd�}|j|jd d�}td |i�}t� ||� t� ||� t tjdddgdd�}|jd d�}|j|d d�}|�� }|d |d< t�||� t�||� td |i�}|jd d�}|j|d d�}td |i�}t� ||� t� ||� d S ) Nr � � zM8[ns])�dtype�pad)�method)�value� �A�bfill) r r r �fillna�values�copy�tm�assert_series_equalr Zassert_frame_equal)�self�series�filledZfilled2�expectedZdf� r" �]/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/series/methods/test_fillna.py�test_fillna_nat s4 z TestSeriesFillNA.test_fillna_natc C sf t dddddgt�d�d�}t�||jdd �� tj|d < t dddddg|jd�}t�|jdd �|� t dddddg|jd�}t�|jd d �|� t ddd ddg|jd�}t�|jdd �|� d}tj t |d�� |�� W d Q R X d}tj t |d�� |jddd� W d Q R X t tj g�}t dg�}|�|�}t dg�}t�||� |�i �}t�||� |�t dt d��}t�||� |�|�}t�||� |�ddi�}t�||� |�ddi�}t�|t tj g�� |�ddd��}t�||� |�t ddd���}t�||� |�t ddd�ddgd��}t�||� t ddd gt d��}t dtj d gt d��}|�|�}t dddgt d��}t�||� t tj ddd gd�} | jddd�}t dtj tj gddd gd�}t�||� | jdd d�}t ddtj gddd gd�}t�||� d d d g} xT| D ]L} t ddtj tj dgd d�} | �| �}t dd| | dgd d�}t�||� �qW d S )!Ng g �?g @g @g @� )�index�ffill)r r �backfillg @)r z'Must specify a fill 'value' or 'method')�matchz(Cannot specify both 'value' and 'method'r )r r r r" )r )r r � �abcZbaci� )�limit�0z1.5z-0.3�float64�object)r r � makeDateIndexr r �np�NaNr& �pytest�raises� ValueError�nanr/ �list) r �datetime_series�ts�exp�msg�s1�s2�resultr! �ser�vals�valr" r" r# � test_fillna@ sl z TestSeriesFillNA.test_fillnac C s� t td�tg�}|�tddd��}t td�tddd�gdd�}t�||� |jddgtddd�d d �}t�||� |jddgtddd�d d �}t�||� |�d �}t td�d g�}t�||� |�� }d |d < t�||� d S ) N�20130101z US/Eastern)�tzz 2013-01-01r/ )r TF�ignore)�errorsZfoor )r r r r r r �wherer )r r? r> r! �ser2r" r" r# �test_fillna_consistency� s$ z(TestSeriesFillNA.test_fillna_consistencyc C sp t dtjg�}|jddd�}t ddg�}t�||� t dtjg�}|jddidd�}t ddg�}t�||� d S )Ng �?r Zinfer)Zdowncastr )r r1 r6 r r r )r r? r> r! r" r" r# �test_fillna_downcast� s z%TestSeriesFillNA.test_fillna_downcastc C s^ t td�td�td�td�g�}|�� }||�}|�tdd��}t td�td�td�tddd�g�}||�}t�||� d }tj t |d �� |�d� W d Q R X |�tdd��}t tdd�td�td�tddd�g�}||�}t�||� |�tddd��}t tddd�td�td�tddd�g�}||�}t�||� |�t � d ��}t tdd�td�td�tddd�g�}||�}t�||� |�t �}t t td�td�tddd�gd d �}||�}t�||� t j|d< ||�}|�� }|�tdd��}t j|d< ||�}t�||� t j|d< ||�}|�� }|�tdd��}tddd�|d< ||�}t�||� d S )NrC �20130102z20130103 9:01:01r )�secondsr i�~ )�daysrL zBvalue should be a 'Timedelta', 'NaT', or array of those. Got 'int')r) i ʚ;zm8[ns])r r )r r �diffr r r r � assert_equalr3 r4 � TypeErrorr1 Z timedelta64r r6 r' r )r �frame_or_seriesr? �td�objr> r! r; r" r" r# �test_timedelta_fillna� s� z&TestSeriesFillNA.test_timedelta_fillnac C s� t td�td�td�td�g�}tj|d< |�� }t td�td�td�td�g�}t�||� |�� }t td�td�td�td�g�}t�||� d}tjt |d�� t t t dg�}W d Q R X t dddgdd �}|j d d �}t�||� d S ) NrC rK z20130103 9:01:01r z containing strings is deprecated)r) z2013-08-05 15:30:00.000001zM8[ns])r r( )r ) r r r1 r6 r' r r r �assert_produces_warning� FutureWarningr r )r r? r> r! r; r" r" r# �test_datetime64_fillna s< z'TestSeriesFillNA.test_datetime64_fillnarD z US/Easternz Asia/Tokyoc C s� t td�ttd�tg�}t ddddg�}|�td��}t td�td�td�td�g�}t�||� t�t|�|� |�td|d��}t td�td|d�td�td|d�g�}t�||� t�t|�|� |�d�}t td�dtd�dgtd�}t�||� t�t|�|� |�td|d�td �d ��}t td�td|d�td�td �g�}t�||� t�t|�|� |�td�td �d ��}t td�td�td�td �g�}t�||� t�t|�|� tdtdtg|d�}t |�}|j d |� d �k�s�t �t�t|�|� |�td��}t td|d�td�td|d�td�g�}t�||� t�t|�|� |�td|d��}tddddg|d�}t |�}t�||� t�t|�|� |�td|d�� � �}tddddg|d�}t |�}t�||� t�t|�|� |�d�}t td|d�dtd|d�dgtd�}t�||� t�t|�|� |�td|d�td �d ��}t td|d�td|d�td|d�td �g�}t�||� t�t|�|� |�td|d�td |d�d ��}t td|d�td|d�td|d�td |d�g�}t�||� t�t|�|� |�td ��}t td|d�td�td|d�td�g�}t�||� t�t|�|� |�td dd��}t td|d�tddd�td|d�tddd�g�}t�||� t�t|�|� d S )Nz2011-01-01 10:00z2011-01-03 10:00FTz2011-01-02 10:00)rD ZAAA)r z2011-01-04 10:00)r r zdatetime64[ns, �]rC z 2013-01-01z US/Pacific) r r r r r r r r/ r r �AssertionErrorZ to_pydatetime)r rD r? Znull_locr> r! �idxr" r" r# �test_datetime64_tz_fillnaR s� z*TestSeriesFillNA.test_datetime64_tz_fillnac C st t td�tg�}t td�td�g�}t�|jdd�|� t ttd�g�}t td�td�g�}t�|jdd�|� d S )Nz2012-11-11 00:00:00+01:00r )r r )r r r r r r )r r? r: r" r" r# � test_fillna_dt64tz_with_method s z/TestSeriesFillNA.test_fillna_dt64tz_with_methodc C sL t tjtd�gddgd�}|�td��}t td�ddgd�}t�||� d S )Nz1 daysr �B)r& r )r r1 r6 r r r r r )r r? r> r! r" r" r# �test_fillna_pytimedelta/ s z(TestSeriesFillNA.test_fillna_pytimedeltac C sh t tddd�tddd�g�}|�tddd��}t tddd�tddd�g�}t�||� |jdksdt�d S )Nz2011-01�M)�freqr z2012-01z Period[M])r r r r r r rY )r r? �resr: r" r" r# �test_fillna_period7 s z#TestSeriesFillNA.test_fillna_periodc C s� t td�td�td�td�g�}tj|d< ||�}|�td��}t td�td�td�td�g�}||�}t�||� |�t�}|}t�||� d S )NrC rK z20130103 9:01:01r Z20130104)r r r1 r6 r r rO r )r rQ r? rS r> r! r" r" r# �test_fillna_dt64_timestamp@ s$ z+TestSeriesFillNA.test_fillna_dt64_timestampc C sV t td�ttd�g�}t�dd�}|�|�}t td�td�td�g�}t�||� d S )Nz 2010-01-01z 2000-01-01z 1975-04-05�ms)r r r r1 Z datetime64r r r )r r? rA r> r! r" r" r# �test_fillna_dt64_non_nao] s z)TestSeriesFillNA.test_fillna_dt64_non_naoc C sb t tjdtjdtjgdddddg�}|�� }|jdd d �}|d ksFt�|jdd �}t�||� d S ) Ng �?g @�z�a�b�c�dr T)r �inplace)r )r r1 r6 r r rY r r )r �x�y� return_valuer! r" r" r# �test_fillna_numeric_inplaceh s $ z,TestSeriesFillNA.test_fillna_numeric_inplacezfill_value, expected_outputrg rh )r r r* r )r r )r& r r* c C sV dt jdt jt jg}tt|ddgd��}tt|ddgd��}|�|�}t�||� d S )Nrg rh )� categories)r1 r6 r r r r r )r � fill_value�expected_output�datar? r: r> r" r" r# �test_fillna_categoricalu s z(TestSeriesFillNA.test_fillna_categoricalri rj �e)rp c C sb dt jdt jt jg}tt|dddddgd��}tt|dddddgd��}|�|�}t�||� d S )Nrg rh ri rj ru )rp )r1 r6 r r r r r )r rq rr rs r? r: r> r" r" r# �+test_fillna_categorical_with_new_categories� s z<TestSeriesFillNA.test_fillna_categorical_with_new_categoriesc C s: dt jdt jt jg}tt|ddgd��}|j}d}tjt|d�� |�d� W d Q R X d}tjt|d�� |�td�� W d Q R X tjt|d�� |�ddd�� W d Q R X d }tjt |d�� |�ddg� W d Q R X d }tjt |d�� |�d � W d Q R X d }tjt |d�� |�t dgdgd��� W d Q R X d S ) Nrg rh )rp z3Cannot setitem on a Categorical with a new category)r) rj z!Length of 'value' does not match.)r r zC"value" parameter must be a scalar or dict, but you passed a "list"zD"value" parameter must be a scalar or dict, but you passed a "tuple")rg rh zP"value" parameter must be a scalar, dict or Series, but you passed a "DataFrame") r1 r6 r r �_valuesr3 r4 r5 r rP r )r rs r? �catr; Zmsg2r" r" r# � test_fillna_categorical_raises� s( z/TestSeriesFillNA.test_fillna_categorical_raisesr �float32r. � fill_typec C sL t tjdg|d�}t ddg|d�}|�|�}t ddg|d�}t�||� d S )Ng333333�?)r r g @)r r1 r6 r r r )r r r{ r? Z fill_valuesr> r! r" r" r# �test_fillna_float_casting� s z*TestSeriesFillNA.test_fillna_float_castingc C sD t tjdgtjd�}|�ddi�}t ddgtjd�}t�||� d S )Ng333333�?)r r r g �?)r r1 r6 rz r r r )r r? r> r! r" r" r# � test_fillna_f32_upcast_with_dict� s z1TestSeriesFillNA.test_fillna_f32_upcast_with_dictc C sH y|j dd� W n2 tk rB } zdt|�ks2t�W d d }~X Y nX d S )NZffil)r )r r5 �strrY )r r8 �instr" r" r# �test_fillna_invalid_method� s z+TestSeriesFillNA.test_fillna_invalid_methodc C sl t tj�ddd��}d}tjt|d�� |�ddg� W d Q R X d}tjt|d�� |�d � W d Q R X d S ) Ni�����d �2 zC"value" parameter must be a scalar or dict, but you passed a "list")r) r r zD"value" parameter must be a scalar or dict, but you passed a "tuple")r r )r r1 �random�randintr3 r4 rP r )r r? r; r" r" r# � test_fillna_listlike_invalid� s z-TestSeriesFillNA.test_fillna_listlike_invalidc C s^ t dddd g�}d}xDdD ]<}x6dD ].}tjt|d�� |jd||d� W d Q R X q$W qW d S ) Nr r r z`Cannot specify both 'value' and 'method'\.|Limit must be greater than 0|Limit must be an integer)���r g �?g @)r( r r r' N)r) )r, r )r r3 r4 r5 r )r r? r; r, r r" r" r# �$test_fillna_method_and_limit_invalid� s z5TestSeriesFillNA.test_fillna_method_and_limit_invalidc C s� t tdddd��}|�� }t|d< |�tdddtjd��}t� ||� t d d d �}t |j � d ��}|�|�}t |d ||d gt d�}t� ||� d S )NZ2020r �UTC)ZperiodsrD r i� r )�tzinfoz 2000-01-01z US/Pacific)rD zdateutil/US/Pacificr )r )r r r r r r r �utcr r r rw Z tz_convertr/ )r r? r! r> r9 rH r" r" r# �+test_fillna_datetime64_with_timezone_tzinfo� s z<TestSeriesFillNA.test_fillna_datetime64_with_timezone_tzinfoc C sf t dddtjgtd�}d}tjt|d�� |�dd d �}W d Q R X t ddddgtd�}t�||� d S )Nr r r )r zqIn a future version of pandas all arguments of Series.fillna except for the argument 'value' will be keyword-only)r) r ) r r1 r6 �floatr rU rV r r )r Zsrsr; r> r! r" r" r# � test_fillna_pos_args_deprecation s z1TestSeriesFillNA.test_fillna_pos_args_deprecationN)$�__name__� __module__� __qualname__r$ rB rI rJ rT rW r3 �markZ parametrizer[ r\ r^ rb rc re ro r1 r6 r rt r rv ry r� r ZALL_REAL_DTYPESr| r} r� r� r� r� r� r" r" r" r# r sR %I"c6 I 0 r c @ sd e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dS )� TestFillnaPadc C s� t tjdtjdtjgdddddg�}|jdd �}t tjddddg|j�}t�||� |jd d �}t ddddtjg|j�}t�||� d S ) Ng �?g @rf rg rh ri rj r' )r r )r r1 r6 r r& r r )r r? r r! r" r" r# �test_fillna_bug s $ z TestFillnaPad.test_fillna_bugc C sB t dddddgt�d�d�}tj|d< t�|�� |jd d �� d S ) Ng g �?g @g @g @r% )r& r r' )r )r r r0 r1 r2 r r' r )r r9 r" r" r# � test_ffill s zTestFillnaPad.test_ffillc C sT t dddg�}d}tjt|d�� |�d�}W d Q R X t dddg�}t�||� d S )Nr r r zPIn a future version of pandas all arguments of Series.ffill will be keyword-only)r) r )r r rU rV r' r )r r? r; r> r! r" r" r# � test_ffill_pos_args_deprecation s z-TestFillnaPad.test_ffill_pos_args_deprecationc C s2 t tdddtjd�dg�}|�� }t�||� d S )Ni� r )r� )r r �pytzr� r' r r )r r r> r" r" r# �,test_ffill_mixed_dtypes_without_missing_data, s z:TestFillnaPad.test_ffill_mixed_dtypes_without_missing_datac C sB t dddddgt�d�d�}tj|d< t�|�� |jd d �� d S ) Ng g �?g @g @g @r% )r& r r )r )r r r0 r1 r2 r r r )r r9 r" r" r# � test_bfill2 s zTestFillnaPad.test_bfillc C sT t dddg�}d}tjt|d�� |�d�}W d Q R X t dddg�}t�||� d S )Nr r r zPIn a future version of pandas all arguments of Series.bfill will be keyword-only)r) r )r r rU rV r r )r r? r; r> r! r" r" r# � test_bfill_pos_args_deprecation7 s z-TestFillnaPad.test_bfill_pos_args_deprecationc C s� t tjdtjdtjgdddddgtd�}|jd d d �}|d ksBt�t tjddddgdddddgtd�}t�|d d � |d d � � t�|d �s�tt�|d ���d S )Ng �?g @rf rg rh ri rj )r r T)r rk r r ) r r1 r6 r� r rY r r �isnan)r rl rn r! r" r" r# � test_pad_nanC s & " zTestFillnaPad.test_pad_nanc C s� t �d�}tt j�d�|d�}|d d� �|�}|jddd�}|d d� �|�jdd�}t j|dd �< t� ||� |d d � �|�}|jd dd�}|d d � �|�jd d�}t j|d d �< t� ||� d S ) N� )r& r r r% )r r, )r ������r r( r ) r1 �aranger r� �randn�reindexr r6 r r )r r& �sr> r! r" r" r# �test_series_fillna_limitQ s z&TestFillnaPad.test_series_fillna_limitc C s� t �d�}tt j�d�|d�}|d d� j|ddd�}|d d� �|�jdd�}t j|dd �< t� ||� |d d � j|d dd�}|d d � �|�jd d�}t j|d d �< t� ||� d S ) Nr� )r& r r r% )r r, )r r� r� r( r ) r1 r� r r� r� r� r r6 r r )r r& r� r> r! r" r" r# � test_series_pad_backfill_limitc s z,TestFillnaPad.test_series_pad_backfill_limitc C sH t tj�ddd��}|jddd�}|d ks.t�t�|jddd�|� d S )Ni����r� r� r' T)r rk F)r r1 r� r� r rY r r )r r? rn r" r" r# �test_fillna_ints s z TestFillnaPad.test_fillna_intc C s� t tttdddddddtjd�g�}|jdd �}t tdddddddtjd�tdddddddtjd�tdddddddtjd�g�}t�||� d S ) Ni� � � � � i�� )r� r )r )r r r r� r� r r r )r rs r r! r" r" r# �$test_datetime64tz_fillna_round_issuey s " z2TestFillnaPad.test_datetime64tz_fillna_round_issueN)r� r� r� r� r� r� r� r� r� r� r� r� r� r� r" r" r" r# r� s r� )r r r �numpyr1 r3 r� Zpandasr r r r r r r r r r Zpandas._testingZ_testingr r r� r" r" r" r# �<module> s 0 {
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка