a
    5gh&                  '   @   st  d Z ddlmZ ddlmZmZmZmZmZ ddlm	Z	 ddl
Z
ddlZddlmZ ddlmZmZ ddlZddlmZ ddlZddlZdd	lmZmZ ddlm  mZ dd
lmZm Z  ddl!Z"ddl!m#Z#m$Z$m%Z%m&Z&m'Z'm(Z( ddl)m*Z+ ddl,m-Z- ddl.m/Z/m0Z0 zddl1Z2W n e3y.   dZ4Y n0 [2dZ4dZ5e"j6j7rvddl5Z5ze58d W n e5j9yt   dZ5Y n0 ej:;dZ<ddddZ=ej>e?e?ddddZ@dd ZAejBjCddejDjEfd ejBFd dG D ]8ZHeIe"jJjKeHZLeMeLejNeLeOddeP d qd G D ]DZHeIe"jJjKeHZLeMeLejNeLeOd!d"eP ejOd#d$d%d& qd'G D ]DZHeIe"jJjKeHZLeMeLejNeLeOd(d)eP ejOd#d$d%d* qjejQddd+d,ZRejQdd-ddd.d/ZSejQg d0d1d2 d3d4d5 ZTeTZUejQd#d6gd7d2 d3d8d9 ZVejQg d:d;d<d= ZWejQg d:d;d>d? ZXejQg d@d;dAdB ZYejQg dCd;dDdE ZZejQg dFd;dGdH Z[ejQg dFd;dIdJ Z\ejQddKdLdMdNdOej]dPe^dQdRgd;dSdT Z_ejQdKdLdMdNdOej]dPe^dQdRgd;dUdV Z`ejQddgd;dWdX ZaejQg dYd;dZd[ ZbejQd\d]gd;d^d_ ZcejQe+jdd`d2 d3dadb ZeeeZfejQdejge"jhgd;dcdd ZieiZjejQe+jkded2 d3dfdg ZlelZmejQe#e&gd;dhdi ZnejQe/e&gdjdkgd3dldm ZoeoZpejQe/e&e"jqgg dnd3dodp ZrejQe/e&e#e"jqgdqd2 d3drds ZsesZtejQdtdu ZuejQdvdw ZvejQdxdy ZwejQe0ddzd{ZxejQe#dd|d}Zyd~d Zzdd Z{e+|de+}de+j}ddde+~de+de+de+de+de+de+dde+dde+jddde+jddde+jddde+jddde+jddde+jddde+jddde+jddde+jddde+jddde+dee/ejddk e+de+de/g e0eg dg de{ ez e/g de/eddde/eddde/eddde/ededde/e"jqe+|dddd#Ze4rfe/e"jqe+|dddZeed< ejQe d;ddj ZeZejQdd eD d;dd ZeZejQdd eD d;dd ZejQe&dddZejQe&dddZejQe&dddZdd Zdd e D ZejQe&dddZejQe&dddZdd e+jD Zi eeeZejQe d;dd ZejQe#dddZejQe#dddZejQe#dddZejQe#dddZejQe&dddZejQe$dd"de ddfe$ddde ddfe%dddǍdfe%dddǍdfe(dddeddfe'ddύdfgd;dd҄ Zg dӢZejQed;ddՄ ZejQe
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-jgd;ddׄ ZejQe
je-je
je-je
je-je
je-je
je-je
je-je
je-jgd;ddل Zg dڢZejQed;dd܄ ZddgZejQed;dd Zee ZejQed;dd ZejQe
je
je
je
je
je
jgd;dd ZejQg dd;dd ZejQg dd;dd ZejQdd ZejQe?ede?f dddZejQe#dddZdddddddddde e edededejeed#deeddddgZe5dureԠe58de58dg d d eD Zee?eփejQeed3dd Zee?ed#d ejQed#d ed#d d3dd ZeZڐddee ejgZe5durJe۠e58d ejQed;dd ZeZejQe+jd;d	d
 ZejQdej]dej^ddddRgd;dd ZejQdej]dej^ddddRgd;dd ZeZejQe+jd;dd ZejQe+jd;dd ZejQddej]dej^ddddRgd;dd ZejQe+jd;dd ZejQe+jd;dd ZejQe(dddZejQe+jd;d d! ZejQe+jd;d"d# ZejQe+je+j d;d$d% ZejQe+jd;d&d' ZejQe+jd;d(d) ZejQe+jd;d*d+ ZejQe+jd;d,d- ZejQe+jd;d.d/ ZejQe+je+j d;d0d1 ZejQe+je+j d;d2d3 ZejQe+jd;d4d5 Z ejQe+jd;d6d7 ZejQe+jd;d8d9 ZejQe+je+j e+j e+j d;d:d; Zd<d=ejgd>gfd<d=e"jd>gfd?d=e"jhd>gfd@dAejgdBgfdCejgejgejggfdCg fdDd=ejgdEgfd?d=ejgdFgfdGdHejgdFgfdId#ejgdEgfdJd#ejgdFgfdKe	d#ejge	dEgfddejgdgfdde"jdgfdLedMejgedNgfdOe(dPejge(dQgfdRedSd#d#ejgedTd#d#gfdUed#ejgedEgfdVed#ejgedEgfdWe%dSe"jhe%dTgfdXe$dd#ejge$ddEgfgZee \Z	Z
ejQee	d3dYdZ ZejQd[d\ ZejQg d]d;d^d_ ZejQd`d e"jKjD d;dadb ZejQddcd2 gd;ddde ZeQ dfdg ZejQdhdidjejgejgejgfejge"jhdfejge"jdfe"je"je"jfgd;dkdl ZejQe+je+je+jgd;dmdn ZejQe+je+jgd;dodp ZejQe+je+jgd;dqdr ZejQe+je+jgd;dsdt ZejQe+je+jgd;dudv ZejQe+je+jgd;dwdx ZejQdydz ZejQedd{d|ZdS (}  a  
This file is very long and growing, but it was decided to not split it yet, as
it's still manageable (2020-03-17, ~1.1k LoC). See gh-31989

Instead of splitting it was decided to define sections here:
- Configuration / Settings
- Autouse fixtures
- Common arguments
- Missing values & co.
- Classes
- Indices
- Series'
- DataFrames
- Operators & Operations
- Data sets/files
- Time zones
- Dtypes
- Misc
    )abc)datedatetimetime	timedeltatimezone)DecimalN)Callable)tzlocaltzutc)
strategies)FixedOffsetutc)DatetimeTZDtypeIntervalDtype)	DataFrameIntervalPeriodSeries	Timedelta	Timestamp)ops)Index
MultiIndexFTUTCz3ignore:Promotion of numbers and bools:FutureWarningreturnc                 C   sd   | j dddd | j dddd | j dddd | j d	dd
d | j dddd | j dddd d S )N--skip-slow
store_truezskip slow tests)actionhelp--skip-networkzskip network tests	--skip-dbzskip db testsz--run-high-memoryzrun high memory tests--only-slowzrun only slow tests--strict-data-filesz0Fail if a test is skipped for missing data file.)Z	addoption)parser r&   _/mounts/lovelace/software/anaconda3/envs/metaDMG/lib/python3.9/site-packages/pandas/conftest.pypytest_addoptionf   s    r(   )itempathmessager   c                 C   s(   | j |r$| tjd|  dS )a  Ignore 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.
    zignore:N)nameendswith
add_markerpytestmarkfilterwarnings)r)   r*   r+   r&   r&   r'   ignore_doctest_warningu   s    r2   c                 C   s<  | d}| d}| d}| d}tjjd|dftjjd|dftjjd|dfg}g d	}| D ]}| d
s~|j dddr|tjd |D ]\}	}
t||	|
 qd|j	v r|tjj
 |t |D ]D\}}}}||jv r|r
|tjd|   q|| q|rbd|jvrb|tjd qbd S )Nr   r#   r!   r"   slownetworkz	--networkdb)	)zDataFrame.appendz%The frame.append method is deprecated)zSeries.appendz&The series.append method is deprecated)zdtypes.common.is_categoricalzis_categorical is deprecated)zCategorical.replacez!Categorical.replace is deprecated)zdtypes.common.is_extension_typez!'is_extension_type' is deprecated)zIndex.is_mixedzIndex.is_mixed is deprecated)zMultiIndex._is_lexsortedz%MultiIndex.is_lexsorted is deprecated)zmissing.mask_zero_div_zerozdivide by zero encountered)z$_validators.validate_axis_style_argszUse named argumentsz--doctest-modulesz--doctest-cythonF)defaultadd_doctest_importsz/frame/zskipping due to zskipping due to --only-slow)	getoptionr/   r0   r3   r4   r5   r.   Zusefixturesr2   ZnodeidZarraymanager suppress_npdev_promotion_warningkeywordsskip)itemsconfigZ	skip_slowZ	only_slowZskip_networkZskip_dbmarksZignored_doctest_warningsr)   r*   r+   r0   ZkwdZskip_if_foundZarg_namer&   r&   r'   pytest_collection_modifyitems   s6    






r?   ci)deadlineZsuppress_health_checkz)MonthBegin MonthEnd BMonthBegin BMonthEndic   )n	normalizez%YearBegin YearEnd BYearBegin BYearEnd         )Z	min_valueZ	max_value)rC   rD   monthz1QuarterBegin QuarterEnd BQuarterBegin BQuarterEndi   )rC   rD   ZstartingMonthc                 C   s   t | d< t| d< dS )z:
    Make `np` and `pd` names available for doctests.
    nppdN)rK   rL   )Zdoctest_namespacer&   r&   r'   r7      s    r7   )Zautousec                   C   s   t dd dS )z<
    Configure settings for all tests and test modules.
    Zchained_assignmentraiseN)rL   Z
set_optionr&   r&   r&   r'   configure_tests   s    rN   )r   rG   indexcolumnsc                 C   s   dt |  S Nzaxis=reprxr&   r&   r'   <lambda>	      rV   )paramsidsc                 C   s   | j S )z@
    Fixture for returning the axis numbers of a DataFrame.
    paramrequestr&   r&   r'   axis	  s    r^   rP   c                 C   s   dt |  S rQ   rR   rT   r&   r&   r'   rV     rW   c                 C   s   | j S )zA
    Fixture for returning aliases of axis 1 of a DataFrame.
    rZ   r\   r&   r&   r'   axis_1  s    r_   )TFN)rX   c                 C   s   | j S )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).
    rZ   r\   r&   r&   r'   observed  s    
r`   c                 C   s   | j S )z6
    Boolean 'ordered' parameter for Categorical.
    rZ   r\   r&   r&   r'   ordered)  s    ra   )firstlastFc                 C   s   | j S )z[
    Valid values for the 'keep' parameter used in
    .duplicated or .drop_duplicates
    rZ   r\   r&   r&   r'   keep1  s    rd   )bothneitherleftrightc                 C   s   | j S )zA
    Fixture for trying all interval 'inclusive' parameters.
    rZ   r\   r&   r&   r'   inclusive_endpoints_fixture:  s    ri   )rg   rh   re   rf   c                 C   s   | j S )z<
    Fixture for trying all interval closed parameters.
    rZ   r\   r&   r&   r'   closedB  s    rj   c                 C   s   | j S )zS
    Secondary closed fixture to allow parametrizing over all pairs of closed.
    rZ   r\   r&   r&   r'   other_closedJ  s    rk   gzipbz2zipxztarZzstdZ	zstandard)r>   c                 C   s   | j S )zK
    Fixture for trying common compression types in compression tests.
    rZ   r\   r&   r&   r'   compressionR  s    rq   c                 C   s   | j S )zk
    Fixture for trying common compression types in compression tests excluding
    uncompressed case.
    rZ   r\   r&   r&   r'   compression_onlyd  s    rr   c                 C   s   | j S )z,
    Fixture that an array is writable.
    rZ   r\   r&   r&   r'   writablev  s    rs   )innerouterrg   rh   c                 C   s   | j S )z:
    Fixture for trying all types of join operations.
    rZ   r\   r&   r&   r'   	join_type~  s    rv   nlargest	nsmallestc                 C   s   | j S )z1
    Fixture for trying all nselect methods.
    rZ   r\   r&   r&   r'   nselect_method  s    ry   c                 C   s
   t | jS Ntype__name__rT   r&   r&   r'   rV     rW   c                 C   s   | j S )z/
    Fixture for each null type in pandas.
    rZ   r\   r&   r&   r'   nulls_fixture  s    r~   c                 C   s   | j S )zL
    Fixture for each null type in pandas, each null type exactly once.
    rZ   r\   r&   r&   r'   unique_nulls_fixture  s    r   c                 C   s
   t | jS rz   r{   rT   r&   r&   r'   rV     rW   c                 C   s   | j S )z-
    Fixture for each NaT type in numpy.
    rZ   r\   r&   r&   r'   np_nat_fixture  s    r   c                 C   s   | j S )z;
    Fixture to parametrize over DataFrame and Series.
    rZ   r\   r&   r&   r'   frame_or_series  s    r   rO   seriesc                 C   s   | j S )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
    rZ   r\   r&   r&   r'   index_or_series  s    r   )rO   r   arrayc                 C   s   | j S )zG
    Fixture to parametrize over Index, Series, and ExtensionArray
    rZ   r\   r&   r&   r'   index_or_series_or_array  s    r   c                 C   s   | j S rz   )r}   rT   r&   r&   r'   rV     rW   c                 C   s   | j S )z]
    Fixture to test behavior for Index, Series, DataFrame, and pandas Array
    classes
    rZ   r\   r&   r&   r'   box_with_array  s    r   c                  C   s   G dd dt } | S )z,
    Fixture for a dictionary subclass.
    c                   @   s   e Zd ZddddZdS )z"dict_subclass.<locals>.TestSubDictNr   c                 _   s   t j| g|R i | d S rz   )dict__init__)selfargskwargsr&   r&   r'   r     s    z+dict_subclass.<locals>.TestSubDict.__init__)r}   
__module____qualname__r   r&   r&   r&   r'   TestSubDict  s   r   )r   )r   r&   r&   r'   dict_subclass  s    r   c                  C   s   G dd dt j} | S )z8
    Fixture for a non-mapping dictionary subclass.
    c                   @   s2   e Zd ZddddZdd Zdd Zd	d
 ZdS )z5non_dict_mapping_subclass.<locals>.TestNonDictMappingNr   c                 S   s
   || _ d S rz   )_data)r   Zunderlying_dictr&   r&   r'   r     s    z>non_dict_mapping_subclass.<locals>.TestNonDictMapping.__init__c                 S   s   | j |S rz   )r   __getitem__)r   keyr&   r&   r'   r      s    zAnon_dict_mapping_subclass.<locals>.TestNonDictMapping.__getitem__c                 S   s
   | j  S rz   )r   __iter__r   r&   r&   r'   r     s    z>non_dict_mapping_subclass.<locals>.TestNonDictMapping.__iter__c                 S   s
   | j  S rz   )r   __len__r   r&   r&   r'   r     s    z=non_dict_mapping_subclass.<locals>.TestNonDictMapping.__len__)r}   r   r   r   r   r   r   r&   r&   r&   r'   TestNonDictMapping  s   r   )r   Mapping)r   r&   r&   r'   non_dict_mapping_subclass  s    r   c                  C   s^   t d} | dd dd dd g }|jdd |jjD |_|jjg dd	d
 |S )z
    DataFrame with 3 level MultiIndex (year, month, day) covering
    first 100 business days from 2000-01-01 with random data
    d   c                 S   s   | j S rz   )yearrT   r&   r&   r'   rV     rW   zAmultiindex_year_month_day_dataframe_random_data.<locals>.<lambda>c                 S   s   | j S rz   )rI   rT   r&   r&   r'   rV     rW   c                 S   s   | j S rz   )dayrT   r&   r&   r'   rV     rW   c                 S   s   g | ]}| d qS )i8)astype).0Zlevr&   r&   r'   
<listcomp>  rW   zCmultiindex_year_month_day_dataframe_random_data.<locals>.<listcomp>)r   rI   r   T)Zinplace)tmZmakeTimeDataFramegroupbysumrO   Z
set_levelslevelsZ	set_names)ZtdfZymdr&   r&   r'   /multiindex_year_month_day_dataframe_random_data  s
    
 r   c                   C   s*   t g dg dgg dg dgddgdS )z;
    2-level MultiIndex, lexsorted, with string names.
    foobarbazquxonetwothree)
r   r   r   rG   rG      r      r   r   )
r   rG   r   r   rG   rG   r   r   rG   r   rb   second)r   codesnames)r   r&   r&   r&   r'   %lexsorted_two_level_string_multiindex  s
    r   c                 C   s(   | }t tjdd|tg ddddS )z2DataFrame with 2 level MultiIndex with random data
   r   )ABCexpr,   )rO   rP   )r   rK   randomrandnr   )r   rO   r&   r&   r'    multiindex_dataframe_random_data)  s    r   c                  C   sT   t g d} t ddg}tg d}tg d}ddg}t| |g||g|dd	S )
zJ
    MultiIndex used to test the general functionality of this object
    r   r   r   )r   r   rG   r   r   r   )r   rG   r   rG   r   rG   rb   r   F)r   r   r   Zverify_integrity)r   rK   r   r   )Z
major_axisZ
minor_axisZmajor_codesZminor_codesZindex_namesr&   r&   r'   _create_multiindex4  s    r   c                   C   s,   t jddgddgtjddddgg d	d
S )zB
    MultiIndex with a level that is a tzaware DatetimeIndex.
    rG   r   ab20130101r   
US/Eastern)Zperiodstzr   )r   )r   Zfrom_productrL   Z
date_ranger&   r&   r&   r'   _create_mi_with_dt64tz_levelH  s    r   r   z
US/Pacific)r   	complex64
complex128int64dtypeint32Zint16Zint8Zuint64Zuint32Zuint16Zuint8Zfloat64float32r   )r   r   r   )rG   r   r   )r   r   rG   rG   r   r   ZInt64ZUInt16ZFloat32booleanzstring[python])#stringr   zdatetime-tzperiodr   intuintrangefloatr   r   Z	num_int64Z	num_int32Z	num_int16Znum_int8Z
num_uint64Z
num_uint32Z
num_uint16Z	num_uint8Znum_float64Znum_float32zbool-object
bool-dtypeZcategoricalintervalemptytuplesmi-with-dt64tz-levelmultirepeatsZnullable_intZnullable_uintZnullable_floatZnullable_boolzstring-pythonzstring[pyarrow]zstring-pyarrowc                 C   s   t | j  S )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\   r&   r&   r'   rO   }  s    c                 C   s   g | ]}t t| ts|qS r&   )
isinstancer   r   r   r   r&   r&   r'   r     s   r   c                 C   s   | j }t|  S )z8
    index fixture, but excluding MultiIndex cases.
    )r[   r   r   )r]   r   r&   r&   r'   
index_flat  s    	r   c                 C   s0   g | ](}|d v s| dstt| ts|qS ))r   r   r   r   r   r   Znum_)
startswithr   r   r   r   r&   r&   r'   r     s
   
c                 C   s   t | j jdd}|j}| jdv rd| }d|d dd  |d< d|d dd  |d< t|S d|d< d|d< t||S d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.
    Tdeep)r   r   r   rz   r   rG   N)r   r[   r   valuestolistr   from_tuplesr|   )r]   indvalsr&   r&   r'   index_with_missing  s    

r   c                  C   s   t  } d| _| S )zC
    Fixture for Series of floats with Index of unique strings
    r   )r   ZmakeStringSeriesr,   sr&   r&   r'   string_series  s    r   c                  C   s   t  } d| _| S )zI
    Fixture for Series of dtype object with Index of unique strings
    objects)r   ZmakeObjectSeriesr,   r   r&   r&   r'   object_series  s    r   c                  C   s   t  } d| _| S )z9
    Fixture for Series of floats with DatetimeIndex
    ts)r   ZmakeTimeSeriesr,   r   r&   r&   r'   datetime_series  s    r   c                 C   s"   t | }tj|}t|| ddS )zHelper for the _series dictr   )rO   r,   )lenrK   r   r   r   )rO   sizedatar&   r&   r'   _create_series  s    r   c                 C   s"   i | ]\}}d | dt |qS )zseries-with-z-indexr   )r   Zindex_idrO   r&   r&   r'   
<dictcomp>  s   r   c                 C   s   t | S )zE
    Fixture for tests on series with changing types of indices.
    r   rO   r&   r&   r'   series_with_simple_index  s    r   c                  C   sH   g dg dg} t |  }t|}tjd}t||d}tj|d< |S )z:
    Fixture with a Series with a 2-level MultiIndex.
    )r   r   r   r   r   r   r   r   )r   r   r   r   r   r   r   r      r   r   )rn   r   r   rK   r   r   r   NaN)Zarraysr   rO   r   Zserr&   r&   r'   series_with_multilevel_index  s    

r   c                 C   s$   i | ]}|j  d tjd|dqS )z-seriesr   )r,   r   )r}   r   Zmake_rand_series)r   r   r&   r&   r'   r     s   c                 C   s   t | j jddS )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[   r   r\   r&   r&   r'   index_or_series_obj  s    r   c                   C   s   t t dS )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   )r   r   getSeriesDatar   r&   r&   r&   r'   	int_frame"  s    r   c                   C   s   t t S )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]
    )r   r   ZgetTimeSeriesDatar&   r&   r&   r'   datetime_frame?  s    r   c                   C   s   t t S )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]
    )r   r   r   r&   r&   r&   r'   float_frame\  s    r   c                	   C   s>   t dddtjdgd ddtjdgd ddd	tdd
S )z
    Fixture for DataFrame of float/int/string columns with RangeIndex
    Columns are ['a', 'b', 'c', 'float32', 'int32'].
          ?r   r   r   r   r   rG   r   )r   r   cr   r   r   )r   rK   r   aranger&   r&   r&   r'   mixed_type_framey  s    r  c                  C   s   t dddt dddt dddt dddt dddt dddt dddt dddt dddt dddg
} ttjt| | dS )zF
    Fixture for Series with a DatetimeIndex that has duplicates.
    i  rG   r   r      rF   r   )r   r   rK   r   r   r   )datesr&   r&   r'   (rand_series_with_duplicate_datetimeindex  s    









r  )rg   rh   rh   g?g      ?z2012-01M)freqz	period[M]z
2012-02-01Dz	period[D]z
2011-01-01r   i  )secondsztimedelta64[ns]c                 C   s   | j S rz   rZ   r\   r&   r&   r'   ea_scalar_and_dtype  s    r  )__add____radd____sub____rsub____mul____rmul____floordiv____rfloordiv____truediv____rtruediv____pow____rpow____mod____rmod__c                 C   s   | j S )zD
    Fixture for dunder names for common arithmetic operations.
    rZ   r\   r&   r&   r'   all_arithmetic_operators  s    r  c                 C   s   | j S )zU
    Fixture for operator and roperator arithmetic, comparison, and logical ops.
    rZ   r\   r&   r&   r'   all_binary_operators  s    "r  c                 C   s   | j S )z
    Fixture for operator and roperator arithmetic functions.

    Notes
    -----
    This includes divmod and rdivmod, whereas all_arithmetic_operators
    does not.
    rZ   r\   r&   r&   r'   all_arithmetic_functions  s    r  )
r   maxminmeanprodZstdvarZmedianZkurtZskewc                 C   s   | j S )z.
    Fixture for numeric reduction names.
    rZ   r\   r&   r&   r'   all_numeric_reductions"  s    r"  allanyc                 C   s   | j S )z.
    Fixture for boolean reduction names.
    rZ   r\   r&   r&   r'   all_boolean_reductions-  s    r%  c                 C   s   | j S )z>
    Fixture for all (boolean + numeric) reduction names.
    rZ   r\   r&   r&   r'   all_reductions8  s    r&  c                 C   s   | j S )z;
    Fixture for operator module comparison functions.
    rZ   r\   r&   r&   r'   comparison_op@  s    r'  )__le____lt____ge____gt__c                 C   s   | j S )zm
    Fixture for dunder names for compare operations except == and !=

    * >=
    * >
    * <
    * <=
    rZ   r\   r&   r&   r'   compare_operators_no_eq_neQ  s    
r,  )__and____rand____or____ror____xor____rxor__c                 C   s   | j S )zY
    Fixture for dunder names for common logical operations

    * |
    * &
    * ^
    rZ   r\   r&   r&   r'   all_logical_operators^  s    r3  c                 C   s
   |  dS )zO
    Returns the configuration for the test setting `--strict-data-files`.
    r$   )r8   )Zpytestconfigr&   r&   r'   strict_data_fileso  s    r4  .)r4  r   c                    s(   t jt jtd  fdd}|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.
    Ztestsc                     sL   t jj g| R  }t j|sHr6td| dntd| d |S )NzCould not find file z  and --strict-data-files is set.zCould not find .)osr*   joinexists
ValueErrorr/   r;   )r   r*   Z	BASE_PATHr4  r&   r'   deco  s    
zdatapath.<locals>.deco)r6  r*   r7  dirname__file__)r4  r;  r&   r:  r'   datapathw  s    r>  c                 C   s   t | ddddS )z*
    The iris dataset as a DataFrame.
    ior   csvziris.csv)rL   Zread_csv)r>  r&   r&   r'   iris  s    rA  z
Asia/Tokyozdateutil/US/Pacificzdateutil/Asia/Singaporez+01:15z-02:15z	UTC+01:15z	UTC-02:15i,  i)hoursr   r   r   c                 C   s   g | ]}t |qS r&   rR   )r   ir&   r&   r'   r     rW   c                 C   s   | j S )zD
    Fixture for trying timezones including default (None): {0}
    rZ   r\   r&   r&   r'   tz_naive_fixture  s    rD  c                 C   s   | j S )z4
    Fixture for trying explicit timezones: {0}
    rZ   r\   r&   r&   r'   tz_aware_fixture  s    rE  r   zdateutil/UTCc                 C   s   | j S )zQ
    Fixture to provide variants of UTC timezone strings and tzinfo objects.
    rZ   r\   r&   r&   r'   utc_fixture  s    rF  c                 C   s   | j S )zR
    Parametrized fixture for string dtypes.

    * str
    * 'str'
    * 'U'
    rZ   r\   r&   r&   r'   string_dtype  s    	rG  pyarrowz1.0.0)min_versionc                 C   s   | j S )za
    Parametrized fixture for string dtypes.

    * 'string[python]'
    * 'string[pyarrow]'
    rZ   r\   r&   r&   r'   nullable_string_dtype  s    rJ  pythonc                 C   s   | j S )zb
    Parametrized fixture for pd.options.mode.string_storage.

    * 'python'
    * 'pyarrow'
    rZ   r\   r&   r&   r'   string_storage  s    rL  c                 C   s   | j S )zK
    Parametrized fixture for bytes dtypes.

    * bytes
    * 'bytes'
    rZ   r\   r&   r&   r'   bytes_dtype  s    rM  c                 C   s   | j S )zN
    Parametrized fixture for object dtypes.

    * object
    * 'object'
    rZ   r\   r&   r&   r'   object_dtype$  s    rN  objectc                 C   s   | j S )zo
    Parametrized fixture for string dtypes.
    * 'object'
    * 'string[python]'
    * 'string[pyarrow]'
    rZ   r\   r&   r&   r'   any_string_dtype/  s    rP  c                 C   s   | j S )z\
    Parametrized fixture for datetime64 dtypes.

    * 'datetime64[ns]'
    * 'M8[ns]'
    rZ   r\   r&   r&   r'   datetime64_dtypeB  s    rQ  c                 C   s   | j S )z^
    Parametrized fixture for timedelta64 dtypes.

    * 'timedelta64[ns]'
    * 'm8[ns]'
    rZ   r\   r&   r&   r'   timedelta64_dtypeM  s    rR  c                	   C   s   t ddddddddS )z-
    Fixture emits fixed Timestamp.now()
    i  rG   rH   r        )r   rI   r   hourminuter   microsecond)r   r&   r&   r&   r'   fixed_now_tsX  s    rX  c                 C   s   | j S )z^
    Parameterized fixture for float dtypes.

    * float
    * 'float32'
    * 'float64'
    rZ   r\   r&   r&   r'   float_numpy_dtypeb  s    	rY  c                 C   s   | j S )zR
    Parameterized fixture for float dtypes.

    * 'Float32'
    * 'Float64'
    rZ   r\   r&   r&   r'   float_ea_dtypen  s    rZ  c                 C   s   | j S )z~
    Parameterized fixture for float dtypes.

    * float
    * 'float32'
    * 'float64'
    * 'Float32'
    * 'Float64'
    rZ   r\   r&   r&   r'   any_float_dtypey  s    r[  c                 C   s   | j S )zg
    Parameterized fixture for complex dtypes.

    * complex
    * 'complex64'
    * 'complex128'
    rZ   r\   r&   r&   r'   complex_dtype  s    	r\  c                 C   s   | j S )z|
    Parameterized fixture for signed integer dtypes.

    * int
    * 'int8'
    * 'int16'
    * 'int32'
    * 'int64'
    rZ   r\   r&   r&   r'   any_signed_int_numpy_dtype  s    r]  c                 C   s   | j S )zx
    Parameterized fixture for unsigned integer dtypes.

    * 'uint8'
    * 'uint16'
    * 'uint32'
    * 'uint64'
    rZ   r\   r&   r&   r'   any_unsigned_int_numpy_dtype  s    
r^  c                 C   s   | j S )z
    Parameterized fixture for any integer dtype.

    * int
    * 'int8'
    * 'uint8'
    * 'int16'
    * 'uint16'
    * 'int32'
    * 'uint32'
    * 'int64'
    * 'uint64'
    rZ   r\   r&   r&   r'   any_int_numpy_dtype  s    r_  c                 C   s   | j S )z
    Parameterized fixture for any nullable integer dtype.

    * 'UInt8'
    * 'Int8'
    * 'UInt16'
    * 'Int16'
    * 'UInt32'
    * 'Int32'
    * 'UInt64'
    * 'Int64'
    rZ   r\   r&   r&   r'   any_int_ea_dtype  s    r`  c                 C   s   | j S )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'
    rZ   r\   r&   r&   r'   any_int_dtype  s    ra  c                 C   s   | j S )z
    Parameterized fixture for any nullable integer dtype and
    any float ea dtypes.

    * 'UInt8'
    * 'Int8'
    * 'UInt16'
    * 'Int16'
    * 'UInt32'
    * 'Int32'
    * 'UInt64'
    * 'Int64'
    * 'Float32'
    * 'Float64'
    rZ   r\   r&   r&   r'   any_numeric_ea_dtype  s    rb  c                 C   s   | j S )z~
    Parameterized fixture for any signed nullable integer dtype.

    * 'Int8'
    * 'Int16'
    * 'Int32'
    * 'Int64'
    rZ   r\   r&   r&   r'   any_signed_int_ea_dtype  s    
rc  c                 C   s   | j S )z
    Parameterized fixture for any (purely) real numeric dtype.

    * int
    * 'int8'
    * 'uint8'
    * 'int16'
    * 'uint16'
    * 'int32'
    * 'uint32'
    * 'int64'
    * 'uint64'
    * float
    * 'float32'
    * 'float64'
    rZ   r\   r&   r&   r'   any_real_numpy_dtype  s    rd  c                 C   s   | j S )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'
    rZ   r\   r&   r&   r'   any_numpy_dtype!  s    "re  c                 C   s   | j S )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'
    rZ   r\   r&   r&   r'   any_numeric_dtypeF  s    $rf  r   r   r  mixedbytes   a   cr   zmixed-integerr   g       @Zfloatingr   integerzmixed-integer-floatdecimal
datetime64z
2013-01-01z
2018-01-01r   r   Z20180101r   i  i  r   r   r   r   c                 C   s    | j \}}tj|td}||fS )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   )r[   rK   r   rO  )r]   Zinferred_dtyper   r&   r&   r'   any_skipna_inferred_dtype  s    )
rn  c                  C   s>   t jddd ddlm}  ddlm} | }d|j_| |dS )	zj
    Get an instance of IPython.InteractiveShell.

    Will raise a skip if IPython is not installed.
    ZIPythonz6.0.0)Z
minversionr   )InteractiveShell)Configz:memory:)r=   )r/   importorskipZIPython.core.interactiveshellro  Ztraitlets.configrp  ZHistoryManagerZ	hist_file)ro  rp  r  r&   r&   r'   ip  s    rr  )ZbsrZcooZcscZcsrZdiaZdokZlilc                 C   s   ddl m} t|| jd S )z-
    Yields scipy sparse matrix classes.
    r   )sparseZ_matrix)Zscipyrs  getattrr[   )r]   rs  r&   r&   r'   spmatrix  s    ru  c                 C   s6   g | ].}t ttj|tjjr|d krttj|qS )Tick)
issubclassrt  rL   offsetsrv  )r   or&   r&   r'   r     s   c                 C   s   | j S )zN
    Fixture for Tick based datetime offsets available for a time series.
    rZ   r\   r&   r&   r'   tick_classes  s    rz  c                 C   s   | S rz   r&   rT   r&   r&   r'   rV     rW   c                 C   s   | j S )zk
    Simple fixture for testing keys in sorting methods.
    Tests None (no key) and the identity key.
    rZ   r\   r&   r&   r'   sort_by_key  s    r{  c                  c   sx   t d ddlm}  ddlm} ddlm} G dd d|}| d|d	d
 | V  |dd  d |j	d< |j
  d S )Nfsspecr   )register_implementation)MemoryFileSystem)	_registryc                       s,   e Zd ZdZdgZdd fddZ  ZS )z fsspectest.<locals>.TestMemoryFStestmemNr   c                    s(   | dd | jd< t jf i | d S )Ntestr   )popr  superr   )r   r   	__class__r&   r'   r     s    z)fsspectest.<locals>.TestMemoryFS.__init__)r}   r   r   protocolr  r   __classcell__r&   r&   r  r'   TestMemoryFS  s   r  r  T)clobber)r/   rq  r|  r}  Zfsspec.implementations.memoryr~  Zfsspec.registryr  r  r  storeclear)r}  r~  registryr  r&   r&   r'   
fsspectest  s    

r  )r   NN)ZEgonZVenkmanN)NCC1701Dr  r  c                 C   s   | j S )zP
    A 3-tuple of names, the first two for operands, the last for a result.
    rZ   r\   r&   r&   r'   r   	  s    r   c                 C   s   | j S )zI
    Parametrize over __setitem__, loc.__setitem__, iloc.__setitem__
    rZ   r\   r&   r&   r'   indexer_sli  s    r  c                 C   s   | j S )z<
    Parametrize over loc.__getitem__, iloc.__getitem__
    rZ   r\   r&   r&   r'   
indexer_li$  s    r  c                 C   s   | j S )z8
    Parametrize over __setitem__, iloc.__setitem__
    rZ   r\   r&   r&   r'   
indexer_si,  s    r  c                 C   s   | j S )z7
    Parametrize over __setitem__, loc.__setitem__
    rZ   r\   r&   r&   r'   
indexer_sl4  s    r  c                 C   s   | j S )z:
    Parametrize over at.__setitem__, loc.__setitem__
    rZ   r\   r&   r&   r'   
indexer_al<  s    r  c                 C   s   | j S )z<
    Parametrize over iat.__setitem__, iloc.__setitem__
    rZ   r\   r&   r&   r'   indexer_ialD  s    r  c                   C   s   t jjjdkS )z>
    Fixture to check if the array manager is being used.
    r   )rL   optionsmodedata_managerr&   r&   r&   r'   using_array_managerL  s    r  c                   C   s   t jjjot jjjdkS )z7
    Fixture to check if Copy-on-Write is enabled.
    block)rL   r  r  Zcopy_on_writer  r&   r&   r&   r'   using_copy_on_writeT  s    r  (   __doc__collectionsr   r   r   r   r   r   rl  r   operatorr6  typingr	   Zdateutil.tzr
   r   Z
hypothesisr   stnumpyrK   r/   Zpytzr   r   Zpandas.util._test_decoratorsutilZ_test_decoratorstdZpandas.core.dtypes.dtypesr   r   ZpandasrL   r   r   r   r   r   r   Zpandas._testingZ_testingr   Zpandas.corer   Zpandas.core.indexes.apir   r   rH  paImportErrorZhas_pyarrowzoneinfocompatZPY39ZZoneInfoZZoneInfoNotFoundErrorr0   r1   r9   r(   ZItemstrr2   r?   settingsZregister_profileZHealthCheckZtoo_slowZload_profilesplitr,   rt  Ztseriesrx  clsZregister_type_strategyZbuildsZintegersZbooleansZfixturer7   rN   r^   Z
axis_framer_   r`   ra   rd   ri   rj   rk   r[   Z
skip_if_norq   rr   rs   rv   ry   ZNULL_OBJECTSr~   Znulls_fixture2nanZNaTr   Zunique_nulls_fixture2ZNP_NAT_OBJECTSr   Znp_nat_fixture2r   r   Zindex_or_series2r   r   r   Zbox_with_array2r   r   r   r   r   r   r   ZmakeStringIndexZmakeDateIndexZmakePeriodIndexZmakeTimedeltaIndexZmakeIntIndexZmakeUIntIndexZmakeRangeIndexZmakeFloatIndexr   ZmakeNumericIndexZmakeBoolIndexrO  r   r   ZmakeCategoricalIndexZmakeIntervalIndexr   rn   r  boolr   idxkeysrO   Zindex_fixture2r   Zindex_flat2r   r   r   r   r   r<   Z_seriesr   r   ZNARROW_NP_DTYPESZ_narrow_seriesr   r   r   r   r   r  r  r  Z_all_arithmetic_operatorsr  addZraddsubZrsubmulZrmultruedivZrtruedivfloordivZ	rfloordivmodZrmodpowZrpoweqneltlegtgeand_Zrand_xorZrxoror_Zror_r  r  Z_all_numeric_reductionsr"  Z_all_boolean_reductionsr%  Z_all_reductionsr&  r'  r,  r3  r4  r>  rA  Z	TIMEZONESextendZTIMEZONE_IDSZparametrize_fixture_docrD  rE  Ztz_aware_fixture2Z_UTCSappendrF  Zutc_fixture2ZSTRING_DTYPESrG  rJ  rL  Zstring_storage2ZBYTES_DTYPESrM  ZOBJECT_DTYPESrN  rP  ZDATETIME64_DTYPESrQ  ZTIMEDELTA64_DTYPESrR  rX  ZFLOAT_NUMPY_DTYPESrY  ZFLOAT_EA_DTYPESrZ  r[  ZCOMPLEX_DTYPESr\  ZSIGNED_INT_NUMPY_DTYPESr]  ZUNSIGNED_INT_NUMPY_DTYPESr^  ZALL_INT_NUMPY_DTYPESr_  ZALL_INT_EA_DTYPESr`  ra  rb  ZSIGNED_INT_EA_DTYPESrc  ZALL_REAL_NUMPY_DTYPESrd  ZALL_NUMPY_DTYPESre  rf  ZNArm  Z_any_skipna_inferred_dtyperY   _rn  rr  ru  __all__rz  r{  r  r   setitemlocZilocr  r  r  r  atr  Ziatr  r  r  r&   r&   r&   r'   <module>   s   



<































%

	
			

















"





	


	
$#$$2	