o
    թZh                     @   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ejj	ddgdd	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]e^_ddd Z`ej	d<g dg dg dg deddddgdd Zaejj	ddgdd	ej	dd dgdd Zbdd Zcejj	dg ddd	dd Zdej	d	g d
dd Zeej	deefdeddde&gddgeddddgdd Zhe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&gdiddejfdejejfgfgdd Zidd Zjdd  Zkej	deeld!dfeefdɃdfgd"d# Zmd$d% Znd&d' Zod(d) Zpd*d+ Zqd,d- Zrej	d.ddgej	d/g eTg g d0g d1eTg d0eTg d1ed2ddjsed2ddt gd3d4 Zuej	d/g eTg ejTg ejvdߍgd5d6 Zwej	d7eg ejxfgd8d9 Zyd:d; Zzd<d= Z{d>d? Z|d@dA Z}ej	dBe&gdCdDgeTddgfe&gdCdEgeTddgfgdFdG Z~dHdI Zej	de&gg dJeg dKgdLdM ZdNdO Zej	dPejdQdRgfejdSdTgfgdUdV ZdWdX ZdYdZ Zd[d\ Zd]d^ Zd_d` Zej	dadLejddddgfdNejejdbdbdbgfdejejejdcdcgfgddde Zej	dfdg dgfdLg dhfdNg difgdjdk ZdS (l  	TestIndexreturnc                 C   s   t tdS )NZabcde)r   listself r   U/var/www/html/lang_env/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sJ |jdksJ |r)t|jtj|dd nt	||j d|d< |d dks<J d S )NTnamecopyr9   string[pyarrow_numpy]dtypeZSOMEBIGLONGSTRINGr   )
r3   r4   r   
isinstancer9   r/   Zassert_extension_array_equalvaluespdassert_numpy_array_equal)r   r#   using_infer_stringr6   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tzr9   )rH   rI   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   r?   r   rJ   rA   Z	Timedeltar>   r3   object_r   )r   rE   r#   resultr   r   r    "test_constructor_from_index_dtlikeY   s"   
z,TestIndex.test_constructor_from_index_dtlikezindex,has_tz)rH   rI   rJ   c                 C   s4   t t|}t|| |r|j|jksJ d S d S r.   )r   r   r/   r5   rJ   )r   r#   Zhas_tzrS   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rH   )r   r   rA   to_datetimer/   r5   )r   dtsexpectedsrS   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 )	NrV   rW   rX      )   rG   dater2   r9   )r   r   r3   randomdefault_rngrQ   r>   r9   r/   r5   r   assert_series_equalrA   Z
infer_freq)r   rC   rZ   r[   dfrS   r>   rH   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[   rS   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[   rS   r   r   r    test_index_ctor_infer_nan_nat   s   	z'TestIndex.test_index_ctor_infer_nan_natz
vals,dtype   r^   rG      r_   intg?g@      @rg   ABCobjc                 C   s*   t ||d}||j|}t|| d S )Nra   )r   Z_simple_newr@   r/   r5   )r   valsr>   r#   rS   r   r   r    test_constructor_simple_new   s   	z%TestIndex.test_constructor_simple_newattrr@   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_   rI   z:Cannot use .astype to convert from timezone-naive dtype torz   zunexpected keywordr'   )rJ   r=   )r   getattrZtz_localizer>   r   r/   r5   r   r)   r*   	TypeErrorrP   r   )
r   Ztz_naive_fixturery   rl   r#   argr>   errmsgrS   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 )NrK   r_   r|   r=   )r   Z
_with_freqr>   r}   r/   r5   r   )r   ry   rl   r#   r>   r@   rS   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.   r?   len)r   r   rl   emptyr   r   r    test_constructor_empty#  s   z TestIndex.test_constructor_emptyzempty,klassrX   c                 c   r   r.   r   r   r   r   r    r   7  r   ro   )stepr^   blueredlevels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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 |j dkrJtjtdd |d W d    d S 1 sCw   Y  d S d}tjt|d |d W d    d S 1 sdw   Y  d S )NzWhen changing to a larger dtyper'   r   r2   z(Cannot change data-type for object array)r>   boolr)   r*   r+   r   NotImplementedErrorr~   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)rP   get_locr9   )r   r#   Zcastedr   r   r    test_astypem  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bobro      r;   )	r   r   is_r   r;   r3   r4   r9   r@   )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ro   r^   rG   TFr=   z$Cannot compare dtypes int64 and boolr'   r   )
r   rQ   r)   r*   r~   Zasofr   reescapestr)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/   rN   rO   )r   r#   r>   requestrC   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   appendrP   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 )	NrG   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#   rS   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[   rS   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)rI   rH   Zhourlyra   c                 S   s   | j S r.   )hourr   r   r   r    r   /  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   rS   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>6      &TestIndex.<lambda>.<locals>.<dictcomp>r   r@   r#   r   r   r    r   6      zTestIndex.<lambda>c                 C   
   t | |S r.   r   r  r   r   r    r   7     
 c                 C   sB   t g d}t tdtjd}|||j|}t|| d S )Nr   rG   r=   )r   r3   r   r   r   r@   r/   r5   )r   r  r[   r#   rS   r   r   r    test_map_dictlike_simple3  s   	z"TestIndex.test_map_dictlike_simplec                 C   r  )Nc                 S   r  r   r   r  r   r   r    r  D  r	  r
  r  r  r   r   r    r   D  r  c                 C   r  r.   r  r  r   r   r    r   E  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=   )r?   r
   r)   skip	is_uniquer3   r   r   r   r   r   r   r>   Z_constructorr   rQ   r   r/   r5   )r   r#   r  r   rngr[   rS   r   r   r    test_map_dictlikeA  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[   rS   r   r   r    )test_map_with_non_function_missing_values`  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      ?rG   r_   c                 S   s   | d S )Nr^   r   r   r   r   r    r   n  r   z1TestIndex.test_map_na_exclusion.<locals>.<lambda>ignore)Z	na_actionr^   r  r   r   r   r    test_map_na_exclusionk  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   t  r   z0TestIndex.test_map_defaultdict.<locals>.<lambda>stuffro   )r  r  r  )r   r   r   r/   r5   )r   r#   Zdefault_dictrS   r[   r   r   r    test_map_defaultdictr  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   ra   r   zDThe behavior of array concatenation with empty entries is deprecatedr'   )r   r/   rN   rO   r   r9   )r   r9   r[   r   r   r   rS   r   r   r    test_append_empty_preserve_namez  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[   rC   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/   rN   rO   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'   NaNr   ro   r^   rG   )r   r   r   rQ   r/   rN   rO   r*  r?   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}   r@   Z	to_seriesr   )r   r1  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#   nr5  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*   KeyErrorr5  )r   r#   r9  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   tolistr5  r/   r5   )r   r#   r6  r5  mixedr7  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^   rG   r   r5  r/   r5   )r   r#   r7  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'   rG   rp   )r   r)   r*   r<  r5  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   )rG   rp   r_   c                 C   s*   t g d}|j|dd}t|| d S )Nr   r  r>  rC  )r   r$   r[   r#   r7  r   r   r    (test_drop_by_numeric_label_errors_ignore  s   z2TestIndex.test_drop_by_numeric_label_errors_ignore)r   r   r   r   )r   rG  r   )rG  r   r   to_droprG  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   rQ   r5  r/   r5   r   r   __repr__r)   r*   r<  )	r   r@   rH  r#   r[   rS   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   r?   r   r)   r  r   r   uniquerepeatr5  r/   r5   r   r   r   r    "test_drop_with_duplicates_in_index3  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   rT  r   index,expected)quxr   r   r   FFTTr=   c                 C   s   | |}t|| d S r.   )isinr/   rB   )r   r@   r#   r[   rS   r   r   r    	test_isinL  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   r?   rg   mathisnanr/   rB   rX  r3   r4   r>   )r   r/  Znulls_fixture2rC   idxr   r   r    test_isin_nan_common_objectX  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)rA   NaTZNAreprr   __name__r)   r*   r~   r   r3   rh   rX  r2  r/   rB   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  rr   r-  c                 C   sZ   |  dd  dg }tg d}t||j||d d|_t||j|dd d S )NZnonexistingrW  rd  r   )r@  r3   r4   r/   rB   rX  r9   )r   rd  r#   r@   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'   rf  )Znlevelsr)   r*   r   rX  )r   r#   rd  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'   rf  )r?   r   r   namesr)   r*   r<  rX  )r   ri  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   rX  r/   rB   )r   r   r#   r[   rS   r   r   r    test_isin_empty  s   
zTestIndex.test_isin_emptypyarrowc                 C   s:   t ddgdd}|d g}tddg}t|| d S )Nr   r   r<   r=   F)r   rX  r3   r4   r/   rB   r   r   r   r    test_isin_arrow_string_null  s   z%TestIndex.test_isin_arrow_string_nullro   r^   rG   rp   TTTT)r   r   r   rV  z
2018-01-01c                 C   s2   t |}||k}tjg dtd}t|| d S )Nrp  r=   )r   r3   r4   r   r/   rB   )r   r@   r#   rS   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;   r9   Zget_level_valuesr/   r5   )r   r#   r9   rd  r[   rS   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asdfra   ro   r   r9   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_typerS   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   rz  r   r    r      s    z0TestIndex.test_str_attribute.<locals>.<listcomp>)r   r@   r}   r   r/   r5   )r   rz  r#   r[   rS   r   r  r    test_str_attribute  s   zTestIndex.test_str_attributer_   )r   r:  )r   32000Z2010Y)startendrH   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   rM  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#   rS   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/   rB   r?   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  rp   r  r   r^   r  r  )r   r   r   r#   r   r  r/   rd   )r   r#   r\   rS   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)ro   r^   rG   r   r   r   ro   rG   r^   r=   r   )r   r   r3   rR   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|   rw  ru  )r   r3   rb   rc   Zpermutationr   r/   rN   RuntimeWarningrx  rP   rQ   r   r5   )r   Z
left_indexZright_indexrS   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xxxra   )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  ra   zJWhen allow_fill=True and fill_value is not None, all indices must be >= -1r'   )ro   r   re  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  ra   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_raisesr9   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   r9   reindex)r   r9   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 Nabcr   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 )Nr  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 )Nr  r   ro   )
r   r   r  r   r3   r   ri   r   r>   r   )r   r#   rS   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   ra   ro  rf  )r   r  r/   r5   )r   r\  Zidx2r[   rS   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^   rG   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 )NrG   r  r   ro   rp   r^   r_   r   r   rj  columnsr#   )r   r   from_arraysr#   r/   rB   )r   mir[   re   rS   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 )NrG   r  r   ro   rp   r^   r_   r   r   r  r  T)r   r   r  r#   r3   r4   r/   rB   )r   re   rS   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 )NrG   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/   rB   )r   rC   r  r#   rS   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 )Nri  ra   rs  )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.rp   )
r)   ZimportorskipZIPython.core.completerr  Zrun_cellr/   rN   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 )Nro   'z$' object has no attribute 'contains'r'   )r?   r   containsr   ra  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)r_   rp   rG   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   r\  rS   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   (   rL   )r   diffr/   r5   )r   rI   expected_resultsr\  rS   r[   r   r   r    test_index_diff:     

zTestIndex.test_index_diffzdecimals, expected_resultsr^  r  rr   )g333333?gffffff@g      @)gGz?g@gGz@c                 C   r  )N)gX9v?g(\@gS@)r   roundr/   r5   )r   Zdecimalsr  r\  rS   r[   r   r   r    test_index_roundJ  r  zTestIndex.test_index_round)ra  
__module____qualname__r)   fixturer   r!   r%   r   parametrizer-   r1   r8   rD   r   r   r   rT   rU   r]   rf   rk   r3   ri   rh   r   rA   r_  rm   rx   r   r   r   iterr
   r   r   r   r   r   r   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,  r0  r4  r8  r=  rB  rD  rE  rF  rK  rN  rS  r4   r   rY  r]  rc  rg  rh  rk  rQ   rl  td
skip_if_norn  rq  rr  rt  ry  r  r   r   r  r  r  r  r   r  r  r  r  r  r  r@   r@  r  r   r  Z
datetime64r  r  r  r  r  r  r  r  r  rY   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  rG   r  )r   r/   r5   r9   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MyNamera   ZNewNameF)Zcheck_names)r   r;   r/   r5   r9   )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^   ro   Zmy_indexra   )r   r3   rh   rL  r/   r5   )r   r\  r[   rS   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.   )r3  r@   r2  r  r   r   r    r4    s   z%TestMixedIntIndex.test_logical_compatrv  r2  r3  r>   Ncategoryzvals,expectedr   r  r^  r  rr   r  rs   rt   ru   c                 C   s4   t ||d}|j|d}t ||d}t|| d S )Nr=   ru  )r   dropnar/   r5   )r   rv  r>   rw   r[   r#   rS   r   r   r    test_dropna  s   zTestMixedIntIndex.test_dropnarU  )r{   
2011-01-02
2011-01-03r{   r  r  )rK   2 days3 daysrK   r  r  )2012-022012-042012-05MrX   )r  r  r_  r  c                 C   s   |j |d}t|| d S )Nru  )r  r/   r5   )r   rv  r#   r[   rS   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  ru  )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   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 NFrO  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  rP  rQ  )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   ra   rG   r  0)r   r   r   r`  )r   Zframe_or_seriesr#   rS   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/   rB   r  r   r   r    test_index_with_tuple_bool  s   z,TestMixedIntIndex.test_index_with_tuple_bool)"ra  r  r  r)   r  r   r!   r  r  r  r  r  r4  r   r  rQ   r3   rh   r  r   rA   r_  r   r   r   r  rY   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   Nr9   ra   r   r   r   r   ro   ZL1ZL2r  c                 C   s   t ||}t|| d S r.   )r   r/   r5   )r   rj   rj  r[   rS   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   rG   Zneitherrp   Zbothr=   )rA   Intervalr   r   rQ   r/   r5   )r   Z	intervalsrS   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   r@   rS   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   rS   r[   r   r   r    test_get_combined_index;  s   z&TestIndexUtils.test_get_combined_index)ra  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}   ra  )r  r#   rz  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)r9   Zfastpathr   r   r   r_   r^   r   )	r)   r*   r~   r   r3   r4   rQ   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^   rG   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=   rp   r   rG   )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   rH   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   )rl   extra_kwargsrS   r[   r   r   r    test_construct_from_memoryview  s   r-  r1  c                 C   sL   t tjg}tdg}| ||}t|| | || }t|| d S r  )r   r3   rh   r4   r/   rB   r;   )r1  r\  r[   rS   r   r   r    test_nan_comparison_same_object  s   
r.  rm  c                  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^   rG   r=   )rm  r   rA   Z
ArrowDtypelist_r   rP  rQ  )par\  r   r   r    #test_is_monotonic_pyarrow_list_type  s   &
r1  )Dcollectionsr   r   	functoolsr   rZ  operatorr   numpyr3   r)   Zpandas.compatr   Zpandas.errorsr   Zpandas.util._test_decoratorsutilZ_test_decoratorsr  Zpandas.core.dtypes.commonr   r   r	   ZpandasrA   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  rQ   r%  r'  r(  r   r   ri   r)  ZALL_REAL_NUMPY_DTYPESr-  r  r  r.  r  r1  r   r   r   r    <module>   s    4	          8 34






