B •xŠ\zFã@sddlmZmZmZddlmZddlZddlmZm Z ddl Z ddl m Z ddl ZddlZddlmZmZddlmZmZddlmmZddlZe jjdd e je jj fd e j !d¡d d „Z"d d„Z#ej$dddd„ƒZ%ej$dddd„ƒZ&ej$dddddddgddd„ƒZ'ej$ddd d!gd"d#„d$d%d&„ƒZ(e(Z)ej$dd gd'd#„d$d(d)„ƒZ*ej$d*d+„ƒZ+ej$dd,dgdd-d.„ƒZ,d/d0d1d2d3d4d5d6d7d8d9d:d;dg¡ej$e-dd?d@„ƒZ/dAdBdCdDdEdFdGdHdIdJg Z0ej$e0ddKdL„ƒZ1dMdNgZ2ej$e2ddOdP„ƒZ3ej4j5j6j7 8¡Z7ej$e9e7ƒddQdR„ƒZ:dSdT„Z;ej$dUdVdWdXdYdZgdd[d\„ƒZdagddbdc„ƒZ?ej$d]d^d_ej=d`ej>dagdddde„ƒZ@ej$dd,gddfdg„ƒZAej$dhdidjdk„ƒZBdldmee ƒgZCerddnlmDZDeC EeDj¡ej$eCddodp„ƒZFej$dqdrdsdtgddudv„ƒZGej$dwdx„ƒZHej$dydz„ƒZIej$d{d|„ƒZJej$d}d~gddd€„ƒZKej$dsdtdd‚gddƒd„„ƒZLej$dsdtdd‚gdd…d†„ƒZMej$dejNejOePd‡ƒe Pdˆ¡gdd‰dŠ„ƒZQeQZRej$dejNejOgdd‹dŒ„ƒZSeSZTdddŽddd‘e ƒeƒed’ƒedƒed“ƒg ZUe VeWeUƒ¡ej$eUdd”d•„ƒƒZXe VeWeUdd…ƒ¡ej$eUdd…dd–d—„ƒƒZYd˜d™dšd›gZZdœddždŸgZ[e\d d¡d¢d£gZ]d¤d¥d¦d§gZ^eZe]Z_e[e^Z`ePd¨d©gZaebdªd«gZceWd¬d­gZdd®d¯gZed°d±gZfegd²gZheid³gZjekd´gZleae_ZmemecedeeefehelejeZnej$edddµd¶„ƒZoej$eadd·d¸„ƒZpej$ecdd¹dº„ƒZqej$e]dd»d¼„ƒZrej$eZdd½d¾„ƒZsej$e_dd¿dÀ„ƒZtej$emddÁd„ƒZuej$enddÃdÄ„ƒZvdÅdÆejNdÇgfesôdÈndÅedƃejNedǃgferd³ndÅdÉejNdÊgfdËejNejNejNgfdËgfdÌdÆejNdÍgfdÎdÆejNdÏgfdÐdÑejNdÏgfdÒdejNdÍgfdÓdejNdÏgfdÔedƒejNed̓gfdÕdejNd,gfdÖe wdסejNe wdØ¡gfdÙe xdÚ¡ejNe xdÛ¡gfdÜedÝddƒejNedÞddƒgfdßedƒejNed̓gfdàedƒejNed̓gfdáe ydÝ¡ejOe ydÞ¡gfdâe zdd¡ejNe zddÍ¡gfgZ{e|e{Ž\Z}Z~ej$e{e}d$dãd䄃Zej$dådæ„ej€jDƒddçd脃Z‚xDdé ƒ¡D]8Z„e…ej†j€e„ƒZ‡e  ˆe‡e j‰e‡e  Šdêdë¡e  ‹¡dì¡q¬WxPdí ƒ¡D]DZ„e…ej†j€e„ƒZ‡e  ˆe‡e j‰e‡e  Šdîdï¡e  ‹¡e jŠddðdñdò¡qòWxPd󠃡D]DZ„e…ej†j€e„ƒZ‡e  ˆe‡e j‰e‡e  Šdôdõ¡e  ‹¡e jŠddðdñdö¡qDWdS)÷é)ÚdateÚtimeÚ timedelta)ÚDecimalN)ÚtzlocalÚtzutc)Ú strategies)Ú FixedOffsetÚutc)ÚPY3ÚuZciiô)ZdeadlineZtimeoutZsuppress_health_checkcCsd|jdddd|jdddd|jdddd|jd dd d|jd dd d|jd ddddS)Nz --skip-slowÚ store_truezskip slow tests)ÚactionÚhelpz--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)Úparser©rú.lib/python3.7/site-packages/pandas/conftest.pyÚpytest_addoptionsrcCs¤d|jkr |j d¡r t d¡d|jkr@|j d¡r@t d¡d|jkr`|j d¡r`t d¡d |jkr€|j d ¡r€t d ¡d |jkr |j d ¡s t d¡dS)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)ÚkeywordsZconfigÚ getoptionÚpytestÚskip)ÚitemrrrÚpytest_runtest_setup.s    rT)ZautousecCst dd¡dS)NZchained_assignmentÚraise)ÚpdZ set_optionrrrrÚconfigure_testsCsrcCst|d<t|d<dS)NÚnpr)rr)Zdoctest_namespacerrrÚ add_importsJsrZbsrZcooZcscZcsrZdiaZdokZlil)ÚparamscCsddlm}t||jdƒS)Nr)ÚsparseZ_matrix)Zscipyr ÚgetattrÚparam)Úrequestr rrrÚspmatrixPs r$éÚindexÚcolumnscCs d |¡S)Nz axis {!r})Úformat)ÚxrrrÚWsr*)rÚidscCs|jS)zB Fixture for returning the axis numbers of a DataFrame. )r")r#rrrÚaxisVsr,cCs d |¡S)Nz axis {!r})r()r)rrrr*bscCs|jS)z? Fixture for returning the axis numbers of a Series. )r")r#rrrÚ axis_seriesbsr-cCs tjdddddlm}|ƒS)zj Get an instance of IPython.InteractiveShell. Will raise a skip if IPython is not installed. ZIPythonz6.0.0)Z minversionr)ÚInteractiveShell)rZ importorskipZIPython.core.interactiveshellr.)r.rrrÚipjs r/FcCs|jS)az 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 compatiblity if we decide to change the default (and would need to warn if this parameter is not passed))r")r#rrrÚobservedwsr0Ú__add__Ú__radd__Ú__sub__Ú__rsub__Ú__mul__Ú__rmul__Ú __floordiv__Ú __rfloordiv__Ú __truediv__Ú __rtruediv__Ú__pow__Ú__rpow__Ú__mod__Ú__rmod__Z__div__Z__rdiv__cCs|jS)zC Fixture for dunder names for common arithmetic operations )r")r#rrrÚall_arithmetic_operatorssr?ÚsumÚmaxÚminZmeanZprodZstdÚvarZmedianZkurtZskewcCs|jS)z- Fixture for numeric reduction names )r")r#rrrÚall_numeric_reductionsšsrDÚallÚanycCs|jS)z- Fixture for boolean reduction names )r")r#rrrÚall_boolean_reductions¥srGcCs|jS)N)r")r#rrrÚcython_table_items°srHcsDg}x:|D]2\‰‰| ˆˆˆf¡|‡‡‡fdd„tDƒ7}q W|S)aÀcombine frame, functions from SelectionMixin._cython_table keys and expected result. Parameters ---------- ndframe : DataFrame or Series func_names_and_expected : Sequence of two items The first item is a name of a NDFrame method ('sum', 'prod') etc. The second item is the expected return value Returns ------- results : list List of three items (DataFrame, function, expected result) cs"g|]\}}|ˆkrˆ|ˆf‘qSrr)Ú.0ÚfuncÚname)ÚexpectedÚ func_nameÚndframerrú Èsz,_get_cython_table_params..)ÚappendÚ _cython_table)rNZfunc_names_and_expectedZresultsr)rLrMrNrÚ_get_cython_table_paramsµs rRÚ__eq__Ú__ne__Ú__le__Ú__lt__Ú__ge__Ú__gt__cCs|jS)zu Fixture for dunder names for common compare operations * >= * > * == * != * < * <= )r")r#rrrÚall_compare_operatorsÍs rYZgzipÚbz2ÚzipZxz)ZmarkscCs|jS)zJ Fixture for trying common compression types in compression tests )r")r#rrrÚ compressionÝsr\cCs|jS)zj Fixture for trying common compression types in compression tests excluding uncompressed case )r")r#rrrÚcompression_onlyæsr]cCs|jS)z+ Fixture that an array is writable )r")r#rrrÚwritableðsr^Úmodule)ZscopecCsddlm}|jS)Nr)Útimezone)Údatetimer`r )r`rrrÚdatetime_tz_utcøs rbr z dateutil/UTC)r`cCs|jS)zP Fixture to provide variants of UTC timezone strings and tzinfo objects )r")r#rrrÚ utc_fixturesrcÚinnerZouterÚleftÚrightcCs|jS)z9 Fixture for trying all types of join operations )r")r#rrrÚ join_type srgcCs | d¡S)Nz--strict-data-files)r)Z pytestconfigrrrÚstrict_data_filessrhcs(tj tj t¡d¡‰‡‡fdd„}|S)a?Get the path to a data file. Parameters ---------- path : str Path to the file, relative to ``pandas/tests/`` Returns ------- path : path including ``pandas/tests``. Raises ------ ValueError If the path doesn't exist and the --strict-data-files option is set. ZtestscsNtjjˆf|žŽ}tj |¡sJˆr6d}t| |¡ƒ‚nd}t | |¡¡|S)Nz6Could not find file {} and --strict-data-files is set.zCould not find {}.)ÚosÚpathÚjoinÚexistsÚ ValueErrorr(rr)ÚargsrjÚmsg)Ú BASE_PATHrhrrÚdeco-s zdatapath..deco)rirjrkÚdirnameÚ__file__)rhrqr)rprhrÚdatapaths rtcCst |ddƒ¡S)z The iris dataset as a DataFrame.Údataziris.csv)rZread_csv)rtrrrÚiris:srvÚnlargestÚ nsmallestcCs|jS)z0 Fixture for trying all nselect methods )r")r#rrrÚnselect_method@sryZbothZneithercCs|jS)z; Fixture for trying all interval closed parameters )r")r#rrrÚclosedHsrzcCs|jS)zR Secondary closed fixture to allow parametrizing over all pairs of closed )r")r#rrrÚ other_closedPsr{ÚnanZNaNcCs|jS)z. Fixture for each null type in pandas )r")r#rrrÚ nulls_fixtureXsr}cCs|jS)zK Fixture for each null type in pandas, each null type exactly once )r")r#rrrÚunique_nulls_fixturecsr~ZUTCz US/Easternz Asia/Tokyozdateutil/US/Pacificzdateutil/Asia/Singaporei,iÔþÿÿcCs|jS)zD Fixture for trying timezones including default (None): {0} )r")r#rrrÚtz_naive_fixturetsrcCs|jS)z4 Fixture for trying explicit timezones: {0} )r")r#rrrÚtz_aware_fixture}sr€Zuint8Zuint16Zuint32Zuint64ZUInt8ZUInt16ZUInt32ZUInt64Zint8Zint16Zint32Zint64ZInt8ZInt16ZInt32ZInt64Zfloat32Zfloat64Z complex64Z complex128ÚstrÚUzdatetime64[ns]zM8[ns]ztimedelta64[ns]zm8[ns]ÚboolÚbytesÚobjectcCs|jS)zMParametrized fixture for string dtypes. * str * 'str' * 'U' )r")r#rrrÚ string_dtype sr†cCs|jS)z^ Parameterized fixture for float dtypes. * float * 'float32' * 'float64' )r")r#rrrÚ float_dtype«s r‡cCs|jS)zg Parameterized fixture for complex dtypes. * complex * 'complex64' * 'complex128' )r")r#rrrÚ complex_dtype¸s rˆcCs|jS)z| Parameterized fixture for signed integer dtypes. * int * 'int8' * 'int16' * 'int32' * 'int64' )r")r#rrrÚ sint_dtypeÅs r‰cCs|jS)zx Parameterized fixture for unsigned integer dtypes. * 'uint8' * 'uint16' * 'uint32' * 'uint64' )r")r#rrrÚ uint_dtypeÔs rŠcCs|jS)z³ Parameterized fixture for any integer dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' )r")r#rrrÚ any_int_dtypeâsr‹cCs|jS)zí Parameterized fixture for any (purely) real numeric dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' * float * 'float32' * 'float64' )r")r#rrrÚany_real_dtypeõsrŒ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' )r")r#rrrÚany_numpy_dtype s#rÚstringÚaÚcZunicodeóaócÚemptyz mixed-integeréZmixedg@Zfloatinggð?Zintegerzmixed-integer-floatÚdecimalZbooleanÚ datetime64z 2013-01-01z 2018-01-01raZ20130101Z20180101riÝiârrZperiodZintervalcCs |j\}}tj|td}||fS)a² Fixture for all inferred dtypes from _libs.lib.infer_dtype The covered (inferred) types are: * 'string' * 'unicode' (if PY2) * 'empty' * 'bytes' (if PY3) * '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 )Zdtype)r"rZarrayr…)r#Zinferred_dtypeÚvaluesrrrÚany_skipna_inferred_dtypeNs* r˜cCs.g|]&}tttj|ƒtjjƒrttj|ƒ‘qSr)Ú issubclassr!rÚoffsetsZTick)rIÚorrrrOsrOcCs|jS)zN Fixture for Tick based datetime offsets available for a time series. )r")r#rrrÚ tick_classessrœz)MonthBegin MonthEnd BMonthBegin BMonthEndiÿÿÿéc)ÚnÚ normalizez%YearBegin YearEnd BYearBegin BYearEndéûÿÿÿéé )Z min_valueZ max_value)ržrŸZmonthz1QuarterBegin QuarterEnd BQuarterBegin BQuarterEndièÿÿÿé)ržrŸZ startingMonth)Œrarrrr•rriZ dateutil.tzrrZ hypothesisrÚstZnumpyrrZpytzr r Z pandas.compatr r Zpandas.util._test_decoratorsÚutilZ_test_decoratorsZtdZpandasrZsettingsZregister_profileZ unlimitedZ HealthCheckZtoo_slowZ load_profilerrZfixturerrr$r,Z axis_framer-r/r0Z_all_arithmetic_operatorsÚextendr?Z_all_numeric_reductionsrDZ_all_boolean_reductionsrGZcoreÚbaseZSelectionMixinrQÚitemsÚlistrHrRrYr"Zskip_if_no_lzmar\r]r^rbZutc_objsr`rPrcrgrhrtrvryrzr{r|ZNaTÚfloatr}Znulls_fixture2r~Zunique_nulls_fixture2Z TIMEZONESZparametrize_fixture_docrrr€ZUNSIGNED_INT_DTYPESZUNSIGNED_EA_INT_DTYPESÚintZSIGNED_INT_DTYPESZSIGNED_EA_INT_DTYPESZALL_INT_DTYPESZALL_EA_INT_DTYPESZ FLOAT_DTYPESÚcomplexZCOMPLEX_DTYPESZ STRING_DTYPESZDATETIME_DTYPESZTIMEDELTA_DTYPESrƒZ BOOL_DTYPESr„Z BYTES_DTYPESr…Z OBJECT_DTYPESZALL_REAL_DTYPESZALL_NUMPY_DTYPESr†r‡rˆr‰rŠr‹rŒrr–Z TimestampZPeriodZIntervalZ_any_skipna_inferred_dtyper[r+Ú_r˜ršÚ__all__rœÚsplitrKr!ZtseriesÚclsZregister_type_strategyZbuildsZintegersZbooleansrrrrÚs"   "         !,         $   (  " 1"