d6.dZddlmZddlmZddlmZmZmZmZm Z ddl m Z ddl Z ddl Z ddlmZmZmZddlmZmZddlZdd lmZddlZddlZdd lmZmZddlmcm 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 l1m2Z2ddl3m4Z4m5Z5 ddl6Z7[7dZ8n #e9$rdZ8YnwxYwdZ:e-j;r(ddl:Z: e:j<dn#e:j=$rdZ:YnwxYwd,dZ>d-dZ?d,dZ@ejABddejCjDfejAEddFD]VZGeHe&jIjJeGZKejLeKejMeKejNdd ejO!Wd"FD]gZGeHe&jIjJeGZKejLeKejMeKejNd#d$ejOejNd%d&'(hd)FD]gZGeHe&jIjJeGZKejLeKejMeKejNd*d+ejOejNd%d&',hejPd,d-ZQejPd.d,d/ZRejPgd0d12d3ZSeSZTejPd%d4gd52d6ZUejPgd78d9ZVejPgd78d:ZWejPddg8d;ZXejPgd<8d=ZYejPgd>8d?ZZejPgd@8dAZ[ejPgd@8dBZ\ejPddCdDdEdFdGej]dHe!j^dIJg8dKZ_ejPdCdDdEdFdGej]dHe!j^dIJg8dLZ`ejPddg8dMZaejPgdN8dOZbejPdPdQg8dRZcejPe0jddS2dTZeeeZfejPdejge&jhg8dUZieiZjejPe0jkdV2dWZlelZmejPe'e*g8dXZnejPe4e*gdYdZg2d[ZoeoZpejPe4e*e&jqggd\2d]ZrejPe4e*e'e&jqgd^2d_ZsesZtejPd`ZuejPdaZvejPdbZwejPd.ddZxejPd/dfZydgZzdhZ{idie0j|djdke0j}djdle0j}djdmndoe0j~djdpe0jdjdqe0jdjdre0jdjdrsdte0jdjdtsdue0jdjdusdve0jdjdvsdwe0jdjdwsdxe0jdjdxsdye0jdjdysdze0jdjdzsd{e0jdjd{sd|e0jdjd|sd}e0jd~ee4ejd~dke0jdjd|sde0jdjd|sde0jdje0jdje4ge5jegdgde{eze4gde4ejdjdse4ejdjdse4ejdjdse4ejdjedse4e&jqe0j|djdsdZe8r.e4e&jqe0j|djdsZeed<ejPe8dZeZejPdeD8dZeZejPdeD8dZejPd0dZejPd0dZejPd0dZdZdeDZejPd0dZejPd0dZde0jDZieeeZejPe8dZejPd/dZejPd/dZejPd/dZejPd/dZejPd0dZejPe(dd$e$dvdfe(dde$d|dfe)dddfe)dddfe,ddne#dnfe+ddfg8dZgdZejPe8dZejPe je2je je2je je2je je2je je2je je2je je2je je je je je je je je2je je2je je2jg8dZejPe je2je je2je je2je je2je je2je je2je je2jg8dZgdZejPeȬ8dZddgZejPeʬ8dZeezZejPe̬8dZejPe je je je je je jg8dZejPgd8dZejPgd8dZgdZejPeѬ8dZejPd„ZejPd1dńZejPd/dƄZddddddddddeeedΦededϦe je ed%Цe edѬЦdҬӦgZe:3e֠e:j<dme:j<dgdԄeDZe!jeeئejPeeج2dՄZe!jeed%dejPed%ded%d2dքZeZddeee jgZe:#eޠe:j<dejPeެ8dلZeZejPe0j8dڄZejPdej]de!j^dۦJg8d܄ZejPdej]de!j^dۦJg8dބZejPdej]de!j^dۦJg8dZeZejPe0j8dZejPe0j8dZejPddej]de!j^dۦJg8dZejPe0j8dZejPe0j8dZejPd2dZejPe0j8dZejPe0j8dZejPe0j8dZejPe0j8dZejPe0j8dZejPe0j8dZejPe0j8dZejPe0j8dZejPe0j8dZejPe0je0jz8dZejPe0je0jze0jze0jz8dZejPe0j8dZ ejPe0j 8dZ ejPe0j 8dZ ejPe0j8dZejPe0j8dZdidejgdgfdide&jdgfdde&jhdgfddejgdgfdejgejgejggfdgfddejgdgfddejgdgfddejgdgfdd%ejgdgfdd%ejgdgfde d%ejge dgfddejgdgfdde&jdgfdejd ejgejd gfdke,d ejge,d gfd edd%d%ejgedd%d%gfdped%ejgedgfded%ejgedgfdoe)de&jhe)dgfde(dd%ejge(ddgfgZee\ZZejPee2dZejPdZejPgd8dZejPde&jJjD8dZejPddg8dZejPdZejPdddejgejgejgfejge&jhdfejge&jdfe&je&je&jfg8d3dZejPe0je0j e0j!g8d Z"ejPe0j e0j!g8d!Z#ejPe0je0j!g8d"Z$ejPe0je0j g8d#Z%ejPe0j&e0j g8d$Z'ejPe0j(e0j!g8d%Z)ejPd4d'Z*ejPd4d(Z+d)d*gZ,e:%e,e:j<d)ejPe,8d5d+Z-dS(6a 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 MultiIndexTFUTCreturnNonec&|ddd|ddd|ddd|d dd |d dd |d dddS)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.) addoption)parsers K/var/www/html/t/fyr/venv311/lib/python3.11/site-packages/pandas/conftest.pypytest_addoptionr.hs ]<>OPPP %lAUVVV [OLLL L7N ]<>STTT  ?item pytest.Itempathstrmessagec|j|r7|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)r0r2r4s r-ignore_doctest_warningr<wsZ y$I  223FW3F3FGGHHHHHIIr/c|d}|d}|d}|d}tjjd|dftjjd|dftjjd|dfg}d g}|D]E}|d s|d d rK|tjd|D]\} } t|| | d|j vr$|tjj |D]^\} } } }| |j vrN| r7|tj d|n|| _|r;d|j vr2|tj dGdS)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) getoptionr9r:r>r?r@r8 usefixturesr<nodeid arraymanagerkeywordsskip)itemsconfig skip_slow only_slow skip_networkskip_dbmarksignored_doctest_warningsr0r2r4r:kwd skip_if_foundarg_names r-pytest_collection_modifyitemsrTs  //I  //I##$455L{++G  69m<  i{C w 4 E E  MM   / 0 0 rwsARaARARr/)paramsidsc|jS)z@ Fixture for returning the axis numbers of a DataFrame. paramrequests r-axisr =r/rpc&dt|Srrrsrus r-rwrw s5FT!WW5F5Fr/c|jS)zA Fixture for returning aliases of axis 1 of a DataFrame. r{r}s r-axis_1r rr/)TFN)rxc|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{r}s r-observedr  =r/c|jS)z6 Boolean 'ordered' parameter for Categorical. r{r}s r-orderedr rr/c|jS)z% Boolean 'skipna' parameter. r{r}s r-skipnar(rr/)firstlastFc|jS)z[ Valid values for the 'keep' parameter used in .duplicated or .drop_duplicates r{r}s r-keepr0 =r/)bothneitherleftrightc|jS)zA Fixture for trying all interval 'inclusive' parameters. r{r}s r-inclusive_endpoints_fixturer9rr/)rrrrc|jS)z< Fixture for trying all interval closed parameters. r{r}s r-closedrArr/c|jS)zS Secondary closed fixture to allow parametrizing over all pairs of closed. r{r}s r- other_closedrIrr/gzipbz2zipxztarzstd zstandard)rOc|jS)zK Fixture for trying common compression types in compression tests. r{r}s r- compressionrQ  =r/c|jS)zk Fixture for trying common compression types in compression tests excluding uncompressed case. r{r}s r-compression_onlyrcrr/c|jS)z, Fixture that an array is writable. r{r}s r-writablerurr/)innerouterrrc|jS)z: Fixture for trying all types of join operations. r{r}s r- join_typer}rr/nlargest nsmallestc|jS)z1 Fixture for trying all nselect methods. r{r}s r-nselect_methodrrr/c*t|jSNtype__name__rus r-rwrwsd1gg6Fr/c|jS)z/ Fixture for each null type in pandas. r{r}s r- nulls_fixturerrr/c|jS)zL Fixture for each null type in pandas, each null type exactly once. r{r}s r-unique_nulls_fixturerrr/c*t|jSrrrus r-rwrwsQ8Hr/c|jS)z- Fixture for each NaT type in numpy. r{r}s r-np_nat_fixturerrr/c|jS)z; Fixture to parametrize over DataFrame and Series. r{r}s r-frame_or_seriesrrr/roseriesc|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{r}s r-index_or_seriesrrr/)rorarrayc|jS)zG Fixture to parametrize over Index, Series, and ExtensionArray r{r}s r-index_or_series_or_arrayrrr/c|jSr)rrus r-rwrws1:r/c|jS)z] Fixture to test behavior for Index, Series, DataFrame, and pandas Array classes r{r}s r-box_with_arrayrrr/c,Gddt}|S)z, Fixture for a dictionary subclass. ceZdZddZdS)"dict_subclass..TestSubDictr r!c0tj|g|Ri|dSr)dict__init__)selfargskwargss r-rz+dict_subclass..TestSubDict.__init__s( M$ 0 0 0 0 0 0 0 0 0r/Nr r!)r __module__ __qualname__rrmr/r- TestSubDictrs( 1 1 1 1 1 1r/r)r)rs r- dict_subclassrs4 11111d111 r/c6Gddtj}|S)z8 Fixture for a non-mapping dictionary subclass. c,eZdZd dZdZd dZd dZd S) 5non_dict_mapping_subclass..TestNonDictMappingr r!c||_dSr)_data)runderlying_dicts r-rz>non_dict_mapping_subclass..TestNonDictMapping.__init__s (DJJJr/c6|j|Sr)r __getitem__)rkeys r-rzAnon_dict_mapping_subclass..TestNonDictMapping.__getitem__s:))#.. .r/r c4|jSr)r__iter__rs r-rz>non_dict_mapping_subclass..TestNonDictMapping.__iter__s:&&(( (r/intc4|jSr)r__len__rs r-rz=non_dict_mapping_subclass..TestNonDictMapping.__len__s:%%'' 'r/Nr)r r )r r)rrrrrrrrmr/r-TestNonDictMappingrs_ ) ) ) ) / / / ) ) ) ) ( ( ( ( ( (r/r)rMapping)rs r-non_dict_mapping_subclassrs6 ( ( ( ( (S[ ( ( ( r/c*tjd}|dddg}|jd|jjD|_|jgdd|S) z DataFrame with 3 level MultiIndex (year, month, day) covering first 100 business days from 2000-01-01 with random data dc|jSr)yearrus r-rwzAmultiindex_year_month_day_dataframe_random_data..sr/c|jSr)rarus r-rwzAmultiindex_year_month_day_dataframe_random_data..s17r/c|jSr)dayrus r-rwzAmultiindex_year_month_day_dataframe_random_data..saer/c8g|]}|dS)i8)astype).0levs r- zCmultiindex_year_month_day_dataframe_random_data..s$%S%S%S3cjj&6&6%S%S%Sr/)rrarT)inplace)tmmakeTimeDataFramegroupbysumro set_levelslevels set_names)tdfymds r-/multiindex_year_month_day_dataframe_random_datar s s # #C ++''):):OOL M M Q Q S SC $$%S%S#)BR%S%S%STTCII000$??? Jr/rcBtgdgdggdgdgddgS)z; 2-level MultiIndex, lexsorted, with string names. foobarbazquxonetwothree) rrrr]r]r r r ) rr]r rr]r]r rr]r rsecond)rcodesnames)rrmr/r-%lexsorted_two_level_string_multiindexrsM ,,,.E.E.EF---/M/M/MN!   r/rc|}ttjdd|t gddS)z2DataFrame with 2 level MultiIndex with random data r )ABCexpr6)rorp)rrerandomrandnr)rros r- multiindex_dataframe_random_datar%sI 2E  AeU???QV5W5W5W   r/ctgd}tddg}tjgd}tjgd}ddg}t||g||g|d S) zJ MultiIndex used to test the general functionality of this object rrr)rrr]r r r )rr]rr]rr]rr F)rr r verify_integrity)rrerr) major_axis minor_axis major_codes minor_codes index_namess r-_create_multiindexr 0s 33344Ju~&&J(---..K(---..KH%K J'K(    r/c jtjddgddgtjdddggd  S) zB MultiIndex with a level that is a tzaware DatetimeIndex. r]r ab20130101r  US/Eastern)periodstzr)r )r from_productrf date_rangermr/r-_create_mi_with_dt64tz_levelr*DsJ  " Q#sR]:q\RRRS%%%   r/stringrrz datetime-tzz US/Pacific)r'periodrrangeint8dtypeint16int32int64uint8uint16uint32uint64float32float64z bool-objectr complex64 complex128)rrr)r]r r )rrr]r]r r Int64UInt16Float32booleanzstring[python]) bool-dtyper:r; categoricalintervalemptytuplesmi-with-dt64tz-levelmultirepeats nullable_int nullable_uintnullable_float nullable_boolz string-pythonzstring[pyarrow]zstring-pyarrowcJt|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|copyr}s r-rorovs  & + + - --r/cBg|]\}}t|t|Srm) isinstancerrrvalues r-rrs;   U*UJ:W:W     r/cN|j}t|S)z8 index fixture, but excluding MultiIndex cases. )r|rMrN)r~rs r- index_flatrTs" -C   ! ! # ##r/cg|]_\}}|dsE|ds0|ds|dvHt|t]|`S)ruintfloat)r-rCrGr@) startswithrPrrQs r-rrs    C NN5 ! !  ~~f%%  ~~g&& AAA5*--B BAAr/cZt|jd}|j}|jdvrT|}d|dddz|d<d|dddz|d<t j|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)rDrErFrrr]N)rMr|rNvaluestolistr from_tuplesr)r~indvalss r-index_with_missingrbs2 w} % * * * 5 5C :D}CCCzz||DGABBK'QT"Xabb\)R%d+++QRtCyyr/rc:tj}d|_|S)zC Fixture for Series of floats with Index of unique strings r)rmakeStringSeriesr6ss r- string_seriesrgs A AF Hr/c:tj}d|_|S)zI Fixture for Series of dtype object with Index of unique strings objects)rmakeObjectSeriesr6res r- object_seriesrks A AF Hr/c:tj}d|_|S)z9 Fixture for Series of floats with DatetimeIndex ts)rmakeTimeSeriesr6res r-datetime_seriesros A AF Hr/ct|}tj|}t ||ddS)zHelper for the _series dictr"F)ror6rN)lenrerrr)rosizedatas r-_create_seriesrts7 u::D 9??4 D $e#E : : ::r/c<i|]\}}d|dt|S)z series-with-z-indexrt)rindex_idros r- rxsA   %$8###^E%:%:   r/c t|S)zE Fixture for tests on series with changing types of indices. rvros r-series_with_simple_indexr{s %  r/cgdgdg}t|}tj|}tjd}t ||}tj|d<|S)z: Fixture with a Series with a 2-level MultiIndex. )rrrrrrrr)rrrrrrrrrzr )rrr_rerrrNaN)arraysrDrorssers r-series_with_multilevel_indexrsn A@@@@@F&\F  "6 * *E 9??1  D U # # #C VCF Jr/cLi|]!}|jdtjd|"S)z-seriesr")r6r0)rrmake_rand_series)rr0s r-rxrxsE  ~ 3E J J Jr/cNt|jdS)z| Fixture for tests on indexes, series and series with a narrow dtype copy to avoid mutation, e.g. setting .name TrZ)_index_or_series_objsr|rNr}s r-index_or_series_objrs" ! / 4 4$ 4 ? ??r/chttjdS)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] r3)rr getSeriesDatarrmr/r- int_framers)4 R%'' ( ( / / 8 88r/cBttjS)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] )rrgetTimeSeriesDatarmr/r-datetime_framer:s4 R)++ , ,,r/cBttjS)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] )rrrrmr/r- float_framerWs4 R%'' ( ((r/c tdddtjdgdzdtjdgdzdd tjd S) z Fixture for DataFrame of float/int/string columns with RangeIndex Columns are ['a', 'b', 'c', 'float32', 'int32']. ?r rrr8r/r]r2)r"r#cr8r2rz)rrerarangermr/r-mixed_type_framertsk x )<<<XqcBhg666   imm    r/ctdddtdddtdddtdddtdddtdddtdddtdddtdddtdddg }ttjt ||S)zF Fixture for Series with a DatetimeIndex that has duplicates. ir]r r r\rz)rrrerrrq)datess r-(rand_series_with_duplicate_datetimeindexrs q!q!q!q!q!q!q!q!q!q! E ")//#e**--U ; ; ;;r/)rrrg?g?z2012-01M)freqz period[M]z 2012-02-01Dz period[D]z 2011-01-01r%i)secondsztimedelta64[ns]c|jSrr{r}s r-ea_scalar_and_dtyper  =r/)__add____radd____sub____rsub____mul____rmul__ __floordiv__ __rfloordiv__ __truediv__ __rtruediv____pow____rpow____mod____rmod__c|jS)zD Fixture for dunder names for common arithmetic operations. r{r}s r-all_arithmetic_operatorsrrr/c|jS)zU Fixture for operator and roperator arithmetic, comparison, and logical ops. r{r}s r-all_binary_operatorsr D =r/c|jS)z Fixture for operator and roperator arithmetic functions. Notes ----- This includes divmod and rdivmod, whereas all_arithmetic_operators does not. r{r}s r-all_arithmetic_functionsrs 6 =r/) countrmaxminmeanprodstdvarmediankurtskewsemc|jS)z. Fixture for numeric reduction names. r{r}s r-all_numeric_reductionsrrr/allanyc|jS)z. Fixture for boolean reduction names. r{r}s r-all_boolean_reductionsr*rr/c|jS)z> Fixture for all (boolean + numeric) reduction names. r{r}s r-all_reductionsr5rr/c|jS)z; Fixture for operator module comparison functions. r{r}s r- comparison_opr=rr/)__le____lt____ge____gt__c|jS)zm Fixture for dunder names for compare operations except == and != * >= * > * < * <= r{r}s r-compare_operators_no_eq_nerNrr/)__and____rand____or____ror____xor____rxor__c|jS)zY Fixture for dunder names for common logical operations * | * & * ^ r{r}s r-all_logical_operatorsr[  =r/)cumsumcumprodcummincummaxc|jS)z0 Fixture for numeric accumulation names r{r}s r-all_numeric_accumulationsrlrr/c,|dS)zO Returns the configuration for the test setting `--strict-data-files`. r*)rC) pytestconfigs r-strict_data_filesrws  ! !"7 8 88r/rCallable[..., str]ctjtjtdfd}|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. testsctjjg|R}tj|s-rt d|dt jd|d|S)NzCould not find file z and --strict-data-files is set.zCould not find .)osr2joinexists ValueErrorr9rH)rr2 BASE_PATHrs r-decozdatapath..decos}w|I----w~~d## 3   Q4QQQ K1$111 2 2 2 r/)rr2rdirname__file__)rrrs` @r-datapathrsN& RW__X66@@I Kr/cBtj|ddddS)z* The iris dataset as a DataFrame. iorscsvziris.csv)rfread_csv)rs r-irisrs$ ;xxfeZ@@ A AAr/z Asia/Tokyozdateutil/US/Pacificzdateutil/Asia/Singaporez+01:15z-02:15z UTC+01:15z UTC-02:15i,i)hoursr\rrc,g|]}t|Srmrs)ris r-rrs+++AQ+++r/c|jS)zD Fixture for trying timezones including default (None): {0} r{r}s r-tz_naive_fixturerrr/c|jS)z4 Fixture for trying explicit timezones: {0} r{r}s r-tz_aware_fixturerrr/rz dateutil/UTCc|jS)zQ Fixture to provide variants of UTC timezone strings and tzinfo objects. r{r}s r- utc_fixturerrr/c|jS)zR Parametrized fixture for string dtypes. * str * 'str' * 'U' r{r}s r- string_dtyper  =r/pyarrowc|jS)za Parametrized fixture for string dtypes. * 'string[python]' * 'string[pyarrow]' r{r}s r-nullable_string_dtyper  =r/pythonc|jSzb Parametrized fixture for pd.options.mode.string_storage. * 'python' * 'pyarrow' r{r}s r-string_storager rr/numpy_nullablec|jSrr{r}s r- dtype_backendr rr/c|jS)zK Parametrized fixture for bytes dtypes. * bytes * 'bytes' r{r}s r- bytes_dtyper .  =r/c|jS)zN Parametrized fixture for object dtypes. * object * 'object' r{r}s r- object_dtyper9r r/objectc|jS)zo Parametrized fixture for string dtypes. * 'object' * 'string[python]' * 'string[pyarrow]' r{r}s r-any_string_dtyperDrr/c|jS)z\ Parametrized fixture for datetime64 dtypes. * 'datetime64[ns]' * 'M8[ns]' r{r}s r-datetime64_dtyperUr r/c|jS)z^ Parametrized fixture for timedelta64 dtypes. * 'timedelta64[ns]' * 'm8[ns]' r{r}s r-timedelta64_dtyper`r r/rc .tdddddddS)z- Fixture emits fixed Timestamp.now() ir]r^r )rrarhourminuter  microsecond)rrmr/r- fixed_now_tsrks*  1RR   r/c|jS)z^ Parameterized fixture for float dtypes. * float * 'float32' * 'float64' r{r}s r-float_numpy_dtyperurr/c|jS)zR Parameterized fixture for float dtypes. * 'Float32' * 'Float64' r{r}s r-float_ea_dtyper!r r/c|jS)z~ Parameterized fixture for float dtypes. * float * 'float32' * 'float64' * 'Float32' * 'Float64' r{r}s r-any_float_dtyper#rr/c|jS)zg Parameterized fixture for complex dtypes. * complex * 'complex64' * 'complex128' r{r}s r- complex_dtyper%rr/c|jS)z| Parameterized fixture for signed integer dtypes. * int * 'int8' * 'int16' * 'int32' * 'int64' r{r}s r-any_signed_int_numpy_dtyper'rr/c|jS)zx Parameterized fixture for unsigned integer dtypes. * 'uint8' * 'uint16' * 'uint32' * 'uint64' r{r}s r-any_unsigned_int_numpy_dtyper)rr/c|jS)z Parameterized fixture for any integer dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' r{r}s r-any_int_numpy_dtyper+rr/c|jS)z Parameterized fixture for any nullable integer dtype. * 'UInt8' * 'Int8' * 'UInt16' * 'Int16' * 'UInt32' * 'Int32' * 'UInt64' * 'Int64' r{r}s r-any_int_ea_dtyper-rr/c|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{r}s r- any_int_dtyper/s . =r/c|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{r}s r-any_numeric_ea_dtyper1s " =r/c|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{r}s r-any_numeric_ea_and_arrow_dtyper3s @ =r/c|jS)z~ Parameterized fixture for any signed nullable integer dtype. * 'Int8' * 'Int16' * 'Int32' * 'Int64' r{r}s r-any_signed_int_ea_dtyper57rr/c|jS)z Parameterized fixture for any (purely) real numeric dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' * float * 'float32' * 'float64' r{r}s r-any_real_numpy_dtyper7Ds $ =r/c|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{r}s r-any_real_numeric_dtyper9Ys ( =r/c|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}s r-any_numpy_dtyper;prr/c|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{r}s r-any_numeric_dtyper=s > =r/r"rmixedbytesacrCz mixed-integerr g@floatingrintegerzmixed-integer-floatdecimal datetime64z 2013-01-01z 2018-01-01r$20180101riirrBcT|j\}}tj|t}||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|rerr)r~inferred_dtyper]s r-any_skipna_inferred_dtyperIs2R%]NF XfF + + +F 6 !!r/ctjddddlm}ddlm}|}d|j_||S) zj Get an instance of IPython.InteractiveShell. Will raise a skip if IPython is not installed. IPythonz6.0.0) minversionr)InteractiveShell)Configz:memory:)rJ)r9 importorskipIPython.core.interactiveshellrMtraitlets.configrNHistoryManager hist_file)rMrNrs r-iprT sj  g6666>>>>>>('''''A!+A  1 % % %%r/)bsrcoocsccsrdiadoklilc>ddlm}t||jdzS)z- Yields scipy sparse matrix classes. r)sparse_matrix)scipyr]getattrr|)r~r]s r-spmatrixras-  67=94 5 55r/cg|]Y}tttj|tjjr |dk?ttj|ZS)Tick) issubclassr`rfoffsetsrc)ros r-rr's[    gbj!,,bjo > > DE;;  ACN;;r/c|jS)zN Fixture for Tick based datetime offsets available for a time series. r{r}s r- tick_classesrh&rr/c|Srrmrus r-rwrw4sr/c|jS)zk Simple fixture for testing keys in sorting methods. Tests None (no key) and the identity key. r{r}s r- sort_by_keyrk4rr/c#Ktjdddlm}ddlm}ddlm}Gdd|}|d|d |V|ddd|j d<|j dS) Nfsspecr)register_implementation)MemoryFileSystem) _registryc*eZdZdZdgZdfd ZxZS) fsspectest..TestMemoryFStestmemNr r!cv|dd|jd<tjdi|dS)Ntestrrm)poprusuperr)rr __class__s r-rz)fsspectest..TestMemoryFS.__init__Hs?!::fd33DIaL EGG  & &v & & & & &r/r)rrrprotocolrur __classcell__)rxs@r- TestMemoryFSrrDsIv ' ' ' ' ' ' ' ' ' 'r/r{rsT)clobber) r9rOrmrnfsspec.implementations.memoryrofsspec.registryrprvrustoreclear)rnroregistryr{s r- fsspectestr=s !!!......>>>>>>555555'''''''''I|TBBBB ,.. LLD!!!Lar/)rNN)EgonVenkmanN)NCC1701Drr#tuple[Hashable, Hashable, Hashable]c|jS)zP A 3-tuple of names, the first two for operands, the last for a result. r{r}s r-r r Ss =r/c|jS)zI Parametrize over __setitem__, loc.__setitem__, iloc.__setitem__ r{r}s r- indexer_slirfrr/c|jS)z< Parametrize over loc.__getitem__, iloc.__getitem__ r{r}s r- indexer_lirnrr/c|jS)z8 Parametrize over __setitem__, iloc.__setitem__ r{r}s r- indexer_sirvrr/c|jS)z7 Parametrize over __setitem__, loc.__setitem__ r{r}s r- indexer_slr~rr/c|jS)z: Parametrize over at.__setitem__, loc.__setitem__ r{r}s r- indexer_alrrr/c|jS)z< Parametrize over iat.__setitem__, iloc.__setitem__ r{r}s r- indexer_ialrrr/boolc6tjjjdkS)z> Fixture to check if the array manager is being used. r)rfoptionsmode data_managerrmr/r-using_array_managerrs :? '7 22r/cbtjjjotjjjdkS)z7 Fixture to check if Copy-on-Write is enabled. block)rfrr copy_on_writerrmr/r-using_copy_on_writers" :? ( TRZ_-IW-TTr/z Europe/Warsawzdateutil/Europe/Warsawc|jS)zE tzinfo for Europe/Warsaw using pytz, dateutil, or zoneinfo. r{r}s r-warsawrrr/r)r0r1r2r3r4r3r r!)r r)r r)r r)rr3r r)r r)r r)r r)r r3(.__doc__ __future__r collectionsrrrrrr rDr operatorrtypingr r r dateutil.tzrr hypothesisrstnumpyrer9pytzrrpandas.util._test_decoratorsutil_test_decoratorstdpandas.core.dtypes.dtypesrrpandasrfrrrrrrrpandas._testing_testingr pandas.corerpandas.core.indexes.apirrrpa has_pyarrow ImportErrorzoneinfoPY39ZoneInfoZoneInfoNotFoundErrorr.r<rTsettingsregister_profile HealthChecktoo_slow load_profilesplitr6r`tseriesreclsregister_type_strategybuildsintegersbooleansfixturerBrnr axis_framerrrrrrrrr| skip_if_norrrrr NULL_OBJECTSrnulls_fixture2nanNaTrunique_nulls_fixture2NP_NAT_OBJECTSrnp_nat_fixture2rrindex_or_series2rrrbox_with_array2rrrrrr r*makeStringIndex makeDateIndexmakePeriodIndexmakeTimedeltaIndexmakeRangeIndex makeIntIndex makeUIntIndexmakeFloatIndex makeBoolIndexrrrrmakeNumericIndexmakeCategoricalIndexmakeIntervalIndexr_rrrrMidxkeysroindex_fixture2rIrT index_flat2rbrgrkrort_seriesr{rNARROW_NP_DTYPES_narrow_seriesrrrrrrrr_all_arithmetic_operatorsraddraddsubrsubmulrmultruedivrtruedivfloordiv rfloordivmodrmodpowrpoweqneltlegtgeand_rand_xorrxoror_ror_rr_all_numeric_reductionsr_all_boolean_reductionsr_all_reductionsrrrr_all_numeric_accumulationsrrrr TIMEZONESextend TIMEZONE_IDSparametrize_fixture_docr3rrtz_aware_fixture2_UTCSappendr utc_fixture2 STRING_DTYPESrrrr string_storage2 BYTES_DTYPESr  OBJECT_DTYPESrrDATETIME64_DTYPESrTIMEDELTA64_DTYPESrrFLOAT_NUMPY_DTYPESrFLOAT_EA_DTYPESr!ALL_FLOAT_DTYPESr#COMPLEX_DTYPESr%SIGNED_INT_NUMPY_DTYPESr'UNSIGNED_INT_NUMPY_DTYPESr)ALL_INT_NUMPY_DTYPESr+ALL_INT_EA_DTYPESr-ALL_INT_DTYPESr/r1ALL_INT_PYARROW_DTYPES_STR_REPRFLOAT_PYARROW_DTYPES_STR_REPRr3SIGNED_INT_EA_DTYPESr5ALL_REAL_NUMPY_DTYPESr7ALL_REAL_DTYPESr9ALL_NUMPY_DTYPESr;ALL_NUMERIC_DTYPESr=NArE_any_skipna_inferred_dtypery_rIrTra__all__rhrkrr setitemlocilocrrrratriatrrrwarsawsrrmr/r-r-s#&#"""""  ''''''  *))))))))   KK KKK  ; OOO %     )    IIII$-M-M-M-Mb $$%1:<%     &&& 8 = = ? ?D '"*$d + +CB YRYskbk#r22kbkmm L L L 4 9 9 ; ;  D '"*$d + +CB  bk"a  !bkmm"+R888     @ E E G G  D '"*$d + +CB  bk#r""!bkmm%"+R@@@    !!!!11111117R7RSSSTS 9~+F+FGGGHG***+++  ,+ ***+++,+e}%%%&%///00010;;;<<<=<;;;<<<=<;;;<<<=<      V=2=#=#=>>>           V=2=#=#=>>>       e}%%%&%:::;;;<; K011121r,F,FGGGHGbfbf-.../.-r(.H.HIIIJI! 6*+++,+vWh,?@@@  A@ #vrx06R6R6RSSSTSvy"(;AUAUVVVWV!   0   (!  b %%!  %%!#2#CL999!  b %% ! &&s++ !  R s # # ! OBOCv . . .! _R_S 0 0 0! _R_S 0 0 0! _R_S 0 0 0! R c 1 1 1! bs(333! bs(333! bs(333! r I666!  r I666!!"#2#B''..v66#!$% ++a/00$$S :::AA+NN%"%c;;;BB<PP*2*3//$$S)) U2YY$j$SS)>)>)> %J%JKK88::   ! !u'''((E)")C..888U929S>>:::eIBIcNN)<<<U929S>>0066iHHHU828$6B$6s$;$;CSTTTUUA!!! D) %++C008IJJJ K KC%(L!"|((**+++ . .,+ .  *0022    $$  $   &,,..        :            ;;;  '--//    !!!!     $ F<E7EnE,1133444@@54@99998----8))))8"<<<<. q " " "MM'7$C$CD s# & & & i(I(IJ   $ $ $k2  3 ' ' '5 Il| 4 4 4 O| , , ,  3   !23      $011121                        5 <=<             $  %$     .///0/!%..///0/*,CC'''('            ???@@@  A@  N N NGFF122232999BBBBB     EGG GIIKKNNK L HYYQ      HYYR u---%  (  'h' 557Hx7H7O7OPQQQ+++++ CC --..yl33343/.CC QRR 01122y},qrr*:;;;<;32% UUWWhl;  LL""5))***u r'((()( &mbmI.F.FGGG     YmbmI&>&>???     YmbmI&>&>???    !r'''('r'((()( &mbmI.F.FGGG r+,,,-,r,---.-r,---.-r)***+*r*+++  ,+ r()))*)r1222  32 r3444  54 r.///0/"r+,,,  -,  r()))*)2r+b.@@AAABA*  ( ) & '   :r.///  0/ r/00010(r)***+*,r*+++!!,+!Hr,---.-FRVS!" RUC ! sBFC ! tRVT"# rvrvrv&' bMsBFA&' sBFC !#rvs#$BFAQ,-RVWWQZZ01rvu%&rue$%MBM,//|9T9TUV))J'':1F1FGH dd4AT1a(8(8 9: 99Q<<167 dd1ggrvttAww '( t bfffTll34((1a.."&((1a..9:36 ( )Q1s;;;,","<;,"d&&&$HHHIII66JI6  #   kk*+++,+*!,    ru       BFBG455565()))*) BG,---.- BF+,,,-,rv'''('()))*)3333UUUU 4 5  NN$8$_55666w s$BB%$B%6C CC