a PfM @sVddlmZddlZddlZddlmmZddl m Z ddl m Z ddl mZmZmZmZmZmZmZmZmZmZddlmZddlmZddlmmZ ej!ddgdd d Z"Gd d d Z#Gd dde#Z$Gddde#Z%Gddde#Z&Gddde#Z'ej()dgdddZ*ddZ+ej()ddej,de-ddgd d!Z.dS)")partialN)is_unsigned_integer_dtype) IntervalDtype) CategoricalCategoricalDtypeCategoricalIndexIndexInterval IntervalIndex date_rangenotna period_rangetimedelta_range) IntervalArrayZfooparamscCs|jSNparam)requestrh/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/indexes/interval/test_constructors.pynamesrc @seZdZdZejgdejfejdddejfe ejddejdejfe ejddej dej fe ed d d ej dej fe d dd dfe d ddddfe ddd dfgdddZddZejde gdejddfe gdejddfe gdejddfe gdej ddfe ddd dfe ddd dfgd d!Zejd"e gdejde gdej de gdej de ddd e ddd gd#d$Zejd"ejgd%ejgd&ejgd'gd(d)Zejd"gejgddejgd*dejgddejgddejgddgd+d,Zejd"ed-ed.ejed.edejed.d/dgd0d1Zejd2eegd3d4Zd5d6Zd7S)8ConstructorTestsz 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 int64dtypei g?Z20180101periodszdS)Nr5r!)r0astyperstrr2r3) r+r6r7r1Zexpected_kwargsexpectedr9iv_dtyper!r:rrrtest_constructor_dtypeGs   z'ConstructorTests.test_constructor_dtyper7c Csxt|j}||}|t|fD]R}td$|f|dd|}Wdn1sX0Y|jjdks Jq dS)Nr4r!r-)rr!r0rBr2Zassert_produces_warningr-)r+r6r7rDr9r!r:rrrtest_constructor_pass_closed]s   2z-ConstructorTests.test_constructor_pass_closedr<r=2cCsl||}|fd|i|}tj}tj|ddtd}|j|ksFJ|jj|ksVJt t||dS)Nr-r.r ) r0npr>arrayobjectr-r!r1r2assert_numpy_array_equal)r+r6r7r-r9r:r8expected_valuesrrrtest_constructor_nanrs z%ConstructorTests.test_constructor_nanuint64cCsv||}|fd|i|}tjgtd}t|dtj}|jsBJ|j|ksPJ|jj |ks`Jt t||dS)Nr-r r!) r0rIrJrKgetattrremptyr-r!r1r2rL)r+r6r7r-r9r:rMr8rrrtest_constructor_emptys  z'ConstructorTests.test_constructor_empty 0123456789Z abcdefghijzr rr,r;mark parametrizerErGnanrNrJrRtuplelistrKr[rrr_rcrrrrr!s|        *         rc@steZdZdZejddZdddZddZej d e j e j fe j e j fgd d Zej d eegd dZdS)TestFromArraysz+Tests specific to IntervalIndex.from_arrayscCstjSr)r from_arraysr+rrrr6szTestFromArrays.constructorr5cCs|dd|dddS)z converts intervals in breaks format to a dictionary of kwargs to specific to the format expected by IntervalIndex.from_arrays Nr.r/)r4r5rr+r7r-rrrr0sz%TestFromArrays.get_kwargs_from_breakscCsttddd}d}tjt|d,t|dd|ddWdn1sT0Ygd}d d g}d }tjt|dt||Wdn1s0YdS) N 01234abcdeTorderedrTrUr.r/)rr/r<r<rz(left and right must have the same length)rrmrWrXrYr rorb)r+datarZr4r5rrrtest_constructor_errorss:z&TestFromArrays.test_constructor_errorszleft_subtype, right_subtypec Cs|tjd|d}tjdd|d}t||}t|tjd}t|tjd}tj}t|j|t|j ||j j |ksxJdS)z:mixed int/float left/right results in float for both sides r r/rN) rIr]r rorr>r2r3r4r5r!r1) r+Z left_subtypeZ right_subtyper4r5r:Z expected_leftZexpected_rightr8rrrtest_mixed_float_ints z#TestFromArrays.test_mixed_float_int interval_clscCstdddd}tdddd}|||}||d|}t||||d}||d}|||}||d|} t|| |d} |d} || | } || d| } t| | dS) Nz 2016-01-01rs)r'unitr?msrUTC)r roZas_unitr2Z assert_equalZ tz_localize)r+ryr4r5r:rCZleft2Zright2Zresult2Z expected2Zleft3Zright3Zresult3Z expected3rrr.test_from_arrays_mismatched_datetimelike_resoss         z=TestFromArrays.test_from_arrays_mismatched_datetimelike_resosN)r5)rdrerfrgrWrhr6r0rvrirjrIrr>rxrr r~rrrrrns   rnc@s@eZdZdZejddZdddZddZd d Z d d Z d S)TestFromBreaksz+Tests specific to IntervalIndex.from_breakscCstjSr)r r^rprrrr6!szTestFromBreaks.constructorr5cCsd|iS)z converts intervals in breaks format to a dictionary of kwargs to specific to the format expected by IntervalIndex.from_breaks r7rrqrrrr0%sz%TestFromBreaks.get_kwargs_from_breakscCsPttddd}d}tjt|dt|Wdn1sB0YdS)NrrTrsrTrU)rrmrWrXrYr r^)r+rurZrrrrv,s z&TestFromBreaks.test_constructor_errorscCs*dg}t|}tg}t||dS)z3breaks of length one produce an empty IntervalIndexrN)r r^r2r3)r+r7r:rCrrrtest_length_one6s  zTestFromBreaks.test_length_onecCs:td}t|j}|jjdus6|jj|jjus6JdS)Nr@)rIr]r r^_data_leftbase_right)r+r7r:rrrtest_left_right_dont_share_data=s  z.TestFromBreaks.test_left_right_dont_share_dataN)r5) rdrerfrgrWrhr6r0rvrrrrrrrs   rc@s8eZdZdZejddZd ddZddZd d Z d S) TestFromTuplesz+Tests specific to IntervalIndex.from_tuplescCstjSr)r from_tuplesrprrrr6GszTestFromTuples.constructorr5cCst|rt|jdt|dkr.d|iStt|dd|dd}t|ttfrbd|iStt |ddt rd| |iSdt |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 testsrruNr.r/r!)rrWskipr!lenrmzip isinstancerlrPr _constructorcomZasarray_tuplesafe)r+r7r-tuplesrrrr0Ks z%TestFromTuples.get_kwargs_from_breakscCsgd}d}tjt|j|ddt|Wdn1sB0Ygd}d}tjt|j|ddt|Wdn1s0Ygd}tjt|j|ddt|Wdn1s0YdS)N)rr/r<rr=z5IntervalIndex.from_tuples received an invalid item, 2)trU)r)r<rz>IntervalIndex.from_tuples requires tuples of length 2, got {t})r)r<rr=)r@)rWrXrYformatr rrb)r+rrZrrrrv]s((z&TestFromTuples.test_constructor_errorscCs>dtjtjfdg}t|}tdtjdg}t||dS)Nr)r<r)rIrkr rr2r3)r+Zna_tupleZ idx_na_tupleZidx_na_elementrrrtest_na_tuplesns zTestFromTuples.test_na_tuplesN)r5) rdrerfrgrWrhr6r0rvrrrrrrDs   rc @seZdZdZejeeeddgddgdddZ ejd d Z d%d d Z ddZ ddZ ddZejdgdfejejgdfeddddeddddgdfeddddeddd dgdfejedddd fgddZejdeejeegd d!Zd"d#Zd$S)&TestClassConstructorsz6Tests specific to the IntervalIndex/Index constructorsintervalr r r)rZidscCs|jSrrr*rrrklassyszTestClassConstructors.klasscCstSr)r rprrrr6sz!TestClassConstructors.constructorr5cst|rt|jdt|dkr.d|iSfddt|dd|ddD}t|trhd|iStt|d dt rd| |iSdt 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 testsrrucs(g|] \}}t|r t||n|qSr)r r ).0r4r5r-rr sz@TestClassConstructors.get_kwargs_from_breaks..Nr.r/r!r )rrWrr!rrrrmrPrrrIrJrK)r+r7r-ivsrrrr0s   z,TestClassConstructors.get_kwargs_from_breakscCsdS)z override the base class implementation since errors are handled differently; checks unnecessary since caught at the Interval level Nr)r+r6rrrrcsz)TestClassConstructors.test_generic_errorscCsdSrrrprrrr[sz-TestClassConstructors.test_constructor_stringcCstddddtddddg}d}tjt|d ||Wdn1sL0Yd }tjt|d |d Wdn1s0Yd }tjt|d |ddgWdn1s0YdS) Nrr/r5rr<rr4z-intervals must all be closed on the same siderUzX(IntervalIndex|Index)\(...\) must be called with a collection of some kind, 5 was passedr@z?type with value 0 is not an interval)r rWrXrbrY)r+rrrZrrrrvs&&z-TestClassConstructors.test_constructor_errorsz data, closedbothneitherrrrr<r@r4cCsLt|tr|}ndd|D}tj||d}|||d}t||dS)NcSs$g|]}t|r|j|jfn|qSr)r r4r5)rZivrrrrzGTestClassConstructors.test_override_inferred_closed..r)rr Z to_tuplesrr2r3)r+r6rur-rrCr:rrrtest_override_inferred_closeds    z3TestClassConstructors.test_override_inferred_closedvalues_constructorcCsXtddtddtddg}||}t|td}t|tus@Jt|jt|dS)Nrr/r<rr ) r rrKtyper2rLvaluesrIrJ)r+r intervalsrr:rrrtest_index_object_dtypes  z-TestClassConstructors.test_index_object_dtypecCsXtddddtddddtddddtdd d dg}t|}t|td }t||dS) Nrr/r4rr<r5rrr=rr )r rrKr2r3)r+rr:rCrrrtest_index_mixed_closeds     z-TestClassConstructors.test_index_mixed_closedN)r5)rdrerfrgrWrhr rrrr6r0rcr[rvrirjrIrkr r^rarrmrJrrrrrrrrvs>     rtimezone)r}z US/PacificGMTcCsjtdd|d}d|d|d}tjddgdd g||d }tj|dd |d d|d }t||dS)NZ2022rr(zinterval[datetime64[ns, z], ]z 2022-01-01z 2022-01-02z 2022-01-03r-r!r.r/r)r r ror2r3)rZinclusive_endpoints_fixturedatesr!r:rCrrrtest_interval_index_subtypesrcCsttjd}d}tjt|dtg|ddWdn1sB0Ytjt|dtg|ddWdn1s~0YdS)Nr4z*closed keyword does not match dtype.closedrUrrF)rrIrrWrXrbr r)r!rZrrrtest_dtype_closed_mismatchs  ,rr!ZFloat64zfloat64[pyarrow]Zpyarrow)ZmarkscCsVddg}t|dd}tj|d|d}|j|ks2Jtj|dd|}t||dS)N)g皙?)rg333333?r4)r1r-rr)rr rr!rAr2r3)r!ZbinsZinterval_dtyper:rCrrr test_ea_dtype s  r)/ functoolsrnumpyrIrWZpandas.util._test_decoratorsutilZ_test_decoratorstdZpandas.core.dtypes.commonrZpandas.core.dtypes.dtypesrZpandasrrrrr r r r r rZpandas._testingZ_testingr2Zpandas.core.arraysrZpandas.core.commoncorecommonrrhrrrnrrrrirjrrrZ skip_if_norrrrrs2   0   6H&2x