U
|
ý°d ã @ s* d dl Zd dlZd dlZd dlmZmZmZ d dlm Z
|
ej dddg¡dd Z ej dddg¡dd Zd
|
d Zej d ddd d gfdd d dgfg¡dd Zdd Zej dddg¡ej dddg¡ej dddg¡dd Zej ddejfdg¡ej dd d d d g¡d d! Zd"d# ZdS )$é N)Ú DataFrameÚIndexÚ
|
date_rangeÚfuncÚffillÚbfillc C sx t dddgdtjtjgdddggtdddgd d
|
d }| dg¡ddg }t|| j}tddgd d
|
}t ||¡ d S ) Né ç ð?g ð¿g @g ÀÚtypeÚaÚbÚidx©Úname©Úcolumns) r ÚnpÚnanr ÚgroupbyÚgetattrr ÚtmZassert_index_equal)r ÚdfÚ
|
df_groupedÚresultÚexpected© r úXd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/groupby/test_missing.pyÚ.test_groupby_column_index_name_lost_fill_funcs s þr c C s t dddgdddgd}t ddtjdgi}tj||gddjdgd}t dd gdtjgdd
|
ggddgd }t|| }t ||¡ d S ) Nr é é )Úfield1Úfield2r )Úaxisr! )Úbyr g @r ) r r r ÚpdÚconcatr r r Úassert_frame_equal)r Zdf1Zdf2r r r r r r Ú(test_groupby_fill_duplicate_column_names s ÿ r' c C sB t ddgddgd} tjtdd | d¡ ¡ W 5 Q R X d S )Nr é ©r r zMust specify a fill)Úmatchr )r ÚpytestZraisesÚ
|
ValueErrorr Úfillna)r r r r Ú test_ffill_missing_arguments' s r. zmethod, expectedr c C s` t tjd dd gdddddgd}| d¡}|j| d}t dtj|ddi}t ||¡ d S )Nr Ústring)Zdtyper r) r ©Úmethod)r r$ Úarrayr r- r r& )r1 r r Úgrpr r r r Ú test_fillna_with_string_dtype. s
|
$
|
r4 c C sÎ t tj ddgtddg¡tddgddd } tjd
|
tjtjd tjtjd tjtjd tjg | d< tjdtjtjdtjtjdtjtjdtjg | d< | jdddj dd}| j
|
jdd
|
dj ddj
|
}t ||¡ d S )NZvalue1Zvalue2z
|
2014-01-01z
|
2014-01-06Ú1Ú2Úidr )Úindexr r é r( é r é! r é, r )Úlevelr" r r0 ) r r$ Z
|
MultiIndexZ from_productr r r r r r- ÚTr r& )r r r r r r Útest_fill_consistency: sF ÿüôôr? r1 ÚdropnaTFÚ has_nan_groupc C s. t ddg}ddddddg}| |¡jdd}|r8tjnd}t d gd
|
|gd
|
¡|d < |jd | d }t||d d }ddddddgddddddgddddddgddddddgddddddgddddddgddddddgddddddgd} | || |f¡}| |¡jdd}
|
|
|
j
|
t ¡|
|
_
|
t ||
|
¡ d S )N)r g¹?)r( gÉ?éÿÿÿÿr r T)Zdropr r r Z group_col)r# r@ )Úlimit))r TT)r TF)r FT)r FF)r TT)r TF)r FT)r FF)r ZreindexZ reset_indexr r r$ ZSeriesr r Úgetr ZastypeÚobjectr r& ) r@ r1 rA Zdf_without_nan_rowsZridxr Zgroup_bZgroupedr Z expected_rowsr r r r Ú test_ffill_handles_nan_groupsf s( ø rF zmin_count, valuer( )rB r ÚfirstÚlastÚmaxÚminc C sp t dgd dtjtjgtjgd d}t| d¡| |d}t |gtjgdtdgddd}t ||¡ d S ) Nr r ©r r Úcr )Ú min_count)r rL r )r8 )r r r r r r r r& )r rM Úvaluer r r r r r Útest_min_count s ("rO c C sb t ddtjgdddgdddgd} | d d
|
g¡}|j}t d g¡t dg¡d }||ks^td S ) Nr r( r r é é é rK r r r ))r r( )r r )r r r r Úindicesr2 ÚAssertionError)r Úgr r r r r Útest_indices_with_missing s
|
$rV )Únumpyr r+ Zpandasr$ r r r Zpandas._testingZ_testingr ÚmarkZ parametrizer r' r. r4 r? rF r rO rV r r r r Ú<module> s.
|
|
ÿ
|
,!
|
