U
|
ý°d ã @ sl d dl Zd dlZd dlmZ d dlmZmZmZm Z m
|
Z
|
m Z m Z d dl mZ ejdd ZG dd dZdS )é N)ÚIncompatibleFrequency)Ú DataFrameÚPeriodÚSeriesÚ TimestampÚ
|
date_rangeÚ period_rangeÚ to_datetimec C s0 d} t d| dd}tt | ¡t | ¡d|dS )zZ
|
Fixture for DataFrame of ints with date_range index
|
|
Columns are ['A', 'B'].
|
é2 ú1/1/1990Z53s©ÚperiodsÚfreq)ÚAÚB©Úindex)r r ÚnpZarange)ÚNÚrng© r ú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_asof.pyÚdate_range_frame s r c @ s e Zd Zdd Zdd Zdd Zdd Zd d
|
Zej
|
d e d e d e d dfe de de ddfg¡dd Z dd Zdd Zdd ZdS )Ú TestFrameAsofc
|
C sÐ | ddi¡}d}tj|j|jdd
df< td|d dd }| |¡}| ¡ d
|
¡ ¡ s^t |jd }|jd }t
|
|}| |¡}| ¡ d
|
¡ ¡ st |j|k|j|k @ }|| } | d k d
|
¡ ¡ sÌt d S ) Nr Úfloatr
|
é é r é Ú25sr é é ) Úastyper ÚnanÚlocr r ÚasofZnotnaÚallÚAssertionErrorÚlist)
|
Úselfr Údfr ÚdatesÚresultZlbZubÚmaskÚrsr r r Ú
|
test_basic s
|
|
|
|
zTestFrameAsof.test_basicc C sô d}|j d |
¡ ddi¡}tj|j|jdd
df< td|d dd }|j|dd
|
}| |¡}t
|
||¡ |j|dd gd
|
}| |¡}t
|
||¡ |j|d d
|
}|j dd d ¡ |¡}d|j dd
< |d |d j¡|d < t
|
||¡ d S )Né
|
r r é é r r r r )Zsubsetr Úright)Úclosedé é )ÚilocÚcopyr! r r" r# r r r$ ÚtmÚassert_frame_equalZresampleZffillZreindexÚdtype)r( r r r) r* r+ Úexpectedr r r Ú test_subset4 s
|
|
zTestFrameAsof.test_subsetc C s d}|j d |
¡ }| d¡}tddgtdtjd}t ||¡ | t dg¡}t
|
t dgddgdd}t ||¡ | d¡}| d¡}t |jtstd S ) Nr/ z
|
1989-12-31r r )r Únamer: Úfloat64)r Úcolumnsr: ÚD)r6 r7 r$ r r r r> r8 Úassert_series_equalr r r9 Z to_periodÚ
|
isinstancer= r r& )r( r r r) r+ r; r r r Ú test_missingN s&
|
ÿ ÿ
|
|
zTestFrameAsof.test_missingc C s0 |t jg dg¡}|t jg}t ||¡ d S )Nr )r r" r$ r8 Z assert_equal)r( Zframe_or_seriesr+ r; r r r Útest_asof_all_nansg s z TestFrameAsof.test_asof_all_nansc C s d}|j }td|dd}ttj|dgd |¡}ttj|dgd}t ||¡ td|dd}ttj|dddgd |¡}ttj|dddgd}t ||¡ ttjd d
|
gddgd d g¡}ttjd gddgd}t ||¡ ttjd d
|
gddgd d ¡}ttjddgd d }t ||¡ d S ) Né r r r r )r r? r ÚCr é r )r r= )
|
r r r r r" r$ r8 r9 r rA )r( r r r r* r+ r; r r r Ú test_all_nansn s zTestFrameAsof.test_all_nanszstamp,expectedz 2018-01-01 23:22:43.325+00:00rG )r= z 2018-01-01 22:33:20.682+01:00r c C s6 t ddgtdtdgd}| |¡}t ||¡ d S )Nr rG z 2018-01-01 21:00:05.001+00:00z 2018-01-01 22:35:10.550+00:00)Údatar )r r r$ r8 rA )r( Zstampr; r) r+ r r r Útest_time_zone_aware_index s þþ
|
z(TestFrameAsof.test_time_zone_aware_indexc C sj | ddi¡}d}tj|j|jdd
df< td|d dd }| |¡}t d ¡ d
|
|d < W 5 Q R X d S ) Nr r r
|
r r r r r r r rF ) r! r r" r# r r r$ r8 Zassert_produces_warning)r( r r) r r* r+ r r r Ú test_is_copy¥ s
|
zTestFrameAsof.test_is_copyc C sX d}t d|dd}ttj |¡|d}d}tjt|d | | d¡¡ W 5 Q R X d S ) Nr
|
r ÚHr r zInput has different freq)Úmatchr@ )
|
r r r ÚrandomZrandnÚpytestZraisesr r$ Zasfreq)r( r r r) Úmsgr r r Ú%test_asof_periodindex_mismatched_freq² s z3TestFrameAsof.test_asof_periodindex_mismatched_freqc C s\ t dddd}tdddg|d d
d}|d }| |g¡}tdg|gd}t ||¡ d S ) Nz
|
2017-01-01ZMSr0 )r r TFéÿÿÿÿr )r r r$ r8 rA )r( ZdtiZserÚtsÚresr; r r r Ú test_asof_preserves_bool_dtype¼ s z,TestFrameAsof.test_asof_preserves_bool_dtypeN)Ú__name__Ú
|
__module__Ú __qualname__r. r< rC rD rH rO ÚmarkZ parametrizer r rJ rK rQ rU r r r r r s& þþûþ
|
|
r )Únumpyr rO Zpandas._libs.tslibsr Zpandasr r r r r r r Zpandas._testingZ_testingr8 Zfixturer r r r r r Ú<module> s $
|
|
