U md$@s2dZddlmZddlmZddlmZmZmZmZm 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 l1m2Z2ddl3m4Z4m5Z5z ddl6Z7Wne8k rHdZ9YnX[7dZ9dZ:e-j;rddl:Z:ze:dddddddZ?ddddZ@ejAjBddejCjDfdejAEddFD]8ZGeHe&jIjJeGZKeLeKejMeKeNd d!eOd"qd#FD]DZGeHe&jIjJeGZKeLeKejMeKeNd$d%eOejNd&d'd(d)q.d*FD]DZGeHe&jIjJeGZKeLeKejMeKeNd+d,eOejNd&d'd(d-q|ejPddd.d/ZQejPdd0ddd1d2ZRejPdd&d3d4gd5d6d7d8d9ZSeSZTejPd&d4gd:d6d7d;d<ZUejPdddgd=d>d?ZVejPdddgd=d@dAZWejPddgd=dBdCZXejPdDdEdgd=dFdGZYejPdHdIdJdKgd=dLdMZZejPdJdKdHdIgd=dNdOZ[ejPdJdKdHdIgd=dPdQZ\ejPddRdSdTdUdVej]dWe!^dXdYgd=dZd[Z_ejPdRdSdTdUdVej]dWe!^dXdYgd=d\d]Z`ejPddgd=d^d_ZaejPd`dadJdKgd=dbdcZbejPdddegd=dfdgZcejPe0jddhd6d7didjZeeeZfejPdejge&jhgd=dkdlZieiZjejPe0jkdmd6d7dndoZlelZmejPe'e*gd=dpdqZnejPe4e*gd3drgd7dsdtZoeoZpejPe4e*e&jqgd3drdugd7dvdwZrejPe4e*e'e&jqgdxd6d7dydzZsesZtejPd{d|ZuejPd}d~ZvejPddZwejPddddZxejPddddZyddZzddZ{e0|de0}de0j}ddde0~de0de0de0jddde0jddde0jddde0jddde0jddde0jddde0jddde0jddde0jddde0jddde0dee4ejddke0jdddde0jdddde0de0de4ge5edddgd&ddge{eze4ddd&d&ddge4eddde4eddde4eddde4ededde4e&jqe0|dddd Ze9re4e&jqe0|dddZeed<ejPed=dd3ZeZejPddeDd=ddZeZejPddeDd=ddZejPddddZejPddddZejPddddZddZddeDZejPddddZejPddddZdde0jDZeeeZejPed=dd„ZejPddddĄZejPddddƄZejPddddȄZejPddddʄZejPdddd̄ZejPe(dd%d͍e$ddKfe(ddd͍e$ddKfe)dddҍdfe)dddҍdfe,ddde#ddfe+ddڍdfgd=dd݄ZddddddddddddddgZejPed=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=ddZddddddddddddg ZejPed=ddZɐddgZejPed=ddZeeZejPed=ddZejPe je je je je je jgd=ddZejPdd d d gd=d d ZejPddddddgd=ddZАddddgZejPed=ddZejPddZejPdddd d!ZejPddd"d#Zdddؐd$d%d&d'd(d)d*eeed+eded,e je ed&d-e ed.d-dd/gZe:dk re֠e:ej]d;e!^d;dYgd=d?d@ZejPdAej]d;e!^d;dYgd=dBdCZeZejPe0jd=dDdEZejPe0jd=dFdGZejPdHdej]de!^d;dYgd=dIdJZejPe0jd=dKdLZejPe0jd=dMdNZejPdOddPdQZejPe0jd=dRdSZejPe0jd=dTdUZejPe0jd=dVdWZejPe0jd=dXdYZejPe0jd=dZd[ZejPe0jd=d\d]ZejPe0jd=d^d_ZejPe0jd=d`daZejPe0jd=dbdcZejPe0je0jd=dddeZejPe0je0je0je0jd=dfdgZejPe0jd=dhdiZ ejPe0j d=djdkZ ejPe0j d=dldmZ ejPe0jd=dndoZejPe0jd=dpdqZdrdsejgdtgfdrdse&jdtgfdudse&jhdtgfdvdwejgdxgfdyejgejgejggfdygfdzdsejgdgfdudsejgd{gfd|d}ejgd{gfd~d&ejgdgfdd&ejgd{gfde d&ejge dgfddejgdgfdde&jdgfdedejgedgfde,dejge,dgfdedd&d&ejgedd&d&gfded&ejgedgfded&ejgedgfde)de&jhe)dgfde(dd&ejge(ddgfgZee\ZZejPeed7ddZejPddZejPdddddddgd=ddZejPdde&jJjDd=ddZejPddd6gd=ddZePddZejPdddejgejgejgfejge&jhdfejge&jdfe&je&je&jfgd=ddddZejPe0je0j e0j!gd=ddZ"ejPe0j e0j!gd=ddZ#ejPe0je0j!gd=ddZ$ejPe0je0j gd=ddZ%ejPe0j&e0j gd=ddZ'ejPe0j(e0j!gd=ddZ)ejPddddZ*ejPddddZ+ddgZ,e:dk r e,e:<dejPe,d=ddddZ-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 MultiIndexFTUTCNonereturncCsd|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+H/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/pandas/conftest.pypytest_addoptionhsr-z pytest.Itemstr)itempathmessager!cCs(|j|r$|tjd|dS)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/r0r1r+r+r,ignore_doctest_warningws r8cCs.|d}|d}|d}|d}tjjd|dftjjd|dftjjd|dfg}d g}|D]}|d s||jd d d r|tjd|D]\} } t|| | qd|j kr|tjj |D]B\} } } }| |j kr| r|tj d|q|| q|r`d|j kr`|tj dq`dS)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/r0r1r6ZkwdZ skip_if_foundZarg_namer+r+r,pytest_collection_modifyitemss8         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=sr=)ZautousecCstdddS)z< Configure settings for all tests and test modules. Zchained_assignmentraiseN)rQZ set_optionr+r+r+r,configure_testssrSindexcolumnscCsdt|SNzaxis=reprxr+r+r,r[)paramsidscCs|jS)z@ Fixture for returning the axis numbers of a DataFrame. paramrequestr+r+r,axissrccCsdt|SrVrWrYr+r+r,r[ r\cCs|jS)zA Fixture for returning aliases of axis 1 of a DataFrame. r_rar+r+r,axis_1 srd)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). r_rar+r+r,observeds recCs|jS)z6 Boolean 'ordered' parameter for Categorical. r_rar+r+r,ordered srfcCs|jS)z% Boolean 'skipna' parameter. r_rar+r+r,skipna(srgfirstlastcCs|jS)z[ Valid values for the 'keep' parameter used in .duplicated or .drop_duplicates r_rar+r+r,keep0srjZbothZneitherleftrightcCs|jS)zA Fixture for trying all interval 'inclusive' parameters. r_rar+r+r,inclusive_endpoints_fixture9srmcCs|jS)z< Fixture for trying all interval closed parameters. r_rar+r+r,closedAsrncCs|jS)zS Secondary closed fixture to allow parametrizing over all pairs of closed. r_rar+r+r, other_closedIsrogzipbz2zipxztarZzstdZ zstandard)rCcCs|jS)zK Fixture for trying common compression types in compression tests. r_rar+r+r, compressionQsrucCs|jS)zk Fixture for trying common compression types in compression tests excluding uncompressed case. r_rar+r+r,compression_onlycsrvcCs|jS)z, Fixture that an array is writable. r_rar+r+r,writableusrwinneroutercCs|jS)z: Fixture for trying all types of join operations. r_rar+r+r, join_type}srznlargest nsmallestcCs|jS)z1 Fixture for trying all nselect methods. r_rar+r+r,nselect_methodsr}cCs t|jSNtype__name__rYr+r+r,r[r\cCs|jS)z/ Fixture for each null type in pandas. r_rar+r+r, nulls_fixturesrcCs|jS)zL Fixture for each null type in pandas, each null type exactly once. r_rar+r+r,unique_nulls_fixturesrcCs t|jSr~rrYr+r+r,r[r\cCs|jS)z- Fixture for each NaT type in numpy. r_rar+r+r,np_nat_fixturesrcCs|jS)z; Fixture to parametrize over DataFrame and Series. r_rar+r+r,frame_or_seriessrseriescCs|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 r_rar+r+r,index_or_seriess rarraycCs|jS)zG Fixture to parametrize over Index, Series, and ExtensionArray r_rar+r+r,index_or_series_or_arraysrcCs|jSr~)rrYr+r+r,r[r\cCs|jS)z] Fixture to test behavior for Index, Series, DataFrame, and pandas Array classes r_rar+r+r,box_with_arraysrcCsGdddt}|S)z, Fixture for a dictionary subclass. c@seZdZddddZdS)z"dict_subclass..TestSubDictrr c_stj|f||dSr~)dict__init__)selfargskwargsr+r+r,rsz+dict_subclass..TestSubDict.__init__N)r __module__ __qualname__rr+r+r+r, TestSubDictsr)r)rr+r+r, dict_subclasssrcCsGdddtj}|S)z8 Fixture for a non-mapping dictionary subclass. c@s>eZdZddddZddZdddd Zd dd d Zd S)z5non_dict_mapping_subclass..TestNonDictMappingrr cSs ||_dSr~)_data)rZunderlying_dictr+r+r,rsz>non_dict_mapping_subclass..TestNonDictMapping.__init__cSs |j|Sr~)r __getitem__)rkeyr+r+r,rszAnon_dict_mapping_subclass..TestNonDictMapping.__getitem__r cSs |jSr~)r__iter__rr+r+r,rsz>non_dict_mapping_subclass..TestNonDictMapping.__iter__intcSs |jSr~)r__len__rr+r+r,rsz=non_dict_mapping_subclass..TestNonDictMapping.__len__N)rrrrrrrr+r+r+r,TestNonDictMappingsr)rMapping)rr+r+r,non_dict_mapping_subclasss rcCs`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|jSr~)yearrYr+r+r,r[r\zAmultiindex_year_month_day_dataframe_random_data..cSs|jSr~)rNrYr+r+r,r[r\cSs|jSr~)dayrYr+r+r,r[r\cSsg|]}|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   rrc CsPtddddgdddggdddd d d d d d d g dd d dd d d dd d g gd d gdS)z; 2-level MultiIndex, lexsorted, with string names. foobarbazquxonetwothreerrLrhsecond)rcodesnames)rr+r+r+r,%lexsorted_two_level_string_multiindexs .rrcCs*|}ttjdd|tdddgdddS) z2DataFrame with 2 level MultiIndex with random data rABCexpr2)rTrU)rrPrandomrandnr)rrTr+r+r, multiindex_dataframe_random_data%s  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 rrrrrrrrLrrrhrF)rrrZverify_integrity)rrPrr)Z major_axisZ minor_axisZ major_codesZ minor_codesZ index_namesr+r+r,_create_multiindex0s rcCs.tjddgddgtjddddgd d d gd S) zB MultiIndex with a level that is a tzaware DatetimeIndex. rLrab20130101r US/Eastern)Zperiodstzrrr)r)rZ from_productrQZ date_ranger+r+r+r,_create_mi_with_dt64tz_levelDsrrz US/Pacific)rint8dtypeint16int32int64uint8uint16uint32uint64float32float64r complex64 complex128rrrrrZInt64ZUInt16ZFloat32booleanzstring[python]) stringrz datetime-tzperiodrrangerrrrrrrrrrz bool-object 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_dictr`copyrar+r+r,rTvs cCsg|]\}}t|ts|qSr+) isinstancerrrvaluer+r+r,rs rcCs|j}t|S)z8 index fixture, but excluding MultiIndex cases. )r`rr)rbrr+r+r, index_flats rcCsDg|]<\}}|ds|ds|ds|dkst|ts|qS)rZuintfloat)rrrr) startswithrrrr+r+r,rs    cCst|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. Tdeep)rrrr~rrLN)rr`rvaluestolistr from_tuplesr)rbindvalsr+r+r,index_with_missings  rrcCst}d|_|S)zC Fixture for Series of floats with Index of unique strings r)rZmakeStringSeriesr2sr+r+r, string_seriessrcCst}d|_|S)zI Fixture for Series of dtype object with Index of unique strings objects)rZmakeObjectSeriesr2rr+r+r, object_seriessrcCst}d|_|S)z9 Fixture for Series of floats with DatetimeIndex ts)rZmakeTimeSeriesr2rr+r+r,datetime_seriessrcCs$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, s rcCst|S)zE Fixture for tests on series with changing types of indices. rrTr+r+r,series_with_simple_indexsrc Cs`ddddddddgddddddddgg}t|}t|}tjd}t||d}tj|d <|S) z: Fixture with a Series with a 2-level MultiIndex. rrrrrrrr)rrrrrPrrrNaN)ZarraysrrTrZserr+r+r,series_with_multilevel_indexs    rcCs$i|]}|jdtjd|dqS)z-seriesr)r2r)rrZmake_rand_series)rrr+r+r,rs cCst|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_objsr`rrar+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_frame:srcCs 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_frameWsrc 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_frametsr c Cstdddtdddtdddtdddtdddtdddtdddtdddtdddtdddg }ttjt||dS)zF Fixture for Series with a DatetimeIndex that has duplicates. irLrrrKr)rrrPrrr)datesr+r+r,(rand_series_with_duplicate_datetimeindexs           r)rkrlg?g?z2012-01M)freqz period[M]z 2012-02-01Dz period[D]z 2011-01-01ri)secondsztimedelta64[ns]cCs|jSr~r_rar+r+r,ea_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. r_rar+r+r,all_arithmetic_operatorssr#cCs|jS)zU Fixture for operator and roperator arithmetic, comparison, and logical ops. r_rar+r+r,all_binary_operatorss"r$cCs|jS)z Fixture for operator and roperator arithmetic functions. Notes ----- This includes divmod and rdivmod, whereas all_arithmetic_operators does not. r_rar+r+r,all_arithmetic_functionssr%countrmaxminZmeanprodZstdvarZmedianZkurtZskewZsemcCs|jS)z. Fixture for numeric reduction names. r_rar+r+r,all_numeric_reductionssr+allanycCs|jS)z. Fixture for boolean reduction names. r_rar+r+r,all_boolean_reductions*sr.cCs|jS)z> Fixture for all (boolean + numeric) reduction names. r_rar+r+r,all_reductions5sr/cCs|jS)z; Fixture for operator module comparison functions. r_rar+r+r, comparison_op=sr0__le____lt____ge____gt__cCs|jS)zm Fixture for dunder names for compare operations except == and != * >= * > * < * <= r_rar+r+r,compare_operators_no_eq_neNs r5__and____rand____or____ror____xor____rxor__cCs|jS)zY Fixture for dunder names for common logical operations * | * & * ^ r_rar+r+r,all_logical_operators[s r<ZcumsumZcumprodZcumminZcummaxcCs|jS)z0 Fixture for numeric accumulation names r_rar+r+r,all_numeric_accumulationslsr=cCs |dS)zO Returns the configuration for the test setting `--strict-data-files`. r))r>)Z pytestconfigr+r+r,strict_data_fileswsr>zCallable[..., str])r>r!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. testscsHtjjf|}tj|sDr2td|dtd|d|S)NzCould not find file z and --strict-data-files is set.zCould not find .)osr0joinexists ValueErrorr5r@)rr0Z BASE_PATHr>r+r,decos  zdatapath..deco)rAr0rBdirname__file__)r>rFr+rEr,datapaths rIcCst|ddddS)z* The iris dataset as a DataFrame. iorcsvziris.csv)rQZread_csv)rIr+r+r,irissrLz Asia/Tokyozdateutil/US/Pacificzdateutil/Asia/Singaporez+01:15z-02:15z UTC+01:15z UTC-02:15i,i)hoursrrcCsg|] }t|qSr+rW)rir+r+r,rscCs|jS)zD Fixture for trying timezones including default (None): {0} r_rar+r+r,tz_naive_fixturesrOcCs|jS)z4 Fixture for trying explicit timezones: {0} r_rar+r+r,tz_aware_fixturesrPrz dateutil/UTCcCs|jS)zQ Fixture to provide variants of UTC timezone strings and tzinfo objects. r_rar+r+r, utc_fixturesrQcCs|jS)zR Parametrized fixture for string dtypes. * str * 'str' * 'U' r_rar+r+r, string_dtypes rRpyarrowcCs|jS)za Parametrized fixture for string dtypes. * 'string[python]' * 'string[pyarrow]' r_rar+r+r,nullable_string_dtypes rTpythoncCs|jSzb Parametrized fixture for pd.options.mode.string_storage. * 'python' * 'pyarrow' r_rar+r+r,string_storage s rWZnumpy_nullablecCs|jSrVr_rar+r+r, dtype_backends rXcCs|jS)zK Parametrized fixture for bytes dtypes. * bytes * 'bytes' r_rar+r+r, bytes_dtype.srYcCs|jS)zN Parametrized fixture for object dtypes. * object * 'object' r_rar+r+r, object_dtype9srZobjectcCs|jS)zo Parametrized fixture for string dtypes. * 'object' * 'string[python]' * 'string[pyarrow]' r_rar+r+r,any_string_dtypeDsr\cCs|jS)z\ Parametrized fixture for datetime64 dtypes. * 'datetime64[ns]' * 'M8[ns]' r_rar+r+r,datetime64_dtypeUsr]cCs|jS)z^ Parametrized fixture for timedelta64 dtypes. * 'timedelta64[ns]' * 'm8[ns]' r_rar+r+r,timedelta64_dtype`sr^rc CstddddddddS)z- Fixture emits fixed Timestamp.now() irLrMr  )rrNrhourminuter microsecond)rr+r+r+r, fixed_now_tsksrdcCs|jS)z^ Parameterized fixture for float dtypes. * float * 'float32' * 'float64' r_rar+r+r,float_numpy_dtypeus recCs|jS)zR Parameterized fixture for float dtypes. * 'Float32' * 'Float64' r_rar+r+r,float_ea_dtypesrfcCs|jS)z~ Parameterized fixture for float dtypes. * float * 'float32' * 'float64' * 'Float32' * 'Float64' r_rar+r+r,any_float_dtypes rgcCs|jS)zg Parameterized fixture for complex dtypes. * complex * 'complex64' * 'complex128' r_rar+r+r, complex_dtypes rhcCs|jS)z| Parameterized fixture for signed integer dtypes. * int * 'int8' * 'int16' * 'int32' * 'int64' r_rar+r+r,any_signed_int_numpy_dtypes ricCs|jS)zx Parameterized fixture for unsigned integer dtypes. * 'uint8' * 'uint16' * 'uint32' * 'uint64' r_rar+r+r,any_unsigned_int_numpy_dtypes rjcCs|jS)z Parameterized fixture for any integer dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' r_rar+r+r,any_int_numpy_dtypesrkcCs|jS)z Parameterized fixture for any nullable integer dtype. * 'UInt8' * 'Int8' * 'UInt16' * 'Int16' * 'UInt32' * 'Int32' * 'UInt64' * 'Int64' r_rar+r+r,any_int_ea_dtypesrlcCs|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' r_rar+r+r, any_int_dtypesrmcCs|jS)z Parameterized fixture for any nullable integer dtype and any float ea dtypes. * 'UInt8' * 'Int8' * 'UInt16' * 'Int16' * 'UInt32' * 'Int32' * 'UInt64' * 'Int64' * 'Float32' * 'Float64' r_rar+r+r,any_numeric_ea_dtypesrncCs|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]' r_rar+r+r,any_numeric_ea_and_arrow_dtypes rocCs|jS)z~ Parameterized fixture for any signed nullable integer dtype. * 'Int8' * 'Int16' * 'Int32' * 'Int64' r_rar+r+r,any_signed_int_ea_dtype7s rpcCs|jS)z Parameterized fixture for any (purely) real numeric dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' * float * 'float32' * 'float64' r_rar+r+r,any_real_numpy_dtypeDsrqcCs|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. r_rar+r+r,any_real_numeric_dtypeYsrrcCs|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' r_rar+r+r,any_numpy_dtypeps"rscCs|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' r_rar+r+r,any_numeric_dtypesrtrrr mixedbytesacrz mixed-integerg@Zfloatingr integerzmixed-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 -------- >>> 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)r`rPrr[)rbZinferred_dtyperr+r+r,any_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:)rB)r5 importorskipZIPython.core.interactiveshellr}Ztraitlets.configr~ZHistoryManagerZ hist_file)r}r~r r+r+r,ip s   rZbsrZcooZcscZcsrZdiaZdokZlilcCsddlm}t||jdS)z- Yields scipy sparse matrix classes. r)sparseZ_matrix)Zscipyrgetattrr`)rbrr+r+r,spmatrixs rcCs6g|].}tttj|tjjr|dkrttj|qS)Tick) issubclassrrQoffsetsr)ror+r+r,r'scCs|jS)zN Fixture for Tick based datetime offsets available for a time series. r_rar+r+r, tick_classes&s rcCs|Sr~r+rYr+r+r,r[4r\cCs|jS)zk Simple fixture for testing keys in sorting methods. Tests None (no key) and the identity key. r_rar+r+r, sort_by_key4srccsxtdddlm}ddlm}ddlm}Gddd|}|d|d d |V|ddd|j d<|j dS) Nfsspecr)register_implementation)MemoryFileSystem) _registrycs,eZdZdZdgZddfdd ZZS)z fsspectest..TestMemoryFStestmemNrr cs$|dd|jd<tjf|dS)Ntestr)poprsuperr)rr __class__r+r,rHsz)fsspectest..TestMemoryFS.__init__)rrrprotocolrr __classcell__r+r+rr, TestMemoryFSDsrrT)clobber) r5rrrZfsspec.implementations.memoryrZfsspec.registryrrrstoreclear)rrregistryrr+r+r, fsspectest=s      r)rNN)ZEgonZVenkmanN)NCC1701Drrz#tuple[Hashable, Hashable, Hashable]cCs|jS)zP A 3-tuple of names, the first two for operands, the last for a result. r_rar+r+r,rSsrcCs|jS)zI Parametrize over __setitem__, loc.__setitem__, iloc.__setitem__ r_rar+r+r, indexer_slifsrcCs|jS)z< Parametrize over loc.__getitem__, iloc.__getitem__ r_rar+r+r, indexer_linsrcCs|jS)z8 Parametrize over __setitem__, iloc.__setitem__ r_rar+r+r, indexer_sivsrcCs|jS)z7 Parametrize over __setitem__, loc.__setitem__ r_rar+r+r, indexer_sl~srcCs|jS)z: Parametrize over at.__setitem__, loc.__setitem__ r_rar+r+r, indexer_alsrcCs|jS)z< Parametrize over iat.__setitem__, iloc.__setitem__ r_rar+r+r, indexer_ialsrboolcCstjjjdkS)z> Fixture to check if the array manager is being used. r)rQoptionsmode data_managerr+r+r+r,using_array_managersrcCstjjjotjjjdkS)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/WarsawcCs|jS)zE tzinfo for Europe/Warsaw using pytz, dateutil, or zoneinfo. r_rar+r+r,warsawsr(.__doc__ __future__r collectionsrrrrrrrzr operatorrAtypingr r r Z dateutil.tzr rZ hypothesisrstnumpyrPr5ZpytzrrZpandas.util._test_decoratorsutilZ_test_decoratorstdZpandas.core.dtypes.dtypesrrZpandasrQrrrrrrrZpandas._testingZ_testingrZ pandas.corerZpandas.core.indexes.apirrrSpa ImportErrorZ has_pyarrowZzoneinfoZPY39ZZoneInfoZZoneInfoNotFoundErrorr-r8rDsettingsZregister_profileZ HealthCheckZtoo_slowZ load_profilesplitr2rZtseriesrclsZregister_type_strategyZbuildsZintegersZbooleansZfixturer=rSrcZ axis_framerdrerfrgrjrmrnror`Z skip_if_norurvrwrzr}Z NULL_OBJECTSrZnulls_fixture2nanZNaTrZunique_nulls_fixture2ZNP_NAT_OBJECTSrZnp_nat_fixture2rrZindex_or_series2rrrZbox_with_array2rrrrrrrZmakeStringIndexZ makeDateIndexZmakePeriodIndexZmakeTimedeltaIndexZmakeRangeIndexZ makeIntIndexZ makeUIntIndexZmakeFloatIndexZ makeBoolIndexrr[rrZmakeNumericIndexZmakeCategoricalIndexZmakeIntervalIndexrrrr rridxkeysrTZindex_fixture2rArZ index_flat2rrrrrZ_seriesrrZNARROW_NP_DTYPESZ_narrow_seriesrrrrrr rrZ_all_arithmetic_operatorsr#addZraddsubZrsubmulZrmultruedivZrtruedivfloordivZ rfloordivmodZrmodpowZrpoweqneltlegtgeand_Zrand_xorZrxoror_Zror_r$r%Z_all_numeric_reductionsr+Z_all_boolean_reductionsr.Z_all_reductionsr/r0r5r<Z_all_numeric_accumulationsr=r>rIrLZ TIMEZONESextendZ TIMEZONE_IDSZparametrize_fixture_docr.rOrPZtz_aware_fixture2Z_UTCSappendrQZ utc_fixture2Z STRING_DTYPESrRrTrWrXZstring_storage2Z BYTES_DTYPESrYZ OBJECT_DTYPESrZr\ZDATETIME64_DTYPESr]ZTIMEDELTA64_DTYPESr^rdZFLOAT_NUMPY_DTYPESreZFLOAT_EA_DTYPESrfZALL_FLOAT_DTYPESrgZCOMPLEX_DTYPESrhZSIGNED_INT_NUMPY_DTYPESriZUNSIGNED_INT_NUMPY_DTYPESrjZALL_INT_NUMPY_DTYPESrkZALL_INT_EA_DTYPESrlZALL_INT_DTYPESrmrnZALL_INT_PYARROW_DTYPES_STR_REPRZFLOAT_PYARROW_DTYPES_STR_REPRroZSIGNED_INT_EA_DTYPESrpZALL_REAL_NUMPY_DTYPESrqZALL_REAL_DTYPESrrZALL_NUMPY_DTYPESrsZALL_NUMERIC_DTYPESrtZNAr{Z_any_skipna_inferred_dtyper^_r|rr__all__rrrrsetitemlocZilocrrrratrZiatrrrZwarsawsrr+r+r+r,s8    $     1                                                 "                         !                         $#$$2&