U
|
®ý°d± ã @ s d dl m Z mZmZ d dlZd dlZd dlZd dlmZm Z m
|
Z
|
m Z m Z m Z mZmZmZmZ d dlmZ d dlmZ G dd dZG dd dZdS ) é )ÚdatetimeÚ timedeltaÚtimezoneN)
|
Ú CategoricalÚ DataFrameÚ DatetimeIndexÚNaTÚPeriodÚSeriesÚ TimedeltaÚ TimestampÚ
|
date_rangeÚisna)Ú period_arrayc @ sæ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd Zej dd d g¡d d Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zej d,d-d-d-d.d-d-gfd-d.d.d/d-d-d.d.d.gfd0d-id-d-d.ejejgfd-d.d1d-d-d.d.ejgfe d-d-ejd.ejejgfe d-d0gd2d-d-d.ejejgfe d-d.d1d-d-d.d.ejgfe d-d.gd3d4gd2d-ejd.d-d.gfg¡d5d6 Z ej d,e d-d.d7d8d9gd-d.d.d8d9gfe d.d8d-d8d-gd-d8d.d8d-gfe e d.d8d-d8d-gd.d7d8d9d-gd:d-d8d.d8d-gfg¡d;d< Z d=d> Z ej d?e!d@dAg¡ej dBe"j#¡ej dCdDdEg¡dFdG Z$dHdI Z%dJdK Z&dLdM Z'dNdO Z(dPdQ Z)ej dRdSdTdUdSgdTdSdTdTdSgdSdTgfdSdTejdSgdTdSdTdTdSgdSdTgfg¡dVdW Z*dUS )XÚTestSeriesFillNAc C s` t dddtjgdd}|jdd}|j|jd d}| ¡ }|jd |jd < t ||¡ t ||¡ t d
|
|i}|jdd}|j|jd d}t d
|
|i}t
|
||¡ t
|
||¡ t tjdddgdd}|jd d}|j|d d}| ¡ }|d |d< t ||¡ t ||¡ t d
|
|i}|jd d}|j|d d}t d
|
|i}t
|
||¡ t
|
||¡ d S ) Nr é é úM8[ns]©ÚdtypeÚpad©Úmethod©Úvalueé ÚAÚbfill) r
|
r Ú_valueÚfillnaÚvaluesÚcopyZilocÚtmÚassert_series_equalr Zassert_frame_equal)ÚselfÚseriesÚfilledZfilled2ÚexpectedZdf© r( ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/series/methods/test_fillna.pyÚtest_fillna_nat s4 z TestSeriesFillNA.test_fillna_natc C s0 d}t jt|d |jddd W 5 Q R X d S )Nz(Cannot specify both 'value' and 'method'©Úmatchr Úffill)r r )ÚpytestÚraisesÚ
|
ValueErrorr )r$ Údatetime_seriesÚmsgr( r( r) Útest_fillna_value_or_methodA s z,TestSeriesFillNA.test_fillna_value_or_methodc C sè 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 5 Q R X d S )Nç ç ð?ç @ç @ç @é ©Úindexr- r r Úbackfillg @r z'Must specify a fill 'value' or 'method'r+ ) r
|
r" Ú makeDateIndexr# r ÚnpÚNaNr; r. r/ r0 )r$ ÚtsÚexpr2 r( r( r) Ú test_fillnaF s
|
z TestSeriesFillNA.test_fillnac C s$ t tjg}t dg}| |¡}t dg}t ||¡ | i ¡}t ||¡ | t dtd¡}t ||¡ | |¡}t ||¡ | ddi¡}t ||¡ | ddi¡}t |t tjg¡ | ddd¡}t ||¡ | t ddd¡}t ||¡ | t dddddgd ¡}t ||¡ d S )
|
Nr r5 r( r r )r r é r9 r: )r
|
r> Únanr r" r# Úobject©r$ Ús1Ús2Úresultr' r( r( r) Útest_fillna_nonscalarZ s*
|
|
|
|
|
z&TestSeriesFillNA.test_fillna_nonscalarc C sX t dddgtd}t dtjdgtd}| |¡}t dddgtd}t ||¡ d S )Nr r r ÚabcZbacr6 )r
|
Úlistr> rD r r" r# rF r( r( r) Útest_fillna_alignsr s
|
|
z#TestSeriesFillNA.test_fillna_alignsc C s t tjdddgd}|jddd}t dtjtjgdddgd}t ||¡ |jddd}t ddtjgdddgd}t ||¡ d S )Nr r r r: iç )Úlimit©r
|
r> rD r r" r# ©r$ ÚserrI r' r( r( r) Útest_fillna_limity s z"TestSeriesFillNA.test_fillna_limitc C s^ dddg}|D ]J}t ddtjtjdgdd}| |¡}t dd||dgd d}t ||¡ qd S )
|
NÚ0z1.5z-0.3r r rC Úfloat64r rE rO )r$ ÚvalsÚvalrQ rI r' r( r( r) Ú test_fillna_dont_cast_strings s
|
|
z.TestSeriesFillNA.test_fillna_dont_cast_stringsc C sØ t tdtg}| tddd¡}t tdtdddgdd}t ||¡ | ddgtddd¡}t ||¡ | ddgtddd¡}t ||¡ | d ¡}t tdd g}t ||¡ | ¡ }d |d
|
< t ||¡ d S ) NÚ20130101ú
|
US/Eastern©Útzú
|
2013-01-01rE r TFZfoor )r
|
r r r r" r# Úwherer! )r$ rQ rI 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 )Nr5 r Úinfer©Zdowncastr rO rP r( r( r) Útest_fillna_downcast« s z%TestSeriesFillNA.test_fillna_downcastc C sN t d¡ t¡}t j|d< t|}|jddd}tt d¡t jd}t ||¡ |j
|
dd}tddddd gt jd}t ||¡ |j dd}tdddd d gt jd}t ||¡ d
|
|d< tddd
|
dd gt j d}|jddd}t ||¡ |j
|
dd}tddd
|
d
|
d gt j d}t ||¡ |j dd}tddd
|
d d gt j d}t ||¡ d S ) Nr9 r r` ra r r r r rC g @) r> ÚarangeÚastyperE rD r
|
r Úint64r" r# r- r rT )r$ ÚarrrQ Úresr' r( r( r) Ú-test_fillna_downcast_infer_objects_to_numeric¹ s,
|
z>TestSeriesFillNA.test_fillna_downcast_infer_objects_to_numericc C s\ t tdtdtdtdg}| ¡ }||}| tdd¡}t tdtdtdtdddg}||}t ||¡ | d¡}| t ¡ d¡}t ||¡ | tdd¡}t tddtdtdtdddg}||}t ||¡ | tddd¡}t tdddtdtdtdddg}||}t ||¡ | t
|
d ¡¡}t tddtdtdtdddg}||}t ||¡ | t ¡}t t tdtdtdddgd
|
d }||}t ||¡ t
|
j |d < ||}| ¡ }| tdd¡}t
|
j |d< ||}t ||¡ t
|
j |d < ||}| ¡ }| tdd¡}tddd|d < ||}t ||¡ d S ) NrX Ú20130102ú20130103 9:01:01r )Úsecondsr iÍ~ )Údaysrk i Ê;zm8[ns]r r )r
|
r Zdiffr r r r" Ú assert_equalrd rE r> Z timedelta64r rD r- r )r$ Úframe_or_seriesrQ ÚtdÚobjrI r' rg r( r( r) Útest_timedelta_fillnaÝ s üÿ
|
üÿ
|
|
üÿ
|
|
üÿ
|
üÿ
|
|
üù
|
|
|
z&TestSeriesFillNA.test_timedelta_fillnac C s t tdtdtdtdg}tj|d< | ¡ }t tdtdtdtdg}t ||¡ | ¡ }t tdtdtdtdg}t ||¡ d S )NrX ri rj r )r
|
r r> rD r- r" r# r rP r( r( r) Útest_datetime64_fillnaB s4 üÿ
|
üÿ üÿz'TestSeriesFillNA.test_datetime64_fillnac C s@ t ttdgdd}t dddgdd}|jdd}t ||¡ d S )Nz2013-08-05 15:30:00.000001r r r< r )r
|
r r r" r# )r$ rQ r' rI r( r( r) Ú test_datetime64_fillna_backfille s ýú z0TestSeriesFillNA.test_datetime64_fillna_backfillr[ rY z
|
Asia/Tokyoc C s t tdttdtg}t ddddg}| td¡}t tdtdtdtdg}t ||¡ t t||¡ | td|d¡}t tdtd|dtdtd|dg}t ||¡ t t||¡ | d¡}t tddtddgtd}t ||¡ t t||¡ | td|dtd d
|
¡}t tdtd|dtdtd g}t ||¡ t t||¡ | tdtd d
|
¡}t tdtdtdtd g}t ||¡ t t||¡ tdtdtg|d}t |}|j d | d ksît
|
t t||¡ | td¡}t td|dtdtd|dtdg}t ||¡ t t||¡ | td|d¡}tddddg|d}t |}t ||¡ t t||¡ | td|d ¡ ¡}tddddg|d}t |}t ||¡ t t||¡ | d¡}t td|ddtd|ddgtd}t ||¡ t t||¡ | td|dtd d
|
¡}t td|dtd|dtd|dtd g}t ||¡ t t||¡ | td|dtd |dd
|
¡}t td|dtd|dtd|dtd |dg}t ||¡ t t||¡ | td ¡}t td|dtdtd|dtdg}t ||¡ t t||¡ | td dd¡}t td|dtddd |¡td|dtddd |¡g}t ||¡ t t||¡ d S )Nz2011-01-01 10:00z2011-01-03 10:00FTz2011-01-02 10:00rZ ZAAAr z2011-01-04 10:00©r r zdatetime64[ns, ú]rX r\ ú
|
US/Pacific) r
|
r r r r" r# r rE r r ÚAssertionErrorZ to_pydatetimeÚ
|
tz_convert)r$ r[ rQ Znull_locrI r' Úidxr( r( r) Útest_datetime64_tz_fillnau sD üÿüÿ
|
|
üÿ
|
üù
|
þÿ
|
üÿ ÿüÿ
|
|
üÿ üù üù
|
|
|
üù
|
þÿ
|
|
|
üÿ
|
|
þÿ
|
|
|
|
üÿ
|
|
üÿ
|
|
üÿ z*TestSeriesFillNA.test_datetime64_tz_fillnac C st t tdtg}t tdtdg}t |jdd|¡ t ttdg}t tdtdg}t |jdd|¡ d S )Nz2012-11-11 00:00:00+01:00r r r )r
|
r r r" r# r )r$ rQ rA r( r( r) Ú test_fillna_dt64tz_with_method? s þÿþÿz/TestSeriesFillNA.test_fillna_dt64tz_with_methodc C sL t tjtdgddgd}| td¡}t tdddgd}t ||¡ d S )Nz1 daysr ÚBr: r )r
|
r> rD r r r r" r# rP r( r( r) Útest_fillna_pytimedeltaT s z(TestSeriesFillNA.test_fillna_pytimedeltac C sh t tdddtdddg}| tddd¡}t tdddtdddg}t ||¡ |jdksdtd S )Nz2011-01ÚM©Úfreqr z2012-01z Period[M])r
|
r r r" r# r rw )r$ rQ rg rA r( r( r) Útest_fillna_period\ s
|
z#TestSeriesFillNA.test_fillna_periodc C s t tdtdtdtdg}tj|d< ||}| td¡}t tdtdtdtdg}||}t ||¡ | t¡}|}t ||¡ d S )NrX ri rj r Z20130104)r
|
r r> rD r r" rm r )r$ rn rQ rp rI r' r( r( r) Útest_fillna_dt64_timestampe s, üÿ
|
üÿ
|
z+TestSeriesFillNA.test_fillna_dt64_timestampc C sV t tdttdg}t dd¡}| |¡}t tdtdtdg}t ||¡ d S )Nz
|
2010-01-01ú
|
2000-01-01z
|
1975-04-05Úms)r
|
r r r> Z
|
datetime64r r" r# )r$ rQ rV rI 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 ) Nr5 r7 ÚzÚaÚbÚcÚdr T)r Úinplacer )r
|
r> rD r! r rw r" r# )r$ ÚxÚyÚ return_valuer' r( r( r) Útest_fillna_numeric_inplace s $ z,TestSeriesFillNA.test_fillna_numeric_inplacezfill_value, expected_outputr r )r r rC r rt r: r rC c C sV dt jdt jt jg}tt|ddgd}tt|ddgd}| |¡}t ||¡ d S )Nr r ©Ú
|
categories©r> rD r
|
r r r" r# ©r$ Z
|
fill_valueZexpected_outputÚdatarQ rA rI r( r( r) Útest_fillna_categorical s
|
|
z(TestSeriesFillNA.test_fillna_categoricalr r Úer c C sb dt jdt jt jg}tt|dddddgd}tt|dddddgd}| |¡}t ||¡ d S )Nr r r r r 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 5 Q R X d}tjt |d | td¡ W 5 Q R X tjt|d | ddd¡ W 5 Q R X d }tjt|d | ddg¡ W 5 Q R X d
|
}tjt|d | d ¡ W 5 Q R X d }tjt|d | t
|
dgdgd¡ W 5 Q R X d S ) Nr r r z3Cannot setitem on a Categorical with a new categoryr+ r z!Length of 'value' does not match.rt úC"value" parameter must be a scalar or dict, but you passed a "list"úD"value" parameter must be a scalar or dict, but you passed a "tuple")r r zP"value" parameter must be a scalar, dict or Series, but you passed a "DataFrame") r> rD r
|
r Ú_valuesr. r/ Ú TypeErrorr r0 r )r$ r rQ Úcatr2 Zmsg2r( r( r) Ú test_fillna_categorical_raisesÇ s* ÿz/TestSeriesFillNA.test_fillna_categorical_raisesr Úfloat32rT Ú fill_typeÚscalarTFc
|
C sè t tjdg|d}t ddg|d}|r2|j d¡}| |¡}t ddg|d}t ||¡ t tjdg|d}| ¡ ¡ }|||< t ||¡ t tjdg|d}|j
|
||dd t ||¡ t tjdg|d}| | |¡} t | |¡ d S )Nç333333ó?r r r6 T)r ) r
|
r> rD r Útyper r" r# r Zto_numpyÚmaskr] )
|
r$ r r r rQ Z fill_valuesrI r' r£ rg r( r( r) Útest_fillna_float_castingæ s"
|
z*TestSeriesFillNA.test_fillna_float_castingc C sD t tjdgtjd}| ddi¡}t ddgtjd}t ||¡ d S )Nr¡ r r r r5 )r
|
r> rD r r r" r# rP r( r( r) Ú test_fillna_f32_upcast_with_dict s z1TestSeriesFillNA.test_fillna_f32_upcast_with_dictc
|
C sH z|j dd W n2 tk
|
rB } zdt|ks2tW 5 d }~X Y nX d S )NZffilr )r r0 Ústrrw )r$ r1 Úinstr( r( r) Útest_fillna_invalid_method s z+TestSeriesFillNA.test_fillna_invalid_methodc C sl t tj ddd¡}d}tjt|d | ddg¡ W 5 Q R X d}tjt|d | d ¡ W 5 Q R X d S )
|
Néÿÿÿéd é2 r r+ r r r )r r )r
|
r> ÚrandomÚrandintr. r/ r r )r$ rQ r2 r( r( r) Ú test_fillna_listlike_invalid s z-TestSeriesFillNA.test_fillna_listlike_invalidc C sb t dddd g}d dddg¡}dD ]8}d D ].}tjt|d
|
|jd||d W 5 Q R X q,q$d S ) Nr r r ú|z*Cannot specify both 'value' and 'method'\.z Limit must be greater than 0zLimit must be an integer)éÿÿÿÿr r5 r6 )r< r r r- Nr+ )rN r )r
|
Újoinr. r/ r0 r )r$ rQ r2 rN 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 ¡}|j jd ksrt| |¡}t |d | |j j¡|d g|j d}t ||¡ d S )NZ2020r ÚUTC)Úperiodsr[ r iä r ©Útzinfor rv rZ zdateutil/US/Pacificr~ r r )r
|
r r! r r r r Úutcr" r# r r rx r Úkindrw r[ )r$ rQ r' rI r@ r^ r( r( r) Ú+test_fillna_datetime64_with_timezone_tzinfo* s
|
þz<TestSeriesFillNA.test_fillna_datetime64_with_timezone_tzinfoz7input, input_fillna, expected_data, expected_categoriesr r| Nc
|
C sF t |}t| |¡}| |¡}| |¡}t ||d} t || ¡ d S )Nr )r r
|
r r" Zassert_categorical_equal)
|
r$ ÚinputZ input_fillnaZ expected_dataZexpected_categoriesr rQ r& rI r' r( r( r) Ú(test_fillna_categorical_accept_same_type> s
|
|
z9TestSeriesFillNA.test_fillna_categorical_accept_same_type)+Ú__name__Ú
|
__module__Ú __qualname__r* r3 rB rJ rM rR rW r_ rb rh rq rr rs r. ÚmarkÚ parametrizerz r{ r} r r r
r r> rD r
|
r r r r Úfloatr" ZALL_REAL_NUMPY_DTYPESr¤ r¥ r¨ r® r² r¹ r» r( r( r( r) r s %
|
$e#
|
J "øþ
|
ÿÿ úýþ
|
|
|
þþr c @ sv e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z e j dddg¡dd ZdS )Ú TestFillnaPadc C s t tjdtjdtjgdddddg}|jdd }t tjddddg|j}t ||¡ |jd
|
d }t ddddtjg|j}t ||¡ d S ) Nr5 r7 r r r r r r- r r )r
|
r> rD r r; r" r# )r$ rQ r& r' r( r( r) Útest_fillna_bugR s $ z TestFillnaPad.test_fillna_bugc C sB t dddddgt d¡d}tj|d< t | ¡ |jd d
|
¡ d S ) Nr4 r5 r6 r7 r8 r9 r: r r- r )r
|
r" r= r> r? r# r- r ©r$ r@ r( r( r) Ú
|
test_ffill\ s
|
zTestFillnaPad.test_ffillc C s2 t tdddtjddg}| ¡ }t ||¡ d S )Niß r rµ )r
|
r Úpytzr· r- r" r# )r$ r% rI r( r( r) Ú,test_ffill_mixed_dtypes_without_missing_dataa s z:TestFillnaPad.test_ffill_mixed_dtypes_without_missing_datac C sB t dddddgt d¡d}tj|d< t | ¡ |jd d
|
¡ d S ) Nr4 r5 r6 r7 r8 r9 r: r r r )r
|
r" r= r> r? r# r r rÄ r( r( r) Ú
|
test_bfillg s
|
zTestFillnaPad.test_bfillc C s¦ t tjdtjdtjgdddddgtd}|jd d
|
d }|d ksBtt tjddddgdddddgtd}t |d d
|d d
¡ t |d ¡s¢tt |d ¡d S )Nr5 r7 r r r r r r r T©r r r r ) r
|
r> rD rÁ r rw r" r# Úisnan)r$ r r r' r( r( r) Ú test_pad_nanl 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 r9 ©r rN r éýÿÿÿéþÿÿÿr r< r ©
|
r> rc r
|
r¬ ZrandnZreindexr rD r" r# ©r$ r; ÚsrI r' r( r( r) Útest_series_fillna_limitz 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 r9 rÍ r rÎ rÏ r< r rÐ rÑ r( r( r) Ú test_series_pad_backfill_limit s
|
z,TestFillnaPad.test_series_pad_backfill_limitc C sH t tj ddd¡}|jddd}|d ks.tt |jddd|¡ d S )Nr© rª r« r- TrÉ F)r
|
r> r¬ r r rw r" r# )r$ rQ r r( r( r) Útest_fillna_int s z TestFillnaPad.test_fillna_intc C s t tttdddddddtjdg}|jdd }t tdddddddtjdtdddddddtjdtdddddddtjdg}t ||¡ d S )
|
Nià é é é é i¡ rµ r r )r
|
r r rÆ r· r r" r# )r$ r r& r' r( r( r) Ú$test_datetime64tz_fillna_round_issue¢ s ÿ ýÿz2TestFillnaPad.test_datetime64tz_fillna_round_issuec C s~ t tjtdd ddd}t| d¡}t|d< ttdtdtdtd td
|
gdd }t|}|jd d }t
|
||¡ d S )NrÌ )Ú nanosecondsr9 Úns)r´ r r z 2262-04-11 23:47:16.854775797z 2262-04-11 23:47:16.854775798z 2262-04-11 23:47:16.854775800z 2262-04-11 23:47:16.854775801r r r ) r r Úmaxr r
|
Z to_periodr r r r" r# )r$ ZdtirQ rf r' r& r( r( r) Útest_fillna_parrµ s& ÿûø
|
z TestFillnaPad.test_fillna_parrÚfuncr r< c C s4 t dddg}t t¡ t|| W 5 Q R X d S )Nr r r )r
|
r" Zassert_produces_warningÚ FutureWarningÚgetattr)r$ rß rQ r( r( r) Ú test_pad_backfill_deprecatedÌ s z*TestFillnaPad.test_pad_backfill_deprecatedN)r¼ r½ r¾ rà rÅ rÇ rÈ rË rÓ rÔ rÕ rÚ rÞ r. r¿ rÀ râ r( r( r( r) r Q s
|
r )r r r Únumpyr> r. rÆ Zpandasr r r r r r
|
r r r r Zpandas._testingZ_testingr" Zpandas.core.arraysr r r r( r( r( r) Ú<module> s 0 <
|
