o d@sdZddlmZddlmZddlmZmZmZmZm Z ddl m Z ddl Z ddl Z ddlmZmZmZddlmZmZddlZdd lmZddlZddlZdd lmZmZddlmm 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-ddl.m/Z0dd l1m2Z2ddl3m4Z4m5Z5zddl6Z7Wn e8ydZ9Ynw[7dZ9dZ:e-j;rddl:Z:ze:dddZ?dddZ@ejAjBddejCjDfd ejAEdd!FD]ZGeHe&jIjJeGZKeLeKejMeKeNd"d#eOd$qd%FD]"ZGeHe&jIjJeGZKeLeKejMeKeNd&d'eOejNd(d)d*d+q d,FD]"ZGeHe&jIjJeGZKeLeKejMeKeNd-d.eOejNd(d)d*d/q2ejPdd0d1ZQejPdd2dd3d4ZRejPgd5d6d7d8d9d:ZSeSZTejPd(d;gdZUejPgd?d@dAdBZVejPgd?d@dCdDZWejPddgd@dEdFZXejPgdGd@dHdIZYejPgdJd@dKdLZZejPgdMd@dNdOZ[ejPgdMd@dPdQZ\ejPddRdSdTdUdVej]dWe!^dXdYgd@dZd[Z_ejPdRdSdTdUdVej]dWe!^dXdYgd@d\d]Z`ejPddgd@d^d_ZaejPgd`d@dadbZbejPdcddgd@dedfZcejPe0jddgd7d8dhdiZeeeZfejPdejge&jhgd@djdkZieiZjejPe0jkdld7d8dmdnZlelZmejPe'e*gd@dodpZnejPe4e*gdqdrgd8dsdtZoeoZpejPe4e*e&jqggdud8dvdwZrejPe4e*e'e&jqgdxd7d8dydzZsesZtejPd{d|ZuejPd}d~ZvejPddZwejPdddZxejPdddZyddZzddZ{ide0|dde0}dde0j}dddde0~dde0dde0dde0jdddde0jdddde0jdddde0jdddde0jdddde0jdddde0jdddde0jdddde0jdddde0jdddde0dee4ejddke0jdddde0jdddde0de0de4ge5egdgde{eze4gde4eddde4eddde4eddde4ededde4e&jqe0|ddddZe9re4e&jqe0|dddZeed<ejPed@ddqZeZejPddeDd@ddZeZejPddeDd@ddZejPdddZejPdddZejPdddZddZddeDZejPddd„ZejPdddĄZdde0jDZieeeZejPed@ddDŽZejPdddɄZejPddd˄ZejPddd̈́ZejPdddτZejPdddфZejPe(dd'dҍe$ddӃfe(dddҍe$ddӃfe)ddd؍dfe)ddd؍dfe,ddde#ddfe+dddfgd@ddZgdZejPed@ddZejPe je2je je2je je2je je2je je2je je2je je2je je je je je je je je2je je2je je2jgd@ddZejPe je2je je2je je2je je2je je2je je2je je2jgd@ddZgdZejPed@ddZddgZejPed@ddZeeZejPed@ddZejPe je je je je je jgd@ddZejPgdd@ddZejPgdd@ddZgdZejPed@ddZejPddZejPdddZejPdddZdddސdddd d d d eeed edede je ed(de eddddgZe:dure֠e:ZejPe0jd@d?d@ZejPe0jd@dAdBZejPe0jd@dCdDZejPe0jd@dEdFZejPe0je0jd@dGdHZejPe0je0je0je0jd@dIdJZejPe0jd@dKdLZ ejPe0j d@dMdNZ ejPe0j d@dOdPZ ejPe0jd@dQdRZejPe0jd@dSdTZddUejgdVgfddUe&jdVgfdWdUe&jhdVgfdXdYejgdZgfd[ejgejgejggfd[gfd\dUejgd]gfdWdUejgd^gfd_d`ejgd^gfdad(ejgd]gfdbd(ejgd^gfdce d(ejge d]gfddejgdgfdde&jdgfddedeejgedfgfde,dgejge,dhgfdiedjd(d(ejgedkd(d(gfded(ejged]gfdled(ejged]gfde)dje&jhe)dkgfdme(dd(ejge(dd]gfgZee\ZZejPeed8dndoZejPdpdqZejPgdrd@dsdtZejPdude&jJjDd@dvdwZejPddxd7gd@dydzZePd{d|ZejPd}d~dejgejgejgfejge&jhdfejge&jdfe&je&je&jfgd@dddZejPe0je0j e0j!gd@ddZ"ejPe0j e0j!gd@ddZ#ejPe0je0j!gd@ddZ$ejPe0je0j gd@ddZ%ejPe0j&e0j gd@ddZ'ejPe0j(e0j!gd@ddZ)ejPdddZ*ejPdddZ+ddgZ,e:du r.e,e:<dejPe,d@dddZ-dS(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)CallableHashableIterator)tzlocaltzutc) strategies) FixedOffsetutc)DatetimeTZDtype IntervalDtype) DataFrameIntervalPeriodSeries Timedelta Timestampcompat)ops)Index MultiIndexFTUTCreturnNonecCsd|jdddd|jdddd|jdddd|jd dd d|jd dd d|jd ddddS)N --skip-slow store_truezskip slow tests)actionhelp--skip-networkzskip network tests --skip-dbz skip db testsz--run-high-memoryzrun high memory tests --only-slowzrun only slow tests--strict-data-filesz0Fail if a test is skipped for missing data file.)Z addoption)parserr*C/app/.heroku/python/lib/python3.10/site-packages/pandas/conftest.pypytest_addoptionhs r,item pytest.ItempathstrmessagecCs,|j|r|tjd|dSdS)aIgnore doctest warning. Parameters ---------- item : pytest.Item pytest test item. path : str Module path to Python object, e.g. "pandas.core.frame.DataFrame.append". A warning will be filtered when item.name ends with in given path. So it is sufficient to specify e.g. "DataFrame.append". message : str Message to be filtered. zignore:N)nameendswith add_markerpytestmarkfilterwarnings)r-r/r1r*r*r+ignore_doctest_warningws r8cCs,|d}|d}|d}|d}tjjd|dftjjd|dftjjd|dfg}d g}|D]c}|d s>|jd d d rT|tjd|D] \} } t|| | qId|j vr`|tjj |D] \} } } }| |j vr| r}|tj d|n|| qb|rd|j vr|tj dq0dS)Nr!r'r%r&slownetworkz --networkdb)zmissing.mask_zero_div_zerozdivide by zero encounteredz--doctest-modulesz--doctest-cythonF)defaultadd_doctest_importsz/frame/zskipping due to zskipping due to --only-slow) getoptionr5r6r9r:r;r4Z usefixturesr8ZnodeidZ arraymanagerkeywordsskip)itemsconfigZ skip_slowZ only_slowZ skip_networkZskip_dbmarksZignored_doctest_warningsr-r/r1r6ZkwdZ skip_if_foundZarg_namer*r*r+pytest_collection_modifyitemss<         rDci)deadlineZsuppress_health_checkz)MonthBegin MonthEnd BMonthBegin BMonthEndic)n normalizez%YearBegin YearEnd BYearBegin BYearEnd )Z min_valueZ max_value)rHrImonthz1QuarterBegin QuarterEnd BQuarterBegin BQuarterEndi)rHrIZ startingMonthcCst|d<t|d<dS)z: Make `np` and `pd` names available for doctests. nppdN)rPrQ)Zdoctest_namespacer*r*r+r=s r=)ZautousecCstdddS)z< Configure settings for all tests and test modules. Zchained_assignmentraiseN)rQZ set_optionr*r*r*r+configure_testssrS)rrLindexcolumnscCdt|SNzaxis=reprxr*r*r+r\)paramsidscC|jS)z@ Fixture for returning the axis numbers of a DataFrame. paramrequestr*r*r+axisrerUcCrVrWrXrZr*r*r+r\ r]cCr`)zA Fixture for returning aliases of axis 1 of a DataFrame. rarcr*r*r+axis_1 rfrg)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). rarcr*r*r+observed rhcCr`)z6 Boolean 'ordered' parameter for Categorical. rarcr*r*r+ordered rfrjcCr`)z% Boolean 'skipna' parameter. rarcr*r*r+skipna(rfrk)firstlastFcCr`)z[ Valid values for the 'keep' parameter used in .duplicated or .drop_duplicates rarcr*r*r+keep0rn)bothneitherleftrightcCr`)zA Fixture for trying all interval 'inclusive' parameters. rarcr*r*r+inclusive_endpoints_fixture9rfrt)rrrsrprqcCr`)z< Fixture for trying all interval closed parameters. rarcr*r*r+closedArfrucCr`)zS Secondary closed fixture to allow parametrizing over all pairs of closed. rarcr*r*r+ other_closedIrfrvgzipbz2zipxztarZzstdZ zstandard)rCcCr`)zK Fixture for trying common compression types in compression tests. rarcr*r*r+ compressionQr|cCr`)zk Fixture for trying common compression types in compression tests excluding uncompressed case. rarcr*r*r+compression_onlycr}r~cCr`)z, Fixture that an array is writable. rarcr*r*r+writableurfr)innerouterrrrscCr`)z: Fixture for trying all types of join operations. rarcr*r*r+ join_type}rfrnlargest nsmallestcCr`)z1 Fixture for trying all nselect methods. rarcr*r*r+nselect_methodrfrcC t|jSNtype__name__rZr*r*r+r\ cCr`)z/ Fixture for each null type in pandas. rarcr*r*r+ nulls_fixturerfrcCr`)zL Fixture for each null type in pandas, each null type exactly once. rarcr*r*r+unique_nulls_fixturerfrcCrrrrZr*r*r+r\rcCr`)z- Fixture for each NaT type in numpy. rarcr*r*r+np_nat_fixturerfrcCr`)z; Fixture to parametrize over DataFrame and Series. rarcr*r*r+frame_or_seriesrfrrTseriescCr`)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 rarcr*r*r+index_or_seriesrir)rTrarraycCr`)zG Fixture to parametrize over Index, Series, and ExtensionArray rarcr*r*r+index_or_series_or_arrayrfrcCr`r)rrZr*r*r+r\cCr`)z] Fixture to test behavior for Index, Series, DataFrame, and pandas Array classes rarcr*r*r+box_with_arrayrorcCsGdddt}|S)z, Fixture for a dictionary subclass. c@seZdZdddZdS)z"dict_subclass..TestSubDictrr c_stj|g|Ri|dSr)dict__init__)selfargskwargsr*r*r+rsz+dict_subclass..TestSubDict.__init__Nrr )r __module__ __qualname__rr*r*r*r+ TestSubDictsr)r)rr*r*r+ dict_subclasssrcCsGdddtj}|S)z8 Fixture for a non-mapping dictionary subclass. c@s2eZdZdddZddZddd Zdd d Zd S)z5non_dict_mapping_subclass..TestNonDictMappingrr 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)rr )rr)rrrrrrrr*r*r*r+TestNonDictMappings   r)rMapping)rr*r*r+non_dict_mapping_subclasss rcCs^td}|ddddddg}|jdd|jjD|_|jjgdd d |S) z DataFrame with 3 level MultiIndex (year, month, day) covering first 100 business days from 2000-01-01 with random data dcSr`r)yearrZr*r*r+r\rzAmultiindex_year_month_day_dataframe_random_data..cSr`r)rNrZr*r*r+r\rcSr`r)dayrZr*r*r+r\rcSsg|]}|dqS)i8)astype).0Zlevr*r*r+ szCmultiindex_year_month_day_dataframe_random_data..)rrNrT)Zinplace)tmZmakeTimeDataFramegroupbysumrTZ set_levelslevelsZ set_names)ZtdfZymdr*r*r+/multiindex_year_month_day_dataframe_random_data s  rrcCs*tgdgdggdgdgddgdS)z; 2-level MultiIndex, lexsorted, with string names. foobarbazquxonetwothree) rrrrLrLrrr) rrLrrrLrLrrrLrrlsecond)rcodesnames)rr*r*r*r+%lexsorted_two_level_string_multiindexs rrcCs(|}ttjdd|tgddddS)z2DataFrame with 2 level MultiIndex with random data r)ABCexpr2)rTrU)rrPrandomrandnr)rrTr*r*r+ multiindex_dataframe_random_data%srcCsTtgd}tddg}tgd}tgd}ddg}t||g||g|dd S) zJ MultiIndex used to test the general functionality of this object rrr)rrrLrrr)rrLrrLrrLrlrF)rrrZverify_integrity)rrPrr)Z major_axisZ minor_axisZ major_codesZ minor_codesZ index_namesr*r*r+_create_multiindex0s  rcCs,tjddgddgtjddddggd d S) zB MultiIndex with a level that is a tzaware DatetimeIndex. rLrab20130101r US/Eastern)Zperiodstzr)r)rZ from_productrQZ date_ranger*r*r*r+_create_mi_with_dt64tz_levelDsrstringrrz datetime-tzz US/Pacific)rZperiodrrangeZint8dtypeZint16int32int64Zuint8Zuint16Zuint32Zuint64float32Zfloat64z bool-objectr complex64 complex128)rrr)rLrr)rrrLrLrrZInt64ZUInt16ZFloat32booleanzstring[python]) bool-dtyperrZ categoricalintervalemptytuplesmi-with-dt64tz-levelmultirepeatsZ nullable_intZ nullable_uintZnullable_floatZ nullable_boolz string-pythonzstring[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_dictrbcopyrcr*r*r+rTvs cCsg|] \}}t|ts|qSr*) isinstancerrrvaluer*r*r+rs rcCs|j}t|S)z8 index fixture, but excluding MultiIndex cases. )rbrr)rdrr*r*r+ index_flats rcCsDg|]\}}|ds|ds|ds|dvst|ts|qS)rZuintfloat)rrrr) startswithrrrr*r*r+rs cCst|jjdd}|j}|jdvr2|}d|ddd|d<d|ddd|d<t|Sd|d<d|d<t||S) z Fixture for indices with missing values. Integer-dtype and empty cases are excluded because they cannot hold missing values. MultiIndex is excluded because isna() is not defined for MultiIndex. Tdeep)rrrrrrLN)rrbrvaluestolistr from_tuplesr)rdindvalsr*r*r+index_with_missings   rrcCt}d|_|S)zC Fixture for Series of floats with Index of unique strings r)rZmakeStringSeriesr2sr*r*r+ string_seriesrcCr)zI Fixture for Series of dtype object with Index of unique strings objects)rZmakeObjectSeriesr2rr*r*r+ object_seriesrrcCr)z9 Fixture for Series of floats with DatetimeIndex ts)rZmakeTimeSeriesr2rr*r*r+datetime_seriesrrcCs$t|}tj|}t||dddS)zHelper for the _series dictrF)rTr2r)lenrPrrr)rTsizedatar*r*r+_create_seriess rcCs"i|] \}}d|dt|qS)z series-with-z-indexr)rZindex_idrTr*r*r+ srcCst|S)zE Fixture for tests on series with changing types of indices. rrTr*r*r+series_with_simple_indexsr cCsHgdgdg}t|}t|}tjd}t||d}tj|d<|S)z: Fixture with a Series with a 2-level MultiIndex. )rrrrrrrr)rrrrrrrrrr)ryrrrPrrrNaN)ZarraysrrTrZserr*r*r+series_with_multilevel_indexs    r cCs$i|]}|jdtjd|dqS)z-seriesr)r2r)rrZmake_rand_series)rrr*r*r+rscCst|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_objsrbrrcr*r*r+index_or_series_objsrcCsttdS)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 getSeriesDatarr*r*r*r+ int_framesrcC ttS)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] )rrZgetTimeSeriesDatar*r*r*r+datetime_frame: rcCr)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] )rrrr*r*r*r+ float_frameWrrc Cs>tdddtjdgdddtjdgdddd tdd S) z Fixture for DataFrame of float/int/string columns with RangeIndex Columns are ['a', 'b', 'c', 'float32', 'int32']. ?rrrrrrLr)rrcrrr)rrPraranger*r*r*r+mixed_type_frametsrc Cstdddtdddtdddtdddtdddtdddtdddtdddtdddtdddg }ttjt||dS)zF Fixture for Series with a DatetimeIndex that has duplicates. irLrrrKr)rrrPrrr)datesr*r*r+(rand_series_with_duplicate_datetimeindexs           r)rrrsrsg?g?z2012-01M)freqz period[M]z 2012-02-01Dz period[D]z 2011-01-01ri)secondsztimedelta64[ns]cCr`rrarcr*r*r+ea_scalar_and_dtyper )__add____radd____sub____rsub____mul____rmul__ __floordiv__ __rfloordiv__ __truediv__ __rtruediv____pow____rpow____mod____rmod__cCr`)zD Fixture for dunder names for common arithmetic operations. rarcr*r*r+all_arithmetic_operatorsrfr0cCr`)zU Fixture for operator and roperator arithmetic, comparison, and logical ops. rarcr*r*r+all_binary_operators"r1cCr`)z Fixture for operator and roperator arithmetic functions. Notes ----- This includes divmod and rdivmod, whereas all_arithmetic_operators does not. rarcr*r*r+all_arithmetic_functionssr3) countrmaxminmeanprodZstdvarZmedianZkurtZskewZsemcCr`)z. Fixture for numeric reduction names. rarcr*r*r+all_numeric_reductionsrfr:allanycCr`)z. Fixture for boolean reduction names. rarcr*r*r+all_boolean_reductions*rfr=cCr`)z> Fixture for all (boolean + numeric) reduction names. rarcr*r*r+all_reductions5rfr>cCr`)z; Fixture for operator module comparison functions. rarcr*r*r+ comparison_op=r!r?)__le____lt____ge____gt__cCr`)zm Fixture for dunder names for compare operations except == and != * >= * > * < * <= rarcr*r*r+compare_operators_no_eq_neNrirD)__and____rand____or____ror____xor____rxor__cCr`)zY Fixture for dunder names for common logical operations * | * & * ^ rarcr*r*r+all_logical_operators[ rK)ZcumsumZcumprodZcumminZcummaxcCr`)z0 Fixture for numeric accumulation names rarcr*r*r+all_numeric_accumulationslrfrMcCs |dS)zO Returns the configuration for the test setting `--strict-data-files`. r()r>)Z pytestconfigr*r*r+strict_data_filesws rNCallable[..., str]cs(tjtjtdfdd}|S)a= 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 --strict-data-files option is set. testscsJtjjg|R}tj|s#rtd|dtd|d|S)NzCould not find file z and --strict-data-files is set.zCould not find .)osr/joinexists ValueErrorr5r@)rr/Z BASE_PATHrNr*r+decos  zdatapath..deco)rRr/rSdirname__file__)rNrWr*rVr+datapaths rZcCst|ddddS)z* The iris dataset as a DataFrame. iorcsvziris.csv)rQZread_csv)rZr*r*r+irissr]z Asia/Tokyozdateutil/US/Pacificzdateutil/Asia/Singaporez+01:15z-02:15z UTC+01:15z UTC-02:15i,i)hoursrrrcCsg|]}t|qSr*rX)rir*r*r+rscCr`)zD Fixture for trying timezones including default (None): {0} rarcr*r*r+tz_naive_fixtureror`cCr`)z4 Fixture for trying explicit timezones: {0} rarcr*r*r+tz_aware_fixturerorarz dateutil/UTCcCr`)zQ Fixture to provide variants of UTC timezone strings and tzinfo objects. rarcr*r*r+ utc_fixturerfrbcCr`)zR Parametrized fixture for string dtypes. * str * 'str' * 'U' rarcr*r*r+ string_dtype rcpyarrowcCr`)za Parametrized fixture for string dtypes. * 'string[python]' * 'string[pyarrow]' rarcr*r*r+nullable_string_dtype rfpythoncCr`zb Parametrized fixture for pd.options.mode.string_storage. * 'python' * 'pyarrow' rarcr*r*r+string_storage rgrjZnumpy_nullablecCr`rirarcr*r*r+ dtype_backendrgrkcCr`)zK Parametrized fixture for bytes dtypes. * bytes * 'bytes' rarcr*r*r+ bytes_dtype.rlcCr`)zN Parametrized fixture for object dtypes. * object * 'object' rarcr*r*r+ object_dtype9rmrnobjectcCr`)zo Parametrized fixture for string dtypes. * 'object' * 'string[python]' * 'string[pyarrow]' rarcr*r*r+any_string_dtypeDr!rpcCr`)z\ Parametrized fixture for datetime64 dtypes. * 'datetime64[ns]' * 'M8[ns]' rarcr*r*r+datetime64_dtypeUrmrqcCr`)z^ Parametrized fixture for timedelta64 dtypes. * 'timedelta64[ns]' * 'm8[ns]' rarcr*r*r+timedelta64_dtype`rmrrrc CstddddddddS)z- Fixture emits fixed Timestamp.now() irLrMr )rrNrhourminuter microsecond)rr*r*r*r+ fixed_now_tsksrxcCr`)z^ Parameterized fixture for float dtypes. * float * 'float32' * 'float64' rarcr*r*r+float_numpy_dtypeurdrycCr`)zR Parameterized fixture for float dtypes. * 'Float32' * 'Float64' rarcr*r*r+float_ea_dtypermrzcCr`)z~ Parameterized fixture for float dtypes. * float * 'float32' * 'float64' * 'Float32' * 'Float64' rarcr*r*r+any_float_dtyperLr{cCr`)zg Parameterized fixture for complex dtypes. * complex * 'complex64' * 'complex128' rarcr*r*r+ complex_dtyperdr|cCr`)z| Parameterized fixture for signed integer dtypes. * int * 'int8' * 'int16' * 'int32' * 'int64' rarcr*r*r+any_signed_int_numpy_dtyperLr}cCr`)zx Parameterized fixture for unsigned integer dtypes. * 'uint8' * 'uint16' * 'uint32' * 'uint64' rarcr*r*r+any_unsigned_int_numpy_dtyperir~cCr`)z Parameterized fixture for any integer dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' rarcr*r*r+any_int_numpy_dtyper}rcCr`)z Parameterized fixture for any nullable integer dtype. * 'UInt8' * 'Int8' * 'UInt16' * 'Int16' * 'UInt32' * 'Int32' * 'UInt64' * 'Int64' rarcr*r*r+any_int_ea_dtyper!rcCr`)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' rarcr*r*r+ any_int_dtypesrcCr`)z Parameterized fixture for any nullable integer dtype and any float ea dtypes. * 'UInt8' * 'Int8' * 'UInt16' * 'Int16' * 'UInt32' * 'Int32' * 'UInt64' * 'Int64' * 'Float32' * 'Float64' rarcr*r*r+any_numeric_ea_dtypesrcCr`)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]' rarcr*r*r+any_numeric_ea_and_arrow_dtypes rcCr`)z~ Parameterized fixture for any signed nullable integer dtype. * 'Int8' * 'Int16' * 'Int32' * 'Int64' rarcr*r*r+any_signed_int_ea_dtype7rircCr`)z Parameterized fixture for any (purely) real numeric dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' * float * 'float32' * 'float64' rarcr*r*r+any_real_numpy_dtypeDsrcCr`)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. rarcr*r*r+any_real_numeric_dtypeYsrcCr`)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' rarcr*r*r+any_numpy_dtypepr2rcCr`)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' rarcr*r*r+any_numeric_dtypesrrrmixedbytesacrz mixed-integerrg@Zfloatingrintegerzmixed-integer-floatdecimal datetime64z 2013-01-01z 2018-01-01rZ20180101riirrcCs |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)rbrPrro)rdZinferred_dtyperr*r*r+any_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:)rB)r5 importorskipZIPython.core.interactiveshellrZtraitlets.configrZHistoryManagerZ hist_file)rrrr*r*r+ip s    r)ZbsrZcooZcscZcsrZdiaZdokZlilcCsddlm}t||jdS)z- Yields scipy sparse matrix classes. r)sparseZ_matrix)Zscipyrgetattrrb)rdrr*r*r+spmatrixs rcCs6g|]}tttj|tjjr|dkrttj|qS)Tick) issubclassrrQoffsetsr)ror*r*r+r's cCr`)zN Fixture for Tick based datetime offsets available for a time series. rarcr*r*r+ tick_classes&rLrcCs|Srr*rZr*r*r+r\4scCr`)zk Simple fixture for testing keys in sorting methods. Tests None (no key) and the identity key. rarcr*r*r+ sort_by_key4rorccsztdddlm}ddlm}ddlm}Gddd|}|d|d d |V|ddd|j d<|j dS) Nfsspecr)register_implementation)MemoryFileSystem) _registrycs(eZdZdZdgZdfdd ZZS)z fsspectest..TestMemoryFStestmemNrr cs(|dd|jd<tjdi|dS)Ntestrr*)poprsuperr)rr __class__r*r+rHsz)fsspectest..TestMemoryFS.__init__r)rrrprotocolrr __classcell__r*r*rr+ TestMemoryFSDsrrT)clobber) r5rrrZfsspec.implementations.memoryrZfsspec.registryrrrstoreclear)rrregistryrr*r*r+ fsspectest=s      r)rNN)ZEgonZVenkmanN)NCC1701Drr#tuple[Hashable, Hashable, Hashable]cCr`)zP A 3-tuple of names, the first two for operands, the last for a result. rarcr*r*r+rSsrcCr`)zI Parametrize over __setitem__, loc.__setitem__, iloc.__setitem__ rarcr*r*r+ indexer_slifrfrcCr`)z< Parametrize over loc.__getitem__, iloc.__getitem__ rarcr*r*r+ indexer_linrfrcCr`)z8 Parametrize over __setitem__, iloc.__setitem__ rarcr*r*r+ indexer_sivrfrcCr`)z7 Parametrize over __setitem__, loc.__setitem__ rarcr*r*r+ indexer_sl~rfrcCr`)z: Parametrize over at.__setitem__, loc.__setitem__ rarcr*r*r+ indexer_alrfrcCr`)z< Parametrize over iat.__setitem__, iloc.__setitem__ rarcr*r*r+ indexer_ialrfrboolcCstjjjdkS)z> Fixture to check if the array manager is being used. r)rQoptionsmode data_managerr*r*r*r+using_array_managersrcCstjjjo tjjjdkS)z7 Fixture to check if Copy-on-Write is enabled. block)rQrrZ copy_on_writerr*r*r*r+using_copy_on_writesrz Europe/Warsawzdateutil/Europe/WarsawcCr`)zE tzinfo for Europe/Warsaw using pytz, dateutil, or zoneinfo. rarcr*r*r+warsawrfrr)r-r.r/r0r1r0rr )rr)rr)rr)rNr0rrO)rr)rr)rr)rr0(.__doc__ __future__r collectionsrrrrrrrr operatorrRtypingr r r Z dateutil.tzr rZ hypothesisrstnumpyrPr5ZpytzrrZpandas.util._test_decoratorsutilZ_test_decoratorstdZpandas.core.dtypes.dtypesrrZpandasrQrrrrrrrZpandas._testingZ_testingrZ pandas.corerZpandas.core.indexes.apirrrepa ImportErrorZ has_pyarrowZzoneinfoZPY39ZZoneInfoZZoneInfoNotFoundErrorr,r8rDsettingsZregister_profileZ HealthCheckZtoo_slowZ load_profilesplitr2rZtseriesrclsZregister_type_strategyZbuildsZintegersZbooleansZfixturer=rSreZ axis_framergrhrjrkrnrtrurvrbZ skip_if_nor|r~rrrZ NULL_OBJECTSrZnulls_fixture2nanZNaTrZunique_nulls_fixture2ZNP_NAT_OBJECTSrZnp_nat_fixture2rrZindex_or_series2rrrZbox_with_array2rrrrrrrZmakeStringIndexZ makeDateIndexZmakePeriodIndexZmakeTimedeltaIndexZmakeRangeIndexZ makeIntIndexZ makeUIntIndexZmakeFloatIndexZ makeBoolIndexrrorrZmakeNumericIndexZmakeCategoricalIndexZmakeIntervalIndexrryrrridxkeysrTZindex_fixture2rArZ index_flat2rrrrrZ_seriesr r ZNARROW_NP_DTYPESZ_narrow_seriesr rrrrrrr Z_all_arithmetic_operatorsr0addZraddsubZrsubmulZrmultruedivZrtruedivfloordivZ rfloordivmodZrmodpowZrpoweqneltlegtgeand_Zrand_xorZrxoror_Zror_r1r3Z_all_numeric_reductionsr:Z_all_boolean_reductionsr=Z_all_reductionsr>r?rDrKZ_all_numeric_accumulationsrMrNrZr]Z TIMEZONESextendZ TIMEZONE_IDSZparametrize_fixture_docr0r`raZtz_aware_fixture2Z_UTCSappendrbZ utc_fixture2Z STRING_DTYPESrcrfrjrkZstring_storage2Z BYTES_DTYPESrlZ OBJECT_DTYPESrnrpZDATETIME64_DTYPESrqZTIMEDELTA64_DTYPESrrrxZFLOAT_NUMPY_DTYPESryZFLOAT_EA_DTYPESrzZALL_FLOAT_DTYPESr{ZCOMPLEX_DTYPESr|ZSIGNED_INT_NUMPY_DTYPESr}ZUNSIGNED_INT_NUMPY_DTYPESr~ZALL_INT_NUMPY_DTYPESrZALL_INT_EA_DTYPESrZALL_INT_DTYPESrrZALL_INT_PYARROW_DTYPES_STR_REPRZFLOAT_PYARROW_DTYPES_STR_REPRrZSIGNED_INT_EA_DTYPESrZALL_REAL_NUMPY_DTYPESrZALL_REAL_DTYPESrZALL_NUMPY_DTYPESrZALL_NUMERIC_DTYPESrZNArZ_any_skipna_inferred_dtyper__rrr__all__rrrrsetitemlocZilocrrrratrZiatrrrZwarsawsrr*r*r*r+s,    $     1                                                "                             !                        $#$$2