U
|
ý°d~F ã @ sü d dl mZ d dlZd dlZd dlmZ d dlmZ d dl m
|
Z
|
m Z m Z m Z mZmZmZmZmZ d dlmZ d dlmZ d dlm mZ ej ddgdd d
|
Z G d d d Z G d d de Z G dd de Z G dd de Z!G dd de Z"dd Z#dS )é )ÚpartialN)Úis_categorical_dtype)Ú IntervalDtype) Ú CategoricalÚCategoricalIndexÚIndexÚIntervalÚ IntervalIndexÚ
|
date_rangeÚnotnaÚ period_rangeÚtimedelta_range)Ú IntervalArrayZfoo©Úparamsc C s | j S ©N©Úparam)Úrequest© r úfd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/interval/test_constructors.pyÚname s r c @ s¾ e Zd ZdZejdddddgejfejddd ejfe ejd
|
d ejd ejfe ejdd ej
|
d ej
|
fe e d dd¡ej d ej fe ddddfe dddddfe ddddfgddd Zdd Zej d e d d d dd!gejd d"fe d d d dd!gejd d#fe d d d dd!gejd d$fe d d d dd!gej d dfe d%d&ddfe dd&ddfg¡d'd( Zej d)e d d d dd!gejd e d d d dd!gej
|
d e d d d dd!gej d e d%d&de dd&dg¡d*d+ Zej d)ejgd ejgd! ejgd, g¡d-d. Zej d)g ejg dd ejg d/d ejg d"d ejg d#d ejg d$d g¡d0d1 Zej d)ed2ed3ejed3ed ejed3d4d g¡d5d6 Zej d7e e g¡d8d9 Z d:d; Z d<S )=ÚConstructorTestszÑ
|
Common tests for all variations of IntervalIndex construction. Input data
|
to be supplied in breaks format, then converted by the subclass method
|
get_kwargs_from_breaks to the expected format.
|
é é é é\ i é
|
Úint64©Údtypeiöÿÿÿé é é é g à?Z20180101©Úperiodsz<M8[ns]z
|
US/Eastern)r& Útzzdatetime64[ns, US/Eastern]z1 dayz<m8[ns]r c C s |j S r r ©Úselfr r r r Úbreaks_and_expected_subtype% s z,ConstructorTests.breaks_and_expected_subtypec C s |\}}| ||¡}|f ||d|}|j|ks6t|j|ksDt|jj|ksTtt |jt |d d
|d¡ t |j
|
t |dd
|d¡ d S )N)Úclosedr éÿÿÿÿr é ) Úget_kwargs_from_breaksr+ ÚAssertionErrorr r ÚsubtypeÚtmÚassert_index_equalÚleftr Úright) r) Ú constructorr* r+ r ÚbreaksÚexpected_subtypeÚ result_kwargsÚresultr r r Útest_constructor7 s z!ConstructorTests.test_constructorzbreaks, subtyper r- é é Úfloat64zdatetime64[ns]ztimedelta64[ns]z
|
2017-01-01é c
|
C sb | | |¡¡}|f |}| |¡}t|d}|t|fD ]"}|f d|i|} t | |¡ q:d S )Nr4 r )r. Zastyper Ústrr1 r2 )
|
r) r5 r6 r0 Zexpected_kwargsÚexpectedr8 Úiv_dtyper r9 r r r Útest_constructor_dtypeD s
|
|
|
z'ConstructorTests.test_constructor_dtyper6 c
|
C sd t |j}| |¡}|t|fD ]>}t d ¡ |f |dd|}W 5 Q R X |jjdks tq d S )Nr3 ©r r+ )r r r. r? r1 Zassert_produces_warningr+ r/ )r) r5 r6 rA r8 r r9 r r r Ú test_constructor_pass_closedZ s
|
|
z-ConstructorTests.test_constructor_pass_closedé2 c C sl | |¡}|f d|i|}tj}tj|d d
td}|j|ksFt|jj|ksVtt
|
t |¡|¡ d S )Nr+ r, r ) r. Únpr= ÚarrayÚobjectr+ r/ r r0 r1 Úassert_numpy_array_equal)r) r5 r6 r+ r8 r9 r7 Úexpected_valuesr r r Útest_constructor_nano s
|
z%ConstructorTests.test_constructor_nanÚuint64c C sv | |¡}|f d|i|}tjg td}t|dtj}|jsBt|j|ksPt|j j
|
|ks`tt t |¡|¡ d S )Nr+ r r ) r. rF rG rH Úgetattrr Úemptyr/ r+ r r0 r1 rI )r) r5 r6 r+ r8 r9 rJ r7 r r r Útest_constructor_empty| s
|
|
z'ConstructorTests.test_constructor_emptyÚ
|
0123456789Z
|
abcdefghijz<U1c C s2 d}t jt|d |f | |¡ W 5 Q R X d S )NúIcategory, object, and string subtypes are not supported for IntervalIndex©Úmatch)ÚpytestÚraisesÚ TypeErrorr. )r) r5 r6 Úmsgr r r Útest_constructor_string s ÿz(ConstructorTests.test_constructor_stringÚcat_constructorc C sD t jddd}t |¡}||}| |¡}|f |}t ||¡ d S )Nr r r )rF Úaranger Ú from_breaksr. r1 r2 )r) r5 rY r6 r@ Z
|
cat_breaksr8 r9 r r r Ú"test_constructor_categorical_valid¦ s
|
|
|
z3ConstructorTests.test_constructor_categorical_validc C s | td¡}d}tjt|d |f ddi| W 5 Q R X d}tjt|d |f ddi| W 5 Q R X d }tjt|d |f ddi| W 5 Q R X td
|
dd }| |¡}d }tjt|d |f | W 5 Q R X | tdd d ¡}d}tjt|d |f | W 5 Q R X d S )Nr z8closed must be one of 'right', 'left', 'both', 'neither'rR r+ Úinvalidz)dtype must be an IntervalDtype, got int64r r z(data type ["']invalid["'] not understoodz
|
2000-01-01r% z:Period dtypes are not supported, use a PeriodIndex insteadr, z+left side of interval must be <= right side)r. ÚrangerT rU Ú
|
ValueErrorrV r )r) r5 ZfillerrW r& Zperiods_kwargsZdecreasing_kwargsr r r Útest_generic_errors² s&
|
z$ConstructorTests.test_generic_errorsN) Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__rT ÚfixturerF r rZ r rL r= r
|
r r* r: ÚmarkÚ parametrizerB rD ÚnanrK rG rO ÚtupleÚlistrH rX r r r\ r` r r r r r s| þõÿ
|
úþ
|
|
|
ûþ
|
|
*
|
úþ
|
üþ
|
|
r c @ sZ e Zd ZdZejdd Zd ddZdd Zej
|
d e j e j fe j e j fg¡d
|
d Zd S )ÚTestFromArraysz+Tests specific to IntervalIndex.from_arraysc C s t jS r )r Ú from_arrays©r) r r r r5 Ö s zTestFromArrays.constructorr4 c C s |dd
|dd
dS )z
|
converts intervals in breaks format to a dictionary of kwargs to
|
specific to the format expected by IntervalIndex.from_arrays
|
Nr, r- )r3 r4 r ©r) r6 r+ r r r r. Ú s z%TestFromArrays.get_kwargs_from_breaksc C s t tddd}d}tjt|d" t |d d
|dd
¡ W 5 Q R X ddd g}d d
|
g}d }tjt|d t ||¡ W 5 Q R X d S ) NÚ
|
01234abcdeT©ZorderedrQ rR r, r- r r; r z(left and right must have the same length)r rj rT rU rV r rl r_ )r) ÚdatarW r3 r4 r r r Útest_constructor_errorsá s ÿ&
|
z&TestFromArrays.test_constructor_errorszleft_subtype, right_subtypec C s| t jd|d}t jdd|d}t ||¡}t|t jd}t|t jd}t j}t |j|¡ t |j |¡ |j
|
j |ksxt dS )z:mixed int/float left/right results in float for both sidesé r r- r N) rF rZ r rl r r= r1 r2 r3 r4 r r0 r/ ) r) Z left_subtypeZ right_subtyper3 r4 r9 Z expected_leftZexpected_rightr7 r r r Útest_mixed_float_intò s z#TestFromArrays.test_mixed_float_intN)r4 )ra rb rc rd rT re r5 r. rr rf rg rF r r= rt r r r r rk Ó s
|
|
ÿrk c @ s@ e Zd ZdZejdd ZdddZdd Zd d
|
Z d d Z
|
d S )ÚTestFromBreaksz+Tests specific to IntervalIndex.from_breaksc C s t jS r )r r[ rm r r r r5 s zTestFromBreaks.constructorr4 c C s d|iS )z
|
converts intervals in breaks format to a dictionary of kwargs to
|
specific to the format expected by IntervalIndex.from_breaks
|
r6 r rn r r r r. s z%TestFromBreaks.get_kwargs_from_breaksc C s< t tddd}d}tjt|d t |¡ W 5 Q R X d S )Nro Trp rQ rR )r rj rT rU rV r r[ )r) rq rW r r r rr s
|
ÿz&TestFromBreaks.test_constructor_errorsc C s* dg}t |¡}t g ¡}t ||¡ dS )z3breaks of length one produce an empty IntervalIndexr N)r r[ r1 r2 )r) r6 r9 r@ r r r Útest_length_one s
|
|
z TestFromBreaks.test_length_onec C s: t d¡}t |¡j}|jjd ks6|jj|jjk s6td S )Nr> ) rF rZ r r[ Ú_dataÚ_leftÚbaseZ_rightr/ )r) r6 r9 r r r Ú test_left_right_dont_share_data# s
|
z.TestFromBreaks.test_left_right_dont_share_dataN)r4 ) ra rb rc rd rT re r5 r. rr rv rz r r r r ru s
|
|
|
ru c @ s8 e Zd ZdZejdd Zd ddZdd Zd d
|
Z d S ) ÚTestFromTuplesz+Tests specific to IntervalIndex.from_tuplesc C s t jS r )r Ú from_tuplesrm r r r r5 - s zTestFromTuples.constructorr4 c C s t |¡r t |j d¡ t|dkr0d|iS tt|dd
|dd
}t|tt frdd|iS t
|
|rzd| |¡iS dt |¡iS )z
|
converts intervals in breaks format to a dictionary of kwargs to
|
specific to the format expected by IntervalIndex.from_tuples
|
z- not relevant IntervalIndex.from_tuples testsr rq Nr, r- )r1 Úis_unsigned_integer_dtyperT Úskipr Úlenrj ÚzipÚ
|
isinstanceri r Ú _constructorÚcomZasarray_tuplesafe)r) r6 r+ Útuplesr r r r. 1 s
|
z%TestFromTuples.get_kwargs_from_breaksc C s® dddg}d}t jt|j|dd t |¡ W 5 Q R X dddg}d}t jt|j|dd t |¡ W 5 Q R X dd d
|
g}t jt|j|dd t |¡ W 5 Q R X d S ) N©r r- r; )r r< z5IntervalIndex.from_tuples received an invalid item, 2)ÚtrR )r; z>IntervalIndex.from_tuples requires tuples of length 2, got {t})r; r r< )r> é )rT rU rV Úformatr r| r_ )r) r rW r r r rr C s
|
|
|
z&TestFromTuples.test_constructor_errorsc C s> dt jt jfdg}t |¡}t dt jdg¡}t ||¡ d S )Nr
)r; r )rF rh r r| r1 r2 )r) Zna_tupleZ idx_na_tupleZidx_na_elementr r r Útest_na_tuplesT s
|
z TestFromTuples.test_na_tuplesN)r4 )
|
ra rb rc rd rT re r5 r. rr r r r r r r{ * s
|
|
r{ c @ s e Zd ZdZejeeeddgddgddd Z ejd d
|
Z
|
d%d d Z dd Z dd Z dd Zej dg dfejejgdfeddddeddddgd feddd deddd dgdfejedddd fg¡d d Zej d eejeeg¡d d! Zd"d# Zd$S )&ÚTestClassConstructorsz6Tests specific to the IntervalIndex/Index constructorsÚintervalr r r )r Úidsc C s |j S r r r( r r r Úklass_ s zTestClassConstructors.klassc C s t S r )r rm r r r r5 g s z!TestClassConstructors.constructorr4 c s t |¡r t |j d¡ t|dkr0d|iS fddt|dd
|dd
D }t|trjd|iS t |rd|
|
|¡iS dt j |t d iS )
|
z¢
|
converts intervals in breaks format to a dictionary of kwargs to
|
specific to the format expected by the IntervalIndex/Index constructors
|
z) not relevant for class constructor testsr rq c s( g | ] \}}t |r t|| n|qS r )r r )Ú.0r3 r4 ©r+ r r Ú
|
<listcomp>v s ÿz@TestClassConstructors.get_kwargs_from_breaks.<locals>.<listcomp>Nr, r- r )r1 r} rT r~ r r r r rj r r rF rG rH )r) r6 r+ Úivsr r r r. k s
|
|
þ
|
z,TestClassConstructors.get_kwargs_from_breaksc C s dS )z
|
override the base class implementation since errors are handled
|
differently; checks unnecessary since caught at the Interval level
|
Nr )r) r5 r r r r` s z)TestClassConstructors.test_generic_errorsc C s d S r r rm r r r rX s z-TestClassConstructors.test_constructor_stringc C s t ddddt ddddg}d}tjt|d || W 5 Q R X d
|
}tjt|d |d W 5 Q R X d }tjt|d |ddg W 5 Q R X d S ) Nr r- r4 r r; r r3 z-intervals must all be closed on the same siderR zX(IntervalIndex|Index)\(...\) must be called with a collection of some kind, 5 was passedr> z?type <class 'numpy.int(32|64)'> with value 0 is not an interval)r rT rU r_ rV )r) r r rW r r r rr s ÿz-TestClassConstructors.test_constructor_errorsz data, closedÚbothÚneitherr r r r; r> r3 c C sL t |tr| ¡ }ndd |D }tj||d}|||d}t ||¡ d S )Nc S s$ g | ] }t |r |j|jfn|qS r )r r3 r4 )r Zivr r r r ¶ s zGTestClassConstructors.test_override_inferred_closed.<locals>.<listcomp>r )r r Z to_tuplesr| r1 r2 )r) r5 rq r+ r r@ r9 r r r Ú test_override_inferred_closed¡ s
|
|
z3TestClassConstructors.test_override_inferred_closedÚvalues_constructorc C sX t ddt ddt ddg}||}t|td}t|tks@tt |jt |¡¡ d S )Nr r- r; r r )
|
r r rH Útyper/ r1 rI ÚvaluesrF rG )r) r Ú intervalsr r9 r r r Útest_index_object_dtype» s
|
z-TestClassConstructors.test_index_object_dtypec C sX t ddddt ddddt ddddt dd d
|
dg}t|}t|td }t ||¡ d S ) Nr r- r3 r r; r4 r r r< r r )r r rH r1 r2 )r) r r9 r@ r r r Útest_index_mixed_closedÇ s ü z-TestClassConstructors.test_index_mixed_closedN)r4 ) ra rb rc rd rT re r r r r r5 r. r` rX rr rf rg rF rh r r[ r^ r rj rG r r r r r r r r \ s@ þ
|
|
|
þþõþ
|
|
ÿ
|
r c C sd t tjd} d}tjt|d tg | dd W 5 Q R X tjt|d tg | dd W 5 Q R X d S )Nr3 z*closed keyword does not match dtype.closedrR r rC )r rF r rT rU r_ r r )r rW r r r Útest_dtype_closed_mismatchÔ s r )$Ú functoolsr ÚnumpyrF rT Zpandas.core.dtypes.commonr Zpandas.core.dtypes.dtypesr Zpandasr r r r r r
|
r r r Zpandas._testingZ_testingr1 Zpandas.core.arraysr Zpandas.core.commonÚcoreÚcommonr re r r rk ru r{ r r r r r r Ú<module> s" ,
|
61&2x
|
