o
    S+ io                     @   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   j/home/app/PaddleOCR-VL-test/.venv_paddleocr/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   )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_tzrU   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srU   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 )	NrX   rY   rZ      )   rI   dater>   r;   )r   r   r3   randomdefault_rngrS   r@   r:   r/   r5   r   assert_series_equalrC   Z
infer_freq)r   rE   r\   r]   dfrU   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]   rU   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]   rU   r   r   r    test_index_ctor_infer_nan_nat   s   	z'TestIndex.test_index_ctor_infer_nan_natz
vals,dtype   r`   rI      ra   intg?g@      @rh   ABCobjc                 C   s*   t ||d}||j|}t|| d S )Nr;   )r   Z_simple_newrB   r/   r5   )r   valsr@   r#   rU   r   r   r    test_constructor_simple_new   s   	z%TestIndex.test_constructor_simple_newattrrB   asi8rm   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-01ra   rK   z:Cannot use .astype to convert from timezone-naive dtype tor{   zunexpected keywordr'   )rL   r?   )r   getattrZtz_localizer@   r   r/   r5   r   r)   r*   	TypeErrorrR   r   )
r   Ztz_naive_fixturerz   rm   r#   argr@   errmsgrU   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   ra   r}   r?   )r   Z
_with_freqr@   r~   r/   r5   r   )r   rz   rm   r#   r@   rB   rU   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   rm   emptyr   r   r    test_constructor_empty#  s   z TestIndex.test_constructor_emptyzempty,klassrZ   c                 c   r   r.   r   r   r   r   r    r   7  r   rp   )stepr`   blueredlevelscodesc                 C   s   t ||sJ t|rJ d S r.   r   )r   r   rm   r   r   r    test_constructor_empty_special2  s   z(TestIndex.test_constructor_empty_special)r   rj   float32int64int32ZperiodrangeZrepeats	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   ra   foobar)rR   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   i1i2Zi3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)deepZbobrp      r=   )	r   r   is_r   r=   r3   r4   r:   rB   )r   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rp   r`   rI   TFr?   z$Cannot compare dtypes int64 and boolr'   r   )
r   rS   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?   Fra      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   rj   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/   rP   rQ   )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   rj   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   appendrR   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 Nrp   r   r   r   r   r    r         c                 S   s   g | ]}||d kfqS )rp   r   r   r   r   r    r     s    )	r   r3   aranger   mapr/   r5   r   from_tuplesr   r#   rU   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 )Nrp   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]   rU   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.   )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   rU   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#   rU   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   rS   r   r/   r5   )r   r#   r  r   rngr]   rU   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`   rp   r   r   r3   ri   r   r/   r5   )r   r  r]   rU   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   ra   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>stuffrp   )r  r  r  )r   r   r   r/   r5   )r   r#   Zdefault_dictrU   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/   rP   rQ   r   r:   )r   r:   r]   r   r   r   rU   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)rj   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"  )rj   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/   rP   rQ   format)r   r'  r   r#   	formattedr]   r   r   r    test_format_bug  s   

"zTestIndex.test_format_bugrx   )rp   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'   NaNr   rp   r`   rI   )r   r   r   rS   r/   rP   rQ   r)  rA   rh   r>   )r   rx   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   r0  r!   r#   r   r   r   r   r    test_logical_compat  s
   zTestIndex.test_logical_compat)r2   r   r   rj   r   c                 C   sx   t |}|ttdd }||}|ttdttd|  }t|| ||d }|dd  }t|| d S )Nra   r   r   rp   )r   r   r   dropr/   r5   )r   r#   nr4  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*   KeyErrorr4  )r   r#   r8  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 )Nra   r   r   r  errorsr   )r   r   r   tolistr4  r/   r5   )r   r#   r5  r4  mixedr6  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   rp   r`   rI   r   r4  r/   r5   )r   r#   r6  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   rq   )r   r)   r*   r;  r4  r,   r   r   r    .test_drop_by_numeric_label_raises_missing_keys  s   "z8TestIndex.test_drop_by_numeric_label_raises_missing_keyszkey,expectedrq   r   )rI   rq   ra   c                 C   s*   t g d}|j|dd}t|| d S )Nr   r  r=  rB  )r   r$   r]   r#   r6  r   r   r    (test_drop_by_numeric_label_errors_ignore  s   z2TestIndex.test_drop_by_numeric_label_errors_ignore)r   r   r   r   )r   rF  r   )rF  r   r   to_droprF  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   rp   z\"\[z\] not found in axis\"r'   )r   rS   r4  r/   r5   r   r   __repr__r)   r*   r;  )	r   rB   rG  r#   r]   rU   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repeatr4  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 )Nra      )r   r   r'  r~   )r   rz   r#   r   r   r    test_is_monotonic_incomparable@  s   
z(TestIndex.test_is_monotonic_incomparable)r   r   quux>   r   r   rS  index,expected)quxr   r   r   FFTTr?   c                 C   s   | |}t|| d S r.   )isinr/   rD   )r   rB   r#   r]   rU   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   rh   mathisnanr/   rD   rW  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   ri   rW  r1  r/   rD   r4   )r   r.  Zfloat_numpy_dtyper@   r   r[  resr   r   r    test_isin_nan_common_float64~  s"   z&TestIndex.test_isin_nan_common_float64levelr   r  )r]  r  rs   r,  c                 C   sZ   |  dd  dg }tg d}t||j||d d|_t||j|dd d S )NZnonexistingrV  rc  r   )r?  r3   r4   r/   rD   rW  r:   )r   rc  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   rp   zToo many levelsr'   re  )Znlevelsr)   r*   r   rW  )r   r#   rc  r   r   r    &test_isin_level_kwarg_bad_level_raises  s   z0TestIndex.test_isin_level_kwarg_bad_level_raiseslabelr]  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'   re  )rA   r   r   namesr)   r*   r;  rW  )r   rh  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   rW  r/   rD   )r   r   r#   r]   rU   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   rW  r3   r4   r/   rD   )r   Zstring_dtype_no_objectr#   rU   r]   r   r   r    test_isin_string_null  s   zTestIndex.test_isin_string_nullrp   r`   rI   rq   TTTT)r   r   r   rU  z
2018-01-01c                 C   s2   t |}||k}tjg dtd}t|| d S )Nrn  r?   )r   r3   r4   r   r/   rD   )r   rB   r#   rU   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:   rc  r]   rU   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;   rp   r   r:   r,   r   r   r    test_slice_keep_name  s   zTestIndex.test_slice_keep_name)r2   r   r   r   r   r   rj   r   c                 C   s2   |j ||d}|}|dkr| }t|| d S )Nhowouter)joinZsort_valuesr/   r5   )r   r#   Z	join_typerU   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   rx  r   r    r      s    z0TestIndex.test_str_attribute.<locals>.<listcomp>)r   rB   r~   r>   r/   r5   )r   rx  r#   r]   rU   r   r}  r    test_str_attribute  s   zTestIndex.test_str_attributera   )r   r9  )r   3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>   rL  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 er  )expand)r   r>   splitr/   r5   )r   r  r]   r#   rU   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a2b1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 )Nr  rq   r  r   r`   r  r  )r   r   r   r#   r>   r  r/   re   )r   r#   r^   rU   r]   r   r   r    test_str_bool_series_indexing1  s
   z'TestIndex.test_str_bool_series_indexing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)rp   r`   rI   r   r   r   rp   rI   r`   r?   r   )r   r   r3   rT   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}   ru  rs  )r   r3   rc   rd   Zpermutationr   r/   rP   RuntimeWarningrv  rR   rS   r   r5   )r   Z
left_indexZright_indexrU   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;   )rp   r   r  ZBACT
fill_valueru   rt   F)Z
allow_fillr  )ru   rt   rv   )r   r   taker3   r4   r/   r5   ri   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'   )rp   r   rd  Tr  )rp   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'   rp   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rt   ru   rv   )rv   ru   rt   Z20130101c                 C   s.   t g d}||_||d j|ksJ d S )N)r   rp   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   rp   )
r   r   r  r   r3   r   rj   r   r@   r   )r   r#   rU   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;   rm  re  )r   r  r/   r5   )r   r[  Zidx2r]   rU   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 )Nra   )rp   rp   r`   r`   r`   r   rp   )r`   rI   rq   rp   r`   )r   r   groupbyr3   r4   r/   Zassert_dict_equalr   r   r   r    test_groupby  s   zTestIndex.test_groupbyzmi,expectedr  rq   ra   )rq      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   rp   rq   r`   ra   r   r   ri  columnsr#   )r   r   from_arraysr#   r/   rD   )r   mir]   rf   rU   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   rp   rq   r`   ra   r   r   r  r  T)r   r   r  r#   r3   r4   r/   rD   )r   rf   rU   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   rp   rq   r`   ra   r   r   r  r  zLengths must matchr'   )r   r   r  r)   r*   r+   r#   )r   r#   rf   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#   rU   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 )Nrh  r;   rq  )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*   r  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.rq   )
r)   ZimportorskipZIPython.core.completerr  Zrun_cellr/   rP   r   Z	CompleterZcompletions)r   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 )Nrp   'z$' object has no attribute 'contains'r'   )rA   r   containsr   r`  r)   r*   r  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)ra   rq   rI   r`   rp   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 valuero   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 )Nrp   first)Zna_positionr   )r   r3   ri   r  r/   r5   r   r[  rU   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   diffr/   r5   )r   rK   expected_resultsr[  rU   r]   r   r   r    test_index_diff:     

zTestIndex.test_index_diffzdecimals, expected_resultsr]  r  rs   )g333333?gffffff@g      @)gGz?g@gGz@c                 C   r  )N)gX9v?g(\@gS@)r   roundr/   r5   )r   Zdecimalsr  r[  rU   r]   r   r   r    test_index_roundJ  r  zTestIndex.test_index_round)r`  
__module____qualname__r)   fixturer   r!   r%   r   parametrizer-   r1   r8   rF   r   r   r   rV   rW   r_   rg   rl   r3   rj   ri   r   rC   r^  rn   ry   r   r   r   iterr
   r   r   r   r   r   r   paramr   r   r   r   r   r   r   r   r   r   rr   r   r   r   r   r   r   r   r  r  filterwarningsr  r   r  r  r  r  r#  r$  r%  r+  r/  r3  r7  r<  rA  rC  rD  rE  rJ  rM  rR  r4   r   rX  r\  rb  rf  rg  rj  rS   rk  rl  ro  rp  rr  rw  r~  r   r   r  r  r  r  r   r  r  r  r  r  r  rB   r?  r  r   r  Z
datetime64r  r  r  r  r  r  r  r  r  r[   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   rp   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 )Nrp   r`   ZMyNamer;   ZNewNameF)Zcheck_names)r   r=   r/   r5   r:   )r   r#   Zindex1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`   rp   Zmy_indexr;   )r   r3   ri   rK  r/   r5   )r   r[  r]   rU   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.   )r2  rB   r1  r  r   r   r    r3    s   z%TestMixedIntIndex.test_logical_compatrt  r1  r2  r@   Ncategoryzvals,expectedr   r  r]  r  rs   r  rt   ru   rv   c                 C   s4   t ||d}|j|d}t ||d}t|| d S )Nr?   rs  )r   dropnar/   r5   )r   rt  r@   rx   r]   r#   rU   r   r   r    test_dropna  s   zTestMixedIntIndex.test_dropnarT  )r|   
2011-01-02
2011-01-03r|   r  r  )rM   2 days3 daysrM   r  r  )2012-022012-042012-05MrZ   )r  r  r^  r  c                 C   s   |j |d}t|| d S )Nrs  )r  r/   r5   )r   rt  r#   r]   rU   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  rs  )r)   r*   r+   r   r  )r   r   r   r   r    test_dropna_invalid_how_raises  s   "z0TestMixedIntIndex.test_dropna_invalid_how_raisesr#   rp   r`   r   r   r^  r  )r  r^  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 NFrN  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)Nrp   rp   r?   F)r   r  rO  rP  )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   r_  )r   Zframe_or_seriesr#   rU   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>   r|  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)"r`  r  r  r)   r  r   r!   r  r  r  r  r  r3  r   r  rS   r3   ri   r  r   rC   r^  r   r   r  r  r[   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   rp   ZL1ZL2r  c                 C   s   t ||}t|| d S r.   )r   r/   r5   )r   rk   ri  r]   rU   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   rp   r   )closedr`   r   rI   Zneitherrq   Zbothr?   )rC   Intervalr   r   rS   r/   r5   )r   Z	intervalsrU   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   rU   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   rU   r]   r   r   r    test_get_combined_index;  s   z&TestIndexUtils.test_get_combined_index)r`  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~   r`  )r  r#   rx  r   r   r    test_generated_op_namesA  s   
r  rm   rp   )rk   r   r   r   )rk   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   )rm   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   ra   r`   r   )	r)   r*   r   r   r3   r4   rS   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   rp   r`   rI   r&   r'   )r   r)   r*   r+   )r[  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?   rq   r   rI   )r3   r   Zreshaper)   r*   r+   r   r   r   r   r4   r#   )r@   r   r6   rf   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   r  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   )rm   extra_kwargsrU   r]   r   r   r    test_construct_from_memoryview  s   r(  r0  c                 C   sL   t tjg}tdg}| ||}t|| | || }t|| d S r  )r   r3   ri   r4   r/   rD   r=   )r0  r[  r]   rU   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   rp   r`   rI   r?   )r*  r   rC   Z
ArrowDtypelist_r   rO  rP  )par[  r   r   r    #test_is_monotonic_pyarrow_list_type  s   &
r-  )Dcollectionsr   r   	functoolsr   rY  operatorr   numpyr3   r)   Zpandas.compatr   Zpandas.errorsr   Zpandas.util._test_decoratorsutilZ_test_decorators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
  rS   r   r"  r#  r   r   rj   r$  ZALL_REAL_NUMPY_DTYPESr(  r  r  r)  Z
skip_if_nor-  r   r   r   r    <module>   s    4	          8 34






