U
|
ý°dC ã @ s d Z ddlmZmZ ddlZddlZddlZddlmZ ddl m
|
Z
|
m Z ddl Z ddl mZmZmZmZ ddlmZ G dd dZej d dd
|
g¡d d Zej d d dg¡dd Zdd ZdS )z
|
Collection of tests asserting things that should be true for
|
any index subclass except for MultiIndex. Makes use of the `index_flat`
|
fixture defined in pandas/conftest.py.
|
é )ÚcopyÚdeepcopyN)ÚIS64)Úis_integer_dtypeÚis_numeric_dtype)ÚCategoricalIndexÚ
|
MultiIndexÚ PeriodIndexÚ
|
RangeIndexc @ sÈ e Zd Zej dddg¡dd Zdd Zdd Zd
|
d Z d d Z
|
dd Z dd Z dd Z dd Zdd Zdd Zdd Zd d Zd d Zd d! Zd"d# Zej d$d%d&d'd(d)d*g¡d+d, Zd-d. ZdS )/Ú
|
TestCommonÚnameNÚnew_namec C s |}|r|}n
|
|j pd}|j|d}|j|ks2tt|jdksDt|jd |ksVt|| j|jk sjt|jd|d}|j|k std S )Nr ©r é F)Úindexr )r Zto_framer ÚAssertionErrorÚlenÚcolumnsÚvalues)Úselfr Ú
|
index_flatÚidxZidx_nameZdf© r úWd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/test_common.pyÚ test_to_frame! s
|
zTestCommon.test_to_framec
|
C s¤ |}| g ¡ |¡st|j|jgfD ]D}t|jtrB||jkrBq&d}tjt|d | |¡ W 5 Q R X q&ddgfD ](}tjt dd | |¡ W 5 Q R X qvd S )NzTCannot remove 1 levels from an index with 1 levels: at least one level must be left.©ÚmatchÚwrongz>'Requested level \(wrong\) does not match index name \(None\)')
|
Z droplevelÚequalsr r Ú
|
isinstanceÚtupleÚpytestÚraisesÚ
|
ValueErrorÚKeyError)r r r ÚlevelÚmsgr r r Útest_droplevel5 s ÿþzTestCommon.test_droplevelc C s` |}d}dgg}t jt|d |j|d W 5 Q R X t jt|d |j|d W 5 Q R X d S )Nz"Index.name must be a hashable typeÚ1r r )Únames)r! r" Ú TypeErrorÚrenameÚ set_names)r r r ÚmessageZrenamedr r r Ú"test_constructor_non_hashable_nameN s z-TestCommon.test_constructor_non_hashable_namec C s* |}t |||jd}t |j|j¡ d S )N©Údtype)Útyper0 ÚtmZ assert_equalÚ_data)r r ÚaÚbr r r Ú test_constructor_unwraps_index] s z)TestCommon.test_constructor_unwraps_indexc C s |}| ¡ }t ||¡ d S )N)Z to_flat_indexr2 Úassert_index_equal)r r r Úresultr r r Útest_to_flat_indexd s z TestCommon.test_to_flat_indexc C sÊ |}d}|j }| |g¡}|j |ks(t|j |ks6t|j|dd}|d ksPt|j |ks^t|j|gksnttjtdd |jddd W 5 Q R X d }|j|dd |j |ks¶t|j|gksÆtd S )
|
Nz#This is the new name for this indexT©ZinplacezLevel must be Noner r4 r ©r% )ÚAÚB)r r, r r+ r) r! r" r# )r r r r Ú original_nameZnew_indÚresr r r r Útest_set_name_methodsk s z TestCommon.test_set_name_methodsc C sT |}t tfD ]&}||}||k s$t| |¡s tq |j ddd}|jdksPtd S )NTZbanana)Údeepr )r r r r r )r r r ÚfuncZidx_copyZnew_copyr r r Útest_copy_and_deepcopy s z!TestCommon.test_copy_and_deepcopyc C s¨ |}t ||ddd}t ||dd}||k s2tt ||¡ | |¡sLt|jdksZt|jdkshttjd|d}tjd|d d
d}|| }|jjdks¤td S )
|
NTÚmario)r r F)r é )r é éÿÿÿÿ) r1 r r2 r7 r r ÚpdÚSeriesr )r r r ÚfirstÚsecondÚs1Ús2Zs3r r r Útest_copy_name s zTestCommon.test_copy_namec C s |}|j ddjdksttjtdd |j ddgd W 5 Q R X t|j d}tjt|d |j dggd W 5 Q R X d S )NrD r z$Length of new names must be 1, got 2r Zluigiz .name must be a hashable type) r r r r! r" r# r1 Ú__name__r* )r r r r& r r r Útest_copy_name2« s zTestCommon.test_copy_name2c C s¢ |}| ¡ }d|jd fD ] }|j|d}t ||¡ qd}tjt|d |jdd W 5 Q R X dt |j
|
¡ ¡ d}tjt |d |jdd W 5 Q R X d S ) Nr r; z.Too many levels: Index has only 1 level, not 4r rF z6Requested level \(wrong\) does not match index name \(z\)r ) Údrop_duplicatesr Úuniquer2 r7 r! r" Ú
|
IndexErrorÚreÚescapeÚ__repr__r$ )r r r Úexpectedr% r8 r& r r r Útest_unique_level¸ s ÿz TestCommon.test_unique_levelc C s$ |}t |st d¡ |dgd }|dg }|jdks<tz|jdksLtW n tk
|
rb Y nX | ¡ }t ||¡ |j
|
st d¡ |j dgd }t j |d< |d d
}| |¡}| |¡} | jdksÐt|j|jksàt| j|jksðt| }
|
t|| gD ] \} } | ¡ }t ||
|
¡ q d S )Nz)Skip check for empty Index and MultiIndexr é TFz%Skip na-check if index cannot hold narE )r r! ÚskipÚ is_uniquer ÚhasnansÚNotImplementedErrorrR r2 r7 Z _can_hold_naÚ_valuesÚnpÚnanZ _shallow_copyr0 Ú enumerate) r r r r Z
|
idx_uniquer8 ÚvalsZ vals_uniqueZidx_nanZidx_unique_nanrW ÚposÚir r r Ú test_uniqueÍ s4
|
|
|
|
|
|
zTestCommon.test_uniquec C sF |}t |tjr,tjjdtd}|j |¡ |j r<t
|
d¡ |d }d||k ¡ }}|dkrft |}|j rÖ|j|dd}||kst|j|dd} || ks t|j|dd}
|
||
|
ksºt|j|dd} || ksÔtnl|jr|j|dd}||ksøt|j|dd} || ksBtn,d} tjt| d |j|dd W 5 Q R X d S )
|
Nz8IntervalIndex.searchsorted does not support Interval arg)Úreasonr" zSkip check for empty Indexr Úleft)ZsideÚrightz0index must be monotonic increasing or decreasingr )r rH Z IntervalIndexr! ÚmarkÚxfailr] ÚnodeÚ
|
add_markerÚemptyrZ Zargminr Zis_monotonic_increasingZ_searchsorted_monotonicr Z searchsortedZis_monotonic_decreasingr" r# ) r r Úrequestr ri ÚvalueZ expected_leftZexpected_rightZssm_leftZ ssm_rightZss_leftZss_rightr& r r r Útest_searchsorted_monotonicõ s< þ
|
z&TestCommon.test_searchsorted_monotonicc C sâ |}t |trt d¡ t|dkr.t d¡ t|}tt|}t|rP|j nd }|||d}t|}t
|
j |t |d ¡} ||j| }
|
t | ¡j|dj} t |
|
j|d| ¡ |t |
|
¡j|d} t |
|
j|d| ¡ d S )NzWRangeIndex is tested in test_drop_duplicates_no_duplicates as it cannot hold duplicatesr zXempty index is tested in test_drop_duplicates_no_duplicates as it cannot hold duplicatesr/ g ø?)Úkeep)r r
|
r! rZ r r1 ÚlistÚsetr r0 r_ ÚrandomÚchoiceÚintr rH rI Ú
|
duplicatedr2 Úassert_numpy_array_equalrQ r7 ) r r rq r ÚholderÚ unique_valuesr0 Ú
|
unique_idxÚnZduplicated_selectionr Úexpected_duplicatedZexpected_droppedr r r Útest_drop_duplicates' s*
|
ÿ ÿ ÿz TestCommon.test_drop_duplicatesc C s |}t |tr|}n2t|}tt|}t|r6|jnd }|||d}tjdgt | dd}t
|
| ¡ |¡ | ¡ }t
|
||¡ ||k std S )Nr/ FÚbool)r r
|
r1 rr rs r r0 r_ Úarrayr r2 rx rw rQ r7 r ) r r r r{ ry rz r0 r} Zresult_droppedr r r Ú"test_drop_duplicates_no_duplicatesJ s
|
z-TestCommon.test_drop_duplicates_no_duplicatesc C s. d}t jt|d |jdd W 5 Q R X d S )Nz6drop_duplicates\(\) got an unexpected keyword argumentr Tr: )r! r" r* rQ )r r r& r r r Ú test_drop_duplicates_inplace` s z'TestCommon.test_drop_duplicates_inplacec C sZ |}t |}t|r t|tr(t d¡ ||d gd }|jdksHt|jdksVtd S )Nz6Skip check for empty Index, MultiIndex, and RangeIndexr rY FT) r1 r r r
|
r! rZ r[ r Zhas_duplicates)r r r ry r r r r Útest_has_duplicatese s
|
z TestCommon.test_has_duplicatesr0 Úint64Úuint64Úfloat64Úcategoryzdatetime64[ns]ztimedelta64[ns]c C sÖ t |tr"dd t|jD |_nd|_d }|jjdkrF|dkrFtj }t
|
|jdkoZ|dk}z*t j ||dd | |¡}W 5 Q R X W n ttttfk
|
r¤ Y d S X t |trÂ|j|jksÒtn|j|jksÒtd S )
|
Nc S s g | ]}d t | qS )r )Ústr)Ú.0rd r r r Ú
|
<listcomp>{ s z9TestCommon.test_astype_preserves_name.<locals>.<listcomp>r Úc)r r r
zstring[pyarrow]r F)Zraise_on_extra_warningsZcheck_stacklevel)r r ÚrangeZnlevelsr) r r0 Úkindr_ ZComplexWarningr r2 Zassert_produces_warningZastyper# r* r] Ú SystemErrorr )r r r0 ÚwarnZis_pyarrow_strr8 r r r Útest_astype_preserves_namet s&
|
ý
|
z%TestCommon.test_astype_preserves_namec C sØ |}|j dd}tjdgt| td}t |j|¡ |jdksDt |j dd}|j
|
}t|dkrfd S t |j rtd S |j tkrd S tj |d< t||}tjdgt| td}d|d< t |j|¡ |jdksÔt d S )NT)rA Fr/ r r )r r_ r r r r2 rx Z_isnanr\ r r^ r r0 r` r1 )r r r r rW r r r r Útest_hasnans_isnans s&
|
|
|
z TestCommon.test_hasnans_isnans)rO Ú
|
__module__Ú __qualname__r! ri Ú parametrizer r' r. r6 r9 r@ rC rN rP rX re rp r~ r r r r r r r r r r s.
|
(2#þ
|
r Ú na_positionZmiddlec C s0 t jtd| d | j|d W 5 Q R X d S )Nzinvalid na_position: r ©r )r! r" r# Ú sort_values)Úindex_with_missingr r r r Ú$test_sort_values_invalid_na_position´ s r rJ Úlastc C s¨ t | tr"|j tjjddd¡ t | ¡ ¡}| |
|
¡ j }t |¡}|dkrft d g| |g¡}nt |d g| g¡}t| || jd}| j|d}t ||¡ d S )Nz,missing value sorting order not well-definedF)rf ÚstrictrJ r/ r )r r rk rl r! ri rj r_ ÚsumZisnaZnotnar ÚsortZ concatenater1 r0 r r2 r7 )r r rn Z missing_countZ not_na_valsZ sorted_valuesrW r8 r r r Ú test_sort_values_with_missingº s
|
ÿÿ
|
r c C s¤ t | trtst d¡ | }|j |¡s,t| ¡ |¡s>t|j }|j
|
|j
|
ksTt|j |j ksdt|j |j ksttt | t tfs|j|jkst|j |j j d S )NÚOverflow)r r r r! rZ ÚTr r Z transposer ÚshapeÚndimÚsizer
|
r Únbytes)r r r r r r Ú test_ndarray_compat_propertiesÕ s
|
r¥ )Ú__doc__r r rT Únumpyr_ r! Z pandas.compatr Zpandas.core.dtypes.commonr r ZpandasrH r r r r
|
Zpandas._testingZ_testingr2 r ri r r r r¥ r r r r Ú<module> s$
|
|
|
