U
|
¬ý°ds8 ã @ s d dl Zd dlZd dlm mZ d dlZd dlm Z m
|
Z
|
m Z m Z m Z mZmZ d dlmZ d dlmZ eje d ddge dddgfe d d
|
d ge d
|
d d gfed ddedddfedddedddfeddddeddddfgdd ddd ZG dd dZG dd d ZG d d d Zd d ZG d!d" d"Ze d#¡Z e d$d% Z e d&d' Z e d(d) Z e ej!j"d*d d
|
d d ged+dd,d-gd.d/gd0d1d2 Z#e ej!j"d*d d
|
d d ged+dd,d-gd.d/gd0d3d4 Z$e d5d6 Z%ej! "d7d8d9d:g¡d;d< Z&dS )=é N)ÚIndexÚIntervalÚ IntervalIndexÚ TimedeltaÚ TimestampÚ
|
date_rangeÚtimedelta_range)Ú IntervalArrayé é é é é ç ç ð?ç @ç @ú0 days©Úperiodsú1 dayZ20170101Z20170102ú
|
US/Eastern©r Útzc C s t | d jS )Nr )ÚstrÚdtype)Úx© r úad:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/interval/test_interval.pyÚ<lambda> ó r )ÚparamsÚidsc C s | j S )zC
|
Fixture for building an IntervalArray from various dtypes
|
)Úparam)Úrequestr r r Úleft_right_dtypes s r% c @ sh e Zd Zej ddededfededfedddedddfg¡ej d ee g¡d
|
d Z
|
d S ) ÚTestAttributesz left, right©r r r r z
|
2018-01-01z
|
2018-01-02r )r Ú constructorc C sH ||f||ft jg}t |dkddg¡}|j||dj}t ||¡ d S )NÚbothF©Úclosed)ÚnpÚnanÚarrayÚ from_tuplesZis_emptyÚtmZassert_numpy_array_equal)Úselfr( ÚleftÚrightr+ ZtuplesÚexpectedÚresultr r r Ú test_is_empty) s z TestAttributes.test_is_emptyN) Ú__name__Ú
|
__module__Ú __qualname__ÚpytestÚmarkÚ parametrizer r r r r6 r r r r r& ( s
|
|
þüþ r& c @ sp e Zd Zej dddddg¡dd Zej ded d
|
dd ej d
|
d d dgdd g¡dd Z
|
dd Z dd Z dS )Ú TestMethodsÚ
|
new_closedr2 r3 r) Úneitherc C s> t jtd|d}| |¡}t jtd|d}t ||¡ d S )Né
|
r* )r Ú from_breaksÚrangeÚ
|
set_closedr0 Ú assert_extension_array_equal)r1 r+ r> r. r5 r4 r r r Útest_set_closed? s
|
zTestMethods.test_set_closedÚotherr r r* r
|
r r c C s t tjddddgdd¡}t dddg¡}d }tjt|d
|
|j ||¡ W 5 Q R X |j
|
||d }| t ¡
|
||¡}t ||¡ d S ) Nr r
|
r r r2 r* TFz+'value.closed' is 'right', expected 'left'.©Úmatch)rF )ÚpdZSeriesr rA r, r. r: ÚraisesÚ
|
ValueErrorZ_whereÚwhereÚastypeÚobjectr0 Zassert_series_equal)r1 rF ZserÚmaskrH Úresr4 r r r Útest_where_raisesG s z TestMethods.test_where_raisesc C sl t dddg¡}| ¡ }t tjtjfdg¡}t ||¡ d}tj t
|
|d |jdt j d W 5 Q R X d S )Nr r
|
r )r r ú=can only insert Interval objects and NA into an IntervalArrayrG ©Z
|
fill_value) r rA Úshiftr/ r, r- r0 Úassert_interval_array_equalr: rJ Ú TypeErrorrI ÚNaT©r1 Úar5 r4 Úmsgr r r Ú
|
test_shiftZ s zTestMethods.test_shiftc C s t tddd¡}| d¡}|jdddgdd}t ||¡ | d¡}|jd ddgdd}t ||¡ d
|
}tjt |d |jd t
|
d d ¡d W 5 Q R X d S )NÚ2000r r r
|
éÿÿÿÿr T)Z
|
allow_fillr rR rG rW ÚnsrS ) r rA r rT Útaker0 rU r: rJ rV r, Z timedelta64rX r r r Útest_shift_datetimef s
|
|
z TestMethods.test_shift_datetimeN) r7 r8 r9 r: r; r< rE r r rA rQ r[ r` r r r r r= > s
|
þþ
|
r= c @ s e Zd Zdd Zdd ZdS )Ú TestSetitemc C sú |\}}|j dd}|j dd}t ||¡}|jjjdkrbd}tjt|d t j
|
|d< W 5 Q R X |jjjdkrd}tjt|d t j |d< W 5 Q R X d S t j |d< t|jgt|d d
}t|jgt|d d
}t ||¡}t ||¡ d S )
|
NT©Údeep)ÚmÚMz='value' should be an interval type, got <.*NaTType'> instead.rG r )ÚiÚuz4Cannot set float NaN to integer-backed IntervalArrayr )Úcopyr Ú from_arraysr ÚsubtypeÚkindr: rJ rV rI rW r, ÚNaNr- r Z _na_valueÚlistr0 rD ) r1 r% r2 r3 r5 rZ Z expected_leftZexpected_rightr4 r r r Ú test_set_naw s$
|
zTestSetitem.test_set_nac C s t td¡}| ¡ }| d¡}d}tjt|d |d |d< W 5 Q R X tjt|d |d d
|d d
< W 5 Q R X tjt|d |d d
|d d
< W 5 Q R X tjt|d |d d d
|d d
< W 5 Q R X tjt|d t|d d d
|d d
< W 5 Q R X tjt|d" |d d d
t
|
¡|d d
< W 5 Q R X tjt|d" |d d d
d¡|d d
< W 5 Q R X g |d d
< t ||¡ d S ) Nr r) z*'value.closed' is 'both', expected 'right'rG r r r] Úcategory) r rA rB rh rC r: rJ rK rm rM rN r0 rU )r1 ÚarrÚorigrF rZ r r r Ú test_setitem_mismatched_closed s(
|
$&& z*TestSetitem.test_setitem_mismatched_closedN)r7 r8 r9 rn rr r r r r ra v s ra c C s* t ddg¡} t| }d}||ks&td S )Nr' )r r
|
zI<IntervalArray>
|
[(0, 1], (1, 2]]
|
Length: 2, dtype: interval[int64, right])r r/ ÚreprÚAssertionError)rp r5 r4 r r r Ú test_repr« s
|
ÿru c @ s e Zd Zdd Zdd ZdS )ÚTestReductionsc
|
C sÚ |\}}|j dd}|j dd}t ||¡}d}dD ]P}tjt|d |j|d W 5 Q R X tjt|d |j|d W 5 Q R X q4d}tjt|d |jdd W 5 Q R X tjt|d |jdd W 5 Q R X d S ) NTrb z2`axis` must be fewer than the number of dimensions)éþÿÿÿr rG )Úaxisz'>=' not supported betweenZfoo) rh r ri r: rJ rK ÚminÚmaxrV )r1 r% r2 r3 rp rZ rx r r r Útest_min_max_invalid_axis¸ s z(TestReductions.test_min_max_invalid_axisc C s |\}}|j dd}|j dd}t ||¡}|js6tt|jsDt|d }|d }t t|¡}tj
|
|¡ | |¡}| dtj ¡} ||}|| } dD ]\}
|
|j|
|
d} | |ks¶tt| t|ksÊt|j|
|
d} | |ksâtt| t|kstq| jdd} t | ¡st| jdd} t | ¡s0t| jdd} | |ksJtt| t|ks`t| jdd} | |ksztt| t|kstd S ) NTrb r r] r
|
)TF)ÚskipnaF)rh r ri Zis_monotonic_increasingrt r r, ZarangeÚlenÚrandomÚshuffler_ Úinsertr- ry Útyperz Úisnan) r1 r% Zindex_or_series_or_arrayr2 r3 rp ZMINÚMAXZindexerZarr_nar| rP r r r Ú test_min_maxË s>
|
|
zTestReductions.test_min_maxN)r7 r8 r9 r{ r r r r r rv · s rv Úpyarrowc C s dd l } ddlm} || ¡ d}|| ¡ d}|| ¡ d}|jdksLt||ksXt||ksdtt|t|ksxtt|t|kstd S )Nr ©ÚArrowIntervalTyper2 r3 )r
Ú(pandas.core.arrays.arrow.extension_typesr Úint64r+ rt Úhash)Úpar Úp1Úp2Zp3r r r Útest_arrow_extension_typeÿ s r c C s\ dd l } ddlm} tjddddj}| |¡}t|j|s@t|jj |j ksRt|jj
|
| ¡ ksft|j d¡ | jdddd gd
|
d ¡st|j d ¡ | jddd dgd
|
d ¡s¶t| d d tddD ¡}|j |¡sàt| j||jd }| |¡sþttjtdd | j|dd W 5 Q R X tjtdd | j||| ¡ dd W 5 Q R X d S )Nr r r r )Úfreqr2 r
|
r r r ©r r3 c S s g | ]}||d dqS )r ©r2 r3 r )Ú.0rf r r r Ú
|
<listcomp> s z$test_arrow_array.<locals>.<listcomp>z&Not supported to convert IntervalArrayrG Úfloat64)r
r r rI Zinterval_ranger. Ú
|
isinstancer rt r+ rj r ÚstorageÚfieldÚequalsrB r: rJ rV r )r r Z intervalsr5 r4 r r r Útest_arrow_array s"
|
((r c C s dd l } ddlm} t ddddg¡}d |d< | |¡}t|j|sHt|jj |j ksZt|jj
|
| ¡ ksnt| jdd dgdd }| jdd dgdd }|j d
|
¡ |¡s¬t|j d ¡ |¡sÂtddd d d d ddd g}| jj|t d dd g¡d}|j |¡s td S )Nr r r r r r r r r r2 r3 r FT)rO )r
r r r rA r. r r rt r+ rj r r r r Z StructArrayZ from_pandasr, )r r rp r5 r2 r3 Úvalsr4 r r r Útest_arrow_array_missing. s$
|
ý r ÚbreaksZ2017ÚD)r r Úfloatzdatetime64[ns])r" c C sö dd l }ddlm} t | ¡}d |d< t d|i¡}| |¡}t| d¡j
|
|sTt | ¡ }t|d j tjsrt t ||¡ | ||g¡}| ¡ }tj||gdd}t ||¡ |j|jg | d¡j
|
dg|jd}| ¡ }t ||dd
¡ d S ) Nr r r rY T)Z ignore_indexr )Úschema)r
r r r rA rI Ú DataFrameÚtabler r r rt Ú to_pandasr Ú IntervalDtyper0 Úassert_frame_equalZ concat_tablesÚconcatÚ chunked_arrayÚcolumnr ) r r r rp Údfr¡ r5 Ztable2r4 r r r Útest_arrow_table_roundtripL s(
|
|
ÿr© c C sx dd l }t | ¡}d |d< t d|i¡}| |¡}| ¡ }|jjd ksJt |
|
¡ }t |d j tj sht t ||¡ d S )Nr r rY )r
r rA rI r r¡ Zreplace_schema_metadatar Úmetadatart r¢ r r r£ r0 r¤ )r r rp r¨ r¡ r5 r r r Ú+test_arrow_table_roundtrip_without_metadatan s
|
|
r« c C s dd l } | ddddddg¡}tjt d¡dd}| |¡}tjtjdddgdddd}t
|
||¡ | | |g¡¡}t
|
||¡ d S ) Nr r r r
|
r r? r* )r ) r
r. rI r£ r, r Z__from_arrow__r rA r0 rD r¦ )r rp r r5 r4 r r r Ú%test_from_arrow_from_raw_struct_array
s
|
ÿ r¬ ÚtimezoneÚUTCz
|
US/PacificÚGMTc C sj t dd| d}d| d| d}tjddgdd g||d
|
}tj|d d
|d d
|d }t ||¡ d S )NZ2022r r zinterval[datetime64[ns, z], ú]z
|
2022-01-01z
|
2022-01-02z
|
2022-01-03)r+ r r] r r* )r r ri r0 Zassert_index_equal)r Zinclusive_endpoints_fixtureÚdatesr r5 r4 r r r Útest_interval_index_subtype s ü
|
|
ÿr² )'Únumpyr, r: Z pandas.util._test_decoratorsÚutilZ_test_decoratorsÚtdZpandasrI r r r r r r r Zpandas._testingZ_testingr0 Zpandas.core.arraysr Zfixturer% r& r= ra ru rv Z
|
skip_if_noZ pyarrow_skipr r r r; r< r© r« r¬ r² r r r r Ú<module> s` $ þû
|
õ
|
85 E
|
|
|
|
ý ý
|
|
