U
|
®ý°dS7 ã @ sÔ d dl Zd dlZd dlmZmZmZmZmZm Z m
|
Z
|
m Z m Z m Z mZmZmZ d dlmZ dd Zdd Zdd Zd d
|
Zd d Zd d Zdd Zdd Zdd Z dd Z dd Z dd Z dd Z ej! "d d d g¡ej! "d d!d e d g¡d"d# Z#d$d% Z$d&d' Z%ej! "d(e
|
d)d*e
|
d+d*ge
|
d)d*e
|
d+d*d,gd-d-gd-d-ej&gfe
|
d)d*e
|
d+d*ge
|
d)d*e
|
d+d*gd-d-gd-d-gfg¡d.d/ Z'd0d1 Z(d2d3 Z)d4d5 Z*ej! "d6d7gd8ggg g gg¡d9d: Z+d;d< Z,d=d> Z-dS )?é N) ÚNAÚ CategoricalÚ Float64DtypeÚIndexÚ
|
MultiIndexÚNaTÚPeriodÚ PeriodIndexÚSeriesÚ TimedeltaÚ TimestampÚ
|
date_rangeÚisnac C sH | |j¡}zt |j|j¡s"tW n tk
|
r8 Y nX |j |j¡sLt|j |j¡s^t|jdd
}| |¡}| ¡ D ]\}}||| ks~tq~| jdd
}| |¡}| ¡ D ]\}}|| | ks¸tq¸| |¡} t | ¡
|
¡ sît| jd d d
}
|
| |
|
¡} | ¡ D ] \}}|| | kstq| ¡ } | | k sDtd S )Né
|
é é ) ÚreindexÚindexÚnpZmay_share_memoryÚAssertionErrorÚAttributeErrorÚis_Z identicalÚitemsÚisnanÚall) Údatetime_seriesZ string_seriesÚidentityZsubIndexZ subSeriesÚidxÚvalZ subIndex2ZsubTSZ stuffSeriesZnonContigIndexZ subNonContigÚresult© r ú_d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/series/methods/test_reindex.pyÚ test_reindex s.
|
|
|
|
r" c C s t ddddgddtjdgd} tjdtjddtjgdd ddddg }}t | |¡| j| ¡ | j d
|
¡| _tj| |¡| j| d d d S ) Nr é é é é é é ©r r ÚobjectF)Zcheck_index_type) r
|
r ÚnanÚtmÚassert_series_equalr Zilocr Úastype)ÚtsÚiÚjr r r! Útest_reindex_nan= s
|
(r2 c C sp t dddd} t| }| td¡}t |jt d¡¡s<t| ¡ }|dd
¡ sXt|d d
|
¡ rltd S )Nz1/1/2000 00:00:00r Z10s©ÚperiodsÚfreqé zM8[ns]éûÿÿÿ) r r
|
r Úranger Z
|
issubdtypeÚdtyper r r Úany)ÚrngZseriesr Úmaskr r r! Útest_reindex_series_add_natI s r= c C s t ddd} ttj d¡| d}| t|jdd
¡}|dd
}|j d ¡|_t
|
||¡ |t|jdd
}t
|
||¡ d S )Nz1/1/2000r ©r4 r) r$ r ) r r
|
r ÚrandomZrandnr Úlistr Ú
|
_with_freqr, r- )r; r/ r Úexpectedr r r! Útest_reindex_with_datetimesU s rC c C s t g d}|j| jdd |j| jdd}| t| j¡}| j d ¡| _t | |¡ | d d d
}d}tjt |d |j| jdd W 5 Q R X d S )Nr) Úpad©Úmethodr zVInvalid fill method\. Expecting pad \(ffill\), backfill \(bfill\) or nearest\. Got foo©ÚmatchÚfoo)
|
r
|
r r r@ rA r, r- ÚpytestÚraisesÚ
|
ValueError)r ÚemptyÚ reindexedr/ Úmsgr r r! Útest_reindex_cornerb s
|
ÿrP c C s¶ t t d¡dd} | d d d
}|j| jdd}|j| jdd}t ||¡ t ddddd d d
|
d
|
d d g
|
}t ||¡ t d dd d dgdddddgd} ddddg}t d d d d g|d}| |¡ ¡ }t || d¡¡ | |¡jdd}t ||¡ t d dd dg|d}| j|dd}t ||¡ t ddddgt d d} d }| t |¡ ¡ }t dddgt |d}t ||¡ t dt
|
ddd} | d ¡j d d}t dt
|
ddd}t ||¡ d S ) Nr Úint64©r9 r rD rE Úffillr r' é r( r& r# r$ ÚaÚbÚcÚdÚer) ÚgÚfÚfloat64Zinfer)ZdowncastTFZabcdZagcZbfill) r
|
r Úaranger r r, r- rS r. r@ r8 ÚshiftZfillna)ÚsÚs2rN Z
|
reindexed2rB Ú new_indexr r r r! Útest_reindex_pady s4 rb c C s¼ t tjddd} ddddg}| j|dd }t t |¡ d¡|}t ||¡ | j|dd
|
d }t d d tjdg|}t ||¡ | j|dddddgd }t d tjtjdg|}t ||¡ d S )Nr rQ rR g¹?gÍÌÌÌÌÌì?g ø?ç @ZnearestrE gÉ?)rF Z tolerancer r& r g333333Ó?g{®Gáz?gÙ?r# ) r
|
r r] r Zaroundr. r, r- r+ )r_ Útargetr rB r r r! Útest_reindex_nearest¢ s re c C sp | d d d
}t tjt|td|jd}| | j¡}|jtjksFt | |jd d d
¡}|jtj
|
kslt d S ©Nr rR r) ) r
|
r ÚzerosÚlenÚintr r r9 Zfloat_r Úint_)r r/ Zint_tsZ reindexed_intr r r! Útest_reindex_int² s rk c C sp | d d d
}t tjt|td|jd}| | j¡}|jtjksFt | |jd d d
¡}|jtj
|
kslt d S rf ) r
|
r rg rh Úboolr r r9 Zobject_r Zbool_)r r/ Úbool_tsZreindexed_boolr r r! Útest_reindex_boolÁ s rn c C sT | dd
}t tjt|td|jd}|j| jdd}t|d d
¡ sPt d S )Nr$ rR r) rD rE )
|
r
|
r rg rh rl r r r r r )r r/ rm Z filled_boolr r r! Útest_reindex_bool_padÑ s ro c C sÞ t ddd} tdddgdd}| | ¡}tttjtjtjgdddgd }| |_t ||¡ ttddgdddgd }d
|
d g|_| d
|
d g¡}t ||¡ ttdtjgdddgd }d dg|_| d dg¡}t ||¡ d S ) NZ20000101r# r> rU rV rW ÚcategoryrR )ÚvaluesÚ
|
categoriesr& r ) r r
|
r r r r+ r r, r- )r r_ r rB r r r! Útest_reindex_categoricalÙ s
|
ÿ
|
|
rs c C sf t dddgdddgd} dddg}d}t}| |¡ |¡ |¡}| |¡ |¡ |¡}t ||¡ d S )Nr& r r# r r) rp )r
|
Ústrr r. r, r- )Úserra Z
|
temp_dtypeZ new_dtyper rB r r r! Ú%test_reindex_astype_order_consistencyñ s
|
rv c C s t dddg} | dddg¡}t ddtjgdddgd}t ||¡ | jdddgdd }t dddgdddgd}t ||¡ t dddg}| dddg¡}t ddtjgdddgd}t ||¡ |jdddgdd }t dddgdddgd}t|jjtj st
|
t ||¡ t dddgt d
|
}| dddg¡}t ddtjgdddgt d }t ||¡ |jdddgd d }t ddd gdddgt d }t ||¡ t d dd g}| dddg¡}t dd tjgdddgt d }t ||¡ |jdddgdd }t dd dgdddgd}t ||¡ d S )Nç ð?rc g @r& r r# r) r ©Ú
|
fill_valuerR ©r r9 rI TF) r
|
r r r+ r, r- Ú
|
issubclassr9 ÚtypeÚintegerr r* )Zfloatsr rB ÚintsZobjectsZboolsr r r! Útest_reindex_fill_valueü s: r r9 zdatetime64[ns]útimedelta64[ns]ry Ústringc C sp |rt d¡ | dkr*|tdkr*td}ttg| d}|jddg|d}td |gddgtd}t ||¡ d S )Nz3Array manager does not promote dtype, hence we failr r rR r& rx rz )
|
rJ Úskipr r r
|
r r r* r, r- )r9 ry Zusing_array_managerru r rB r r r! Ú+test_reindex_fill_value_datetimelike_upcast/ s
|
r c C s\ t dddd} t dddd}ttd| d }d
|
}tjt|d |j|d d W 5 Q R X d S )NZ20131101zAmerica/Chicagor% )Útzr4 Z20131103r ÚHr3 r) zLCannot compare dtypes datetime64\[ns, America/Chicago\] and datetime64\[ns\]rG rS rE )r r
|
r8 rJ rK Ú TypeErrorr )r Znewidxr_ rO r r r! Ú/test_reindex_datetimeindexes_tz_naive_and_awareA s ÿr c C s6 t dd ddg¡} t tgd dd}t | |¡ d S )Nzdatetime64[ns, UTC]rR r r& r )r
|
r r r, Z assert_equal)r rB r r r! Ú"test_reindex_empty_series_tz_dtypeN s r z+p_values, o_values, values, expected_valuesZ2019Q1zQ-DECZ2019Q2ZAllrw c C sB t | }t|}t||d}| |¡}t||d}t ||¡ d S )Nr) )r r r
|
r r, r- ) Zp_valuesZo_valuesrq Zexpected_valuesZ period_indexZ object_indexru r rB r r r! Ú$test_reindex_periodindex_with_objectU s
|
r c C s> t ddg} d}tjt|d | ddgd¡ W 5 Q R X d S )Nr& r zCreindex\(\) takes from 1 to 2 positional arguments but 3 were givenrG r# F©r
|
rJ rK r r ©ru rO r r r! Útest_reindex_too_many_argsq s r c C sD t ddg} d}tjt|d | jddgddgd W 5 Q R X d S )Nr& r z4reindex\(\) got multiple values for argument 'index'rG r# r' r) r r r r r! Útest_reindex_double_indexy s r c C s@ t ddg} | jddgd}t ddgddgd}t ||¡ d S )Nr& r r r) )r
|
r r, r- )ru r rB r r r! Útest_reindex_no_posargs s r rq rU Úxc C sh t tt| d t | ¡dd}|jt dg¡dd}t tdg| d gg g gddd}t ||¡ d S )Nr r* rz rV )Úlevelr& )ÚlevelsÚcodes)
|
r
|
r8 rh r Z from_arraysr r Úarrayr, r- )rq ru r rB r r r! Ú test_reindex_empty_with_level s ÿ ÿr c C sN t ddddgdd} d}tjt|d | jddddd gd
|
d W 5 Q R X d S ) Nr& r r# rp rR z:Cannot setitem on a Categorical with a new category \(-1\)rG r' r$ éÿÿÿÿrx r r r r r! Ú test_reindex_missing_category s r c C s t dtgt d} | td¡}|jj}t dtj tj g¡}t
|
||¡ t
|
d ¡4 t |¡}t dtj tj gt d}t
|
||¡ W 5 Q R X d S )Nrw rR r# r& r )r
|
r r r r8 rq Ú_datar r ÚNaNr, Zassert_numpy_array_equalZassert_produces_warningÚlogr- )r_ Z s_reindexr rB Z
|
result_logZ expected_logr r r! Ú#test_reindexing_with_float64_NA_log s
|
r ).Únumpyr rJ Zpandasr r r r r r r r r
|
r r r r Zpandas._testingZ_testingr, r" r2 r= rC rP rb re rk rn ro rs rv r ÚmarkZ parametrizer r r r+ r r r r r r r r r r r! Ú<module> sV < ' ) 3
|
üüùþ
|
|
|
