a < bs%@sbdZddlmZddlmZmZmZmZmZddlm Z ddl Z ddl Z ddl m Z mZddlZddlmZddlZddlZddlmZmZddlmmZdd lmZmZddlZ dd lm!Z!m"Z"m#Z#m$Z$m%Z%m&Z&ddl'm(Z)dd l*m+Z+dd l,m-Z-m.Z.z ddl/Z0Wne1y"d Z2Yn0[0dZ2ej34dZ5ddZ6ddZ7ej8j9ddej:j;fdej8e?e j@jAe>ZBeCeBejDeBeEddeFdqrd=D]DZ>e?e j@jAe>ZBeCeBejDeBeEddeFejEdddd qd!=D]DZ>e?e j@jAe>ZBeCeBejDeBeEd"d#eFejEdddd$qejGd%d&ZHejGdd'd(d)ZIejGgd*d+d,d-d.d/ZJeJZKejGgd0d1d2d3ZLejGgd0d1d4d5ZMejGgd6d1d7d8ZNejGgd9d1d:d;ZOejGgd<d1d=d>ZPejGgd<d1d?d@ZQejGddAdBdCdDejRdEeSdFdGgd1dHdIZTejGdAdBdCdDejRdEeSdFdGgd1dJdKZUejGdd gd1dLdMZVejGgdNd1dOdPZWejGdQdRgd1dSdTZXejGe)jYdUd,d-dVdWZZeZZ[ejGdej\e j]gd1dXdYZ^e^Z_ejGe)j`dZd,d-d[d\ZaeaZbejGe!e$gd1d]d^ZcejGe-e$gd_d`gd-dadbZdedZeejGe-e$e jfggdcd-dddeZgejGe-e$e!e jfgdfd,d-dgdhZhehZiejGdidjZjejGdkdlZkejGdmdnZlejGdodpZmejGdqdrZndsdtZodudvZpe)qdwe)rdwe)sdwe)jsdwdxdye)tdwe)udwe)vdwe)wdwe)xdwe)ydwe)jzdwdzd{e)jzdwd|d{e)jzdwd}d{e)jzdwd~d{e)jzdwdd{e)jzdwdd{e)jzdwdd{e)jzdwdd{e)jzdwdd{e)jzdwdd{e){de)|dwe)}dwe-ge.~egdgdepeoe-gde-edwdd{e-edwdd{e-edwdd{e-edwedd{e-e jfe)rdwdd{d!Ze2re-e jfe)rdwdd{Zeed<ejGed1dd_ZeZejGddeDd1ddZeZejGddeDd1ddZejGddeDd1ddZejGddZejGddZejGddZddZddeDZejGddZejGddZdde)jDZieeeZejGed1ddZejGddZejGddZejGddZejGddZejGddZejGe"dddedzdfe"dddeddfe#ddddfe#ddddfe&dddyeddyfe%ddōdfgd1ddȄZgdɢZejGed1dd˄ZejGe je+je je+je je+je je+je je+je je+je je+je je je je je je je je+je je+je je+jgd1dd̈́ZejGe je+je je+je je+je je+je je+je je+je je+jgd1ddτZgdТZejGed1dd҄ZddgZejGed1ddքZeeZejGed1dd؄ZejGe je je je je je jgd1ddڄZejGgdۢd1dd݄ZejGgdޢd1ddZejGddZejGddZejGddZddddddddddee edededejeeddeeddddgZddeDZeeeɃejGeed-ddZeeeddejGeddeddd-ddZeZejGddeeejgd1ddZeZejGe)jd1ddZejGdejRdejSddddGgd1ddZejGdejRdejSddddGgd1ddZeZejGe)jd1dd ZejGe)jd1d d ZejGd dejRdejSddddGgd1d dZejGe)jd1ddZejGe)jd1ddZejGddZejGe)jd1ddZejGe)jd1ddZejGe)je)jd1ddZejGe)jd1ddZejGe)jd1ddZejGe)jd1dd ZejGe)jd1d!d"ZejGe)jd1d#d$ZejGe)je)jd1d%d&ZejGe)je)jd1d'd(ZejGe)jd1d)d*ZejGe)jd1d+d,ZejGe)jd1d-d.Zd/d0ej\d1gfd/d0e jd1gfd2d3ej\d4gfd5ej\ej\ej\gfd5gfd6d0ej\d7gfd8d0ej\d9gfd:d;ej\d9gfde dej\e d7gfddej\d gfdde jd gfd?ed@ej\edAgfdBe&dCej\e&dDgfdEedFddej\edGddgfdHedej\ed7gfdIedej\ed7gfdJe#dFe j]e#dGgfdKe"ddej\e"dd7gfgZee\ZZejGeed-dLdMZejGdNdOZejGgdPd1dQdRZejGdSde jAjDd1dTdUZejGddVd,gd1dWdXZeGdYdZZejGd[d\d]ej\ej\ej\fej\e j]dfej\e jdfe je je jfgd1d^d_ZejGe)je)je)jgd1d`daZejGe)je)jgd1dbdcZejGe)je)jgd1dddeZ ejGe)j e)jgd1dfdgZ ejGe)j e)jgd1dhdiZ ejGdjdkZdS(la 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 )abc)datedatetimetime timedeltatimezone)DecimalN)tzlocaltzutc) strategies) FixedOffsetutc)DatetimeTZDtype IntervalDtype) DataFrameIntervalPeriodSeries Timedelta Timestamp)ops)Index MultiIndexFTz3ignore:Promotion of numbers and bools:FutureWarningcCsd|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 tests--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#_/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/pandas/conftest.pypytest_addoptionYsr%c CsD|d}|d}|d}|d}|d}tjjd|dftjjd|dftjjd |dfg}|D]}|d s|jd d d r|tjdd|jvr|tjj |t |D]@\} } } } | |j vr| r|tj d| q|| q|rd|j vr|tj dd|j vrd|sd|tj dqddS)Nrr rrrslownetworkz --networkdbz--doctest-modulesz--doctest-cythonF)defaultadd_doctest_importsz/frame/zskipping due to zskipping due to --only-slowZ high_memoryz=skipping high memory test since --run-high-memory was not set) getoptionpytestmarkr&r'r(Z add_markerZ usefixturesZnodeidZ arraymanager suppress_npdev_promotion_warningkeywordsskip) itemsconfigZ skip_slowZ only_slowZ skip_networkZskip_dbZrun_high_memorymarksitemr-ZkwdZ skip_if_foundZarg_namer#r#r$pytest_collection_modifyitemshs>         r5ci)deadlineZsuppress_health_checkz)MonthBegin MonthEnd BMonthBegin BMonthEndic)n normalizez%YearBegin YearEnd BYearBegin BYearEnd )Z min_valueZ max_value)r9r:monthz1QuarterBegin QuarterEnd BQuarterBegin BQuarterEndi)r9r:Z startingMonthcCst|d<t|d<dS)z: Make `np` and `pd` names available for doctests. nppdN)rArB)Zdoctest_namespacer#r#r$r*sr*)ZautousecCstdddS)z< Configure settings for all tests and test modules. Zchained_assignmentraiseN)rBZ set_optionr#r#r#r$configure_testssrD)rr=indexcolumnscCsdt|S)Nzaxis=reprxr#r#r$rK)paramsidscCs|jS)z@ Fixture for returning the axis numbers of a DataFrame. paramrequestr#r#r$axissrS)TFN)rMcCs|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). rOrQr#r#r$observeds rTcCs|jS)z6 Boolean 'ordered' parameter for Categorical. rOrQr#r#r$orderedsrU)firstlastFcCs|jS)z[ Valid values for the 'keep' parameter used in .duplicated or .drop_duplicates rOrQr#r#r$keepsrX)bothneitherleftrightcCs|jS)zA Fixture for trying all interval 'inclusive' parameters. rOrQr#r#r$inclusive_endpoints_fixturesr])r[r\rYrZcCs|jS)z< Fixture for trying all interval closed parameters. rOrQr#r#r$closedsr^cCs|jS)zS Secondary closed fixture to allow parametrizing over all pairs of closed. rOrQr#r#r$ other_closedsr_gzipbz2zipxzZzstdZ zstandard)r3cCs|jS)zK Fixture for trying common compression types in compression tests. rOrQr#r#r$ compressionsrdcCs|jS)zk Fixture for trying common compression types in compression tests excluding uncompressed case. rOrQr#r#r$compression_only/srecCs|jS)z, Fixture that an array is writable. rOrQr#r#r$writable@srf)innerouterr[r\cCs|jS)z: Fixture for trying all types of join operations. rOrQr#r#r$ join_typeHsrinlargest nsmallestcCs|jS)z1 Fixture for trying all nselect methods. rOrQr#r#r$nselect_methodPsrlcCs t|jSNtype__name__rIr#r#r$rK[rLcCs|jS)z/ Fixture for each null type in pandas. rOrQr#r#r$ nulls_fixture[srqcCs|jS)zL Fixture for each null type in pandas, each null type exactly once. rOrQr#r#r$unique_nulls_fixturefsrrcCs t|jSrmrnrIr#r#r$rKrrLcCs|jS)z- Fixture for each NaT type in numpy. rOrQr#r#r$np_nat_fixturersrscCs|jS)z; Fixture to parametrize over DataFrame and Series. rOrQr#r#r$frame_or_seriessrtrEseriescCs|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 rOrQr#r#r$index_or_seriess rv)rEruarraycCs|jS)zG Fixture to parametrize over Index, Series, and ExtensionArray rOrQr#r#r$index_or_series_or_arraysrxcCs|jSrm)rprIr#r#r$rKrLcCs|jS)z] Fixture to test behavior for Index, Series, DataFrame, and pandas Array classes rOrQr#r#r$box_with_arraysrycCsGdddt}|S)z, Fixture for a dictionary subclass. c@seZdZddZdS)z"dict_subclass..TestSubDictc_stj|g|Ri|dSrm)dict__init__)selfargskwargsr#r#r$r{sz+dict_subclass..TestSubDict.__init__N)rp __module__ __qualname__r{r#r#r#r$ TestSubDictsr)rz)rr#r#r$ dict_subclasssrcCsGdddtj}|S)z8 Fixture for a non-mapping dictionary subclass. c@s,eZdZddZddZddZddZd S) z5non_dict_mapping_subclass..TestNonDictMappingcSs ||_dSrm)_data)r|Zunderlying_dictr#r#r$r{sz>non_dict_mapping_subclass..TestNonDictMapping.__init__cSs |j|Srm)r __getitem__)r|keyr#r#r$rszAnon_dict_mapping_subclass..TestNonDictMapping.__getitem__cSs |jSrm)r__iter__r|r#r#r$rsz>non_dict_mapping_subclass..TestNonDictMapping.__iter__cSs |jSrm)r__len__rr#r#r$rsz=non_dict_mapping_subclass..TestNonDictMapping.__len__N)rprrr{rrrr#r#r#r$TestNonDictMappingsr)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 dcSs|jSrm)yearrIr#r#r$rKrLzAmultiindex_year_month_day_dataframe_random_data..cSs|jSrm)r?rIr#r#r$rKrLcSs|jSrm)dayrIr#r#r$rKrLcSsg|]}|dqS)i8)astype).0Zlevr#r#r$ rLzCmultiindex_year_month_day_dataframe_random_data..)rr?rT)Zinplace)tmZmakeTimeDataFramegroupbysumrEZ set_levelslevelsZ set_names)ZtdfZymdr#r#r$/multiindex_year_month_day_dataframe_random_datas   rcCs*tgdgdggdgdgddgdS)z; 2-level MultiIndex, lexsorted, with string names. foobarbazquxonetwoZthree) rrrr=r=rrr) rr=rrr=r=rrr=rrVsecond)rcodesnames)rr#r#r#r$%lexsorted_two_level_string_multiindexs rcCs(|}ttjdd|tgddddS)z2DataFrame with 2 level MultiIndex with random data r)ABCexpname)rErF)rrArandomrandnr)rrEr#r#r$ multiindex_dataframe_random_datasrcCsTtgd}tddg}tgd}tgd}ddg}t||g||g|dd S) zJ MultiIndex used to test the general functionality of this object rrr)rrr=rrr)rr=rr=rr=rVrF)rrrZverify_integrity)rrArwr)Z major_axisZ minor_axisZ major_codesZ minor_codesZ index_namesr#r#r$_create_multiindexs  rcCs,tjddgddgtjddddggd d S) zB MultiIndex with a level that is a tzaware DatetimeIndex. r=rab20130101r US/Eastern)Zperiodstzr)r)rZ from_productrBZ date_ranger#r#r#r$_create_mi_with_dt64tz_levelsrrz US/Pacific)rint64dtypeint32Zint16Zint8Zuint64Zuint32Zuint16Zuint8Zfloat64float32r)rrr)r=rr)rrr=r=rrZInt64ZUInt16ZFloat32booleanzstring[python])!unicodestringrz datetime-tzperiodrintuintrangefloatZ num_int64Z num_int32Z num_int16Znum_int8Z num_uint64Z num_uint32Z num_uint16Z num_uint8Z num_float64Z num_float32boolZ 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_dictrPcopyrQr#r#r$rECs cCsg|]}tt|ts|qSr#) isinstancerrrrr#r#r$rWsrcCs|j}t|S)z8 index fixture, but excluding MultiIndex cases. rPrrrRrr#r#r$ index_flatVs rcCs(g|] }tt|tst|jr|qSr#)rrrZ is_uniquerr#r#r$rhscCs|j}t|S)z1 index_flat with uniqueness requirement. rrr#r#r$index_flat_uniquegs rcCs0g|](}|dvs|dstt|ts|qS))rrrrrZnum_) startswithrrrrr#r#r$rws  cCst|jjdd}|j}|jdvrd|}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)rrrrmrr=N)rrPrvaluestolistr from_tuplesro)rRindvalsr#r#r$index_with_missingvs  rcCst}d|_|S)zC Fixture for Series of floats with Index of unique strings ru)rZmakeStringSeriesrsr#r#r$ string_seriessrcCst}d|_|S)zI Fixture for Series of dtype object with Index of unique strings Zobjects)rZmakeObjectSeriesrrr#r#r$ object_seriessrcCst}d|_|S)z9 Fixture for Series of floats with DatetimeIndex ts)rZmakeTimeSeriesrrr#r#r$datetime_seriessrcCs"t|}tj|}t||ddS)zHelper for the _series dictr)rEr)lenrArrr)rEsizedatar#r#r$_create_seriess rcCs"i|]\}}d|dt|qS)z series-with-z-indexr)rZindex_idrEr#r#r$ srcCst|S)zE Fixture for tests on series with changing types of indices. rrEr#r#r$series_with_simple_indexsrcCsHgdgdg}t|}t|}tjd}t||d}tj|d<|S)z: Fixture with a Series with a 2-level MultiIndex. )rrrrrrrr)rrrrrrrrrr)rbrrrArrrNaN)ZarraysrrErZserr#r#r$series_with_multilevel_indexs    rcCs(i|] }|jdtjdd|qS)z-seriesrr)rprZmakeFloatSeriesr)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_objsrPrrQr#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_framesrcCs 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_framesrcCs ttS)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_frame1src Cs>tdddtjdgdddtjdgdddd tdd S) z Fixture for DataFrame of float/int/string columns with RangeIndex Columns are ['a', 'b', 'c', 'float32', 'int32']. ?rrrrrr=r)rrcrrr)rrArwaranger#r#r#r$mixed_type_frameNsrc Cstdddtdddtdddtdddtdddtdddtdddtdddtdddtdddg }ttjt||dS)zF Fixture for Series with a DatetimeIndex that has duplicates. ir=rrr<r)rrrArrr)datesr#r#r$(rand_series_with_duplicate_datetimeindex`s           r)r[r\r\g?g?z2012-01M)freqz period[M]z 2012-02-01Dz period[D]z 2011-01-01ri)secondsztimedelta64[ns]cCs|jSrmrOrQr#r#r$ea_scalar_and_dtypexsr)__add____radd____sub____rsub____mul____rmul__ __floordiv__ __rfloordiv__ __truediv__ __rtruediv____pow____rpow____mod____rmod__cCs|jS)zD Fixture for dunder names for common arithmetic operations. rOrQr#r#r$all_arithmetic_operatorssrcCs|jS)zU Fixture for operator and roperator arithmetic, comparison, and logical ops. rOrQr#r#r$all_binary_operatorss"rcCs|jS)z Fixture for operator and roperator arithmetic functions. Notes ----- This includes divmod and rdivmod, whereas all_arithmetic_operators does not. rOrQr#r#r$all_arithmetic_functionssr) rmaxminmeanprodZstdvarZmedianZkurtZskewcCs|jS)z. Fixture for numeric reduction names. rOrQr#r#r$all_numeric_reductionssrallanycCs|jS)z. Fixture for boolean reduction names. rOrQr#r#r$all_boolean_reductionssrcCs|jS)z> Fixture for all (boolean + numeric) reduction names. rOrQr#r#r$all_reductions srcCs|jS)z; Fixture for operator module comparison functions. rOrQr#r#r$ comparison_opsr)__le____lt____ge____gt__cCs|jS)zm Fixture for dunder names for compare operations except == and != * >= * > * < * <= rOrQr#r#r$compare_operators_no_eq_ne&s r )__and____rand____or____ror____xor____rxor__cCs|jS)zY Fixture for dunder names for common logical operations * | * & * ^ rOrQr#r#r$all_logical_operators3s r'cCs |dS)zO Returns the configuration for the test setting `--strict-data-files`. r!)r+)Z pytestconfigr#r#r$strict_data_filesDsr(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. testscsLtjjg|R}tj|sHr6td|dntd|d|S)NzCould not find file z and --strict-data-files is set.zCould not find .)ospathjoinexists ValueErrorr,r0)r}r,Z BASE_PATHr(r#r$decoas  zdatapath..deco)r+r,r-dirname__file__)r(r1r#r0r$datapathLs r4cCst|ddddS)z* The iris dataset as a DataFrame. iorcsvziris.csv)rBZread_csv)r4r#r#r$irisosr7UTCz Asia/Tokyozdateutil/US/Pacificzdateutil/Asia/Singaporez+01:15z-02:15z UTC+01:15z UTC-02:15i,i)hoursrrrcCsg|] }t|qSr#rG)rir#r#r$rrLcCs|jS)zD Fixture for trying timezones including default (None): {0} rOrQr#r#r$tz_naive_fixturesr;cCs|jS)z4 Fixture for trying explicit timezones: {0} rOrQr#r#r$tz_aware_fixturesr<r z dateutil/UTCcCs|jS)zQ Fixture to provide variants of UTC timezone strings and tzinfo objects. rOrQr#r#r$ utc_fixturesr=cCs|jS)zR Parametrized fixture for string dtypes. * str * 'str' * 'U' rOrQr#r#r$ string_dtypes r>pyarrowz1.0.0) min_versioncCs|jS)za Parametrized fixture for string dtypes. * 'string[python]' * 'string[pyarrow]' rOrQr#r#r$nullable_string_dtypesrApythoncCs|jS)zb Parametrized fixture for pd.options.mode.string_storage. * 'python' * 'pyarrow' rOrQr#r#r$string_storages rCcCs|jS)zK Parametrized fixture for bytes dtypes. * bytes * 'bytes' rOrQr#r#r$ bytes_dtypesrDcCs|jS)zN Parametrized fixture for object dtypes. * object * 'object' rOrQr#r#r$ object_dtypesrEobjectcCs|jS)zo Parametrized fixture for string dtypes. * 'object' * 'string[python]' * 'string[pyarrow]' rOrQr#r#r$any_string_dtypesrGcCs|jS)z\ Parametrized fixture for datetime64 dtypes. * 'datetime64[ns]' * 'M8[ns]' rOrQr#r#r$datetime64_dtypesrHcCs|jS)z^ Parametrized fixture for timedelta64 dtypes. * 'timedelta64[ns]' * 'm8[ns]' rOrQr#r#r$timedelta64_dtypesrIc CstddddddddS)z- Fixture emits fixed Timestamp.now() ir=r>r )rr?rhourminuter microsecond)rr#r#r#r$ fixed_now_ts&srOcCs|jS)z^ Parameterized fixture for float dtypes. * float * 'float32' * 'float64' rOrQr#r#r$float_numpy_dtype0s rPcCs|jS)zR Parameterized fixture for float dtypes. * 'Float32' * 'Float64' rOrQr#r#r$float_ea_dtype<srQcCs|jS)z~ Parameterized fixture for float dtypes. * float * 'float32' * 'float64' * 'Float32' * 'Float64' rOrQr#r#r$any_float_dtypeGs rRcCs|jS)zg Parameterized fixture for complex dtypes. * complex * 'complex64' * 'complex128' rOrQr#r#r$ complex_dtypeUs rScCs|jS)z| Parameterized fixture for signed integer dtypes. * int * 'int8' * 'int16' * 'int32' * 'int64' rOrQr#r#r$any_signed_int_numpy_dtypeas rTcCs|jS)zx Parameterized fixture for unsigned integer dtypes. * 'uint8' * 'uint16' * 'uint32' * 'uint64' rOrQr#r#r$any_unsigned_int_numpy_dtypeos rUcCs|jS)z Parameterized fixture for any integer dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' rOrQr#r#r$any_int_numpy_dtype|srVcCs|jS)z Parameterized fixture for any nullable integer dtype. * 'UInt8' * 'Int8' * 'UInt16' * 'Int16' * 'UInt32' * 'Int32' * 'UInt64' * 'Int64' rOrQr#r#r$any_int_ea_dtypesrWcCs|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' rOrQr#r#r$ any_int_dtypesrXcCs|jS)z Parameterized fixture for any nullable integer dtype and any float ea dtypes. * 'UInt8' * 'Int8' * 'UInt16' * 'Int16' * 'UInt32' * 'Int32' * 'UInt64' * 'Int64' * 'Float32' * 'Float64' rOrQr#r#r$any_numeric_ea_dtypesrYcCs|jS)z~ Parameterized fixture for any signed nullable integer dtype. * 'Int8' * 'Int16' * 'Int32' * 'Int64' rOrQr#r#r$any_signed_int_ea_dtypes rZcCs|jS)z Parameterized fixture for any (purely) real numeric dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' * float * 'float32' * 'float64' rOrQr#r#r$any_real_numpy_dtypesr[cCs|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' rOrQr#r#r$any_numpy_dtypes"r\rrrbytesacrz mixed-integerrmixedg@Zfloatingrintegerzmixed-integer-floatdecimal datetime64z 2013-01-01z 2018-01-01rrZ20180101riirrrrcCs |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 -------- >>> import pandas._libs.lib as 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)rPrArwrF)rRZinferred_dtyperr#r#r$any_skipna_inferred_dtype2s) rdcCs>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:)r2)r, importorskipZIPython.core.interactiveshellreZtraitlets.configrfZHistoryManagerZ hist_file)rerfrr#r#r$ipes   rh)ZbsrZcooZcscZcsrZdiaZdokZlilcCsddlm}t||jdS)z- Yields scipy sparse matrix classes. r)sparseZ_matrix)ZscipyrigetattrrP)rRrir#r#r$spmatrixxs rkcCs.g|]&}tttj|tjjrttj|qSr#) issubclassrjrBoffsetsZTick)ror#r#r$rscCs|jS)zN Fixture for Tick based datetime offsets available for a time series. rOrQr#r#r$ tick_classess rocCs|Srmr#rIr#r#r$rKrLcCs|jS)zk Simple fixture for testing keys in sorting methods. Tests None (no key) and the identity key. rOrQr#r#r$ sort_by_keysrpccsxtdddlm}ddlm}ddlm}Gddd|}|d|d d |V|ddd|j d<|j dS) Nfsspecr)register_implementation)MemoryFileSystem) _registrycs&eZdZdZdgZfddZZS)z fsspectest..TestMemoryFStestmemNcs(|dd|jd<tjfi|dS)Ntestr)poprvsuperr{)r|r~ __class__r#r$r{sz)fsspectest..TestMemoryFS.__init__)rprrprotocolrvr{ __classcell__r#r#ryr$ TestMemoryFSsr}ruT)clobber) r,rgrqrrZfsspec.implementations.memoryrsZfsspec.registryrtrwrvstoreclear)rrrsregistryr}r#r#r$ fsspectests      r)rNN)ZEgonZVenkmanN)NCC1701DrrcCs|jS)zP A 3-tuple of names, the first two for operands, the last for a result. rOrQr#r#r$rsrcCs|jS)zI Parametrize over __setitem__, loc.__setitem__, iloc.__setitem__ rOrQr#r#r$ indexer_slisrcCs|jS)z8 Parametrize over __setitem__, iloc.__setitem__ rOrQr#r#r$ indexer_sisrcCs|jS)z7 Parametrize over __setitem__, loc.__setitem__ rOrQr#r#r$ indexer_slsrcCs|jS)z: Parametrize over at.__setitem__, loc.__setitem__ rOrQr#r#r$ indexer_alsrcCs|jS)z< Parametrize over iat.__setitem__, iloc.__setitem__ rOrQr#r#r$ indexer_ialsrcCstjjjdkS)z> Fixture to check if the array manager is being used. rw)rBoptionsmodeZ data_managerrQr#r#r$using_array_managersr(__doc__ collectionsrrrrrrrbroperatorr+Z dateutil.tzr r Z hypothesisr stZnumpyrAr,Zpytzr r Zpandas.util._test_decoratorsutilZ_test_decoratorstdZpandas.core.dtypes.dtypesrrZpandasrBrrrrrrZpandas._testingZ_testingrZ pandas.corerZpandas.core.indexes.apirrr?pa ImportErrorZ has_pyarrowr-filterwarningsr.r%r5settingsZregister_profileZ HealthCheckZtoo_slowZ load_profilesplitrrjZtseriesrmclsZregister_type_strategyZbuildsZintegersZbooleansZfixturer*rDrSZ axis_framerTrUrXr]r^r_rPZ skip_if_nordrerfrirlZ NULL_OBJECTSrqZnulls_fixture2nanZNaTrrZunique_nulls_fixture2ZNP_NAT_OBJECTSrsZnp_nat_fixture2rtrvZindex_or_series2rwrxryZbox_with_array2rrrrrrrZmakeUnicodeIndexZmakeStringIndexZ makeDateIndexZmakePeriodIndexZmakeTimedeltaIndexZ makeIntIndexZ makeUIntIndexZmakeRangeIndexZmakeFloatIndexZmakeNumericIndexZ makeBoolIndexZmakeCategoricalIndexZmakeIntervalIndexrrbrrrridxkeysrEZindex_fixture2rZ index_flat2rrrrrrr1Z_seriesrrZNARROW_NP_DTYPESZ_narrow_seriesrrrrrrrrZ_all_arithmetic_operatorsraddZraddsubZrsubmulZrmultruedivZrtruedivfloordivZ rfloordivmodZrmodpowZrpoweqneltlegtgeand_Zrand_xorZrxoror_Zror_rrZ_all_numeric_reductionsrZ_all_boolean_reductionsrZ_all_reductionsrrr r'r(r4r7Z TIMEZONESZ TIMEZONE_IDSZparametrize_fixture_docstrr;r<Ztz_aware_fixture2r=Z utc_fixture2Z STRING_DTYPESr>rArCZstring_storage2Z BYTES_DTYPESrDZ OBJECT_DTYPESrErGZDATETIME64_DTYPESrHZTIMEDELTA64_DTYPESrIrOZFLOAT_NUMPY_DTYPESrPZFLOAT_EA_DTYPESrQrRZCOMPLEX_DTYPESrSZSIGNED_INT_NUMPY_DTYPESrTZUNSIGNED_INT_NUMPY_DTYPESrUZALL_INT_NUMPY_DTYPESrVZALL_INT_EA_DTYPESrWrXrYZSIGNED_INT_EA_DTYPESrZZALL_REAL_NUMPY_DTYPESr[ZALL_NUMPY_DTYPESr\ZNArcZ_any_skipna_inferred_dtyperN_rdrhrk__all__rorprrsetitemlocZilocrrratrZiatrrr#r#r#r$s         /                                                 #                                 "                             & $  2