a OfӼ$@s6dZddlmZddlmZddlmZmZmZmZm 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@z ddlAZBWneCydZDYn0[BdZDddlEZEzeEFdWneEjGydZEYn0ddddZHdddddddZIddddZJejKjLgZMe=ejNe=d kreMOejKjPejQjRd!deSeMd"ejQTd!d#UD]8ZVeWe'jXjYeVZZe[eZej\eZe]d$d%e^d&qDd'UD]DZVeWe'jXjYeVZZe[eZej\eZe]d(d)e^ej]d*d+d,d-qd.UD]DZVeWe'jXjYeVZZe[eZej\eZe]d/d0e^ej]d*d+d,d1qej_dd2ddd3d4Z`ej_dd2ddd5d6Zaej_gd7d8d9d:d;d<ZbebZcej_d*d=gd>d9d:d?d@Zdej_gdAdBdCdDZeej_gdAdBdEdFZfej_ddgdBdGdHZgej_gdIdBdJdKZhej_gdLdBdMdNZiej_gdOdBdPdQZjej_gdOdBdRdSZkej_ddTdUdVdWdXejldYe"mdZd[gdBd\d]Znej_dTdUdVdWdXejldYe"mdZd[gdBd^d_Zoej_ddgdBd`daZpej_gdbdBdcddZqej_dedfgdBdgdhZrej_e6jsdid9d:djdkZtetZuej_dejve'jwgdBdldmZxexZyej_e6jzdnd9d:dodpZ{e{Z|ej_e)e.gdBdqdrZ}ej_e:e.gdsdtgd:dudvZ~e~Zej_e:e.e'jggdwd:dxdyZej_e:e.e)e'jgdzd9d:d{d|ZeZej_d}dd~dZej_ddddZej_ddZej_ddddZej_ddddZddZddZe:ddedDe1ddde1dddde2dddde3dddde-de:eddde:eddde:eddde:eddde:eddde:eddde:eddde:eddde:eddde:eddde:ddgd)ede:ddgd)ede:ejddddejddde:ejddddejddde(edde+ejdddde:ge;egdgdeee:gde:eddde:eddde:eddde:ededde:e'jddedDddd ZeDrle:e'jddedDddZeed<ej_edBddsZeZej_ddeDdBddZeZej_ddeDdBddZej_dddd„Zej_ddddĄZej_ddddƄZddȄZddʄeDZej_dddd̄Zddʄe6jDZieeeZej_edBddτZddʄe6jDZieeeeZej_edBdd҄Zej_ddddԄZej_ddddքZej_dddd؄Zej_e*dd)dٍe%ddڃfe*dddٍe%ddڃfe,dddߍdfe,dddߍdfe0ddde$dddfe/dddfgdBddZej_e6jdBddZej_e je8je je8je je8je je8je je8je je8je je8je je je je je je je je8je je8je je8jgdBddZej_e je8je je8je je8je je8je je8je je8je je8jgdBddZgdZej_edBddZddgZej_edBddZeeZej_edBddZej_e je je je je je jgdBddZej_gddBddZej_gddBddZgdZej_edBddZej_ddZej_dd d d d Zdddd ddddddeeededede je ed*de eddddgZeEdu re۠eEFdeEFdgddeDZe"ee݃ej_eed:ddZe"eed*dej_ed*ded*dd:ddZeZdd eee jgZeEdu reOeEFdej_edBd!d"ZeZej_gd#dBd$d%ZeZej_e6jdBd&d'Zej_dejlde"md(d[gdBd)d*Zej_d+ejld(e"md(d[ejld,e"md(d[gdBd-d.Zej_d/ejld(e"md(d[gdBd0d1ZeZej_e6jdBd2d3Zej_e6jdBd4d5Zej_d6dejlde"md(d[ejld7e"md(d[gdBd8d9Zej_e6jdBd:d;Zej_e6jdBd<d=Zej_d>dd?d@Zej_e6jdBdAdBZej_e6jdBdCdDZej_e6jdBdEdFZej_e6jdBdGdHZej_e6jdBdIdJZej_e6jdBdKdLZej_e6jdBdMdNZej_e6jdBdOdPZej_e6jdBdQdRZ ej_e6je6jdBdSdTZ ej_e6je6je6j e6j dBdUdVZ ej_e6jdBdWdXZej_e6jdBdYdZZej_e6jdBd[d\Zej_e6jdBd]d^Zej_e6jdBd_d`Zej_e6jdBdadbZdcddejvdegfdcdde'jdegfdfdde'jwdegfdgdhejvdigfdjejvejvejvgfdjgfdkddejvdlgfdfddejvdmgfdndoejvdmgfdpd*ejvdlgfdqd*ejvdmgfdre d*ejve dlgfddejvdgfdde'jdgfdsedtejvedugfdve0dwejve0dxgfdyedzd*d*ejved{d*d*gfd|d}ejvd~gfded*ejvedlgfded*ejvedlgfde,dze'jwe,d{gfde*dd*ejve*ddlgfgZee\ZZej_eed:ddZej_ddZ ej_gddBddZ!ej_dde'jYj"DdBddZ#ej_ddd9gdBddZ$ej_dddejvejvejvfejve'jwdfejve'jdfe'je'je'jfgdBddddZ%ej_e6j&e6j'e6j(gdBddZ)ej_e6j'e6j(gdBddZ*ej_e6j&e6j(gdBddZ+ej_e6j&e6j'gdBddZ,ej_e6j-e6j'gdBddZ.ej_e6j/e6j(gdBddZ0ej_ddddZ1ej_ddddZ2ej_ddddZ3ej_ddddZ4ddgZ5eEdure5OeEFdej_e5dBddddZ6e_ddZ7dS(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)HashableIteratorFTUTCNonereturncCs|jdddddS)N--no-strict-data-files store_falsez6Don't fail if a test is skipped for missing data file.)actionhelp)Z addoption)parserr0H/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/conftest.pypytest_addoptionos r2z pytest.Itemstr)itempathmessager*cCs(|j|r$|tjd|dS)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_markerpytestmarkfilterwarnings)r4r5r6r0r0r1ignore_doctest_warningws r<cCsL|dp|jddd}gd}|rH|D]}|D]\}}t|||q0q(dS)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.idxmaxrA)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_warningsr4r5r6r0r0r1pytest_collection_modifyitemss/ rEz6.83.2ci)deadlineZsuppress_health_checkz)MonthBegin MonthEnd BMonthBegin BMonthEndic)n normalizez%YearBegin YearEnd BYearBegin BYearEnd )Z min_valueZ max_value)rIrJmonthz1QuarterBegin QuarterEnd BQuarterBegin BQuarterEndi)rIrJZ startingMonth)ZautousecCst|d<t|d<dS)z: Make `np` and `pd` names available for doctests. nppdN)rQrR)Zdoctest_namespacer0r0r1add_doctest_importssrScCstdddS)z< Configure settings for all tests and test modules. Zchained_assignmentraiseN)rRZ set_optionr0r0r0r1configure_tests srU)rrMindexcolumnscCsdt|SNzaxis=reprxr0r0r1r])paramsidscCs|jS)z@ Fixture for returning the axis numbers of a DataFrame. paramrequestr0r0r1axissrerWcCsdt|SrXrYr[r0r0r1r]r^cCs|jS)zA Fixture for returning aliases of axis 1 of a DataFrame. rarcr0r0r1axis_1srf)TFN)r_cCs|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). rarcr0r0r1observed's rgcCs|jS)z6 Boolean 'ordered' parameter for Categorical. rarcr0r0r1ordered4srhcCs|jS)z% Boolean 'skipna' parameter. rarcr0r0r1skipna<sri)firstlastFcCs|jS)z[ Valid values for the 'keep' parameter used in .duplicated or .drop_duplicates rarcr0r0r1keepDsrl)bothneitherleftrightcCs|jS)zA Fixture for trying all interval 'inclusive' parameters. rarcr0r0r1inclusive_endpoints_fixtureMsrq)rorprmrncCs|jS)z< Fixture for trying all interval closed parameters. rarcr0r0r1closedUsrrcCs|jS)zS Secondary closed fixture to allow parametrizing over all pairs of closed. rarcr0r0r1 other_closed]srsgzipbz2zipxztarZzstdZ zstandard)ZmarkscCs|jS)zK Fixture for trying common compression types in compression tests. rarcr0r0r1 compressionesrycCs|jS)zk Fixture for trying common compression types in compression tests excluding uncompressed case. rarcr0r0r1compression_onlywsrzcCs|jS)z, Fixture that an array is writable. rarcr0r0r1writablesr{)innerouterrorpcCs|jS)z: Fixture for trying all types of join operations. rarcr0r0r1 join_typesr~nlargest nsmallestcCs|jS)z1 Fixture for trying all nselect methods. rarcr0r0r1nselect_methodsrcCs t|jSNtype__name__r[r0r0r1r]r^cCs|jS)z/ Fixture for each null type in pandas. rarcr0r0r1 nulls_fixturesrcCs|jS)zL Fixture for each null type in pandas, each null type exactly once. rarcr0r0r1unique_nulls_fixturesrcCs t|jSrrr[r0r0r1r]r^cCs|jS)z- Fixture for each NaT type in numpy. rarcr0r0r1np_nat_fixturesrcCs|jS)z; Fixture to parametrize over DataFrame and Series. rarcr0r0r1frame_or_seriessrrVseriescCs|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 rarcr0r0r1index_or_seriess r)rVrarraycCs|jS)zG Fixture to parametrize over Index, Series, and ExtensionArray rarcr0r0r1index_or_series_or_arraysrcCs|jSr)rr[r0r0r1r]r^cCs|jS)z] Fixture to test behavior for Index, Series, DataFrame, and pandas Array classes rarcr0r0r1box_with_arraysrz type[dict]cCsGdddt}|S)z, Fixture for a dictionary subclass. c@seZdZddddZdS)z"dict_subclass..TestSubDictr(r)c_stj|g|Ri|dSr)dict__init__)selfargskwargsr0r0r1rsz+dict_subclass..TestSubDict.__init__N)r __module__ __qualname__rr0r0r0r1 TestSubDictsr)r)rr0r0r1 dict_subclasssrztype[abc.Mapping]cCsGdddtj}|S)z8 Fixture for a non-mapping dictionary subclass. c@s>eZdZddddZddZdddd Zd dd d Zd S)z5non_dict_mapping_subclass..TestNonDictMappingr(r)cSs ||_dSr)_data)rZunderlying_dictr0r0r1r sz>non_dict_mapping_subclass..TestNonDictMapping.__init__cSs |j|Sr)r __getitem__)rkeyr0r0r1rszAnon_dict_mapping_subclass..TestNonDictMapping.__getitem__r&cSs |jSr)r__iter__rr0r0r1rsz>non_dict_mapping_subclass..TestNonDictMapping.__iter__intcSs |jSr)r__len__rr0r0r1rsz=non_dict_mapping_subclass..TestNonDictMapping.__len__N)rrrrrrrr0r0r0r1TestNonDictMapping sr)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)rWrVcSs|jSr)yearr[r0r0r1r]*r^zAmultiindex_year_month_day_dataframe_random_data..cSs|jSr)rOr[r0r0r1r]*r^cSs|jSr)dayr[r0r0r1r]*r^cSsg|]}|dqS)i8)astype).0Zlevr0r0r1 ,r^zCmultiindex_year_month_day_dataframe_random_data..)rrOrT)Zinplace)rrQrandom default_rngstandard_normalr"listobjectrgroupbysumrVZ set_levelslevelsZ set_names)ZtdfZymdr0r0r1/multiindex_year_month_day_dataframe_random_datas  rr#cCs*tgdgdggdgdgddgdS)z; 2-level MultiIndex, lexsorted, with string names. foobarbazZquxonetwothree) rrrrMrMrrrr) rrMrrrMrMrrrMrrjsecond)rcodesnames)r#r0r0r0r1%lexsorted_two_level_string_multiindex1s rrcCs,|}ttjdd|tgddddS)z2DataFrame with 2 level MultiIndex with random datar) r)ArCexpr7rVrW)rrQrrrr")rrVr0r0r1 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)rrrMrrr)rrMrrMrrMrjrF)rrrZverify_integrity)r"rQrr#)Z major_axisZ minor_axisZ major_codesZ minor_codesZ index_namesr0r0r1_create_multiindexJs  rcCs*tjddgddgtddddggd d S) zB MultiIndex with a level that is a tzaware DatetimeIndex. rMrab20130101r US/Easternrtzr)r)r#Z from_productrr0r0r0r1_create_mi_with_dt64tz_level^srcCsg|]}d|qSZpandas_r0rir0r0r1rjr^rrz 2020-01-01)rz US/PacificrDrz1 day)startrrint8rint16int32int64uint8uint16uint32uint64float32float64 complex64y? complex128abcde)num)rrr)rMrr)rrrMrMrrZInt64ZUInt16ZFloat32booleancCsg|]}d|qSrr0rr0r0r1rr^zstring[python]) stringrz datetime-tzperiodrrangerrrrrrrrrrz bool-object bool-dtyperrZ categoricalintervalemptytuplesmi-with-dt64tz-levelmultirepeatsZ nullable_intZ nullable_uintZnullable_floatZ nullable_boolz string-pythoncCsg|]}d|qSrr0rr0r0r1rr^zstring[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_dictrbcopyrcr0r0r1rVs cCsg|]\}}t|ts|qSr0) isinstancer#rrvaluer0r0r1rscCs|j}t|S)z8 index fixture, but excluding MultiIndex cases. )rbrr)rdrr0r0r1 index_flats rcCs0g|](\}}|ds|dvst|ts|qS))rZuintfloat)rrrr) startswithrr#rr0r0r1rs   cCst|jjdd}|j}|jdvrh|}d|ddd|d<d|ddd|d<t|Sd|d<d|d<t||SdS) 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)rrrrrrMN)rrbrvaluestolistr# from_tuplesr)rdindvalsr0r0r1index_with_missings   rrcCs4ttjdtjddtddtdDtdddS)zC Fixture for Series of floats with Index of unique strings rg?cSsg|]}d|qS)Zi_r0rr0r0r1rr^z!string_series..rrVr7)rrQarangerr"rrr0r0r0r1 string_seriess r cCs<ddtdD}tddtdDtd}t||dtdS)zI Fixture for Series of dtype object with Index of unique strings cSsg|]}d|qSZfoo_r0rr0r0r1rr^z!object_series..r cSsg|]}d|qS)Zbar_r0rr0r0r1rr^robjectsrVr7r)rr"rr)datarVr0r0r1 object_seriessrcCs&ttjddtddddddS)z9 Fixture for Series of floats with DatetimeIndex rr rrrtsr )rrQrrrrr0r0r0r1datetime_seriess  rcCs*t|}tjd|}t||dddS)zHelper for the _series dictrrF)rVr7r)lenrQrrrr)rVsizerr0r0r1_create_seriessrcCs"i|]\}}d|dt|qS)z series-with-z-indexr)rZindex_idrVr0r0r1 srcCst|S)zE Fixture for tests on series with changing types of indices. rrVr0r0r1series_with_simple_indexsrc Cs8i|]0}|jdttdddtdDd|dqS)-seriesr cSsg|]}d|qS)zi-r0rr0r0r1rr^z.rr)rrrrrr0r0r1rs cCst|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_objsrbrrcr0r0r1index_or_series_obj$srcCsi|]}|jdt|qS)r)rrrr0r0r1r-scCst|jjddS)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_objsrbrrcr0r0r1index_or_series_memory_obj:sr cCs<ttjdtjdtddtdDtdttdtddS)zj Fixture for DataFrame of ints with index of unique strings Columns are ['A', 'B', 'C', 'D'] r rrcSsg|]}d|qSr r0rr0r0r1rPr^zint_frame..r rr)rrQZonesrr"rrrr0r0r0r1 int_frameGs r"cCs6ttjddtddtdDttddS)zm Fixture for DataFrame of floats with index of unique strings Columns are ['A', 'B', 'C', 'D']. rr!cSsg|]}d|qSr r0rr0r0r1r^r^zfloat_frame..r rr)rrQrrrr"rrr0r0r0r1 float_frameUs  r#c Cstdddtdddtdddtdddtdddtdddtdddtdddtdddtdddg }ttjdt||dS)zF Fixture for Series with a DatetimeIndex that has duplicates. irMrrrrLr)rrrQrrrr)datesr0r0r1(rand_series_with_duplicate_datetimeindexcs           r%)rorprpg?g?z2012-01M)rz period[M]z 2012-02-01z period[D]z 2011-01-01r)rs)unitri)secondsztimedelta64[ns]cCs|jSrrarcr0r0r1ea_scalar_and_dtype{sr*cCs|jS)zD Fixture for dunder names for common arithmetic operations. rarcr0r0r1all_arithmetic_operatorssr+cCs|jS)zU Fixture for operator and roperator arithmetic, comparison, and logical ops. rarcr0r0r1all_binary_operatorss"r,cCs|jS)z Fixture for operator and roperator arithmetic functions. Notes ----- This includes divmod and rdivmod, whereas all_arithmetic_operators does not. rarcr0r0r1all_arithmetic_functionssr-) countrmaxminmeanprodZstdvarZmedianZkurtZskewZsemcCs|jS)z. Fixture for numeric reduction names. rarcr0r0r1all_numeric_reductionssr4allanycCs|jS)z. Fixture for boolean reduction names. rarcr0r0r1all_boolean_reductionssr7cCs|jS)z> Fixture for all (boolean + numeric) reduction names. rarcr0r0r1all_reductionssr8cCs|jS)z; Fixture for operator module comparison functions. rarcr0r0r1 comparison_op sr9)__le____lt____ge____gt__cCs|jS)zm Fixture for dunder names for compare operations except == and != * >= * > * < * <= rarcr0r0r1compare_operators_no_eq_nes r>)__and____rand____or____ror____xor____rxor__cCs|jS)zY Fixture for dunder names for common logical operations * | * & * ^ rarcr0r0r1all_logical_operators(s rE)ZcumsumZcumprodZcumminZcummaxcCs|jS)z0 Fixture for numeric accumulation names rarcr0r0r1all_numeric_accumulations9srFcCs |dS)zR Returns the configuration for the test setting `--no-strict-data-files`. r+)rB)Z pytestconfigr0r0r1strict_data_filesDsrGzCallable[..., str])rGr*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|sFr4td|dtd|d|S)NzCould not find file z' and --no-strict-data-files is not set.zCould not find .)osr5joinexists ValueErrorr9skip)rr5Z BASE_PATHrGr0r1decoas  zdatapath..deco)rIr5rJdirname__file__)rGrOr0rNr1datapathLs rRz Asia/Tokyozdateutil/US/Pacificzdateutil/Asia/Singaporez+01:15z-02:15z UTC+01:15z UTC-02:15i,i)hoursrrrcCsg|] }t|qSr0rYrr0r0r1rr^cCs|jS)zD Fixture for trying timezones including default (None): {0} rarcr0r0r1tz_naive_fixturesrTcCs|jS)z4 Fixture for trying explicit timezones: {0} rarcr0r0r1tz_aware_fixturesrUrz dateutil/UTCcCs|jS)zQ Fixture to provide variants of UTC timezone strings and tzinfo objects. rarcr0r0r1 utc_fixturesrV)r'msusnscCs|jS)z& datetime64 units we support. rarcr0r0r1r(sr(cCs|jS)zR Parametrized fixture for string dtypes. * str * 'str' * 'U' rarcr0r0r1 string_dtypes rZpyarrowcCs|jS)za Parametrized fixture for string dtypes. * 'string[python]' * 'string[pyarrow]' rarcr0r0r1nullable_string_dtypes r\python pyarrow_numpycCs|jS)zx Parametrized fixture for pd.options.mode.string_storage. * 'python' * 'pyarrow' * 'pyarrow_numpy' rarcr0r0r1string_storagesr_Znumpy_nullablecCs|jS)zb Parametrized fixture for pd.options.mode.string_storage. * 'python' * 'pyarrow' rarcr0r0r1 dtype_backends r`cCs|jS)zK Parametrized fixture for bytes dtypes. * bytes * 'bytes' rarcr0r0r1 bytes_dtypesracCs|jS)zN Parametrized fixture for object dtypes. * object * 'object' rarcr0r0r1 object_dtypesrbrzstring[pyarrow_numpy]cCs|jS)zo Parametrized fixture for string dtypes. * 'object' * 'string[python]' * 'string[pyarrow]' rarcr0r0r1any_string_dtypesrccCs|jS)z\ Parametrized fixture for datetime64 dtypes. * 'datetime64[ns]' * 'M8[ns]' rarcr0r0r1datetime64_dtype-srdcCs|jS)z^ Parametrized fixture for timedelta64 dtypes. * 'timedelta64[ns]' * 'm8[ns]' rarcr0r0r1timedelta64_dtype8srerc CstddddddddS)z- Fixture emits fixed Timestamp.now() irMrNr )rrOrhourminuter microsecond)rr0r0r0r1 fixed_now_tsCsrkcCs|jS)z^ Parameterized fixture for float dtypes. * float * 'float32' * 'float64' rarcr0r0r1float_numpy_dtypeMs rlcCs|jS)zR Parameterized fixture for float dtypes. * 'Float32' * 'Float64' rarcr0r0r1float_ea_dtypeYsrmcCs|jS)z~ Parameterized fixture for float dtypes. * float * 'float32' * 'float64' * 'Float32' * 'Float64' rarcr0r0r1any_float_dtypeds rncCs|jS)zg Parameterized fixture for complex dtypes. * complex * 'complex64' * 'complex128' rarcr0r0r1 complex_dtypers rocCs|jS)z| Parameterized fixture for signed integer dtypes. * int * 'int8' * 'int16' * 'int32' * 'int64' rarcr0r0r1any_signed_int_numpy_dtype~s rpcCs|jS)zx Parameterized fixture for unsigned integer dtypes. * 'uint8' * 'uint16' * 'uint32' * 'uint64' rarcr0r0r1any_unsigned_int_numpy_dtypes rqcCs|jS)z Parameterized fixture for any integer dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' rarcr0r0r1any_int_numpy_dtypesrrcCs|jS)z Parameterized fixture for any nullable integer dtype. * 'UInt8' * 'Int8' * 'UInt16' * 'Int16' * 'UInt32' * 'Int32' * 'UInt64' * 'Int64' rarcr0r0r1any_int_ea_dtypesrscCs|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' rarcr0r0r1 any_int_dtypesrtcCs|jS)z Parameterized fixture for any nullable integer dtype and any float ea dtypes. * 'UInt8' * 'Int8' * 'UInt16' * 'Int16' * 'UInt32' * 'Int32' * 'UInt64' * 'Int64' * 'Float32' * 'Float64' rarcr0r0r1any_numeric_ea_dtypesrucCs|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]' rarcr0r0r1any_numeric_ea_and_arrow_dtypes rvcCs|jS)z~ Parameterized fixture for any signed nullable integer dtype. * 'Int8' * 'Int16' * 'Int32' * 'Int64' rarcr0r0r1any_signed_int_ea_dtypes rwcCs|jS)z Parameterized fixture for any (purely) real numeric dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' * float * 'float32' * 'float64' rarcr0r0r1any_real_numpy_dtypesrxcCs|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. rarcr0r0r1any_real_numeric_dtype1srycCs|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' rarcr0r0r1any_numpy_dtypeHs"rzcCs|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]' rarcr0r0r1any_real_nullable_dtypemsr{cCs|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' rarcr0r0r1any_numeric_dtypesr|rrcmixedbytesacrz mixed-integerrg@Zfloatingg?integerzmixed-integer-floatdecimal datetime64z 2013-01-01z 2018-01-01rrZ20180101riicomplexy??y@@rrrrcCs |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)rbrQrr)rdZinferred_dtyperr0r0r1any_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. ZIPythonz6.0.0)Z minversionr)InteractiveShell)Configz:memory:)rD)r9 importorskipZIPython.core.interactiveshellrZtraitlets.configrZHistoryManagerZ hist_file)rrr}r0r0r1ips   r)ZbsrZcooZcscZcsrZdiaZdokZlilcCstd}t||jdS)z- Yields scipy sparse matrix classes. z scipy.sparseZ_matrix)r9rgetattrrb)rdsparser0r0r1spmatrixs rcCs6g|].}tttj|tjjr|dkrttj|qS)Tick) issubclassrrRoffsetsr)ror0r0r1rscCs|jS)zN Fixture for Tick based datetime offsets available for a time series. rarcr0r0r1 tick_classess rcCs|Srr0r[r0r0r1r],r^cCs|jS)zk Simple fixture for testing keys in sorting methods. Tests None (no key) and the identity key. rarcr0r0r1 sort_by_key,sr)rNN)ZEgonZVenkmanN)NCC1701Drrz#tuple[Hashable, Hashable, Hashable]cCs|jS)zP A 3-tuple of names, the first two for operands, the last for a result. rarcr0r0r1r5srcCs|jS)zI Parametrize over __setitem__, loc.__setitem__, iloc.__setitem__ rarcr0r0r1 indexer_sliHsrcCs|jS)z< Parametrize over loc.__getitem__, iloc.__getitem__ rarcr0r0r1 indexer_liPsrcCs|jS)z8 Parametrize over __setitem__, iloc.__setitem__ rarcr0r0r1 indexer_siXsrcCs|jS)z7 Parametrize over __setitem__, loc.__setitem__ rarcr0r0r1 indexer_sl`srcCs|jS)z: Parametrize over at.__setitem__, loc.__setitem__ rarcr0r0r1 indexer_alhsrcCs|jS)z< Parametrize over iat.__setitem__, iloc.__setitem__ rarcr0r0r1 indexer_ialpsrboolcCstddddkS)z> Fixture to check if the array manager is being used. mode.data_managerTZsilentrrr0r0r0r1using_array_managerxsrcCstjjjduotddddkS)z7 Fixture to check if Copy-on-Write is enabled. TrrblockrRoptionsmodeZ copy_on_writerr0r0r0r1using_copy_on_writesrcCstjjjdkotddddkS)z? Fixture to check if Copy-on-Write is in warning mode. warnrTrrrr0r0r0r1warn_copy_on_writesrcCstjjjduS)z= Fixture to check if infer string option is enabled. T)rRrfutureZ infer_stringr0r0r0r1using_infer_stringsrz Europe/Warsawzdateutil/Europe/WarsawcCs|jS)zE tzinfo for Europe/Warsaw using pytz, dateutil, or zoneinfo. rarcr0r0r1warsawsrcCsdS)N)r[r^r0r0r0r0r1arrow_string_storagesr(8__doc__ __future__r collectionsrrrrrrrr operatorrItypingr r Z dateutil.tzr r Z hypothesisrstnumpyrQr9ZpytzrrZpandas._config.configrZpandas.util._test_decoratorsutilZ_test_decoratorstdZpandas.core.dtypes.dtypesrrZpandasrRrrrrrrrrrrrr Zpandas._testingZ_testingtmZ pandas.corer!Zpandas.core.indexes.apir"r#Zpandas.util.versionr$collections.abcr%r&r[pa ImportErrorZ has_pyarrowZzoneinfoZZoneInfoZZoneInfoNotFoundErrorr2r<rEZ HealthCheckZtoo_slowZhypothesis_health_checks __version__appendZdiffering_executorssettingsZregister_profiletupleZ load_profilesplitr7rZtseriesrclsZregister_type_strategyZbuildsZintegersZbooleansZfixturerSrUreZ axis_framerfrgrhrirlrqrrrsrbZ skip_if_noryrzr{r~rZ NULL_OBJECTSrZnulls_fixture2nanZNaTrZunique_nulls_fixture2ZNP_NAT_OBJECTSrZnp_nat_fixture2rrZindex_or_series2rrrZbox_with_array2rrrrrrrrr rrrZ from_breaksZlinspacerrvrridxkeysrVZindex_fixture2rCrZ index_flat2rr rrrZ_seriesrZNARROW_NP_DTYPESZ_narrow_seriesrrZPYTHON_DATA_TYPESZ_typ_objects_seriesrr r"r#r%r*Zarithmetic_dunder_methodsr+addZraddsubZrsubmulZrmultruedivZrtruedivfloordivZ rfloordivmodZrmodpowZrpoweqneltlegtgeand_Zrand_xorZrxoror_Zror_r,r-Z_all_numeric_reductionsr4Z_all_boolean_reductionsr7Z_all_reductionsr8r9r>rEZ_all_numeric_accumulationsrFrGrRZ TIMEZONESextendZ TIMEZONE_IDSZparametrize_fixture_docr3rTrUZtz_aware_fixture2Z_UTCSrVZ utc_fixture2r(Zunit2Z STRING_DTYPESrZr\r_r`Zstring_storage2Z BYTES_DTYPESraZ OBJECT_DTYPESrbrcZDATETIME64_DTYPESrdZTIMEDELTA64_DTYPESrerkZFLOAT_NUMPY_DTYPESrlZFLOAT_EA_DTYPESrmZALL_FLOAT_DTYPESrnZCOMPLEX_DTYPESroZSIGNED_INT_NUMPY_DTYPESrpZUNSIGNED_INT_NUMPY_DTYPESrqZALL_INT_NUMPY_DTYPESrrZALL_INT_EA_DTYPESrsZALL_INT_DTYPESrtruZALL_INT_PYARROW_DTYPES_STR_REPRZFLOAT_PYARROW_DTYPES_STR_REPRrvZSIGNED_INT_EA_DTYPESrwZALL_REAL_NUMPY_DTYPESrxZALL_REAL_DTYPESryZALL_NUMPY_DTYPESrzZALL_REAL_NULLABLE_DTYPESr{ZALL_NUMERIC_DTYPESr|ZNArZ_any_skipna_inferred_dtyper`_rrr__all__rrrsetitemlocZilocrrrratrZiatrrrrrZwarsawsrrr0r0r0r1sB     8      <                                           (                             %                        $#$$2