uj>UdZddlmZddlmZddlZddlmZmZmZm Z m Z m Z ddl m Z ddlZddlZddlZddlmZmZddlZddlmZmZddlZddlZdd lmZdd lmZddlm cm!Z"dd l#m$Z$m%Z%ddl&Z'ddl(m)Z*dd l+m,Z,dd l-m.Z.m/Z/er ddl0m1Z1m2Z2m3Z3 ddl4Z5[5dZ6eddZ8gdZ9dJdZ:dJdZ;dZ<dKdZ=dJdZ>ej~ddJdZ@ej~gddd ZAej~ddg!d"ZBej~gd#!d$ZCej~ddg!d%ZDej~ddg!d&ZEej~ddg!d'ZFej~gd(!d)ZGej~gd*!d+ZHej~gd,!d-ZIej~gd,!d.ZJej~dd/d0d1d2d3ejd4e"jd56g!d7ZMej~d/d0d1d2d3ejd4e"jd56g!d8ZNej~ddg!d9ZOej~gd:!d;ZPej~dZQej~ddg!d?ZRej~ddg!d@ZSej~gdA!dBZTej~ddg!dCZUej~ddg!dDZVej~ddg!dEZWej~dg!dFZXej~e*jdGdHZZeZZ[ej~deje'jg!dIZ^e^Z_ej~e*jdJdKZaeaZbej~e'je'jg!dLZeej~e.e'jgdMdNgdOZfej~e.e'je'jggdPdQZhej~e.e'je'je'jgdRdSZieiZjej~dLdTZkej~dMdUZlej~dVZmej~dNdWZnej~ dOdXZodYZpdZZqid[e.erd\Dcgc]}d]| c}es^d_e.erd\Dcgc]}d]| c}d^d`e'jdad\bdce'jdad\ddedfe'jdad\dghdie'jdjd\dgkdle'jd\dme.ejd\dm^dne.ejd\dn^doe.ejd\do^dpe.ejd\dp^dqe.ejd\dq^dre.ejd\dr^dse.ejd\ds^dte.ejd\dt^due.ejd\du^dve.ejd\dv^idwe.ddgdxzes^dye.ddgdxzey^dze.ejd\dz^d{ejd\dz^zzd|e.ejd\d|^d{ejd\d|^zzd}e'je{d~dzde'jjejdddde.gde/jegdgdddeqdepde.gdde.gdde.ejd\d^de.ejd\d^de.ejd\d^de.ejd\jeyd^de.e'jerd\Dcgc]}d]| c}d^Ze6r6e.e'jerd\Dcgc]}d]| c}d^Zeed<ej~ej !dZej~eDcgc] }|dk7s | c}!dZej~ejDcgc]\}}e|e/r|c}}!dZej~ejDcgc]\}}e|e/s|dk7r|c}}!dZeZeZej~ejDcgc]&\}}|jds|dvs e|e/s|(c}}!dZej~ejDcgc]&\}}|jds|dvs e|e/s|(c}}!dZej~dPdZej~dPdZej~dPdZdZejDcic]\}}d|de|c}}Zej~dPdZe*j,Dcic]D}|j.de'jerderdDcgc]}d| c}d|Fc}}ZieeeZej~ej !dZejDcic] \}}d|vr||c}}Zej~ej !dZe*j:Dcic]#}|j.de'j|%c}ZieeeeZej~ej !dZej~dOdZej~dOdZej~dPdZej~e'jHddxe%dpdfe'jHdde%dvdfe'jJdddfe'jJddgdfe'jLdd«jOdëe$ddīfe'jPdŬƫdfg!dȄZej~e*jT!dɄZej~ejXe,jZej\e,j^ej`e,jbejde,jfejhe,jjejle,jnejpe,jrejtejvejxejzej|ej~eje,jeje,jeje,jg!dʄZej~ejXe,jZej\e,j^ej`e,jbejde,jfejhe,jjejle,jnejpe,jrg!d˄Zej~e*j!d̄Zej~e*j!d̈́Zej~e*j!d΄Zej~e*j!dτZej~gdТ!dфZej~gdҢ!dӄZej~e*j!dԄZej~dՄZej~dQdքZddddddddddeeee e d߬e e ddgZe8eeՐje8jde8jde8jde8jdde8jd׫feDcgc] }e| c}Ze"jee٫ej~ee٬dZe"jeeddej~eddedddZddeegZe8eސje8jej~eެ!dZeZej~gd!dZeZej~e*j!dZej~de'jfejde'jfe"jd6ejdejfe"jd6dejfggddZej~dejde"jd6g!dZej~ejdejfe"jd6ejde'jfe"jd6g!dZej~dejde"jd6g!dZej~de'jfejde'jfe"jd6ejdejfe"jd6dejfggddZej~dejde"jd6g!dZeZeZej~e*j!dZej~e*j!dZej~ejd[de'jfejde'jfe"jd6ejdejfe"jd6dejfggddZeZej~e*j!dZej~e*j!dZej~dRdZej~e*j!dZej~e*j!dZej~e*j!dZej~e*j!dZej~e*j!dZej~e*j !dZej~e*j!dZ ej~e*j!dZ ej~e*j!dZ ej~e*j!dZej~e*je*jz!d Zej~e*je*jze*j"ze*j$z!d Zej~e*j(!d Zej~e*j,!d Zej~e*j0!d Zej~e*j4!dZej~e*j8!dZej~e*j<!dZd_dejdgfd_de'jdgfdde'jdgfddejdgfdejejejgfdgfddejdgfddejdgfddejdgfddejdgfddejdgfde d߫eje dgfddejdgfdde'jdgfdej@dejej@dgfd`e'jLd eje'jLd!gfd"ed#dd߫ejed$dd߫gfd%d&ejd'gfdie d߫eje dgfd(e d߫eje dgfdfe'jJd)e'je'jJd*gfde'jHdd߫eje'jHddgfgZ!e!Dcgc]}|d c}Z"ej~e!e"d+Z#ej~d,Z$ej~d-Z%ejLd.Z'ej~e'jPjRDcgc]P}e*e+e'jP|e'jPjXr|d/k7re+e'jP|Rc}!d0Z-ej~dd1g!d2Z.ej~d3d4d5ejejejfeje'jdfeje'jdfe'je'je'jfg!dSd6Z/ej~e*j`e*jbe*jdg!d7Z3ej~e*jbe*jdg!d8Z4ej~e*j`e*jdg!d9Z5ej~e*j`e*jbg!d:Z6ej~e*jne*jbg!d;Z8ej~e*jre*jdg!d<Z:ej~ddg!dTd=Z;ej~dUd>ZdBe?dC<e8#e>je8jd@ej~e>!dVdDZ@ej~dEZAej~dFGdHZBej~ddg!dIZCy#e7$rdZ6YswxYwcc}wcc}wcc}wcc}wcc}wcc}}wcc}}wcc}}wcc}}wcc}}wcc}wcc}}wcc}}wcc}wcc}wcc}wcc}w(Wa This file is very long and growing, but it was decided to not split it yet, as it's still manageable (2020-03-17, ~1.1k LoC). See gh-31989 Instead of splitting it was decided to define sections here: - Configuration / Settings - Autouse fixtures - Common arguments - Missing values & co. - Classes - Indices - Series' - DataFrames - Operators & Operations - Data sets/files - Time zones - Dtypes - Misc ) annotations)abcN)UTCdatedatetimetime timedeltatimezone)Decimal) TYPE_CHECKINGAny)tzlocaltzutcusing_string_dtype)import_optional_dependency)DatetimeTZDtype IntervalDtype)ops)Index MultiIndex)CallableHashableIteratorTFpytzignore)errors)z6single_cpu: tests that should run on a single cpu onlyzslow: mark a test as slowznetwork: mark a test as networkz2db: tests requiring a database (mysql or postgres)z(clipboard: mark a pd.read_clipboard testz4arm_slow: mark a test as slow for arm64 architecturez4high_memory: mark a test as requiring >5GB of memorycX|jddddd|jddd y) Nz--no-strict-data-files store_falsestrict_data_filesTz6Don't fail if a test is skipped for missing data file.)actiondestdefaulthelp--run-high-memory store_truez.Run tests marked high_memory (>5GB of memory).)r!r$) addoption)parsers Z/var/www/html/python-recuperauto-other/env/lib/python3.12/site-packages/pandas/conftest.pypytest_addoptionr*fsB    E   =c>tD]}|jd|y)Nmarkers)PANDAS_MARKERSaddinivalue_line)configmarkers r)pytest_configurer2us! 3 623r+crddl}ddl|jjfd}||j_y)NrcXj|rd|jvry|||S)N.T) isfunction __qualname__)selfmoduleobjectinspectorigs r) _from_modulez)pytest_sessionstart.._from_modules2   f %#1D1D*DD&&))r+)doctestr; DocTestFinderr=)sessionr>r=r;r<s @@r)pytest_sessionstartrAzs1  - -D**6G&r+strc|jj|r2|jtjj d|yy)a~Ignore doctest warning. Parameters ---------- item : pytest.Item pytest test item. path : str Module path to Python object, e.g. "pandas.DataFrame.append". A warning will be filtered when item.name ends with in given path. So it is sufficient to specify e.g. "DataFrame.append". message : str Message to be filtered. zignore:N)nameendswith add_markerpytestmarkfilterwarnings)itempathmessages r)ignore_doctest_warningrMs= yy$  22WWI3FGH r+cJ|jdsGtjjd}|D]"}d|jvs|j |$|jdxs|jdd}gd }|r|D]}|D]\}}t |||yy) Nr%zneed --run-high-memory to run)reason high_memoryz--doctest-modulesz--doctest-cythonF)r#))zapi.interchange.from_dataframe"The DataFrame Interchange Protocol)is_int64_dtypezis_int64_dtype is deprecated)is_interval_dtypezis_interval_dtype is deprecated)is_period_dtypezis_period_dtype is deprecated)is_datetime64tz_dtypez#is_datetime64tz_dtype is deprecated)is_categorical_dtypez"is_categorical_dtype is deprecated) is_sparsezis_sparse is deprecated)z&CategoricalDtype._from_values_or_dtypezConstructing a Categorical)zDataFrame.__dataframe__rQ)zDataFrameGroupBy.corrwithz'DataFrameGroupBy.corrwith is deprecated)to_pytimedeltaz2The behavior of TimedeltaProperties.to_pytimedelta)zSeries.autocorr#invalid value encountered in divide)z Series.corrrY)zmissing.mask_zero_div_zerozdivide by zero encountered)zTimestamp.utcfromtimestampz(Timestamp.utcfromtimestamp is deprecated)zBaseOffset.name.__get__z!The 'name' property is deprecated)zProperties.freqz3A future version of pandas will return a BaseOffset) getoptionrGrHskipkeywordsrFrM)itemsr0skip_high_memoryrJ is_doctestignored_doctest_warningsrKrLs r)pytest_collection_modifyitemsras   / 0!;;++3R+S 2D - 01 2!!"56&:J:JE;K;J  4 ros 5r+)paramsidsc|jS)z@ Fixture for returning the axis numbers of a DataFrame. paramrequests r)axisrw ==r+)rpc|jS)a Pass in the observed keyword to groupby for [True, False] This indicates whether categoricals should return values for values which are not in the grouper [False / None], or only values which appear in the grouper [True]. [None] is supported for future compatibility if we decide to change the default (and would need to warn if this parameter is not passed). rsrus r)observedrz ==r+)TFNc|jS)z6 Boolean 'ordered' parameter for Categorical. rsrus r)orderedr}rxr+c|jS)z% Boolean 'dropna' parameter. rsrus r)dropnarrxr+c|jS)z# Boolean 'sort' parameter. rsrus r)sortrrxr+c|jS)z% Boolean 'skipna' parameter. rsrus r)skipnarrxr+)firstlastFc|jS)z[ Valid values for the 'keep' parameter used in .duplicated or .drop_duplicates rsrus r)keepr ==r+)bothneitherleftrightc|jS)zA Fixture for trying all interval 'inclusive' parameters. rsrus r)inclusive_endpoints_fixturer rxr+)rrrrc|jS)z< Fixture for trying all interval closed parameters. rsrus r)closedr(rxr+c|jS)zS Secondary closed fixture to allow parametrizing over all pairs of closed. rsrus r) other_closedr0rxr+gzipbz2zipxztarzstd zstandard)marksc|jS)zK Fixture for trying common compression types in compression tests. rsrus r) compressionr8 ==r+c|jS)zk Fixture for trying common compression types in compression tests excluding uncompressed case. rsrus r)compression_onlyrJrr+c|jS)z, Fixture that an array is writable. rsrus r)writabler\rxr+)innerouterrrc|jS)z: Fixture for trying all types of join operations. rsrus r) join_typerdrxr+nlargest nsmallestc|jS)z1 Fixture for trying all nselect methods. rsrus r)nselect_methodrlrxr+c|jS)z2 Fixture for 'na_action' argument in map. rsrus r) na_actionrtrxr+c|jS)zJ Fixture for 'ascending' argument in sort_values/sort_index/rank. rsrus r) ascendingr|rxr+)averageminmaxrdensec|jS)z. Fixture for 'rank' argument in rank. rsrus r) rank_methodrrxr+c|jS)z5 Fixture for 'as_index' argument in groupby. rsrus r)as_indexrrxr+c|jS)z6 Fixture for 'cache' argument in to_datetime. rsrus r)cacherrxr+c|jS)z> Fixture for parallel keyword argument for numba.jit. rsrus r)parallelrrxr+c|jS)z; Fixture for nogil keyword argument for numba.jit. rsrus r)nogilrrxr+c,t|jSNtype__name__rms r)rorosd1g6F6Fr+c|jS)z/ Fixture for each null type in pandas. rsrus r) nulls_fixturerrxr+c|jS)zL Fixture for each null type in pandas, each null type exactly once. rsrus r)unique_nulls_fixturerrxr+c,t|jSrrrms r)rorosQ8H8Hr+c|jS)z- Fixture for each NaT type in numpy. rsrus r)np_nat_fixturerrxr+c|jS)z; Fixture to parametrize over DataFrame and Series. rsrus r)frame_or_seriesrrxr+riseriesc|jS)z Fixture to parametrize over Index and Series, made necessary by a mypy bug, giving an error: List item 0 has incompatible type "Type[Series]"; expected "Type[PandasObject]" See GH#29725 rsrus r)index_or_seriesrr{r+)rirarrayc|jS)zG Fixture to parametrize over Index, Series, and ExtensionArray rsrus r)index_or_series_or_arrayrrxr+c|jSr)rrms r)roros QZZr+c|jS)z] Fixture to test behavior for Index, Series, DataFrame, and pandas Array classes rsrus r)box_with_arrayr ==r+c>Gddtttf}|S)z, Fixture for a dictionary subclass. ceZdZddZy)"dict_subclass..TestSubDictc6tj|g|i|yr)dict__init__)r8argskwargss r)rz+dict_subclass..TestSubDict.__init__ s MM$ 0 0 0r+NreturnNone)r __module__r7rrlr+r) TestSubDictr s 1r+r)rr )rs r) dict_subclassrs 1d38n1 r+cRGddtjttf}|S)z8 Fixture for a non-mapping dictionary subclass. c*eZdZddZdZddZddZy) 5non_dict_mapping_subclass..TestNonDictMappingc||_yr)_data)r8underlying_dicts r)rz>non_dict_mapping_subclass..TestNonDictMapping.__init__s (DJr+c8|jj|Sr)r __getitem__)r8keys r)rzAnon_dict_mapping_subclass..TestNonDictMapping.__getitem__s::))#. .r+c6|jjSr)r__iter__r8s r)rz>non_dict_mapping_subclass..TestNonDictMapping.__iter__ s::&&( (r+c6|jjSr)r__len__rs r)rz=non_dict_mapping_subclass..TestNonDictMapping.__len__#s::%%' 'r+Nr)rz Iterator[Any])rint)rrr7rrrrrlr+r)TestNonDictMappingrs ) / ) (r+r)rMappingr )rs r)non_dict_mapping_subclassrs# (S[[c2 ( r+c tjtjj dj dt tdtjddd}|jd d d gj}|jj|jjDcgc]}|jd c}|_ |jjgd d|Scc}w)z DataFrame with 3 level MultiIndex (year, month, day) covering first 100 business days from 2000-01-01 with random data )dABCD 2000-01-01rBperiodsfreq)rjric|jSr)yearrms r)rozAmultiindex_year_month_day_dataframe_random_data..7s r+c|jSr)monthrms r)rozAmultiindex_year_month_day_dataframe_random_data..7s 177r+c|jSr)dayrms r)rozAmultiindex_year_month_day_dataframe_random_data..7s aeer+i8)rrrT)inplace)re DataFramerdrandom default_rngstandard_normalrlist date_rangegroupbysumri set_levelslevelsastype set_names)tdfymdlevs r)/multiindex_year_month_day_dataframe_random_datar,s ,, a 00:d6l#mmL#C@ C ++'):OL M Q Q SC $$#))BRBR%S3cjj&6%STCIII0$? J&Ts?Dc:tgdgdggdgdgddgS)z; 2-level MultiIndex, lexsorted, with string names. foobarbazquxonetwothree) rrrrhrhrrr r ) rrhrrrhrhrrrhrrsecond)rcodesnames)rrlr+r)%lexsorted_two_level_string_multiindexr$>s+ ,.EF-/MN! r+c|}tjtjj dj d|t gddS)z2DataFrame with 2 level MultiIndex with random datar) r )ArCexprDrirj)rerrdrrr r)r$ris r) multiindex_dataframe_random_datar,JsD 2E << a 009oE2 r+ctgd}tddg}tjgd}tjgd}ddg}t||g||g|d S) zJ MultiIndex used to test the general functionality of this object rrr)rrrhrr r )rrhrrhrrhrr!F)rr"r#verify_integrity)rrdrr) major_axis minor_axis major_codes minor_codes index_namess r)_create_multiindexr4Wsh 34Ju~&J((-.K((-.KH%K J'K(  r+c ntjddgddgtjdddggd  S) zB MultiIndex with a level that is a tzaware DatetimeIndex. rhrab20130101r  US/Easternrtzr)r#)r from_productrer rlr+r)_create_mi_with_dt64tz_levelr=ks9  " " Q#sR]]:q\RS% r+r:r&pandas_dtypestringrz 2020-01-01)rz datetime-tzz US/Pacificr:periodDrr z1 day)startrrrangeint8int16int32int64uint8uint16uint32uint64float32float64z bool-object bool-dtype complex64y? complex128 categoricalabcdrintervalr )numemptytuples)rrr)rhrr )strictmi-with-dt64tz-levelmultimixed-int-string)rr6rhr7rcrepeats)rrrhrhrr nullable_intInt64 nullable_uintUInt16nullable_floatFloat32 nullable_boolbooleanz string-pythonzstring[python]zstring[pyarrow]zstring-pyarrowcHt|jjdS)z Fixture for many "simple" kinds of indices. These indices are unlikely to cover corner cases, e.g. - no names - no NaTs/NaNs - no values near implementation bounds - ... Fdeep indices_dictrtcopyrus r)riris!  & + + + 77r+cHt|jjdS)zD index fixture, but excluding types that are not orderable. Frjrlrus r)index_sortablerps!  & + + + 77r+cL|j}t|jdS)z8 index fixture, but excluding MultiIndex cases. Frj)rtrmrn)rvrs r) index_flatrrs& --C   ! !u ! --r+cHt|jjdS)zI index_flat fixture, but excluding types that are not orderable. Frjrlrus r)index_flat_sortablerts!  & + + + 77r+)ruintfloat)rErYr`rQc8t|j}|jdvrC|j}dg|ddd|d<dg|ddd|d<tj|S|j j }d|d<d|d<t||dS)z Fixture for indices with missing values. Integer-dtype and empty cases are excluded because they cannot hold missing values. MultiIndex is excluded because isna() is not defined for MultiIndex. )rZr\r]NrrhFrn)rmrttolistr from_tuples_valuesrnrrvindvalss r)index_with_missingrs( w}} %C}}CCzz|&$q'!"+&Q(48AB<(R%%d++{{!QRtCyE**r+)rErYr`rQr^ct|j}|jj}d|d<d|d<t ||dS)zQ index_with_missing fixture, but excluding types that are not orderable. NrrxFry)rmrtr|rnrr}s r)index_with_missing_sortablersI w}} %C ;;   DDGDH 49T &&r+c tjtjdtjdzt t dDcgc]}d| c}dScc}w)zC Fixture for Series of floats with Index of unique strings r?g?i_rrirD)reSeriesrdarangerOrrEis r) string_seriesrsO 99 "BJJ'#-uRy1!r!X12  1s A"ctdDcgc]}d| }}ttdDcgc]}d| c}}tj||dtScc}wcc}w)zI Fixture for Series of dtype object with Index of unique strings rfoo_bar_objectsrirDr@)rErrerr:)rdataris r) object_seriesr%s] !&b *1d1#J *D * uRy1!T!:1 2E 99TYf EE +1s A A#ctjtjj dj dtj ddddS)z9 Fixture for Series of floats with DatetimeIndex rrrrrtsr)rerrdrrr r rlr+r)datetime_seriesr/sE 99 a 004mmL"3?  r+ct|}tjjdj |}t j ||ddS)zHelper for the _series dictrr6F)rirDrn)lenrdrrr rer)risizers r)_create_seriesr;s? u:D 99  # 3 3D 9D 99TSu ==r+z series-with-z-indexct|S)zE Fixture for tests on series with changing types of indices. )rris r)series_with_simple_indexrHs %  r+z-seriesrzi-r6rcHt|jjdS)z| Fixture for tests on indexes, series and series with a narrow dtype copy to avoid mutation, e.g. setting .name Frj)_index_or_series_objsrtrnrus r)index_or_series_objr[s! ! / 4 4% 4 @@r+cHt|jjdS)zR index_or_series_obj fixture, but excluding types that are not orderable. Frj)_index_or_series_objs_orderablertrnrus r)index_or_series_obj_orderablerks! +7== 9 > >E > JJr+cHt|jjdS)z Fixture for tests on indexes, series, series with a narrow dtype and series with empty objects type copy to avoid mutation, e.g. setting .name Frj)_index_or_series_memory_objsrtrnrus r)index_or_series_memory_objrs! ( 6 ; ; ; GGr+c tjtjdtjt t dDcgc]}d| c}t tdScc}w)zj Fixture for DataFrame of ints with index of unique strings Columns are ['A', 'B', 'C', 'D'] rrr?rrrr+)rerrdonesrIrrEr rs r) int_framersS << rxx(r3AtA3Z34d6l# 3s A1c tjtjj dj dt tdDcgc]}d| c}t tdScc}w)zm Fixture for DataFrame of floats with index of unique strings Columns are ['A', 'B', 'C', 'D']. rrrrrr+) rerrdrrr rrEr rs r) float_framers^ << a 009r3AtA3Z34d6l# 3s A:ctdddtdddtdddtdddtdddtdddtdddtdddtdddtdddg }tjtjj dj t||S)zF Fixture for Series with a DatetimeIndex that has duplicates. irhrr rrPr)rrerrdrrr r)datess r)(rand_series_with_duplicate_datetimeindexrs q!q!q!q!q!q!q!q!q!q! E 99RYY**1-==c%jIQV WWr+)rrrg?g?z2012-01M)rz period[M]z 2012-02-01z period[D]z 2011-01-01r9)r;s)unitr;i)secondsztimedelta64[us]c|jS)z; Fixture that tests each scalar and datetime type. rsrus r)ea_scalar_and_dtyper" ==r+c|jS)zD Fixture for dunder names for common arithmetic operations. rsrus r)all_arithmetic_operatorsrrxr+c|jS)zU Fixture for operator and roperator arithmetic, comparison, and logical ops. rsrus r)all_binary_operatorsrD ==r+c|jS)z Fixture for operator and roperator arithmetic functions. Notes ----- This does not include divmod and rdivmod, which return a tuple and so cannot be used interchangeably with the other operators. rsrus r)all_arithmetic_functionsrs6 ==r+c|jS)z. Fixture for numeric reduction names. rsrus r)all_numeric_reductionsr%rxr+c|jS)z. Fixture for boolean reduction names. rsrus r)all_boolean_reductionsr-rxr+c|jS)z> Fixture for all (boolean + numeric) reduction names. rsrus r)all_reductionsr5rxr+c|jS)z; Fixture for operator module comparison functions. rsrus r) comparison_opr=rxr+)__le____lt____ge____gt__c|jS)zm Fixture for dunder names for compare operations except == and != * >= * > * < * <= rsrus r)compare_operators_no_eq_nerEr{r+)__and____rand____or____ror____xor____rxor__c|jS)zY Fixture for dunder names for common logical operations * | * & * ^ rsrus r)all_logical_operatorsrR ==r+c|jS)z0 Fixture for numeric accumulation names rsrus r)all_numeric_accumulationsr`rxr+c$|jdS)zR Returns the configuration for the test setting `--no-strict-data-files`. r )rZ) pytestconfigs r)r r ks  ! !"5 66r+ctjjtjjtdfd}|S)aD Get the path to a data file. Parameters ---------- path : str Path to the file, relative to ``pandas/tests/`` Returns ------- path including ``pandas/tests``. Raises ------ ValueError If the path doesn't exist and the --no-strict-data-files option is not set. testsctjjg|}tjj|s*rt d|dt j d|d|S)NzCould not find file z' and --no-strict-data-files is not set.zCould not find r5)osrKjoinexists ValueErrorrGr[)rrK BASE_PATHr s r)decozdatapath..decoscww||I--ww~~d#  *4&0WX KK/$q1 2 r+)rrKrdirname__file__)r rrs` @r)datapathrss2& RWW__X6@I Kr+rz Asia/Tokyozdateutil/US/Pacificzdateutil/Asia/Singaporez+01:15z-02:15z UTC+01:15z UTC-02:15rh)hoursrxrr*i,ic|jS)zD Fixture for trying timezones including default (None): {0} rsrus r)tz_naive_fixturerrr+c|jS)z4 Fixture for trying explicit timezones: {0} rsrus r)tz_aware_fixturerrr+utcz dateutil/UTCc|jS)zQ Fixture to provide variants of UTC timezone strings and tzinfo objects. rsrus r) utc_fixturerrxr+)rmsusnsc|jS)z& datetime64 units we support. rsrus r)rrrxr+c|jS)zR Parametrized fixture for string dtypes. * str * 'str' * 'U' rsrus r) string_dtyper ==r+pythonpyarrow)string=string[python]string=string[pyarrow]string=str[pyarrow]string=str[python]cL|j\}}tj||S)z Parametrized fixture for string dtypes. * 'string[python]' (NA variant) * 'string[pyarrow]' (NA variant) * 'str' (NaN variant, with pyarrow) * 'str' (NaN variant, without pyarrow) )rtre StringDtypervstoragena_values r)string_dtype_no_objectrs#0 GX >>'8 ,,r+c|jS)za Parametrized fixture for string dtypes. * 'string[python]' * 'string[pyarrow]' rsrus r)nullable_string_dtyper ==r+c:tj|jS)zq Parametrized fixture for string dtypes backed by Pyarrow. * 'str[pyarrow]' * 'string[pyarrow]' )rerrtrus r)pyarrow_string_dtypers >>7== ))r+c|jSzb Parametrized fixture for pd.options.mode.string_storage. * 'python' * 'pyarrow' rsrus r)string_storager/rr+c|jS)z Parametrized fixture for StringDtype storage and na_value. * 'python' + pd.NA * 'pyarrow' + pd.NA * 'pyarrow' + np.nan rsrus r)string_dtype_argumentsr?s, ==r+numpy_nullablec|jSrrsrus r) dtype_backendr Xrr+c|jS)zK Parametrized fixture for bytes dtypes. * bytes * 'bytes' rsrus r) bytes_dtyper mrr+c|jS)zN Parametrized fixture for object dtypes. * object * 'object' rsrus r) object_dtyperxrr+)z string=objectrrrrct|jtjr |jS|j\}}t j ||S)z Parametrized fixture for string dtypes. * 'object' * 'string[python]' (NA variant) * 'string[pyarrow]' (NA variant) * 'str' (NaN variant, with pyarrow) * 'str' (NaN variant, without pyarrow) ) isinstancertrdr@rerrs r)any_string_dtypersA2'--*}}$MM~~gx00r+c|jS)z\ Parametrized fixture for datetime64 dtypes. * 'datetime64[ns]' * 'M8[ns]' rsrus r)datetime64_dtyperrr+c|jS)z^ Parametrized fixture for timedelta64 dtypes. * 'timedelta64[ns]' * 'm8[ns]' rsrus r)timedelta64_dtyperrr+c :tjdddddddS)z- Fixture emits fixed Timestamp.now() irh r )rrrhourminuter! microsecond)re Timestamprlr+r) fixed_now_tsrs$ << 1RR r+c|jS)z^ Parameterized fixture for float dtypes. * float * 'float32' * 'float64' rsrus r)float_numpy_dtyper!rr+c|jS)zR Parameterized fixture for float dtypes. * 'Float32' * 'Float64' rsrus r)float_ea_dtyper#rr+c|jS)z~ Parameterized fixture for float dtypes. * float * 'float32' * 'float64' * 'Float32' * 'Float64' rsrus r)any_float_dtyper%rr+c|jS)zg Parameterized fixture for complex dtypes. * complex * 'complex64' * 'complex128' rsrus r) complex_dtyper'rr+c|jS)z Parameterized fixture for complex and numpy float dtypes. * complex * 'complex64' * 'complex128' * float * 'float32' * 'float64' rsrus r)complex_or_float_dtyper)s ==r+c|jS)z| Parameterized fixture for signed integer dtypes. * int * 'int8' * 'int16' * 'int32' * 'int64' rsrus r)any_signed_int_numpy_dtyper+rr+c|jS)zx Parameterized fixture for unsigned integer dtypes. * 'uint8' * 'uint16' * 'uint32' * 'uint64' rsrus r)any_unsigned_int_numpy_dtyper-r{r+c|jS)z Parameterized fixture for any integer dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' rsrus r)any_int_numpy_dtyper/#rr+c|jS)z Parameterized fixture for any nullable integer dtype. * 'UInt8' * 'Int8' * 'UInt16' * 'Int16' * 'UInt32' * 'Int32' * 'UInt64' * 'Int64' rsrus r)any_int_ea_dtyper15s ==r+c|jS)a. Parameterized fixture for any nullable integer dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' * 'UInt8' * 'Int8' * 'UInt16' * 'Int16' * 'UInt32' * 'Int32' * 'UInt64' * 'Int64' rsrus r) any_int_dtyper3Fs. ==r+c|jS)z Parameterized fixture for any nullable integer dtype and any float ea dtypes. * 'UInt8' * 'Int8' * 'UInt16' * 'Int16' * 'UInt32' * 'Int32' * 'UInt64' * 'Int64' * 'Float32' * 'Float64' rsrus r)any_numeric_ea_dtyper5`rr+c|jS)a Parameterized fixture for any nullable integer dtype and any float ea dtypes. * 'UInt8' * 'Int8' * 'UInt16' * 'Int16' * 'UInt32' * 'Int32' * 'UInt64' * 'Int64' * 'Float32' * 'Float64' * 'uint8[pyarrow]' * 'int8[pyarrow]' * 'uint16[pyarrow]' * 'int16[pyarrow]' * 'uint32[pyarrow]' * 'int32[pyarrow]' * 'uint64[pyarrow]' * 'int64[pyarrow]' * 'float32[pyarrow]' * 'float64[pyarrow]' rsrus r)any_numeric_ea_and_arrow_dtyper7vs@ ==r+c|jS)z~ Parameterized fixture for any signed nullable integer dtype. * 'Int8' * 'Int16' * 'Int32' * 'Int64' rsrus r)any_signed_int_ea_dtyper9r{r+c|jS)z Parameterized fixture for any (purely) real numeric dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' * float * 'float32' * 'float64' rsrus r)any_real_numpy_dtyper;s$ ==r+c|jS)a  Parameterized fixture for any (purely) real numeric dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' * float * 'float32' * 'float64' and associated ea dtypes. rsrus r)any_real_numeric_dtyper=s( ==r+c|jS)a Parameterized fixture for all numpy dtypes. * bool * 'bool' * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' * float * 'float32' * 'float64' * complex * 'complex64' * 'complex128' * str * 'str' * 'U' * bytes * 'bytes' * 'datetime64[ns]' * 'M8[ns]' * 'timedelta64[ns]' * 'm8[ns]' * object * 'object' rsrus r)any_numpy_dtyper?rr+c|jS)a Parameterized fixture for all real dtypes that can hold NA. * float * 'float32' * 'float64' * 'Float32' * 'Float64' * 'UInt8' * 'UInt16' * 'UInt32' * 'UInt64' * 'Int8' * 'Int16' * 'Int32' * 'Int64' * 'uint8[pyarrow]' * 'uint16[pyarrow]' * 'uint32[pyarrow]' * 'uint64[pyarrow]' * 'int8[pyarrow]' * 'int16[pyarrow]' * 'int32[pyarrow]' * 'int64[pyarrow]' * 'float[pyarrow]' * 'double[pyarrow]' rsrus r)any_real_nullable_dtyperAs: ==r+c|jS)a Parameterized fixture for all numeric dtypes. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' * float * 'float32' * 'float64' * complex * 'complex64' * 'complex128' * 'UInt8' * 'Int8' * 'UInt16' * 'Int16' * 'UInt32' * 'Int32' * 'UInt64' * 'Int64' * 'Float32' * 'Float64' rsrus r)any_numeric_dtyperCs> ==r+r_mixedbytesacz mixed-integerg@floatingg?integerzmixed-integer-floatdecimal datetime64z 2013-01-01z 2018-01-01r820180101riicomplexy??y@@r20132018c^|j\}}tj|t}||fS)a Fixture for all inferred dtypes from _libs.lib.infer_dtype The covered (inferred) types are: * 'string' * 'empty' * 'bytes' * 'mixed' * 'mixed-integer' * 'mixed-integer-float' * 'floating' * 'integer' * 'decimal' * 'boolean' * 'datetime64' * 'datetime' * 'date' * 'timedelta' * 'time' * 'period' * 'interval' Returns ------- inferred_dtype : str The string for the inferred dtype from _libs.lib.infer_dtype values : np.ndarray An array of object dtype that will be inferred to have `inferred_dtype` Examples -------- >>> from pandas._libs import lib >>> >>> def test_something(any_skipna_inferred_dtype): ... inferred_dtype, values = any_skipna_inferred_dtype ... # will pass ... assert lib.infer_dtype(values, skipna=True) == inferred_dtype r?)rtrdrr:)rvinferred_dtypevaluess r)any_skipna_inferred_dtyperSZs0R%]]NF XXfF +F 6 !!r+ctjddddlm}ddlm}|}d|j _d|j _|| S) zj Get an instance of IPython.InteractiveShell. Will raise a skip if IPython is not installed. IPythonz6.0.0) minversionr)InteractiveShell)Configz:memory:F)r0) rG importorskipIPython.core.interactiveshellrWtraitlets.configrXHistoryManager hist_fileenabled)rWrXr_s r)ipr_sJ  g6> (A!+A$A 1 %%r+c#RKt5ddddy#1swYyxYww)zJUses mpl_cleanup_context to ensure matplotlib is cleaned up around a test.N)mpl_cleanup_contextrlr+r) mpl_cleanuprbs%   s ' '$'c#Ktjd}tjd}tjd}|jj} |j 5|j ddddd|jj |jj||jdtjdy#1swYexYw#|jj |jj||jdtjdwxYww)a* Ensure Matplotlib is cleaned up around a test. Before a test is run: 1) Set the backend to "template" to avoid requiring a GUI. After a test is run: 1) Reset units registry 2) Reset rc_context 3) Close all figures See matplotlib/testing/decorators.py#L24. matplotlibzmatplotlib.unitszmatplotlib.pyplottemplateNallrh) rGrYregistryrn rc_contextuseclearupdateclosegccollect)mpl mpl_unitspltorig_units_registrys r)raras"   l +C##$67I   1 2C#,,113  ^^   GGJ      "!!"56 % 1      "!!"56 % 1 s8AEC3-C'C3 AE'C0,C33AEETickc|jS)zN Fixture for Tick based datetime offsets available for a time series. rsrus r) tick_classesrurr+c|Srrlrms r)rorosr+c|jS)zk Simple fixture for testing keys in sorting methods. Tests None (no key) and the identity key. rsrus r) sort_by_keyrxrr+)rNN)EgonVenkmanN)NCC1701Dr{r{c|jS)zP A 3-tuple of names, the first two for operands, the last for a result. rsrus r)r#r#s ==r+c|jS)zI Parametrize over __setitem__, loc.__setitem__, iloc.__setitem__ rsrus r) indexer_slir~rxr+c|jS)z< Parametrize over loc.__getitem__, iloc.__getitem__ rsrus r) indexer_lirrxr+c|jS)z8 Parametrize over __setitem__, iloc.__setitem__ rsrus r) indexer_sirrxr+c|jS)z7 Parametrize over __setitem__, loc.__setitem__ rsrus r) indexer_slr rxr+c|jS)z: Parametrize over at.__setitem__, loc.__setitem__ rsrus r) indexer_alrrxr+c|jS)z< Parametrize over iat.__setitem__, iloc.__setitem__ rsrus r) indexer_ialrrxr+c#Ktjd|j5|jrtjjnddddy#1swYyxYww)z Fixture to check if performance warnings are enabled. Either produces ``PerformanceWarning`` if they are enabled, otherwise ``False``. zmode.performance_warningsFN)reoption_contextrtrPerformanceWarningrus r)performance_warningr%sM  6 FG.5mmbii**FGGGs!A#+A A#A A#ctS)z= Fixture to check if infer string option is enabled. rrlr+r)using_infer_stringr/s  r+cNtjjjduS)NT)reoptionsfuturepython_scalarsrlr+r)using_python_scalarsr7s ::   + +t 33r+z Europe/Warsawzdateutil/Europe/Warsawz list[Any]_warsawsc|jS)zE tzinfo for Europe/Warsaw using pytz, dateutil, or zoneinfo. rsrus r)warsawrArxr+cf|ttjz }|j|S)z Generate a unique file for testing use. See link for removal policy. https://docs.pytest.org/en/7.1.x/how-to/tmp_path.html#the-default-base-temporary-directory )rBuuiduuid4touch)tmp_path file_paths r) temp_filerIs) 3tzz|,,I OO r+r@)scopec#zKtjj5}|dddy#1swYyxYwwr)rG MonkeyPatchcontext)mps r) monkeysessionrTs3    # # %s;/ ;8;c#K|j}tjd| 5|dddy#1swYyxYww)Nzfuture.distinguish_nan_and_na)rtrer)rvopts r)using_nan_is_narZs; --C  :G D s$A4 A=Ar)rJz pytest.ItemrKrBrLrBrr)rztype[dict[Any, Any]])rztype[abc.Mapping[Any, Any]])rr)rz pd.DataFrame)rz pd.Series)r rBrzCallable[..., str])rz pd.Timestamp)rz#tuple[Hashable, Hashable, Hashable])rzIterator[bool | type[Warning]])rbool)rrB(D__doc__ __future__r collectionsr contextlibrrrrr r rJr rmoperatorrtypingr r r dateutil.tzrrnumpyrdrGpandas._configrpandas.compat._optionalrpandas.util._test_decoratorsutil_test_decoratorstdpandas.core.dtypes.dtypesrrpandasrepandas._testing_testingtm pandas.corerpandas.core.indexes.apirrcollections.abcrrrrpa has_pyarrow ImportErrorrr.r*r2rArMrafixturergrwrzr}rrrrrrrrt skip_if_norrrrrrrrrrrr NULL_OBJECTSrnulls_fixture2nanNaTrunique_nulls_fixture2NP_NAT_OBJECTSrnp_nat_fixture2rrrrrrrbox_with_array2rrrr$r,r4r=rEr:r  period_rangetimedelta_range RangeIndexrrCategoricalIndexr  IntervalIndex from_breakslinspacer{rrrmidxkeysrirpr]rrrrt index_flat2index_flat2_sortable startswithrrrrrr_seriesrNARROW_NP_DTYPESr_narrow_seriesrrrrPYTHON_DATA_TYPES_typ_objects_seriesrrrrrIntervalPeriodras_unit Timedeltararithmetic_dunder_methodsraddraddsubrsubmulrmultruedivrtruedivfloordiv rfloordivmodrmodpowrpoweqneltlegtgeand_rand_xorrxoror_ror_rrnumeric_reductionsrboolean_reductionsrrcomparison_opsrrrnumeric_accumulationsrr r TIMEZONESextend FixedOffsetrepr TIMEZONE_IDSparametrize_fixture_docrBrr_UTCSappendrr utc_fixture2runit2 STRING_DTYPESrNArrrrrr string_storage2string_dtype_arguments2 BYTES_DTYPESr  OBJECT_DTYPESrr@rany_string_dtype2DATETIME64_DTYPESrTIMEDELTA64_DTYPESrrFLOAT_NUMPY_DTYPESr!FLOAT_EA_DTYPESr#ALL_FLOAT_DTYPESr%COMPLEX_DTYPESr'COMPLEX_FLOAT_DTYPESr)SIGNED_INT_NUMPY_DTYPESr+UNSIGNED_INT_NUMPY_DTYPESr-ALL_INT_NUMPY_DTYPESr/ALL_INT_EA_DTYPESr1ALL_INT_DTYPESr3r5ALL_INT_PYARROW_DTYPES_STR_REPRFLOAT_PYARROW_DTYPES_STR_REPRr7SIGNED_INT_EA_DTYPESr9ALL_REAL_NUMPY_DTYPESr;ALL_REAL_DTYPESr=ALL_NUMPY_DTYPESr?ALL_REAL_NULLABLE_DTYPESrAALL_NUMERIC_DTYPESrCrK_any_skipna_inferred_dtyperqrSr_rbcontextmanagerraoffsets__all__ issubclassgetattrrsrurxr#setitemlocilocr~rrratriatrrrrr__annotations__rrrr)rrvalueindex_idrir@pairos00000000r)r4s (#    ->))    K!&: 3 6,I$*        V=2==#=>    e}%&:;< K012h'()e}%&BCDe}%&e}%&e}%&w r,FGHbffbff-./-r((.HIJ! bii012ryy)/BC D ryy"((39UVW 299bllBHH 5;O!  0"   (+ eE"I6qwqc]6fE+ eE"I6qwqc]6eD+  lB7+=2==rlK +  oboolBSA + ###'2CH + ]R]]2 + E)"))B-v .+ U9299R= 0+ U9299R= 0+ U9299R= 0+ U9299R= 0+ eIBIIbM2+ eIBIIbM2+ eIBIIbM2+ uYRYYr])4!+"uYRYYr])4#+$5$*&9%+&%u )6'+( "K(4)"))Bk2R+RR)+.% "L)D9299R|3T,TT/+4&2&&tF|a'785+6  ,,[R[[CR-HI7+8 U2Y9+: $j$$ !9T:;+@8:A+B  !C+D67E+Fu'(G+HE)"))B-w7I+JU9299R=9K+LeIBIIbM;M+NU9299R=//5YGO+PUr3AGA3-3;KLQ+ X r;AGA3-;CTU VC%(L!"|((*+ 8, 8|Qs>P7PQR8S8*002 U*UJ:W  .  .',,.  C%,8J1J  88 *',,.  C NN3 4AA5*-   + +.',,.  C NN3 4UU5*-   ' 'FF>(--/ %8*F#^E%::  !!$$   ~~g   b E"I6qBqc(6S!F<E7EnE,1134A5A,113#U$J#6;;=>K?K?A>R>R5:u~~g  % 00        388:;H<H    XX. !1 %}Wg'FG #S )=G+LM 93 '5 ?     255 i'}r}}Y/GH i(  i0HI 266       YmbmmI&>?    !0r'(r''() 255 i'}r}}Y/GH i(  i0HI 266    1! 1$%r++,-r,,-.r,,-.r))*+r**+ , r(()*r../ 0 r112 3 r334 5 r../0"r++, -  r(()*2r++b.@.@@AB*   (( ) && '   :r../ 0 r//01(r))*+,r**+!,!Hr2234>r,,-.FRVVS!" RUUC ! sBFFC ! tRVVT"# rvvrvvrvv&' bMsBFFA&' sBFFC !#rvvs#$BFFAQ,-RVVWQZ01rvvu%&ruue$%MBMM,/|9TUV,",,z*BFFLBLL4LMN d4AT1a(8 9:()9Q<167 d1grvvtAw '(  &!2669299V+<=>+"++a#RVV[R[[A->?@383DG 1s;,"<,"d&&*  @##  gbjj!,bjjoo >1;  A k*+,!,    ruu    BFFBGG456()* BGG,-. BFF+,-rvv'(()*e}%G&G  44'(@A)A OODMM/23x !i ! e}%&]@KT76N4<&R  & D B  7(#@ ,Dn sA|< A|  A| ! A| A|! , A|&7A|&)A|+ ;A|+ .A|1 ;+A|7 +A|= "A}!1A} A}  A}2A}>(A}EA}l* A}$n;AA})|A||A|} A}