l2i+ 8dZddlmZddlmZddlmZmZmZmZm 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<[Zdejgd=5d?Zeejdd@dAdBdCdDejdEe"jdFGg5dHZhejd@dAdBdCdDejdEe"jdFGg5dIZiejddg5dJZjejgdK5dLZkejdMdNg5dOZleje0jdP/dQZnenZoejdeje'jg5dRZrerZseje0jdS/dTZueuZveje(e+g5dUZweje4e+gdVdWg/dXZxexZyeje4e+e'jggdY/dZZ{eje4e+e(e'jgd[/d\Z|e|Z}ejd]Z~ejd^Zejd_Zejd.d`Zej d/daZdbZdcZidde0j dedfe0j dedge0j dedhidje0jdedke0jdedle0jdedme0jdedmndoe0jdedondpe0jdedpndqe0jdedqndre0jdedrndse0jdedsndte0jdedtndue0jdedundve0jdedvndwe0jdedwndxe0jdyjee4ej j#dzj%dydke0j&dedwnjd{e0j&dedwnjd|e0j(dee0j*dee4ge5j,egd}gd~eee4gde4ej0dedne4ej0dedne4ej0dedne4ej0dejedne4e'je0j dedndZe=r0e4e'je0j dednZeed<ejej95dZeZejej?Dcgc]\}}e|e5r|c}}5dZeZejej?Dcgc]&\}}|jGds|dvs e|e5s|(c}}5dZejd0dZejd0dZejd0dZdZej?Dcic]\}}d|de|c}}Zejd0dZejd0dZe0jXDcic]&}|jZde0j\d|(c}ZieeeZejej95dZe0jdDcic]}|jZde+|c}ZieeeeZejej95dZejd/dZejd/dZejd/dZejd/dZejd0dZeje)dd!e%dqdfe)dde%dwdfe*dddfe*dddfe-ddie$ddfe,ddfg5dZeje0jx5dZeje j|e2j~e je2je je2je je2je je2je je2je je2je je je je je je je je2je je2je je2jg5dZeje j|e2j~e je2je je2je je2je je2je je2je je2jg5dZgdZejeڬ5dZddgZejeܬ5dZeezZejeެ5dZeje je je je je je jg5dZejgd5dZejgd5dZgdZeje5dZejd„Zejd1dÄZejd1dĄZejd2dńZejd/dƄZddddddddddeeedΫededϫe j<e ed"Ыe edѬЫdҬӫgZe?4eje?jdhe?jdgeDcgc] }e| c}Ze"jeeejee/dԄZe"jeed"dejed"ded"d/dՄZeZddeee j<gZe?!eje?jdeje5d؄ZeZeje0j5dلZejdejde"jdګGg5dۄZejdejde"jdګGejde"jdګGg5dބZejdejde"jdګGg5dZeZeje0j5dZeje0j5dZejddejde"jdګGejde"jdګGg5dZeje0j5dZeje0j5dZejd3dZeje0j 5dZeje0j5dZ eje0j5dZ eje0j5dZ eje0j5dZeje0j 5dZeje0j$5dZeje0j(5dZeje0j,5dZeje0j(e0jz5dZeje0j(e0jze0j2ze0j4z5dZeje0j85dZeje0j<5dZeje0j@5dZ!eje0jD5dZ#eje0jH5dZ%dddejdgfddde'jLdgfdde'jdgfddejdgfdejejejgfdgfddejdzgfddejdgfddejdgfdd"ejdzgfdd"ejdgfde d"eje dzgfddejdgfdde'jLdgfdejNdejejNdgfdfe-d eje-d gfd ed d"d"ejed d"d"gfddejdgfdked"ejedzgfded"ejedzgfdje*d e'je*d gfde)dd"eje)ddzgfgZ(ee(\Z)Z*eje(e)/dZ+ejdZ,ejgd5dZ-eje'jj\Dcgc]K}e/eQe'j|e'jj`r|dk7reQe'j|Mc}5dZ1ejddg5dZ2ejdZ3ejdddejejejfeje'jdfeje'jLdfe'jLe'jLe'jLfg5d4dZ4eje0jje0jle0jng5d Z8eje0jle0jng5d!Z9eje0jje0jng5d"Z:eje0jje0jlg5d#Z;eje0jxe0jlg5d$Z=eje0j|e0jng5d%Z?ejd5d&Z@ejd5d'ZAd(d)gZBe?#eBje?jd(ejeB5d6d*ZCejd+ZDy#e>$rdZ=YwxYw#e?j$rdZ?YwxYwcc}}wcc}}wcc}}wcc}wcc}wcc}wcc}w(7a 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)HashableIteratorTFUTCc,|jdddy)N--no-strict-data-files store_falsez6Don't fail if a test is skipped for missing data file.)actionhelp) addoption)parsers /var/www/tkim.planitai.co.jp/gemegg/20251205-worksheet-17AUAHQQB0nZVje9QXtdISEPtMoa8RXKkuCk4R5iHbd4/project/python/venv/lib/python3.12/site-packages/pandas/conftest.pypytest_addoptionr*hs   Ec|jj|r2|jtjj d|yy)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)itempathmessages r)ignore_doctest_warningr6ps= yy$  22WWI3FGH r+cd|jdxs|jdd}gd}|D]}|rE|jtjj d|D]\}}t |||d|j vsY|jtjjy)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)r?zSeries.replace without 'value')z Series.idxminThe behavior of Series.idxmin)z Series.idxmaxThe behavior of Series.idxmax)zSeriesGroupBy.idxminr@)zSeriesGroupBy.idxmaxrA)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)rCz,DataFrame.fillna with 'method' is deprecatedadd_doctest_importsz/frame/) getoptionr/r0r1 usefixturesr6nodeid arraymanager)itemsconfig is_doctestignored_doctest_warningsr3r4r5s r)pytest_collection_modifyitemsrMs!!"56&:J:JE;K;J ) V 6  OOFKK334IJ K!9 < g&tT7; <  # OOFKK44 5 6r+z6.83.2ci)deadlinesuppress_health_checkz)MonthBegin MonthEnd BMonthBegin BMonthEndic)n normalizez%YearBegin YearEnd BYearBegin BYearEnd ) min_value max_value)rRrSmonthz1QuarterBegin QuarterEnd BQuarterBegin BQuarterEndi)rRrS startingMonthc(t|d<t|d<y)z: Make `np` and `pd` names available for doctests. nppdN)r^r_)doctest_namespaces r)rDrDs !d dr+)autousec0tjddy)z< Configure settings for all tests and test modules. chained_assignmentraiseN)r_ set_optionr+r)configure_testsrgs MM&0r+)rrVindexcolumnscdt|SNzaxis=reprxs r)rp s5a ARr+)paramsidsc|jS)z@ Fixture for returning the axis numbers of a DataFrame. paramrequests r)axisrx  ==r+ricdt|Srkrlrns r)rprpsuT!WI5Fr+c|jS)zA Fixture for returning aliases of axis 1 of a DataFrame. rtrvs r)axis_1r|ryr+)TFN)rqc|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). rtrvs r)observedr~  ==r+c|jS)z6 Boolean 'ordered' parameter for Categorical. rtrvs r)orderedr-ryr+c|jS)z% Boolean 'skipna' parameter. rtrvs r)skipnar5ryr+)firstlastFc|jS)z[ Valid values for the 'keep' parameter used in .duplicated or .drop_duplicates rtrvs r)keepr= ==r+)bothneitherleftrightc|jS)zA Fixture for trying all interval 'inclusive' parameters. rtrvs r)inclusive_endpoints_fixturerFryr+)rrrrc|jS)z< Fixture for trying all interval closed parameters. rtrvs r)closedrNryr+c|jS)zS Secondary closed fixture to allow parametrizing over all pairs of closed. rtrvs r) other_closedrVryr+gzipbz2zipxztarzstd zstandard)marksc|jS)zK Fixture for trying common compression types in compression tests. rtrvs r) compressionr^ ==r+c|jS)zk Fixture for trying common compression types in compression tests excluding uncompressed case. rtrvs r)compression_onlyrprr+c|jS)z, Fixture that an array is writable. rtrvs r)writablerryr+)innerouterrrc|jS)z: Fixture for trying all types of join operations. rtrvs r) join_typerryr+nlargest nsmallestc|jS)z1 Fixture for trying all nselect methods. rtrvs r)nselect_methodrryr+c,t|jSNtype__name__rns r)rprpsd1g6F6Fr+c|jS)z/ Fixture for each null type in pandas. rtrvs r) nulls_fixturerryr+c|jS)zL Fixture for each null type in pandas, each null type exactly once. rtrvs r)unique_nulls_fixturerryr+c,t|jSrrrns r)rprpsQ8H8Hr+c|jS)z- Fixture for each NaT type in numpy. rtrvs r)np_nat_fixturerryr+c|jS)z; Fixture to parametrize over DataFrame and Series. rtrvs r)frame_or_seriesrryr+rhseriesc|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 rtrvs r)index_or_seriesrrr+)rhrarrayc|jS)zG Fixture to parametrize over Index, Series, and ExtensionArray rtrvs r)index_or_series_or_arrayrryr+c|jSr)rrns r)rprps 1::r+c|jS)z] Fixture to test behavior for Index, Series, DataFrame, and pandas Array classes rtrvs r)box_with_arrayrrr+c$Gddt}|S)z, Fixture for a dictionary subclass. ceZdZddZy)"dict_subclass..TestSubDictc6tj|g|i|yr)dict__init__)selfargskwargss r)rz+dict_subclass..TestSubDict.__init__s MM$ 0 0 0r+NreturnNone)r __module__ __qualname__rrfr+r) TestSubDictrs 1r+r)r)rs r) dict_subclassrs 1d1 r+c8Gddtj}|S)z8 Fixture for a non-mapping dictionary subclass. c*eZdZddZdZddZddZy) 5non_dict_mapping_subclass..TestNonDictMappingc||_yr)_data)runderlying_dicts r)rz>non_dict_mapping_subclass..TestNonDictMapping.__init__s (DJr+c8|jj|Sr)r __getitem__)rkeys r)rzAnon_dict_mapping_subclass..TestNonDictMapping.__getitem__ s::))#. .r+c6|jjSr)r__iter__rs r)rz>non_dict_mapping_subclass..TestNonDictMapping.__iter__ s::&&( (r+c6|jjSr)r__len__rs r)rz=non_dict_mapping_subclass..TestNonDictMapping.__len__s::%%' 'r+Nr)rr )rint)rrrrrrrrfr+r)TestNonDictMappingrs ) / ) (r+r)rMapping)rs r)non_dict_mapping_subclassrs (S[[ ( r+cdtjd}|jdddgj}|jj |jj Dcgc]}|jdc}|_|jjgdd|Scc}w) z DataFrame with 3 level MultiIndex (year, month, day) covering first 100 business days from 2000-01-01 with random data dc|jSr)yearrns r)rpzAmultiindex_year_month_day_dataframe_random_data..s r+c|jSr)rZrns r)rpzAmultiindex_year_month_day_dataframe_random_data..s 177r+c|jSr)dayrns r)rpzAmultiindex_year_month_day_dataframe_random_data..s aeer+i8)rrZrT)inplace) tmmakeTimeDataFramegroupbysumrh set_levelslevelsastype set_names)tdfymdlevs r)/multiindex_year_month_day_dataframe_random_datars  s #C ++'):OL M Q Q SC $$#))BRBR%S3cjj&6%STCIII0$? J&Ts(B-c:tgdgdggdgdgddgS)z; 2-level MultiIndex, lexsorted, with string names. foobarbazquxonetwothree) rrrrVrVrrr) rrVrrrVrVrrrVrrsecond)rcodesnames)rrfr+r)%lexsorted_two_level_string_multiindexr&s+ ,.EF-/MN! r+c|}ttjjdj d|t gddS)z2DataFrame with 2 level MultiIndex with random datar) r)ABCexpr-)rhri)rr^random default_rngstandard_normalr)rrhs r) multiindex_dataframe_random_datar2s@ 2E  a 009oE2 r+ctgd}tddg}tjgd}tjgd}ddg}t||g||g|d S) zJ MultiIndex used to test the general functionality of this object rrr)rrrVrrr)rrVrrVrrVrrF)rrrverify_integrity)rr^rr) major_axis minor_axis major_codes minor_codes index_namess r)_create_multiindexr?sh 34Ju~&J((-.K((-.KH%K J'K(  r+c ntjddgddgtjdddggd  S) zB MultiIndex with a level that is a tzaware DatetimeIndex. rVrab20130101r US/Eastern)periodstzr)r)r from_productr_ date_rangerfr+r)_create_mi_with_dt64tz_levelr$Ss9  " " Q#sR]]:q\RS% r+stringrrz datetime-tzz US/Pacific)r!periodrrangeint8dtypeint16int32int64uint8uint16uint32uint64float32float64z bool-objectr r complex64 complex128)rrr)rVrr)rrrVrVrrInt64UInt16Float32booleanzstring[python]) bool-dtyper4r5 categoricalintervalemptytuplesmi-with-dt64tz-levelmultirepeats nullable_int nullable_uintnullable_float nullable_boolz string-pythonzstring[pyarrow]zstring-pyarrowcDt|jjS)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_dictrucopyrvs r)rhrhs  & + + --r+cH|j}t|jS)z8 index fixture, but excluding MultiIndex cases. )rurGrH)rwrs r) index_flatrJs! --C   ! ! ##r+)ruintfloat)r'r=rAr:cPt|jjd}|jj}|jdvrA|j }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)r>r?r@rrrVN)rGrurHvaluestolistr from_tuplesr)rwindvalss r)index_with_missingrVs. w}} % * * * 5C ::?? D}}CCzz|DGABK'QT"Xab\)R%%d++QRtCyr+c<tj}d|_|S)zC Fixture for Series of floats with Index of unique strings r)rmakeStringSeriesr-ss r) string_seriesr[s A AF Hr+c<tj}d|_|S)zI Fixture for Series of dtype object with Index of unique strings objects)rmakeObjectSeriesr-rYs r) object_seriesr_s A AF Hr+c<tj}d|_|S)z9 Fixture for Series of floats with DatetimeIndex ts)rmakeTimeSeriesr-rYs r)datetime_seriesrcs A AF Hr+ct|}tjjdj |}t ||ddS)zHelper for the _series dictrrF)rhr-rH)lenr^rrrr)rhsizedatas r)_create_seriesrhs; u:D 99  # 3 3D 9D $e#E ::r+z series-with-z-indexct|S)zE Fixture for tests on series with changing types of indices. )rhrhs r)series_with_simple_indexrks %  r+cgdgdg}t|}tj|}tjj dj d}t||}tj|jd<|S)z: Fixture with a Series with a 2-level MultiIndex. )rrrrrrrr)rrrrrrrrrrjr) rrrSr^rrrrnaniloc)arraysr>rhrgsers r)series_with_multilevel_indexrrsn A@F&\F  " "6 *E 99  # 3 3A 6D U #C&&CHHQK Jr+z-seriesr)r-r*cHt|jjdS)z| Fixture for tests on indexes, series and series with a narrow dtype copy to avoid mutation, e.g. setting .name TrN)_index_or_series_objsrurHrvs r)index_or_series_objrus! ! / 4 4$ 4 ??r+cHt|jjdS)z Fixture for tests on indexes, series, series with a narrow dtype and series with empty objects type copy to avoid mutation, e.g. setting .name TrN)_index_or_series_memory_objsrurHrvs r)index_or_series_memory_objrx4s! ( 6 ; ; ; FFr+cZttjjdS)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] r-)rr getSeriesDatarrfr+r) int_framer{As#4 R%%' ( / / 88r+c<ttjS)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] )rrgetTimeSeriesDatarfr+r)datetime_framer~^s4 R))+ ,,r+c<ttjS)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] )rrrzrfr+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']. ?rrr r2r)rVr,)rrcr2r,rj)rr^rarangerfr+r)mixed_type_framersT xx )<XXqcBhg6  iim  r+cztdddtdddtdddtdddtdddtdddtdddtdddtdddtdddg }ttjj dj t ||S)zF Fixture for Series with a DatetimeIndex that has duplicates. irVrrrUrj)rrr^rrrre)datess r)(rand_series_with_duplicate_datetimeindexrs q!q!q!q!q!q!q!q!q!q! E "))''*::3u:Fe TTr+)rrrg?g?z2012-01M)freqz period[M]z 2012-02-01Dz period[D]z 2011-01-01rrZ)unitr!i)secondsztimedelta64[ns]c|jSrrtrvs r)ea_scalar_and_dtyper ==r+c|jS)zD Fixture for dunder names for common arithmetic operations. rtrvs r)all_arithmetic_operatorsrryr+c|jS)zU Fixture for operator and roperator arithmetic, comparison, and logical ops. rtrvs r)all_binary_operatorsrD ==r+c|jS)z Fixture for operator and roperator arithmetic functions. Notes ----- This includes divmod and rdivmod, whereas all_arithmetic_operators does not. rtrvs r)all_arithmetic_functionsrs6 ==r+) countrmaxminmeanprodstdvarmediankurtskewsemc|jS)z. Fixture for numeric reduction names. rtrvs r)all_numeric_reductionsr3ryr+allanyc|jS)z. Fixture for boolean reduction names. rtrvs r)all_boolean_reductionsr>ryr+c|jS)z> Fixture for all (boolean + numeric) reduction names. rtrvs r)all_reductionsrIryr+c|jS)z; Fixture for operator module comparison functions. rtrvs r) comparison_oprQrr+)__le____lt____ge____gt__c|jS)zm Fixture for dunder names for compare operations except == and != * >= * > * < * <= rtrvs r)compare_operators_no_eq_nerbrr+)__and____rand____or____ror____xor____rxor__c|jS)zY Fixture for dunder names for common logical operations * | * & * ^ rtrvs r)all_logical_operatorsro ==r+)cumsumcumprodcummincummaxc|jS)z0 Fixture for numeric accumulation names rtrvs r)all_numeric_accumulationsrryr+c$|jdS)zR Returns the configuration for the test setting `--no-strict-data-files`. r#)rE) pytestconfigs r)strict_data_filesrs  ! !": ;;r+c:ttjdz S)Ntests)r __file__parentrfr+r) tests_pathrs > 7 **r+c|dz dz S)Niorgrf)rs r)tests_io_data_pathrs  v %%r+ctjjtjjtdfd}|S)aD Get the path to a data file. Parameters ---------- path : str Path to the file, relative to ``pandas/tests/`` Returns ------- path including ``pandas/tests``. Raises ------ ValueError If the path doesn't exist and the --no-strict-data-files option is not set. rctjjg|}tjj|s*rt d|dt j d|d|S)NzCould not find file z' and --no-strict-data-files is not set.zCould not find .)osr4joinexists ValueErrorr0skip)rr4 BASE_PATHrs r)decozdatapath..decoscww||I--ww~~d#  *4&0WX KK/$q1 2 r+)rr4rdirnamer)rrrs` @r)datapathrs2& RWW__X6@I Kr+c>tj|ddddS)z* The iris dataset as a DataFrame. rrgcsvziris.csv)r_read_csv)rs r)irisrs ;;xfeZ@ AAr+z Asia/Tokyozdateutil/US/Pacificzdateutil/Asia/Singaporez+01:15z-02:15z UTC+01:15z UTC-02:15i,i)hoursrPrrc|jS)zD Fixture for trying timezones including default (None): {0} rtrvs r)tz_naive_fixturerrr+c|jS)z4 Fixture for trying explicit timezones: {0} rtrvs r)tz_aware_fixturerrr+rz dateutil/UTCc|jS)zQ Fixture to provide variants of UTC timezone strings and tzinfo objects. rtrvs r) utc_fixturerryr+c|jS)zR Parametrized fixture for string dtypes. * str * 'str' * 'U' rtrvs r) string_dtyper ==r+pyarrowc|jS)za Parametrized fixture for string dtypes. * 'string[python]' * 'string[pyarrow]' rtrvs r)nullable_string_dtyper ==r+python pyarrow_numpyc|jS)zx Parametrized fixture for pd.options.mode.string_storage. * 'python' * 'pyarrow' * 'pyarrow_numpy' rtrvs r)string_storager-rr+numpy_nullablec|jS)zb Parametrized fixture for pd.options.mode.string_storage. * 'python' * 'pyarrow' rtrvs r) dtype_backendr?rr+c|jS)zK Parametrized fixture for bytes dtypes. * bytes * 'bytes' rtrvs r) bytes_dtyperS ==r+c|jS)zN Parametrized fixture for object dtypes. * object * 'object' rtrvs r) object_dtyper^rr+objectzstring[pyarrow_numpy]c|jS)zo Parametrized fixture for string dtypes. * 'object' * 'string[python]' * 'string[pyarrow]' rtrvs r)any_string_dtyperirr+c|jS)z\ Parametrized fixture for datetime64 dtypes. * 'datetime64[ns]' * 'M8[ns]' rtrvs r)datetime64_dtyper{rr+c|jS)z^ Parametrized fixture for timedelta64 dtypes. * 'timedelta64[ns]' * 'm8[ns]' rtrvs r)timedelta64_dtyperrr+c &tdddddddS)z- Fixture emits fixed Timestamp.now() irVrWr )rrZrhourminuter microsecond)rrfr+r) fixed_now_tsrs  1RR r+c|jS)z^ Parameterized fixture for float dtypes. * float * 'float32' * 'float64' rtrvs r)float_numpy_dtyperrr+c|jS)zR Parameterized fixture for float dtypes. * 'Float32' * 'Float64' rtrvs r)float_ea_dtyper rr+c|jS)z~ Parameterized fixture for float dtypes. * float * 'float32' * 'float64' * 'Float32' * 'Float64' rtrvs r)any_float_dtyper rr+c|jS)zg Parameterized fixture for complex dtypes. * complex * 'complex64' * 'complex128' rtrvs r) complex_dtyper rr+c|jS)z| Parameterized fixture for signed integer dtypes. * int * 'int8' * 'int16' * 'int32' * 'int64' rtrvs r)any_signed_int_numpy_dtyperrr+c|jS)zx Parameterized fixture for unsigned integer dtypes. * 'uint8' * 'uint16' * 'uint32' * 'uint64' rtrvs r)any_unsigned_int_numpy_dtyperrr+c|jS)z Parameterized fixture for any integer dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' rtrvs r)any_int_numpy_dtyperrr+c|jS)z Parameterized fixture for any nullable integer dtype. * 'UInt8' * 'Int8' * 'UInt16' * 'Int16' * 'UInt32' * 'Int32' * 'UInt64' * 'Int64' rtrvs r)any_int_ea_dtyperrr+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' rtrvs 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' rtrvs r)any_numeric_ea_dtyper$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]' rtrvs r)any_numeric_ea_and_arrow_dtyper:s@ ==r+c|jS)z~ Parameterized fixture for any signed nullable integer dtype. * 'Int8' * 'Int16' * 'Int32' * 'Int64' rtrvs r)any_signed_int_ea_dtyper]rr+c|jS)z Parameterized fixture for any (purely) real numeric dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' * float * 'float32' * 'float64' rtrvs r)any_real_numpy_dtyperjs$ ==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. rtrvs r)any_real_numeric_dtyper!s( ==r+c|jS)a Parameterized fixture for all numpy dtypes. * bool * 'bool' * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' * float * 'float32' * 'float64' * complex * 'complex64' * 'complex128' * str * 'str' * 'U' * bytes * 'bytes' * 'datetime64[ns]' * 'M8[ns]' * 'timedelta64[ns]' * 'm8[ns]' * object * 'object' rtrvs 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' rtrvs r)any_numeric_dtyper%s> ==r+rmixedbytesacr=z mixed-integerg@floatingrintegerzmixed-integer-floatdecimal datetime64z 2013-01-01z 2018-01-01r20180101riicomplexy??y@@rr<c^|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))rur^rr)rwinferred_dtyperQs r)any_skipna_inferred_dtyper2s0R%]]NF XXfF +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:)rJ)r0 importorskipIPython.core.interactiveshellr6traitlets.configr7HistoryManager hist_file)r6r7rs r)ipr=/s<  g6>(A!+A 1 %%r+)bsrcoocsccsrdiadoklilc^tjd}t||jdzS)z- Yields scipy sparse matrix classes. z scipy.sparse_matrix)r0r8getattrru)rwsparses r)spmatrixrIBs*   0F 67==94 55r+Tickc|jS)zN Fixture for Tick based datetime offsets available for a time series. rtrvs r) tick_classesrLLrr+c|Srrfrns r)rprpZsr+c|jS)zk Simple fixture for testing keys in sorting methods. Tests None (no key) and the identity key. rtrvs r) sort_by_keyrOZrr+c#Ktjdddlm}ddlm}ddlm}Gdd|}|d|d ||jddd|jd<|jjyw) Nfsspecr)register_implementation)MemoryFileSystem) _registryc*eZdZdZdgZdfd ZxZS) fsspectest..TestMemoryFStestmemNcb|jdd|jd<t| di|y)Ntestrrf)poprYsuperr)rr __class__s r)rz)fsspectest..TestMemoryFS.__init__ns+!::fd3DIIaL G  &v &r+r)rrrprotocolrYr __classcell__)r\s@r) TestMemoryFSrVjsv ' 'r+r_rWT)clobber) r0r8rQrRfsspec.implementations.memoryrSfsspec.registryrTrZrYstoreclear)rRrSregistryr_s r) fsspectestrfcsp !.>5'''I|TB . LLD!LasBB)rNN)EgonVenkmanN)NCC1701Driric|jS)zP A 3-tuple of names, the first two for operands, the last for a result. rtrvs r)rrys ==r+c|jS)zI Parametrize over __setitem__, loc.__setitem__, iloc.__setitem__ rtrvs r) indexer_slirlryr+c|jS)z< Parametrize over loc.__getitem__, iloc.__getitem__ rtrvs r) indexer_lirnryr+c|jS)z8 Parametrize over __setitem__, iloc.__setitem__ rtrvs r) indexer_sirpryr+c|jS)z7 Parametrize over __setitem__, loc.__setitem__ rtrvs r) indexer_slrrryr+c|jS)z: Parametrize over at.__setitem__, loc.__setitem__ rtrvs r) indexer_alrtryr+c|jS)z< Parametrize over iat.__setitem__, iloc.__setitem__ rtrvs r) indexer_ialrvryr+cPtjjjdk(S)z> Fixture to check if the array manager is being used. r)r_optionsmode data_managerrfr+r)using_array_managerr{s ::?? ' '7 22r+ctjjjxr'tjjjdk(S)z7 Fixture to check if Copy-on-Write is enabled. block)r_rxry copy_on_writerzrfr+r)using_copy_on_writers0 ::?? ( ( TRZZ__-I-IW-TTr+z Europe/Warsawzdateutil/Europe/Warsawc|jS)zE tzinfo for Europe/Warsaw using pytz, dateutil, or zoneinfo. rtrvs r)warsawrryr+cy)N)rrrfrfr+r)arrow_string_storagers 'r+r)r3z pytest.Itemr4strr5rrr)rr)rr)rr)rr )rrrzCallable[..., str])rr)rz#tuple[Hashable, Hashable, Hashable])rbool)rr(E__doc__ __future__r collectionsrrrrrr r,r operatorrpathlibr typingr r dateutil.tzrr hypothesisrstnumpyr^r0pytzrrpandas.util._test_decoratorsutil_test_decoratorstdpandas.core.dtypes.dtypesrrpandasr_rrrrrrpandas._testing_testingr pandas.corerpandas.core.indexes.apirrpandas.util.versionrcollections.abcrr rpa has_pyarrow ImportErrorzoneinfoZoneInfoZoneInfoNotFoundErrorr*r6rM HealthChecktoo_slowhypothesis_health_checks __version__appenddiffering_executorssettingsregister_profiletuple load_profilesplitr-rGtseriesoffsetsclsregister_type_strategybuildsintegersbooleansfixturerDrgrx axis_framer|r~rrrrrrru skip_if_norrrrr NULL_OBJECTSrnulls_fixture2rnNaTrunique_nulls_fixture2NP_NAT_OBJECTSrnp_nat_fixture2rrindex_or_series2rrrbox_with_array2rrrrrrr$makeStringIndex makeDateIndexmakePeriodIndexmakeTimedeltaIndexmakeRangeIndex makeIntIndex makeUIntIndexmakeFloatIndex makeBoolIndexrrrrrmakeNumericIndexmakeCategoricalIndexmakeIntervalIndexrSrrrrGidxkeysrhindex_fixture2rI isinstancerJ index_flat2 startswithrVr[r_rcrh_seriesrkrrNARROW_NP_DTYPESrmake_rand_series_narrow_seriesrtruPYTHON_DATA_TYPES_typ_objects_seriesrwrxr{r~rrrrarithmetic_dunder_methodsraddraddsubrsubmulrmultruedivrtruedivfloordiv rfloordivmodrmodpowrpoweqneltlegtgeand_rand_xorrxoror_ror_rr_all_numeric_reductionsr_all_boolean_reductionsr_all_reductionsrrrr_all_numeric_accumulationsrrrrrr TIMEZONESextendrm TIMEZONE_IDSparametrize_fixture_docrrrtz_aware_fixture2_UTCSr utc_fixture2 STRING_DTYPESrrrrstring_storage2 BYTES_DTYPESr OBJECT_DTYPESrrDATETIME64_DTYPESrTIMEDELTA64_DTYPESrrFLOAT_NUMPY_DTYPESrFLOAT_EA_DTYPESr ALL_FLOAT_DTYPESr COMPLEX_DTYPESr SIGNED_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_DTYPESr!ALL_NUMPY_DTYPESr#ALL_NUMERIC_DTYPESr%NAr-_any_skipna_inferred_dtyperr_r2r=rI__all__ issubclassrJrLrOrfrsetitemlocrorlrnrprratrtiatrvr{rwarsawsrr)rvalueindex_idrhr*ios0000000r)r"s*&#  '  *) (  KHeI$=6@'22;;< : ! !"gh&77##J$:$:$N$NO $$ 89%   & 8 = = ?D "**$$d +CB YRYYskbkk#r2kbkkm L 4 9 9 ; D "**$$d +CB  bkk"a !bkkm"++R8    @ E E G D "**$$d +CB  bkk#r"!bkkm%"++R@   !!1117RST 9~+FGH*+ , *+,e}%&/01;<=;<=;<=      V=2==#=>        V=2==#=>    e}%&:;< K012r,FGHbffbff-./-r((.HIJ! 6*+,vWh,?@ A #vrxx06RSTvy"((;AUVW!  0     (!  b  %!   %!#2##CL9!  b  % ! &&&s+ !  R  s # ! OBOOCv .! _R__S 0! _R__S 0! _R__S 0! R  c 1! bs(3! bs(3! bs(3! r  I6!  r  I6!!"#2##B'..v6#!$ --a0@@DqHI$$$S :AA+N%"%%c;BB<P*2**3/$$$S) 2Y$j$$S)> %JK8:  !'()"))C.89299S>:IBIIcN)<9299S>006iH8288$6B$6$6s$;CSTUA! D +++C08IJ KC%(L!"|((*+ ., .*002 U*UJ:W  $  $ ',,.  C NN3 4AA5*-    :      ;(--/ %8*F#^E%::  !!  $$$  ~~g 3 3 3E JJ F<E7EnE,1134@5@<>;O;O27u~~gu -        388:;G<G998--8))8"UU. q "M'7$CD s# & i(IJ   $k2  3 '5 l| 4  6  3 !23   r3345                        5 <=<             $ %$   ./0!%../0*,CC'(         ?@ A  NG123<<++&&BBB     G IN LL YQ   YR u-%  (   H  l + H  e $  "++AQ+ C -.yl34/C QR 012y},qr*:;<3% UWhll;  LL"""5)*u r''() &mbmmI.FG     YmbmmI&>? _MBMM),DE  YmbmmI&>?    !r'(r''() &mbmmI.FG ,MBMM)4LM  r++,-r,,-.r,,-.r))*+r**+ , r(()*r112 3 r334 5 r../0"r++, -  r(()*2r++b.@.@@AB*   (( ) && '   :r../ 0 r//01(r))*+,r**+!,!Hr,,-.FRVVS!" RUUC ! sBFFC ! tRVVT"# rvvrvvrvv&' bMsBFFA&' sBFFC !#rvvs#$BFFAQ,-RVVWQZ01rvvu%&ruue$%MBMM,/|9TUV)J':1FGH d4AT1a(8 9:()9Q<167 d1grvvtAw '( t bfffTl34(1a."&&(1a.9:36 ( )Q1s;,"<,"d&&$HI6J6##  gbjj!,bjjoo >1;  A k*+,*!,    ruu    BFFBGG456()* BGG,-. BFF+,-rvv'(()*33UU 4 5  NN$8$$_56w ((M<K%%Hv " X >$| ,P si%An:1AoAo 0Ao %+Ao& 2Ao,+Ao21Ao7C Ao<b9AApn:AooAooAooAo