U
|
ý°dØj ã @ s¦ d dl Zd dlZd dlm mZ d dlmZm Z m
|
Z
|
m Z m Z m Z mZmZmZmZ d dlmZ d dlmZ G dd dZdd Zdd Zej d
|
d d g¡d d Z dS )é N)
|
Ú CategoricalÚ DataFrameÚ DatetimeIndexÚNaTÚ PeriodIndexÚSeriesÚTimedeltaIndexÚ TimestampÚ
|
date_rangeÚ to_datetime)Ú_check_mixed_floatc @ sØ e Zd Zdd Zejdd Zdd Zdd Zd d
|
Z d d Z
|
d d Z dd Z dd Z dd Zdd Zdd Zdd Zdd Zd d Zd d Zej d!d"d"d#gd"d"gg¡d$d% Zd&d' Zej d(d)d*ejd+g¡d,d- Z d.d/ Z d0d1 Z d2d3 Z d4d5 Z d6d7 Z!d8d9 Z"ej d:e#e$g¡d;d< Z%ej d:e#e$g¡d=d> Z&d?d@ Z'dAdB Z(dCdD Z)dEdF Z*dGdH Z+dIdJ Z,dKdL Z-dMdN Z.dOdP Z/dQdR Z0dSdT Z1dUdV Z2ej3ej d(dWdWdWdXg¡dYdZ Z4d[d\ Z5d]d^ Z6d_S )`Ú
|
TestFillNAc C s& t tjgd ttdtgtjddgd}dddg|_|d d
}|jddidd t d gd dtddgdddgd}dddg|_t ||¡ |sÂt |j
|
d d
d
|
f |j
|
d d
d
|
f ¡sÂt t |j
|
d d
df |j
|
d d
df ¡rît |s"t |j
|
d d
df |j
|
d d
df ¡s"t d S ) Né é Úfooé )ÚAÚBÚCr T©Úinplaceç @r ) r ÚnpÚnanr r ÚcolumnsÚfillnaÚtmÚassert_frame_equalÚ shares_memoryÚilocÚAssertionError)ÚselfÚusing_copy_on_writeÚdfÚorigÚexpected© r& ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_fillna.pyÚ*test_fillna_dict_inplace_nonunique_columns s $ÿ ÿ ,,z5TestFillNA.test_fillna_dict_inplace_nonunique_columnsc C s t dt j¡}t|dd}|d jddd |rPt |d d
df ¡ ¡ sltn |d d
df dk ¡ sltt|j j
|
dkstt |j |¡std S ) N)é( é2 F)Úcopyr éÿÿÿÿTr r ) r Úfullr r r ÚisnanÚallr ÚlenZ_mgrZarraysr Úvalues)r! r" Úarrr# r& r& r' Útest_fillna_on_column_view. s z%TestFillNA.test_fillna_on_column_viewc 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 ¡ ¡ st|j|jdd
df |j|jd df k ¡ sÌtd}tj t
|
|d | ¡ W 5 Q R X d }tj t
|
|d |jdd
|
d W 5 Q R X d S ) Né r éûÿÿÿr Úpad©Úmethodz'Must specify a fill 'value' or 'method'©Úmatchz(Cannot specify both 'value' and 'method'Úffill) r r ÚlocÚindexr r/ r r. ÚpytestÚraisesÚ
|
ValueError)r! Údatetime_frameÚtfZ zero_filledZpaddedÚmsgr& r& r' Útest_fillna_datetime? s
|
$ &*ÿ
|
z TestFillNA.test_fillna_datetimec C sT |}t j|j|jdd
df< t j|j|jdd
df< |jdd |jdd d S )
|
Nr4 é r éöÿÿÿr r ©Úvaluer6 r7 )r r r< r= r )r! Úfloat_string_frameÚmfr& r& r' Útest_fillna_mixed_typeT 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 r ÚD©r rF r rG r ©Údtyper6 r7 )Úreindexr r r< r= r r )r! Zmixed_float_framerJ Úresultr& r& r' Útest_fillna_mixed_float\ s z"TestFillNA.test_fillna_mixed_floatc C s8 t dgd}dD ]"}|jj|dd |jj|d qd S )NÚxrM )r6 ÚbackfillT©r8 r r7 )r rS r )r! r# Úmr& r& r' Útest_fillna_emptyf s z TestFillNA.test_fillna_emptyc C s t ddtjdgddtjdgddtjdgg}| ddi¡}t ddddgddddgddddgg}t ||¡ |jddidd}t ||¡ |d kstd S )NÚaÚbÚcr r Tr )r r r r r r r )r! r# rQ r% Z return_valuer& r& r' Útest_fillna_different_dtypem 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 r4 iç )r r )r r r ©Úlimit)
|
r r ÚrandomÚrandnr r r+ r r r ©r! r# r% rQ r& r& r' Útest_fillna_limit_and_value} s
|
|
z&TestFillNA.test_fillna_limit_and_valuec C sl t ttdgtdtgd}| ¡ }|d |j|jd df ¡|d< |jd|d id}t ||¡ d S )Nz2014-1-1z2013-1-1)ÚDateÚDate2rd r re rG ) r r r r+ r r< r= r r rb r& r& r' Útest_fillna_datelike s
|
|
þÿ"z TestFillNA.test_fillna_datelikec C s t dtdtgi}t dtdtdgi}t |jdd|¡ t dttdgi}t dtdtdgi}t |jdd|¡ d S )Nr z2012-11-11 00:00:00+01:00r6 r7 Úbfill)r r r r r r )r! r# Úexpr& r& r' Útest_fillna_tzaware s$ þÿÿþÿÿz TestFillNA.test_fillna_tzawarec C s` t tddddddtjtjgd}|jdd }t tddddd
|
d d d gd}t ||¡ d S ) NÚ20130101é z
|
US/Eastern)ÚperiodsÚtzr r ©r r r6 r7 ç ð?r )r r
|
r r r r r ©r! r# rQ 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} |jdddd } t | |¡ d
|
}tjt |d |jd d dd W 5 Q R X |jdd} t | |¡ |j
|
dgd} t | | ¡ |
|
¡ } t | | ¡ t tjdtjgddgd}t|dddgd}t dddgddgd}t|dddgd}| d¡} t | |¡ d S )Nr r r )Ú
|
categoriesrX rY Úd©ÚcatsÚvalsrG z3Cannot setitem on a Categorical with a new categoryr9 rk rZ r6 r7 ru )Zsubset) r r r r r r r r> r? Ú TypeErrorZdropna)r! Úcatrv r# Zcat2Zvals2Z df_exp_fillZcat3Zvals3Zdf_exp_drop_catsZcat4Zvals4Zdf_exp_drop_allÚresrC rZ 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|ddd}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 rt r ÚcategoryrN z2011-01-01 09:00z2016-01-01 23:45rX rG z2011-01ÚM)Úfreqz1 daysz2 days)r r r r r r r r ru Úassert_series_equalrv r r r r )
|
r! rx Úvalr# Zmedianry Zv_exprz rQ Ú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 )NrX ro r Úinfer©Zdowncastr ©r r r r r r rp r& r& r' Útest_fillna_downcast s z TestFillNA.test_fillna_downcastc C s0 |dddgdd}|j ddd}t ||¡ d S ) Nr r r ÚobjectrN Ú Fr )r r Ú assert_equal)r! Úframe_or_seriesÚobjrQ r& r& r' Útest_fillna_downcast_false s z%TestFillNA.test_fillna_downcast_falsec C s |dddgt jd}|jdt t j¡d}| t j¡}t ||¡ | t j¡}|jddd}|}t ||¡ |jdt t j¡d}t ||¡ d S )Nr r r rN r r r ) r Úint64r rO Úint32Úastyper r Úfloat64) r! r r ry r% Úobj2Zres2Ú expected2Zres3r& r& r' Útest_fillna_downcast_noop% s z$TestFillNA.test_fillna_downcast_noopr r r c C sX t tjddg|d}t d ¡ | ddi¡}W 5 Q R X | ¡ }d|d< t ||¡ d S )Nr r ©r= r r ç )r r r r Úassert_produces_warningr r+ r )r! r r# rQ r% r& r& r' Ú-test_fillna_dictlike_value_duplicate_colnames8 s z8TestFillNA.test_fillna_dictlike_value_duplicate_colnamesc C sÊ t dddgdddddgd }|j}tt d
|
¡gd dddddgd }t ||¡ | d¡}t ddddgdddddgd }t ||¡ t t dddgd d }| d¡}t dt dddgd }t ||¡ d S )Nr r r r r r rk r4 r r ©r= r )r= r rO r )
|
r Zdtypesr r rO r r r r Úrangerp r& r& r' Ú test_fillna_dtype_conversionC s "
|
|
z'TestFillNA.test_fillna_dtype_conversionr r r ro c C s@ t dtjgddgd}| tj|¡}| |¡}t ||¡ d S )Nr ro r rn )r r r Úreplacer r r )r! r r# r% rQ r& r& r' Ú*test_fillna_dtype_conversion_equiv_replaceT s
|
z5TestFillNA.test_fillna_dtype_conversion_equiv_replacec C s t ddtjgtdddddd gdd d gd
|
td ddd }| d ¡}t ddd gtdddddd gdd d gd
|
td ddd }t ||¡ t ddtjgtdtdtgddd gdd d gd
|
td ddd }| d ¡}t ddd gtdtdd gddd gdd d gd
|
td ddd }t ||¡ d S )Nr, éþÿÿÿrj r )rl r ÚbarZfoo2Zbar2©r r r rL Z20130110r ú?z
|
2013-01-01z
|
2013-01-02) r r r r
|
r r r r r rp r& r& r' Ú test_fillna_datetime_columns[ sH
|
|
ü
|
ù
|
|
ü
|
ù
|
ü
|
ù
|
ü
|
ù z'TestFillNA.test_fillna_datetime_columnsc C sP t j|j|jd d
df< t j|j|jdd
df< t | ¡ |jdd¡ d S )Nr4 r r5 r; r7 )r r r< r= r r r; r ©r! rA r& r& r' Ú
|
test_ffill s
|
ÿzTestFillNA.test_ffillc C sP t j|j|jd d
df< t j|j|jdd
df< t | ¡ |jdd¡ d S )Nr4 r r5 rg r7 )r r r< r= r r rg r r¢ r& r& r' Ú
|
test_bfill s
|
ÿzTestFillNA.test_bfillc 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 © Nr\ rk r r r6 r4 )r8 r_ r7 éýÿÿÿr rT r © r Zaranger r` ra rP r r r r r ©r! r= r# rQ r% r& r& r' Ú test_frame_pad_backfill_limit 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 r¥ r§ r¨ r& r& r' Útest_frame_fillna_limit§ s
|
z"TestFillNA.test_frame_fillna_limitc C s( t tj dd¡ t¡}| tj¡ d S )Nr\ rk )r r r` ra r Úintr r )r! r# 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 5 Q R X d S )Nr\ rk z Limit must be greater than 0r9 r r5 r^ © r r r` ra r r> r? r@ r ©r! r r# rC r& r& r' Útest_fillna_positive_limitÁ s z%TestFillNA.test_fillna_positive_limitc C sH t tj dd¡ |¡}d}tjt|d |jddd W 5 Q R X d S )Nr\ rk zLimit must be an integerr9 r g à?r^ r® r¯ r& r& r' Útest_fillna_integer_limitÉ s z$TestFillNA.test_fillna_integer_limitc C sè t tj dd¡}tj|jd d
df< tj|jdd
df< |jdd}||k sRt|jddd t ||¡ |jddidd }|d ksttj|jd d
df< tj|jdd
df< |jd
|
d }||k sÊt|jd
|
dd t ||¡ d S ) Nr\ rk r éüÿÿÿr r rG T©rH r r; r7 rU )
|
r r r` ra r r< r r r r ©r! r# 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 5 Q R X d S )Nr r r rk ©rX rY rZ r r4 )rX rY rX rY r] )rX rY rs zcolumn by columnr9 )Úaxis) r r r r r+ r r ÚmaxÚto_dictr> r? ÚNotImplementedErrorrp r& r& r' Útest_fillna_dict_seriesç s" ýÿ z"TestFillNA.test_fillna_dict_seriesc C sÎ t tjddtjtjgdddtjtjgtjddddgdtdd}t tjdd d
|
d gd d dddgdgd dtdd}| |¡}t tjddtjd gdddtjdgtjddddgdtdd}t ||¡ d S )Nr r r rk r¶ ZVWXYZr r\ rE é r) r* é< éF éP éZ r r4 )rX rY r ZVWXuZ)r r r Úlistr r r )r! r# Zdf2rQ r% r& r& r' Útest_fillna_dataframe s. ýú
|
ýú
|
ýú z TestFillNA.test_fillna_dataframec C s t j dd¡}t j|d d
d d d
f< t|}|jddd}|jjddj}t ||¡ | dd d
|
¡ |jddd}|
|
t ¡jddd}t ||¡ d S ) Nr\ r r; r )r8 r· r6 r7 é r r4 ) r r` ra r r r ÚTr r Úinsertr Úfloat)r! r2 r# rQ r% r& r& r' Útest_fillna_columns& s z TestFillNA.test_fillna_columnsc C s* t jtdd |jdd W 5 Q R X d S )NZffilr9 r7 )r> r? r@ r )r! Ú float_framer& r& r' Útest_fillna_invalid_method4 s z%TestFillNA.test_fillna_invalid_methodc C s d}t jt| d¡d | ddg¡ W 5 Q R X t jt| d¡d | d¡ W 5 Q R X d}t jt|d |jd d
d f |¡ W 5 Q R X d S )
|
NzA"value" parameter must be a scalar or dict, but you passed a "{}"rÁ r9 r r Útuple)r r zP"value" parameter must be a scalar, dict or Series, but you passed a "DataFrame"r )r> r? rw Úformatr r )r! rÈ rC r& r& r' Útest_fillna_invalid_value8 s ÿz$TestFillNA.test_fillna_invalid_valuec C s^ dd t dddD }tj dd¡}tt d||d}|jdd }|j ¡ |j ¡ ksZtd S )
|
Nc S s g | ]}d t | qS )zCOL.)Ústr)Ú.0Úir& r& r' Ú
|
<listcomp>I s z9TestFillNA.test_fillna_col_reordering.<locals>.<listcomp>r4 r r, rE )r= r Údatar; r7 ) r r r` Zrandr r r Útolistr )r! ÚcolsrÑ r# Úfilledr& r& r' Útest_fillna_col_reorderingH 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 ) Nr4 rE r rF r r rG rM )r r r< r= r r/ r rP )r! rÈ rI rJ rÔ Z empty_floatrQ r& r& r' Útest_fill_cornerO s $ zTestFillNA.test_fill_cornerc C sH t ddtjgi}|jddiddid}t dddgi}t ||¡ d S )NÚcol1r r r r r
rp r& r& r' Útest_fillna_downcast_dict] s z$TestFillNA.test_fillna_downcast_dictc C s2 t tjdtjdgddtjdgtjtjtjdgtjdtjdggtdd}|jdd dd
|
}|jdd dd
|
}t td dd d gd d ddtjdgtjd tjtjgtddddgd d d ddddgd}t td dd d gd d tddd dgd d d d tjd gtddddgd d d ddddgd}t ||¡ t ||¡ d S )Nr r r rk r r4 ÚABCDrM éd ©r· rH r_ r rN r r )r r r rÁ r r r r )r! r# rQ Zresult2r% r r& r& r' Ú"test_fillna_with_columns_and_limitd s8 üù ü
|
ù ü
|
ù
|
z-TestFillNA.test_fillna_with_columns_and_limitc C sF t tdd gtdd gd}| ¡ }|jtjdd t ||¡ d S )Nz
|
2018-05-30z
|
2018-09-30)Zdate1Zdate2Tr )r r r+ r r r r r r´ r& r& r' Ú test_fillna_datetime_inplace s
|
|
þÿz'TestFillNA.test_fillna_datetime_inplacec C s t tjdtjdgddtjdgtjtjtjdgtjdtjdggtdd}|jdd dd
|
}||k sdt|jdd dd d t ||¡ d S ) Nr r r rk r r4 rÙ rM rÚ rÛ T)r· rH r_ r )r r r rÁ r r r r r´ r& r& r' Ú0test_fillna_inplace_with_columns_limit_and_value s üù
|
z;TestFillNA.test_fillna_inplace_with_columns_limit_and_valuer, ©rS Úyc C s t tjdgtjdgd}| ¡ }|d d
}|j|dd t ddgddgd}t ||¡ |rpt ||¡ n t ||¡ d S )Nr rß Tr r, r ç ð¿)r r r r+ r r r )r! r r" r# Zdf_origZ result_viewr% r& r& r' Ú test_inplace_dict_update_view¦ s z(TestFillNA.test_inplace_dict_update_viewc C s² t ddtjdtjgdtjtjddgdtjdddgtjddtjd gd
|
}|jd ddd }t d dddgddtjdgdtjddgdd ddgddddggddddgd}t ||¡ d S ) Nr4 r r\ r] r é r r é )r× Úcol2Úcol3Úcol4r* )r_ r· g @g @g (@g I@r ro g $@r ç @g 2@r× rå ræ rç rM r
rp r& r& r' Ú)test_single_block_df_with_horizontal_axisµ s$ üÿ
|
|
|
û
|
ø
|
z4TestFillNA.test_single_block_df_with_horizontal_axisc C sº t tjddgddtjgddtjgd}t dddgdddgddtjgd}t | ddi¡|¡ t | ddd ¡|¡ t dddgddd
|
gddtjgd}t | ddd ¡|¡ d S ) Nr rè ro ))rS rX ©rS rY )rà rZ rá rS r, r )rS rê g À)rê rS )r r r r r r )r! Zpdfr% r& r& r' Ú"test_fillna_with_multi_index_frameÌ s*
|
|
|
ýÿ
|
ýÿ
|
ýÿz-TestFillNA.test_fillna_with_multi_index_frameN)7Ú__name__Ú
|
__module__Ú __qualname__r( ÚtdZ&skip_array_manager_not_yet_implementedr3 rD rK rR rW r[ rc rf ri rq r{ r r r r r> ÚmarkÚ parametrizer r r r r r¡ r£ r¤ r© rª r¬ r« rÆ r° r± rµ r» r rÇ rÉ rÌ rÕ rÖ rØ rÜ rÝ rÞ Z skip_array_manager_invalid_testrâ ré rë r& r& r& r' r sb
|
|
*$
|
|
|
,
|
|
#$ r c C sf t ddddgddddgddddggddd d
|
gd } | jddd }| d ¡}| ¡ ¡ ¡ d ksbtd S )Nr ro r r r rè Úi1Úi2Zi3Úf1rM r r )r Zpivotr ZisnaÚsumr )r# Z df_nonconsolrQ r& r& r' Ú!test_fillna_nonconsolidated_frameé s
|
|
|
ý
|
ú
|
rö c C sr t d d gd d ggddgd} t d¡ | jddddd W 5 Q R X t ddgddggddgd}t | |¡ d S )
|
Nr r rM Fr r rn Tr³ )r r r r r )r# r% r& r& r' Útest_fillna_nones_inplaceø s þ r÷ Úfuncr6 rT c C s8 t ddddgi}t t¡ t|| W 5 Q R X d S )NrX r r r )r r r Ú FutureWarningÚgetattr)rø r# r& r& r' Ú test_pad_backfill_deprecated s rû ) Únumpyr r> Z pandas.util._test_decoratorsÚutilZ_test_decoratorsrï Zpandasr r r r r r r r r
|
r Zpandas._testingZ_testingr Zpandas.tests.frame.commonr r rö r÷ rð rñ rû r& r& r& r' Ú<module> s 0 X
|
