i ?UdZddlmZddlmZddlmZmZmZmZm Z ddl m Z ddl Z ddl Z ddlZddlmZmZddlZddlmZmZddlZdd lmZddlZddlZdd lmZddlmcm 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.m/Z/m0Z0m1Z1m2Z2ddl3m4Z5dd l6m7Z7ddl8m9Z9m:Z:er ddl;mZ> ddl?Z@[@dZAeddZCdUdZDdZEdVdZFdUdZGejjddddejjejjfejjddjD]eZOePe&jjeOZSejeSejeSejddej gd!jD]xZOePe&jjeOZSejeSejeSejd"d#ejejd$d%&'zd(jD]xZOePe&jjeOZSejeSejeSejd)d*ejejd$d%&+zejd,dUd-ZYejd,dUd.ZZejgd/d01d2Z[ejddg3d4Z\ejgd53d6Z]ejddg3d7Z^ejddg3d8Z_ejddg3d9Z`ejgd:3d;Zaejgd<3d=Zbejgd>3d?Zcejgd>3d@ZdejddAdBdCdDdEejdFe!jdGHg3dIZgejdAdBdCdDdEejdFe!jdGHg3dJZhejddg3dKZiejgdL3dMZjejdNdOg3dPZkejddg3dQZlejddg3dRZmejgdS3dTZnejddg3dUZoejddg3dVZpejddg3dWZqejdg3dXZrejdg3dYZseje5jdZ1d[ZueuZvejdeje&jg3d\ZyeyZzeje5jd]1d^Z|e|Z}eje(e-g3d_Z~eje9e-gd`dag1dbZeje9e-e&jggdc1ddZeje9e-e(e&jgde1dfZeZejdWdgZejdXdhZejdiZejdYdjZej dZdkZdlZdmZidne9edoDcgc]}dp| c}eqdre9edoDcgc]}dp| c}dqdse0dtdoudve0dtdodwxdye1dtdodz{d|e2d}dodz~de,dode9ejdodqde9ejdodqde9ejdodqde9ejdodqde9ejdodqde9ejdodqde9ejdodqde9ejdodqde9ejdodqde9ejdodqide9ddgd#zeqde9ddgd#zeqde9ejdodqdejdodqzzde9ejdodqdejdodqzzde'eddzde*j ej"dddde9gde:j$egdgdddedede9gdde9ejdodqde9ejdodqde9ejdodqde9ejdoj)edqde9e&jedoDcgc]}dp| c}dqZeAr7e9e&jedoDcgc]}dp| c}dqZeed<ejej/3dZejej3Dcgc]\}}e|e:r|c}}3dZejej3Dcgc]&\}}|j9ds|dvs e|e:s|(c}}3dZejd[dZejd[dZejd[dZdZej3Dcic]\}}d|de|c}}Zejd[dZe5jHDcic]:}|jJde-ededDcgc]}d| c}d|<c}}ZieeeZejej/3dZe5jRDcic]}|jJde-|c}ZieeeeZejej/3dZejdZdZejdZdZejd[dZeje)dd#e$dd«fe)ddĬe$dd«fe+ddƬǫdfe+ddzǫdfe/dd̬ͫjadΫe#dd̬ϫfe.dЬѫdfg3dӄZeje5jd3dԄZeje jhe7jje jle7jne jpe7jre jte7jve jxe7jze j|e7j~e je7je je je je je je je je7je je7je je7jg3dՄZeje jhe7jje jle7jne jpe7jre jte7jve jxe7jze j|e7j~e je7jg3dքZgdעZejeЬ3d؄ZddgZejeҬ3dۄZeezZejeԬ3d܄Zeje je je je je je jg3d݄Zejgdޢ3d߄Zejgd3dZgdZeje٬3dZejdZejd\dZddddddddddeee je ed$e eddgZeCeeސjeCjdeCjdeCjdeCjdweCjdfeDcgc] }e| c}Ze!jeeejee1dZe!jeed$dejed$ded$d1dZddee jgZeCejeCjeje3dZeZejgd3dZeZeje5j3dZejde&jfejde&jfe!jdHejdejfe!jdHdejfggd1dZejdejde!jdHg3dZejejdejfe!jdHejde&jfe!jdHg3dZejdejde!jdHg3dZejde&jfejde&jfe!jdHejdejfe!jdHdejfggd1dZejdejde!jdHg3dZeZeZeje5j3dZeje5j3dZejejdnde&jfejde&jfe!jdHejdejfe!jdHdejfggd1d ZeZeje5j3d Zeje5j3d Zejd]d Zeje5j3d Zeje5j 3dZeje5j3dZ eje5j3dZ eje5j3dZ eje5j3dZeje5j 3dZeje5j$3dZeje5j(3dZeje5j,3dZeje5j(e5j z3dZeje5j(e5j ze5j2ze5j4z3dZeje5j83dZeje5j<3dZeje5j@3dZ!eje5jD3dZ#eje5jH3dZ%eje5jL3dZ'drdejdgfdrde&jdgfd de&jdgfd!d"ejd#gfdejejejgfdgfd$dejdgfd dejd%gfd&d'ejd%gfd(d$ejdgfd)d$ejd%gfd*e d$eje dgfddejdgfdde&jdgfd+ejPd,ejejPd-gfdse/d.eje/d/gfd0ed1d$d$ejed2d$d$gfd3d4ejd5gfd|ed$ejedgfd6ed$ejedgfdye+d1e&je+d2gfde)dd$eje)ddgfgZ)e)Dcgc]}|d c}Z*eje)e*1d7Z+ejd8Z,ejd9Z-eje&jj\Dcgc]K}e/ePe&j|e&jj`r|d:k7rePe&j|Mc}3d;Z1ejdd<g3d=Z2ejd>d?d@ejejejfeje&jdfeje&jdfe&je&je&jfg3d^dAZ3eje5jhe5jje5jlg3dBZ7eje5jje5jlg3dCZ8eje5jhe5jlg3dDZ9eje5jhe5jjg3dEZ:eje5jve5jjg3dFZ<eje5jze5jlg3dGZ>ejddg3d_dHZ?ejd`dIZ@ejd`dJZAdKdLgZBdMeCdN<eC#eBjeCjdKejeB3dadOZDejdPZEejdQRdSZFejddg3dTZGy#eB$rdZAYwxYwcc}wcc}wcc}wcc}wcc}}wcc}}wcc}}wcc}wcc}}wcc}wcc}wcc}wcc}w(ba 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_CHECKINGAny)tzlocaltzutc) strategies)import_optional_dependency)DatetimeTZDtype IntervalDtype) CategoricalIndex DataFrameInterval IntervalIndexPeriod RangeIndexSeries Timedelta Timestamp date_range period_rangetimedelta_range)ops)Index MultiIndex)CallableHashableIteratorTFpytzignore)errorsc,|jdddy)N--no-strict-data-files store_falsez6Don't fail if a test is skipped for missing data file.)actionhelp) addoption)parsers d/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/pandas/conftest.pypytest_addoptionr0hs   Ecrddl}ddl|jjfd}||j_y)NrcXj|rd|jvry|||S)N.T) isfunction __qualname__)selfmoduleobjectinspectorigs r/ _from_modulez)pytest_sessionstart.._from_modulezs2   f %#1D1D*DD&&))r1)doctestr: DocTestFinderr<)sessionr=r<r:r;s @@r/pytest_sessionstartr@ps1  - -D**6G&r1strc|jj|r2|jtjj d|yy)a~Ignore doctest warning. Parameters ---------- item : pytest.Item pytest test item. path : str Module path to Python object, e.g. "pandas.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_warningrLs= yy$  22WWI3FGH r1c|jdxs|jdd}gd}|r|D]}|D]\}}t|||yy)Nz--doctest-modulesz--doctest-cythonF)default))zapi.interchange.from_dataframe"The DataFrame Interchange Protocol)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)z&CategoricalDtype._from_values_or_dtypezConstructing a Categorical)zDataFrame.__dataframe__rO)DataFrameGroupBy.fillnaz%DataFrameGroupBy.fillna is deprecated)zDataFrameGroupBy.corrwithz'DataFrameGroupBy.corrwith is deprecated)zNDFrame.replacezSeries.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.idxminrW)zSeriesGroupBy.idxmaxrX)to_pytimedeltaz2The behavior of TimedeltaProperties.to_pytimedelta)zNDFrame.reindex_likez'keyword argument 'method' is deprecated)zmissing.mask_zero_div_zerozdivide by zero encountered)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)rVz3DataFrameGroupBy.fillna with 'method' is deprecated) read_parquetz1Passing a BlockManager to DataFrame is deprecated) getoptionrL)itemsconfig is_doctestignored_doctest_warningsrIrJrKs r/pytest_collection_modifyitemsr`sg!!"56&:J:JE;K;J % ND!9 g&tT7;":r1 pandas_ci)databasedeadline max_examplessuppress_health_checkz)MonthBegin MonthEnd BMonthBegin BMonthEndic)n normalizez%YearBegin YearEnd BYearBegin BYearEnd ) min_value max_value)rhrimonthz1QuarterBegin QuarterEnd BQuarterBegin BQuarterEndi)rhri startingMonth)autousec(t|d<t|d<y)z: Make `np` and `pd` names available for doctests. nppdN)rurv)doctest_namespaces r/add_doctest_importsrxs !d dr1c0tjddy)z< Configure settings for all tests and test modules. chained_assignmentraiseN)rv set_optionr1r/configure_testsr~ s MM&0r1)rrlindexcolumnsc d|S)Nzaxis=r}xs r/rs 5r1)paramsidsc|jS)z@ Fixture for returning the axis numbers of a DataFrame. paramrequests r/axisr ==r1)rc|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). rrs r/observedr ==r1)TFNc|jS)z6 Boolean 'ordered' parameter for Categorical. rrs r/orderedr+rr1c|jS)z% Boolean 'dropna' parameter. rrs r/dropnar3rr1c|jS)z# Boolean 'sort' parameter. rrs r/sortr;rr1c|jS)z% Boolean 'skipna' parameter. rrs r/skipnarCrr1)firstlastFc|jS)z[ Valid values for the 'keep' parameter used in .duplicated or .drop_duplicates rrs r/keeprK ==r1)bothneitherleftrightc|jS)zA Fixture for trying all interval 'inclusive' parameters. rrs r/inclusive_endpoints_fixturerTrr1)rrrrc|jS)z< Fixture for trying all interval closed parameters. rrs r/closedr\rr1c|jS)zS Secondary closed fixture to allow parametrizing over all pairs of closed. rrs r/ other_closedrdrr1gzipbz2zipxztarzstd zstandard)marksc|jS)zK Fixture for trying common compression types in compression tests. rrs r/ compressionrl ==r1c|jS)zk Fixture for trying common compression types in compression tests excluding uncompressed case. rrs r/compression_onlyr~rr1c|jS)z, Fixture that an array is writable. rrs r/writablerrr1)innerouterrrc|jS)z: Fixture for trying all types of join operations. rrs r/ join_typerrr1nlargest nsmallestc|jS)z1 Fixture for trying all nselect methods. rrs r/nselect_methodrrr1c|jS)z2 Fixture for 'na_action' argument in map. rrs r/ na_actionrrr1c|jS)zJ Fixture for 'na_action' argument in sort_values/sort_index/rank. rrs r/ ascendingrrr1)averageminmaxrdensec|jS)z. Fixture for 'rank' argument in rank. rrs r/ rank_methodrrr1c|jS)z5 Fixture for 'as_index' argument in groupby. rrs r/as_indexrrr1c|jS)z6 Fixture for 'cache' argument in to_datetime. rrs r/cacherrr1c|jS)z> Fixture for parallel keyword argument for numba.jit. rrs r/parallelrrr1c|jS)z; Fixture for nogil keyword argument for numba.jit. rrs r/nogilrrr1c|jS)z> Fixture for nopython keyword argument for numba.jit. rrs r/nopythonrrr1c,t|jSNtype__name__rs r/rrsd1g6F6Fr1c|jS)z/ Fixture for each null type in pandas. rrs r/ nulls_fixturerrr1c|jS)zL Fixture for each null type in pandas, each null type exactly once. rrs r/unique_nulls_fixturerrr1c,t|jSrrrs r/rrsQ8H8Hr1c|jS)z- Fixture for each NaT type in numpy. rrs r/np_nat_fixturerrr1c|jS)z; Fixture to parametrize over DataFrame and Series. rrs r/frame_or_seriesrrr1rseriesc|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 rrs r/index_or_seriesrrr1)rrarrayc|jS)zG Fixture to parametrize over Index, Series, and ExtensionArray rrs r/index_or_series_or_arrayr,rr1c|jSr)rrs r/rr4s1::r1c|jS)z] Fixture to test behavior for Index, Series, DataFrame, and pandas Array classes rrs r/box_with_arrayr4rr1c$Gddt}|S)z, Fixture for a dictionary subclass. ceZdZddZy)"dict_subclass..TestSubDictc6tj|g|i|yr)dict__init__)r7argskwargss r/rz+dict_subclass..TestSubDict.__init__Gs MM$ 0 0 0r1NreturnNone)r __module__r6rr}r1r/ TestSubDictrFs 1r1r)r)rs r/ dict_subclassr@s 1d1 r1c8Gddtj}|S)z8 Fixture for a non-mapping dictionary subclass. c*eZdZddZdZddZddZy) 5non_dict_mapping_subclass..TestNonDictMappingc||_yr)_data)r7underlying_dicts r/rz>non_dict_mapping_subclass..TestNonDictMapping.__init__Ts (DJr1c8|jj|Sr)r __getitem__)r7keys r/rzAnon_dict_mapping_subclass..TestNonDictMapping.__getitem__Ws::))#. .r1c6|jjSr)r__iter__r7s r/rz>non_dict_mapping_subclass..TestNonDictMapping.__iter__Zs::&&( (r1c6|jjSr)r__len__rs r/rz=non_dict_mapping_subclass..TestNonDictMapping.__len__]s::%%' 'r1Nr)rr$)rint)rrr6rrrrr}r1r/TestNonDictMappingrSs ) / ) (r1r)rMapping)rs r/non_dict_mapping_subclassr Ms (S[[ ( r1c ttjjdj dt t dtddd}|jd d d gj}|jj|jjDcgc]}|jd c}|_ |jjgd d|Scc}w)z DataFrame with 3 level MultiIndex (year, month, day) covering first 100 business days from 2000-01-01 with random data )dABCD 2000-01-01r Bperiodsfreq)rrc|jSr)yearrs r/rzAmultiindex_year_month_day_dataframe_random_data..qsr1c|jSr)rprs r/rzAmultiindex_year_month_day_dataframe_random_data..qs177r1c|jSr)dayrs r/rzAmultiindex_year_month_day_dataframe_random_data..qsaeer1i8)rrprT)inplace)rrurandom default_rngstandard_normalr listrgroupbysumr set_levelslevelsastype set_names)tdfymdlevs r//multiindex_year_month_day_dataframe_random_datar(fs  a 00:d6l#s= C ++'):OL M Q Q SC $$#))BRBR%SBR3cjj&6BR%STCIII0$? J&Ts+C0c:tgdgdggdgdgddgS)z; 2-level MultiIndex, lexsorted, with string names. foobarbazquxonetwothree) rrrrlrlr r r3r3) rrlr rrlrlr rrlr rsecond)r"codesnames)r!r}r1r/%lexsorted_two_level_string_multiindexr7xs+ ,.EF-/MN! r1c|}ttjjdj d|t gddS)z2DataFrame with 2 level MultiIndex with random datar ) r3)ArCexprCrr)rrurrrr )r7rs r/ multiindex_dataframe_random_datar?s@ 2E  a 009oE2 r1ctgd}tddg}tjgd}tjgd}ddg}t||g||g|d S) zJ MultiIndex used to test the general functionality of this object r*r0r1)rrrlr r3r3)rrlrrlrrlrr4F)r"r5r6verify_integrity)r rurr!) major_axis minor_axis major_codes minor_codes index_namess r/_create_multiindexrGsh 34Ju~&J((-.K((-.KH%K J'K(  r1c Ztjddgddgtdddggd  S) zB MultiIndex with a level that is a tzaware DatetimeIndex. rlr ab20130101r3 US/Easternrtzr/)r6)r! from_productrr}r1r/_create_mi_with_dt64tz_levelrPs5  " " Q#sZ A,OP% r1r9r9pandas_dtypestringrz 2020-01-01)rz datetime-tzz US/PacificrMperiodDrrz1 day)startrrrangeint8int16int32int64uint8uint16uint32uint64float32float64z bool-object bool-dtype complex64y? complex128 categoricalabcdr intervalr  )numemptytuples)r+r,r-)rlr r3)strictmi-with-dt64tz-levelmultirepeats)rrrlrlr r  nullable_intInt64 nullable_uintUInt16nullable_floatFloat32 nullable_boolbooleanz string-pythonzstring[python]zstring[pyarrow]zstring-pyarrowcHt|jjdS)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 - ... Fdeep) indices_dictrcopyrs r/rrs!  & + + + 77r1cL|j}t|jdS)z8 index fixture, but excluding MultiIndex cases. Frz)rr|r})rrs r/ index_flatrs& --C   ! !u ! --r1)ruintfloat)rXrkrprcc8t|j}|jdvrC|j}dg|ddd|d<dg|ddd|d<tj|S|j j }d|d<d|d<t||dS)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. )rlrnroNrrlF)r})r|rtolistr! from_tuplesvaluesr}r)rindvalss r/index_with_missingrs( w}} %C}}CCzz|&$q'!"+&Q(48AB<(R%%d++zz QRtCyE**r1c ttjdtjdzt t dDcgc]}d| c}dScc}w)zC Fixture for Series of floats with Index of unique strings rRg?i_rrrC)rruarangerbr rXis r/ string_seriesr!sO  "BJJ'#-uRy1y!r!Xy12  1s ActdDcgc]}d| }}ttdDcgc]}d| c}}t||dtScc}wcc}w)zI Fixture for Series of dtype object with Index of unique strings rfoo_bar_objectsrrCrS)rXr rr9)rdatars r/ object_seriesr-s^ !&b * 1d1#J D * uRy1y!T!:y1 2E $e)6 BB +1s A Acttjjdj dt ddddS)z9 Fixture for Series of floats with DatetimeIndex r rrrrtsr)rrurrrrr}r1r/datetime_seriesr7s=  a 004r<  r1ct|}tjjdj |}t ||ddS)zHelper for the _series dictr rIF)rrCr})lenrurrrr)rsizers r/_create_seriesrCs; u:D 99  # 3 3D 9D $e#E ::r1z series-with-z-indexct|S)zE Fixture for tests on series with changing types of indices. )rrs r/series_with_simple_indexrPs %  r1z-seriesrzi-rIrcHt|jjdS)z| Fixture for tests on indexes, series and series with a narrow dtype copy to avoid mutation, e.g. setting .name Frz)_index_or_series_objsrr}rs r/index_or_series_objrcs! ! / 4 4% 4 @@r1cHt|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 Frz)_index_or_series_memory_objsrr}rs r/index_or_series_memory_objrys! ( 6 ; ; ; GGr1c ttjdtjt t dDcgc]}d| c}t t dScc}w)zj Fixture for DataFrame of ints with index of unique strings Columns are ['A', 'B', 'C', 'D'] rr rRrrrr>)rruonesr\r rXrrs r/ int_framersS  rxx(r3AtA3Z34d6l# 3s A'c ttjjdj dt t dDcgc]}d| c}t tdScc}w)zm Fixture for DataFrame of floats with index of unique strings Columns are ['A', 'B', 'C', 'D']. r rrrrr>)rrurrrr rXrrs r/ float_framers^  a 009r3AtA3Z34d6l# 3s A0cztdddtdddtdddtdddtdddtdddtdddtdddtdddtdddg }ttjj dj t ||S)zF Fixture for Series with a DatetimeIndex that has duplicates. irlr r3r rkr)rrrurrrr)datess r/(rand_series_with_duplicate_datetimeindexrs q!q!q!q!q!q!q!q!q!q! E "))''*::3u:Fe TTr1)rrrg?g?z2012-01M)rz period[M]z 2012-02-01z period[D]z 2011-01-01rL)rNs)unitrNi)secondsztimedelta64[us]c|jS)z; Fixture that tests each scalar and datetime type. rrs r/ea_scalar_and_dtyper" ==r1c|jS)zD Fixture for dunder names for common arithmetic operations. rrs r/all_arithmetic_operatorsrrr1c|jS)zU Fixture for operator and roperator arithmetic, comparison, and logical ops. rrs r/all_binary_operatorsrD ==r1c|jS)z Fixture for operator and roperator arithmetic functions. Notes ----- This includes divmod and rdivmod, whereas all_arithmetic_operators does not. rrs r/all_arithmetic_functionsrs6 ==r1) countr rrmeanprodstdvarmediankurtskewsemc|jS)z. Fixture for numeric reduction names. rrs r/all_numeric_reductionsr.rr1allanyc|jS)z. Fixture for boolean reduction names. rrs r/all_boolean_reductionsr9rr1c|jS)z> Fixture for all (boolean + numeric) reduction names. rrs r/all_reductionsrDrr1c|jS)z; Fixture for operator module comparison functions. rrs r/ comparison_oprL ==r1)__le____lt____ge____gt__c|jS)zm Fixture for dunder names for compare operations except == and != * >= * > * < * <= rrs r/compare_operators_no_eq_ner]rr1)__and____rand____or____ror____xor____rxor__c|jS)zY Fixture for dunder names for common logical operations * | * & * ^ rrs r/all_logical_operatorsrj ==r1)cumsumcumprodcummincummaxc|jS)z0 Fixture for numeric accumulation names rrs r/all_numeric_accumulationsr{rr1c$|jdS)zR Returns the configuration for the test setting `--no-strict-data-files`. r))r[) pytestconfigs r/strict_data_filesrs  ! !": ;;r1ctjjtjjtdfd}|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. testsctjjg|}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 r4)osrJjoinexists ValueErrorrFskip)rrJ BASE_PATHrs r/decozdatapath..decoscww||I--ww~~d#  *4&0WX KK/$q1 2 r1)rrJrdirname__file__)rrrs` @r/datapathrs2& RWW__X6@I Kr1UTCz Asia/Tokyozdateutil/US/Pacificzdateutil/Asia/Singaporez+01:15z-02:15z UTC+01:15z UTC-02:15)hoursrr+r=i,ic|jS)zD Fixture for trying timezones including default (None): {0} rrs r/tz_naive_fixturerrr1c|jS)z4 Fixture for trying explicit timezones: {0} rrs r/tz_aware_fixturerrr1utcz dateutil/UTCc|jS)zQ Fixture to provide variants of UTC timezone strings and tzinfo objects. rrs r/ utc_fixturerrr1)rmsusnsc|jS)z& datetime64 units we support. rrs r/rrrr1c|jS)zR Parametrized fixture for string dtypes. * str * 'str' * 'U' rrs r/ string_dtyper ==r1pythonpyarrow)string=string[python]string=string[pyarrow]string=str[pyarrow]string=str[python]cL|j\}}tj||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) )rrv StringDtyperstoragena_values r/string_dtype_no_objectrs#0 GX >>'8 ,,r1c|jS)za Parametrized fixture for string dtypes. * 'string[python]' * 'string[pyarrow]' rrs r/nullable_string_dtyper* ==r1c:tj|jS)zq Parametrized fixture for string dtypes backed by Pyarrow. * 'str[pyarrow]' * 'string[pyarrow]' )rvrrrs r/pyarrow_string_dtyper:s >>7== ))r1c|jSzb Parametrized fixture for pd.options.mode.string_storage. * 'python' * 'pyarrow' rrs r/string_storagerJrr1c|jS)z Parametrized fixture for StringDtype storage and na_value. * 'python' + pd.NA * 'pyarrow' + pd.NA * 'pyarrow' + np.nan rrs r/string_dtype_argumentsrZs, ==r1numpy_nullablec|jSrrrs r/ dtype_backendr!srr1c|jS)zK Parametrized fixture for bytes dtypes. * bytes * 'bytes' rrs r/ bytes_dtyper# ==r1c|jS)zN Parametrized fixture for object dtypes. * object * 'object' rrs r/ object_dtyper&r$r1)z string=objectr r r rct|jtjr |jS|j\}}t j ||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) ) isinstancerrurSrvrrs r/any_string_dtyper)sA2'--*}}$MM~~gx00r1c|jS)z\ Parametrized fixture for datetime64 dtypes. * 'datetime64[ns]' * 'M8[ns]' rrs r/datetime64_dtyper+r$r1c|jS)z^ Parametrized fixture for timedelta64 dtypes. * 'timedelta64[ns]' * 'm8[ns]' rrs r/timedelta64_dtyper-r$r1c &tdddddddS)z- Fixture emits fixed Timestamp.now() irlrmr  )rrprhourminuter4 microsecond)rr}r1r/ fixed_now_tsr4s  1RR r1c|jS)z^ Parameterized fixture for float dtypes. * float * 'float32' * 'float64' rrs r/float_numpy_dtyper6rr1c|jS)zR Parameterized fixture for float dtypes. * 'Float32' * 'Float64' rrs r/float_ea_dtyper8r$r1c|jS)z~ Parameterized fixture for float dtypes. * float * 'float32' * 'float64' * 'Float32' * 'Float64' rrs r/any_float_dtyper:rr1c|jS)zg Parameterized fixture for complex dtypes. * complex * 'complex64' * 'complex128' rrs r/ complex_dtyper<rr1c|jS)z Parameterized fixture for complex and numpy float dtypes. * complex * 'complex64' * 'complex128' * float * 'float32' * 'float64' rrs r/complex_or_float_dtyper>s ==r1c|jS)z| Parameterized fixture for signed integer dtypes. * int * 'int8' * 'int16' * 'int32' * 'int64' rrs r/any_signed_int_numpy_dtyper@#rr1c|jS)zx Parameterized fixture for unsigned integer dtypes. * 'uint8' * 'uint16' * 'uint32' * 'uint64' rrs r/any_unsigned_int_numpy_dtyperB1rr1c|jS)z Parameterized fixture for any integer dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' rrs r/any_int_numpy_dtyperD>rr1c|jS)z Parameterized fixture for any nullable integer dtype. * 'UInt8' * 'Int8' * 'UInt16' * 'Int16' * 'UInt32' * 'Int32' * 'UInt64' * 'Int64' rrs r/any_int_ea_dtyperFPrr1c|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' rrs r/ any_int_dtyperHas. ==r1c|jS)z Parameterized fixture for any nullable integer dtype and any float ea dtypes. * 'UInt8' * 'Int8' * 'UInt16' * 'Int16' * 'UInt32' * 'Int32' * 'UInt64' * 'Int64' * 'Float32' * 'Float64' rrs r/any_numeric_ea_dtyperJ{rr1c|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]' rrs r/any_numeric_ea_and_arrow_dtyperLs@ ==r1c|jS)z~ Parameterized fixture for any signed nullable integer dtype. * 'Int8' * 'Int16' * 'Int32' * 'Int64' rrs r/any_signed_int_ea_dtyperNrr1c|jS)z Parameterized fixture for any (purely) real numeric dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' * float * 'float32' * 'float64' rrs r/any_real_numpy_dtyperPs$ ==r1c|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. rrs r/any_real_numeric_dtyperRs( ==r1c|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' rrs r/any_numpy_dtyperTrr1c|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]' rrs r/any_real_nullable_dtyperVs: ==r1c|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' rrs r/any_numeric_dtyperX2s> ==r1cmixedbytesacz mixed-integerg@floatingg?integerzmixed-integer-floatdecimal datetime64z 2013-01-01z 2018-01-01rK20180101riicomplexy??y@@rc^|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 rR)rrurr9)rinferred_dtypers r/any_skipna_inferred_dtyperfus0R%]]NF XXfF +F 6 !!r1ctjddddlm}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:)r])rF importorskipIPython.core.interactiveshellrjtraitlets.configrkHistoryManager hist_file)rjrkrYs r/iprqs<  g6>(A!+A 1 %%r1c#Ktjd}tjd}tjd}|jj} |j 5|j ddddd|jj |jj||jdtjdy#1swYexYw#|jj |jj||jdtjdwxYww)a* Ensure Matplotlib is cleaned up around a test. Before a test is run: 1) Set the backend to "template" to avoid requiring a GUI. After a test is run: 1) Reset units registry 2) Reset rc_context 3) Close all figures See matplotlib/testing/decorators.py#L24. matplotlibzmatplotlib.unitszmatplotlib.pyplottemplateNrrl) rFrlregistryr} rc_contextuseclearupdateclosegccollect)mpl mpl_unitspltorig_units_registrys r/ mpl_cleanuprs"   l +C##$67I   1 2C#,,113  ^^  GGJ     "!!"56 % 1     "!!"56 % 1 s8AEC3-C'C3 AE'C0,C33AEETickc|jS)zN Fixture for Tick based datetime offsets available for a time series. rrs r/ tick_classesrrr1c|Srr}rs r/rrsr1c|jS)zk Simple fixture for testing keys in sorting methods. Tests None (no key) and the identity key. rrs r/ sort_by_keyrrr1)r+NN)EgonVenkmanN)NCC1701Drrc|jS)zP A 3-tuple of names, the first two for operands, the last for a result. rrs r/r6r6s ==r1c|jS)zI Parametrize over __setitem__, loc.__setitem__, iloc.__setitem__ rrs r/ indexer_slirrr1c|jS)z< Parametrize over loc.__getitem__, iloc.__getitem__ rrs r/ indexer_lirrr1c|jS)z8 Parametrize over __setitem__, iloc.__setitem__ rrs r/ indexer_sirrr1c|jS)z7 Parametrize over __setitem__, loc.__setitem__ rrs r/ indexer_slrrr1c|jS)z: Parametrize over at.__setitem__, loc.__setitem__ rrs r/ indexer_alr&rr1c|jS)z< Parametrize over iat.__setitem__, iloc.__setitem__ rrs r/ indexer_ialr.rr1c#Ktjd|j5|jrtjjnddddy#1swYyxYww)z Fixture to check if performance warnings are enabled. Either produces ``PerformanceWarning`` if they are enabled, otherwise ``False``. zmode.performance_warningsFN)rvoption_contextrr'PerformanceWarningrs r/performance_warningr6sA  6 F.5mmbii**F G F Fs!A#+A A#A A#cNtjjjduS)z= Fixture to check if infer string option is enabled. T)rvoptionsfuture infer_stringr}r1r/using_infer_stringr@s ::   ) )T 11r1cNtjjjduS)NT)rvrrpython_scalarsr}r1r/using_python_scalarsrHs ::   + +t 33r1z Europe/Warsawzdateutil/Europe/Warsawz list[Any]_warsawsc|jS)zE tzinfo for Europe/Warsaw using pytz, dateutil, or zoneinfo. rrs r/warsawrRrr1cf|ttjz }|j|S)z Generate a unique file for testing use. See link for removal policy. https://docs.pytest.org/en/7.1.x/how-to/tmp_path.html#the-default-base-temporary-directory )rAuuiduuid4touch)tmp_path file_paths r/ temp_filerZs) 3tzz|,,I OO r1r?)scopec#zKtjj5}|dddy#1swYyxYwwr)rF MonkeyPatchcontext)mps r/ monkeysessionres*    # # % & % %s;/ ;8;c#K|j}tjd| 5|dddy#1swYyxYww)Nzfuture.distinguish_nan_and_na)rrvr)ropts r/using_nan_is_narks3 --C  :G D  E D Ds$A4 A=Ar)rIz pytest.ItemrJrArKrArr)rz type[dict])rztype[abc.Mapping])rr!)rr)rr)rrArzCallable[..., str])rr)rz#tuple[Hashable, Hashable, Hashable])rzIterator[bool | type[Warning]])rbool)rrA(H__doc__ __future__r collectionsrrrrrr r`r r{operatorrtypingr r r dateutil.tzr r hypothesisrstnumpyrurFpandas.compat._optionalrpandas.util._test_decoratorsutil_test_decoratorstdpandas.core.dtypes.dtypesrrpandasrvrrrrrrrrrrrrpandas._testing_testingtm pandas.corerpandas.core.indexes.apir r!collections.abcr"r#r$r pa has_pyarrow ImportErrorr%r0r@rLr`settingsregister_profile HealthChecktoo_slowdiffering_executors load_profilesplitrCgetattrtseriesoffsetsclsregister_type_strategybuildsintegersbooleansfixturerxr~rrrrrrrrrrr skip_if_norrrrrrrrrrrrr NULL_OBJECTSrnulls_fixture2nanNaTrunique_nulls_fixture2NP_NAT_OBJECTSrnp_nat_fixture2rrrrrbox_with_array2rr r(r7r?rGrPrXr9rrr from_breakslinspacerrr#r|idxkeysrr\r(r startswithrrrrr_seriesrNARROW_NP_DTYPESr_narrow_seriesrrPYTHON_DATA_TYPES_typ_objects_seriesrrrrras_unitrarithmetic_dunder_methodsraddraddsubrsubmulrmultruedivrtruedivfloordiv rfloordivmodrmodpowrpoweqneltlegtgeand_rand_xorrxoror_ror_rr_all_numeric_reductionsr_all_boolean_reductionsr_all_reductionsrrrr_all_numeric_accumulationsrrrr TIMEZONESextend FixedOffsetrepr TIMEZONE_IDSparametrize_fixture_docrArr_UTCSappendr utc_fixture2runit2 STRING_DTYPESrNArrrrrr!string_storage2string_dtype_arguments2 BYTES_DTYPESr# OBJECT_DTYPESr&rSr)any_string_dtype2DATETIME64_DTYPESr+TIMEDELTA64_DTYPESr-r4FLOAT_NUMPY_DTYPESr6FLOAT_EA_DTYPESr8ALL_FLOAT_DTYPESr:COMPLEX_DTYPESr<COMPLEX_FLOAT_DTYPESr>SIGNED_INT_NUMPY_DTYPESr@UNSIGNED_INT_NUMPY_DTYPESrBALL_INT_NUMPY_DTYPESrDALL_INT_EA_DTYPESrFALL_INT_DTYPESrHrJALL_INT_PYARROW_DTYPES_STR_REPRFLOAT_PYARROW_DTYPES_STR_REPRrLSIGNED_INT_EA_DTYPESrNALL_REAL_NUMPY_DTYPESrPALL_REAL_DTYPESrRALL_NUMPY_DTYPESrTALL_REAL_NULLABLE_DTYPESrVALL_NUMERIC_DTYPESrXra_any_skipna_inferred_dtyperrfrqr__all__ issubclassrrrr6setitemlocilocrrrratriatrrrrr__annotations__rrrr)rrvalueindex_idrrSpairos00000000r/rFs,!(#   ' >))        K!&:6,I$1        V=2==#=>    e}%&:;< K012h'()e}%&BCDe}%&e}%&e}%&w vr,FGHbffbff-./-r((.HIJ! 6*+,vWh,?@ A vrxx06RSTvy"((;AUVW!  0"   (* eE"I6Iqwqc]I6fE* eE"I6Iqwqc]I6eD* <4*:lB<H *  l<#> * wE * Z^* E)"))B-v .* U9299R= 0* U9299R= 0* U9299R= 0* U9299R= 0* eIBIIbM2* eIBIIbM2* eIBIIbM2* uYRYYr])4!*"uYRYYr])4#*$5$*&9%*&%u )6'*( "K(4)"))Bk2R+RR)*.% "L)D9299R|3T,TT/*4#DL1$455*6) ))+"++a"*EF7*8 U2Y9*: $j$$ !9T:;*@8:A*B  !C*Du'(E*FE)"))B-w7G*HU9299R=9I*JeIBIIbM;K*LU9299R=//5YGM*NUr3AGA3-3;KLO* V r;AGA3-;CTU VC%(L!"|((*+ 8, 8*002 2 U*UJ:W2  .  .',,. .JC NN3 4AA5*- .  + +4CC;(--/ /%8*F#^E%::/  !!$$ % ~~g b E"I6IqBqc(I6S!% F<E7EnE,1134A5A<>;O;O;O%u~~gu -;O        388:;H<H    UU. q "M'7$CD s# & i(IJ   $k2  3 '5 l| 4 < ?     255 i'}r}}Y/GH i(  i0HI 266       YmbmmI&>?    !0r'(r''() 255 i'}r}}Y/GH i(  i0HI 266    1! 1$%r++,-r,,-.r,,-.r))*+r**+ , r(()*r../ 0 r112 3 r334 5 r../0"r++, -  r(()*2r++b.@.@@AB*   (( ) && '   :r../ 0 r//01(r))*+,r**+!,!Hr2234>r,,-.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:3832DG2 1s;,"<,"d&&$@## #A gbjj!,bjjoo >1;  A# k*+,!,    ruu    BFFBGG456()* BGG,-. BFF+,-rvv'(()*e}%G&G2244'(@A)A OODMM/23x !i ! e}%&g@Kp76L4<(  Z  7(D ,DZ s.A|=; A}  A} & A} A} A} A} +A}% A}+'A}6: A}1 A}6A}<F9A~m> A~o9AA~ |=A}}A}}1A}6