Wh&&-dZddlmZddlmZmZmZmZmZddlm Z ddl Z ddl Z ddl m Z ddlmZmZddlZddlmZddlZddlZdd lmZmZddlmcmZdd lmZm Z ddl!Z"dd l!m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(ddl)m*Z+dd l,m-Z-dd l.m/Z/m0Z0 ddl1Z2[2dZ3n #e4$rdZ3YnwxYwdZ5e"j6j7r(ddl5Z5 e5j8dn#e5j9$rdZ5YnwxYwej:;dZ<d2dZ=dej>de?de?ddfdZ@dZAejBCddejDjEfejBFddGD]VZHeIe"jJjKeHZLejMeLejNeLejOddejPWdGD]gZHeIe"jJjKeHZLejMeLejNeLejOd d!ejPejOd"d#$%hd&GD]gZHeIe"jJjKeHZLejMeLejNeLejOd'd(ejPejOd"d#$)hejQd2d*ZRejQd+d2d,ZSejQgd-d./d0ZTeTZUejQd"d1gd2/d3ZVejQgd45d6ZWejQgd45d7ZXejQgd85d9ZYejQgd:5d;ZZejQgd<5d=Z[ejQgd<5d>Z\ejQdd?d@dAdBdCej]dDej^dEFg5dGZ_ejQd?d@dAdBdCej]dDej^dEFg5dHZ`ejQddg5dIZaejQgdJ5dKZbejQdLdMg5dNZcejQe+jddO/dPZeeeZfejQdejge"jhg5dQZieiZjejQe+jkdR/dSZlelZmejQe#e&g5dTZnejQe/e&gdUdVg/dWZoeoZpejQe/e&e"jqggdX/dYZrejQe/e&e#e"jqgdZ/d[ZsesZtejQd\ZuejQd]ZvejQd^ZwejQde0fd_ZxejQde#fd`ZydaZzdbZ{idce+j|dddee+j}dddfe+j}dddghdie+j~dddje+jdddke+jdddle+jdddme+jdddne+jdddoe+jdddodpe+jdddpdqe+jdddrsdte+jdddusdve+jdddwsdxe+jdddysdze+jddd{sd|e+jddd}sid~e+jdddsde+jdddsde+jdddsde+jdddsde+jdede/ejddkde+jddde+jddde/gde0jegdgdde{dezde/gdde/ejdddsde/ejdddsde/ejdddsde/ejddedsde/e"jqe+j|dddsiZe3r.e/e"jqe+j|dddsZeed<ejQe5dZeZejQdeD5dZeZejQdeD5dZejQde&fdZejQde&fdZejQde&fdZdZdeDZejQde&fdZejQde&fdZde+jDZieeeZejQe5dZejQde#fdZejQde#fdZejQde#fdZejQde#fdZejQde&fdZejQe$dd!e drdfe$dde ddfe%dddfe%dddfe(ddhedhfe'ddfg5d„ZgdâZejQe5dĄZejQe 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-jg5dńZejQe je-je je-je je-je je-je je-je je-je je-jg5dƄZgdǢZejQeȬ5dȄZddgZejQeʬ5d˄ZeezZejQe̬5d̄ZejQe je je je je je jg5d̈́ZejQgd΢5dτZejQgdТ5dфZejQd҄ZejQde?de de?ffdՄZejQde#fdքZddddddddddeeedަededߦejeed"eeddgZe53eԠe5j8dge5j8dgdeDZeje?e֦ejQee֬/dZeje?ed"dejQed"ded"d/dZeZddeeejgZe5#e۠e5j8dejQe۬5dZeZejQe+j5dZejQdej]dej^ddFg5dZejQdej]dej^ddFg5dZeZejQe+j5dZejQe+j5dZejQddej]dej^ddFg5dZejQe+j5dZejQe+j5dZejQde(fdZejQe+j5dZejQe+j5dZejQe+je+jz5dZejQe+j5dZejQe+j5dZejQe+j5dZejQe+j5dZejQe+j5dZejQe+je+jz5dZejQe+je+jz5dZejQe+j5dZejQe+j5dZejQe+j5dZejQe+je+jze+jze+jz5dZdcdejgdgfdcde"jdgfdde"jhdgfd d ejgd gfdejgejgejggfdgfd dejgd gfddejgdgfddejgdgfdd"ejgd gfdd"ejgdgfde d"ejge d gfddejgdgfdde"jdgfdejdejgejdgfdee(dejge(dgfdedd"d"ejgedd"d"gfdjed"ejged gfded"ejged gfdie%de"jhe%dgfde$dd"ejge$dd gfgZee\Z Z ejQee /dZ ejQdZ ejQgd5d Z ejQd!e"jKjD5d"ZejQdd#g5d$ZejQd%ZejQd&d'd(ejgejgejgfejge"jhdfejge"jdfe"je"je"jfg5d)ZejQe+je+je+jg5d*ZejQe+je+jg5d+ZejQe+je+jg5d,ZejQe+je+jg5d-ZejQe+je+jg5d.ZejQe+je+jg5d/ZejQd0ZejQdefd1ZdS(3a 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)Callable)tzlocaltzutc) strategies) FixedOffsetutc)DatetimeTZDtype IntervalDtype) DataFrameIntervalPeriodSeries Timedelta Timestamp)ops)Index MultiIndexTFUTCz3ignore:Promotion of numbers and bools:FutureWarningreturnc&|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 U/var/www/html/movieo_spanner_bot/venv/lib/python3.11/site-packages/pandas/conftest.pypytest_addoptionr)fs ]<>OPPP %lAUVVV [OLLL L7N ]<>STTT  ?itempathmessagec|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)r+r,r-s r(ignore_doctest_warningr5usZ y$I  223FW3F3FGGHHHHHIIr*c|d}|d}|d}|d}tjjd|dftjjd|dftjjd|dfg}gd }|D]_}|d s|d d rK|tjd|D]\} } t|| | d|j vr$|tjj |t|D]^\} } } }| |j vrN| r7|tj d|n|| _|r;d|j vr2|tj dadS)Nrr$r"r#slownetworkz --networkdb) )zDataFrame.appendz%The frame.append method is deprecated)z Series.appendz&The series.append method is deprecated)zdtypes.common.is_categoricalzis_categorical is deprecated)zCategorical.replacez!Categorical.replace is deprecated)zdtypes.common.is_extension_typez!'is_extension_type' is deprecated)zIndex.is_mixedzIndex.is_mixed is deprecated)zMultiIndex._is_lexsortedz%MultiIndex.is_lexsorted is deprecated)zmissing.mask_zero_div_zerozdivide by zero encountered)z$_validators.validate_axis_style_argszUse named argumentsz--doctest-modulesz--doctest-cythonF)defaultadd_doctest_importsz/frame/zskipping due to zskipping due to --only-slow) getoptionr2r3r7r8r9r1 usefixturesr5nodeid arraymanager suppress_npdev_promotion_warningkeywordsskip)itemsconfig skip_slow only_slow skip_networkskip_dbmarksignored_doctest_warningsr+r,r-r3kwd skip_if_foundarg_names r(pytest_collection_modifyitemsrNs%  //I  //I##$455L{++G  69m<  i{C w 4 E   MM   / 0 0 rq sARaARARr*)paramsidsc|jS)z@ Fixture for returning the axis numbers of a DataFrame. paramrequests r(axisry  =r*rjc&dt|Srlrmros r(rqrqs5FT!WW5F5Fr*c|jS)zA Fixture for returning aliases of axis 1 of a DataFrame. rurws r(axis_1r}rzr*)TFN)rrc|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). rurws r(observedr  =r*c|jS)z6 Boolean 'ordered' parameter for Categorical. rurws r(orderedr)rzr*)firstlastFc|jS)z[ Valid values for the 'keep' parameter used in .duplicated or .drop_duplicates rurws r(keepr1 =r*)bothneitherleftrightc|jS)zA Fixture for trying all interval 'inclusive' parameters. rurws r(inclusive_endpoints_fixturer:rzr*)rrrrc|jS)z< Fixture for trying all interval closed parameters. rurws r(closedrBrzr*c|jS)zS Secondary closed fixture to allow parametrizing over all pairs of closed. rurws r( other_closedrJrzr*gzipbz2zipxztarzstd zstandard)rIc|jS)zK Fixture for trying common compression types in compression tests. rurws r( compressionrR  =r*c|jS)zk Fixture for trying common compression types in compression tests excluding uncompressed case. rurws r(compression_onlyrdrr*c|jS)z, Fixture that an array is writable. rurws r(writablervrzr*)innerouterrrc|jS)z: Fixture for trying all types of join operations. rurws r( join_typer~rzr*nlargest nsmallestc|jS)z1 Fixture for trying all nselect methods. rurws r(nselect_methodrrzr*c*t|jSNtype__name__ros r(rqrqsd1gg6Fr*c|jS)z/ Fixture for each null type in pandas. rurws r( nulls_fixturerrzr*c|jS)zL Fixture for each null type in pandas, each null type exactly once. rurws r(unique_nulls_fixturerrzr*c*t|jSrrros r(rqrqsQ8Hr*c|jS)z- Fixture for each NaT type in numpy. rurws r(np_nat_fixturerrzr*c|jS)z; Fixture to parametrize over DataFrame and Series. rurws r(frame_or_seriesrrzr*riseriesc|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 rurws r(index_or_seriesrs  =r*)rirarrayc|jS)zG Fixture to parametrize over Index, Series, and ExtensionArray rurws r(index_or_series_or_arrayrrzr*c|jSr)rros r(rqrqs1:r*c|jS)z] Fixture to test behavior for Index, Series, DataFrame, and pandas Array classes rurws r(box_with_arrayrrr*c,Gddt}|S)z, Fixture for a dictionary subclass. ceZdZddZdS)"dict_subclass..TestSubDictrNc0tj|g|Ri|dSr)dict__init__)selfargskwargss r(rz+dict_subclass..TestSubDict.__init__s( M$ 0 0 0 0 0 0 0 0 0r*rN)r __module__ __qualname__rrgr*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(eZdZddZdZdZdZdS)5non_dict_mapping_subclass..TestNonDictMappingrNc||_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*c4|jSr)r__iter__rs r(rz>non_dict_mapping_subclass..TestNonDictMapping.__iter__s:&&(( (r*c4|jSr)r__len__rs r(rz=non_dict_mapping_subclass..TestNonDictMapping.__len__s:%%'' 'r*r)rrrrrrrrgr*r(TestNonDictMappingrsU ) ) ) ) / / / ) ) ) ( ( ( ( (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)yearros r(rqzAmultiindex_year_month_day_dataframe_random_data..sr*c|jSr)r[ros r(rqzAmultiindex_year_month_day_dataframe_random_data..s17r*c|jSr)dayros r(rqzAmultiindex_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*)rr[rT)inplace)tmmakeTimeDataFramegroupbysumri set_levelslevels set_names)tdfymds r(/multiindex_year_month_day_dataframe_random_datars s # #C ++''):):OOL M M Q Q S SC $$%S%S#)BR%S%S%STTCII000$??? Jr*cBtgdgdggdgdgddgS)z; 2-level MultiIndex, lexsorted, with string names. foobarbazquxonetwothree) rrrrWrWrrr) rrWrrrWrWrrrWrrsecond)rcodesnames)rrgr*r(%lexsorted_two_level_string_multiindexrsM ,,,.E.E.EF---/M/M/MN!   r*c|}ttjdd|t gddS)z2DataFrame with 2 level MultiIndex with random data r)ABCexpr/)rirj)rr_randomrandnr)rris 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)rrrWrrr)rrWrrWrrWrrF)rrrverify_integrity)rr_rr) major_axis minor_axis major_codes minor_codes index_namess r(_create_multiindexr4s 33344Ju~&&J(---..K(---..KH%K J'K(    r*c jtjddgddgtjdddggd  S) zB MultiIndex with a level that is a tzaware DatetimeIndex. rWrab20130101r US/Eastern)periodstzr)r)r from_productr` date_rangergr*r(_create_mi_with_dt64tz_levelr!HsJ  " Q#sR]:q\RRRS%%%   r*stringrrz datetime-tzz US/Pacific)rperiodrintuintrangefloat complex64 complex128 num_int64int64dtype num_int32int32 num_int16int16num_int8int8 num_uint64uint64 num_uint32uint32 num_uint16uint16 num_uint8uint8 num_float64float64 num_float32float32z bool-objectr bool-dtype categoricalintervalemptytuples)rrr)rWrrmi-with-dt64tz-levelmultirepeats)rrrWrWrr nullable_intInt64 nullable_uintUInt16nullable_floatFloat32 nullable_boolbooleanz string-pythonzstring[python]zstring[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_dictrvcopyrws r(riri}s  & + + - --r*cRg|]$}tt|t"|%Srg) isinstancerQrrrs r(rrs<   :l36G+T+T     r*cN|j}t|S)z8 index fixture, but excluding MultiIndex cases. )rvrQrR)rxrs r( index_flatrWs" -C   ! ! # ##r*cg|]=}|dv|dtt|t;|>S))r$r%r&rCrGr@num_) startswithrTrQrrUs r(rrs`     M M M~~f%% N<,j99 N  N M Mr*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)rDrErFrrrWN)rQrvrRvaluestolistr from_tuplesr)rxindvalss r(index_with_missingrds. w} % * * * 5 5C :D}CCCzz||DGABBK'QT"Xabb\)R%d+++QRtCyyr*c:tj}d|_|S)zC Fixture for Series of floats with Index of unique strings r)rmakeStringSeriesr/ss r( string_seriesris A AF Hr*c:tj}d|_|S)zI Fixture for Series of dtype object with Index of unique strings objects)rmakeObjectSeriesr/rgs r( object_seriesrms A AF Hr*c:tj}d|_|S)z9 Fixture for Series of floats with DatetimeIndex ts)rmakeTimeSeriesr/rgs r(datetime_seriesrqs A AF Hr*ct|}tj|}t ||dS)zHelper for the _series dictr)rir/)lenr_r rr)risizedatas r(_create_seriesrvs5 u::D 9??4 D $e# . . ..r*c<i|]\}}d|dt|S)z series-with-z-indexrv)rindex_idris r( rzsA   %$8###^E%:%:   r*c t|S)zE Fixture for tests on series with changing types of indices. rxris 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)rrrrrrrrr|r)rrrar_r rrNaN)arraysrDrirusers r(series_with_multilevel_indexrsn A@@@@@F&\F  "6 * *E 9??1  D U # # #C VCF Jr*cLi|]!}|jdtjd|"S)z-seriesr)r/r-)rrmake_rand_series)rr-s r(rzrz sE  ~ 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 Tr\)_index_or_series_objsrvrRrws 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] r+)rr getSeriesDatarrgr*r( int_framer"s)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] )rrgetTimeSeriesDatargr*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] )rrrrgr*r( float_framer\s4 R%'' ( ((r*c tdddtjdgdzdtjdgdzdd tjd S) z Fixture for DataFrame of float/int/string columns with RangeIndex Columns are ['a', 'b', 'c', 'float32', 'int32']. ?rrrr?r,rWr/)rrcr?r/r|)rr_rarangergr*r(mixed_type_framerysk x )<<<XqcBhg666   imm    r*ctdddtdddtdddtdddtdddtdddtdddtdddtdddtdddg }ttjt ||S)zF Fixture for Series with a DatetimeIndex that has duplicates. irWrrrVr|)rrr_r rrs)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-01ri)secondsztimedelta64[ns]c|jSrrurws 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. rurws r(all_arithmetic_operatorsrrzr*c|jS)zU Fixture for operator and roperator arithmetic, comparison, and logical ops. rurws 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. rurws r(all_arithmetic_functionsrs 6 =r*) rmaxminmeanprodstdvarmediankurtskewc|jS)z. Fixture for numeric reduction names. rurws r(all_numeric_reductionsr"rzr*allanyc|jS)z. Fixture for boolean reduction names. rurws r(all_boolean_reductionsr-rzr*c|jS)z> Fixture for all (boolean + numeric) reduction names. rurws r(all_reductionsr8rzr*c|jS)z; Fixture for operator module comparison functions. rurws r( comparison_opr@rr*)__le____lt____ge____gt__c|jS)zm Fixture for dunder names for compare operations except == and != * >= * > * < * <= rurws r(compare_operators_no_eq_nerQrr*)__and____rand____or____ror____xor____rxor__c|jS)zY Fixture for dunder names for common logical operations * | * & * ^ rurws r(all_logical_operatorsr^  =r*c,|dS)zO Returns the configuration for the test setting `--strict-data-files`. r%)r<) pytestconfigs r(strict_data_filesros  ! !"7 8 88r*r.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 .)osr,joinexists ValueErrorr2rB)rr, BASE_PATHrs r(decozdatapath..decos}w|I----w~~d## 7  7 Q4QQQ 5d555666 r*)rr,rdirname__file__)rrrs` @r(datapathrwsN& RW__X66@@I       Kr*cBtj|ddddS)z* The iris dataset as a DataFrame. iorucsvziris.csv)r`read_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^rr c,g|]}t|Srgrm)ris r(rrs+++AQ+++r*c|jS)zD Fixture for trying timezones including default (None): {0} rurws r(tz_naive_fixturerrr*c|jS)z4 Fixture for trying explicit timezones: {0} rurws r(tz_aware_fixturerrr*rz dateutil/UTCc|jS)zQ Fixture to provide variants of UTC timezone strings and tzinfo objects. rurws r( utc_fixturerrzr*c|jS)zR Parametrized fixture for string dtypes. * str * 'str' * 'U' rurws r( string_dtyper  =r*pyarrowz1.0.0) min_versionc|jS)za Parametrized fixture for string dtypes. * 'string[python]' * 'string[pyarrow]' rurws r(nullable_string_dtyperrr*pythonc|jS)zb Parametrized fixture for pd.options.mode.string_storage. * 'python' * 'pyarrow' rurws r(string_storagers  =r*c|jS)zK Parametrized fixture for bytes dtypes. * bytes * 'bytes' rurws r( bytes_dtyper  =r*c|jS)zN Parametrized fixture for object dtypes. * object * 'object' rurws r( object_dtyper$rr*objectc|jS)zo Parametrized fixture for string dtypes. * 'object' * 'string[python]' * 'string[pyarrow]' rurws r(any_string_dtyper/ =r*c|jS)z\ Parametrized fixture for datetime64 dtypes. * 'datetime64[ns]' * 'M8[ns]' rurws r(datetime64_dtyper Brr*c|jS)z^ Parametrized fixture for timedelta64 dtypes. * 'timedelta64[ns]' * 'm8[ns]' rurws r(timedelta64_dtyper Mrr*c .tdddddddS)z- Fixture emits fixed Timestamp.now() irWrXr )rr[rhourminuter microsecond)rrgr*r( fixed_now_tsrXs*  1RR   r*c|jS)z^ Parameterized fixture for float dtypes. * float * 'float32' * 'float64' rurws r(float_numpy_dtyperbrr*c|jS)zR Parameterized fixture for float dtypes. * 'Float32' * 'Float64' rurws r(float_ea_dtypernrr*c|jS)z~ Parameterized fixture for float dtypes. * float * 'float32' * 'float64' * 'Float32' * 'Float64' rurws r(any_float_dtyperyrr*c|jS)zg Parameterized fixture for complex dtypes. * complex * 'complex64' * 'complex128' rurws r( complex_dtyperrr*c|jS)z| Parameterized fixture for signed integer dtypes. * int * 'int8' * 'int16' * 'int32' * 'int64' rurws r(any_signed_int_numpy_dtyperrr*c|jS)zx Parameterized fixture for unsigned integer dtypes. * 'uint8' * 'uint16' * 'uint32' * 'uint64' rurws r(any_unsigned_int_numpy_dtyperrr*c|jS)z Parameterized fixture for any integer dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' rurws r(any_int_numpy_dtyper!rr*c|jS)z Parameterized fixture for any nullable integer dtype. * 'UInt8' * 'Int8' * 'UInt16' * 'Int16' * 'UInt32' * 'Int32' * 'UInt64' * 'Int64' rurws 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' rurws 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' rurws r(any_numeric_ea_dtyper's " =r*c|jS)z~ Parameterized fixture for any signed nullable integer dtype. * 'Int8' * 'Int16' * 'Int32' * 'Int64' rurws r(any_signed_int_ea_dtyper)rr*c|jS)z Parameterized fixture for any (purely) real numeric dtype. * int * 'int8' * 'uint8' * 'int16' * 'uint16' * 'int32' * 'uint32' * 'int64' * 'uint64' * float * 'float32' * 'float64' rurws r(any_real_numpy_dtyper+ s $ =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' rurws r(any_numpy_dtyper-!rr*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' rurws r(any_numeric_dtyper/Fs H =r*rrmixedbytesacz mixed-integerrg@floatingrintegerzmixed-integer-floatdecimal datetime64z 2013-01-01z 2018-01-01r20180101riircT|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 -------- >>> 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,)rvr_rr)rxinferred_dtyper_s r(any_skipna_inferred_dtyper;s2R%]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:)rD)r2 importorskipIPython.core.interactiveshellr?traitlets.configr@HistoryManager hist_file)r?r@rs r(iprFsj  g6666>>>>>>('''''A!+A  1 % % %%r*)bsrcoocsccsrdiadoklilc>ddlm}t||jdzS)z- Yields scipy sparse matrix classes. r)sparse_matrix)scipyrOgetattrrv)rxrOs r(spmatrixrSs-  67=94 5 55r*cg|]Y}tttj|tjjr |dk?ttj|ZS)Tick) issubclassrRr`offsetsrU)ros r(rrs[    gbj!,,bjo > > DE;;  ACN;;r*c|jS)zN Fixture for Tick based datetime offsets available for a time series. rurws r( tick_classesrZrr*c|Srrgros r(rqrqsr*c|jS)zk Simple fixture for testing keys in sorting methods. Tests None (no key) and the identity key. rurws r( sort_by_keyr]rr*c#Ktjdddlm}ddlm}ddlm}Gdd|}|d|d |V|ddd|j d<|j dS) Nfsspecr)register_implementation)MemoryFileSystem) _registryc*eZdZdZdgZdfd ZxZS) fsspectest..TestMemoryFStestmemNrcv|dd|jd<tjdi|dS)Ntestrrg)poprgsuperr)rr __class__s r(rz)fsspectest..TestMemoryFS.__init__s?!::fd33DIaL EGG  & &v & & & & &r*r)rrrprotocolrgr __classcell__)rjs@r( TestMemoryFSrdsIv ' ' ' ' ' ' ' ' ' 'r*rmreT)clobber) r2rAr_r`fsspec.implementations.memoryrafsspec.registryrbrhrgstoreclear)r`raregistryrms r( fsspectestrts !!!......>>>>>>555555'''''''''I|TBBBB ,.. LLD!!!Lar*)rNN)EgonVenkmanN)NCC1701Drwrwc|jS)zP A 3-tuple of names, the first two for operands, the last for a result. rurws r(rr rr*c|jS)zI Parametrize over __setitem__, loc.__setitem__, iloc.__setitem__ rurws r( indexer_slirzrzr*c|jS)z< Parametrize over loc.__getitem__, iloc.__getitem__ rurws r( indexer_lir|$rzr*c|jS)z8 Parametrize over __setitem__, iloc.__setitem__ rurws r( indexer_sir~,rzr*c|jS)z7 Parametrize over __setitem__, loc.__setitem__ rurws r( indexer_slr4rzr*c|jS)z: Parametrize over at.__setitem__, loc.__setitem__ rurws r( indexer_alr<rzr*c|jS)z< Parametrize over iat.__setitem__, iloc.__setitem__ rurws r( indexer_ialrDrzr*c6tjjjdkS)z> Fixture to check if the array manager is being used. r)r`optionsmode data_managerrgr*r(using_array_managerrLs :? '7 22r*cbtjjjotjjjdkS)z7 Fixture to check if Copy-on-Write is enabled. block)r`rr copy_on_writerrgr*r(using_copy_on_writerTs" :? ( TRZ_-IW-TTr*r( __doc__ collectionsrrrrrrr6r operatorrtypingr dateutil.tzr r hypothesisr stnumpyr_r2pytzrrpandas.util._test_decoratorsutil_test_decoratorstdpandas.core.dtypes.dtypesrrpandasr`rrrrrrpandas._testing_testingr pandas.corerpandas.core.indexes.apirrrpa has_pyarrow ImportErrorzoneinfocompatPY39ZoneInfoZoneInfoNotFoundErrorr3r4r@r)Itemstrr5rNsettingsregister_profile HealthChecktoo_slow load_profilesplitr/rRtseriesrWclsregister_type_strategybuildsintegersbooleansfixturer;rhry axis_framer}rrrrrrrv skip_if_norrrrr NULL_OBJECTSrnulls_fixture2nanNaTrunique_nulls_fixture2NP_NAT_OBJECTSrnp_nat_fixture2rrindex_or_series2rrrbox_with_array2rrrrrrr!makeStringIndex makeDateIndexmakePeriodIndexmakeTimedeltaIndex makeIntIndex makeUIntIndexmakeRangeIndexmakeFloatIndexrmakeNumericIndex makeBoolIndexrr rmakeCategoricalIndexmakeIntervalIndexrarrboolrQidxkeysriindex_fixture2rW index_flat2rdrirmrqrvrC_seriesr}rNARROW_NP_DTYPES_narrow_seriesrrrrrrrr_all_arithmetic_operatorsraddraddsubrsubmulrmultruedivrtruedivfloordiv rfloordivmodrmodpowrpoweqneltlegtgeand_rand_xorrxoror_ror_rr_all_numeric_reductionsr_all_boolean_reductionsr_all_reductionsrrrrrrr TIMEZONESextend TIMEZONE_IDSparametrize_fixture_docrrtz_aware_fixture2_UTCSappendr utc_fixture2 STRING_DTYPESrrrstring_storage2 BYTES_DTYPESr OBJECT_DTYPESrrDATETIME64_DTYPESr TIMEDELTA64_DTYPESr rFLOAT_NUMPY_DTYPESrFLOAT_EA_DTYPESrrCOMPLEX_DTYPESrSIGNED_INT_NUMPY_DTYPESrUNSIGNED_INT_NUMPY_DTYPESrALL_INT_NUMPY_DTYPESr!ALL_INT_EA_DTYPESr#r%r'SIGNED_INT_EA_DTYPESr)ALL_REAL_NUMPY_DTYPESr+ALL_NUMPY_DTYPESr-r/NAr7_any_skipna_inferred_dtypers_r;rFrS__all__rZr]rtrsetitemlocilocrzr|r~ratriatrrrrgr*r(rs#( ''''''  *))))))))   KK KKK 9> OOO %     )$*;#=#=9$$     IICI#I$IIII$8M8M8Mx $$%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***+++  ,+ ***+++,+///00010;;;<<<=<;;;<<<=<;;;<<<=<      V=2=#=#=>>>           V=2=#=#=>>>       e}%%%&%:::;;;<; K011121r,F,FGGGHGbfbf-.../.-r(.H.HIIIJI! 6*+++,+ 6?( 3   #vrx06R6R6RSSSTSvy"(;AUAUVVVWV!   0   z($  b %%$  %%$#2#CL999$  b %% $ &&s++ $  ?2?3   $ B S ! !$ R s # #$ R s # #$""3''..{;;$#"#C((// ==$$$S888$$$S888$$$S888$##Cv666$ %"%c:::!$"%"%c:::#$$$%"%c:::%$&$$S888'$(&2&s)<<<)$*&2&s)<<<+$,#2#B''..v66-$.%% ++a/00/$0*2*3//1$2$$S))3$4 UU2YY5$6 $j$SS)>)>)> %J%JKK7$888::9$:   ! !;$<uu'''((=$>EE)")C..888?$@UU929S>>:::A$BeeIBIcNN)<<>0066iHHHE$$FUU828$6B$6s$;$;CSTTTUUG$$ J) %++C008IJJJ K KC%(L!"|((**+++ . .,+ .  #    $$  $           : v     v         ///  '--//    !v!!!! f     $ F<E7EnE,1133444@@54@9999998- ----8)Y))))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 N999c(:DBiBBBB     EGG GIIKKNNK L HYYQ      HYYR u---%  (  'h' 557Hx7H7O7OPQQQ+++++ CC --..yl33343/.CC QRR 01122y},qrr*:;;;<;32% UUWWhl;  LL""5))***u r'((()( ]R]9'%R%R%R     YmbmI7&S&S&STTT    !r'''('r'((()( ]R]9'%R%R%R    r+,,,-,r,---.-ir,---.-r)***+*r,r/AABBB  CB r()))*)r1222  32 r3444  54 r.///0/"r+,,,  -,  r.1EEFFFGF2r+b.@@AAABA&r.///  0/ r/00010(r*+++!!,+!H #       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'''('()))*)333UTUUUUUUs$BBB/C CC