U
|
ý°d· ã @ sÀ d dl m Z mZ d dlZd dlZd dlZd dlmZ d dl m
|
m Z d dl Zd dl mZmZmZmZmZmZmZmZ d dlmZ d dlmZ d dl m m Z G dd dZ!G dd d Z"dS )
|
é )ÚdatetimeÚ timedeltaN)Údateutil_gettz)Ú CategoricalÚCategoricalIndexÚ DataFrameÚIndexÚ
|
MultiIndexÚSeriesÚ
|
date_rangeÚisna)ÚCategoricalDtypec @ sZ e Zd Zdd Zdd Zdd Zdd Zej d d
|
d d d dgdddd dgg¡dd Z
|
dS )ÚTestReindexSetIndexc C sd t tj d¡}tddddd}tddddd}| |¡}t |j|¡ | |¡}t |j|¡ d S ) Né z
|
2011/01/01ÚMú
|
US/Eastern©ÚperiodsÚfreqÚtzZ2013ÚAz
|
Asia/Tokyo) r ÚnpÚrandomr Ú set_indexÚtmÚassert_index_equalÚindexÚreindex)ÚselfÚdfZidx1Zidx2© r ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_reindex.pyÚ(test_dti_set_index_reindex_datetimeindex s
|
|
z<TestReindexSetIndex.test_dti_set_index_reindex_datetimeindexc C s~ t tdddtddddddd}ttj dd¡d g|d
|
}t tddd tddd dddd}| |¡}|jj|jkszt d S ) Niß é
|
é é ÚHr )r r é Úa©Úcolumnsr é )
|
r r r r r Úrandnr r r ÚAssertionError)r r r Ú new_indexÚresultr r r! Ú'test_dti_set_index_reindex_freq_with_tz+ s
|
ÿ
|
ÿ
|
z;TestReindexSetIndex.test_dti_set_index_reindex_freq_with_tzc C s< t dtdi}t |jd¡}||d< | d¡}| ¡ }d S )Nr r# é ÚB)r ÚrangeÚpdZcutr r Ú reset_index)r r Úserr r r! Ú"test_set_reset_index_intervalindex8 s
|
|
z6TestReindexSetIndex.test_set_reset_index_intervalindexc C s¦ t dddgd dtjtjd¡}| dg¡}g |d< | ¡ }t ddddgtdd dtjtjtjd
|
¡}t ||¡ | ddg¡}g |d< | ¡ }t ||¡ d S ) Nr( ÚbÚc©r* zdatetime64[ns]©r( r8 r9 Údr r) ©r( r8 r9 r< )
|
r Úastyper Úint64Úfloat64r r5 r3 r Úassert_frame_equal)r r Zdf1r/ ÚexpectedÚdf2r r r! Ú test_setitem_reset_index_dtypes@ s ÿ ÿ z3TestReindexSetIndex.test_setitem_reset_index_dtypesz timezone, year, month, day, hourzAmerica/ChicagoiÝ é é r$ zAmerica/Santiagoiå é r% c
|
C s t |}tj||||dd|d}tj||||dd|d}t||gddgd d¡ dd g¡} t | tdd gd d gd d¡¡ d S )
|
Nr )ÚyearÚmonthÚdayÚhourÚminuteÚfoldÚtzinfor$ r( r8 )r Úvalsr Ú1Ú2)Úgettzr4 Ú Timestampr r r r rA )
|
r ÚtimezonerH rI rJ rK Z test_timezoneZ transition_1Z transition_2r r r r! Ú test_reindex_timestamp_with_foldR s: ù ù
|
ÿþÿþz4TestReindexSetIndex.test_reindex_timestamp_with_foldN) Ú__name__Ú
|
__module__Ú __qualname__r" r0 r7 rD ÚpytestÚmarkÚ parametrizerU r r r r! r s þr c @ s^ e Zd Zdd Zdd Zejdd Zdd Ze j
|
d d
|
d d d d gfde j d d d gfdd d d d gfg¡dd Zdd Zdd Zdd Zdd Ze j
|
ddd d d d gfd d d d e j gfg¡d!d" Zd#d$ Ze j
|
d%d&d'd(gd)gd*d&d(d+gd)gd(gd)gd(d'd&gd)gd&d(d'gd)gd&d'gd)gd&d(gd)gd(d'gd)gd'd(d&gd)gd+d*gd,gd(d&d'gd,gg ¡d-d. Ze j
|
d/d0d1d2d0d3d4d5d6d7d5d8d9d:d;d<d:d=d>g¡d?d@ Ze j
|
dAdBdCdDgd d dEd d dFdGdHdIdJdKdLdMdNdOdPdQdRdSdTggdDdCdBgd d d d dEdKdLdMdHdIdJdFdGdRdSdTdPdQdNdOggdCdDgd d dHdIdJdKdLdMdPdQdRdSdTg gdDdBgd d d d dEdKdLdMdFdGdRdSdTdNdOggg¡dUdV Ze j
|
dWedXd d d d dHdEdFdJdIg d)gedYd d d d dHdFdJdIgd)gedZd d d dFdJdIgd)ged[d d dEdIdFgd)ged\d d dEdHdIgd)ged]d d d d dEdHdFdJdIg d)ged^d dEdHgd)ged_g d,gg¡d`da Zdbdc Zddde Zdfdg Z dhdi Z djdk Z dldm Z dndo Z dpdq Z!drds Z"dtdu Z#dvdw Z$dxdy Z%dzd{ Z&d|d} Z'd~d Z(dd Z)dd Z*dd
Z+dd Z,dd Z-dd Z.dd Z/dd Z0dd Z1dd Z2dd Z3dd Z4dd Z5e j
|
dde6dGddde6dGdddd
|
iddig¡dd Z7e j
|
d¡e8g e9g g¡e j
|
d¢e8g e9g e8d(d'ge9d(d'ge8d(d(ge9d(d(gg¡d£d¤ Z:e j
|
d¥d¦d§g¡d¨d© Z;e j
|
dªe9g d(gd«e8d(ge8d(gfe9g d(gd«e8d'ge8d'gfe9g d(gd«e9d(ge9d(gfe9g d(gd«e9d'ge9d'gfg¡d¬d Z<dS )®ÚTestDataFrameSelectReindexc C s d}t tj |d |¡}t |¡}tj |¡ |j|dd}t |d j|d j¡rZt |j||j
|
dd}t |d j|d j¡rt d S )Nr# T©r* Úcopyr ©r* r r^ ) r r r r, ÚarangeÚshuffler Ú shares_memoryZ_valuesr- r )r ÚNr Úcolsr/ Úresult2r r r! Útest_reindex_copiesz s
|
z.TestDataFrameSelectReindex.test_reindex_copiesc C sä d}t tj |d |¡dd}t |¡}tj |¡ |j|dd}|rht |d jj |d jj ¡st
|
n t |d jj |d jj ¡rt
|
|j||j dd}|rÀt |d jj |d jj ¡sàt
|
n t |d jj |d jj ¡ràt
|
d S )Nr# ZFloat64©ÚdtypeTr] r r_ ) r r r r, r` ra r rb ÚarrayÚ_datar- r )r Zusing_copy_on_writerc r rd r/ re r r r! Útest_reindex_copies_ea s
|
" "z1TestDataFrameSelectReindex.test_reindex_copies_eac C s t dddj dd¡}t|ddgtdd}|jd }| ¡ }|jtd
|
ddd g|d }t|d ¡ |g |d ¡ |g |gd
|
d t d}t
|
||¡ |jtd
|
|d}t
|
||ddg ¡ |jtd
|
ddd gdd }t|d ¡ |g |d ¡ |g |gd
|
d }t
|
||¡ d S )Nú
|
2016-01-01r ©r rF r+ r r2 r) ©r r rG ÚC©r r* Ú
|
fill_value)r r2 ro rg )r rq ) r ÚvaluesZreshaper r3 ÚilocÚdater ÚtolistÚobjectr rA )r Úarrr ÚtsÚfvÚresrB r r r! Ú test_reindex_date_fill_value s*
|
,þ ÿ,ÿz7TestDataFrameSelectReindex.test_reindex_date_fill_valuec
|
C st t dgd dgd dgd ttdd dddd d
|
d d gd d ddg¡}ddddg}tjdg|gddgd}| |¡}t dgd |tjdd tjgd ddg¡}t
|
||¡ t dgd |ddd d gd ddg¡}|j|dd}t
|
||¡ |j|dd}t
|
||¡ t dgd |ddd d gd ddg¡}|j|dd}t
|
||¡ |j|dd}t
|
||¡ d S )Néÿÿÿÿé r r$ rF r r2 ro ÚDÚEÚFÚGr; r( r8 ç à?ç @g @g333333@)ÚnamesrG Úbfill©ÚmethodÚbackfillÚpadÚffill) r Úlistr3 r r Ú from_productr r Únanr rA )r r r. Znew_multi_indexÚ reindexedrB Zreindexed_with_backfillingZreindexed_with_paddingr r r! Ú test_reindex_with_multi_index» sF & ýÿú
|
ÿþ ÿþ ÿþ z8TestDataFrameSelectReindex.test_reindex_with_multi_indexzmethod,expected_valuesÚnearestr r$ r+ r r c
|
C s> t dttdi}t ddddg¡}t d|i|d}|j||d}t ||¡ |j||d d
|
}t ||¡ |j||d d d d gd
|
}t ||¡ |d d d
}|j|d d d
|d}t ||¡ d d dd g}|j| }|j|| |d}t ||¡ |dkrdn|dkrdn|} |d d d
j|| d}t ||¡ d S )NÚxr1 繿çÍÌÌÌÌÌì?çñ?ç ø?©r r r$ ©r Ú tolerancer| rF r r+ r r ) r r r3 r ri r r rA rs )
|
r r Zexpected_valuesr ÚtargetrB ÚactualÚe2Z new_orderZswitched_methodr r r! Útest_reindex_methods s(
|
ÿz/TestDataFrameSelectReindex.test_reindex_methodsc C s t dttdi}t ddddg¡}t ddddtjgi|d }|j|d
|
d d }t ||¡ t ddtjdtjgi|d }|j|d
|
d dddgd }t ||¡ d S )Nr r1 r r r r r r$ r r gÉ?r r g{®Gáz?gÙ?g¹?) r r r3 r ri r r r rA )r r r rB r r r r! Ú$test_reindex_methods_nearest_special* s z?TestDataFrameSelectReindex.test_reindex_methods_nearest_specialc C sZ |}t dd|d}tdttdi|d}| d¡}|j|d d
dd}t ||¡ d S ) Nz
|
2019-01-01r1 )r r r r rF r r )r r r r3 Úheadr r rA )r Ztz_aware_fixturer Úidxr rB r r r r! Útest_reindex_nearest_tz6 s
|
z2TestDataFrameSelectReindex.test_reindex_nearest_tzc C sF t dg¡}tt dg¡d}t|d}|j|dd}t ||¡ d S )Nz2016-06-26 14:27:26+00:00z2016-07-04 14:00:59+00:00r r r )r4 Z DatetimeIndexr r r rA )r Údtir rB r/ r r r! Ú#test_reindex_nearest_tz_empty_frame@ s
|
|
z>TestDataFrameSelectReindex.test_reindex_nearest_tz_empty_framec C s t dddd}ttj t|¡|d}| td¡}t |d j t d¡¡sRt
|
t |¡d }|d d
¡ stt
|
|d d
¡ rt
|
d S )
|
Nz1/1/2000 00:00:00r# Z10s©r r ©r r2 é r2 úM8[ns]éûÿÿÿ)r r r r r, Úlenr r3 Z
|
issubdtyperh r- Úcomr ÚallÚany)r Úrngr r/ Úmaskr r r! Útest_reindex_frame_add_natH s z5TestDataFrameSelectReindex.test_reindex_frame_add_natzmethod, exp_valuesr rF r
g ð?r g @c C sb |ddddgt ddddd d
|
}t d dddd }|j||t d ¡d }|||d
|
}t ||¡ d S )Nr r$ r+ rF z2020-01-01 00:00:00rG r& ÚUTCr r z2020-01-01 00:01:00z1 hourr )r r r4 Z Timedeltar Z assert_equal)r Zframe_or_seriesr Z
|
exp_valuesÚobjr. r/ rB r r r! Ú!test_reindex_frame_tz_ffill_bfillS s
|
þ z<TestDataFrameSelectReindex.test_reindex_frame_tz_ffill_bfillc C s dddgdddgdddgdddgg}dddgdddgdddgdddgdddgt jt jt jgg}t|}|jddddd d
|
gd dd }t|}t ||¡ d S ) Nr r2 ro r~ r r$ r+ rF rG r1 r )r Úlimit)r r r r r rA )r ÚdataZexp_datar r/ rB r r r! Útest_reindex_limitb s $ú z-TestDataFrameSelectReindex.test_reindex_limitzidx, check_index_typero r2 r Tr r~ Fc s t tdtdd d d
dddgd tj dd d
|
¡d d d dg} fdd}t tt||¡} |¡j|d d} j | |¡}t
|
j |||d d S )NÚ BBBBAACCCÚ abcdeabcdr| r# é é rF r éè é ©ÚjimÚjoeÚjolieZjoliner¼ r½ r¾ c s t d | k ¡ ¡d S ©Nr¼ r ©r ZnonzeroZto_numpy©Úval©r r r! Úf s zKTestDataFrameSelectReindex.test_reindex_level_verify_first_level.<locals>.f©Úlevel©Úcheck_index_type) r r r r ÚrandintÚ concatenateÚmapr r rs r rA )r r rÈ ÚicolrÄ ÚiÚleftÚrightr rà r! Ú%test_reindex_level_verify_first_levelr s üÿ
|
z@TestDataFrameSelectReindex.test_reindex_level_verify_first_levelr )Úmid)rÑ Úbtm)rÑ rÒ Útop)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Ò rÑ c sø t dgd dgd dgd dgd d gd
|
d gd
|
d gd dgd
|
d gd dgd
|
d gd t d d dD ¡tj d¡ d
|
¡d d dddg} fdd}t tt||¡} |¡j |dd} j
|
| |¡}t ||¡ d S )NrÑ r1 rÒ é rÓ r} Ú3rdr+ Ú1strF Ú2ndc S s g | ]}t jjd |ddqS ©r¹ F)Úreplace©r r Úchoice©Ú.0r r r r! Ú
|
<listcomp>½ s ÿz\TestDataFrameSelectReindex.test_reindex_level_verify_first_level_repeats.<locals>.<listcomp>©r+ rF rF r+ rF r+ rF r+ r· r# r» r¼ r½ r¾ c s t d | k ¡ ¡d S r¿ rÀ rÁ rà r r! rÄ Ç s zSTestDataFrameSelectReindex.test_reindex_level_verify_first_level_repeats.<locals>.frÅ ) r r rÊ r r, Úroundr rË r r rs r rA )r r rÌ rÄ rÍ rÎ rÏ r rà r! Ú-test_reindex_level_verify_first_level_repeats s> ÿþýüûúù þÿíÿ
|
zHTestDataFrameSelectReindex.test_reindex_level_verify_first_level_repeatsz idx, indexerrÖ r× rÕ rG rÔ rº r1 r r} r# rE é é é é é r¥ é é c C sØ t dgd dgd dgd dgd d gd
|
d gd
|
d gd dgd
|
d gd dgd
|
d gd t d d dD ¡tj d¡ d
|
¡d d}dddg}| |¡j|dd}|j| |¡}t
|
||¡ d S )NrÑ r1 rÒ rÔ rÓ r} rÕ r+ rÖ rF r× c S s g | ]}t jjd |ddqS rØ rÚ rÜ r r r! rÞ î s ÿzPTestDataFrameSelectReindex.test_reindex_level_verify_repeats.<locals>.<listcomp>rß r· r# r» r¼ r½ r¾ rÅ ) r r rÊ r r, rà r r rs r rA )r r Úindexerr rÌ rÎ rÏ r r r! Ú!test_reindex_level_verify_repeatsÏ s: ÿþýüûúù þÿíÿ
|
z<TestDataFrameSelectReindex.test_reindex_level_verify_repeatsz idx, indexer, check_index_typeZabcdeÚabcdÚabcZecaZedcZeadbcZedwqZwqc C s| t tdtdd d d
dddgd tj dd d
|
¡d }d d dg}| |¡j|d d}|j| |¡}tj |||d d S )Nrµ r¶ r| r# r· r¸ rF r r¹ rº r» r¼ r½ r¾ rÅ rÇ )
|
r r r r rÉ r r rs r rA )r r ré rÈ r rÌ rÎ rÏ r r r! Útest_reindex_level_verifyû s üÿ
|
z4TestDataFrameSelectReindex.test_reindex_level_verifyc C s t dddd}tj dd¡}t||tdd}t||dd d
|
gd dd g tdd}d }tjt|d |j |j
|
dd W 5 Q R X tjt|d |j |j
|
dd W 5 Q R X tjt|d |j |j
|
dd W 5 Q R X tjt|d |j |j
|
dd W 5 Q R X d S )Nz
|
2013-08-01r r2 r£ r$ r ©r r* rF rG r1 r r+ z0index must be monotonic increasing or decreasing©Úmatchr r r r
r ) r r r r, r r rY ÚraisesÚ
|
ValueErrorr r )r Zdrr³ r Zdf_revÚmsgr r r! Ú"test_non_monotonic_reindex_methods s &z=TestDataFrameSelectReindex.test_non_monotonic_reindex_methodsc C sz t ddgtjddgt dd¡dd}| ddg¡}t dtjgtjdtjgt dd¡ddddgd }t ||¡ d S )
|
Nr r$ r? rg r¤ r+ ç r@ r ) r r4 ri Z SparseDtyper r r r rA ©r r r/ rB r r r! Útest_reindex_sparse& s "ÿþûz.TestDataFrameSelectReindex.test_reindex_sparsec C sv t jdd}| |j¡}|jD ]j}|| ¡ D ]X\}}||jkrxt |¡rbt || | ¡svtq||| | kstq.t |¡s.tq.q | ¡ D ] \}}t |j|j¡stq| t
|
g ¡}t |jdksÐt| |jd d d
¡} | jD ]r}| | ¡ D ]`\}}||jkrNt |¡r6t || | ¡sLtn||| | ks\tqüt |¡sütqüqì| ¡ D ] \}}t |j| j¡shtqh|j|jdd}|j|jks¬t| g ¡}|j sÂtt |jt |jksÜt| g ¡}| |j¡}t |jt |jks tt |jt |jks&t| t |j¡}|j d ¡}
|
t |j|
|
¡ | ¡ } t | |¡ | |k srtd S )Nr¸ )Znperr r+ F)r^ )r ZmakeTimeSeriesr r r* Úitemsr Úisnanr- Z equalContentsr r¨ Úemptyr Z
|
_with_freqr rA ) r Ú float_frameZdatetime_seriesZnewFrameÚcolr r ZseriesZ
|
emptyFrameZnonContigFramerB r/ r r r! Ú test_reindex5 sL
|
|
|
|
|
|
z'TestDataFrameSelectReindex.test_reindexc C s4 t ddgddgddgddggdtjddgd d
|
gd }tjddtjddtjgdddddd dg }}t | |¡|j| ¡ |j d ¡|_tj| |¡|j| dd t ddtjdgdtjdtjgddddgd}t
|
|j ¡|d< t
|
d¡|d d¡|d< | dddg¡ ¡ }|jddddgd }t ||¡ d S ) Nr$ r+ rF r1 r} rE rº r% r½ r¼ rî r rv FrÇ r( r8 r9 z
|
2015-03-22z
|
2012-01-08rG )Úotherrt Úamountrt z
|
2015-06-18Údeltarþ rÿ r: )r r r r rA r rs r r> r4 Ú to_datetimert Úshiftr r5 )r r rÍ ÚjrÎ rÏ r r r! Útest_reindex_nant s( ý,
|
ýÿ z+TestDataFrameSelectReindex.test_reindex_nanc C sÔ t tj d¡}t|t t|¡d}t t d¡dd}| |¡}|jj dksRt
|
| t t d¡dd¡}|jj dkszt
|
t tj d¡}t|j t t|¡d}t t d¡dd}|j|d}|j j dksÐt
|
d S )Nr# r Ziname©ÚnameZtmpnamer: )r
|
r r Zrandr r` r¨ r r r r- r ÚTr* )r Úsr rÍ r r r! Útest_reindex_name_remains s
|
z4TestDataFrameSelectReindex.test_reindex_name_remainsc C sr | |jd d d
¡}|d jtjks*t| |j¡}|d jtjksJt|j ddgd}|d jtjksntd S )Nr+ r r2 r: )r r rh r r? r- r@ )r Ú int_frameÚsmallerZbiggerr r r! Útest_reindex_int¢ s z+TestDataFrameSelectReindex.test_reindex_intc C sb |j dddgd}t |d |d ¡ t |d ¡ ¡ s<td|ksHt|j g d}|js^td S )Nr r2 r r: ro )r r Zassert_series_equalr rù rª r- rú )r rû Z new_framer r r! Útest_reindex_columns s z/TestDataFrameSelectReindex.test_reindex_columnsc C sz t dddgdddgddd ggd
|
d d gd
|
d d gtd }|jtdd}t tjddtjdtjgtjddtjdtjgtjddtjd tjggd
|
d d gtdtd }t ||¡ |jtddd}t tjdddddgtjdddddgtjdddd d ggd
|
d d gtdtd }t ||¡ |jtddd}t dddddtjgdddddtjgdddd d tjggd
|
d d gtdtd }t ||¡ d S )NrE râ rã é é r% é é é! r$ r+ rG )r³ r r* rh r r: r )r* r r
)r Úfloatr r3 r r r rA rö r r r! Útest_reindex_columns_method¸ sN üýø
|
ýø
|
ýø
|
z6TestDataFrameSelectReindex.test_reindex_columns_methodc C s¤ t t d¡tdddtdddtdddgdddgd }td
|
d d d }ddg}|j|djj}|j||d jj}|j|dj|djj}||kst||ks td S )N©rF rF iÜ r$ r+ rF r( r8 r9 rî ú
|
2012-01-01z
|
2012-01-03r< )r r r: ) r r Úonesr r r r r r- )r r Z time_freqZ some_colsZ
|
index_freqZ both_freqZseq_freqr r r! Útest_reindex_axesë s ý z,TestDataFrameSelectReindex.test_reindex_axesc C s: t tj dd¡}| ttd¡}t |jdd
¡ ¡ s@t
|
|jtddd}| td¡ d¡}t ||¡ |jtddd }| ¡ }d|d< t ||¡ |jtddd }| ¡ }d|d< t ||¡ |jtdd
|
d }| ¡ }d
|
|d< t ||¡ d
|
|d
|
< |jtddd}| td¡ d¡}t ||¡ d S ) Nr# rG r¥ r§ r )rq r1 rõ )r* rq Úfoo)r r r r, r r r3 rù rr rª r- Zfillnar rA r^ rö r r r! Útest_reindex_fill_valueû s, z2TestDataFrameSelectReindex.test_reindex_fill_valuec C sl t ddgddgd|d}|jtdddddgdd }t ddddgddddgddd
|
|d}t ||¡ d S ) Nr$ r+ ©r( r8 rg rë r rF r# ©r* r rq r= )r r r r rA )r Z any_unsigned_int_numpy_dtyper r/ rB r r r! Ú#test_reindex_uint_dtypes_fill_value s þz>TestDataFrameSelectReindex.test_reindex_uint_dtypes_fill_valuec C sZ t dddgi|d}|jtddddgdd}t tdddg|ddd }t ||¡ d S )
|
Nr( r$ r+ rg Úabr r# r r )r r r r
|
r rA )r Zany_numeric_ea_dtyper r/ rB r r r! Ú/test_reindex_single_column_ea_index_and_columns& s ÿzJTestDataFrameSelectReindex.test_reindex_single_column_ea_index_and_columnsc C s¤ t j d¡}t|ddddddddddg
|
d}| ¡ }ttt||_t|ttt|d}t
|
||¡ d}t j t |d |jttt|d W 5 Q R X d S )
|
Nr# r$ r+ rF rG r1 r ú/cannot reindex on an axis with duplicate labelsrï )r r r, r r^ r r3 r¨ r r rA rY rñ rò r )r rw r r/ rB ró r r r! Útest_reindex_dups/ s z,TestDataFrameSelectReindex.test_reindex_dupsc C s t dddgdddgdddggdddgd}d}tjt|d |jdgd W 5 Q R X tjt|d |jdd gd W 5 Q R X d S )
|
Nr$ r1 g @Úbarr( r: r rï r )r rY rñ rò r ©r r ró r r r! Ú#test_reindex_with_duplicate_columns? s ÿz>TestDataFrameSelectReindex.test_reindex_with_duplicate_columnsc C s¢ t dddgdddgd}t ddtjgddtjgddddgd }| dddg¡}t ||¡ |jdddgdd
|
}t ||¡ |jdddgd d
|
}t ||¡ d S ) Nr$ r+ rF rG r1 r r¤ r r ©Úaxisr )r r r r r rA )r r rB r/ r r r! Útest_reindex_axis_styleJ s ÿ z2TestDataFrameSelectReindex.test_reindex_axis_stylec C sR t dddgdddgd}d}tjt|d | d
|
dgd d d g¡ W 5 Q R X d S )Nr$ r+ rF rG r1 r r¤ zCreindex\(\) takes from 1 to 2 positional arguments but 3 were givenrï r r r2 ro ©r rY rñ Ú TypeErrorr r# r r r! Ú test_reindex_positional_raisesY s z9TestDataFrameSelectReindex.test_reindex_positional_raisesc C sî t dddgdddgd}tjtdd |jd
|
dgd gdd W 5 Q R X tjtdd |jd
|
dgd gd d W 5 Q R X tjtdd |jd
|
dgd d W 5 Q R X tjtdd |jd
|
dgdd W 5 Q R X tjtdd |jd
|
dgdd W 5 Q R X tjtdd |jd
|
dgd
|
dgdd W 5 Q R X tjtdd |jd
|
dgd
|
gd gd W 5 Q R X tjtdd |jd
|
dgd d W 5 Q R X tjtdd |jd
|
dgdd W 5 Q R X tjtdd |jd
|
dgd
|
dgd W 5 Q R X d S )Nr$ r+ rF rG r1 r r¤ zCannot specify both 'axis'rï r r )r* r& r )r r& r* )r r* r& zCannot specify all)Úlabelsr r* zmultiple values©r+ r( )r r r r r! Ú test_reindex_axis_style_raisesa s* "" z9TestDataFrameSelectReindex.test_reindex_axis_style_raisesc C sN t dddgdddgd}|jddgdgd}t dddgi}t ||¡ d S )Nr$ r+ rF r¤ r r r: ©r r r rA rö r r r! Ú!test_reindex_single_named_indexer s z<TestDataFrameSelectReindex.test_reindex_single_named_indexerc C s\ t dddgdddgdddggdd d
|
gd d d gd}| d dg¡}|jd dgd}|jd dgd}|jd dgdd}|jd dgdd}||||fD ]}t ||¡ q|jd d gd}|jd d gdd}|jd d gdd}||fD ]}t ||¡ qÞ|jd dgd d gd}|jd d gd dgd}|jd dgddjd d gdd}||fD ]}t ||¡ qDd S )Nr$ r+ rF rG r1 r r} r( r8 r9 r< ÚerÄ rî r r, r )r+ r& r% r: r) r. )r r Zres1Zres2Zres3Zres4Zres5rz r r r! Ú test_reindex_api_equivalence s, ý z7TestDataFrameSelectReindex.test_reindex_api_equivalencec C s t tjdtdt ddd¡ddgd}| t d¡¡}|jjtjksHt t
|
|d d s\t |jt d d
|
}|jjtjks~t t
|
|d ¡ st d S ) N)r# r+ rg r r· r+ rî r# r$ rF r: ) r r r Úboolr` r rr rh Zobject_r- r r3 rª )r Úframer r r r! Útest_reindex_boolean¨ s ÿz/TestDataFrameSelectReindex.test_reindex_booleanc C s> |j dddgd}d|ks t|j ddgd}d|ks:td S )Nr r r2 r: )r r- )r Zfloat_string_framer r r r! Útest_reindex_objectsµ s z/TestDataFrameSelectReindex.test_reindex_objectsc C sh t dddg}ti jdddgd}|j|d}t |j|¡ |jd d
|
d gd}|d jtjksdt d S ) Nr( r8 r9 r$ r+ rF r r: r r2 r )
|
r r r r r r* rh r r@ r- )r r
|
r Údmr r r r r! Útest_reindex_corner¼ s z.TestDataFrameSelectReindex.test_reindex_cornerc C s¾ t ddgddgtjtjgddgddggd d
|
gd d tjd dgd}|jd d dgd}|jdddg }t ||¡ |jdgd}|jdg }t ||¡ |jd gd}|jdg }t ||¡ d S )Nr$ r+ rF rG r} rÔ rº r# r( r8 g Y@g @Y@g Y@g ÀY@r) r )r r r r rs r rA rö r r r! Útest_reindex_with_nansÆ s $ý z1TestDataFrameSelectReindex.test_reindex_with_nansc C sV t tj dd¡}|jtdtdd}| ttd¡jtdd}t ||¡ t tj ddd¡}|jtdtdd}| ttd¡jtdd}t ||¡ t tj ddd¡}|jtdtdd}| td¡jtdd}t ||¡ t tj d d¡d
|
d d d gd}|jddgd d gd}| ddg¡jd d gd}t ||¡ d S )NrF rG rî r: r r# r r+ r1 y ð?r( r8 r9 r$ )
|
r r r r, r r3 r r rA rÉ rö r r r! Útest_reindex_multiÙ s z-TestDataFrameSelectReindex.test_reindex_multic C s t tdddgttddddg¡}tdtt|i|d}|jd d
|
d dd d ddg }| |¡}tdd d
|
d dd d dt j
|
dg i|d}t ||¡ d S )Nr( r8 r9 r rF r& r£ r r r$ r+ rG r1 r rÔ ) r r r r r r3 r¨ rs r r r r rA )r Zmidxr rC r/ rB r r r! Ú#test_reindex_multi_categorical_timeö s þÿ
|
$z>TestDataFrameSelectReindex.test_reindex_multi_categorical_timec C s. t dtjdddittdttdddd }| d
|
d d g¡}t d dtjgttdd d¡}t
|
j ||dd | d
|
d g¡}t d dgttdd d¡}t
|
j ||dd | d g¡}t tjgtd gd d¡}t
|
j ||dd | dg¡}t tjgtdgd d¡}t
|
j ||dd td}| t d
|
d g|d¡}t d tjgttd t|¡d d¡}t
|
j ||dd | t d
|
g|d¡}t d gttd
|
t|¡d d¡}t
|
j ||dd | d
|
d d g¡}t d dtjgttdd d¡}t
|
j ||dd | d
|
d g¡}t d dgttdd d¡}t
|
j ||dd | d g¡}t tjgtd gd d¡}t
|
j ||dd | t d
|
d g|dd¡}t d tjgttd t|dd¡d d¡}t
|
j ||dd | t d
|
dgd
|
dgd¡}t d tjgttd td
|
dg¡d d¡}t
|
j ||dd t dtjdddittdttdddd }d }tjt|d | d
|
d g¡ W 5 Q R X d }tjt| d ¡d |jd
|
gd d! W 5 Q R X tjt| d"¡d |jd
|
gdd# W 5 Q R X tjt| d$¡d |jd
|
gd%d& W 5 Q R X d S )'Nr rF r? rg rì Zcaber2 )rh r r r( r8 r0 r r$ Zaber¤ TrÇ r r< ©Ú
|
categoriesZae)r< Úordered)r= Úadr Zaabbcar rï z<argument {} is not implemented for CategoricalIndex\.reindexr r r rÆ rÅ r² r+ )r² )r r r` r r ÚCDTr r r
|
r r rA r r> rY rñ rò ÚNotImplementedErrorÚformat)r r r/ rB ZcatsrC ró r r r! Ú"test_reindex_with_categoricalindex s ÿü ÿ ÿþ ÿþ ÿ $ÿþ$ÿþ ÿüz=TestDataFrameSelectReindex.test_reindex_with_categoricalindexc C s< t tj¡}t|j}|ddddddddd d
|
d h ks8td S ) Nr r+ r r* r& r² r^ rÆ r rq r )ÚinspectÚ signaturer r ÚsetÚ
|
parametersr- )r ÚsigrF r r r! Útest_reindex_signature` s
|
õz1TestDataFrameSelectReindex.test_reindex_signaturec C s t ddg¡}tddgddgg|ddgd }t ddd
|
g¡}|j|dd d }tddgddgddgg|ddgd }t ||¡ d S ) Nr )r< r0 r r} rF rG r Úyrî )ÚhrÍ r )r& r )r Ú from_tuplesr r r rA )r Úmir Zmi2r/ rB r r r! Ú(test_reindex_multiindex_ffill_added_rowsq s $zCTestDataFrameSelectReindex.test_reindex_multiindex_ffill_added_rowsÚkwargs)Úsecondsr r Nc C s\ t dddd}tg tg dddgd}|j|f|}tdtjgd i|d }t ||¡ d S )
|
NZ2020Z30srF )Ústartr r Útimer r( rî r )r r r r r4 ZNAr rA )r rN r r r/ rB r r r! Útest_reindex_empty_frame{ s
|
z3TestDataFrameSelectReindex.test_reindex_empty_frameÚsrc_idxÚcat_idxc C s< t |dgdd}|j|d}t dg|dd}t ||¡ d S )NÚKZf8)r* r rh r: )r r* rh r. )r rS rT r r/ rB r r r! Útest_reindex_empty s z-TestDataFrameSelectReindex.test_reindex_emptyrh úm8[ns]r¦ c C sR t tdtdg¡}tddd}t dd¡}|dkrN||d }t dd¡}t||d
|
}t j
|
|d d d
< | ¡ }|j td g¡}|j td g¡}|j|||d} t|d ¡ |g |d ¡ |g tjdgd |jdd|d
|
}
|
| jd d g tk ¡ st| jd t j
|
kst| jd |ks.t| jd |ksBtt | |
|
¡ d S )NZABCDEr+ rl r# rm ÚNaTÚnsrW r r rF r$ r rp r rg )r r$ r rn )r| r )r| r$ ) r r r r3 r r Z timedelta64Z
|
datetime64r
|
r4 rX Zunstackr Úappendr r* r r ru ri rh Zdtypesrv rª r- rs r rA ) r rh rL r¡ ry r6 r r r* rz rB r r r! Ú#test_reindex_datetimelike_to_object§ s0 ýú z>TestDataFrameSelectReindex.test_reindex_datetimelike_to_objectz index_df,index_res,index_expr; c C s0 t |d}|j|d}t |d}t ||¡ d S )Nr r. )r Zindex_dfZ index_resZ index_expr r/ rB r r r! Útest_reindex_not_categoryÊ s
|
|
z4TestDataFrameSelectReindex.test_reindex_not_category)=rV rW rX rf rk ÚtdZ&skip_array_manager_not_yet_implementedr{ r rY rZ 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 r r! r$ r' r* r- r/ r1 r4 r5 r7 r8 r9 r: rB rH rM r rR r r rV r[ r\ r r r r! r\ v sJ
|
Kýþ
|
|
þ
|
|
|
|
|
ôþ
|
îþ
|
"*þ*þ$(öþ
|
øþ
|
? 3!
|
#
|
[
|
üþ
|
þþ
|
|
|
|
øþ
|
" ý ý ý ýðþr\ )#r r rC Únumpyr rY Z pandas._libs.tslibs.timezonesr rR Z pandas.util._test_decoratorsÚutilZ_test_decoratorsr] Zpandasr4 r r r r r r
|
r r Zpandas._testingZ_testingr Zpandas.api.typesr r? Zpandas.core.commonÚcoreÚcommonr© r r\ r r r r! Ú<module> s (
|
Y
|
