U
|
®ý°d0 ã @ s^ d dl Zd dlZd dlmZ d dlmZmZmZm Z m
|
Z
|
m Z m Z d dl mZ G dd dZdS )é N)ÚIncompatibleFrequency)Ú DatetimeIndexÚSeriesÚ TimestampÚ
|
date_rangeÚisnaÚnotnaÚoffsetsc @ sD e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dS )ÚTestSeriesAsofc s t d d¡j t fddtdD }ttj d¡|d}| |j
|
d ¡}|j dks\t ||d kslt t d d¡}||t | kst d S )
|
NÚ20130101Únsc s g | ]} d | qS )é2 © )Ú.0Úi©Útsr ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/series/methods/test_asof.pyÚ
|
<listcomp> s zDTestSeriesAsof.test_asof_nanosecond_index_access.<locals>.<listcomp>éd ©Úindexr Z
|
nanosecondz 2013-01-01 00:00:00.000000050)r Zas_unitÚ_valuer Úranger ÚnpÚrandomÚrandnÚasofr Ú
|
resolutionÚAssertionErrorZ
|
datetime64)ÚselfZdtiZserZ first_valueZ expected_tsr r r Ú!test_asof_nanosecond_index_access s z0TestSeriesAsof.test_asof_nanosecond_index_accessc C s d}t d|dd}ttj |¡|d}tj|jdd
< t d|d d d}| |¡}t| ¡ sbt
|
|j d
|
}|j d }| t |¡}t| ¡ st
|
|j d
|
}|j d }|j |k|j |k @ }|| } | || k ¡ sØt
|
||j |j |k d }
|
|| |
|
kst
|
d S ) Nr ú1/1/1990Ú53s©ZperiodsÚfreqr é é é Ú25sé r ) r r r r r ÚnanÚilocr r Úallr r Úlist) r ÚNÚrngr ÚdatesÚresultÚlbÚubÚmaskÚrsÚvalr r r Ú
|
test_basic$ s$
|
|
|
|
|
zTestSeriesAsof.test_basicc C sô d}t d|dd}tt |¡|dd}tj|jdd
< tj|jd d
|
< | |jd ¡}| |jd ¡}||d ksvt||d kst| t |jd ¡}||d ksªt| |jd ¡}||d ksÊt|jd t
|
¡ }t | |¡¡sðtd S )Nr' r" r# r$ Úfloat)r Zdtypeé é
|
r& é é é é r* r( r ) r r r ÚarangeÚNaNr, r r r Ústrr ÚBDayÚisnan)r r/ r0 r Úval1Úval2r2 Údr r r Ú test_scalar= s zTestSeriesAsof.test_scalarc C s. t dddd}tt t|¡|d}| d¡ ¡ }| |j¡}tddddd d d
|
d
|
d d d d d g t ddddd}t
|
||¡ tj |j d
|
d
< | |j¡}tddddddd
|
d
|
d d d d d g t ddddd}t
|
||¡ tj |j dd
< | |j¡}tddddddd
|
d
|
d d d d dg t ddddd}t
|
||¡ d S )Nz1/1/2000z1/2/2000Z4h©r% r Z2hr é é r( r? r: g @éýÿÿÿg @) r r r r@ ÚlenZresampleZmeanr r ÚtmÚassert_series_equalr+ r, )r r0 ÚsÚrr2 Úexpectedr r r Ú test_with_nanW s. þ þ þz TestSeriesAsof.test_with_nanc C sø ddl m}m} d}|d|dd}ttj |¡|d}tj|jdd
< t d|d
|
d d}|
|
|¡}t | ¡ srt |jd }|jd } |
|
t|¡}t | ¡ s¤t |jd }|jd } ||jjdd }
|
|
|
|k|
|
| k @ } || } | || k ¡ sôt tj|jdd
< tj|jdd
< |
|
|jd ¡} |
|
|jd ¡}| |d ksFt ||d ksXt |
|
t|jd ¡} | |d ks~t |
|
|jd
|
¡|d
|
kst |jd ¡ t ¡ }t|
|
|¡sÆt d}tjt|d |
|
| d¡¡ W 5 Q R X d S )Nr )Ú PeriodIndexÚ period_ranger r" ÚHr$ r r& r' r( Z37minr* rI r: r; r< r= r> r? zInput has different freq©ÚmatchÚD)ÚpandasrT rU r r r r r+ r, r r r r- r r r. ÚvaluesrB Z to_timestampr rC r ÚpytestÚraisesr Zasfreq)r rT rU r/ r0 r r1 r2 r3 r4 Zpixr5 r6 rE rF rG Úmsgr r r Útest_periodindext s@
|
|
|
|
|
z TestSeriesAsof.test_periodindexc C s² t dddgtdtdtdgd}|jjr0ttjtdd | |jd
|
¡ W 5 Q R X d }t d |d d}t t
|
j |¡|d}tjtdd |j|jd
|
dd W 5 Q R X d S )NrJ rK r( r Z20130103Z20130102r zrequires a sorted indexrW r r; r" r# r$ znot valid for SeriesZfoo)Zsubset) r r r Zis_monotonic_increasingr r\ r] Ú
|
ValueErrorr r r r r )r rP r/ r0 r r r Ú test_errors© s þ zTestSeriesAsof.test_errorsc C sÎ d}t d|dd}t d|d dd}ttj|d |¡}ttj|d}t ||¡ t d|d ddd }ttj|d |¡}t|stttjd d
|
dd gd d d dg¡}ttjd dgd d }t ||¡ d S )Nr r" r# r$ r( r) r r rJ rK r? Útest)r Únamer: ) r r r r+ r rN rO r r )r r/ r0 r1 r2 rR Údater r r Ú test_all_nans» s "z TestSeriesAsof.test_all_nansN)
|
Ú__name__Ú
|
__module__Ú __qualname__r! r8 rH rS r_ ra re r r r r r
|
s 5r
|
)Únumpyr r\ Zpandas._libs.tslibsr rZ r r r r r r r Zpandas._testingZ_testingrN r
|
r r r r Ú<module> s
|
$
|
