o h@sVdZddlmZddlmZddlmZmZmZmZm Z ddl m Z ddl Z ddl Z ddlmZmZddlmZmZddlZdd lmZddlZddlZdd lmZmZdd lmZddlm m!Z"dd l#m$Z$m%Z%ddl&Z'dd l&m(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3ddl4m5Z6ddl7m8Z8ddl9m:Z:m;Z;ddlm?Z?m@Z@zddlAZBWn eCydZDYnw[BdZDddlEZEzeEFdWn eEjGydZEYnwdddZHdddZIdd d!ZJejKjLgZMe=ejNe=d"kreMOejKjPejQjRd#deSeMd$ejQTd#d%UD]ZVeWe'jXjYeVZZe[eZej\eZe]d&d'e^d(qd)UD]"ZVeWe'jXjYeVZZe[eZej\eZe]d*d+e^ej]d,d-d.d/q:d0UD]"ZVeWe'jXjYeVZZe[eZej\eZe]d1d2e^ej]d,d-d.d3qaej_dd4dd5d6Z`ej_dd4dd7d8Zaej_gd9d:d;d<d=d>ZbebZcej_d,d?gd@d;d<dAdBZdej_gdCdDdEdFZeej_gdCdDdGdHZfej_ddgdDdIdJZgej_gdKdDdLdMZhej_gdNdDdOdPZiej_gdQdDdRdSZjej_gdQdDdTdUZkej_ddVdWdXdYdZejld[e"md\d]gdDd^d_Znej_dVdWdXdYdZejld[e"md\d]gdDd`daZoej_ddgdDdbdcZpej_gdddDdedfZqej_dgdhgdDdidjZrej_e6jsdkd;d<dldmZtetZuej_dejve'jwgdDdndoZxexZyej_e6jzdpd;d<dqdrZ{e{Z|ej_e)e.gdDdsdtZ}ej_e:e.gdudvgd<dwdxZ~e~Zej_e:e.e'jggdyd<dzd{Zej_e:e.e)e'jgd|d;d<d}d~ZeZej_dddZej_dddZej_ddZej_dddZej_dddZddZddZide:ddedDde1dddde1ddddde2ddddde3ddddde-dde:edddde:edddde:edddde:edddde:edddde:edddde:edddde:edddde:edddde:edddde:ddgd+ede:ddgd+ede:ejddddejddde:ejddddejddde(edde+ejdddde:ge;egdgdeee:gde:eddde:eddde:eddde:ededde:e'jddedDdddZeDrKe:e'jddedDddZeed<ej_edDdduZeZej_ddeDdDddńZeZej_ddeDdDddȄZej_ddd˄Zej_ddd̈́Zej_dddτZddфZddӄeDZej_dddՄZddӄe6jDZieeeZej_edDdd؄Zddӄe6jDZieeeeZej_edDddۄZej_ddd݄Zej_ddd߄Zej_dddZej_e*dd+de%ddfe*ddde%ddfe,ddddfe,ddddfe0ddde$dddfe/dddfgdDddZej_e6jdDddZej_e je8je je8je je8je je8je je8je je8je je8je je je je je je je je8je je8je je8jgdDddZej_e je8je je8je je8je je8je je8je je8je je8jgdDddZgdZej_edDddZddgZej_edDddZeeZej_edDddZej_e je je je je je jgdDddZej_gddDdd Zej_gd dDd d Zgd Zej_edDddZej_ddZej_dddZddddddddddeeededede je ed,de eddd d!gZeEdure۠eEFdeEFdgd"deDZe"ee݃ej_eed<d#d$Ze"eed,dej_ed,ded,dd<d%d&ZeZd'd(eee jgZeEdureOeEFdej_edDd)d*ZeZej_gd+dDd,d-ZeZej_e6jdDd.d/Zej_dejlde"md0d]gdDd1d2Zej_d3ejld0e"md0d]ejld4e"md0d]gdDd5d6Zej_d7ejld0e"md0d]gdDd8d9ZeZej_e6jdDd:d;Zej_e6jdDd<d=Zej_d>dejlde"md0d]ejld?e"md0d]gdDd@dAZej_e6jdDdBdCZej_e6jdDdDdEZej_ddGdHZej_e6jdDdIdJZej_e6jdDdKdLZej_e6jdDdMdNZej_e6jdDdOdPZej_e6jdDdQdRZej_e6jdDdSdTZej_e6jdDdUdVZej_e6jdDdWdXZej_e6jdDdYdZZ ej_e6j dDd[d\Z ej_e6je6jdDd]d^Z ej_e6je6je6j e6jdDd_d`Zej_e6jdDdadbZej_e6jdDdcddZej_e6jdDdedfZej_e6jdDdgdhZej_e6jdDdidjZej_e6jdDdkdlZddmejvdngfddme'jdngfdodme'jwdngfdpdqejvdrgfdsejvejvejvgfdsgfdtdmejvdugfdodmejvdvgfdwdxejvdvgfdyd,ejvdugfdzd,ejvdvgfd{e d,ejve dugfddejvdgfdde'jdgfd|ed}ejved~gfde0dejve0dgfdedd,d,ejvedd,d,gfddejvdgfded,ejvedugfded,ejvedugfde,de'jwe,dgfde*dd,ejve*ddugfgZee\ZZ ej_eed<ddZ!ej_ddZ"ej_gddDddZ#ej_dde'jYj$DdDddZ%ej_ddd;gdDddZ&ej_dddejvejvejvfejve'jwdfejve'jdfe'je'je'jfgdDdddZ'ej_e6j(e6j)e6j*gdDddZ+ej_e6j)e6j*gdDddZ,ej_e6j(e6j*gdDddZ-ej_e6j(e6j)gdDddZ.ej_e6j/e6j)gdDddZ0ej_e6j1e6j*gdDddZ2ej_dddZ3ej_dddZ4ej_dddZ5ej_dddZ6ddgZ7eEdu re7OeEFdej_e7dDdddZ8e_ddZ9dS(a 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) TYPE_CHECKINGCallable)tzlocaltzutc) strategies) FixedOffsetutc _get_option)DatetimeTZDtype IntervalDtype) CategoricalIndex DataFrameInterval IntervalIndexPeriod RangeIndexSeries Timedelta Timestamp date_range period_rangetimedelta_range)ops)Index MultiIndex)Version)HashableIteratorFTUTCreturnNonecCs|jdddddS)N--no-strict-data-files store_falsez6Don't fail if a test is skipped for missing data file.)actionhelp) addoption)parserr0A/var/www/html/env/lib/python3.10/site-packages/pandas/conftest.pypytest_addoptionos  r2item pytest.ItempathstrmessagecCs,|j|r|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)r3r5r7r0r0r1ignore_doctest_warningws r>cCsP|dp |jddd}gd}|r$|D]}|D] \}}t|||qqdSdS)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)DataFrameGroupBy.fillnaz%DataFrameGroupBy.fillna is deprecated)NDFrame.replacezThe 'method' keyword)rGzSeries.replace without 'value')z NDFrame.clipz4Downcasting behavior in Series and DataFrame methods)z Series.idxminThe behavior of Series.idxmin)z Series.idxmaxThe behavior of Series.idxmax)zSeriesGroupBy.fillnaz"SeriesGroupBy.fillna is deprecated)zSeriesGroupBy.idxminrH)zSeriesGroupBy.idxmaxrI)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)rFz3DataFrameGroupBy.fillna with 'method' is deprecated)rFz,DataFrame.fillna with 'method' is deprecated) read_parquetz1Passing a BlockManager to DataFrame is deprecated) getoptionr>)itemsconfig is_doctestignored_doctest_warningsr3r5r7r0r0r1pytest_collection_modifyitemss/ rQz6.83.2ci)deadlinesuppress_health_checkz)MonthBegin MonthEnd BMonthBegin BMonthEndic)n normalizez%YearBegin YearEnd BYearBegin BYearEnd ) min_value max_value)rVrWmonthz1QuarterBegin QuarterEnd BQuarterBegin BQuarterEndi)rVrW startingMonth)autousecCst|d<t|d<dS)z: Make `np` and `pd` names available for doctests. nppdN)rbrc)doctest_namespacer0r0r1add_doctest_importss recCstdddS)z< Configure settings for all tests and test modules. chained_assignmentraiseN)rc set_optionr0r0r0r1configure_tests ri)rrZindexcolumnscCdt|SNzaxis=reprxr0r0r1rs)paramsidscC|jS)z@ Fixture for returning the axis numbers of a DataFrame. paramrequestr0r0r1axisr|rlcCrmrnrorqr0r0r1rsrtcCrw)zA Fixture for returning aliases of axis 1 of a DataFrame. rxrzr0r0r1axis_1r}r~)TFN)rucCrw)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). rxrzr0r0r1observed' rcCrw)z6 Boolean 'ordered' parameter for Categorical. rxrzr0r0r1ordered4r}rcCrw)z% Boolean 'skipna' parameter. rxrzr0r0r1skipna<r}r)firstlastFcCrw)z[ Valid values for the 'keep' parameter used in .duplicated or .drop_duplicates rxrzr0r0r1keepDr)bothneitherleftrightcCrw)zA Fixture for trying all interval 'inclusive' parameters. rxrzr0r0r1inclusive_endpoints_fixtureMr}r)rrrrcCrw)z< Fixture for trying all interval closed parameters. rxrzr0r0r1closedUr}rcCrw)zS Secondary closed fixture to allow parametrizing over all pairs of closed. rxrzr0r0r1 other_closed]r}rgzipbz2zipxztarzstd zstandard)markscCrw)zK Fixture for trying common compression types in compression tests. rxrzr0r0r1 compressionercCrw)zk Fixture for trying common compression types in compression tests excluding uncompressed case. rxrzr0r0r1compression_onlywrrcCrw)z, Fixture that an array is writable. rxrzr0r0r1writabler}r)innerouterrrcCrw)z: Fixture for trying all types of join operations. rxrzr0r0r1 join_typer}rnlargest nsmallestcCrw)z1 Fixture for trying all nselect methods. rxrzr0r0r1nselect_methodr}rcC t|jSNtype__name__rqr0r0r1rs cCrw)z/ Fixture for each null type in pandas. rxrzr0r0r1 nulls_fixturer}rcCrw)zL Fixture for each null type in pandas, each null type exactly once. rxrzr0r0r1unique_nulls_fixturer}rcCrrrrqr0r0r1rsrcCrw)z- Fixture for each NaT type in numpy. rxrzr0r0r1np_nat_fixturer}rcCrw)z; Fixture to parametrize over DataFrame and Series. rxrzr0r0r1frame_or_seriesr}rrkseriescCrw)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 rxrzr0r0r1index_or_seriesrr)rkrarraycCrw)zG Fixture to parametrize over Index, Series, and ExtensionArray rxrzr0r0r1index_or_series_or_arrayr}rcCrwr)rrqr0r0r1rscCrw)z] Fixture to test behavior for Index, Series, DataFrame, and pandas Array classes rxrzr0r0r1box_with_arrayrr type[dict]cCsGdddt}|S)z, Fixture for a dictionary subclass. c@seZdZdddZdS)z"dict_subclass..TestSubDictr(r)c_stj|g|Ri|dSr)dict__init__)selfargskwargsr0r0r1rsz+dict_subclass..TestSubDict.__init__Nr(r))r __module__ __qualname__rr0r0r0r1 TestSubDictsr)r)rr0r0r1 dict_subclasssrtype[abc.Mapping]cCsGdddtj}|S)z8 Fixture for a non-mapping dictionary subclass. c@s2eZdZdddZddZddd Zdd d Zd S)z5non_dict_mapping_subclass..TestNonDictMappingr(r)cSs ||_dSr)_data)runderlying_dictr0r0r1r  z>non_dict_mapping_subclass..TestNonDictMapping.__init__cSs |j|Sr)r __getitem__)rkeyr0r0r1rs zAnon_dict_mapping_subclass..TestNonDictMapping.__getitem__r&cS |jSr)r__iter__rr0r0r1rrz>non_dict_mapping_subclass..TestNonDictMapping.__iter__intcSrr)r__len__rr0r0r1rrz=non_dict_mapping_subclass..TestNonDictMapping.__len__Nr)r(r&)r(r)rrrrrrrr0r0r0r1TestNonDictMapping s   r)rMapping)rr0r0r1non_dict_mapping_subclasss rcCsttjddttdtdtddddd }| d d d d d d g }|j dd|j j D|_ |j jgddd|S)z DataFrame with 3 level MultiIndex (year, month, day) covering first 100 business days from 2000-01-01 with random data )dABCDdtype 2000-01-01rBperiodsfreq)rlrkcSrwr)yearrqr0r0r1rs*rzAmultiindex_year_month_day_dataframe_random_data..cSrwr)r^rqr0r0r1rs*rcSrwr)dayrqr0r0r1rs*rcSsg|]}|dqS)i8)astype).0levr0r0r1 ,zCmultiindex_year_month_day_dataframe_random_data..)rr^rT)inplace)rrbrandom default_rngstandard_normalr"listobjectrgroupbysumrk set_levelslevels set_names)tdfymdr0r0r1/multiindex_year_month_day_dataframe_random_datas  rr#cCs*tgdgdggdgdgddgdS)z; 2-level MultiIndex, lexsorted, with string names. foobarbazquxonetwothree) rrrrZrZrrrr) rrZrrrZrZrrrZrrsecond)rcodesnames)r#r0r0r0r1%lexsorted_two_level_string_multiindex1s rrcCs,|}ttjdd|tgddddS)z2DataFrame with 2 level MultiIndex with random datar) r)ArCexpr8rkrl)rrbrrrr")rrkr0r0r1 multiindex_dataframe_random_data=s rcCsTtgd}tddg}tgd}tgd}ddg}t||g||g|dd S) zJ MultiIndex used to test the general functionality of this object rrr)rrrZrrr)rrZrrZrrZrrF)rrrverify_integrity)r"rbrr#) major_axis minor_axis major_codes minor_codes index_namesr0r0r1_create_multiindexJs  rcCs*tjddgddgtddddggd d S) zB MultiIndex with a level that is a tzaware DatetimeIndex. rZrab20130101r US/Easternrtzr)r)r# from_productrr0r0r0r1_create_mi_with_dt64tz_level^sr stringcCg|]}d|qSpandas_r0rir0r0r1rjrrrrz 2020-01-01)rz datetime-tzz US/Pacificr periodDrrz1 day)startrrrangeint8rint16int32int64uint8uint16uint32uint64float32float64z bool-object complex64y? complex128abcde)num)rrr)rZrr)rrrZrZrrInt64UInt16Float32booleancCrrr0rr0r0r1rrzstring[python]) bool-dtyper"r# categoricalintervalemptytuplesmi-with-dt64tz-levelmultirepeats nullable_int nullable_uintnullable_float nullable_boolz string-pythoncCrrr0rr0r0r1rrzstring[pyarrow]zstring-pyarrowcCst|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_dictrycopyrzr0r0r1rks cCsg|] \}}t|ts|qSr0) isinstancer#rrvaluer0r0r1rs cCs|j}t|S)z8 index fixture, but excluding MultiIndex cases. )ryr8r9)r{rr0r0r1 index_flats r=cCs0g|]\}}|ds|dvst|ts|qS))ruintfloat)rr/r3r,) startswithr:r#r;r0r0r1rscCst|jjdd}|j}|jdvr4|}d|ddd|d<d|ddd|d<t|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)r0r1r2rrrZN)r8ryr9valuestolistr# from_tuplesr)r{indvalsr0r0r1index_with_missings    rIrcCs4ttjdtjddtddtdDtdddS)zC Fixture for Series of floats with Index of unique strings rg?cSr)i_r0rr0r0r1rrz!string_series..rrkr8)rrbaranger!r"rrr0r0r0r1 string_seriess rNcCs<ddtdD}tddtdDtd}t||dtdS)zI Fixture for Series of dtype object with Index of unique strings cSrfoo_r0rr0r0r1rrz!object_series..rJcSr)bar_r0rr0r0r1rrrobjectsrkr8r)rr"rr)datarkr0r0r1 object_seriessrUcCs&ttjddtddddddS)z9 Fixture for Series of floats with DatetimeIndex rrJrrrtsrL)rrbrrrrr0r0r0r1datetime_seriess  rWcCs*t|}tjd|}t||dddS)zHelper for the _series dictrrF)rkr8r9)lenrbrrrr)rksizerTr0r0r1_create_seriessrZcCs"i|] \}}d|dt|qS)z series-with-z-indexrZ)rindex_idrkr0r0r1 sr]cCst|S)zE Fixture for tests on series with changing types of indices. r[rkr0r0r1series_with_simple_indexsr_c Cs8i|]}|jdttdddtdDd|dqS)-seriesrJcSr)zi-r0rr0r0r1rrz.rrS)rrrrrr0r0r1r]s  cCt|jjddS)z| Fixture for tests on indexes, series and series with a narrow dtype copy to avoid mutation, e.g. setting .name TrA)_index_or_series_objsryr9rzr0r0r1index_or_series_obj$srdcCsi|] }|jdt|qS)r`)rrrar0r0r1r]-scCrb)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 TrA)_index_or_series_memory_objsryr9rzr0r0r1index_or_series_memory_obj:srfcCs<ttjdtjdtddtdDtdttdtddS)zj Fixture for DataFrame of ints with index of unique strings Columns are ['A', 'B', 'C', 'D'] rJrrcSrrOr0rr0r0r1rPrzint_frame..rJrr)rrbonesrr"rrrr0r0r0r1 int_frameGs ricCs6ttjddtddtdDttddS)zm Fixture for DataFrame of floats with index of unique strings Columns are ['A', 'B', 'C', 'D']. rrgcSrrOr0rr0r0r1r^rzfloat_frame..rJrr)rrbrrrr"rrr0r0r0r1 float_frameUs  rjc Cstdddtdddtdddtdddtdddtdddtdddtdddtdddtdddg }ttjdt||dS)zF Fixture for Series with a DatetimeIndex that has duplicates. irZrrrrYr^)rrrbrrrrX)datesr0r0r1(rand_series_with_duplicate_datetimeindexcs           rl)rrrg?g?z2012-01M)rz period[M]z 2012-02-01z period[D]z 2011-01-01r )r s)unitr i)secondsztimedelta64[ns]cCrwrrxrzr0r0r1ea_scalar_and_dtype{rqcCrw)zD Fixture for dunder names for common arithmetic operations. rxrzr0r0r1all_arithmetic_operatorsr}rscCrw)zU Fixture for operator and roperator arithmetic, comparison, and logical ops. rxrzr0r0r1all_binary_operators"rtcCrw)z Fixture for operator and roperator arithmetic functions. Notes ----- This includes divmod and rdivmod, whereas all_arithmetic_operators does not. rxrzr0r0r1all_arithmetic_functionssrv) countrmaxminmeanprodstdvarmediankurtskewsemcCrw)z. Fixture for numeric reduction names. rxrzr0r0r1all_numeric_reductionsr}rallanycCrw)z. Fixture for boolean reduction names. rxrzr0r0r1all_boolean_reductionsr}rcCrw)z> Fixture for all (boolean + numeric) reduction names. rxrzr0r0r1all_reductionsr}rcCrw)z; Fixture for operator module comparison functions. rxrzr0r0r1 comparison_op rrr)__le____lt____ge____gt__cCrw)zm Fixture for dunder names for compare operations except == and != * >= * > * < * <= rxrzr0r0r1compare_operators_no_eq_nerr)__and____rand____or____ror____xor____rxor__cCrw)zY Fixture for dunder names for common logical operations * | * & * ^ rxrzr0r0r1all_logical_operators( r)cumsumcumprodcummincummaxcCrw)z0 Fixture for numeric accumulation names rxrzr0r0r1all_numeric_accumulations9r}rcCs |dS)zR Returns the configuration for the test setting `--no-strict-data-files`. r*)rL) pytestconfigr0r0r1strict_data_filesDs rCallable[..., str]cs(tjtjtdfdd}|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. testscsJtjjg|R}tj|s#rtd|dtd|d|S)NzCould not find file z' and --no-strict-data-files is not set.zCould not find .)osr5joinexists ValueErrorr;skip)rr5 BASE_PATHrr0r1decoas  zdatapath..deco)rr5rdirname__file__)rrr0rr1datapathLs rz Asia/Tokyozdateutil/US/Pacificzdateutil/Asia/Singaporez+01:15z-02:15z UTC+01:15z UTC-02:15i,i)hoursrCrrcCsg|]}t|qSr0rorr0r0r1rscCrw)zD Fixture for trying timezones including default (None): {0} rxrzr0r0r1tz_naive_fixturerrcCrw)z4 Fixture for trying explicit timezones: {0} rxrzr0r0r1tz_aware_fixturerrrz dateutil/UTCcCrw)zQ Fixture to provide variants of UTC timezone strings and tzinfo objects. rxrzr0r0r1 utc_fixturer}r)rnmsusnscCrw)z& datetime64 units we support. rxrzr0r0r1ror}rocCrw)zR Parametrized fixture for string dtypes. * str * 'str' * 'U' rxrzr0r0r1 string_dtype rpyarrowcCrw)za Parametrized fixture for string dtypes. * 'string[python]' * 'string[pyarrow]' rxrzr0r0r1nullable_string_dtype rpython pyarrow_numpycCrw)zx Parametrized fixture for pd.options.mode.string_storage. * 'python' * 'pyarrow' * 'pyarrow_numpy' rxrzr0r0r1string_storagerrnumpy_nullablecCrw)zb Parametrized fixture for pd.options.mode.string_storage. * 'python' * 'pyarrow' rxrzr0r0r1 dtype_backendrrcCrw)zK Parametrized fixture for bytes dtypes. * bytes * 'bytes' rxrzr0r0r1 bytes_dtypercCrw)zN Parametrized fixture for object dtypes. * object * 'object' rxrzr0r0r1 object_dtyperrrzstring[pyarrow_numpy]cCrw)zo Parametrized fixture for string dtypes. * 'object' * 'string[python]' * 'string[pyarrow]' rxrzr0r0r1any_string_dtyperrcCrw)z\ Parametrized fixture for datetime64 dtypes. * 'datetime64[ns]' * 'M8[ns]' rxrzr0r0r1datetime64_dtype-rrcCrw)z^ Parametrized fixture for timedelta64 dtypes. * 'timedelta64[ns]' * 'm8[ns]' rxrzr0r0r1timedelta64_dtype8rrrc CstddddddddS)z- Fixture emits fixed Timestamp.now() irZr[r )rr^rhourminuter microsecond)rr0r0r0r1 fixed_now_tsCsrcCrw)z^ Parameterized fixture for float dtypes. * float * 'float32' * 'float64' rxrzr0r0r1float_numpy_dtypeMrrcCrw)zR Parameterized fixture for float dtypes. * 'Float32' * 'Float64' rxrzr0r0r1float_ea_dtypeYrrcCrw)z~ Parameterized fixture for float dtypes. * float * 'float32' * 'float64' * 'Float32' * 'Float64' rxrzr0r0r1any_float_dtypedrrcCrw)zg Parameterized fixture for complex dtypes. * complex * 'complex64' * 'complex128' rxrzr0r0r1 complex_dtyperrrcCrw)z Parameterized fixture for complex and numpy float dtypes. * complex * 'complex64' * 'complex128' * float * 'float32' * 'float64' rxrzr0r0r1complex_or_float_dtype~s rcCrw)z| Parameterized fixture for signed integer dtypes. * int * 'int8' * 'int16' * 'int32' * 'int64' rxrzr0r0r1any_signed_int_numpy_dtyperrcCrw)zx Parameterized fixture for unsigned integer dtypes. * 'uint8' * 'uint16' * 'uint32' * 'uint64' rxrzr0r0r1any_unsigned_int_numpy_dtyperrcCrw)z Parameterized fixture for any integer dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' rxrzr0r0r1any_int_numpy_dtyperrcCrw)z Parameterized fixture for any nullable integer dtype. * 'UInt8' * 'Int8' * 'UInt16' * 'Int16' * 'UInt32' * 'Int32' * 'UInt64' * 'Int64' rxrzr0r0r1any_int_ea_dtyperrrcCrw)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' rxrzr0r0r1 any_int_dtypesrcCrw)z Parameterized fixture for any nullable integer dtype and any float ea dtypes. * 'UInt8' * 'Int8' * 'UInt16' * 'Int16' * 'UInt32' * 'Int32' * 'UInt64' * 'Int64' * 'Float32' * 'Float64' rxrzr0r0r1any_numeric_ea_dtypesrcCrw)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]' rxrzr0r0r1any_numeric_ea_and_arrow_dtypes rcCrw)z~ Parameterized fixture for any signed nullable integer dtype. * 'Int8' * 'Int16' * 'Int32' * 'Int64' rxrzr0r0r1any_signed_int_ea_dtyperrcCrw)z Parameterized fixture for any (purely) real numeric dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' * float * 'float32' * 'float64' rxrzr0r0r1any_real_numpy_dtype+srcCrw)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. rxrzr0r0r1any_real_numeric_dtype@srcCrw)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' rxrzr0r0r1any_numpy_dtypeWrurcCrw)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]' rxrzr0r0r1any_real_nullable_dtype|srcCrw)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' rxrzr0r0r1any_numeric_dtypesrrcmixedbytesacr/z mixed-integerrg@floatingg?integerzmixed-integer-floatdecimal datetime64z 2013-01-01z 2018-01-01r20180101riicomplexy??y@@rr.cCs |j\}}tj|td}||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_dtyperDr0r0r1any_skipna_inferred_dtypes )rcCs>tjdddddlm}ddlm}|}d|j_||dS) zj Get an instance of IPython.InteractiveShell. Will raise a skip if IPython is not installed. IPythonz6.0.0) minversionr)InteractiveShell)Configz:memory:)rN)r; importorskipIPython.core.interactiveshellrtraitlets.configrHistoryManager hist_file)rrrr0r0r1ips    r)bsrcoocsccsrdiadoklilcCstd}t||jdS)z- Yields scipy sparse matrix classes. z scipy.sparse_matrix)r;rgetattrry)r{sparser0r0r1spmatrix#s rcCs6g|]}tttj|tjjr|dkrttj|qS)Tick) issubclassrrcoffsetsr)ror0r0r1r.s cCrw)zN Fixture for Tick based datetime offsets available for a time series. rxrzr0r0r1 tick_classes-rrcCs|Srr0rqr0r0r1rs;scCrw)zk Simple fixture for testing keys in sorting methods. Tests None (no key) and the identity key. rxrzr0r0r1 sort_by_key;rr)rNN)EgonVenkmanN)NCC1701Drr#tuple[Hashable, Hashable, Hashable]cCrw)zP A 3-tuple of names, the first two for operands, the last for a result. rxrzr0r0r1rDsrcCrw)zI Parametrize over __setitem__, loc.__setitem__, iloc.__setitem__ rxrzr0r0r1 indexer_sliWr}rcCrw)z< Parametrize over loc.__getitem__, iloc.__getitem__ rxrzr0r0r1 indexer_li_r}rcCrw)z8 Parametrize over __setitem__, iloc.__setitem__ rxrzr0r0r1 indexer_sigr}rcCrw)z7 Parametrize over __setitem__, loc.__setitem__ rxrzr0r0r1 indexer_slor}rcCrw)z: Parametrize over at.__setitem__, loc.__setitem__ rxrzr0r0r1 indexer_alwr}r cCrw)z< Parametrize over iat.__setitem__, iloc.__setitem__ rxrzr0r0r1 indexer_ialr}r boolcCstddddkS)z> Fixture to check if the array manager is being used. mode.data_managerTsilentrrr0r0r0r1using_array_managerrjrcCstjjjduotddddkS)z7 Fixture to check if Copy-on-Write is enabled. Tr r blockrcoptionsmode copy_on_writerr0r0r0r1using_copy_on_writercCstjjjdkotddddkS)z? Fixture to check if Copy-on-Write is in warning mode. warnr Tr rrr0r0r0r1warn_copy_on_writerrcCstjjjduS)z= Fixture to check if infer string option is enabled. T)rcrfuture infer_stringr0r0r0r1using_infer_stringsrz Europe/Warsawzdateutil/Europe/WarsawcCrw)zE tzinfo for Europe/Warsaw using pytz, dateutil, or zoneinfo. rxrzr0r0r1warsawr}rcCsdS)N)rrr0r0r0r0r1arrow_string_storagesrr)r3r4r5r6r7r6r(r))r(r)r(r)r(r#)r(r)r(r)rr6r(r)r(r)r(r)r(r )r(r6(:__doc__ __future__r collectionsrrrrrrrr operatorrtypingr r dateutil.tzr r hypothesisrstnumpyrbr;pytzrrpandas._config.configrpandas.util._test_decoratorsutil_test_decoratorstdpandas.core.dtypes.dtypesrrpandasrcrrrrrrrrrrrr pandas._testing_testingtm pandas.corer!pandas.core.indexes.apir"r#pandas.util.versionr$collections.abcr%r&rpa ImportError has_pyarrowzoneinfoZoneInfoZoneInfoNotFoundErrorr2r>rQ HealthChecktoo_slowhypothesis_health_checks __version__appenddiffering_executorssettingsregister_profiletuple load_profilesplitr8rtseriesrclsregister_type_strategybuildsintegersbooleansfixturererir| axis_framer~rrrrrrrry skip_if_norrrrr NULL_OBJECTSrnulls_fixture2nanNaTrunique_nulls_fixture2NP_NAT_OBJECTSrnp_nat_fixture2rrindex_or_series2rrrbox_with_array2rrrrrrr rrMrr r from_breakslinspacerFrrr8idxkeysrkindex_fixture2rMr= index_flat2rIrNrUrWrZ_seriesr_NARROW_NP_DTYPES_narrow_seriesrcrdPYTHON_DATA_TYPES_typ_objects_seriesrerfrirjrlrqarithmetic_dunder_methodsrsaddraddsubrsubmulrmultruedivrtruedivfloordiv rfloordivmodrmodpowrpoweqneltlegtgeand_rand_xorrxoror_ror_rtrv_all_numeric_reductionsr_all_boolean_reductionsr_all_reductionsrrrr_all_numeric_accumulationsrrr TIMEZONESextend TIMEZONE_IDSparametrize_fixture_docr6rrtz_aware_fixture2_UTCSr utc_fixture2rounit2 STRING_DTYPESrrrrstring_storage2 BYTES_DTYPESr OBJECT_DTYPESrrDATETIME64_DTYPESrTIMEDELTA64_DTYPESrrFLOAT_NUMPY_DTYPESrFLOAT_EA_DTYPESrALL_FLOAT_DTYPESrCOMPLEX_DTYPESrCOMPLEX_FLOAT_DTYPESrSIGNED_INT_NUMPY_DTYPESrUNSIGNED_INT_NUMPY_DTYPESrALL_INT_NUMPY_DTYPESrALL_INT_EA_DTYPESrALL_INT_DTYPESrrALL_INT_PYARROW_DTYPES_STR_REPRFLOAT_PYARROW_DTYPES_STR_REPRrSIGNED_INT_EA_DTYPESrALL_REAL_NUMPY_DTYPESrALL_REAL_DTYPESrALL_NUMPY_DTYPESrALL_REAL_NULLABLE_DTYPESrALL_NUMERIC_DTYPESrNAr_any_skipna_inferred_dtyperv_rrr__all__rrrsetitemlocilocrrrratr iatr rrrrwarsawsrrr0r0r0r1sp     8        <                                           (                            %                       $#$$2