o Ehe@s dZddlmZddlmZddlmZmZmZmZm 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:ddedDedde:ddedDddde1dddde1ddddde2ddddde3ddddde-dde:edddde:edddde:edddde:edddde:edddde:edddde:edddde:edddde:edddde:edddide:ddgd+edde:ddgd+edde:ejddddejdddde:ejddddejdddde(eddde+ejddddde:gde;egdgddedede:gdde:edddde:edddde:edddde:ededdde:e'jddedDddZeDrxe:e'jddedDddZeed<ej_edDdduZeZej_ddeDdDddӄZeZej_ddeDdDddքZej_dddلZej_dddۄZej_ddd݄Zdd߄ZddeDZej_dddZdde6jDZieeeZej_edDddZdde6jDZieeeeZej_edDddZej_dddZej_dddZej_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 je8jgdDdd Zgd Zej_edDd d Zϐd dgZej_edDddZeeZej_edDddZej_e je je je je je jgdDddZej_gddDddZej_gddDddZgdZej_edDddZej_ddZej_dϐd!d"Zdddd#d$d%d&d'd(d)eeed*eded+e je ed,d,e ed-d,d.d/gZeEdure۠eEFdeEFdgd0deDZe"ee݃ej_eed<d1d2Ze"eed,dej_ed,ded,dd<d3d4ZeZd5d6eee jgZeEdureOeEFdej_edDd7d8ZeZej_gd9dDd:d;ZeZej_e6jdDd<d=Zej_d>e'jfejld?e'jfe"md?d]ejld?ejvfe"md?d]d>ejvfggd@d<dAdBZej_dejlde"md?d]gdDdCdDZej_ejld?ejvfe"md?d]ejld?e'jfe"md?d]gdDdEdFZej_d>ejld?e"md?d]gdDdGdHZej_d>e'jfejld?e'jfe"md?d]ejld?ejvfe"md?d]d>ejvfggd@d<dIdJZej_dKejld?e"md?d]gdDdLdMZeZeZej_e6jdDdNdOZej_e6jdDdPdQZej_edd>e'jfejld?e'jfe"md?d]ejld?ejvfe"md?d]d>ejvfggdRd<dSdTZej_e6jdDdUdVZej_e6jdDdWdXZej_dАdZd[Zej_e6jdDd\d]Zej_e6jdDd^d_Zej_e6jdDd`daZej_e6jdDdbdcZej_e6jdDdddeZej_e6jdDdfdgZ ej_e6j dDdhdiZ ej_e6j dDdjdkZ ej_e6jdDdldmZej_e6jdDdndoZej_e6je6jdDdpdqZej_e6je6je6je6jdDdrdsZej_e6jdDdtduZej_e6jdDdvdwZej_e6jdDdxdyZej_e6jdDdzd{Zej_e6jdDd|d}Zej_e6j dDd~dZ!ddejvdgfdde'jdgfdde'jwdgfddejvdgfdejvejvejvgfdgfddejvdgfddejvdgfddejvdgfdd,ejvdgfdd,ejvdgfde d,ejve dgfddejvdgfdde'jdgfde"dejve"dgfde0dejve0dgfdedd,d,ejvedd,d,gfddejvdgfded,ejvedgfded,ejvedgfde,de'jwe,dgfde*dd,ejve*ddgfgZ#ee#\Z$Z%ej_e#e$d<ddZ&ej_ddZ'ej_gddDddZ(ej_dde'jYj)DdDddZ*ej_ddd;gdDddZ+ej_dddejvejvejvfejve'jwdfejve'jdfe'je'je'jfgdDdѐddZ,ej_e6j-e6j.e6j/gdDddZ0ej_e6j.e6j/gdDddZ1ej_e6j-e6j/gdDddZ2ej_e6j-e6j.gdDddZ3ej_e6j4e6j.gdDddZ5ej_e6j6e6j/gdDddZ7ej_dҐddZ8ej_dҐddZ9ej_dҐddZ:ej_dҐddZ;ddgZdS(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)HashableIteratorFTZUTCreturnNonecCs|jdddddS)N--no-strict-data-files store_falsez6Don't fail if a test is skipped for missing data file.)actionhelp)Z addoption)parserr./lib/python3.10/site-packages/pandas/conftest.pypytest_addoptionos  r0item 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)nameendswithZ add_markerpytestZmarkfilterwarnings)r1r3r5r.r.r/ignore_doctest_warningws r:cCsP|dp |jddd}gd}|r$|D]}|D] \}}t|||qqdSdS)Nz--doctest-modulesz--doctest-cythonF)default))Zis_int64_dtypezis_int64_dtype is deprecated)Zis_interval_dtypezis_interval_dtype is deprecated)Zis_period_dtypezis_period_dtype is deprecated)Zis_datetime64tz_dtypez#is_datetime64tz_dtype is deprecated)Zis_categorical_dtypez"is_categorical_dtype is deprecated)Z is_sparsezis_sparse is deprecated)DataFrameGroupBy.fillnaz%DataFrameGroupBy.fillna is deprecated)NDFrame.replacezThe 'method' keyword)r=zSeries.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.idxminr>)zSeriesGroupBy.idxmaxr?)zmissing.mask_zero_div_zerozdivide by zero encountered)Z 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)r<z3DataFrameGroupBy.fillna with 'method' is deprecated)r<z,DataFrame.fillna with 'method' is deprecated)Z read_parquetz1Passing a BlockManager to DataFrame is deprecated) getoptionr:)itemsconfigZ is_doctestZignored_doctest_warningsr1r3r5r.r.r/pytest_collection_modifyitemss/ rCz6.83.2Zci)ZdeadlineZsuppress_health_checkz)MonthBegin MonthEnd BMonthBegin BMonthEndic)n normalizez%YearBegin YearEnd BYearBegin BYearEnd )Z min_valueZ max_value)rErFmonthz1QuarterBegin QuarterEnd BQuarterBegin BQuarterEndi)rErFZ startingMonth)ZautousecCst|d<t|d<dS)z: Make `np` and `pd` names available for doctests. nppdN)rMrN)Zdoctest_namespacer.r.r/add_doctest_importss rOcCstdddS)z< Configure settings for all tests and test modules. Zchained_assignmentraiseN)rNZ set_optionr.r.r.r/configure_tests rQ)rrIindexcolumnscCdt|SNzaxis=reprxr.r.r/r[)paramsidscC|jS)z@ Fixture for returning the axis numbers of a DataFrame. paramrequestr.r.r/axisrdrTcCrUrVrWrYr.r.r/r[r\cCr_)zA Fixture for returning aliases of axis 1 of a DataFrame. r`rbr.r.r/axis_1rerf)TFN)r]cCr_)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). r`rbr.r.r/observed' rgcCr_)z6 Boolean 'ordered' parameter for Categorical. r`rbr.r.r/ordered4rericCr_)z% Boolean 'skipna' parameter. r`rbr.r.r/skipna<rerj)firstZlastFcCr_)z[ Valid values for the 'keep' parameter used in .duplicated or .drop_duplicates r`rbr.r.r/keepDrl)bothneitherleftrightcCr_)zA Fixture for trying all interval 'inclusive' parameters. r`rbr.r.r/inclusive_endpoints_fixtureMrerr)rprqrnrocCr_)z< Fixture for trying all interval closed parameters. r`rbr.r.r/closedUrerscCr_)zS Secondary closed fixture to allow parametrizing over all pairs of closed. r`rbr.r.r/ other_closed]rertgzipbz2zipxztarZzstdZ zstandard)ZmarkscCr_)zK Fixture for trying common compression types in compression tests. r`rbr.r.r/ compressionerzcCr_)zk Fixture for trying common compression types in compression tests excluding uncompressed case. r`rbr.r.r/compression_onlywr{r|cCr_)z, Fixture that an array is writable. r`rbr.r.r/writablerer})innerZouterrprqcCr_)z: Fixture for trying all types of join operations. r`rbr.r.r/ join_typerernlargestZ nsmallestcCr_)z1 Fixture for trying all nselect methods. r`rbr.r.r/nselect_methodrercC t|jSNtype__name__rYr.r.r/r[ cCr_)z/ Fixture for each null type in pandas. r`rbr.r.r/ nulls_fixturerercCr_)zL Fixture for each null type in pandas, each null type exactly once. r`rbr.r.r/unique_nulls_fixturerercCrrrrYr.r.r/r[rcCr_)z- Fixture for each NaT type in numpy. r`rbr.r.r/np_nat_fixturerercCr_)z; Fixture to parametrize over DataFrame and Series. r`rbr.r.r/frame_or_seriesrerrSseriescCr_)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 r`rbr.r.r/index_or_seriesrhr)rSrarraycCr_)zG Fixture to parametrize over Index, Series, and ExtensionArray r`rbr.r.r/index_or_series_or_arrayrercCr_r)rrYr.r.r/r[cCr_)z] Fixture to test behavior for Index, Series, DataFrame, and pandas Array classes r`rbr.r.r/box_with_arrayrmr type[dict]cCsGdddt}|S)z, Fixture for a dictionary subclass. c@seZdZdddZdS)z"dict_subclass..TestSubDictr'r(c_stj|g|Ri|dSr)dict__init__)selfargskwargsr.r.r/rsz+dict_subclass..TestSubDict.__init__Nr'r()r __module__ __qualname__rr.r.r.r/ TestSubDictsr)r)rr.r.r/ 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)rZunderlying_dictr.r.r/r  z>non_dict_mapping_subclass..TestNonDictMapping.__init__cSs |j|Sr)r __getitem__)rkeyr.r.r/rs zAnon_dict_mapping_subclass..TestNonDictMapping.__getitem__r&cS |jSr)r__iter__rr.r.r/rrz>non_dict_mapping_subclass..TestNonDictMapping.__iter__intcSrr)r__len__rr.r.r/rrz=non_dict_mapping_subclass..TestNonDictMapping.__len__Nr)r'r&)r'r)rrrrrrrr.r.r.r/TestNonDictMapping s   r)rMapping)rr.r.r/non_dict_mapping_subclasss rcCsttjddttdtddddd}|d d d d d d g }|j d d|j j D|_ |j j gddd|S)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)rTrScSr_r)yearrYr.r.r/r[*rzAmultiindex_year_month_day_dataframe_random_data..cSr_r)rKrYr.r.r/r[*rcSr_r)dayrYr.r.r/r[*rcSsg|]}|dqS)Zi8)astype).0Zlevr.r.r/ ,zCmultiindex_year_month_day_dataframe_random_data..)rrKrT)Zinplace)rrMrandom default_rngstandard_normalr"listrgroupbysumrSZ set_levelslevelsZ set_names)ZtdfZymdr.r.r//multiindex_year_month_day_dataframe_random_datas   rr#cCs*tgdgdggdgdgddgdS)z; 2-level MultiIndex, lexsorted, with string names. foobarbazZquxonetwoZthree) rrrrIrIrrrr) rrIrrrIrIrrrIrrksecond)rcodesnames)r#r.r.r.r/%lexsorted_two_level_string_multiindex1s rrcCs,|}ttjdd|tgddddS)z2DataFrame with 2 level MultiIndex with random datar) r)ArCZexpr6rSrT)rrMrrrr")rrSr.r.r/ 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)rrrIrrr)rrIrrIrrIrkrF)rrrZverify_integrity)r"rMrr#)Z major_axisZ minor_axisZ major_codesZ minor_codesZ index_namesr.r.r/_create_multiindexJs  rcCs*tjddgddgtddddggd d S) zB MultiIndex with a level that is a tzaware DatetimeIndex. rIrab20130101r US/Easternrtzr)r)r#Z from_productrr.r.r.r/_create_mi_with_dt64tz_level^srobjectcCg|]}d|qSZpandas_r.rir.r.r/rjrrrdtypestringcCrrr.rr.r.r/rkrrz 2020-01-01)rz datetime-tzz US/PacificrZperiodDrrz1 day)startrrrangeZint8Zint16Zint32int64Zuint8Zuint16Zuint32Zuint64Zfloat32float64z bool-object bool-dtypeZ complex64y?Z complex128Z categoricalZabcdintervale)Znumemptytuples)rrr)rIrrmi-with-dt64tz-levelmultirepeats)rrrIrIrrZ nullable_intZInt64Z nullable_uintZUInt16Znullable_floatZFloat32Z nullable_boolZbooleanz string-pythoncCrrr.rr.r.r/rrzstring[python]cCrrr.rr.r.r/rrzstring[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_dictracopyrbr.r.r/rSs cCsg|] \}}t|ts|qSr.) isinstancer#rrvaluer.r.r/rs cCs|j}t|S)z8 index fixture, but excluding MultiIndex cases. )rarr)rcrr.r.r/ index_flats rcCs0g|]\}}|ds|dvst|ts|qS))rZuintfloat)rrrr) startswithrr#rr.r.r/rscCst|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. TZdeep)rrrrrrIN)rrarvaluestolistr# from_tuplesr)rcZindZvalsr.r.r/index_with_missings    rrcCs0ttjdtjddtddtdDddS)zC Fixture for Series of floats with Index of unique strings rg?cSr)Zi_r.rr.r.r/rrz!string_series..rrSr6)rrMarangerr"rr.r.r.r/ string_seriess rcCs8ddtdD}tddtdD}t||dtdS)zI Fixture for Series of dtype object with Index of unique strings cSrZfoo_r.rr.r.r/rrz!object_series..rcSr)Zbar_r.rr.r.r/rrZobjectsrSr6r)rr"rr)datarSr.r.r/ object_seriessrcCs&ttjddtddddddS)z9 Fixture for Series of floats with DatetimeIndex rrrrrtsr)rrMrrrrr.r.r.r/datetime_seriess  rcCs*t|}tjd|}t||dddS)zHelper for the _series dictrrF)rSr6r)lenrMrrrr)rSsizerr.r.r/_create_seriessr cCs"i|] \}}d|dt|qS)z series-with-z-indexr )rZindex_idrSr.r.r/ sr cCst|S)zE Fixture for tests on series with changing types of indices. r rSr.r.r/series_with_simple_indexsrc Cs8i|]}|jdttdddtdDd|dqS)-seriesrcSr)zi-r.rr.r.r/rrz.rr)rrrrrr.r.r/r s  cCt|jjddS)z| Fixture for tests on indexes, series and series with a narrow dtype copy to avoid mutation, e.g. setting .name Tr)_index_or_series_objsrarrbr.r.r/index_or_series_obj%srcCsi|] }|jdt|qS)r)rrrr.r.r/r .scCr)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 Tr)_index_or_series_memory_objsrarrbr.r.r/index_or_series_memory_obj;srcCs4ttjdtjdtddtdDttddS)zj Fixture for DataFrame of ints with index of unique strings Columns are ['A', 'B', 'C', 'D'] rrrcSrrr.rr.r.r/rQrzint_frame..rrr)rrMZonesrr"rrr.r.r.r/ int_frameHs  rcCs6ttjddtddtdDttddS)zm Fixture for DataFrame of floats with index of unique strings Columns are ['A', 'B', 'C', 'D']. rrcSrrr.rr.r.r/r_rzfloat_frame..rrr)rrMrrrr"rrr.r.r.r/ float_frameVs  rc Cstdddtdddtdddtdddtdddtdddtdddtdddtdddtdddg }ttjdt||dS)zF Fixture for Series with a DatetimeIndex that has duplicates. irIrrrrHr)rrrMrrrr )Zdatesr.r.r/(rand_series_with_duplicate_datetimeindexds           r)rprqrqg?g?z2012-01M)rz period[M]z 2012-02-01z period[D]z 2011-01-01r)rs)unitri)Zsecondsztimedelta64[ns]cCr_rr`rbr.r.r/ea_scalar_and_dtype|rcCr_)zD Fixture for dunder names for common arithmetic operations. r`rbr.r.r/all_arithmetic_operatorsrer cCr_)zU Fixture for operator and roperator arithmetic, comparison, and logical ops. r`rbr.r.r/all_binary_operators"r!cCr_)z Fixture for operator and roperator arithmetic functions. Notes ----- This includes divmod and rdivmod, whereas all_arithmetic_operators does not. r`rbr.r.r/all_arithmetic_functionssr#) countrmaxminZmeanZprodZstdvarZmedianZkurtZskewZsemcCr_)z. Fixture for numeric reduction names. r`rbr.r.r/all_numeric_reductionsrer(allanycCr_)z. Fixture for boolean reduction names. r`rbr.r.r/all_boolean_reductionsrer+cCr_)z> Fixture for all (boolean + numeric) reduction names. r`rbr.r.r/all_reductionsrer,cCr_)z; Fixture for operator module comparison functions. r`rbr.r.r/ comparison_op rr-)__le____lt____ge____gt__cCr_)zm Fixture for dunder names for compare operations except == and != * >= * > * < * <= r`rbr.r.r/compare_operators_no_eq_nerhr2)__and____rand____or____ror____xor____rxor__cCr_)zY Fixture for dunder names for common logical operations * | * & * ^ r`rbr.r.r/all_logical_operators) r9)ZcumsumZcumprodZcumminZcummaxcCr_)z0 Fixture for numeric accumulation names r`rbr.r.r/all_numeric_accumulations:rer;cCs |dS)zR Returns the configuration for the test setting `--no-strict-data-files`. r))r@)Z pytestconfigr.r.r/strict_data_filesEs r<Callable[..., 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. ZtestscsJtjjg|R}tj|s#rtd|dtd|d|S)NzCould not find file z' and --no-strict-data-files is not set.zCould not find .)osr3joinexists ValueErrorr8skip)rr3Z BASE_PATHr<r.r/decobs  zdatapath..deco)r?r3r@dirname__file__)r<rEr.rDr/datapathMs rHz Asia/Tokyozdateutil/US/Pacificzdateutil/Asia/Singaporez+01:15z-02:15z UTC+01:15z UTC-02:15i,i)ZhoursrrrcCsg|]}t|qSr.rWrr.r.r/rscCr_)zD Fixture for trying timezones including default (None): {0} r`rbr.r.r/tz_naive_fixturermrIcCr_)z4 Fixture for trying explicit timezones: {0} r`rbr.r.r/tz_aware_fixturermrJrz dateutil/UTCcCr_)zQ Fixture to provide variants of UTC timezone strings and tzinfo objects. r`rbr.r.r/ utc_fixturererK)rmsusnscCr_)z& datetime64 units we support. r`rbr.r.r/rrercCr_)zR Parametrized fixture for string dtypes. * str * 'str' * 'U' r`rbr.r.r/ string_dtype rOZpythonpyarrow)string=string[python]string=string[pyarrow]string=str[pyarrow]string=str[python]cCs|j\}}t||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) )rarN StringDtypercZstorageZna_valuer.r.r/string_dtype_no_objects  rXcCr_)za Parametrized fixture for string dtypes. * 'string[python]' * 'string[pyarrow]' r`rbr.r.r/nullable_string_dtype rYcCs tj|jS)zq Parametrized fixture for string dtypes backed by Pyarrow. * 'str[pyarrow]' * 'string[pyarrow]' )rNrVrarbr.r.r/pyarrow_string_dtypes r[cCr_zb Parametrized fixture for pd.options.mode.string_storage. * 'python' * 'pyarrow' r`rbr.r.r/string_storage rZr]cCr_)z Parametrized fixture for StringDtype storage and na_value. * 'python' + pd.NA * 'pyarrow' + pd.NA * 'pyarrow' + np.nan r`rbr.r.r/string_dtype_argumentssr^Znumpy_nullablecCr_r\r`rbr.r.r/ dtype_backend5rZr_cCr_)zK Parametrized fixture for bytes dtypes. * bytes * 'bytes' r`rbr.r.r/ bytes_dtypeJr`cCr_)zN Parametrized fixture for object dtypes. * object * 'object' r`rbr.r.r/ object_dtypeUrarb)z string=objectrRrSrTrUcCs*t|jtjr |jS|j\}}t||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) )rrarMrrNrVrWr.r.r/any_string_dtype`s  rccCr_)z\ Parametrized fixture for datetime64 dtypes. * 'datetime64[ns]' * 'M8[ns]' r`rbr.r.r/datetime64_dtyperardcCr_)z^ Parametrized fixture for timedelta64 dtypes. * 'timedelta64[ns]' * 'm8[ns]' r`rbr.r.r/timedelta64_dtyperarerc CstddddddddS)z- Fixture emits fixed Timestamp.now() irIrJr )rrKrZhourZminuterZ microsecond)rr.r.r.r/ fixed_now_tssrhcCr_)z^ Parameterized fixture for float dtypes. * float * 'float32' * 'float64' r`rbr.r.r/float_numpy_dtyperPricCr_)zR Parameterized fixture for float dtypes. * 'Float32' * 'Float64' r`rbr.r.r/float_ea_dtyperarjcCr_)z~ Parameterized fixture for float dtypes. * float * 'float32' * 'float64' * 'Float32' * 'Float64' r`rbr.r.r/any_float_dtyper:rkcCr_)zg Parameterized fixture for complex dtypes. * complex * 'complex64' * 'complex128' r`rbr.r.r/ complex_dtyperPrlcCr_)z Parameterized fixture for complex and numpy float dtypes. * complex * 'complex64' * 'complex128' * float * 'float32' * 'float64' r`rbr.r.r/complex_or_float_dtypes rmcCr_)z| Parameterized fixture for signed integer dtypes. * int * 'int8' * 'int16' * 'int32' * 'int64' r`rbr.r.r/any_signed_int_numpy_dtyper:rncCr_)zx Parameterized fixture for unsigned integer dtypes. * 'uint8' * 'uint16' * 'uint32' * 'uint64' r`rbr.r.r/any_unsigned_int_numpy_dtyperhrocCr_)z Parameterized fixture for any integer dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' r`rbr.r.r/any_int_numpy_dtyper{rpcCr_)z Parameterized fixture for any nullable integer dtype. * 'UInt8' * 'Int8' * 'UInt16' * 'Int16' * 'UInt32' * 'Int32' * 'UInt64' * 'Int64' r`rbr.r.r/any_int_ea_dtyperrqcCr_)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' r`rbr.r.r/ any_int_dtype srrcCr_)z Parameterized fixture for any nullable integer dtype and any float ea dtypes. * 'UInt8' * 'Int8' * 'UInt16' * 'Int16' * 'UInt32' * 'Int32' * 'UInt64' * 'Int64' * 'Float32' * 'Float64' r`rbr.r.r/any_numeric_ea_dtype:srscCr_)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]' r`rbr.r.r/any_numeric_ea_and_arrow_dtypePs rtcCr_)z~ Parameterized fixture for any signed nullable integer dtype. * 'Int8' * 'Int16' * 'Int32' * 'Int64' r`rbr.r.r/any_signed_int_ea_dtypesrhrucCr_)z Parameterized fixture for any (purely) real numeric dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' * float * 'float32' * 'float64' r`rbr.r.r/any_real_numpy_dtypesrvcCr_)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. r`rbr.r.r/any_real_numeric_dtypesrwcCr_)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' r`rbr.r.r/any_numpy_dtyper"rxcCr_)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]' r`rbr.r.r/any_real_nullable_dtypesrycCr_)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' r`rbr.r.r/any_numeric_dtypesrzrcZmixedbytesacz mixed-integerrg@Zfloatingg?Zintegerzmixed-integer-floatdecimal datetime64z 2013-01-01z 2018-01-01rZ20180101riicomplexy??y@@rcCs |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)rarMrr)rcZinferred_dtyperr.r.r/any_skipna_inferred_dtype2s )rcCs>tjdddddlm}ddlm}|}d|j_||dS) zj Get an instance of IPython.InteractiveShell. Will raise a skip if IPython is not installed. ZIPythonz6.0.0)Z minversionr)InteractiveShell)Configz:memory:)rB)r8 importorskipZIPython.core.interactiveshellrZtraitlets.configrZHistoryManagerZ hist_file)rrr{r.r.r/ipes    r)ZbsrZcooZcscZcsrZdiaZdokZlilcCstd}t||jdS)z- Yields scipy sparse matrix classes. z scipy.sparseZ_matrix)r8rgetattrra)rcZsparser.r.r/spmatrixxs rcCs6g|]}tttj|tjjr|dkrttj|qS)Tick) issubclassrrNoffsetsr)ror.r.r/rs cCr_)zN Fixture for Tick based datetime offsets available for a time series. r`rbr.r.r/ tick_classesr:rcCs|Srr.rYr.r.r/r[scCr_)zk Simple fixture for testing keys in sorting methods. Tests None (no key) and the identity key. r`rbr.r.r/ sort_by_keyrmr)rNN)ZEgonZVenkmanN)NCC1701Drr#tuple[Hashable, Hashable, Hashable]cCr_)zP A 3-tuple of names, the first two for operands, the last for a result. r`rbr.r.r/rsrcCr_)zI Parametrize over __setitem__, loc.__setitem__, iloc.__setitem__ r`rbr.r.r/ indexer_slirercCr_)z< Parametrize over loc.__getitem__, iloc.__getitem__ r`rbr.r.r/ indexer_lirercCr_)z8 Parametrize over __setitem__, iloc.__setitem__ r`rbr.r.r/ indexer_sirercCr_)z7 Parametrize over __setitem__, loc.__setitem__ r`rbr.r.r/ indexer_slrercCr_)z: Parametrize over at.__setitem__, loc.__setitem__ r`rbr.r.r/ indexer_alrercCr_)z< Parametrize over iat.__setitem__, iloc.__setitem__ r`rbr.r.r/ indexer_ialrerboolcCstddddkS)z> Fixture to check if the array manager is being used. mode.data_managerTZsilentrrr.r.r.r/using_array_managerrRrcCstjjjduotddddkS)z7 Fixture to check if Copy-on-Write is enabled. TrrblockrNoptionsmodeZ copy_on_writerr.r.r.r/using_copy_on_writercCstjjjdkotddddkS)z? Fixture to check if Copy-on-Write is in warning mode. warnrTrrrr.r.r.r/warn_copy_on_writerrcCstjjjduS)z= Fixture to check if infer string option is enabled. T)rNrZfutureZ infer_stringr.r.r.r/using_infer_stringsrz Europe/Warsawzdateutil/Europe/WarsawcCr_)zE tzinfo for Europe/Warsaw using pytz, dateutil, or zoneinfo. r`rbr.r.r/warsawrercCsdS)N)rQZ pyarrow_numpyr.r.r.r.r/arrow_string_storagesrr)r1r2r3r4r5r4r'r()r'r)r'r)r'r#)r'r)r'r)r<r4r'r=)r'r)r'r)r'r)r'r4(?__doc__Z __future__r collectionsrrrrrrrr operatorr?typingr r Z dateutil.tzr r Z hypothesisrstZnumpyrMr8ZpytzrrZpandas._config.configrZpandas.util._test_decoratorsutilZ_test_decoratorsZtdZpandas.core.dtypes.dtypesrrZpandasrNrrrrrrrrrrrr Zpandas._testingZ_testingZtmZ pandas.corer!Zpandas.core.indexes.apir"r#Zpandas.util.versionr$Zcollections.abcr%r&rQpa ImportErrorZ has_pyarrowZzoneinfoZZoneInfoZZoneInfoNotFoundErrorr0r:rCZ HealthCheckZtoo_slowZhypothesis_health_checks __version__appendZdiffering_executorsZsettingsZregister_profiletupleZ load_profilesplitr6rZtseriesrclsZregister_type_strategyZbuildsZintegersZbooleansZfixturerOrQrdZ axis_framerfrgrirjrlrrrsrtraZ skip_if_norzr|r}rrZ NULL_OBJECTSrZnulls_fixture2nanZNaTrZunique_nulls_fixture2ZNP_NAT_OBJECTSrZnp_nat_fixture2rrZindex_or_series2rrrZbox_with_array2rrrrrrrrrrrrZ from_breaksZlinspacerrwrridxkeysrSZindex_fixture2rArZ index_flat2rrrrr Z_seriesrZNARROW_NP_DTYPESZ_narrow_seriesrrZPYTHON_DATA_TYPESZ_typ_objects_seriesrrrrrrZarithmetic_dunder_methodsr addZraddsubZrsubmulZrmultruedivZrtruedivfloordivZ rfloordivmodZrmodpowZrpoweqneltlegtgeand_Zrand_xorZrxoror_Zror_r!r#Z_all_numeric_reductionsr(Z_all_boolean_reductionsr+Z_all_reductionsr,r-r2r9Z_all_numeric_accumulationsr;r<rHZ TIMEZONESextendZ TIMEZONE_IDSZparametrize_fixture_docr4rIrJZtz_aware_fixture2Z_UTCSrKZ utc_fixture2rZunit2Z STRING_DTYPESrOZNArXrYr[r]r^r_Zstring_storage2Zstring_dtype_arguments2Z BYTES_DTYPESr`Z OBJECT_DTYPESrbrrcZDATETIME64_DTYPESrdZTIMEDELTA64_DTYPESrerhZFLOAT_NUMPY_DTYPESriZFLOAT_EA_DTYPESrjZALL_FLOAT_DTYPESrkZCOMPLEX_DTYPESrlZCOMPLEX_FLOAT_DTYPESrmZSIGNED_INT_NUMPY_DTYPESrnZUNSIGNED_INT_NUMPY_DTYPESroZALL_INT_NUMPY_DTYPESrpZALL_INT_EA_DTYPESrqZALL_INT_DTYPESrrrsZALL_INT_PYARROW_DTYPES_STR_REPRZFLOAT_PYARROW_DTYPES_STR_REPRrtZSIGNED_INT_EA_DTYPESruZALL_REAL_NUMPY_DTYPESrvZALL_REAL_DTYPESrwZALL_NUMPY_DTYPESrxZALL_REAL_NULLABLE_DTYPESryZALL_NUMERIC_DTYPESrzrZ_any_skipna_inferred_dtyper^_rrr__all__rrrsetitemlocZilocrrrrZatrZiatrrrrrZwarsawsrrr.r.r.r/s     8        <                                           !"#$%)                        %                              $#$$2