ujmH UddlmZddlmZddlmZddlZddlZddlm Z ddl Z ddl m Z m Z ddlZddlmZddlmZmZmZdd lmZddlZdd lmZmZmZmZmZmZdd l m!Z!m"Z"m#Z#dd l$m%Z%m&Z&dd l'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;mm?Z?m@Z@ddlAmBZBmCZCmDZDmEZEddlFmGZGmHZHmIZIddlJmKZKddlLmMZMe rddlNmOZOmPZPddlQmRZRddlSmTZTmUZUddlFmVZVgdZWdeXd<gdZYdeXd<eZdddd gZ[deXd!<gd"Z\deXd#<eWe[zZ]eYe\zZ^ge]e^Z_deXd$<e`d%d&gZadeXd'<d(d)gZbdeXd*<geaebZcdeXd+<edd,d-gZedeXd.<er d/gZfdeXd0<n egd1d/gZfdeXd0<geeeaZhdeXd2<d3d4gZideXd5<d6d7gZjdeXd8<ekd9gZldeXd:<emd;gZndeXd<<eod=gZpdeXd><eae]zZqebe^zZrgeqerZsdeXd?<geseeZtdeXd@<eqeezefzeizejzelzepzenzZuejejejejejejejejgZ~egeZe`edeeeeeeekemeegZdAdBdCe Zdejeje`dDejedEgZejejfDcgc]}dFD] }|dG| c}}ZerddlZejejejej$gZejejejej(gZeezZeDcgc]}ege|c}Zejej0gZeDcgc]}ege|c}Zej6dHdIgZej:gZej>gZejBdJejBdKejDdLejDdMgZejHejJgZdNDcgc]}dOD]}ejN||P c}}ZdNDcgc]}ejR|c}ZejVgZeezezezezezezezezezZeezZngZgZgZgZeaerzezZgdQZgdRZejdejfejhejjejlejngZgdSZdTdUgZeezZgdVZdkdldWZdmdXZGdYdZeZGd[d\eZdnd]Z dod^Zdpd_Zdqd`ZdqdaZdqdbZdqdcZdqddZdqdeZgdNZdrdfZdsdgZdtdhZ du dvdiZgdjZycc}}wcc}wcc}wcc}}wcc}w)w) annotations)ThreadPoolExecutor)DecimalN) byteorder) TYPE_CHECKINGAny)using_string_dtype)can_set_locale get_locales set_locale) HAS_PYARROW) ArrowDtype DataFrameIndex MultiIndex RangeIndexSeries)round_trip_pathlibround_trip_picklewrite_to_compressed)assert_produces_warningmaybe_produces_warning)assert_almost_equalassert_attr_equalassert_categorical_equalassert_class_equalassert_contains_all assert_copyassert_datetime_array_equalassert_dict_equal assert_equalassert_extension_array_equalassert_frame_equalassert_index_equal!assert_indexing_slices_equivalentassert_interval_array_equalassert_is_sortedassert_metadata_equivalentassert_numpy_array_equalassert_period_array_equalassert_series_equalassert_sp_array_equalassert_timedelta_array_equalraise_assert_detail) get_dtypeget_obj)decompress_fileraises_chained_assignment_error set_timezonewith_csv_dialect)ArrowExtensionArrayBaseMaskedArrayNumpyExtensionArray)NDArrayBackedExtensionArray) extract_array)CallableIterable)AbstractContextManager)DtypeNpDtype)ExtensionArray)uint8uint16uint32uint64z list[NpDtype]UNSIGNED_INT_NUMPY_DTYPES)UInt8UInt16UInt32UInt64z list[Dtype]UNSIGNED_INT_EA_DTYPESint8int16int32int64SIGNED_INT_NUMPY_DTYPES)Int8Int16Int32Int64SIGNED_INT_EA_DTYPESALL_INT_DTYPESfloat32float64FLOAT_NUMPY_DTYPESFloat32Float64FLOAT_EA_DTYPESALL_FLOAT_DTYPES complex64 complex128COMPLEX_DTYPESU STRING_DTYPESstrCOMPLEX_FLOAT_DTYPESzdatetime64[ns]zM8[ns]DATETIME64_DTYPESztimedelta64[ns]zm8[ns]TIMEDELTA64_DTYPESbool BOOL_DTYPESbytes BYTES_DTYPESobject OBJECT_DTYPESALL_REAL_DTYPESALL_NUMERIC_DTYPES<>)littlebignanNaN) YMWDhmsmsusnspsfsasNaTryrzr{r|)ryrzr{r|)NUTCz US/Pacificz US/Eastern)unittz)__add____radd____sub____rsub____mul____rmul__ __floordiv__ __rfloordiv__ __truediv__ __rtruediv____pow____rpow____mod____rmod__)__eq____ne____le____lt____ge____gt__) countsummaxminmeanprodstdvarmediankurtskewsemallany)cumsumcumprodcummincummaxcr|tjurSt|tr*t t j |j}|Stj|d}|S|turt|d}|S|tur t|}|S|turDt|j}|r'|j}tj|gdzd}|S|t jus|t jurt j|}|S|tur t|}|St!|)z Helper function to wrap the expected output of a test in a given box_class. Parameters ---------- expected : np.ndarray, Index, Series box_cls : {Index, Series, DataFrame} Returns ------- subclass of box_cls F)copyT) ignore_index)pdarray isinstancerr7npasarray_valuesrrrto_frameTconcatndarrayto_arrayNotImplementedError)expectedbox_cls transposes c/var/www/html/python-recuperauto-other/env/lib/python3.12/site-packages/pandas/_testing/__init__.py box_expectedr6s)"(( h +*2::h6F6F+GHH, O)xxu5H( O' E .$ O# F (# O I (#,,.   zzHyy(adCH O BJJ 'RXX"588H% O H H% O"'**cdt|dd}|tj|St|dS)zQ Similar to pd.array, but does not cast numpy dtypes to nullable dtypes. dtypeNT) extract_numpy)getattrrrr9)objrs rrr_s2 C$ 'E }zz# D 11rc8eZdZddgZeddZeddZy)SubclassedSeriestestattrnamecdS)Nct|i|SNrargskwargss rz/SubclassedSeries._constructor..v'7'H'Hrselfs r _constructorzSubclassedSeries._constructoros IHrcdS)Nct|i|SrSubclassedDataFramers rrz9SubclassedSeries._constructor_expanddim..z':D'KF'Krrrs r_constructor_expanddimz'SubclassedSeries._constructor_expanddimxKKrNreturnr:)__name__ __module__ __qualname__ _metadatapropertyrrrrrrrls6V$I IILLrrc6eZdZdgZeddZeddZy)rrcdS)Nct|i|Srrrs rrz2SubclassedDataFrame._constructor..rrrrs rrz SubclassedDataFrame._constructorrrcdS)Nct|i|Srrrs rrz9SubclassedDataFrame._constructor_sliced..rrrrs r_constructor_slicedz'SubclassedDataFrame._constructor_slicedsHHrNr)rrrrrrrrrrrr}s3 I LLIIrrcJtj}|j||zS)aW Convert list of CSV rows to single CSV-formatted string for current OS. This method is used for creating expected value of to_csv() method. Parameters ---------- rows_list : List[str] Each element represents the row of csv. Returns ------- str Expected output of to_csv() in current OS. )oslinesepjoin) rows_listseps rconvert_rows_list_to_csv_strrs! **C 88I  $$rc0ddl}|j|dS)a$ Helper function to mark pytest.raises that have an external error message. Parameters ---------- expected_exception : Exception Expected error to raise. Returns ------- Callable Regular `pytest.raises` function with `match` equal to `None`. rN)match)pytestraises)expected_exceptionrs rexternal_error_raisedrs  ==+4= 88rc|jd} tt|}|S#t$rtt|ddfd}Y|SwxYw)z The operator function for a given op name. Parameters ---------- op_name : str The op name, in form of "add" or "__add__". Returns ------- function A function performing the operation. _Nc||Srr)xyrops rrz"get_op_from_name..s #a)r)striproperatorAttributeError)op_name short_opnameoprs @rget_op_from_namersY==%L$ X| , I $h QR 01 # I $s&!A  A c|Srrrs rgetitemr  Hrc|Srrrs rsetitemr r rc|jSr)locrs rrr 55Lrc|jSr)ilocrs rrrs 66Mrc|jSr)atrs rrrs 44Krc|jSr)iatrs rrrrrc`tj|tj|k\r|S|S)z2 Find the higher of two datetime64 units. )_UNITSindex)leftrights rget_finest_unitrs(||DV\\%00 Lrc|jDchc]4}|jD]}||jdkDr |j!6c}}Scc}}w)z Addresses of the non-empty buffers backing a pyarrow ChunkedArray. Zero-length buffers are excluded because distinct empty arrays can be handed out the same address, which would look like sharing. r) iterchunksbufferssizeaddress)pa_datachunkbufs r_pa_buffer_addressesr$s['')  ==?   ?sxx!|    s9Act|tjr0ttjrtj|St|tjr t|St|tryt|t rt fd|jDSt|ttfrLtttfr t|jjSt|jSt|trt|jSt|tjjj rt|j"St|tjjj$r.t|j&xst|j(St|t*rbtt*rdk(r/|j<j>dj@}t|StCtE|tE)z- Pandas-compat for np.shares_memory. Fc36K|]}t|ywr) shares_memory).0codesrs r z shares_memory..sH5=.Hsrr)#rrrr'rrr_codesrrrr8_ndarrayrcorearrays SparseArray sp_values IntervalArray_left_rightr5r$ _pa_array isdisjointr6_data_maskrlen_mgrblocksvaluesrtype)rr left_addrs right_addrsarrs ` rr'r'sZ$ # 5"**(Ee,, D"** %UD))$ #$ #HDKKHHH$( eeV_ - u}}= =T\\511$34T]]E22$223T^^U33$445TZZ/T=e3TT$+, e0 1-dnn=J.u?K!,,[99 9##D%0 0$( e_ - U[[9] EKK> TZZ/S=U3SS$ "s499+;+;'<'Aiiq!((S%(( d4j$u+ 66rc Ft|5}|g}n t|}|r&tj|}|j | g}t |D]$}|j |j |g|& t|kr|rj |D]} | j dddy#t$r&}ddl} | jd|d|dYd}~ld}~wwxYw#t|kr|rjwwwxYw#1swYyxYw)N) max_workersrz Spawning z threads failed with error z@ (likely due to resource limits on the system running the tests)) rlist threadingBarrierappendrangesubmit RuntimeErrorrskipr8abortresult) closurerA arguments pass_barriertpebarrierfuturesrerfs rrun_multithreadedrT:s-  4  IYI '' 4G   W % G;' @zszz'>I>? @7|k)l  A HHJ 3   KKK=),,   7|k)l /;)/sG9D4B>;C0<8D> C-C(#C0(C--C00$DDD )IALL_INT_EA_DTYPESALL_INT_NUMPY_DTYPESALL_NUMPY_DTYPESALL_REAL_NUMPY_DTYPESrfrhr^rcENDIANrZrWNARROW_NP_DTYPESNP_NAT_OBJECTS NULL_OBJECTSrjrSrNr`rdrIrDrrrrrrrrrr r!r"r#r$r%r&r'r(r)r*rr+r,r-rrr rr1rr/rr r0rr rrrrr.r2rrrTr r3r r'rr4r)T)rrrrrrerr)rrrzExtensionArray | np.ndarray)rz list[str]rra)rztype[Exception]rr<)rrarr:)rrrr)rrarrarra)r!zpa.ChunkedArrayrzset[int])rrrrrre)NF) rLr:rAintrMzIterable | NonerNrerNone) __future__rconcurrent.futuresrdecimalrrrsysrrCtypingrrnumpyrpandas._configr pandas._config.localizationr r r pandas.compatr pandasrrrrrrrpandas._testing._iorrrpandas._testing._warningsrrpandas._testing.assertersrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.pandas._testing.compatr/r0pandas._testing.contextsr1r2r3r4pandas.core.arraysr5r6r7pandas.core.arrays._mixinsr8pandas.core.constructionr9collections.abcr:r; contextlibr<pandas._typingr=r>r?rD__annotations__rIr]rNrSrVrUrTfloatrWrZr[complexr^r`rarbrcrdrerfrgrhrirjrXALL_REAL_EXTENSION_DTYPESrkrlrWfloat16rUrJrKrLr@rArBrZrBtuplerFdictset frozenset bytearray memoryviewPYTHON_DATA_TYPESrYrqrNAr\ datetime64 timedelta64r[pyarrowparCUNSIGNED_INT_PYARROW_DTYPESrMSIGNED_INT_PYARROW_DTYPESALL_INT_PYARROW_DTYPESALL_INT_PYARROW_DTYPES_STR_REPRrVFLOAT_PYARROW_DTYPESFLOAT_PYARROW_DTYPES_STR_REPR decimal128DECIMAL_PYARROW_DTYPESstringSTRING_PYARROW_DTYPESbinaryBINARY_PYARROW_DTYPEStime32time64TIME_PYARROW_DTYPESdate32date64DATE_PYARROW_DTYPES timestampDATETIME_PYARROW_DTYPESdurationTIMEDELTA_PYARROW_DTYPESbool_BOOL_PYARROW_DTYPESALL_PYARROW_DTYPES ALL_REAL_PYARROW_DTYPES_STR_REPRALL_REAL_NULLABLE_DTYPESarithmetic_dunder_methodscomparison_dunder_methodseqnegtgeltlecomparison_opsnumeric_reductionsboolean_reductionsall_reductionsnumeric_accumulationsrrrrrrrr r rrrrrrr$r'rT__all__)clsrtyprs0000rrs"1  - & 0  C22 2,S=R&M M*-vw)QQ$GkG03JJ*-AAI 4I7HI I%*Iy$AMA )955 G"4 G G+G& \B B"%M;&"%uc!2M;2$Jn$J7I$JkJ"2H!=;=#4h"?K? &> [)"G, k,$h/ {/*-AA+.??S!6S9RSS"EO"En"EKE     JJJJGGHHHHHHIIII        " $Y /bffbffeElBEE75>J  r~~.  t(#+288:yryy{IBIIK"U!(HBHHJ HBHHJ O8;TT(>'!$JsO'# 'BJJL*"**,7(<%!$JsO%!,bmmAq12&RYY[M&RYY[M  # $ $ $  %299;  4, ;   $2&& ?VVd  D 1V#288:,     !          " " # #   (*GG%%'!&(#')$225UU"Y KK KK KK KK KK KK  U^#&88A&R 2LvL" I) I%(9'99*:   ! 27p"&        FJ a 4'% WsV*V0V5 #V:8W