U
|
ý°dR7 ã @ sÀ d Z ddlmZmZ ddlmZ ddlZddlZddlm Z m
|
Z
|
m Z m Z m Z mZmZmZmZmZmZmZmZmZmZmZ ddlmZ G dd dZ G dd d Z G d
|
d d Z G d d d Z dS )z7
|
Tests for the Index constructor conducting inference.
|
é )ÚdatetimeÚ timedelta)ÚDecimalN)ÚNAÚ CategoricalÚCategoricalIndexÚ DatetimeIndexÚIndexÚ IntervalIndexÚ
|
MultiIndexÚNaTÚ PeriodIndexÚSeriesÚTimedeltaIndexÚ TimestampÚarrayÚ
|
date_rangeÚ period_rangeÚtimedelta_rangec @ sj e Zd Zdd Zdd Zej dede j
|
e dg¡dd Z ej d
|
de j
|
g¡ej d e eeg¡d d Zej deddddddddg ¡dd Zej dddg¡ej d dddge jdddged g¡d d Zd d! Zd"d# Zd$d% Zej d&d'd(g¡ej d)ed*e d+¡fed,e d+¡fg¡d-d. Z ej d/ddg¡d0d1 Z ej d/ddg¡d2d3 Z dS )4Ú TestIndexConstructorInferencec C s> t jddgtd}t|}|jtks(tt|jtks:td S )NTF©Údtype)Únpr Úobjectr r ÚAssertionErrorr ©ÚselfÚarrÚres© r úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/test_index_new.pyÚtest_object_all_bools# s z3TestIndexConstructorInference.test_object_all_boolsc C sF t jtdtdgtd}t|}|jtks0tt|jtksBtd S )Né é r )r r Úcomplexr r r r r r r r r Útest_object_all_complex, s z5TestIndexConstructorInference.test_object_all_complexÚvalNÚnanc C s¼ t |g}t|}|jdkr&| ¡ ¡ s*tt|d d d
}|jdkrR| ¡ ¡ sVtttj|td}|jdkr~| ¡ ¡ stttj|tdd d d
}|jdkr´| ¡ ¡ s¸td S )Núdatetime64[ns]éÿÿÿÿr ) r r r ZisnaÚallr r r r )r r& ÚvaluesÚidxr r r Útest_infer_nat5 s z,TestIndexConstructorInference.test_infer_natÚna_valueÚvtypec C s2 dd|fg}t ||}t |¡}t ||¡ d S )N)r" Ztwoç @)r r Ú from_tuplesÚtmÚassert_index_equal)r r. r/ r+ ÚresultÚexpectedr r r Ú!test_construction_list_tuples_nanG s
|
z?TestIndexConstructorInference.test_construction_list_tuples_nanr Úint64Úint32Úint16Zint8Úuint64Zuint32Zuint16Zuint8c C s8 t ddddg|d}t dddd g|d}t ||¡ d S )
|
Nr r" r# é r g ç ð?ç @r0 ©r r2 r3 )r r r5 r4 r r r Ú test_constructor_int_dtype_floatP s z>TestIndexConstructorInference.test_constructor_int_dtype_floatÚ
|
cast_indexTFÚvalsr c C s< |rt |td}nt |}t|t ks*t|jtks8td S ©Nr )r ÚboolÚtyper r ©r r@ rA Úindexr r r Ú!test_constructor_dtypes_to_objectZ s
|
z?TestIndexConstructorInference.test_constructor_dtypes_to_objectc C s* t td}t|td}t|t r&td S ©Né r )r Úranger r Ú
|
isinstancer )r Úcir4 r r r Ú&test_constructor_categorical_to_objectg s zDTestIndexConstructorInference.test_constructor_categorical_to_objectc C s4 t dddd}t|}t ||¡ t|ts0td S )Nz2012-1-1ÚMr; )ÚfreqÚperiods)r r r2 r3 rK r r )r ZxpÚrsr r r Ú"test_constructor_infer_periodindexm s z@TestIndexConstructorInference.test_constructor_infer_periodindexc C s0 t dddd}t|}t|}t|ts,td S )Nz1/1/2000é ÚD©rP rO )r Úlistr rK r r )r ÚrngrP r4 r r r Útest_from_list_of_periodss s z7TestIndexConstructorInference.test_from_list_of_periodsÚposr r" zklass,dtype,ctorr( Únatztimedelta64[ns]c C sÒ t |trd S |ttg}|j|ks(t|g}| ||¡ d } |tkrptttg}tj j
|
dd}
|
|j |
|
¡ t } t|} t | ¡ t | |¡ W 5 Q R X ttj|td} t | ¡ t | |¡ W 5 Q R X d S )Nz%Broken with np.NaT ctor; see GH 31884)Úreasonr )rK r r r r Úinsertr r ÚpytestÚmarkZxfailÚnodeZ
|
add_markerÚDeprecationWarningr2 Zassert_produces_warningr3 r r r ) r rY Úklassr ZctorZ nulls_fixtureÚrequestr5 ÚdataÚwarnr^ r4 r r r Ú"test_constructor_infer_nat_dt_likez s$
|
z@TestIndexConstructorInference.test_constructor_infer_nat_dt_likeÚ swap_objsc C s` t d¡t d¡g}|r&|d d d
}t|td}t t||¡ t tt j|td|¡ d S )NrZ r) r )r Ú
|
datetime64Ú timedelta64r r r2 r3 r ©r rf rc r5 r r r Ú-test_constructor_mixed_nat_objs_infers_object¢ s zKTestIndexConstructorInference.test_constructor_mixed_nat_objs_infers_objectc C sb t dddddt d¡g}|r,|d d d
}t|}t t||¡ t ttj|td|¡ d S ) Niå é é é é* Únowr) r ) r r rg r r2 r3 r r r ri r r r Ú(test_constructor_datetime_and_datetime64 s zFTestIndexConstructorInference.test_constructor_datetime_and_datetime64) Ú__name__Ú
|
__module__Ú __qualname__r! r% r] r^ Ú parametrizer r r' Úfloatr- rV ÚtupleÚiterr6 Úintr? r rC rG rM rR rX r rg r rh re rj rp r r r r r " sB
|
þ
|
ÿ þþ
|
|
r c @ sP e Zd Zdd Zej dedddg¡dd Zej dedddg¡d d
|
Z d d Z
|
d d Z dd Z dd Z dd Zdd Zej dddg¡dd Zej ddd d ge dd d g¡ejdd d ged d d d!gejd d d!ged g¡d"d# Zej ddd d gd d d!ge d d d!g¡ejdd d ged ejd d d!ged g¡d$d% Zej ddd d gejdd d ged ejd&d'gd(d ed)dded)dd gg¡d*d+ Zej d,d-d.g¡ej dee e d&¡e d'¡g¡eed)dded)dd gg¡d/d0 Zej d,d-d.g¡ej de e dd1¡e dd1¡g¡e de dgg¡d2d3 Z d4S )5ÚTestDtypeEnforcedc C s, t dg|d}t|td}|jtks(td S )Nr r )r r r r r )r Zany_numeric_ea_dtyper r, r r r Ú*test_constructor_object_dtype_with_ea_data» s z<TestDtypeEnforced.test_constructor_object_dtype_with_ea_datar Úfloat64r: Úcategoryc C sH t td}t ||d}|j|ks&tt td|d}|j|ksDtd S rH )r rJ r r )r r rW r4 r r r Ú.test_constructor_range_values_mismatched_dtype s
|
z@TestDtypeEnforced.test_constructor_range_values_mismatched_dtypec C s, t dddg}t||d}|j|ks(td S )Nr" r# r; r )r r r r )r r Úcatr4 r r r Ú;test_constructor_categorical_values_mismatched_non_ea_dtypeÌ s zMTestDtypeEnforced.test_constructor_categorical_values_mismatched_non_ea_dtypec C s t ddd}t|}t||j}t ||¡ | d¡}t|}t||j}t ||¡ t t d¡}t|}t||jd}t ||¡ d S )Nú
|
2016-01-01r; ©rP z
|
Asia/TokyorI r )
|
r r r r r2 r3 Z tz_localizer
|
Ú from_breaksrJ )r Údtir~ r4 Údti2Zcat2ÚiiZcat3r r r Ú4test_constructor_categorical_values_mismatched_dtypeÓ s
|
zFTestDtypeEnforced.test_constructor_categorical_values_mismatched_dtypec C s^ t ddd}t|dd}t|}t ||¡ t dddd}t|dd}t|}t ||¡ d S )Nr r; r r| r z
|
US/Pacific)rP Útz)r r r r2 r3 )r r r4 r5 r r r r Ú7test_constructor_ea_values_mismatched_categorical_dtypeã s zITestDtypeEnforced.test_constructor_ea_values_mismatched_categorical_dtypec C s2 t dddd}t|dd}t|}t ||¡ d S )Nr r; rT rU r| r )r r r r2 r3 )r Úpir4 r5 r r r Ú/test_constructor_period_values_mismatched_dtypeî s zATestDtypeEnforced.test_constructor_period_values_mismatched_dtypec C s0 t ddd}t|dd}t|}t ||¡ d S )Nz4 DaysrI r r| r )r r r r2 r3 )r Ztdir4 r5 r r r Ú4test_constructor_timedelta64_values_mismatched_dtypeô s zFTestDtypeEnforced.test_constructor_timedelta64_values_mismatched_dtypec C s: t ddd}t |¡}t|dd}t|}t ||¡ d S )Nr r; r r| r )r r
|
r r r r2 r3 )r r r
r4 r5 r r r Ú1test_constructor_interval_values_mismatched_dtypeû s
|
|
zCTestDtypeEnforced.test_constructor_interval_values_mismatched_dtypec C s2 t ddd}t|dd}| d¡}t ||¡ d S )Nr r; r z Period[D]r rT )r r Z to_periodr2 r3 )r r r4 r5 r r r Ú:test_constructor_datetime64_values_mismatched_period_dtype s
|
zLTestDtypeEnforced.test_constructor_datetime64_values_mismatched_period_dtyper7 c C s6 t jg}d}tjt|d t||d W 5 Q R X d S )Nzcannot convert©Úmatchr )r r' r] ÚraisesÚ
|
ValueErrorr )r r rc Úmsgr r r Ú%test_constructor_int_dtype_nan_raises s z7TestDtypeEnforced.test_constructor_int_dtype_nan_raisesrA r" r# r; r r< r= r0 c C s" |}t ||d}|j|ks td S rB ©r r r )r rA Zany_int_numpy_dtyper rF r r r Ú test_constructor_dtypes_to_int s z0TestDtypeEnforced.test_constructor_dtypes_to_intc C s" |}t ||d}|j|ks td S rB r )r rA Zfloat_numpy_dtyper rF r r r Ú test_constructor_dtypes_to_float s z2TestDtypeEnforced.test_constructor_dtypes_to_floatz
|
2011-01-01z
|
2011-01-02r( iÛ c C s t |dd}t|tstd S )Nr| r )r rK r r )r rA rF r r r Ú&test_constructor_dtypes_to_categorical/ s
|
z8TestDtypeEnforced.test_constructor_dtypes_to_categoricalr@ TFc C sH |r.t |td}t|t s t|jtksDtnt |}t|tsDtd S rB )r r rK r r r rE r r r Ú#test_constructor_dtypes_to_datetime< s z5TestDtypeEnforced.test_constructor_dtypes_to_datetimerT c C sH |r.t |td}t|t s t|jtksDtnt |}t|tsDtd S rB )r r rK r r r rE r r r Ú$test_constructor_dtypes_to_timedeltaM s z6TestDtypeEnforced.test_constructor_dtypes_to_timedeltaN) rq rr rs rz r] r^ rt r r} r r r r r r r r r r rx ru r r r r r rg r rh r r r r r r ry ¸ st
|
|
|
úþ
|
ûþ
|
|
üþ
|
þþ þþry c @ sf e Zd Zej deeg¡dd Zdd Z ej de
|
d¡e
|
dd d
|
g¡e d d d jg¡dd ZdS )Ú TestIndexConstructorUnwrappingra c C s> t dt dt dg}t|}t|}||}t ||¡ d S )NZ20110101Z20120101Z20130101)r r r r2 r3 )r ra Zstampsr5 Úserr4 r r r Ú!test_constructor_from_series_dt64b s
|
z@TestIndexConstructorUnwrapping.test_constructor_from_series_dt64c C s6 t dddg}t|j}tdddg}t ||¡ d S )Nr" r# r; )r r r r2 r3 )r r r4 r5 r r r Ú test_constructor_no_pandas_arrayj s
|
z?TestIndexConstructorUnwrapping.test_constructor_no_pandas_arrayr rI ÚaÚbÚcz
|
2000-01-01r; r c C s2 G dd d}t |}t ||}t ||¡ d S )Nc @ s, e Zd ZddddZdejdddZdS )zOTestIndexConstructorUnwrapping.test_constructor_ndarray_like.<locals>.ArrayLikeN)Úreturnc S s
|
|| _ d S ©N©r )r r r r r Ú__init__} s zXTestIndexConstructorUnwrapping.test_constructor_ndarray_like.<locals>.ArrayLike.__init__c S s | j S r¢ r£ )r r r r r Ú __array__ s zYTestIndexConstructorUnwrapping.test_constructor_ndarray_like.<locals>.ArrayLike.__array__)N)rq rr rs r¤ r Zndarrayr¥ r r r r Ú ArrayLike| s r¦ r> )r r r¦ r5 r4 r r r Ú test_constructor_ndarray_likep s z<TestIndexConstructorUnwrapping.test_constructor_ndarray_likeN)rq rr rs r] r^ rt r r r r r Zaranger r r+ r§ r r r r r _ s
|
ýþr c @ s e Zd Zdd ZdS )ÚTestIndexConstructionErrorsc C s> d}t jt|d" tt tj¡jd gdd W 5 Q R X d S )NzIThe elements provided in the data cannot all be casted to the dtype int64r r" r7 r )r] r Ú OverflowErrorr r Ziinfor: Úmax)r r r r r Ú test_constructor_overflow_int64 s ÿz;TestIndexConstructionErrors.test_constructor_overflow_int64N)rq rr rs r« r r r r r¨ s r¨ ) Ú__doc__r r Údecimalr Únumpyr r] Zpandasr r r r r r
|
r r r r r r r r r r Zpandas._testingZ_testingr2 r ry r r¨ r r r r Ú<module> s H ()
|
