B ̵aF@sdZddlmZddlmZmZmZmZmZddlm 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.ej/0d Z1ddZ2ddZ3ddZ4ej5j6ddej7j8fdej59dxDd:D]8Z;ee;Z?e@e?ejAe?eBddeCdqPWxPd:D]DZ;ee;Z?e@e?ejAe?eBddeCejBddd d!qWxPd":D]DZ;ee;Z?e@e?ejAe?eBd#d$eCejBddd d%qWejDd&d'd(d)ZEejDd&d'd*d+ZFejDddd,d-gd.d/d0d1d2ZGeGZHejDd&d3dgd4d5d6ZIejDd&d3dgd4d7d8ZJejDd9d:d3gd4d;d<ZKejDd=d>d?d@gd4dAdBZLejDd=d>d?d@gd4dCdDZMejDddEdFdGdHgd4dIdJZNejDdEdFdGdHgd4dKdLZOejDd&d3gd4dMdNZPejDdOdPd=d>gd4dQdRZQejDdSdTgd4dUdVZRejDe)jSdWd/d0dXdYZTeTZUejDdejVe jWgd4dZd[ZXeXZYejDe!e$gd4d\d]ZZejDe-e$gd,d^gd0d_d`Z[e[Z\ejDe-e$e j]gd,d^dagd0dbdcZ^ejDdddeZ_ejDdfdgZ`ejDdhdiZaejDdjdkZbdldmZcdndoZde)edpe)fdpe)gdpe)jgdpdqdre)hdpe)idpe)jdpe)kdpe)ldpe)mdpe)ndse)odpe)pdpe-ge.qerdtdudvgddwdxgedece-dddddwdwgdyZsejDestd4dzd,ZueuZvejDd{d|esDd4d}d~ZwewZxejDdd|esDd4ddZyejDddZzejDddZ{ejDddZ|ejDddZ}ddZ~ddesDZejDddZejDddZejejejejejejejejgZddeDZejDetd4ddZeseeZejDetd4ddZejDddZejDddZejDddZejDddZejDddZejDddZejDe"dddedd>fe"dddedd>fe#ddddfe#ddddfe&dddreddrfe%dddfgd4ddZddddddddddddddgZejDed4ddĄZejDe 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+jgd4ddƄZejDe je+je je+je je+je je+je je+je je+je je+jgd4ddȄZddddddddddg ZejDed4ddԄZddgZejDed4dd؄ZeeZejDed4ddڄZejDddddddgd4ddZejDddddgd4ddZejDddddddgd4ddZejDddZejDddZejDddZddddddddddee edededejeeddeedddtdgZdd|eDZeeeƒejDeed0ddZeeeddejDeddeddd0ddZeZejDddeeejgd4ddZeZejDe)jd4d d ZejDd ej̐d ej͐d dddgd4ddZejDdej̐d ej͐d dddgd4ddZeZejDe)jd4ddZejDe)jd4ddZejDdd ej̐d ej͐d dddgd4ddZejDe)jd4ddZejDe)jd4dd ZejDe)jd4d!d"ZejDe)jd4d#d$ZejDe)je)jd4d%d&ZejDe)jd4d'd(ZejDe)jd4d)d*ZejDe)jd4d+d,ZejDe)jd4d-d.ZejDe)jd4d/d0ZejDe)je)jd4d1d2ZejDe)je)jd4d3d4ZejDe)jd4d5d6ZejDe)jd4d7d8ZejDe)jd4d9d:Zd;dd?ejVd@gfdAejVejVejVgfdAgfdBdjDd4dfdgZejDddhd/gd4didjZeDdkdlZejDdmdndogd4dpdqZejDe)je)je)jgd4drdsZejDe)je)jgd4dtduZejDe)je)jgd4dvdwZejDe)je)jgd4dxdyZejDdzd{ZdS(|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 )abc)datedatetimetime timedeltatimezone)DecimalN)tzlocaltzutc) strategies) FixedOffsetutc)DatetimeTZDtype IntervalDtype) DataFrameIntervalPeriodSeries Timedelta Timestamp)ops)Index MultiIndexz3ignore:Promotion of numbers and bools:FutureWarningcCsd|jdddd|jdddd|jdddd|jd dd d|jd dd d|jd ddddS)Nz --skip-slow store_truezskip slow tests)actionhelpz--skip-networkzskip network testsz --skip-dbz skip db testsz--run-high-memoryzrun high memory testsz --only-slowzrun only slow testsz--strict-data-filesz0Fail if a test is skipped for missing data file.)Z addoption)parserr.lib/python3.7/site-packages/pandas/conftest.pypytest_addoptionPs rcCsd|jkr |jdr tdd|jkr@|jdr@tdd|jkr`|jdr`tdd |jkr|jd rtd d |jkr|jd stddS)NZslowz --skip-slowzskipping due to --skip-slowz --only-slowzskipping due to --only-slowZnetworkz--skip-networkzskipping due to --skip-networkZdbz --skip-dbzskipping due to --skip-dbZ high_memoryz--run-high-memoryz=skipping high memory test since --run-high-memory was not set)keywordsconfig getoptionpytestskip)itemrrrpytest_runtest_setup_s    r&cCs4x.|D]&}d|jkr"|tjj|tqWdS)Nz/frame/)ZnodeidZ add_markerr#markZ arraymanager suppress_npdev_promotion_warning)itemsr%rrrpytest_collection_modifyitemsrs  r*Zcii)ZdeadlineZsuppress_health_checkz)MonthBegin MonthEnd BMonthBegin BMonthEndic)n normalizez%YearBegin YearEnd BYearBegin BYearEnd )Z min_valueZ max_value)r,r-monthz1QuarterBegin QuarterEnd BQuarterBegin BQuarterEndi)r,r-Z startingMonthT)ZautousecCstdddS)z< Configure settings for all tests and test modules. Zchained_assignmentraiseN)pdZ set_optionrrrrconfigure_testssr6cCst|d<t|d<dS)z: Make `np` and `pd` names available for doctests. npr5N)r7r5)Zdoctest_namespacerrr add_importssr8indexcolumnscCsdt|S)Nzaxis=)repr)xrrrr=)paramsidscCs|jS)z@ Fixture for returning the axis numbers of a DataFrame. )param)requestrrraxissrCF)r?cCs|jS)a Pass in the observed keyword to groupby for [True, False] This indicates whether categoricals should return values for values which are not in the grouper [False / None], or only values which appear in the grouper [True]. [None] is supported for future compatibility if we decide to change the default (and would need to warn if this parameter is not passed). )rA)rBrrrobserveds rDcCs|jS)z6 Boolean 'ordered' parameter for Categorical. )rA)rBrrrorderedsrEfirstZlastcCs|jS)z[ Valid values for the 'keep' parameter used in .duplicated or .drop_duplicates )rA)rBrrrkeepsrGleftrightZbothZneithercCs|jS)z< Fixture for trying all interval closed parameters. )rA)rBrrrclosedsrJcCs|jS)zS Secondary closed fixture to allow parametrizing over all pairs of closed. )rA)rBrrr other_closedsrKZgzipbz2zipZxzcCs|jS)zK Fixture for trying common compression types in compression tests. )rA)rBrrr compressionsrNcCs|jS)zk Fixture for trying common compression types in compression tests excluding uncompressed case. )rA)rBrrrcompression_onlysrOcCs|jS)z, Fixture that an array is writable. )rA)rBrrrwritable srPinnerZoutercCs|jS)z: Fixture for trying all types of join operations. )rA)rBrrr join_typesrRnlargest nsmallestcCs|jS)z1 Fixture for trying all nselect methods. )rA)rBrrrnselect_methodsrUcCs t|jS)N)type__name__)r<rrrr=%r>cCs|jS)z/ Fixture for each null type in pandas. )rA)rBrrr nulls_fixture%srXcCs|jS)zL Fixture for each null type in pandas, each null type exactly once. )rA)rBrrrunique_nulls_fixture0srYcCs|jS)z; Fixture to parametrize over DataFrame and Series. )rA)rBrrrframe_or_series@srZseriescCs|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 )rA)rBrrrindex_or_seriesIs r\arraycCs|jS)zG Fixture to parametrize over Index, Series, and ExtensionArray )rA)rBrrrindex_or_series_or_array\sr^cCsGdddt}|S)z, Fixture for a dictionary subclass. c@seZdZddZdS)z"dict_subclass..TestSubDictc_stj|f||dS)N)dict__init__)selfargskwargsrrrr`ksz+dict_subclass..TestSubDict.__init__N)rW __module__ __qualname__r`rrrr TestSubDictjsrf)r_)rfrrr dict_subclassdsrgcCsGdddtj}|S)z8 Fixture for a non-mapping dictionary subclass. c@s,eZdZddZddZddZddZd S) z5non_dict_mapping_subclass..TestNonDictMappingcSs ||_dS)N)_data)raZunderlying_dictrrrr`xsz>non_dict_mapping_subclass..TestNonDictMapping.__init__cSs |j|S)N)rh __getitem__)rakeyrrrri{szAnon_dict_mapping_subclass..TestNonDictMapping.__getitem__cSs |jS)N)rh__iter__)rarrrrk~sz>non_dict_mapping_subclass..TestNonDictMapping.__iter__cSs |jS)N)rh__len__)rarrrrlsz=non_dict_mapping_subclass..TestNonDictMapping.__len__N)rWrdrer`rirkrlrrrrTestNonDictMappingwsrm)rMapping)rmrrrnon_dict_mapping_subclassqs rocCs`td}|ddddddg}|jdd|jjD|_|jjdd d gd d |S) z DataFrame with 3 level MultiIndex (year, month, day) covering first 100 business days from 2000-01-01 with random data dcSs|jS)N)year)r<rrrr=r>zAmultiindex_year_month_day_dataframe_random_data..cSs|jS)N)r2)r<rrrr=r>cSs|jS)N)day)r<rrrr=r>cSsg|]}|dqS)Zi8)astype).0Zlevrrr szCmultiindex_year_month_day_dataframe_random_data..rqr2rrT)Zinplace)tmZmakeTimeDataFramegroupbysumr9Z set_levelslevelsZ set_names)ZtdfZymdrrr/multiindex_year_month_day_dataframe_random_datas   rzc Csvtddddgdddggdddd d d d d d d g dd d dd d d dd d g gd d gd}ttjdd |tdddgdddS)z2DataFrame with 2 level MultiIndex with random datafoobarbazquxonetwothreerr0rFsecond)rycodesnames ABCZexp)name)r9r:)rrr7randomrandnr)r9rrr multiindex_dataframe_random_datas . rcCshtddddg}tddg}tdddd d d g}tddddddg}d d g}t||g||g|d dS)zJ MultiIndex used to test the general functionality of this object r{r|r}r~rrrr0rrrFrF)ryrrZverify_integrity)rr7r]r)Z major_axisZ minor_axisZ major_codesZ minor_codesZ index_namesrrr_create_multiindexs rcCs.tjddgddgtjddddgd d d gd S) zB MultiIndex with a level that is a tzaware DatetimeIndex. r0rab20130101rz US/Eastern)Zperiodstzrrr)r)rZ from_productr5Z date_rangerrrr_create_mi_with_dt64tz_levelsrrpz US/Pacific)rrr{r|r}rr)Zunicodestringrz datetime-tzperiodrintuintrangefloatboolZ categoricalintervalemptytupleszmi-with-dt64tz-levelmultirepeatscCst|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_dictrAcopy)rBrrrr9s cCsg|]}tt|ts|qSr) isinstancerr)rtrjrrrrusrucCs|j}t|S)z8 index fixture, but excluding MultiIndex cases. )rArr)rBrjrrr index_flats rcCs&g|]}|dkrtt|ts|qS))rrrrr)rrr)rtrjrrrruscCst|jjdd}|j}|jdkrd|}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. T)deep)rzmi-with-dt64tz-levelr)Nrr0N)rrArvaluestolistr from_tuplesrV)rBZindZvalsrrrindex_with_missings  rcCstggtjdS)N)r9dtype)rr7float64rrrr empty_series#srcCst}d|_|S)zC Fixture for Series of floats with Index of unique strings r[)rvZmakeStringSeriesr)srrr string_series(srcCst}d|_|S)zI Fixture for Series of dtype object with Index of unique strings Zobjects)rvZmakeObjectSeriesr)rrrr object_series2srcCst}d|_|S)z9 Fixture for Series of floats with DatetimeIndex Zts)rvZmakeTimeSeriesr)rrrrdatetime_series<srcCs"t|}tj|}t||ddS)zHelper for the _series dictr)r9r)lenr7rrr)r9sizedatarrr_create_seriesFs rcCs"i|]\}}t|d|dqS)z series-with-z-index)r)rtZindex_idr9rrr MsrcCst|S)zE Fixture for tests on series with changing types of indices. )r)r9rrrseries_with_simple_indexSsrc Cs`ddddddddgddddddddgg}t|}t|}tjd}t||d}tj|d <|S) z: Fixture with a Series with a 2-level MultiIndex. r|r}r~r{rr)r9r)rMrrr7rrrZNaN)Zarraysrr9rZserrrrseries_with_multilevel_index[s    rcCs(i|] }tjdd||jdqS)r)rz-series)rvZmakeFloatSeriesrsrW)rtrrrrrvscCst|jS)z: Fixture for Series with low precision data types )_narrow_seriesrAr)rBrrr narrow_series|srcCst|jjddS)z| Fixture for tests on indexes, series and series with a narrow dtype copy to avoid mutation, e.g. setting .name T)r)_index_or_series_objsrAr)rBrrrindex_or_series_objsrcCstS)N)rrrrr empty_framesrcCsttdS)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] int64)rrv getSeriesDatarsrrrr 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] )rrvZgetTimeSeriesDatarrrrdatetime_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] )rrvrrrrr float_framesrc Cs>tdddtjdgdddtjdgdddd tdd S) z Fixture for DataFrame of float/int/string columns with RangeIndex Columns are ['a', 'b', 'c', 'float32', 'int32']. g?rr{rfloat32)rr0int32)rrcrr)r9)rr7r]Zarangerrrrmixed_type_framesrc Cstdddtdddtdddtdddtdddtdddtdddtdddtdddtdddg }ttjt||dS)zF Fixture for Series with a DatetimeIndex that has duplicates. ir0rrr/)r9)rrr7rrr)Zdatesrrr(rand_series_with_duplicate_datetimeindexs         r)rHrIrg?g?rz2012-01M)Zfreqz period[M]z 2012-02-01Dz period[D]z 2011-01-01z US/Eastern)Zsecondsztimedelta64[ns]cCs|jS)N)rA)rBrrrea_scalar_and_dtypesr__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. )rA)rBrrrall_arithmetic_operators@srcCs|jS)zU Fixture for operator and roperator arithmetic, comparison, and logical ops. )rA)rBrrrall_binary_operatorsHs"rcCs|jS)z Fixture for operator and roperator arithmetic functions. Notes ----- This includes divmod and rdivmod, whereas all_arithmetic_operators does not. )rA)rBrrrall_arithmetic_functionsmsrrxmaxminZmeanZprodZstdvarZmedianZkurtZskewcCs|jS)z. Fixture for numeric reduction names. )rA)rBrrrall_numeric_reductionssrallanycCs|jS)z. Fixture for boolean reduction names. )rA)rBrrrall_boolean_reductionssrcCs|jS)z> Fixture for all (boolean + numeric) reduction names. )rA)rBrrrall_reductionssr__eq____ne____le____lt____ge____gt__cCs|jS)zu Fixture for dunder names for common compare operations * >= * > * == * != * < * <= )rA)rBrrrall_compare_operatorss rcCs|jS)zm Fixture for dunder names for compare operations except == and != * >= * > * < * <= )rA)rBrrrcompare_operators_no_eq_nes r__and____rand____or____ror____xor____rxor__cCs|jS)zY Fixture for dunder names for common logical operations * | * & * ^ )rA)rBrrrall_logical_operatorss rcCs |dS)zO Returns the configuration for the test setting `--strict-data-files`. z--strict-data-files)r")Z pytestconfigrrrstrict_data_filessrcs(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. ZtestscsJtjjf|}tj|sFr4td|dntd|d|S)NzCould not find file z and --strict-data-files is set.zCould not find .)ospathjoinexists ValueErrorr#r$)rbr) BASE_PATHrrrdecos zdatapath..deco)rrrdirname__file__)rrr)rrrdatapaths rcCst|ddddS)z* The iris dataset as a DataFrame. iorZcsvziris.csv)r5Zread_csv)rrrririssrZUTCz Asia/Tokyozdateutil/US/Pacificzdateutil/Asia/Singaporez+01:15z-02:15z UTC+01:15z UTC-02:15i,i)Zhoursr)rcCsg|] }t|qSr)r;)rtirrrru.scCs|jS)zD Fixture for trying timezones including default (None): {0} )rA)rBrrrtz_naive_fixture1srcCs|jS)z4 Fixture for trying explicit timezones: {0} )rA)rBrrrtz_aware_fixture:srr z dateutil/UTCcCs|jS)zQ Fixture to provide variants of UTC timezone strings and tzinfo objects. )rA)rBrrr utc_fixtureGsrcCs|jS)zR Parametrized fixture for string dtypes. * str * 'str' * 'U' )rA)rBrrr string_dtypeUs rzstring[python]zstring[pyarrow]Zpyarrowz1.0.0)Z min_version)ZmarkscCs|jS)za Parametrized fixture for string dtypes. * 'string[python]' * 'string[pyarrow]' )rA)rBrrrnullable_string_dtypeasrZpythoncCs|jS)zb Parametrized fixture for pd.options.mode.string_storage. * 'python' * 'pyarrow' )rA)rBrrrstring_storagess r cCs|jS)zK Parametrized fixture for bytes dtypes. * bytes * 'bytes' )rA)rBrrr bytes_dtypesr cCs|jS)zN Parametrized fixture for object dtypes. * object * 'object' )rA)rBrrr object_dtypesr objectcCs|jS)zo Parametrized fixture for string dtypes. * 'object' * 'string[python]' * 'string[pyarrow]' )rA)rBrrrany_string_dtypesr cCs|jS)z\ Parametrized fixture for datetime64 dtypes. * 'datetime64[ns]' * 'M8[ns]' )rA)rBrrrdatetime64_dtypesrcCs|jS)z^ Parametrized fixture for timedelta64 dtypes. * 'timedelta64[ns]' * 'm8[ns]' )rA)rBrrrtimedelta64_dtypesrcCs|jS)z^ Parameterized fixture for float dtypes. * float * 'float32' * 'float64' )rA)rBrrr float_dtypes rcCs|jS)zR Parameterized fixture for float dtypes. * 'Float32' * 'Float64' )rA)rBrrrfloat_ea_dtypesrcCs|jS)z~ Parameterized fixture for float dtypes. * float * 'float32' * 'float64' * 'Float32' * 'Float64' )rA)rBrrr any_float_allowed_nullable_dtypes rcCs|jS)zg Parameterized fixture for complex dtypes. * complex * 'complex64' * 'complex128' )rA)rBrrr complex_dtypes rcCs|jS)z| Parameterized fixture for signed integer dtypes. * int * 'int8' * 'int16' * 'int32' * 'int64' )rA)rBrrr sint_dtypes rcCs|jS)zx Parameterized fixture for unsigned integer dtypes. * 'uint8' * 'uint16' * 'uint32' * 'uint64' )rA)rBrrr uint_dtypes rcCs|jS)z Parameterized fixture for any integer dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' )rA)rBrrr any_int_dtypesrcCs|jS)z Parameterized fixture for any nullable integer dtype. * 'UInt8' * 'Int8' * 'UInt16' * 'Int16' * 'UInt32' * 'Int32' * 'UInt64' * 'Int64' )rA)rBrrrany_nullable_int_dtype$srcCs|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' )rA)rBrrrany_int_or_nullable_int_dtype5srcCs|jS)z Parameterized fixture for any nullable integer dtype and any float ea dtypes. * 'UInt8' * 'Int8' * 'UInt16' * 'Int16' * 'UInt32' * 'Int32' * 'UInt64' * 'Int64' * 'Float32' * 'Float64' )rA)rBrrrany_nullable_numeric_dtypeOsrcCs|jS)z~ Parameterized fixture for any signed nullable integer dtype. * 'Int8' * 'Int16' * 'Int32' * 'Int64' )rA)rBrrrany_signed_nullable_int_dtypecs rcCs|jS)z Parameterized fixture for any (purely) real numeric dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' * float * 'float32' * 'float64' )rA)rBrrrany_real_dtypepsrcCs|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' )rA)rBrrrany_numpy_dtypes"rrrrbytesacrz mixed-integerZmixedg@Zfloatingg?Zintegerzmixed-integer-floatdecimalZboolean 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)rAr7r]r )rBZinferred_dtyperrrrany_skipna_inferred_dtypes) r"cCs>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:)r!)r# importorskipZIPython.core.interactiveshellr#Ztraitlets.configr$ZHistoryManagerZ hist_file)r#r$rrrrips   r&ZbsrZcooZcscZcsrZdiaZdokZlilcCsddlm}t||jdS)z- Yields scipy sparse matrix classes. r)sparseZ_matrix)Zscipyr'getattrrA)rBr'rrrspmatrixs r)cCs.g|]&}tttj|tjjrttj|qSr) issubclassr(r5offsetsZTick)rtorrrruscCs|jS)zN Fixture for Tick based datetime offsets available for a time series. )rA)rBrrr tick_classess r-cCs|S)Nr)r<rrrr=&r>cCs|jS)zk Simple fixture for testing keys in sorting methods. Tests None (no key) and the identity key. )rA)rBrrr sort_by_key&sr.ccsxtdddlm}ddlm}ddlm}Gddd|}|d|d d |V|ddd|j d<|j dS) Nfsspecr)register_implementation)MemoryFileSystem) _registrycs&eZdZdZdgZfddZZS)z fsspectest..TestMemoryFStestmemNcs$|dd|jd<tjf|dS)Ntestr)popr4superr`)rarc) __class__rrr`:sz)fsspectest..TestMemoryFS.__init__)rWrdreZprotocolr4r` __classcell__rr)r7r TestMemoryFS6sr9r3T)Zclobber) r#r%r/r0Zfsspec.implementations.memoryr1Zfsspec.registryr2r5r4Zstoreclear)r0r1registryr9rrr fsspectest/s      r<)r{NN)ZEgonZVenkmanN)NCC1701Dr=r=cCs|jS)zP A 3-tuple of names, the first two for operands, the last for a result. )rA)rBrrrrEs rcCs|jS)zI Parametrize over __setitem__, loc.__setitem__, iloc.__setitem__ )rA)rBrrr indexer_sliSsr>cCs|jS)z8 Parametrize over __setitem__, iloc.__setitem__ )rA)rBrrr indexer_si[sr?cCs|jS)z7 Parametrize over __setitem__, loc.__setitem__ )rA)rBrrr indexer_slcsr@cCs|jS)z: Parametrize over at.__setitem__, loc.__setitem__ )rA)rBrrr indexer_alksrAcCstjjjdkS)z> Fixture to check if the array manager is being used. r])r5ZoptionsmodeZ data_manager)rBrrrusing_array_managerssrC(__doc__ collectionsrrrrrrr roperatorrZ dateutil.tzr r Z hypothesisr stZnumpyr7r#Zpytzr r Zpandas.util._test_decoratorsutilZ_test_decoratorsZtdZpandas.core.dtypes.dtypesrrZpandasr5rrrrrrZpandas._testingZ_testingrvZ pandas.corerZpandas.core.indexes.apirrr'filterwarningsr(rr&r*ZsettingsZregister_profileZ HealthCheckZtoo_slowZ load_profilesplitrr(Ztseriesr+clsZregister_type_strategyZbuildsZintegersZbooleansZfixturer6r8rCZ axis_framerDrErGrJrKrNrOrPrRrUZ NULL_OBJECTSrXZnulls_fixture2nanZNaTrYZunique_nulls_fixture2rZr\Zindex_or_series2r]r^rgrorzrrrZmakeUnicodeIndexZmakeStringIndexZ makeDateIndexZmakePeriodIndexZmakeTimedeltaIndexZ makeIntIndexZ makeUIntIndexZmakeRangeIndexZmakeFloatIndexZ makeBoolIndexZmakeCategoricalIndexZmakeIntervalIndexrrMrkeysr9Zindex_fixture2rZ index_flat2rrrrrrr)Z_seriesrrZfloat16rZint8Zint16rZuint8Zuint16Zuint32Z_narrow_dtypesrrrrrrrrrrrZ_all_arithmetic_operatorsraddZraddsubZrsubmulZrmultruedivZrtruedivfloordivZ rfloordivmodZrmodpowZrpoweqneltlegtgeand_Zrand_xorZrxoror_Zror_rrZ_all_numeric_reductionsrZ_all_boolean_reductionsrZ_all_reductionsrrrrrrrZ TIMEZONESZ TIMEZONE_IDSZparametrize_fixture_docstrrrZtz_aware_fixture2rZ utc_fixture2Z STRING_DTYPESrrAZ skip_if_norr Zstring_storage2Z BYTES_DTYPESr Z OBJECT_DTYPESr r ZDATETIME64_DTYPESrZTIMEDELTA64_DTYPESrZ FLOAT_DTYPESrZFLOAT_EA_DTYPESrrZCOMPLEX_DTYPESrZSIGNED_INT_DTYPESrZUNSIGNED_INT_DTYPESrZALL_INT_DTYPESrZALL_EA_INT_DTYPESrrrZSIGNED_EA_INT_DTYPESrZALL_REAL_DTYPESrZALL_NUMPY_DTYPESrZNAr!Z_any_skipna_inferred_dtyper@_r"r&r)__all__r-r.r<rsetitemZlocZilocr>r?r@ZatrArCrrrrsv         &   "      $     "            #   , * 04  0         ' $  34  "  ($"$