U
|
®ý°d'Ú ã @ s& d dl Z d dlZd dlZd dlZd dlZd dlmZmZm Z m
|
Z
|
m Z d dl m Z d dlmZ ejddddgdd d
|
ZG d d d Zej d ddidddg¡ej dddgddgg¡dd Zdd Zej d dddddd dd d g¡d d Zd!d" Zd#d$ Zd%d& Zd'd( Z d)d* Z dS )+é N)ÚIndexÚ TimedeltaÚ
|
merge_asofÚread_csvÚ to_datetime)Ú
|
MergeErrorÚsÚmsÚusÚns)Úparamsc C s | j S )z-
|
Resolution for datetimelike dtypes.
|
)Úparam)Úrequest© r úad:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/reshape/merge/test_merge_asof.pyÚunit s r c @ s e Zd ZdddZejdd Zejdd Zejdd Zejd
|
d Z ejd d Z
|
ejdd Z dd Z dd Z dd Zdd Zdd Zdd Zd d Zd d Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- Zd.d/ Zd0d1 Z d2d3 Z d4d5 Z d6d7 Z d8d9 Z d:d; Z!d<d= Z"d>d? Z#d@dA Z$dBdC Z%ej&j'dDe(dEe)j*dFdGgdHdIgdJdKdL Z+dMdN Z,dOdP Z-dQdR Z.dSdT Z/dUdV Z0dWdX Z1dYdZ Z2d[d\ Z3d]d^ Z4d_d` Z5dadb Z6dcdd Z7dedf Z8dgdh Z9didj Z:dkdl Z;dmdn Z<dodp Z=dqdr Z>dsdt Z?dudv Z@dwdx ZAdydz ZBej&j'd{d|d} d~d} gddgdJej& 'dddg¡dd
ZCdd ZDdd ZEdd ZFdd ZGdd ZHdd ZIdd ZJdS )Ú TestAsOfMergeFc C sD |ddd|}t |}|r4|jddgddjdd }t|j|_|S )
|
NZreshapeÚmergeÚdataÚtimeÚtickerÚlast)ZkeepT©Údrop)r Zdrop_duplicatesÚ reset_indexr r )ÚselfÚdatapathÚnameÚdedupeÚpathÚxr r r Ú read_data s ÿ zTestAsOfMerge.read_datac C s | |d¡S )Nz
|
trades.csv©r! ©r r r r r Útrades& s zTestAsOfMerge.tradesc C s | j |dddS )Nz
|
quotes.csvT©r r" r# r r r Úquotes* s zTestAsOfMerge.quotesc C s | |d¡S )Núasof.csvr" r# r r r Úasof. s zTestAsOfMerge.asofc C s | |d¡S )Nz tolerance.csvr" r# r r r Ú tolerance2 s zTestAsOfMerge.tolerancec C s | |d¡S )Nzallow_exact_matches.csvr" r# r r r Úallow_exact_matches6 s z!TestAsOfMerge.allow_exact_matchesc C s | |d¡S )Nz%allow_exact_matches_and_tolerance.csvr" r# r r r Ú!allow_exact_matches_and_tolerance: s z/TestAsOfMerge.allow_exact_matches_and_tolerancec C s t dddgdddgd¡}t ddd d
|
d gddd d
|
d gd ¡}t dddgdddgdd d gd ¡}t||dd}t ||¡ dS )údoc-string examplesé é é
|
ÚaÚbÚc©r0 Úleft_valé é é é ©r0 Ú right_val©r0 r4 r: ©ÚonN©ÚpdÚ DataFramer ÚtmÚassert_frame_equal©r ÚleftÚrightÚexpectedÚresultr r r Útest_examples1> s $ ÿz TestAsOfMerge.test_examples1c C s² |dkrt d¡ tjtdddddg d| d¡ddd d d
|
gd d d d dgdddddgdddddgd}tjtdddddddd g d| d¡d dddd d
|
d dgd d d d d d d d!gd"d#d$d%d"d&d'd(gd)ddd*d+gd}t||ddd, t||ddtd-d. tjtdddddg d| d¡ddd d d
|
gd d d d dgdddddgtj d tj tj tj gtj d$tj tj tj gd/ddddd*d+gd}t||ddtd0d1d2}t
|
||¡ d3S )4r, r zNThis test is invalid for unit='s' because that would round the trades['time']]ú20160525 13:30:00.023z20160525 13:30:00.038ú20160525 13:30:00.048zM8[ú]ÚMSFTÚGOOGÚAAPLçùI@ç\Âõ(@çÂõ(\@ç X@éK é éd )r r ÚpriceÚquantityr r rV rW ©Úcolumnsú20160525 13:30:00.030ú20160525 13:30:00.041ú20160525 13:30:00.049z20160525 13:30:00.072z20160525 13:30:00.075ç @ç\Âõ(üI@ç
ëQ¸þI@çÂõ(\X@gáz®GJ@ç=
|
×£p@ç{®GáúI@ç=
|
×£pýI@ç J@çq=
|
×£X@g×£p=
|
@g¤p=
|
×J@)r r ÚbidÚaskrf rg ©r= ÚbyZ2ms©r= ri r) )r r rV rW rf rg Ú10msF©r= ri r) r* N) ÚpytestÚskipr? r@ r Úastyper r ÚnpÚnanrA rB )r r r$ r& rF rG r r r Útest_examples2J s ÿûÿ
|
ø ô
|
ñøÿ
|
õ ø
|
è
|
å ûÿ
|
ø òïúz TestAsOfMerge.test_examples2c C s t dddgdddgd¡}t ddd d
|
d gddd d
|
d gd ¡}t dddgdddgdd
|
tjgd ¡}t||ddd}t ||¡ dS )r, r- r. r/ r0 r1 r2 r3 r5 r6 r7 r8 r9 r; Úforward©r= Ú directionN©r? r@ rp rq r rA rB rC r r r Útest_examples3£ s $ ÿz TestAsOfMerge.test_examples3c C s t dddgdddgd¡}t ddd d
|
d gddd d
|
d gd ¡}t dddgdddgdd
|
d gd ¡}t||ddd}t ||¡ dS )r, r- r. r/ r0 r1 r2 r3 r5 r6 r7 r8 r9 r; Únearestrt Nr> rC r r r Útest_examples4± s $ ÿz TestAsOfMerge.test_examples4c C s$ |}t ||ddd}t ||¡ d S )Nr r rh ©r rA rB ©r r$ r( r& rF rG r r r Ú
|
test_basic¿ s zTestAsOfMerge.test_basicc C sN |}|j d¡|_ |j d¡|_ |j d¡|_ t||ddd}t ||¡ d S )NÚcategoryr r rh )r ro r rA rB r{ r r r Útest_basic_categoricalÅ s z$TestAsOfMerge.test_basic_categoricalc C sB |}| d¡}t||dddd}|j|_||j }t ||¡ d S )Nr Tr )Ú
|
left_indexÚright_onri )Ú set_indexr ÚindexrY rA rB r{ r r r Útest_basic_left_indexÎ s
|
ÿ
|
z#TestAsOfMerge.test_basic_left_indexc C s0 |}| d¡}t||dddd}t ||¡ d S )Nr Tr )Úleft_onÚ right_indexri ©r r rA rB r{ r r r Útest_basic_right_indexÜ s
|
ÿz$TestAsOfMerge.test_basic_right_indexc C s@ | d¡}| d¡}| d¡}t||dddd}t ||¡ d S )Nr Tr ©r r
ri r r{ r r r Ú!test_basic_left_index_right_indexå s
|
|
|
ÿz/TestAsOfMerge.test_basic_left_index_right_indexc C sF | ddg¡}| d¡}tjtdd t||ddd W 5 Q R X d S )Nr rV z left can only have one index©ÚmatchT©r r
©r rm Úraisesr r ©r r$ r& r r r Útest_multi_index_leftï s
|
z#TestAsOfMerge.test_multi_index_leftc C sF | d¡}| ddg¡}tjtdd t||ddd W 5 Q R X d S )Nr rf z right can only have one indexr Tr r r r r r Útest_multi_index_rightö s
|
z$TestAsOfMerge.test_multi_index_rightc C sH | d¡}| d¡}d}tjt|d t||dddd W 5 Q R X d S )Nr z:Can only pass argument "left_on" OR "left_index" not both.r rV T)r r r
r ©r r$ r& Úmsgr r r Útest_on_and_index_left_oný s
|
|
ÿz'TestAsOfMerge.test_on_and_index_left_onc C sH | d¡}| d¡}d}tjt|d t||dddd W 5 Q R X d S )Nr z<Can only pass argument "right_on" OR "right_index" not both.r rf T)r r r
r r r r r Útest_on_and_index_right_on s
|
|
ÿz(TestAsOfMerge.test_on_and_index_right_onc C s& |}t ||dddd}t ||¡ d S )Nr r )r= Úleft_byÚright_byrz r{ r r r Útest_basic_left_by_right_by s ÿz)TestAsOfMerge.test_basic_left_by_right_byc C sL |}||j dk }t||ddd}tj|j|j dkddgf< t ||¡ d S )NrL r r rh rf rg )r r rp rq ÚlocrA rB )r r$ r( r& rF ÚqrG r r r Útest_missing_right_by s
|
z#TestAsOfMerge.test_missing_right_byc C s` t jtdddddgdddddgdd d d
|
d gd d d d dgdddddgddddddgd}t jtdddddd gddddddgd
|
d ddd dgd d d d d d!gd d"d#d$d%d&gd'dddd(d)gd}t jtdddddgdddddgdd d d
|
d gd d d d dgdddddgtjd d d tjgtjd"d%d tjgd*dddddd(d)gd}t||dddgd+}t ||¡ d S ),NrI ú20160525 13:30:00.046rJ ú20160525 13:30:00.050rL rM rN ÚARCAÚNSDQÚBATSrO rP rQ rR rS rT rU ©r r ÚexchrV rW r r r¢ rV rW rX rZ r[ ú20160525 13:30:00.045r\ ç®Gáz@r^ r_ r] r` rb rc rd ra re ©r r r¢ rf rg rf rg ©r r r¢ rV rW rf rg rh ©r? r@ r rp rq r rA rB ©r r$ r& rF rG r r r Ú test_multiby! sj ûÿ ó ðúÿ
|
ò ïûÿ ñîzTestAsOfMerge.test_multibyc C s` t jtdddddgdddddgdd d d
|
d gd d d d dgdddddgddddddgd}t jtdddddd gddddddgd
|
d ddd dgd d d d d d!gd d"d#d$d%d&gd'dddd(d)gd}t jtdddddgdddddgdd d d
|
d gd d d d dgdddddgtjd d d tjgtjd"d%d tjgd*dddddd(d)gd}t||dddgd+}t ||¡ d S ),NrI r rJ r r r- r5 r r r rO rP rQ rR rS rT rU r¡ r r r¢ rV rW rX rZ r[ r£ r\ r¤ r^ r_ r] r` rb rc rd ra re r¥ rf rg r¦ rh r§ r¨ r r r Ú test_multiby_heterogeneous_typesb sj ûÿ ó ðúÿ
|
ò ïûÿ ñîz.TestAsOfMerge.test_multiby_heterogeneous_typesc C sL t jtdddgtdddgtdddgtdddggddd gd
|
d¡}t jtd ddd gtd ddd gtddddgtddddggddd dgd
|
d¡}t jtdddd gtdddd gtddddgtddddggddd dgd
|
d¡}t||dddd gd}t ||¡ tjt dd t||dddd gdgd W 5 Q R X d S )NZ20160602r- r0 r5 Z20160603r1 r Zk1Zk2rX Z20160502ç ð?ç @Z20160503g @g @ÚvalueTr z(left_by and right_by must be same lengthr )r r
r r )
|
r? r@ r r r rA rB rm r r rC r r r Útest_multiby_indexed£ sd üùø
|
ü
|
ùø
|
ü
|
ùø
|
ÿ ÿúz"TestAsOfMerge.test_multiby_indexedc C sH | |d¡}| |d¡}| j |ddd}t||ddd}t ||¡ d S ) Nz asof2.csvz trades2.csvz quotes2.csvTr% r r rh )r! r rA rB )r r rF r$ r& rG r r r Ú test_basic2Õ s
|
zTestAsOfMerge.test_basic2c C s> dd }||}||}||}t ||dd}t ||¡ d S )Nc S s | | j dk jdddjddS )NrL r r- )ZaxisTr )r r r ©r r r r Ú<lambda>ß s
|
ÿþz0TestAsOfMerge.test_basic_no_by.<locals>.<lambda>r r< rz )r r$ r( r& ÚfrF rG r r r Útest_basic_no_byÝ s ÿz TestAsOfMerge.test_basic_no_byc C s d}t jt|d t||dddd W 5 Q R X t jtdd t||ddgdd W 5 Q R X t jtdd t||dd W 5 Q R X d S )
|
Nz6incompatible merge keys \[1\] .* must be the same typer r rf r )r r ri z can only asof on a key for leftrh )ri ©rm r r r r r r r Útest_valid_join_keysì s z"TestAsOfMerge.test_valid_join_keysc C sL t ||g¡ ddg¡jdd}t||ddd}| |d¡}t ||¡ d S )Nr r Tr rh r' )r? ÚconcatÚ sort_valuesr r r! rA rB )r r r$ r& r rG rF r r r Útest_with_duplicatesø s ÿþÿ z"TestAsOfMerge.test_with_duplicatesc C sz t dddgdddgd¡}t dddgdddgd¡}t||dd}t dddgdddgdddgd¡}t ||¡ d S ) Nr- r6 r5 )Úkeyr4 )r¹ r: r¹ r< )r¹ r4 r: r> ©r Údf1Údf2rG rF r r r Útest_with_duplicates_no_on s ÿz(TestAsOfMerge.test_with_duplicates_no_onc C s4 d}t jt|d t||dddd W 5 Q R X d S )Nz/allow_exact_matches must be boolean, passed foor r r Úfoo©r= ri r* r´ r r r r Ú test_valid_allow_exact_matches s ÿz,TestAsOfMerge.test_valid_allow_exact_matchesc C s t ||ddtdd t | ¡ | ¡ dddd d}tjt|d t ||dddd W 5 Q R X tjt|d t | ¡ | ¡ ddd d W 5 Q R X d
|
}tjt|d t ||ddtd d W 5 Q R X tjt|d t | ¡ | ¡ ddd d W 5 Q R X d S ) Nr r Z1srj r r- z6incompatible tolerance .*, must be compat with type .*r r« ztolerance must be positiveéÿÿÿÿ)r r r rm r r r r r r Útest_valid_tolerance sH û û ÿûz"TestAsOfMerge.test_valid_tolerancec C sà |j ddd}|j ddd}|jjr(t|jjr4ttjtdd t||ddd W 5 Q R X | d¡}|jjstt|jjrttjtdd t||ddd W 5 Q R X | d¡}|jjsÀt|jjsÌtt||ddd d S ) Nr F)Z ascendingzleft keys must be sortedr r rh zright keys must be sorted)r· r Zis_monotonic_increasingÚAssertionErrorrm r Ú
|
ValueErrorr r r r r Útest_non_sortedA s
|
|
z TestAsOfMerge.test_non_sortedÚ tolerance_tsÚ1dayr- )Údaysr zdatetime.timedelta)Úidsc C s& t ||dd|d}|}t ||¡ d S )Nr r rj rz )r rÆ r$ r& r) rG rF r r r Útest_toleranceX s ÿz TestAsOfMerge.test_tolerancec C s t dddgdddgd¡}t ddd d
|
d gddd d
|
d gd ¡}t dddgdddgdtjd gd ¡}t||dddd}t ||¡ d S )Nr- r. r/ r0 r1 r2 r3 r5 r6 r8 é r9 r; rs ©r= ru r) rv rC r r r Útest_tolerance_forwardd s $ ÿz$TestAsOfMerge.test_tolerance_forwardc C s t dddgdddgd¡}t ddd d
|
d gddd d
|
d gd ¡}t dddgdddgdtjd gd ¡}t||dddd}t ||¡ d S )Nr- r. r/ r0 r1 r2 r3 r5 r6 r8 rË r9 r; rx rÌ rv rC r r r Útest_tolerance_nearestq s $ ÿz$TestAsOfMerge.test_tolerance_nearestc C s¸ t t jtdddt d¡|dt d¡d¡}t t jtdddt d¡|dtdd ¡}t ||d
|
t
|
d d }t t jtdddt d¡|dt d¡td d¡}t ||¡ d S )Nz
|
2016-01-02ÚDr. ÚUTC)ÚstartÚfreqÚperiodsÚtzr )ÚdateÚvalue1z
|
2016-01-01ZABCDE)rÕ Úvalue2rÕ z1 day©r= r) ZBCDEE)rÕ rÖ r× ) r? r@ Ú
|
date_ranger ÚpytzÚtimezonerp ZarangeÚlistr r rA rB )r r rD rE rG rF r r r Útest_tolerance_tz~ sH ûøÿ ûøÿ û÷ÿ z TestAsOfMerge.test_tolerance_tzc C s t dddgdddgd¡}t dd d
|
d d gdd d
|
d d gd ¡}t dddgdddgdd
|
tjgd¡}t||dddd}t ||¡ d S )Nçñ?g @gÍÌÌÌÌÌ%@r0 r1 r2 r3 r« ç @gffffff
|
@g @g '@r9 r- r; rx ç à?rÌ rv rC r r r Útest_tolerance_float© s ÿ
|
ýÿz"TestAsOfMerge.test_tolerance_floatc C sF | d¡}| d¡}| d¡}t||dddtdd}t ||¡ d S )Nr Tr rÇ )r r
ri r) )r r r rA rB )r r$ r& r) rF rG r r r Útest_index_tolerance» s
|
|
|
úz"TestAsOfMerge.test_index_tolerancec C s& t ||dddd}|}t ||¡ d S )Nr r Fr¿ rz )r r$ r& r* rG rF r r r Útest_allow_exact_matchesË s ÿz&TestAsOfMerge.test_allow_exact_matchesc C s t dddgdddgd¡}t ddd d
|
d gddd d
|
d gd ¡}t dddgdddgdd
|
d gd ¡}t||dddd}t ||¡ d S )Nr- r. r/ r0 r1 r2 r3 r5 r6 r8 rË r9 r; rs F©r= ru r* r> rC r r r Ú test_allow_exact_matches_forwardÒ s $ ÿ ÿz.TestAsOfMerge.test_allow_exact_matches_forwardc C s t dddgdddgd¡}t ddd d
|
d gddd d
|
d gd ¡}t dddgdddgdd d gd ¡}t||dddd}t ||¡ d S )Nr- r. r/ r0 r1 r2 r3 r5 r6 r8 rË r9 r; rx Frä r> rC r r r Ú test_allow_exact_matches_nearestá s $ ÿ ÿz.TestAsOfMerge.test_allow_exact_matches_nearestc C s, t ||ddtddd}|}t ||¡ d S )Nr r Z100msFrl )r r rA rB )r r$ r& r+ rG rF r r r Ú&test_allow_exact_matches_and_toleranceð s úz4TestAsOfMerge.test_allow_exact_matches_and_tolerancec C sæ t tdgdgd¡}t tddgddgd¡}t||dd }t tdgdgdgd
|
¡}t ||¡ t||dd d }t tdgdgdgd
|
¡}t ||¡ t||dd td d}t tdgdgtjgd
|
¡}t ||¡ d S )Nú2016-07-15 13:30:00.030Úbob©r Úusernameú2016-07-15 13:30:00.000r- r5 ©r Úversionr r< ©r rë rî F)r= r* rk ©r= r* r) ) r? r@ r r rA rB r rp rq rº r r r Ú'test_allow_exact_matches_and_tolerance2þ sP ÿÿüÿ ýÿ ýÿ ûýÿz5TestAsOfMerge.test_allow_exact_matches_and_tolerance2c C s t tddgddgd¡}t tddgddgd¡}t||d d
|
td d }t tddgddgtjtjgd ¡}t ||¡ d S )Nrè ré Zcharlierê rì r- r5 rí r Frk rð rï ) r? r@ r r r rp rq rA rB rº r r r Ú'test_allow_exact_matches_and_tolerance30 s< ÿüÿÿüÿ ûÿ
|
ûÿ z5TestAsOfMerge.test_allow_exact_matches_and_tolerance3c C s t dddgdddgd¡}t ddd d
|
d gddd d
|
d gd ¡}t dddgdddgtjd
|
d gd ¡}t||ddddd}t ||¡ d S )Nr- r. r/ r0 r1 r2 r3 r6 é r7 rË r9 r; rs F©r= ru r* r) rv rC r r r Ú.test_allow_exact_matches_and_tolerance_forwardU s $ ÿúz<TestAsOfMerge.test_allow_exact_matches_and_tolerance_forwardc C s t dddgdddgd¡}t ddd d
|
d gddd d d gd ¡}t dddgdddgtjd d gd¡}t||ddddd}t ||¡ d S )Nr- r. r/ r0 r1 r2 r3 r6 ró r7 rË r8 r9 r; rx Frô rv rC r r r Ú.test_allow_exact_matches_and_tolerance_nearesti s $ ÿúz<TestAsOfMerge.test_allow_exact_matches_and_tolerance_nearestc
|
C sÀ t dddddgdddddgd d
|
d d d gd¡}t dddddgdddddgdddddgd¡}t dddddgdddddgd d
|
d d d gdtjdddgd¡}t||d d
|
dd}t ||¡ d S )Nr- r. r/ é é ÚXÚYÚZr0 r1 r2 ÚdÚe©r0 r1 r4 r7 rË é ©r0 r1 r: ©r0 r1 r4 r: rs ©r= ri ru rv rC r r r Útest_forward_by} s* ýÿ ýÿ üÿ z TestAsOfMerge.test_forward_byc
|
C s¾ t dddddgdddddgd d
|
d d d gd¡}t dddddgdddddgdddddgd¡}t dddddgdddddgd d
|
d d d gdddddgd¡}t||d d
|
dd}t ||¡ d S )Nr- r. r/ r÷ rø rù rû rú r0 r1 r2 rü rý rþ r7 rË rÿ r r rx r r> rC r r r Útest_nearest_by s* ýÿ ýÿ üÿ z TestAsOfMerge.test_nearest_byc C sú t jtdddddgdddddgd d
|
d d d gddddgd}t jtddddddddgddddddddgddddd d d d gd ddd!gd}t||ddd"}t jtdddddgdddddgd d
|
d d d gddddd gd#dddd!gd}t ||¡ d S )$Nz20160525 13:30:00.020rZ z20160525 13:30:00.040r z20160525 13:30:00.060r- r5 r6 rÞ g333333ó?gÍÌÌÌÌÌô?gffffffö?g ø?)r r¹ rÖ r r¹ rÖ rX z20160525 13:30:00.015z20160525 13:30:00.025z20160525 13:30:00.035z20160525 13:30:00.055z20160525 13:30:00.065gÍÌÌÌÌÌ @g@gffffff@g333333@rß gÍÌÌÌÌÌ@g@gffffff@)r r¹ r× r× rh )r r¹ rÖ r× )r? r@ r r rA rB rº r r r Ú test_by_int¹ s` ûÿ õ òøÿ òïûÿ ô
|
ñzTestAsOfMerge.test_by_intc
|
C sº t jdddddddgtdd d
|
d gd }t jd ddgdddgdd dgd }| d ¡jdd}t||d d}t jtddddddddgdddddddgdd
|
d dgd }t ||¡ d S )Ng
|
×£p=
|
@gHPüs×b?gáz®G!9@gÍÌÌÌÌ@u@gHáz®Ç>@gC@gÞqäò?ÚABCDEFG)rV Úsymbolr rV rX ç r« ç Y@ç-C ëâ6?ç{®Gáz?ç©?)rV Úmpvr Tr r< ZBGACEDF)r rV r ©r? r@ rÜ r· r r rA rB rº r r r Ú test_on_floatö s( þûþýú zTestAsOfMerge.test_on_floatc C sà t |¡j}tjdddddddgtdd d
|
d gd }||j|_tjd dddgtddd dgd }||j|_| d ¡jdd}t ||d d}tjtddddddddgtddd
|
d dgd }||j|_t
|
||¡ d S )Nr. r5 é rU éN éx éO r )r r r r rX r éP é} Úxyzw)r rG rG Tr r< ÚBACEGDFZxxxxxyz)r r rG ) rp ÚdtypeÚtyper? r@ rÜ r r· r r rA rB ©r Zany_real_numpy_dtyper r» r¼ rG rF r r r Útest_on_specialized_type s, þ þ ýú z&TestAsOfMerge.test_on_specialized_typec C s$ t |¡j}tjdddddddgddd dd ddgtd
|
d d d dgd}||j|_tjddddgdddd gtdddd dgd}||j|_| d¡jdd}t ||dd d}tjtdddd d dddgdddddddgt j
|
dt j
|
t j
|
t j
|
ddgd d d ddgd}||j|_t ||¡ d S ) Nr. r5 r rU r r r r- r6 r )r r¹ r r r¹ r rX r r r r )r r¹ rG rG Tr rh r r Úy)r r¹ r rG ) rp r r r? r@ rÜ r r· r r rq rA rB r r r r Ú test_on_specialized_type_by_int3 s4 ýú þ ü
|
ù z-TestAsOfMerge.test_on_specialized_type_by_intc C s4 t jtddddddddddg ddddd d
|
d d d g ddddgd}t jdddddddddg dddddddddg dddddddd dg d ddd gd}| d¡jd d }| d¡jd d }t||ddd!}t jtddddddddddg dddd
|
dd d d d g dddd ddddd g d"dddd gd}t ||¡ d S )#NZ AAABBBCCCr- r5 r6 g®Gáz
|
@gÜ×sF
|
@g£¼
|
@g)\Âõ()@g®Gáz.)@g )@gfffff¢w@g33333£w@g ¤w@)r r¢ rV r r¢ rV rX r r« r ç @g @@r
|
r r g¹?g Ð?)r¢ rV r r Tr rh )r r¢ rV r r rº r r r Útest_on_float_by_intW sX ÷ýðýú ÷ ò
|
ïz"TestAsOfMerge.test_on_float_by_intc C sp d}t dddgdddgd¡}t dd d
|
d d gdd d
|
d d gd ¡}tjt|d t||dd W 5 Q R X d S )Nú@Incompatible merge dtype, .*, both sides must have numeric dtyper- r. r/ r0 r1 r2 ©r4 r0 r5 r6 r7 r8 ©r: r0 r r< ©r? r@ rm r r r ©r r rD rE r r r Ú test_merge_datatype_error_raises s
|
$z.TestAsOfMerge.test_merge_datatype_error_raisesc
|
C s| d}t dddgt dddg¡d¡}t dd d
|
d d gt dd dd dg¡d¡}tjt|d t||dd W 5 Q R X d S )NzHincompatible merge keys \[0\] .* both sides category, but not equal onesr- r. r/ r0 r1 r2 r! r5 r6 r7 r8 rù r" r r< )r? r@ Ú Categoricalrm r r r r$ r r r Ú,test_merge_datatype_categorical_error_raises s ÿÿ þÿz:TestAsOfMerge.test_merge_datatype_categorical_error_raisesc C s` t dgdgt dg¡d¡}t||dddgd}t dgdgt dg¡d¡}t ||¡ d S )Nr )r r Úzr r r( rh )r? r@ r& r rA rB )r ÚdfrG rF r r r Ú:test_merge_groupby_multiple_column_with_categorical_column¬ s zHTestAsOfMerge.test_merge_groupby_multiple_column_with_categorical_columnÚfuncc C s | S ©Nr r° r r r r± ´ ó zTestAsOfMerge.<lambda>c C s t | S r, )r r° r r r r± ´ r- ÚnumericÚdatetimeÚsiderD rE c C s d| d}|ddt jg}|dddg}t |dddgd ¡}t |d
|
d d gd ¡}tjt|d, |dkr~t||dd nt||dd W 5 Q R X d S )Nz"Merge keys contain null values on z sider« r g $@r0 r1 r2 r3 r- r7 rË r9 r rD r< )rp rq r? r@ rm r rÄ r )r r+ r0 r ZnullsZ non_nullsZdf_nullr) r r r Útest_merge_on_nans³ s z TestAsOfMerge.test_merge_on_nansc C s t jt jddg|d}|jjdkr4t |jj¡j}nt |jj¡j}||d< t
|
|dddgdd d
|
gd d d gd¡}t
|
|dddgddd
|
gdddgd¡}t ||ddgdd}t
|
|dddgdd d
|
gd d d gd¡}tjtj tj tj gt d|d< t ||¡ d S )Nr r- ©r )ÚiÚur5 ÚHELLOÚToZYouró r7 r0 r2 rý ©Úby_col1Úby_col2Úon_colr ÚWORLDZWideZWebr1 rü r² r8 r9 r: ©ri r= )r8 r9 r: Úvalue_xÚvalue_y)r? ÚarrayZNAr Úkindrp ZiinfoZ numpy_dtypeÚmaxZfinfor@ r rq ÚobjectrA rB )r Zany_numeric_ea_dtypeZarrZmax_valrD rE rG rF r r r Útest_by_nullableÅ s: üÿüÿ üÿ z TestAsOfMerge.test_by_nullablec C s t t dg¡ d¡dgdgd¡}t t dg¡ d¡dgdgd¡}t||dd d
|
}t jt jddd dddggdd d d gd}t ||¡ d S )Nú
|
2018-01-01rÐ r5 r0 )Úby_colr: Úvaluesr- r1 rE r: r< ©rÔ Zvalues_xZvalues_yrX )r? r@ Ú DatetimeIndexÚ tz_localizer Ú TimestamprA rB ©r rD rE rG rF r r r Útest_merge_by_col_tz_awareñ s$ ýÿýÿ
|
þz(TestAsOfMerge.test_merge_by_col_tz_awarec C s´ t t dg¡ d¡dgdgdgd¡}t t dg¡ d¡dgdgd gd¡}t||d
|
d gd d }t jt jddddddggd
|
d d dgd}tjtjgt d|d< t
|
||¡ d S )NrD rÐ r5 r5 r0 r7 r; r- r1 r8 r9 r: r< rG r= rX r2 r> ) r? r@ rH rI r rJ rp r? rq rB rA rB rK r r r Útest_by_mixed_tz_aware s* üÿüÿ
|
þz$TestAsOfMerge.test_by_mixed_tz_awarec C sP |dkrt d¡ tjttddddddgdd d
|
d d dgd dgd}t |d d¡ d| d¡|d < tjttdd ddddgdd d
|
d d dgd dgd}t |d d¡ d| d¡|d < tjttddddddgdd d
|
d d dgdtj d
|
d tj tj gd ddgd}t |d d¡ d| d¡|d < t
|
||d t ddd}t ||¡ d S )Nr zHThis test is invalid with unit='s' because that would round left['time']r r. r/ rø é r r- r5 r6 ró r rD rX r zm8[rK é r÷ é rE Z1msrx )r= r) ru )rm rn r? r@ rÜ ÚzipZ to_timedeltaro rp rq r r rA rB )r r rD rE rF rG r r r Ú test_timedelta_tolerance_nearest" sB ÿ$þ"$þ"ýÿø " ÿz.TestAsOfMerge.test_timedelta_tolerance_nearestc C s´ t dddgdddgd¡}t dd d
|
gdddgd ¡}|d |¡|d < |d |¡|d < t dddgdddgtjd dgd¡}|d |¡|d < t||d dd}t ||¡ d S )Nr r/ rN r- r5 r6 r3 r. rø r r9 r0 r« r¬ r; rØ )r? r@ ro rp rq r rA rB )r Zany_int_numpy_dtyperD rE rF rG r r r Útest_int_type_toleranceL s ÿz%TestAsOfMerge.test_int_type_tolerancec C s t jddddd}t jdddd gd
|
g|d d
d }t |d gd dg d¡}t||ddgd}t jdddd g|d d
d gd dg dt jdddddd}t ||¡ t||ddgd}t j|d gd dg tjdddd gdtdd dddgd}t ||¡ d S ) Nz
|
2019-10-01Z30minr. rÐ )rÒ rÓ rÔ gÍÌÌÌÌÌì?gé?gffffffæ?g333333ã?Úxyzr- )rY r g®Gáz®@ró g
ëQ¸
@)Ú from_dateÚabcTrU )rD rE r r r6 )rT rU rV z2019-10-01 00:30:00©r )rD rE r
r )rU rV rT r r5 ) r? rÙ r@ r rA rB rp rq r )r r rD rE rG rF r r r Útest_merge_index_column_tz\ sH " ÿ
|
|
ý ÿú
|
ÿýúz(TestAsOfMerge.test_merge_index_column_tzc C s¶ |dkrt d¡ tjddd|dtdd |¡ }tjdd |d
|
}tjd d it |¡d }tjddit |¡d }tjd ddt |¡d }t||ddtddd}t
|
||¡ d S )Nr z?This test is invalid with unit='s' because that would round dr1z1/1/2020z 1/20/2020Z2D)rÑ ÚendrÒ r gÙ?)Úsecondsz2/1/2020)rÑ rY r Úval1r¾ rW Úval2Úbar)r[ r\ Trà )r r
r) ) rm rn r? rÙ r Zas_unitr@ rH r rA rB )r r Zdr1Zdr2r» r¼ rF rG r r r Ú%test_left_index_right_index_tolerance} s6 ÿ ÿþ ÿûz3TestAsOfMerge.test_left_index_right_index_toleranceN)F)KÚ__name__Ú
|
__module__Ú __qualname__r! rm Úfixturer$ r& r( r) r* r+ rH rr rw ry r| r~ r r r r r r r r r r© rª r® r¯ r³ rµ r¸ r½ rÀ r rÅ ÚmarkÚ parametrizer r/ Ú timedeltarÊ rÍ rÎ rÝ rá râ rã rå ræ rç rñ rò rõ rö r r r r r r r r% r' r* r1 rC rL rM rR rS rX r^ r r r r r s¤
|
|
|
|
|
|
|
|
Y
|
|
AA2
|
.ý
|
+2% = $9 ÿ ,*!r Úkwargsr= r Tr r z2019-06-01 00:09:12z2019-06-01 00:10:29r« c C sP t jd|i|d}t jd|i|d}tjtdd t||f| W 5 Q R X d S )Nr rW r r r# )rf r rD rE r r r Ú#test_merge_asof_non_numerical_dtype s þrg c C sj t dddgdddgd¡} t dddgdd d
|
gd ¡}tjtd d t| |ddddd W 5 Q R X d S )NZ12Z13Z15r0 r1 r2 )r0 Ú left_val1rü rý r² r3 r r rh r4 )r r r r r# ©rD rE r r r Ú*test_merge_asof_non_numerical_dtype_object« s þúrj )r
r Ú left_time)r r
rE )r r c C s t dddgdd}tjdddgdd dgd
|
|d }tjd dd dddgidd dddgd }t||f| }tjdddgdd dgdddgd|d }t ||¡ d S )Nr- r. r/ Útest©r r0 r1 r2 ró )rD rk rW rE r5 r6 r7 r8 )rD rk rE )r r? r@ r rA rB )rf r rD rE rG rF r r r Ú test_merge_asof_index_behavior½ s
|
& þrn c C s t jddddgitdddgdd d
|
} t jd d d dgitdddgdd d
|
}t| |ddd}t dddgdddgd d dgd¡}t ||¡ d S )Nr1 r/ rË r÷ r- r5 r6 r0 rm rW r2 rN é é r ©r r ©r0 r1 r2 )r? r@ r r rA rB ©rD rE rG rF r r r Ú&test_merge_asof_numeri_column_in_indexÓ s
|
&&$rt c C s´ t jddddgit jjdddgddd ggdd
|
gd d } t jd d ddgit jjdddgddd
|
ggddgd d }t| |ddd}t dddgdddgd ddgd¡}t ||¡ d S )Nr1 r/ rË r÷ r- r5 r6 r0 r2 r( )ÚnamesrW rN ro rp r r rq rr )r? r@ Z
|
MultiIndexZ from_arraysr rA rB rs r r r Ú+test_merge_asof_numeri_column_in_multiindexÝ s "þ "þ$rv c C sÈ t jddddgitdddgdd d
|
} t jd d d dgitdddgdd d
|
}tjtdd t| |ddd W 5 Q R X | ¡ ddg¡} | ¡ dd g¡}tjtdd t| |ddd W 5 Q R X d S )Nr1 r/ rË r÷ Ú1Ú2Ú3r0 rm rW r2 rN ro rp ÚmÚnÚor r rq ) r? r@ r rm r r r r r ri r r r Ú3test_merge_asof_numeri_column_in_index_object_dtypeí s &&þþr} c C sÜ t ddgt d¡t d¡gd¡} t jt d¡ddd }t d
|
d d d gi¡}t|| |dddd}t d d d gtjddg|d¡}t ||¡ t| |d|ddd}t ddgt d¡t d¡gd d gd¡}t ||¡ d S )Nr5 r7 z2021/01/01 00:37z2021/01/01 01:40)r0 Útsz2021/01/01 00:00r6 Z1h)rÑ rÓ rÒ r1 ró é r8 r~ FZbackward)r r r* ru )r1 r0 r~ )r0 r~ r1 ) r? r@ rJ rÙ r rp rq rA rB )rE Zts_mergerD rG rF r r r Útest_merge_asof_array_as_on sH þÿ ÿú úýÿr c C s¸ t jdddggdddgd} t jdddggdddgd}tjtdd t| |dd W 5 Q R X tjtdd t| |ddd
|
W 5 Q R X tjtdd t| |ddd
|
W 5 Q R X d S ) Nr- r5 r0 r4 rX r: zcolumn label 'a'r r< rq )r? r@ rm r rÄ r ri r r r Ú+test_merge_asof_raise_for_duplicate_columns+ s r ) r/ Únumpyrp rm rÚ Zpandasr? r r r r r Zpandas._testingZ_testingrA Zpandas.core.reshape.merger rb r r rc rd rg rj rn rt rv r} r r r r r r Ú<module> sX
|
ÿþ ýþ
|
|
)
|
