·oe+0dZddlmZddlmZddlmZmZmZmZm Z ddl m Z ddl Z ddl Z ddlmZddlmZmZdd lmZmZddlZdd lmZddlZddlZdd lmZmZddlm cm!Z"dd l#m$Z$m%Z%ddl&Z'dd l&m(Z(m)Z)m*Z*m+Z+m,Z,m-Z-ddl.m/Z0ddl1m2Z2ddl3m4Z4m5Z5ddl6m7Z7erddl8m9Z9m:Z: ddl;Z<[$rdZ=YnwxYwddl?Z? e?j@dn#e?jA$rdZ?YnwxYwdZ`ejYddg<d?ZaejYgd@<dAZbejYgdB<dCZcejYgdD<dEZdejYgdD<dFZeejYddGdHdIdJdKejfdLe"jgdMNg<dOZhejYdGdHdIdJdKejfdLe"jgdMNg<dPZiejYddg<dQZjejYgdR<dSZkejYdTdUg<dVZlejYe0jmdW6dXZnenZoejYdejpe'jqg<dYZrerZsejYe0jtdZ6d[ZueuZvejYe(e+g<d\ZwejYe4e+gd]d^g6d_ZxexZyejYe4e+e'jzggd`6daZ{ejYe4e+e(e'jzgdb6dcZ|e|Z}ejYddZ~ejYdeZejYdfZejYd>dhZejYd?djZdkZdlZidme0jdndoe0jdndpe0jdndqrdse0jdndte0jdndue0jdndve0jdndvwdxe0jdndxwdye0jdndywdze0jdndzwd{e0jdnd{wd|e0jdnd|wd}e0jdnd}wd~e0jdnd~wde0jdndwde0jdndwde0jdee4ejdddke0jdndwde0jdndwde0jdne0jdne4ge5jegdgdeee4gde4ejdndwe4ejdndwe4ejdndwe4ejdnedwe4e'jze0jdndwdZe=r.e4e'jze0jdndwZeed<ejYe<dZeZejYdeD<dZeZejYdeD<dZejYd@dZejYd@dZejYd@dZdZdeDZejYd@dZejYd@dZde0jDZieeeZejYe<dZde0jDZieeeeZejYe<dZejYd?dZejYd?dZejYd?dZejYd?dZejYd@dZejYe)dd(e%dzdfe)dde%ddfe*dddfe*dddfe-ddre$ddfe,ddfg<dZejYe0j<dZejYe je2je je2je je2je je2je je2je je2je je2je je je je je je je je2je je2je je2jg<dZejYe je2je je2je je2je je2je je2je je2je je2jg<dZgdZejYe֬<dZddgZejYeج<dZeezZejYeڬ<d„ZejYe je je je je je jg<dÄZejYgdĢ<dńZejYgdƢ<dDŽZgdȢZejYe߬<dɄZejYdʄZejYdAd̄ZejYdAd̈́ZejYdBdЄZejYd?dфZddddddddddeeed٦ededڦe je ed)ۦe edܬۦdݬަgZe?3ee?j@dqe?j@dgd߄eDZe"jeeejYee6dZe"jeed)dejYed)ded)d6dZeZddeee jgZe?#eIe?j@dejYe<dZeZejYe0j<dZejYdejfde"jgdNg<dZejYdejfde"jgdNejfde"jgdNg<dZejYdejfde"jgdNg<dZeZejYe0j<dZejYe0j<dZejYddejfde"jgdNejfde"jgdNg<dZejYe0j<dZejYe0j<dZejYdCdZejYe0j<dZejYe0j<dZejYe0j<dZejYe0j<dZejYe0j <dZ ejYe0j <dZ ejYe0j <dZejYe0j<dZejYe0j<dZejYe0je0jz<dZejYe0je0jze0jze0jz<dZejYe0j<dZejYe0j<dZejYe0j<dZejYe0j<dZejYe0j<dZ dmdejpdgfdmde'j!dgfdde'jqdgfd d ejpd gfd ejpejpejpgfd gfd dejpdgfddejpdgfddejpdgfdd)ejpdgfdd)ejpdgfde d)ejpe dgfddejpdgfdde'j!dgfdej"dejpej"dgfdoe-dejpe-dgfdedd)d)ejpedd)d)gfddejpdgfdted)ejpedgfded)ejpedgfdse*de'jqe*dgfd e)dd)ejpe)ddgfgZ#ee#\Z$Z%ejYe#e$6d!Z&ejYd"Z'ejYgd#<d$Z(ejYd%e'jSj)D<d&Z*ejYdd'g<d(Z+ejYd)Z,ejYd*d+d,ejpejpejpfejpe'jqdfejpe'j!dfe'j!e'j!e'j!fg<dDd.Z-ejYe0j.e0j/e0j0g<d/Z1ejYe0j/e0j0g<d0Z2ejYe0j.e0j0g<d1Z3ejYe0j.e0j/g<d2Z4ejYe0j5e0j/g<d3Z6ejYe0j7e0j0g<d4Z8ejYdEd6Z9ejYdEd7Z:d8d9gZ;e?%e;Ie?j@d8ejYe;<dFd:Z<ejYd;Z=dS(Ga 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)abc)datedatetimetime timedeltatimezone)DecimalN)Path) TYPE_CHECKINGCallable)tzlocaltzutc) strategies) FixedOffsetutc)DatetimeTZDtype IntervalDtype) DataFrameIntervalPeriodSeries Timedelta Timestamp)ops)Index MultiIndex)Version)HashableIteratorTFUTCreturnNonec6|ddddS)N--no-strict-data-files store_falsez6Don't fail if a test is skipped for missing data file.)actionhelp) addoption)parsers d/mounts/lovelace/software/anaconda3/envs/kraken-biom/lib/python3.11/site-packages/pandas/conftest.pypytest_addoptionr,hs2   Eitem pytest.Itempathstrmessagec|j|r7|tjd|dSdS)aIgnore doctest warning. Parameters ---------- item : pytest.Item pytest test item. path : str Module path to Python object, e.g. "pandas.core.frame.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)r.r0r2s r+ignore_doctest_warningr:psZ y$I  223FW3F3FGGHHHHHIIr-cd|dp|dd}gd}|D]|}|rK|tjd|D]\}}t |||d|jvr$|tjj}dS)Nz--doctest-modulesz--doctest-cythonF)default))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)NDFrame.replacezThe 'method' keyword)rCzSeries.replace without 'value')z Series.idxminThe behavior of Series.idxmin)z Series.idxmaxThe behavior of Series.idxmax)zSeriesGroupBy.idxminrD)zSeriesGroupBy.idxmaxrE)zmissing.mask_zero_div_zerozdivide by zero encountered) to_pydatetimez>The behavior of DatetimeProperties.to_pydatetime is deprecated)z pandas.core.generic.NDFrame.boolzY(Series|DataFrame).bool is now deprecated and will be removed in future version of pandas)z!pandas.core.generic.NDFrame.firstzqfirst is deprecated and will be removed in a future version. Please create a mask and filter using `.loc` instead)zResampler.fillnaz+DatetimeIndexResampler.fillna is deprecated)DataFrameGroupBy.fillnaz3DataFrameGroupBy.fillna with 'method' is deprecated)rGz,DataFrame.fillna with 'method' is deprecatedadd_doctest_importsz/frame/) getoptionr6r7r8 usefixturesr:nodeid arraymanager)itemsconfig is_doctestignored_doctest_warningsr.r0r2s r+pytest_collection_modifyitemsrQs!!"566&:J:JE;K;;J ) ) ) V 6 6  < OOFK334IJJ K K K!9 < < g&tT7;;;;  # # OOFK4 5 5 5 6 6r-z6.83.2ci)deadlinesuppress_health_checkz)MonthBegin MonthEnd BMonthBegin BMonthEndic)n normalizez%YearBegin YearEnd BYearBegin BYearEnd ) min_value max_value)rVrWmonthz1QuarterBegin QuarterEnd BQuarterBegin BQuarterEndi)rVrW startingMonthc.t|d<t|d<dS)z: Make `np` and `pd` names available for doctests. nppdN)rbrc)doctest_namespaces r+rHrHs !d dr-)autousec0tjdddS)z< Configure settings for all tests and test modules. chained_assignmentraiseN)rc set_optionr-r+configure_testsrks M&00000r-)rrZindexcolumnsc&dt|SNzaxis=reprxs r+rt sARaARARr-)paramsidsc|jS)z@ Fixture for returning the axis numbers of a DataFrame. paramrequests r+axisr|  =r-rmc&dt|Srorprrs r+rtrts5FT!WW5F5Fr-c|jS)zA Fixture for returning aliases of axis 1 of a DataFrame. rxrzs r+axis_1rr}r-)TFN)ruc|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). rxrzs r+observedr   =r-c|jS)z6 Boolean 'ordered' parameter for Categorical. rxrzs r+orderedr-r}r-c|jS)z% Boolean 'skipna' parameter. rxrzs r+skipnar5r}r-)firstlastFc|jS)z[ Valid values for the 'keep' parameter used in .duplicated or .drop_duplicates rxrzs r+keepr= =r-)bothneitherleftrightc|jS)zA Fixture for trying all interval 'inclusive' parameters. rxrzs r+inclusive_endpoints_fixturerFr}r-)rrrrc|jS)z< Fixture for trying all interval closed parameters. rxrzs r+closedrNr}r-c|jS)zS Secondary closed fixture to allow parametrizing over all pairs of closed. rxrzs r+ other_closedrVr}r-gzipbz2zipxztarzstd zstandard)marksc|jS)zK Fixture for trying common compression types in compression tests. rxrzs r+ compressionr^  =r-c|jS)zk Fixture for trying common compression types in compression tests excluding uncompressed case. rxrzs r+compression_onlyrprr-c|jS)z, Fixture that an array is writable. rxrzs r+writablerr}r-)innerouterrrc|jS)z: Fixture for trying all types of join operations. rxrzs r+ join_typerr}r-nlargest nsmallestc|jS)z1 Fixture for trying all nselect methods. rxrzs r+nselect_methodrr}r-c*t|jSNtype__name__rrs r+rtrtsd1gg6Fr-c|jS)z/ Fixture for each null type in pandas. rxrzs r+ nulls_fixturerr}r-c|jS)zL Fixture for each null type in pandas, each null type exactly once. rxrzs r+unique_nulls_fixturerr}r-c*t|jSrrrrs r+rtrtsQ8Hr-c|jS)z- Fixture for each NaT type in numpy. rxrzs r+np_nat_fixturerr}r-c|jS)z; Fixture to parametrize over DataFrame and Series. rxrzs r+frame_or_seriesrr}r-rlseriesc|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 rxrzs r+index_or_seriesrrr-)rlrarrayc|jS)zG Fixture to parametrize over Index, Series, and ExtensionArray rxrzs r+index_or_series_or_arrayrr}r-c|jSr)rrrs r+rtrts1:r-c|jS)z] Fixture to test behavior for Index, Series, DataFrame, and pandas Array classes rxrzs r+box_with_arrayrrr-c,Gddt}|S)z, Fixture for a dictionary subclass. ceZdZddZdS)"dict_subclass..TestSubDictr"r#c0tj|g|Ri|dSr)dict__init__)selfargskwargss r+rz+dict_subclass..TestSubDict.__init__s( M$ 0 0 0 0 0 0 0 0 0r-Nr"r#)r __module__ __qualname__rrjr-r+ TestSubDictrs( 1 1 1 1 1 1r-r)r)rs r+ dict_subclassrs4 11111d111 r-c6Gddtj}|S)z8 Fixture for a non-mapping dictionary subclass. c,eZdZd dZdZd dZd dZd S) 5non_dict_mapping_subclass..TestNonDictMappingr"r#c||_dSr)_data)runderlying_dicts r+rz>non_dict_mapping_subclass..TestNonDictMapping.__init__s (DJJJr-c6|j|Sr)r __getitem__)rkeys r+rzAnon_dict_mapping_subclass..TestNonDictMapping.__getitem__ s:))#.. .r-r c4|jSr)r__iter__rs r+rz>non_dict_mapping_subclass..TestNonDictMapping.__iter__ s:&&(( (r-intc4|jSr)r__len__rs r+rz=non_dict_mapping_subclass..TestNonDictMapping.__len__s:%%'' 'r-Nr)r"r )r"r)rrrrrrrrjr-r+TestNonDictMappingrs_ ) ) ) ) / / / ) ) ) ) ( ( ( ( ( (r-r)rMapping)rs r+non_dict_mapping_subclassrs6 ( ( ( ( (S[ ( ( ( r-c*tjd}|dddg}|jd|jjD|_|jgdd|S) z DataFrame with 3 level MultiIndex (year, month, day) covering first 100 business days from 2000-01-01 with random data dc|jSr)yearrrs r+rtzAmultiindex_year_month_day_dataframe_random_data..sr-c|jSr)r^rrs r+rtzAmultiindex_year_month_day_dataframe_random_data..s17r-c|jSr)dayrrs r+rtzAmultiindex_year_month_day_dataframe_random_data..saer-c8g|]}|dS)i8)astype).0levs r+ zCmultiindex_year_month_day_dataframe_random_data..!s$%S%S%S3cjj&6&6%S%S%Sr-)rr^rT)inplace)tmmakeTimeDataFramegroupbysumrl set_levelslevels set_names)tdfymds r+/multiindex_year_month_day_dataframe_random_datars s # #C ++''):):OOL M M Q Q S SC $$%S%S#)BR%S%S%STTCII000$??? Jr-rcBtgdgdggdgdgddgS)z; 2-level MultiIndex, lexsorted, with string names. foobarbazquxonetwothree) rrrrZrZrrr) rrZrrrZrZrrrZrrsecond)rcodesnames)rrjr-r+%lexsorted_two_level_string_multiindexr &sM ,,,.E.E.EF---/M/M/MN!   r-rc|}ttjdd|t gddS)z2DataFrame with 2 level MultiIndex with random datar) r)ABCexpr4)rlrm)rrbrandom default_rngstandard_normalr)r rls r+ multiindex_dataframe_random_datar2sX 2E  a  0099oooE222   r-ctgd}tddg}tjgd}tjgd}ddg}t||g||g|d S) zJ MultiIndex used to test the general functionality of this object rrr)rrrZrrr)rrZrrZrrZrr F)rr r verify_integrity)rrbrr) major_axis minor_axis major_codes minor_codes index_namess r+_create_multiindexr?s 33344Ju~&&J(---..K(---..KH%K J'K(    r-c jtjddgddgtjdddggd  S) zB MultiIndex with a level that is a tzaware DatetimeIndex. rZrab20130101r US/Eastern)periodstzr)r )r from_productrc date_rangerjr-r+_create_mi_with_dt64tz_levelr)SsJ  " Q#sR]:q\RRRS%%%   r-stringrrz datetime-tzz US/Pacific)r&periodrrangeint8dtypeint16int32int64uint8uint16uint32uint64float32float64z bool-objectrr complex64 complex128)rrr)rZrr)rrrZrZrrInt64UInt16Float32booleanzstring[python]) bool-dtyper9r: categoricalintervalemptytuplesmi-with-dt64tz-levelmultirepeats nullable_int nullable_uintnullable_float nullable_boolz string-pythonzstring[pyarrow]zstring-pyarrowcJt|jS)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 - ... ) indices_dictrycopyrzs r+rlrls  & + + - --r-cBg|]\}}t|t|Srj) isinstancerrrvalues r+rrs;   U*UJ:W:W     r-cN|j}t|S)z8 index fixture, but excluding MultiIndex cases. )ryrLrM)r{rs r+ index_flatrSs" -C   ! ! # ##r-ctg|]5\}}|ds|dvt|t3|6S))ruintfloat)r,rBrFr?) startswithrOrrPs r+rrs_    C NN3 4 4   AAA5*--B BAAr-c~t|jd}|j}|jdvrT|}d|dddz|d<d|dddz|d<t j|Sd|d<d|d<t||S) 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. Tdeep)rCrDrErrrZN)rLryrMvaluestolistr from_tuplesr)r{indvalss r+index_with_missingras. w} % * * * 5 5C :??  D}CCCzz||DGABBK'QT"Xabb\)R%d+++QRtCyyr-rc:tj}d|_|S)zC Fixture for Series of floats with Index of unique strings r)rmakeStringSeriesr4ss r+ string_seriesrfs A AF Hr-c:tj}d|_|S)zI Fixture for Series of dtype object with Index of unique strings objects)rmakeObjectSeriesr4rds r+ object_seriesrjs A AF Hr-c:tj}d|_|S)z9 Fixture for Series of floats with DatetimeIndex ts)rmakeTimeSeriesr4rds r+datetime_seriesrns A AF Hr-ct|}tjd|}t ||ddS)zHelper for the _series dictrr!F)rlr4rM)lenrbrrrr)rlsizedatas r+_create_seriesrssG u::D 9  # # 3 3D 9 9D $e#E : : ::r-c<i|]\}}d|dt|S)z series-with-z-indexrs)rindex_idrls r+ rwsA   %$8###^E%:%:   r-c t|S)zE Fixture for tests on series with changing types of indices. rurls r+series_with_simple_indexrzs %  r-cgdgdg}t|}tj|}tjdd}t||}tj|j d<|S)z: Fixture with a Series with a 2-level MultiIndex. )rrrrrrrr)rrrrrrrrrryr) rrr^rbrrrrnaniloc)arraysrCrlrrsers r+series_with_multilevel_indexrs A@@@@@F&\F  "6 * *E 9  # # 3 3A 6 6D U # # #C&CHQK Jr-cLi|]!}|jdtjd|"S)-seriesr!)r4r/)rrmake_rand_seriesrr/s r+rwrwsE  ~ 3E J J Jr-cNt|jdS)z| Fixture for tests on indexes, series and series with a narrow dtype copy to avoid mutation, e.g. setting .name TrY)_index_or_series_objsryrMrzs r+index_or_series_objrs" ! / 4 4$ 4 ? ??r-c>i|]}|jdt|S)r)rrrs r+rwrw's827u~u r-cNt|jdS)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 TrY)_index_or_series_memory_objsryrMrzs r+index_or_series_memory_objr4s" ( 6 ; ; ; F FFr-chttjdS)a6 Fixture for DataFrame of ints with index of unique strings Columns are ['A', 'B', 'C', 'D'] A B C D vpBeWjM651 1 0 1 0 5JyxmrP1En -1 0 0 0 qEDaoD49U2 -1 1 0 0 m66TkTfsFe 0 0 0 0 EHPaNzEUFm -1 0 -1 0 fpRJCevQhi 2 0 0 0 OlQvnmfi3Q 0 0 -2 0 ... .. .. .. .. uB1FPlz4uP 0 0 0 1 EcSe6yNzCU 0 0 -1 0 L50VudaiI8 -1 1 -2 0 y3bpw4nwIp 0 -1 0 0 H0RdLLwrCT 1 1 0 0 rY82K0vMwm 0 0 0 0 1OPIUjnkjk 2 0 0 0 [30 rows x 4 columns] r2)rr getSeriesDatarrjr-r+ int_framerAs)4 R%'' ( ( / / 8 88r-cBttjS)a Fixture for DataFrame of floats with DatetimeIndex Columns are ['A', 'B', 'C', 'D'] A B C D 2000-01-03 -1.122153 0.468535 0.122226 1.693711 2000-01-04 0.189378 0.486100 0.007864 -1.216052 2000-01-05 0.041401 -0.835752 -0.035279 -0.414357 2000-01-06 0.430050 0.894352 0.090719 0.036939 2000-01-07 -0.620982 -0.668211 -0.706153 1.466335 2000-01-10 -0.752633 0.328434 -0.815325 0.699674 2000-01-11 -2.236969 0.615737 -0.829076 -1.196106 ... ... ... ... ... 2000-02-03 1.642618 -0.579288 0.046005 1.385249 2000-02-04 -0.544873 -1.160962 -0.284071 -1.418351 2000-02-07 -2.656149 -0.601387 1.410148 0.444150 2000-02-08 -1.201881 -1.289040 0.772992 -1.445300 2000-02-09 1.377373 0.398619 1.008453 -0.928207 2000-02-10 0.473194 -0.636677 0.984058 0.511519 2000-02-11 -0.965556 0.408313 -1.312844 -0.381948 [30 rows x 4 columns] )rrgetTimeSeriesDatarjr-r+datetime_framer^s4 R)++ , ,,r-cBttjS)a Fixture for DataFrame of floats with index of unique strings Columns are ['A', 'B', 'C', 'D']. A B C D P7GACiRnxd -0.465578 -0.361863 0.886172 -0.053465 qZKh6afn8n -0.466693 -0.373773 0.266873 1.673901 tkp0r6Qble 0.148691 -0.059051 0.174817 1.598433 wP70WOCtv8 0.133045 -0.581994 -0.992240 0.261651 M2AeYQMnCz -1.207959 -0.185775 0.588206 0.563938 QEPzyGDYDo -0.381843 -0.758281 0.502575 -0.565053 r78Jwns6dn -0.653707 0.883127 0.682199 0.206159 ... ... ... ... ... IHEGx9NO0T -0.277360 0.113021 -1.018314 0.196316 lPMj8K27FA -1.313667 -0.604776 -1.305618 -0.863999 qa66YMWQa5 1.110525 0.475310 -0.747865 0.032121 yOa0ATsmcE -0.431457 0.067094 0.096567 -0.264962 65znX3uRNG 1.528446 0.160416 -0.109635 -0.032987 eCOBvKqf3e 0.235281 1.622222 0.781255 0.392871 xSucinXxuV -1.263557 0.252799 -0.552247 0.400426 [30 rows x 4 columns] )rrrrjr-r+ float_framer{s4 R%'' ( ((r-c tdddtjdgdzdtjdgdzdd tjd S) z Fixture for DataFrame of float/int/string columns with RangeIndex Columns are ['a', 'b', 'c', 'float32', 'int32']. ?rrrr7r.rZr1)r!r"cr7r1ry)rrbrarangerjr-r+mixed_type_framersk x )<<<XqcBhg666   imm    r-ctdddtdddtdddtdddtdddtdddtdddtdddtdddtdddg }ttjdt ||S)zF Fixture for Series with a DatetimeIndex that has duplicates. irZrrrYry)rrrbrrrrp)datess r+(rand_series_with_duplicate_datetimeindexrs q!q!q!q!q!q!q!q!q!q! E ")''**::3u::FFe T T TTr-)rrrg?g?z2012-01M)freqz period[M]z 2012-02-01Dz period[D]z 2011-01-01r$re)unitr&i)secondsztimedelta64[ns]c|jSrrxrzs r+ea_scalar_and_dtyper  =r-c|jS)zD Fixture for dunder names for common arithmetic operations. rxrzs r+all_arithmetic_operatorsrr}r-c|jS)zU Fixture for operator and roperator arithmetic, comparison, and logical ops. rxrzs r+all_binary_operatorsr D =r-c|jS)z Fixture for operator and roperator arithmetic functions. Notes ----- This includes divmod and rdivmod, whereas all_arithmetic_operators does not. rxrzs r+all_arithmetic_functionsrs 6 =r-) countrmaxminmeanprodstdvarmediankurtskewsemc|jS)z. Fixture for numeric reduction names. rxrzs r+all_numeric_reductionsr3r}r-allanyc|jS)z. Fixture for boolean reduction names. rxrzs r+all_boolean_reductionsr>r}r-c|jS)z> Fixture for all (boolean + numeric) reduction names. rxrzs r+all_reductionsrIr}r-c|jS)z; Fixture for operator module comparison functions. rxrzs r+ comparison_oprQrr-)__le____lt____ge____gt__c|jS)zm Fixture for dunder names for compare operations except == and != * >= * > * < * <= rxrzs r+compare_operators_no_eq_nerbrr-)__and____rand____or____ror____xor____rxor__c|jS)zY Fixture for dunder names for common logical operations * | * & * ^ rxrzs r+all_logical_operatorsro  =r-)cumsumcumprodcummincummaxc|jS)z0 Fixture for numeric accumulation names rxrzs r+all_numeric_accumulationsrr}r-c,|dS)zR Returns the configuration for the test setting `--no-strict-data-files`. r%)rI) pytestconfigs r+strict_data_filesrs  ! !": ; ;;r-r c:ttjdz S)Ntests)r __file__parentrjr-r+ tests_pathrs >> 7 **r-c|dz dz S)Niorrrj)rs r+tests_io_data_pathrs  v %%r-rCallable[..., str]ctjtjtdfd}|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. rctjjg|R}tj|s-rt d|dt jd|d|S)NzCould not find file z' and --no-strict-data-files is not set.zCould not find .)osr0joinexists ValueErrorr7skip)rr0 BASE_PATHrs r+decozdatapath..decos}w|I----w~~d## 3   X4XXX K1$111 2 2 2 r-)rr0rdirnamer)rrrs` @r+datapathrsN& RW__X66@@I Kr-cBtj|ddddS)z* The iris dataset as a DataFrame. rrrcsvziris.csv)rcread_csv)rs r+irisrs$ ;xxfeZ@@ A AAr-z Asia/Tokyozdateutil/US/Pacificzdateutil/Asia/Singaporez+01:15z-02:15z UTC+01:15z UTC-02:15i,i)hoursr[rrc,g|]}t|Srjrp)ris r+rrs+++AQ+++r-c|jS)zD Fixture for trying timezones including default (None): {0} rxrzs r+tz_naive_fixturerrr-c|jS)z4 Fixture for trying explicit timezones: {0} rxrzs r+tz_aware_fixturerrr-rz dateutil/UTCc|jS)zQ Fixture to provide variants of UTC timezone strings and tzinfo objects. rxrzs r+ utc_fixturerr}r-c|jS)zR Parametrized fixture for string dtypes. * str * 'str' * 'U' rxrzs r+ string_dtyper  =r-pyarrowc|jS)za Parametrized fixture for string dtypes. * 'string[python]' * 'string[pyarrow]' rxrzs r+nullable_string_dtyper  =r-python pyarrow_numpyc|jS)zx Parametrized fixture for pd.options.mode.string_storage. * 'python' * 'pyarrow' * 'pyarrow_numpy' rxrzs r+string_storager-rr-numpy_nullablec|jS)zb Parametrized fixture for pd.options.mode.string_storage. * 'python' * 'pyarrow' rxrzs r+ dtype_backendr ?rr-c|jS)zK Parametrized fixture for bytes dtypes. * bytes * 'bytes' rxrzs r+ bytes_dtyper S  =r-c|jS)zN Parametrized fixture for object dtypes. * object * 'object' rxrzs r+ object_dtyper^r r-objectzstring[pyarrow_numpy]c|jS)zo Parametrized fixture for string dtypes. * 'object' * 'string[python]' * 'string[pyarrow]' rxrzs r+any_string_dtyperirr-c|jS)z\ Parametrized fixture for datetime64 dtypes. * 'datetime64[ns]' * 'M8[ns]' rxrzs r+datetime64_dtyper{r r-c|jS)z^ Parametrized fixture for timedelta64 dtypes. * 'timedelta64[ns]' * 'm8[ns]' rxrzs r+timedelta64_dtyperr r-rc .tdddddddS)z- Fixture emits fixed Timestamp.now() irZr[r )rr^rhourminuter  microsecond)rrjr-r+ fixed_now_tsrs*  1RR   r-c|jS)z^ Parameterized fixture for float dtypes. * float * 'float32' * 'float64' rxrzs r+float_numpy_dtyperrr-c|jS)zR Parameterized fixture for float dtypes. * 'Float32' * 'Float64' rxrzs r+float_ea_dtyper r r-c|jS)z~ Parameterized fixture for float dtypes. * float * 'float32' * 'float64' * 'Float32' * 'Float64' rxrzs r+any_float_dtyper"rr-c|jS)zg Parameterized fixture for complex dtypes. * complex * 'complex64' * 'complex128' rxrzs r+ complex_dtyper$rr-c|jS)z| Parameterized fixture for signed integer dtypes. * int * 'int8' * 'int16' * 'int32' * 'int64' rxrzs r+any_signed_int_numpy_dtyper&rr-c|jS)zx Parameterized fixture for unsigned integer dtypes. * 'uint8' * 'uint16' * 'uint32' * 'uint64' rxrzs r+any_unsigned_int_numpy_dtyper(rr-c|jS)z Parameterized fixture for any integer dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' rxrzs r+any_int_numpy_dtyper*rr-c|jS)z Parameterized fixture for any nullable integer dtype. * 'UInt8' * 'Int8' * 'UInt16' * 'Int16' * 'UInt32' * 'Int32' * 'UInt64' * 'Int64' rxrzs r+any_int_ea_dtyper,rr-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' rxrzs r+ any_int_dtyper. s . =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' rxrzs r+any_numeric_ea_dtyper0$s " =r-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]' rxrzs r+any_numeric_ea_and_arrow_dtyper2:s @ =r-c|jS)z~ Parameterized fixture for any signed nullable integer dtype. * 'Int8' * 'Int16' * 'Int32' * 'Int64' rxrzs r+any_signed_int_ea_dtyper4]rr-c|jS)z Parameterized fixture for any (purely) real numeric dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' * float * 'float32' * 'float64' rxrzs r+any_real_numpy_dtyper6js $ =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. rxrzs r+any_real_numeric_dtyper8s ( =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' rxrzs r+any_numpy_dtyper:rr-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' rxrzs r+any_numeric_dtyper<s > =r-r!rmixedbytesacrBz mixed-integerg@floatingrintegerzmixed-integer-floatdecimal datetime64z 2013-01-01z 2018-01-01r#20180101riicomplexy??y@@rrAcT|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.)ryrbrr)r{inferred_dtyper\s r+any_skipna_inferred_dtyperIs2R%]NF XfF + + +F 6 !!r-ctjddddlm}ddlm}|}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:)rN)r7 importorskipIPython.core.interactiveshellrMtraitlets.configrNHistoryManager hist_file)rMrNrs r+iprT/sj  g6666>>>>>>('''''A!+A  1 % % %%r-)bsrcoocsccsrdiadoklilcZtjd}t||jdzS)z- Yields scipy sparse matrix classes. z scipy.sparse_matrix)r7rOgetattrry)r{sparses r+spmatrixr`Bs+   0 0F 67=94 5 55r-cg|]Y}tttj|tjjr |dk?ttj|ZS)Tick) issubclassr^rcoffsetsrb)ros r+rrMs[    gbj!,,bjo > > DE;;  ACN;;r-c|jS)zN Fixture for Tick based datetime offsets available for a time series. rxrzs r+ tick_classesrgLrr-c|Srrjrrs r+rtrtZsr-c|jS)zk Simple fixture for testing keys in sorting methods. Tests None (no key) and the identity key. rxrzs r+ sort_by_keyrjZrr-c#Ktjdddlm}ddlm}ddlm}Gdd|}|d|d |V|ddd|j d<|j dS) Nfsspecr)register_implementation)MemoryFileSystem) _registryc*eZdZdZdgZdfd ZxZS) fsspectest..TestMemoryFStestmemNr"r#cv|dd|jd<tjdi|dS)Ntestrrj)poprtsuperr)rr __class__s r+rz)fsspectest..TestMemoryFS.__init__ns?!::fd33DIaL EGG  & &v & & & & &r-r)rrrprotocolrtr __classcell__)rws@r+ TestMemoryFSrqjsIv ' ' ' ' ' ' ' ' ' 'r-rzrrT)clobber) r7rOrlrmfsspec.implementations.memoryrnfsspec.registryrorurtstoreclear)rmrnregistryrzs r+ fsspectestrcs !!!......>>>>>>555555'''''''''I|TBBBB ,.. LLD!!!Lar-)rNN)EgonVenkmanN)NCC1701Drr#tuple[Hashable, Hashable, Hashable]c|jS)zP A 3-tuple of names, the first two for operands, the last for a result. rxrzs r+r r ys =r-c|jS)zI Parametrize over __setitem__, loc.__setitem__, iloc.__setitem__ rxrzs r+ indexer_slirr}r-c|jS)z< Parametrize over loc.__getitem__, iloc.__getitem__ rxrzs r+ indexer_lirr}r-c|jS)z8 Parametrize over __setitem__, iloc.__setitem__ rxrzs r+ indexer_sirr}r-c|jS)z7 Parametrize over __setitem__, loc.__setitem__ rxrzs r+ indexer_slrr}r-c|jS)z: Parametrize over at.__setitem__, loc.__setitem__ rxrzs r+ indexer_alrr}r-c|jS)z< Parametrize over iat.__setitem__, iloc.__setitem__ rxrzs r+ indexer_ialrr}r-boolc6tjjjdkS)z> Fixture to check if the array manager is being used. r)rcoptionsmode data_managerrjr-r+using_array_managerrs :? '7 22r-cbtjjjotjjjdkS)z7 Fixture to check if Copy-on-Write is enabled. block)rcrr copy_on_writerrjr-r+using_copy_on_writers" :? ( TRZ_-IW-TTr-z Europe/Warsawzdateutil/Europe/Warsawc|jS)zE tzinfo for Europe/Warsaw using pytz, dateutil, or zoneinfo. rxrzs r+warsawrr}r-cdS)N)rrrjrjr-r+arrow_string_storagers ' 'r-r)r.r/r0r1r2r1r"r#)r"r)r"r)r"r)r"r )rr1r"r)r"r)r"r)r"r)r"r1(>__doc__ __future__r collectionsrrrrrr rCr operatorrpathlibr typingr r dateutil.tzrr hypothesisrstnumpyrbr7pytzrrpandas.util._test_decoratorsutil_test_decoratorstdpandas.core.dtypes.dtypesrrpandasrcrrrrrrpandas._testing_testingr pandas.corerpandas.core.indexes.apirrpandas.util.versionrcollections.abcrr rpa has_pyarrow ImportErrorzoneinfoZoneInfoZoneInfoNotFoundErrorr,r:rQ HealthChecktoo_slowhypothesis_health_checks __version__appenddiffering_executorssettingsregister_profiletuple load_profilesplitr4r^tseriesrdclsregister_type_strategybuildsintegersbooleansfixturerHrkr| axis_framerrrrrrrrry skip_if_norrrrr NULL_OBJECTSrnulls_fixture2r}NaTrunique_nulls_fixture2NP_NAT_OBJECTSrnp_nat_fixture2rrindex_or_series2rrrbox_with_array2rrrr rrr)makeStringIndex makeDateIndexmakePeriodIndexmakeTimedeltaIndexmakeRangeIndex makeIntIndex makeUIntIndexmakeFloatIndex makeBoolIndexrrrrrmakeNumericIndexmakeCategoricalIndexmakeIntervalIndexr^rrrrLidxkeysrlindex_fixture2rMrS index_flat2rarfrjrnrs_seriesrzrNARROW_NP_DTYPES_narrow_seriesrrPYTHON_DATA_TYPES_typ_objects_seriesrrrrrrrrarithmetic_dunder_methodsraddraddsubrsubmulrmultruedivrtruedivfloordiv rfloordivmodrmodpowrpoweqneltlegtgeand_rand_xorrxoror_ror_rr_all_numeric_reductionsr_all_boolean_reductionsr_all_reductionsrrrr_all_numeric_accumulationsrrrrrr TIMEZONESextend TIMEZONE_IDSparametrize_fixture_docr1rrtz_aware_fixture2_UTCSr utc_fixture2 STRING_DTYPESrrrr string_storage2 BYTES_DTYPESr  OBJECT_DTYPESrrDATETIME64_DTYPESrTIMEDELTA64_DTYPESrrFLOAT_NUMPY_DTYPESrFLOAT_EA_DTYPESr ALL_FLOAT_DTYPESr"COMPLEX_DTYPESr$SIGNED_INT_NUMPY_DTYPESr&UNSIGNED_INT_NUMPY_DTYPESr(ALL_INT_NUMPY_DTYPESr*ALL_INT_EA_DTYPESr,ALL_INT_DTYPESr.r0ALL_INT_PYARROW_DTYPES_STR_REPRFLOAT_PYARROW_DTYPES_STR_REPRr2SIGNED_INT_EA_DTYPESr4ALL_REAL_NUMPY_DTYPESr6ALL_REAL_DTYPESr8ALL_NUMPY_DTYPESr:ALL_NUMERIC_DTYPESr<NArD_any_skipna_inferred_dtyperv_rIrTr`__all__rgrjrr setitemlocr~rrrratriatrrrwarsawsrrrjr-r+r4s%&#"""""  ''''''  *)))))))) ('''''  KK KKK He%HHHIIII$=6=6=6=6@'2;< 7: !""ggh&7&777##J$:$NOOO $$% 899%     &&& 8 = = ? ?D '"*$d + +CB YRYskbk#r22kbkmm L L L 4 9 9 ; ;  D '"*$d + +CB  bk"a  !bkmm"+R888     @ E E G G  D '"*$d + +CB  bk#r""!bkmm%"+R@@@    !!!!11111117R7RSSSTS 9~+F+FGGGHG***+++  ,+ ***+++,+e}%%%&%///00010;;;<<<=<;;;<<<=<;;;<<<=<      V=2=#=#=>>>           V=2=#=#=>>>       e}%%%&%:::;;;<; K011121r,F,FGGGHGbfbf-.../.-r(.H.HIIIJI! 6*+++,+vWh,?@@@  A@ #vrx06R6R6RSSSTSvy"(;AUAUVVVWV!   0       (!  b %%!  %%!#2#CL999!  b %% ! &&s++ !  R s # # ! OBOCv . . .! _R_S 0 0 0! _R_S 0 0 0! _R_S 0 0 0! R c 1 1 1! bs(333! bs(333! bs(333! r I666!  r I666!!"#2#B''..v66#!$% --a00@@DDqHII$$S :::AA+NN%"%c;;;BB<PP*2*3//$$S)) U2YY$j$SS)>)>)> %J%JKK88::   ! !u'''((E)")C..888U929S>>:::eIBIcNN)<<<U929S>>0066iHHHU828$6B$6s$;$;CSTTTUUA!!! D) %++C008IJJJ K KC%(L!"|((**+++ . .,+ .  *0022    $$  $   &,,..        :            ;;;  '--//    !!!!     $ F<E7EnE,1133444@@54@;=;O        388::;;;GG<;G99998----8))))8"UUUU. q " " "MM'7$C$CD s# & & & i(I(IJ   $ $ $k2  3 ' ' '5 Il| 4 4 4 O 6 6 6  3   !23      r344454                        5 <=<             $  %$     .///0/!%..///0/*,CC'''('            ???@@@  A@  N N NGFF122232<<<++++&&&&BBBBB     EGG GIIKKNNK L HYYQ      HYYR u---%  (   H l + + H e $ $  ,++++ CC --..yl33343/.CC QRR 01122y},qrr*:;;;<;32% UUWWhl;  LL""5))***u r'((()( &mbmI.F.FGGG     YmbmI&>&>??? _MBM),D,DEEE  YmbmI&>&>???    !r'''('r'((()( &mbmI.F.FGGG ,MBM)4L4LMMM  r+,,,-,r,---.-r,---.-r)***+*r*+++  ,+ r()))*)r1222  32 r3444  54 r.///0/"r+,,,  -,  r()))*)2r+b.@@AAABA*  ( ) & '   :r.///  0/ r/00010(r)***+*,r*+++!!,+!Hr,---.-FRVS!" RUC ! sBFC ! tRVT"# rvrvrv&' bMsBFA&' sBFC !#rvs#$BFAQ,-RVWWQZZ01rvu%&rue$%MBM,//|9T9TUV))J'':1F1FGH dd4AT1a(8(8 9:()99Q<<167 dd1ggrvttAww '( t bfffTll34((1a.."&((1a..9:36 ( )Q1s;;;,","<;,"d&&&$HHHIII66JI6  #   kk*+++,+*!,    ru       BFBG455565()))*) BG,---.- BF+,,,-,rv'''('()))*)3333UUUU 4 5  NN$8$_55666w (((((s$%B--B76B7?C CC