U
|
ý°dA ã @ s® d dl Zd dlZd dlZd dlmZmZ d dlmZ dd Z
|
dd Z dd Z d d
|
Z d d Zd d Zdd Zej ddgdggg g gg¡dd Zdd Zdd Zdd ZdS ) é N)ÚIndexÚ
|
MultiIndexc C s´ | t| d d
¡\}}t|ts(t|jddgks:tdd |jD ddgksVt| t| ¡\}}t|tsvt|d kst|jddgkstdd |jD ddgks°td S )Né ÚfirstÚsecondc S s g | ]
|
}|j qS © ©Úname©Ú.0Úlevelr r ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/multi/test_reindex.pyÚ
|
<listcomp> s z test_reindex.<locals>.<listcomp>c S s g | ]
|
}|j qS r r r
|
r r r r s )ÚreindexÚlistÚ
|
isinstancer ÚAssertionErrorÚnamesÚlevels)ÚidxÚresultÚindexerr r r Ú test_reindex s r c
|
C sÞ t dg}| j|dd\}}|j| dd\}}| j|ddd}| j|ddd}| |¡s\tt ddd g¡}tj||d
|
d | |¡stt dd dd dd g¡} tj|| d
|
d t j
|
t d d | j| ddd W 5 Q R X d S )NZoner ©r Úright)r ÚhowÚleftr é r F)Z check_dtypeéÿÿÿÿzFill method not supported©ÚmatchÚpad)Úmethodr ) r r ÚjoinÚequalsr ÚnpÚarrayÚtmÚassert_numpy_array_equalÚpytestÚraisesÚ TypeError)
|
r ÚindexÚtargetr Ztarget2Zindexer2Z exp_indexZ
|
exp_index2Z exp_indexerZ exp_indexer2r r r Útest_reindex_level s
|
r. c C sÄ | ¡ } | ¡ }d d g | _|_t ddgddgg¡}| g ¡d jd d gksRt| t g ¡¡d jd d gkstt| | ¡ ¡d jd d gkst| |j ¡d jd d gks²t| | ¡ ¡d jd d gksÒt| |j ¡d jd d gksðtddg| _| g ¡d jddgkst| t g ¡¡d jddgks<t| | ¡ ¡d jddgks^t| |j ¡d jddgks~t| | ¡ ¡d jddgks t| |j ¡d jddgksÀtd S )Né r é r r ÚfooÚbar)
|
Úcopyr r Ú from_productr r r% r& ÚtolistÚvalues)r r- Z other_dtyper r r Ú;test_reindex_preserves_names_when_target_is_list_or_ndarray. s" "
|
$" "r7 c C sb t jddgddggddgd} | jg ddd jddgks>t| jg ddd jddgks^td S ) Nr r/ ÚaÚbr1 r2 )r r )r r4 r r r )r r r r Ú=test_reindex_lvl_preserves_names_when_target_is_list_or_arrayG s r: c C sÚ t ddgddgg¡} | jg ddd jd jjtjks<t| jg ddd jd jjtj ksbtt
|
ddg¡}t
|
j dd d
|
d }t ||g¡}|jg ddd jd j|jks²t|jg ddd jd j|jksÖtd S ) Nr r/ r8 r9 r r1 r2 z
|
2016-01-01r z
|
US/Pacific)ZperiodsÚtz) r r4 r r ÚdtypeÚtyper% Úint64r Zobject_ÚpdZ CategoricalZ
|
date_range)r ÚcatZdtiÚmir r r Ú@test_reindex_lvl_preserves_type_if_target_is_empty_list_or_arrayN s &&$rB c C sT t j| jt jd}| | ¡}t ||¡ tjt dd | j| dd W 5 Q R X d S )N©r< zInvalid fill methodr Úinvalid)r" )
|
r% ÚarangeÚsizeÚintpZ get_indexerr' r( r) r* Ú
|
ValueError)r ÚexpectedÚactualr r r Útest_reindex_base\ s
|
|
rK c C sb t ddddg¡} tjt d¡| d}t dddg¡}d}tjt|d | |¡ W 5 Q R X d S )N)r r )r/ r/ )r r r ©r, z'cannot handle a non-unique multi-index!r )
|
r Ú from_tuplesr? ÚSeriesr% rE r) r* rH r )r r8 Únew_idxÚmsgr r r Útest_reindex_non_uniquef s rQ r6 r8 Úxc C sj t | ¡}|jt dg¡dd\}}t dg| d gg g gd}tjg |jd}t ||¡ t ||¡ d S )Nr9 r r r/ )r ÚcodesrC ) r Z from_arraysr r% r& r< r' Úassert_index_equalr( )r6 r r Zresult_indexerrI Zexpected_indexerr r r Ú test_reindex_empty_with_levelp s
|
rU c C sh ddddg} t | d d
¡}t| }| |¡\}}t ||¡ tjddddgtjd }t ||¡ d S )
|
N)ÚirV )rV Új)rW rV rW r r r/ r rC )
|
r rM r r r' rT r% r& rG r( )ÚkeysrA r Úresr rI r r r Útest_reindex_not_all_tuples{ s rZ c C sn t dddg¡} tjdddg| d}t ddddd d
|
d d d g ¡}tjtdd |j|ddd W 5 Q R X d S )N)r0 ÚA)r r[ )r ÚBg{®Gáz?g{®Gáz?gú~j¼t?rL )r0 r\ )r ÚC)é r\ )r^ r] )é r\ )r_ r] zFlimit argument only valid if doing pad, backfill or nearest reindexingr r r/ )Z
|
fill_valueÚlimit)r rM r? rN r) r* rH r )r ZdfrO r r r Ú&test_reindex_limit_arg_with_multiindex s& ÷ÿþra c C s^ t ddg¡} t ddg¡}tjddg| d}tjddg|d}| |¡}|}t ||¡ d S )N))r8 Nr/ ))r9 Nr r/ r rL )r rM r? Z DataFrameZ reindex_liker' Zassert_frame_equal)r, Úindex2Z df1_dtypeZ df2_dtyper rI r r r Ú+test_reindex_with_none_in_nested_multiindex¢ s
|
rc )Únumpyr% r) Zpandasr? r r Zpandas._testingZ_testingr' r r. r7 r: rB rK rQ ÚmarkZ parametrizerU rZ ra rc r r r r Ú<module> s
|
|
|
|
|
