fume-manage-python.git

U  
®ý°d>7ã@sdZddlmZmZddlZddlmZmZddlZddl    Z    ddl
Z
ddlmZ ddlZddlmZmZmZmZmZddlmZddZd    d
Ze    j ddd g¡ddZddZddZddZddZ ddZ!ddZ"ddZ#ddZ$d d!Z%d"d#Z&d$d%Z'd&d'Z(dS)(z<
Also test support for datetime64[ns] in Series / DataFrame
é)ÚdatetimeÚ    timedeltaN)ÚgettzÚtzutc©Úindex)Ú    DataFrameÚSeriesÚ    TimestampÚ
date_rangeÚperiod_rangec    Csêtdtdddtdddd}tt t|¡|d}|ddksDt|ddksTt|d    dksdt|td
dddks|t|ttd
dddksttj    t
dd |dW5QRXt |dd|td
ddtd
dd¡dS)NúWOM-1FRIéÕééÚ©ÚfreqÚstartÚendré0ú1/2/2009z2009-1-2iÙéz^'2009-1-3'$©Úmatchz2009-1-3z3/6/2009ú
2009-06-05ééé) rrr    ÚnpÚarangeÚlenÚAssertionErrorr
ÚpytestÚraisesÚKeyErrorÚtmÚassert_series_equal©ZdtiÚs©r)úad:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/series/indexing/test_datetime.pyÚtest_fancy_getitems"
 
ÿ
ÿr+cCstdtdddtdddd}tt t|¡|d}d|d<|ddksLtd    |d
<|dd    ksdtd|d
d<|dd dk ¡stdS)Nr rrrrréÿÿÿÿréþÿÿÿréýÿÿÿré6)rrr    rrr r!Úallr'r)r)r*Útest_fancy_setitem2s
 
ÿr1Ú    tz_sourceÚpytzZdateutilc    Cs,|dkrtj}ndd}d}td|d|dd}ttj |¡|d    }| ¡}d
|d<|d|d<t     ||¡| ¡}d
|d <|d|d <t     ||¡| ¡}d
|t
dddd|dd<|d|t
dddd|dd<t     ||¡| ¡}tdddd |d¡}|  ¡}d
||<|d||<t     ||¡dS)Nr3cSs|dkrtSt|S)NÚUTC)rr)Úxr)r)r*Ú<lambda>Fóz2test_getitem_setitem_datetime_tz.<locals>.<lambda>é2ú1/1/1990ÚHú
US/Eastern©ÚperiodsrÚtzrrz1990-01-01 09:00:00+00:00éz1990-01-01 03:00:00-06:00éÆré    r4)Útzinforú
US/Central)r3Útimezonerr    rÚrandomÚrandnÚcopyr%r&rr
Útz_localizeZ to_pydatetime)r2ZtzgetÚNÚrngÚtsÚresultÚdtr)r)r*Ú test_getitem_setitem_datetime_tz@s0rNc
 
Csd}td|ddd}ttj |¡|d}|d}|d}||ksDt| ¡}d    |d<|d|d<t ||¡|dd
}|dd}t ||¡| ¡}d    |dd
<|dd|dd
<t ||¡d}d
}||j    |k|j    |k@}|dd}t ||¡d}d }||j    |k|j    |k@}|dd}t ||¡d}t
dddd}|t|t |d¡fD]2}    t jtt t|    ¡d||    W5QRXq\| ¡}|d||<|j    jtks¶tt |j    dd| t¡¡|j    d|ksætd}t jt|d||t
ddddW5QRX| ¡}t jt|dd    ||t
dddd<W5QRXt jt|dd||t
dddd<W5QRXt ||¡|}t
dddd}d}t jt|d||j    |k|j    |k@W5QRXt| |j¡}tt
dddd |j¡}||j    |k|j    |k@}|dd}t ||¡||j    d}|d}||ksTt||j    dd}|dd}t ||¡| ¡}d    ||j    dd<|jdd|jdd<t ||¡|d}|dd}t ||¡| ¡}d    |d<|dd|d<t ||¡dS)Nr8r9r:r;r<rz1990-01-01 04:00:00r?rz1990-01-01 07:00:00éz1990-01-01 04:00:00-0500z1990-01-01 07:00:00-0500z:Cannot compare tz-naive and tz-aware datetime-like objectsr@rÚnsrr,ééczJInvalid comparison between dtype=datetime64\[ns, US/Eastern\] and datetimez
1990-01-02ér)rr    rrErFr!rGr%r&rrr
Z
datetime64r"r#r$ÚreÚescapeÚreprZdtypeÚobjectZassert_index_equalZastypeÚ    TypeErrorrHrBÚiloc)
rIrJrKrLÚexpectedÚlbÚrbÚmsgZnaiveÚkeyr)r)r*Ú"test_getitem_setitem_datetimeindexfs   "r_cCsd}td|dd}ttj |¡|d}|d}|d}||ksBt| ¡}d|d<|d|d<t ||¡|dd    }|dd
}t ||¡| ¡}d|dd    <|dd
|dd    <t ||¡d}d    }||j    |k|j    |k@}|dd
}t ||¡||j    d}|d}||kst||j    dd
}|dd
}t ||¡| ¡}d||j    dd
<|j
dd
|j
dd
<t ||¡dS)Nr8r9r:©r=rrz 1990-01-01 04r?rz 1990-01-01 07rO)rr    rrErFr!rGr%r&rrY)rIrJrKrLrZr[r\r)r)r*Ú test_getitem_setitem_periodindexÑs@rac    CsÖtdd}| d¡}tt||d}td}tjtt     t
|¡d||W5QRXd||<||dkspttt||d}|ddd}tjtt     t
|¡d||W5QRXd||<||dksÒtdS)    Nú1/1/2000z1/7/2000rrz1/8/2000rrr,)rÚrepeatr    r r
r"r#r$rTrUrVr!)rr(Zstampr)r)r*Útest_datetime_indexingýs
 
rdc
    Csþ|}|j ¡}|D]}||}|j|k}|j|k ¡}||}|dkrTt ||¡nt ||d¡| ¡}d||<tt     |d|¡|jd}t ||¡qt
ddd}    tjt t t|    ¡d||    W5QRXd|t
ddd<|t
ddddksútdS)NrrriÐrr)rÚuniqueÚsumr%r&Zassert_almost_equalrGr    rÚwhererr"r#r$rTrUrVr!)
Z(rand_series_with_duplicate_datetimeindexrKZuniquesÚdaterLÚmaskÚtotalrZÚcpr^r)r)r*Ú*test_indexing_with_duplicate_datetimeindexs&
 
rlcCs0| tdd¡g}tdd}tdd}tddd    d
d}d}t|D]P}| |¡| ||¡| |||¡| ||||¡|d |7}qBtj dt    |dd
¡}|D]}||||d<q°t
tj t    |d¡|tdd}    |d }
|    j |
}||    j kst|    j|t    |    j|gdks,tdS)NÚ_SIZE_CUTOFFéèr)Úsecondsi ¡)ÚmicrosecondsiÛéréééLrrr?ÚABCD)rÚcolumns)ÚsetattrÚlibindexrrÚrangeÚappendrrEÚrandintr rrFÚlistrr!Úloc)ÚmonkeypatchÚdatesÚsecZhalf_secÚdÚnÚiZduplicate_positionsÚpÚdfÚposÚ    timestampr)r)r*Ú!test_loc_getitem_over_size_cutoff5s*
 
 
 
 
rcCs| tdd¡d}tdd|d}|jjs,tttj     t
|¡|d}|d}||}||jksbt||t
|j|gd    kstdS)
NrmrnrtrbÚT)rr=rrr) rwrxrZ_engineZover_size_thresholdr!r    rrErFr rr})r~rÚidxr(rrr)r)r*Ú+test_indexing_over_size_cutoff_period_indexWsrc    s"tddd}ttj t|¡|dt ddddddg¡jD] }|}|}||ksRt    qRfdd    }|t
ddtjt d
d|t
ddW5QRX|t
dd |t
dd|t
dd|t
ddd ¡}d}|j d¡|_t ||¡dS)Nz
2011-01-01z
2011-01-15)rrrrr?éüÿÿÿcs:| ¡}| ¡}|}|j d¡|_t ||¡dS)N)rGÚ
sort_indexrÚ
_with_freqr%r&)ZslobjrLrZ©rKZts2r)r*Úcomparews
z(test_indexing_unordered.<locals>.comparez,Value based partial slicing on non-monotonicrz
2010-12-30z
2011-01-16z    2011-01-6z
2011-01-06z    2011-01-8z
2011-01-12Z2011)rr    rrEÚrandr ÚpdÚconcatrr!Úslicer"r#r$rrr%r&)rJÚtrZrLrr)rr*Útest_indexing_unorderedks&(
rcCsdttdddddd}tt t|¡|d}| tj d¡¡}|d}|j    D]}|j
dksLtqLdS)NrrrrÚMr`rZ2005)rrr    rrr ZtakerEZpermutationrÚyearr!)rJrKrLrr)r)r*Útest_indexing_unordered2s
rc    Cs¨tdddd}ttj t|¡|d}|d}d|_td|i}tj    t
dd|dW5QRXd    |d<|d}d|_d    |jd
<tj    t
dd|dW5QRXdS)Nz2001-1-1rrrr`rÚ2001ÚArr)rr)rr    rrErr Únamerr"r#r$r})rrKrZrr)r)r*Ú test_indexings
rcCsptdddd}ttt||d}|d}t ||¡tdddd}ttt||d}|d}t ||¡dS)    Nú2013-05-31 00:00z2013-05-31 23:00r:)rrrrz2013-05z2013-05-31 23:59ÚS)rr    ryr r%r&©rrKrZr)r)r*Ú)test_getitem_str_month_with_datetimeindex¶sr¡c
CsLtdttdddddddg}ttt||d}|d    }t ||¡dS)
NriÝrééé;i?BrZ2013)r
rr    ryr r%r&r r)r)r*Ú(test_getitem_str_year_with_datetimeindexÃsþr¥c    Csttj dd¡dddddgtdddd    d
d}tjtdd |dW5QRXd}tjt|d ||jdW5QRXdS)NrÚopenÚhighÚlowÚcloseÚvolumez2012-01-02 18:01:00rCr()r=r>r)rvrz^'2012-01-02 18:01:02'$rz2012-01-02 18:01:02z8Timestamp\('2012-01-02 18:01:02-0600', tz='US/Central'\)r)    rrrErrr"r#r$r)rr]r)r)r*Ú*test_getitem_str_second_with_datetimeindexÍsýr«))Ú__doc__rrrTZdateutil.tzrrÚnumpyrr"r3Zpandas._libsrrxZpandasrrr    r
rrZpandas._testingZ_testingr%r+r1ÚmarkZparametrizerNr_rardrlrrrrrr¡r¥r«r)r)r)r*Ú<module>s4
%k,"$