o
    Eho                     @   sh  d dl mZ d dlmZ d dlmZ d dlZd dlZd dlZd dlZ	d dl
Z
d dlmZ d dlmZ d dlm  mZ d dlmZmZmZ d dlZd dlmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z# d dl$m%Z& d d	l'm(Z(m)Z)m*Z*m+Z+m,Z, G d
d dZ-G dd dZ.G dd dZ/e
j01dg ddd Z2e
j01deedgdeedgdeedgdee dgdeee3ddeee4d dgdee(dge5dee)dgd gdgdd Z6dd Z7d d! Z8e
j01d"de	j9e	j:e	j;gd#d$ Z<e
j01d%e(i ggd&d' e&j=D ei ge i ged(d)igd*d+ Z>e
j01d,ej?ej@gd-d. ZAeBd/d0d1 ZCdS )2    )defaultdict)datetime)partialN)IS64)InvalidIndexError)is_any_real_numeric_dtypeis_numeric_dtypeis_object_dtype)CategoricalIndex	DataFrameDatetimeIndexIntervalIndexPeriodIndex
RangeIndexSeriesTimedeltaIndex
date_rangeperiod_rangetimedelta_range)Index
MultiIndex_get_combined_indexensure_indexensure_index_from_sequencesc                   @   s  e Zd ZejdefddZdd Zejj	ddgdd	d
d Z
dd Zejj	ddgdd	dd Zdd Zej	dddgej	deddddddeddddeddddeddddgdd Zej	d eddddd!dfedddddfedddddfgd"d# Zd$d% Zd&d' Zd(d) Zej	d*eejejfed+ejfgd,d- Zej	d.g d/d0fd1ejd2d3gd4fd5d6d7ejgd8fgd9d: Zej	d;d<d=gej	d>eegd?d@ Zej	d;d<d=gej	d>eegdAdB Z ej	dCg e!g dDdE dFD gej	d>ee"eegdGdH Z#ej	dIe$g ddJe$fe$e!g ddJe$fe$dKdE dFD ddJe$fe%dLdMe%fe&dLdNgdOdPggg g gdQe&fgdRdS Z'ejj	dg dTdd	dUdV Z(ejj	ddej)dWejj*dXdYdZd[d\d]gdd	d^d_ Z+ejj	dg d`dd	dadb Z,dcdd Z-ej	deedfdggeg dhg digdjdk Z.dldm Z/dndo Z0dpdq Z1ejj	ddgdd	drds Z2dtdu Z3ejj	dg dvdd	ej	dwe4ej5gdxdy Z6ejj	dg dvdd	dzd{ Z7d|d} Z8d~d Z9dd Z:ej	deddddeddddedddgdd Z;dd Z<ej	ddd dd gdd Z=ej	ddd dd gej>ddd Z?ej	de@g dg ddddddgdd ZAdd ZBdd ZCej	dddgdd ZDejj	dg ddgd	dd ZEejj	dg ddgd	dd ZFdd ZGdd ZHej	dg dg digdd ZIej	dddgdd ZJejj	dg ddd	dd ZKejj	dg ddd	ej	dddgddggdd ZLejj	dg ddd	ddÄ ZMddń ZNddǄ ZOej	ddeg dʢfg dˢedLdNgfgdd̈́ ZPej	d<g d΢g dϢg dТgej	dddfgdfdggddԄ ZQej>dddք ZRej	d;g dעddل ZSej	d<g dڢh dۣgej	deg dݢeTg dޢfeg ejTg eUdߍfgdd ZVdd ZWdd ZXej	dddgej	deg dݢeg dejdߍgdd ZYdd ZZej	ddddejgdd Z[ej	d]g e@e\dߍeTg gdd Z]dd Z^ej	d<g dg dg dg deddddgdd Z_ejj	ddgdd	ej	ddd gdd Z`dd Zaejj	dg ddd	dd Zbej	dg d	d
d Zcej	deeddeddde&eddgeddddgdd Zfej	ddeg diddgdggfdeg diddgdggfde&ediddejfdejejfgfgdd Zgdd Zhdd Ziej	deejd dfeeddɃdfgd!d" Zkd#d$ Zld%d& Zmd'd( Znd)d* Zod+d, Zpej	d-ddgej	d.g eTg g d/g d0eTg d/eTg d0ed1ddjqed1ddr gd2d3 Zsej	d.g eTg ejTg ejtdߍgd4d5 Zuej	d6eg ejvfgd7d8 Zwd9d: Zxd;d< Zyd=d> Zzd?d@ Z{ej	dAe&edBdCgeTddgfe&edBdDgeTddgfgdEdF Z|dGdH Z}ej	de&eg dIeg dJgdKdL Z~dMdN Zej	dOejdPdQgfejdRdSgfgdTdU ZdVdW ZdXdY ZdZd[ Zd\d] Zd^d_ Zej	d`dLejddddgfdNejejdadadagfdejejejdbdbgfgdcdd Zej	dedg dffdLg dgfdNg dhfgdidj ZdS (k  	TestIndexreturnc                 C   s   t tdS )NZabcde)r   listself r   >lib/python3.10/site-packages/pandas/tests/indexes/test_base.pysimple_index.      zTestIndex.simple_indexc                 C   s"   |}|d }| |du sJ d S )Nr   T)Z$_can_hold_identifiers_and_holds_name)r   r!   indexkeyr   r   r    test_can_hold_identifiers2   s   z#TestIndex.test_can_hold_identifiersr#   r   T)Zindirectc                 C   sD   t jtdd |d d d f  W d    d S 1 sw   Y  d S )NMulti-dimensional indexingmatch)pytestraises
ValueErrorr   r#   r   r   r    test_new_axis7   s   "zTestIndex.test_new_axisc                 C   s   t || d S N)tmassert_contains_allr,   r   r   r    test_constructor_regular?   s   z"TestIndex.test_constructor_regularstringc                 C   s.   t |}t|}t|| t|| d S r.   )nparrayr   r/   r0   assert_index_equal)r   r#   arr	new_indexr   r   r    test_constructor_castingB   s   
z"TestIndex.test_constructor_castingc                 C   s   t tddd}t|}t |ddd}t|t sJ |jdks"J |r1t|jt	j|dd nt
||j d|d	< |d	 dksDJ d S )
Nabcnamer:   Tcopyr:   strdtypeZSOMEBIGLONGSTRINGr   )r   r   r3   r4   
isinstancer:   r/   Zassert_extension_array_equalvaluespdassert_numpy_array_equal)r   using_infer_stringr#   r6   r7   r   r   r    test_constructor_copyJ   s   
zTestIndex.test_constructor_copycast_as_objFz2015-01-01 10:00D   z
US/EasternzGreen Eggs & Ham)freqperiodstzr:   )rJ   rK   1 daysz
2015-01-01c                 C   s   |r!t jtdd t|t}W d    n1 sw   Y  nt|}t || t|trZ|j	|j	ks8J |r\|t
jdd7 }t|td}|jtjksPJ t|t|ks^J d S d S d S )NzDtype inferencer'   2   )Znanosecondsr?   )r/   assert_produces_warningFutureWarningr   astypeobjectr5   rA   r   rL   rC   Z	Timedeltar@   r3   object_r   )r   rG   r#   resultr   r   r    "test_constructor_from_index_dtlikeY   s"   
z,TestIndex.test_constructor_from_index_dtlikezindex,has_tz)rJ   rK   rL   c                 C   s4   t t|}t|| |r|j|jksJ d S d S r.   )r   r   r/   r5   rL   )r   r#   Zhas_tzrT   r   r   r    #test_constructor_from_series_dtlike}   s   z-TestIndex.test_constructor_from_series_dtlikec                 C   s>   g d}t |dd}tt|}t |dd}t|| d S )Nz1-1-1990z2-1-1990z3-1-1990z4-1-1990z5-1-1990MSrJ   )r   r   rC   to_datetimer/   r5   )r   dtsexpectedsrT   r   r   r    !test_constructor_from_series_freq   s
   z+TestIndex.test_constructor_from_series_freqc                 C   s   g d}t |dd}ttjdd}||d< t |d dd}|s%tnd}|d j|ks0J d|_t	|| t
|dd}t|d | |sVt|d }|dksXJ d S d S )	NrW   rX   rY      )   rI   dater>   r;   )r   r   r3   randomdefault_rngrR   r@   r:   r/   r5   r   assert_series_equalrC   Z
infer_freq)r   rE   r[   r\   dfrT   r@   rJ   r   r   r    'test_constructor_from_frame_series_freq   s    z1TestIndex.test_constructor_from_frame_series_freqc                 C   s2   t jg}t|t jd}t|dd}t|| d S )Nr?   float)r3   nanr   float64r/   r5   )r   datar\   rT   r   r   r    test_constructor_int_dtype_nan   s   z(TestIndex.test_constructor_int_dtype_nanzklass,dtype,na_valzdatetime64[ns]c                 C   sP   ||g}||}|j |ksJ t|}t|| tt|}t|| d S r.   )r@   r   r/   r5   r3   r4   )r   klassr@   Zna_valZna_listr\   rT   r   r   r    test_index_ctor_infer_nan_nat   s   	z'TestIndex.test_index_ctor_infer_nan_natz
vals,dtype   r_   rI      r`   intg?g@      @rg   ABCobjc                 C   s*   t ||d}||j|}t|| d S )Nr;   )r   Z_simple_newrB   r/   r5   )r   valsr@   r#   rT   r   r   r    test_constructor_simple_new   s   	z%TestIndex.test_constructor_simple_newattrrB   asi8rl   c           
      C   sR  t ddd}t||}||}|j}|d u}d}|dkr+t||}	t|	| n.|tu rMtj	t
dd |||d W d    n1 sGw   Y  n|||d}	t|	| |dkr|r~tj	t
|d t|| W d    n1 sxw   Y  nt||}	t|	| n|||d	}	t|	| |dkrtt||}	t|	| n0|tu rtj	t
dd |||d W d    n1 sw   Y  n|t||d}	t|	| |dkr|rtj	t
|d tt|| W d    d S 1 sw   Y  d S tt||}	t|	| d S |t||d	}	t|	| d S )
N
2011-01-01r`   rK   z:Cannot use .astype to convert from timezone-naive dtype torz   zunexpected keywordr'   )rL   r?   )r   getattrZtz_localizer@   r   r/   r5   r   r)   r*   	TypeErrorrQ   r   )
r   Ztz_naive_fixturery   rl   r#   argr@   errmsgrT   r   r   r     test_constructor_dtypes_datetime   sZ   


$z*TestIndex.test_constructor_dtypes_datetimec                 C   s^   t ddd}|d }|j}t||}|||d}t|| |t||d}t|| d S )NrM   r`   r|   r?   )r   Z
_with_freqr@   r}   r/   r5   r   )r   ry   rl   r#   r@   rB   rT   r   r   r    !test_constructor_dtypes_timedelta  s   

z+TestIndex.test_constructor_dtypes_timedeltavaluec                 c       | ]}|V  qd S r.   r   .0_r   r   r    	<genexpr>#      zTestIndex.<genexpr>r   c                 C   s&   ||}t ||sJ t|rJ d S r.   rA   len)r   r   rl   emptyr   r   r    test_constructor_empty#  s   z TestIndex.test_constructor_emptyzempty,klassrY   c                 c   r   r.   r   r   r   r   r    r   7  r   ro   )stepr_   ZblueZredlevelsZcodesc                 C   s   t ||sJ t|rJ d S r.   r   )r   r   rl   r   r   r    test_constructor_empty_special2  s   z(TestIndex.test_constructor_empty_special)r   ri   float32int64int32ZperiodrangeZrepeatsZ	timedeltaZtuplesuint64uint32c                 C   s   | d d S )Ni8)viewr,   r   r   r    test_view_with_args@  s   zTestIndex.test_view_with_argscategoricalzgh-25464reason)Zmarksbool-object
bool-dtyper   c                 C   s   |j tkr&d}tjt|d |d W d    d S 1 sw   Y  d S d}tjt|d |d W d    d S 1 s@w   Y  d S )NzWhen changing to a larger dtyper'   r   zCannot change data-type for array of references\.|Cannot change data-type for object array\.|Cannot change data-type for array of strings\.|)r@   boolr)   r*   r+   r   r~   r   r#   r   r   r   r    'test_view_with_args_object_array_raisesU  s   
""z1TestIndex.test_view_with_args_object_array_raises)r   r   r   c                 C   s6   | d}|d d|_| d}|jdksJ d S )Nr   r`   foobar)rQ   get_locr:   )r   r#   Zcastedr   r   r    test_astypen  s
   


zTestIndex.test_astypec                 C   s"   t g dt g dsJ d S Nabcr   equalsr   r   r   r    test_equals_object~  s   "zTestIndex.test_equals_objectcompr   r   )r   r   dr   c                 C   s   t g d|rJ d S r   r   )r   r   r   r   r    test_not_equals_object  s   z TestIndex.test_not_equals_objectc                 C   s   t g d}t g d}||sJ |d}||sJ ||r&J |d}||s2J t g d}t g ddd}||rGJ d S )Nr   foo)r   r   r   r   )r   r   F)Ztupleize_cols)r   	identicalrenamer   )r   Zi1Zi2Zi3Zi4r   r   r    test_identical  s   

zTestIndex.test_identicalc                 C   s.  t td}||sJ ||    sJ |t tdr'J || r0J ||jddr;J ||d d  rFJ |ttdrRJ || s[J | }d|_||siJ ||spJ |t |jrzJ ttdd}t |dd}t |dd}||rJ d S )N
   F)ZdeepZbobro      r=   )	r   r   is_r   r=   r3   r4   r:   rB   )r   ZindZind2r6   Zind1r   r   r    test_is_  s$   zTestIndex.test_is_c                 C   s   t g d}t ddgtd}d}tjt|d ||d  W d    n1 s)w   Y  tjttt	|d || W d    n1 sJw   Y  tjttt	|d || W d    d S 1 slw   Y  d S )Nro   r_   rI   TFr?   z$Cannot compare dtypes int64 and boolr'   r   )
r   rR   r)   r*   r~   Zasofr   reescaper>   )r   leftrightr   r   r   r     test_asof_numeric_vs_bool_raises  s   "z*TestIndex.test_asof_numeric_vs_bool_raisesc                 C   s   t jt|td}d|ddd< || }t|D ]\}}|||ks%J q|t| }t|D ]\}}|||ks=J q0d S )Nr?   Fr`      r_   )r3   Zonesr   r   	enumerater   r   )r   r#   Z
bool_indexZ	sub_indexivalr   r   r    test_booleanindex  s   zTestIndex.test_booleanindexc                 C   s4   |}|g d }|D ]}||| | ksJ q
d S )Nr   )r   )r   r!   r#   slr   r   r   r    
test_fancy  s
   zTestIndex.test_fancy)r2   r   r   r   r   ri   r   r@   c                 C   s   |t ju r|r|jdkr|tjjdd t jg |d}t|g |jd}|g  	|s/J |t jkrWt
jtdd || 	|sEJ W d    d S 1 sPw   Y  d S || 	|s`J d S )Nr2   znumpy behavior is buggyr   r?   zis deprecatedr'   )r3   bool_r@   Zapplymarkerr)   markxfailr4   typer   r/   rO   rP   )r   r#   r@   requestrE   Z	empty_arrempty_indexr   r   r    test_empty_fancy  s   
"zTestIndex.test_empty_fancyc                 C   st   t jg t jd}t|g |jd}|g  |sJ d}tjt|d ||  W d    d S 1 s3w   Y  d S )Nr?   z)arrays used as indices must be of integerr'   )	r3   r4   ri   r   r@   r   r)   r*   
IndexError)r   r#   Z
empty_farrr   r   r   r   r    test_empty_fancy_raises  s   
"z!TestIndex.test_empty_fancy_raisesc                 C   sz   |}t ddd}||}||}tt||d}t|| t|| t|| t|| t|| d S )Nz
2019-01-01r   r|   O)	r   unionr   r3   appendrQ   r/   r5   r0   )r   r!   r#   
date_indexZ	first_catZ
second_catZappendedr   r   r    test_union_dt_as_obj  s   

zTestIndex.test_union_dt_as_objc                 C   sr   t tdtjd}|dd }t dd |D }t|| |dd }tdd |D }t|| d S )	NrI   r?   c                 S   s   | fS r.   r   xr   r   r    <lambda>      z0TestIndex.test_map_with_tuples.<locals>.<lambda>c                 S   s   g | ]}|fqS r   r   r   r   r   r   r    
<listcomp>  s    z2TestIndex.test_map_with_tuples.<locals>.<listcomp>c                 S   s   | | dkfS Nro   r   r   r   r   r    r         c                 S   s   g | ]}||d kfqS )ro   r   r   r   r   r    r     s    )	r   r3   aranger   mapr/   r5   r   from_tuplesr   r#   rT   r\   r   r   r    test_map_with_tuples  s   zTestIndex.test_map_with_tuplesc                 C   s>   g d}t t|g d}|dd }t|t| d S )Nr   barbazr   c                 S   s   | d S Nr   r   r   r   r   r    r         z3TestIndex.test_map_with_tuples_mi.<locals>.<lambda>)r   r   zipr   r/   r5   r   )r   Zfirst_levelZmulti_indexZreduced_indexr   r   r    test_map_with_tuples_mi  s   z!TestIndex.test_map_with_tuples_mi
2020-01-01r   1 dayr|   c                 C   s,   t dgd }|dd }t|| d S )Nro   r   c                 S      dS r   r   r   r   r   r    r   )      zATestIndex.test_map_tseries_indices_return_index.<locals>.<lambda>)r   r   r/   r5   r   r#   r\   rT   r   r   r    %test_map_tseries_indices_return_index  s   	z/TestIndex.test_map_tseries_indices_return_indexc                 C   sN   t tdddddd}|dd }ttjdd	d
dd}tj||dd d S )Nr      h)rK   rJ   Zhourlyr;   c                 S   s   | j S r.   )Zhourr   r   r   r    r   0  r   zGTestIndex.test_map_tseries_indices_accsr_return_index.<locals>.<lambda>r   r?   Texact)r   r   r   r   r3   r   r/   r5   )r   r   rT   r\   r   r   r    +test_map_tseries_indices_accsr_return_index,  s   z5TestIndex.test_map_tseries_indices_accsr_return_indexmapperc                 C      dd t | |D S )Nc                 S      i | ]\}}||qS r   r   r   er   r   r   r    
<dictcomp>7      &TestIndex.<lambda>.<locals>.<dictcomp>r   rB   r#   r   r   r    r   7      zTestIndex.<lambda>c                 C   
   t | |S r.   r   r   r   r   r    r   8     
 c                 C   sB   t g d}t tdtjd}|||j|}t|| d S )Nr   rI   r?   )r   r3   r   r   r   rB   r/   r5   )r   r   r\   r#   rT   r   r   r    test_map_dictlike_simple4  s   	z"TestIndex.test_map_dictlike_simplec                 C   r   )Nc                 S   r   r   r   r   r   r   r    r   E  r   r   r   r   r   r   r    r   E  r  c                 C   r  r.   r  r   r   r   r    r   F  r  z3ignore:PeriodDtype\[B\] is deprecated:FutureWarningc                 C   s   t |trtd n|jstd tjt|ddtjd}|j	r't
g }n%t|jr5|j||jd}nt|t
u rH|jtkrHt
||jd}nt
|}||||}t|| d S )NzTested in test_categoricalzCannot map duplicated indexr   r?   )rA   r
   r)   skip	is_uniquer3   r   r   r   r   r   r   r@   Z_constructorr   rR   r   r/   r5   )r   r#   r   r   rngr\   rT   r   r   r    test_map_dictlikeB  s   




zTestIndex.test_map_dictlike)r          @r   )r   r_   r  r#   r   r  r   c                 C   s2   t dtjdg}t g d|}t|| d S )Nr  r   )r_   ro   r   r   r3   rh   r   r/   r5   )r   r   r\   rT   r   r   r    )test_map_with_non_function_missing_valuesa  s   z3TestIndex.test_map_with_non_function_missing_valuesc                 C   s@   t dtjdtjdg}|jdd dd}|d }t|| d S )	Ng      ?rI   r`   c                 S   s   | d S )Nr_   r   r   r   r   r    r   o  r   z1TestIndex.test_map_na_exclusion.<locals>.<lambda>ignore)Z	na_actionr_   r  r   r   r   r    test_map_na_exclusionl  s   zTestIndex.test_map_na_exclusionc                 C   sF   t g d}tdd }d|d< ||}t g d}t|| d S )Nr   c                   S   r   )Nblankr   r   r   r   r    r   u  r   z0TestIndex.test_map_defaultdict.<locals>.<lambda>stuffro   )r  r  r  )r   r   r   r/   r5   )r   r#   Zdefault_dictrT   r\   r   r   r    test_map_defaultdicts  s   
zTestIndex.test_map_defaultdictzname,expected)r   r   )r   Nc                 C   sj   t g dd}t g d|d}d}tjt|d ||}W d    n1 s'w   Y  |j|ks3J d S )Nr   r;   r   zDThe behavior of array concatenation with empty entries is deprecatedr'   )r   r/   rO   rP   r   r:   )r   r:   r\   r   r   r   rT   r   r   r    test_append_empty_preserve_name{  s   z)TestIndex.test_append_empty_preserve_namezindex, expected))r2   F)r   Fr   Fr   F)r   T)r   T)r   T)r   Tr   F)ri   T)r   Tc                 C   s   t ||u sJ d S r.   )r   )r   r#   r\   r   r   r    test_is_numeric  s   zTestIndex.test_is_numeric))r2   T)r   Tr  r  )r   F)r   F)r   F)r   Fr  )ri   F)r   Fc                 C   s*   |r|j dkr|rd}t||u sJ d S )Nr2   F)r@   r	   )r   r#   r\   rE   r   r   r    test_is_object  s   zTestIndex.test_is_objectc                 C   s   |   d S r.   )Z_summaryr,   r   r   r    test_summary  s   zTestIndex.test_summaryc                 C   s   t  }d}t|ds:t|g}tjt|d | }W d    n1 s(w   Y  t|d g}||ks:J tjt|d tg   W d    d S 1 sSw   Y  d S )NIndex\.format is deprecatedZ000r'   r   )	r   nowr>   endswithr   r/   rO   rP   format)r   r  r   r#   	formattedr\   r   r   r    test_format_bug  s   

"zTestIndex.test_format_bugrw   )ro   y       @      @      @c                 C   s   t |}|| t|td}d}tjt|d | }W d    n1 s'w   Y  t|t	r3dnt
|}t
|d t
|d t
|d |g}||ksOJ |d |u sWJ d S )	Nr?   r  r'   ZNaNr   ro   r_   rI   )r   r   r   rR   r/   rO   rP   r  rA   rg   r>   )r   rw   nulls_fixturer#   r   r  Z	null_reprr\   r   r   r    test_format_missing  s   

$zTestIndex.test_format_missingopanyallc                 C   sN   |}t || }|t |j| ksJ t | | }t|t|ks%J d S r.   )r}   rB   Z	to_seriesr   )r   r$  r!   r#   r   r   r   r   r    test_logical_compat  s
   zTestIndex.test_logical_compat)r2   r   r   ri   r   c                 C   sx   t |}|ttdd }||}|ttdttd|  }t|| ||d }|dd  }t|| d S )Nr`   r   r   ro   )r   r   r   dropr/   r5   )r   r#   nr(  droppedr\   r   r   r    test_drop_by_str_label  s   
z TestIndex.test_drop_by_str_labelkeysr   1c                 C   s>   t jtdd || W d    d S 1 sw   Y  d S )N r'   )r)   r*   KeyErrorr(  )r   r#   r,  r   r   r    *test_drop_by_str_label_raises_missing_keys  s   "z4TestIndex.test_drop_by_str_label_raises_missing_keysc                 C   s   t |}|ttdd }| dg }|j|dd}|ttdttd|  }t|| |jddgdd}|tt| }t|| d S )Nr`   r   r   r  errorsr   )r   r   r   tolistr(  r/   r5   )r   r#   r)  r(  Zmixedr*  r\   r   r   r    $test_drop_by_str_label_errors_ignore  s   z.TestIndex.test_drop_by_str_label_errors_ignorec                 C   s2   t g d}|d}t ddg}t|| d S )Nr   ro   r_   rI   r   r(  r/   r5   )r   r#   r*  r\   r   r   r    test_drop_by_numeric_label_loc  s   
z(TestIndex.test_drop_by_numeric_label_locc                 C   sN   t g d}tjtdd |ddg W d    d S 1 s w   Y  d S )Nr   r.  r'   rI   rp   )r   r)   r*   r/  r(  r,   r   r   r    .test_drop_by_numeric_label_raises_missing_keys  s   "z8TestIndex.test_drop_by_numeric_label_raises_missing_keyszkey,expectedrp   r   )rI   rp   r`   c                 C   s*   t g d}|j|dd}t|| d S )Nr   r  r1  r5  )r   r$   r\   r#   r*  r   r   r    (test_drop_by_numeric_label_errors_ignore  s   z2TestIndex.test_drop_by_numeric_label_errors_ignore)r   r   r   r   )r   r9  r   )r9  r   r   to_dropr9  c           	   	   C   s   t |}t dgtd}||}t|| ||d }|d |d gfD ]}||}t|| q&||d }dt|d   d}|d |d gfD ]}tj	t
|d || W d    n1 sjw   Y  qQd S )Nr   r?   r   ro   z\"\[z\] not found in axis\"r'   )r   rR   r(  r/   r5   r   r   __repr__r)   r*   r/  )	r   rB   r:  r#   r\   rT   ZremovedZdrop_mer   r   r   r    test_drop_tuple  s    

zTestIndex.test_drop_tuplec                 C   sp   t |dkst|trtd t|trtstd | d}|dd  }|	|d }t
|| d S )Nr   z,Test doesn't make sense for empty MultiIndexz=Cannot test IntervalIndex with int64 dtype on 32 bit platformr_   )r   rA   r   r)   r  r   r   uniquerepeatr(  r/   r5   r   r   r   r    "test_drop_with_duplicates_in_index4  s   

z,TestIndex.test_drop_with_duplicates_in_indexis_monotonic_increasingis_monotonic_decreasingZ!_is_strictly_monotonic_increasingZ!_is_strictly_monotonic_decreasingc                 C   s$   t dt dg}t||rJ d S )Nr`      )r   r   r  r}   )r   ry   r#   r   r   r    test_is_monotonic_incomparable@  s   
z(TestIndex.test_is_monotonic_incomparable)r   r   quux>   r   rE  r   index,expected)quxr   r   r   FFTTr?   c                 C   s   | |}t|| d S r.   )isinr/   rD   )r   rB   r#   r\   rT   r   r   r    	test_isinM  s   
	zTestIndex.test_isinc                 C   s   t d|g}t|tr+t|tr+t|r+t|r+t||gt	ddg d S ||u r@t||gt	ddg d S |rX|j
dkrXt||gt	ddg d S t||gt	ddg d S )Nr   FTr2   )r   rA   rg   mathZisnanr/   rD   rI  r3   r4   r@   )r   r"  Znulls_fixture2rE   idxr   r   r    test_isin_nan_common_objectY  s6   



z%TestIndex.test_isin_nan_common_objectc                 C   s   |}|t ju s|t ju rJdtt|j }tjt|d t	d|g|d W d    n1 s0w   Y  t	dt
jg|d}||g rHJ d S t	d|g|d}|t
jg}t|t
ddg |t jg}t|t
ddg d S )Nz=float\(\) argument must be a string or a (real )?number, not r'         ?r?   FT)rC   NaTZNAreprr   __name__r)   r*   r~   r   r3   rh   rI  r%  r/   rD   r4   )r   r"  Zfloat_numpy_dtyper@   r   rL  resr   r   r    test_isin_nan_common_float64~  s"   z&TestIndex.test_isin_nan_common_float64levelr   r  )rN  r  rr   r!  c                 C   sZ   |  dd  dg }tg d}t||j||d d|_t||j|dd d S )NZnonexistingrH  rT  r   )r3  r3   r4   r/   rD   rI  r:   )r   rT  r#   rB   r\   r   r   r    test_isin_level_kwarg  s
   	zTestIndex.test_isin_level_kwargc              	   C   sZ   d|j |j d  fD ] }tjtdd |jg |d W d    n1 s%w   Y  q
d S )Nr   ro   zToo many levelsr'   rV  )Znlevelsr)   r*   r   rI  )r   r#   rT  r   r   r    &test_isin_level_kwarg_bad_level_raises  s   z0TestIndex.test_isin_level_kwarg_bad_level_raiseslabelrN  r   Zxyzzyc                 C   s   t |tr|ddg|jdd   }d| d}n|d}d| d}tjt|d |jg |d	 W d    d S 1 s?w   Y  d S )
Nr   r   r_   z'Level z not found'zRequested level \(z$\) does not match index name \(foo\)r'   rV  )rA   r   r   namesr)   r*   r/  rI  )r   rY  r#   r   r   r   r    &test_isin_level_kwarg_bad_label_raises  s   

"z0TestIndex.test_isin_level_kwarg_bad_label_raisesc                 C   s4   t ddg}tddg}||}t|| d S )Nr   r   F)r   r3   r4   rI  r/   rD   )r   r   r#   r\   rT   r   r   r    test_isin_empty  s   
zTestIndex.test_isin_emptyc                 C   s:   t ddg|d}|d g}tddg}t|| d S )Nr   r   r?   F)r   rI  r3   r4   r/   rD   )r   Zstring_dtype_no_objectr#   rT   r\   r   r   r    test_isin_string_null  s   zTestIndex.test_isin_string_nullro   r_   rI   rp   TTTT)r   r   r   rG  z
2018-01-01c                 C   s2   t |}||k}tjg dtd}t|| d S )Nr_  r?   )r   r3   r4   r   r/   rD   )r   rB   r#   rT   r\   r   r   r    test_boolean_cmp  s   zTestIndex.test_boolean_cmpz
name,levelr   r   c                 C   s,   |  }|r	||_||}t|| d S r.   )r=   r:   Zget_level_valuesr/   r5   )r   r#   r:   rT  r\   rT   r   r   r    test_get_level_values  s
   
zTestIndex.test_get_level_valuesc                 C   s,   t ddgdd}|j|dd  jksJ d S )Nr   r   Zasdfr;   ro   r   r:   r,   r   r   r    test_slice_keep_name  s   zTestIndex.test_slice_keep_name)r2   r   r   r   r   r   ri   r   c                 C   s2   |j ||d}|}|dkr| }t|| d S )Nhowouter)joinZsort_valuesr/   r5   )r   r#   Z	join_typerT   r\   r   r   r    test_join_self  s
   zTestIndex.test_join_selfmethod)striprstriplstripc                    sB   t g d}t  fdd|jD }t|j  }t|| d S )N)z jackzjill z jesse Zfrankc                    s   g | ]	}t t |qS r   )r}   r>   r   r   ri  r   r    r      s    z0TestIndex.test_str_attribute.<locals>.<listcomp>)r   rB   r}   r>   r/   r5   )r   ri  r#   r\   rT   r   rn  r    test_str_attribute  s   zTestIndex.test_str_attributer`   )r   r-  )r   3Z2000Z2010Y)startendrJ   c                 C   s@   t jtdd |jd W d    d S 1 sw   Y  d S )Nzonly use .str accessorr'   r_   )r)   r*   AttributeErrorr>   r>  r,   r   r   r    test_str_attribute_raises  s   
"z#TestIndex.test_str_attribute_raiseszexpand,expectedNr   r   fc                 C   s>   t g d}|d ur|jj|d}n|j }t|| d S )N)za b czd erv  )expand)r   r>   splitr/   r5   )r   rw  r\   r#   rT   r   r   r    test_str_split  s
   
zTestIndex.test_str_splitc                 C   sF   t g d}|jd}tg d}t|| t|tjs!J d S )Na1a2Zb1Zb2r   )TTFF)	r   r>   
startswithr3   r4   r/   rD   rA   Zndarrayr   r   r   r    test_str_bool_return(  s
   zTestIndex.test_str_bool_returnc                 C   sR   t g d}ttd|d}||jjd }ttdddgd}t|| d S )Nrz  rp   r  r   r_   r{  r|  )r   r   r   r#   r>   r}  r/   rd   )r   r#   r]   rT   r\   r   r   r    test_str_bool_series_indexing1  s
   z'TestIndex.test_str_bool_series_indexingZabcdc                 C   s   dt |v }||ksJ d S )Nr>   )dirr   r   r   r    test_tab_completion9  s   zTestIndex.test_tab_completionc                 C   sX   t g d}|dd t ddgtjdsJ |ddg t ddgtjds*J d S )N)ro   r_   rI   r   r   r   ro   rI   r_   r?   r   )r   r   r3   rS   r,   r   r   r    !test_indexing_doesnt_change_classA  s   $(z+TestIndex.test_indexing_doesnt_change_classc                 C   s   t tjdd}tddd}tt |j	|dd}W d    n1 s(w   Y  tt |
t|
t}W d    n1 sHw   Y  t|| d S )Nr_      r   r   r|   rf  rd  )r   r3   rb   rc   Zpermutationr   r/   rO   RuntimeWarningrg  rQ   rR   r   r5   )r   Z
left_indexZright_indexrT   r\   r   r   r    test_outer_join_sortG  s   zTestIndex.test_outer_join_sortc                 C   s   t tddd}|tg d}t tddd}t|| |jtg ddd}t dd	tjgdd}t|| |jtg dd
dd}t g ddd}t|| d S )NABCxxxr;   )ro   r   r  ZBACT
fill_valuert   rs   F)Z
allow_fillr  )rt   rs   ru   )r   r   taker3   r4   r/   r5   rh   r   r   r   r    test_take_fill_valueS  s   zTestIndex.test_take_fill_valuec                 C   s   t tddd}d}tjt|d |jtg ddd W d    n1 s(w   Y  tjt|d |jtg d	dd W d    d S 1 sLw   Y  d S )
Nr  r  r;   zJWhen allow_fill=True and fill_value is not None, all indices must be >= -1r'   )ro   r   rU  Tr  )ro   r   )r   r   r)   r*   r+   r  r3   r4   r   r   r   r     test_take_fill_value_none_raisesd  s   "z*TestIndex.test_take_fill_value_none_raisesc                 C   sX   t tddd}tjtdd |tddg W d    d S 1 s%w   Y  d S )Nr  r  r;   zout of boundsr'   ro   r  )r   r   r)   r*   r   r  r3   r4   r,   r   r   r    test_take_bad_bounds_raisesp  s   "z%TestIndex.test_take_bad_bounds_raisesr:   labelsrs   rt   ru   )ru   rt   rs   Z20130101c                 C   s.   t g d}||_||d j|ksJ d S )N)r   ro   r_   r   )r   r:   reindex)r   r:   r  r#   r   r   r    8test_reindex_preserves_name_if_target_is_list_or_ndarrayu  s   zBTestIndex.test_reindex_preserves_name_if_target_is_list_or_ndarrayc                 C   s.   t td}||d jj|jjksJ d S Nr9   r   r   r   r  r@   r   )r   r  r#   r   r   r    <test_reindex_preserves_type_if_target_is_empty_list_or_array  s   "zFTestIndex.test_reindex_preserves_type_if_target_is_empty_list_or_arrayzlabels,dtypec                 C   s*   t td}||d jj|ksJ d S r  r  )r   r  r@   r#   r   r   r    :test_reindex_doesnt_preserve_type_if_target_is_empty_index  s   zDTestIndex.test_reindex_doesnt_preserve_type_if_target_is_empty_indexc                 C   s8   |}t td}t g |d}||d j|ksJ d S )Nr9   r?   r   )r   r   r  r@   )r   Zany_real_numpy_dtyper@   r#   r  r   r   r    Btest_reindex_doesnt_preserve_type_if_target_is_empty_index_numeric  s   zLTestIndex.test_reindex_doesnt_preserve_type_if_target_is_empty_index_numericc                 C   sl   t td}|tt g tjt g tjgg g gd }|jd jj	tjks(J |jd jj	tjks4J d S )Nr9   r   ro   )
r   r   r  r   r3   r   ri   r   r@   r   )r   r#   rT   r   r   r    -test_reindex_no_type_preserve_target_empty_mi  s    z7TestIndex.test_reindex_no_type_preserve_target_empty_mic                 C   sR   t g ddd}t g ddd}t g ddd}|j|dd\}}t|| d S )Nr   r   r;   r^  rV  )r   r  r/   r5   )r   rL  Zidx2r\   rT   r   r   r   r    test_reindex_ignoring_level  s
   z%TestIndex.test_reindex_ignoring_levelc                 C   sJ   t td}|tg d}t ddgt g dd}t|| d S )Nr`   )ro   ro   r_   r_   r_   r   ro   )r_   rI   rp   ro   r_   )r   r   groupbyr3   r4   r/   Zassert_dict_equalr   r   r   r    test_groupby  s   zTestIndex.test_groupbyzmi,expectedr  rp   r`   )rp      c                 C   sH   t ddgdgtjddgddggdd	gd
d}|j|k}t|| d S )NrI   r  r   ro   rp   r_   r`   r   r   rZ  columnsr#   )r   r   from_arraysr#   r/   rD   )r   mir\   re   rT   r   r   r    test_equals_op_multiindex  s   

z#TestIndex.test_equals_op_multiindexc                 C   sX   t ddgdgtjddgddggdd	gd
d}|j|jk}tddg}t|| d S )NrI   r  r   ro   rp   r_   r`   r   r   r  r  T)r   r   r  r#   r3   r4   r/   rD   )r   re   rT   r\   r   r   r    "test_equals_op_multiindex_identify  s   z,TestIndex.test_equals_op_multiindex_identifyr  r  )   	   r   c                 C   sl   t ddgdgtjddgddggdd	gd
d}tjtdd |j|k W d    d S 1 s/w   Y  d S )NrI   r  r   ro   rp   r_   r`   r   r   r  r  zLengths must matchr'   )r   r   r  r)   r*   r+   r#   )r   r#   re   r   r   r    +test_equals_op_mismatched_multiindex_raises  s   "z5TestIndex.test_equals_op_mismatched_multiindex_raisesc                 C   s@   t g d}tg d}||k}tg d}t|| d S )Nr  r   )FFF)r   r   r   r3   r4   r/   rD   )r   rE   r  r#   rT   r\   r   r   r    &test_equals_op_index_vs_mi_same_length  s
   z0TestIndex.test_equals_op_index_vs_mi_same_lengthzdt_conv, arg
2000-01-01
2000-01-02z01:02:03z01:02:04c                 C   s$   t |dd}|j||jksJ d S )NrY  r;   rb  )r   Zdt_convr   r#   r   r   r    !test_dt_conversion_preserves_name  s   	z+TestIndex.test_dt_conversion_preserves_namec                 C   sF   t g d}tjtdd d|_W d    d S 1 sw   Y  d S )Nr   zCan't set attributer'   F)r   r)   r*   rt  r  r,   r   r   r    #test_cached_properties_not_settable  s   "z-TestIndex.test_cached_properties_not_settablec              	   C   s   t jddd ddlm} d}|| tjd dd. |d	 t|j	d
d W d    n1 s4w   Y  W d    d S W d    d S 1 sLw   Y  d S )NZIPythonz6.0.0)Z
minversionr   )provisionalcompleterz+import pandas as pd; idx = pd.Index([1, 2])F)Zraise_on_extra_warningsr  zidx.rp   )
r)   ZimportorskipZIPython.core.completerr  Zrun_cellr/   rO   r   Z	CompleterZcompletions)r   Zipr  coder   r   r    test_tab_complete_warning  s   

"z#TestIndex.test_tab_complete_warningc                 C   sh   t |tr|d d S dt|j d}tjt|d |d W d    d S 1 s-w   Y  d S )Nro   'z$' object has no attribute 'contains'r'   )rA   r   containsr   rQ  r)   r*   rt  r   r   r   r    test_contains_method_removed  s   
"z&TestIndex.test_contains_method_removedc                 C   s>  t g d}tjtdd |jdd W d    n1 sw   Y  tjtdd |jddgd W d    n1 s=w   Y  tjtdd |jdgd W d    n1 s[w   Y  t g d	}|jdgd}t|d
 | t g d	}|jdd}t|d
 | t g d}|jdd}t|d
 | d S )N)r`   rp   rI   r_   ro   z(ascending must be a single bool value orr'   True)Z	ascendingz3ascending must be a list of bool values of length 1Tzascending must be a bool valuern   r   F)r   r)   r*   	Exception	sortlevelr/   r5   r   r   r   r    test_sortlevel  s*   zTestIndex.test_sortlevelc                 C   s<   t dtjg}|jddd }t tjdg}t|| d S )Nro   first)Zna_positionr   )r   r3   rh   r  r/   r5   r   rL  rT   r\   r   r   r    test_sortlevel_na_position3  s   z$TestIndex.test_sortlevel_na_positionzperiods, expected_results   r   c                 C   .   t g d}||}t |}t|| d S )N)r   r  r   (   rN   )r   Zdiffr/   r5   )r   rK   expected_resultsrL  rT   r\   r   r   r    test_index_diff:     

zTestIndex.test_index_diffzdecimals, expected_resultsrN  r  rr   )g333333?gffffff@g      @)gGz?g@gGz@c                 C   r  )N)gX9v?g(\@gS@)r   roundr/   r5   )r   Zdecimalsr  rL  rT   r\   r   r   r    test_index_roundJ  r  zTestIndex.test_index_round)rQ  
__module____qualname__r)   fixturer   r!   r%   r   parametrizer-   r1   r8   rF   r   r   r   rU   rV   r^   rf   rk   r3   ri   rh   r   rC   rO  rm   rx   r   r   r   iterr
   r   r   r   r   r   r   Zparamr   r   r   r   r   r   r   r   r   r   rq   r   r   r   r   r   r   r   r   r  filterwarningsr
  r   r  r  r  r  r  r  r  r   r#  r'  r+  r0  r4  r6  r7  r8  r<  r?  rD  r4   r   rJ  rM  rS  rW  rX  r[  rR   r\  r]  r`  ra  rc  rh  ro  r   r   ru  ry  r~  r  r   r  r  r  r  r  r  rB   r3  r  r   r  Z
datetime64r  r  r  r  r  r  r  r  r  rZ   to_timedeltar  r  r  r  r  r  r  r  r   r   r   r    r   -   s   






9 	

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





	"  		"&	
	r   c                   @   sx  e Zd ZejdefddZdd Zdd Zdd	 Z	d
d Z
dd Zdd Zejdddgejddedgejdg dg dfg dg dfddejdgg dfg dg dfdejddgg dfgd d! Zejdddgejd"eg d#eg d#fed$d%d&ejgeg d#feg d'eg d'feejd(d)d*ejgeg d'feg d+d,d-eg d+d,d-feg d.d,d-eg d+d,d-fgd/d0 Zd1d2 Zejd3eejgeejd4ged4d5ejged6d7ejged8ged8d9geg d:ed;d8ggd<d= Zejdg d>ejd?d@dAgdBdC ZdDdE ZdFdG Zej dHdIdJ Z!dS )KTestMixedIntIndexr   c                 C   s   t g dS )N)r   r   ro   r   r_   r   r   r   r   r   r    r!   _  r"   zTestMixedIntIndex.simple_indexc                 C   s@   |}t jtdd |  W d    d S 1 sw   Y  d S Nz'>|<' not supportedr'   )r)   r*   r~   argsortr   r!   r#   r   r   r    test_argsortc  s   
"zTestMixedIntIndex.test_argsortc                 C   sB   |}t jtdd t| W d    d S 1 sw   Y  d S r  )r)   r*   r~   r3   r  r  r   r   r    test_numpy_argsorth  s   "z$TestMixedIntIndex.test_numpy_argsortc                 C   s   |}t ||ddd}t ||dd}||usJ t|| |jdks&J |jdks-J td|d}td|d d	 d}|| }|jjdksIJ d S )
NTZmarior<   Fr   r_   r  rI   r  )r   r/   r5   r:   r   r#   )r   r!   r#   r  seconds1s2Zs3r   r   r    test_copy_namem  s   z TestMixedIntIndex.test_copy_namec                 C   s`   t ddgdd}| }t|| |jdd}tj||dd |jdks'J |jdks.J d S )Nro   r_   ZMyNamer;   ZNewNameF)Zcheck_names)r   r=   r/   r5   r:   )r   r#   Zindex1Zindex2r   r   r    test_copy_name2  s   z!TestMixedIntIndex.test_copy_name2c                 C   sB   t dtjddgdd}t dtjdgdd}| }t|| d S )Nr_   ro   Zmy_indexr;   )r   r3   rh   r=  r/   r5   )r   rL  r\   rT   r   r   r    test_unique_na  s   z TestMixedIntIndex.test_unique_nac                 C   s4   |}|  |j  ksJ | |j ksJ d S r.   )r&  rB   r%  r  r   r   r    r'    s   z%TestMixedIntIndex.test_logical_compatre  r%  r&  r@   Ncategoryzvals,expectedr   r  rN  r  rr   r  rs   rt   ru   c                 C   s4   t ||d}|j|d}t ||d}t|| d S )Nr?   rd  )r   dropnar/   r5   )r   re  r@   rw   r\   r#   rT   r   r   r    test_dropna  s   zTestMixedIntIndex.test_dropnarF  )r{   
2011-01-02
2011-01-03r{   r  r  )rM   2 days3 daysrM   r  r  )2012-022012-042012-05MrY   )r  r  rO  r  c                 C   s   |j |d}t|| d S )Nrd  )r  r/   r5   )r   re  r#   r\   rT   r   r   r    test_dropna_dt_like  s   z%TestMixedIntIndex.test_dropna_dt_likec                 C   sL   d}t jt|d tg djdd W d    d S 1 sw   Y  d S )Nzinvalid how option: xxxr'   r   r  rd  )r)   r*   r+   r   r  )r   r   r   r   r    test_dropna_invalid_how_raises  s   "z0TestMixedIntIndex.test_dropna_invalid_how_raisesr#   ro   r_   r   r   rO  r  )r  rO  r  r   c                 C   s<   |j du sJ |jdu sJ |jdu sJ |jdu sJ d S NFr@  r,   r   r   r    test_is_monotonic_na  s   z&TestMixedIntIndex.test_is_monotonic_na)Zf8zm8[ns]zM8[us]unique_firstTFc                 C   sp   t g d|d}|r!|jdu sJ |jdu sJ |jdu sJ d S |jdu s(J |jdu s/J |jdu s6J d S )N)Nro   ro   r?   F)r   r  rA  rB  )r   r@   r  r#   r   r   r    test_is_monotonic_unique_na  s   z-TestMixedIntIndex.test_is_monotonic_unique_nac                 C   s8   t g ddd}|ttd|d}dt|v sJ d S )Nr   r   r;   rI   r  0)r   r   r   rP  )r   Zframe_or_seriesr#   rT   r   r   r    test_int_name_format  s   z&TestMixedIntIndex.test_int_name_formatc                 C   sV   t dd tdD }d}tjt|d t| W d    d S 1 s$w   Y  d S )Nc                 S   s   g | ]}t |qS r   )r>   rm  r   r   r    r     r  z>TestMixedIntIndex.test_str_to_bytes_raises.<locals>.<listcomp>r   z2^'str' object cannot be interpreted as an integer$r'   )r   r   r)   r*   r~   bytesr   r   r   r    test_str_to_bytes_raises  s
   
"z*TestMixedIntIndex.test_str_to_bytes_raisesz2ignore:elementwise comparison failed:FutureWarningc                 C   s2   t g d}|dk}tg d}t|| d S )N)r   )r   r   r   r   r  )FFT)r   r3   r4   r/   rD   r  r   r   r    test_index_with_tuple_bool  s   z,TestMixedIntIndex.test_index_with_tuple_bool)"rQ  r  r  r)   r  r   r!   r  r  r  r  r  r'  r   r  rR   r3   rh   r  r   rC   rO  r   r   r  r  rZ   r  r  r  r  r  r  r  r   r   r   r    r  [  s    










r  c                   @   s   e Zd Zejdg dgdeg dfg dgdgeg dddfddgddggdedgddggd	d	gd	d
ggfddgddggddgedgddggd	d	gd	d
ggddgdfgdd Zdd Z	dd Z
dd ZdS )TestIndexUtilszdata, names, expectedr   Nr:   r;   r   r   r   r   ro   ZL1ZL2r  c                 C   s   t ||}t|| d S r.   )r   r/   r5   )r   rj   rZ  r\   rT   r   r   r     test_ensure_index_from_sequences  s   
z/TestIndexUtils.test_ensure_index_from_sequencesc                 C   s`   t jddddt jddddt jddddt jdd	d
dg}t|}t|td}t|| d S )Nr   ro   r   )closedr_   r   rI   Zneitherrp   Zbothr?   )rC   Intervalr   r   rR   r/   r5   )r   Z	intervalsrT   r\   r   r   r    (test_ensure_index_mixed_closed_intervals#  s   z7TestIndexUtils.test_ensure_index_mixed_closed_intervalsc                 C   sF   dt t jjg}t|}t||ksJ t|dd}t|| d S )Nr   r   r?   )	r3   Ziinfor   maxr   r   r   r/   r5   )r   rB   rT   r\   r   r   r    test_ensure_index_uint64/  s
   z'TestIndexUtils.test_ensure_index_uint64c                 C   s    t g }tg }t|| d S r.   )r   r   r/   r5   )r   rT   r\   r   r   r    test_get_combined_index;  s   z&TestIndexUtils.test_get_combined_index)rQ  r  r  r)   r   r  r   r   r  r  r  r  r   r   r   r    r    s&    &
r  opname)eqneleltgegtaddZraddsubZrsubmulZrmultruedivZrtruedivfloordivZ	rfloordivpowZrpowmoddivmodc                 C   s(   d|  d} t || }|j| ksJ d S )N__)r}   rQ  )r  r#   ri  r   r   r    test_generated_op_namesA  s   
r  rl   ro   )rj   r   r   r   )rj   r@   r   c                 C   s>   t jtdd | dd W d    d S 1 sw   Y  d S )Nzunexpected keyword argumentr'   r   )r   )r)   r*   r~   )rl   r   r   r    ,test_index_subclass_constructor_wrong_kwargs`  s   "r	  c                  C   s  d} t jt| d ttjddgtdddd W d    n1 s"w   Y  t jt| d ttjg d	d
dddd W d    n1 sGw   Y  t jt| d tdddddd W d    n1 sgw   Y  t jt| d tg dddd W d    d S 1 sw   Y  d S )Nz[Uu]nexpected keyword argumentr'   r   r   r?   testT)r:   Zfastpathr   r   r   r`   r_   r   )	r)   r*   r~   r   r3   r4   rR   r   r
   )r   r   r   r    test_deprecated_fastpaths  s   "r  c                  C   sP   t g d} tjtdd | d d d f  W d    d S 1 s!w   Y  d S )N)r   ro   r_   rI   r&   r'   )r   r)   r*   r+   )rL  r   r   r    test_shape_of_invalid_index  s   "r  r@   c                 C   s   d}t dddd}tjt|d t|| d W d    n1 s$w   Y  t|dd}tjt|d t|| d W d    n1 sIw   Y  tdt	d}tjt|d t j
ddggd | d|_W d    d S 1 suw   Y  d S )	Nz Index data must be 1-dimensionalr  r_   r'   r?   rp   r   rI   )r3   r   Zreshaper)   r*   r+   r   r   r   r   r4   r#   )r@   r   r6   re   Zserr   r   r    test_validate_1d_input  s   "r  zklass, extra_kwargsc                    s   g | ]
  fd di gqS )c                    s   t |  dS )Nr?   r  r   Zdtypr   r    r     r   z<listcomp>.<lambda>r   )r   r   r  r    r     s    r   rJ   rq  c                 C   sJ   | t tddfi |}| ttddfi |}tj||dd d S )Ni  i  Tr   )
memoryviewr3   r   r   r   r/   r5   )rl   Zextra_kwargsrT   r\   r   r   r    test_construct_from_memoryview  s   r  r$  c                 C   sL   t tjg}tdg}| ||}t|| | || }t|| d S r  )r   r3   rh   r4   r/   rD   r=   )r$  rL  r\   rT   r   r   r    test_nan_comparison_same_object  s   
r  pyarrowc                  C   sF   dd l } tdgddggt| |  d}|jrJ |jr!J d S )Nr   ro   r_   rI   r?   )r  r   rC   Z
ArrowDtypeZlist_r   rA  rB  )parL  r   r   r    #test_is_monotonic_pyarrow_list_type  s   &
r  )Dcollectionsr   r   	functoolsr   rK  operatorr   Znumpyr3   r)   Zpandas.compatr   Zpandas.errorsr   Zpandas.util._test_decoratorsutilZ_test_decoratorsZtdZpandas.core.dtypes.commonr   r   r	   ZpandasrC   r
   r   r   r   r   r   r   r   r   r   r   Zpandas._testingZ_testingr/   Zpandas.core.indexes.apir   r   r   r   r   r   r  r  r   r  r  r   r  rR   r	  r  r  r   r   ri   r  ZALL_REAL_NUMPY_DTYPESr  r  r  r  Z
skip_if_nor  r   r   r   r    <module>   s    4	          8 34






