a
    Pf                     @   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ed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    W/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/indexes/test_base.pysimple_index.   s    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   sB   t jtdd  |d d d f  W d    n1 s40    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s4J |rRt|jtj|dd nt	||j d|d< |d dksxJ 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   |rBt jtdd t|t}W d    qJ1 s60    Y  nt|}t || t|tr|j	|j	kspJ |r|t
jdd7 }t|td}|jtjksJ t|t|ksJ 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   s0   t t|}t|| |r,|j|jks,J 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}|sJtnd}|d j|ks`J d|_t	|| t
|dd}t|d | |st|d }|dksJ 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   sT  t ddd}t||}||}|j}|d u}d}|dkrVt||}	t|	| n\|tu rtj	t
dd |||d W d    q1 s0    Y  n|||d}	t|	| |dkr|rtj	t
|d t|| W d    n1 s0    Y  nt||}	t|	| n|||d	}	t|	| |dkr\tt||}	t|	| nd|tu rtj	t
dd |||d W d    n1 s0    Y  n|t||d}	t|	| |dkr4|rtj	t
|d" tt|| W d    n1 s0    Y  ntt||}	t|	| n|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   sL    

,
.

.
4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   s   | ]
}|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   s   | ]
}|V  qd S 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rHd}tjt|d |d W d    q1 s<0    Y  n|j dkrtjtdd |d W d    q1 s0    Y  n<d}tjt|d |d W d    n1 s0    Y  d S )NzWhen changing to a larger dtyper'   r   r2   zYCannot change data-type for array of references|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s2J d S )Nr   r_   foobar)rP   get_locr9   )r   r#   Zcastedr    r    r!   test_astypep  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 ||rLJ |d}||sdJ t g d}t g ddd}||rJ 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   s0  t td}||sJ ||    s8J |t tdrNJ || r`J ||jddrvJ ||d d  rJ |ttdrJ || sJ | }d|_||sJ ||sJ |t |jrJ ttdd}t |dd}t |dd}||r,J 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 sR0    Y  tjttt	|d || W d    n1 s0    Y  tjttt	|d || W d    n1 s0    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s0J q0|t| }t|D ]\}}|||ks`J q`d 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rt
jtdd" || 	|sJ W d    q1 s0    Y  n|| 	|sJ 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    
2zTestIndex.test_empty_fancyc                 C   sr   t jg t jd}t|g |jd}|g  |s4J d}tjt|d ||  W d    n1 sd0    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>  r   z0TestIndex.test_map_with_tuples.<locals>.<lambda>c                 S   s   g | ]
}|fqS r    r    r   r   r    r    r!   
<listcomp>  r   z2TestIndex.test_map_with_tuples.<locals>.<listcomp>c                 S   s   | | dkfS Nro   r    r   r    r    r!   r     r   c                 S   s   g | ]}||d kfqS )ro   r    r   r    r    r!   r     r   )	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     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   s   dS r   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   2  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   s   dd t | |D S )Nc                 S   s   i | ]\}}||qS r    r    r   er   r    r    r!   
<dictcomp>9  r   &TestIndex.<lambda>.<locals>.<dictcomp>r   r@   r#   r    r    r!   r   9  r   zTestIndex.<lambda>c                 C   s
   t | |S r.   r   r  r    r    r!   r   :  r   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_simple6  s    	z"TestIndex.test_map_dictlike_simplec                 C   s   dd t | |D S )Nc                 S   s   i | ]\}}||qS r    r    r   r    r    r!   r   G  r   r  r  r  r    r    r!   r   G  r   c                 C   s
   t | |S r.   r  r  r    r    r!   r   H  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	rNt
g }nJt|jrj|j||jd}n.t|t
u r|jtkrt
||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_dictlikeD  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_valuesc  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   q  r   z1TestIndex.test_map_na_exclusion.<locals>.<lambda>ignore)Z	na_actionr^   r  r   r    r    r!   test_map_na_exclusionn  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   s   dS )Nblankr    r    r    r    r!   r   w  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_defaultdictu  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 sN0    Y  |j|ksfJ 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_name}  s    (z)TestIndex.test_append_empty_preserve_namezindex, expected))r2   F)r   Fr   Fr   F)r   T)r   T)r   T)r   Tr   F)ri   T)r   Tc                 C   s   t ||u sJ d S r.   )r   )r   r#   r[   r    r    r!   test_is_numeric  s    zTestIndex.test_is_numeric))r2   T)r   Tr  r  )r   F)r   F)r   F)r   Fr  )ri   F)r   Fc                 C   s*   |r|j dkr|rd}t||u s&J d S )Nr2   F)r>   r	   )r   r#   r[   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stt|g}tjt|d | }W d    n1 sP0    Y  t|d g}||kstJ tjt|d tg   W d    n1 s0    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 sN0    Y  t|t	rfdnt
|}t
|d t
|d t
|d |g}||ksJ |d |u sJ 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sJJ d S r.   )r}   r@   Z	to_seriesr   )r   r%  r"   r#   r   r   r    r    r!   test_logical_compat  s
    zTestIndex.test_logical_compat)r2   r   r   ri   r   c                 C   sx   t |}|ttdd }||}|ttdttd|  }t|| ||d }|dd  }t|| d S )Nr_   r   r   ro   )r   r   r   dropr/   r5   )r   r#   nr)  droppedr[   r    r    r!   test_drop_by_str_label  s    
z TestIndex.test_drop_by_str_labelkeysr   1c                 C   s<   t jtdd || W d    n1 s.0    Y  d S )N r'   )r)   r*   KeyErrorr)  )r   r#   r-  r    r    r!   *test_drop_by_str_label_raises_missing_keys  s    z4TestIndex.test_drop_by_str_label_raises_missing_keysc                 C   s   t |}|ttdd }| dg }|j|dd}|ttdttd|  }t|| |jddgdd}|tt| }t|| d S )Nr_   r   r   r  errorsr   )r   r   r   tolistr)  r/   r5   )r   r#   r*  r)  mixedr+  r[   r    r    r!   $test_drop_by_str_label_errors_ignore  s    z.TestIndex.test_drop_by_str_label_errors_ignorec                 C   s2   t g d}|d}t ddg}t|| d S )Nr   ro   r^   rG   r   r)  r/   r5   )r   r#   r+  r[   r    r    r!   test_drop_by_numeric_label_loc  s    
z(TestIndex.test_drop_by_numeric_label_locc                 C   sL   t g d}tjtdd |ddg W d    n1 s>0    Y  d S )Nr   r/  r'   rG   rp   )r   r)   r*   r0  r)  r,   r    r    r!   .test_drop_by_numeric_label_raises_missing_keys  s    z8TestIndex.test_drop_by_numeric_label_raises_missing_keyszkey,expectedrp   r   )rG   rp   r_   c                 C   s*   t g d}|j|dd}t|| d S )Nr   r  r2  r7  )r   r$   r[   r#   r+  r    r    r!   (test_drop_by_numeric_label_errors_ignore  s    z2TestIndex.test_drop_by_numeric_label_errors_ignore)r   r   r   r   )r   r;  r   )r;  r   r   to_dropr;  c           	   	   C   s   t |}t dgtd}||}t|| ||d }|d |d gfD ]}||}t|| qL||d }dt|d   d}|d |d gfD ]<}tj	t
|d || W d    q1 s0    Y  qd S )Nr   r=   r   ro   z\"\[z\] not found in axis\"r'   )r   rQ   r)  r/   r5   r   r   __repr__r)   r*   r0  )	r   r@   r<  r#   r[   rS   ZremovedZdrop_mer   r    r    r!   test_drop_tuple  s    

zTestIndex.test_drop_tuplec                 C   sp   t |dkst|tr td t|tr8ts8t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repeatr)  r/   r5   r   r    r    r!   "test_drop_with_duplicates_in_index6  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_incomparableB  s    
z(TestIndex.test_is_monotonic_incomparable)r   r   quux>   r   rG  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_isinO  s    	
zTestIndex.test_isinc                 C   s   t d|g}t|trTt|trTt|rTt|rTt||gt	ddg nt||u r|t||gt	ddg nL|r|j
dkrt||gt	ddg nt||gt	ddg d S )Nr   FTr2   )r   r?   rg   mathisnanr/   rB   rK  r3   r4   r>   )r   r#  Znulls_fixture2rC   idxr    r    r!   test_isin_nan_common_object[  s6    



z%TestIndex.test_isin_nan_common_objectc                 C   s   |}|t ju s|t ju rdtt|j }tjt|d  t	d|g|d W d    n1 s`0    Y  t	dt
jg|d}||g rJ 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   rK  r&  r/   rB   r4   )r   r#  Zfloat_numpy_dtyper>   r   rO  resr    r    r!   test_isin_nan_common_float64  s     .z&TestIndex.test_isin_nan_common_float64levelr   r  )rQ  r  rr   r!  c                 C   sZ   |  dd  dg }tg d}t||j||d d|_t||j|dd d S )NZnonexistingrJ  rW  r   )r4  r3   r4   r/   rB   rK  r9   )r   rW  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    q1 sJ0    Y  qd S )Nr   ro   zToo many levelsr'   rY  )Znlevelsr)   r*   r   rK  )r   r#   rW  r    r    r!   &test_isin_level_kwarg_bad_level_raises  s    z0TestIndex.test_isin_level_kwarg_bad_level_raiseslabelrQ  r   Zxyzzyc                 C   s   t |tr4|ddg|jdd   }d| d}n|d}d| d}tjt|d |jg |d	 W d    n1 s|0    Y  d S )
Nr   r   r^   z'Level z not found'zRequested level \(z$\) does not match index name \(foo\)r'   rY  )r?   r   r   namesr)   r*   r0  rK  )r   r\  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   rK  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   rK  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   rI  z
2018-01-01c                 C   s2   t |}||k}tjg dtd}t|| d S )Nrc  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   rW  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   rm  r    r!   r     r   z0TestIndex.test_str_attribute.<locals>.<listcomp>)r   r@   r}   r   r/   r5   )r   rm  r#   r[   rS   r    rr  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    n1 s00    Y  d S )Nzonly use .str accessorr'   r^   )r)   r*   AttributeErrorr   r@  r,   r    r    r!   test_str_attribute_raises  s    
z#TestIndex.test_str_attribute_raiseszexpand,expectedNr   r   fc                 C   s>   t g d}|d ur$|jj|d}n
|j }t|| d S )N)za b czd 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sBJ 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_indexing4  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_completion<  s    zTestIndex.test_tab_completionc                 C   sX   t g d}|dd t ddgtjds0J |ddg t ddgtjdsTJ 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_classD  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 sP0    Y  tt& |
t|
t}W d    n1 s0    Y  t|| d S )Nr^      r   r   r|   rj  rh  )r   r3   rb   rc   Zpermutationr   r/   rN   RuntimeWarningrk  rP   rQ   r   r5   )r   Z
left_indexZright_indexrS   r[   r    r    r!   test_outer_join_sortJ  s    ,4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_valueV  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 sP0    Y  tjt|d( |jtg d	dd W d    n1 s0    Y  d S )
Nr  r  ra   zJWhen allow_fill=True and fill_value is not None, all indices must be >= -1r'   )ro   r   rX  Tr  )ro   r   )r   r   r)   r*   r+   r  r3   r4   r   r    r    r!    test_take_fill_value_none_raisesg  s    6z*TestIndex.test_take_fill_value_none_raisesc                 C   sV   t tddd}tjtdd$ |tddg W d    n1 sH0    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_raisess  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_ndarrayx  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s4J 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sPJ |jd jj	tjkshJ 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   rb  rY  )r   r  r/   r5   )r   rO  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   r]  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   sj   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    n1 s\0    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 )Nr\  ra   rf  )r   Zdt_convr   r#   r    r    r!   !test_dt_conversion_preserves_name  s    	z+TestIndex.test_dt_conversion_preserves_namec                 C   sD   t g d}tjtdd d|_W d    n1 s60    Y  d S )Nr   zCan't set attributer'   F)r   r)   r*   ry  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J |d	" t|j	d
d W d    n1 sh0    Y  W d    n1 s0    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   sd   t |tr|d nJdt|j d}tjt|d |d W d    n1 sV0    Y  d S )Nro   'z$' object has no attribute 'contains'r'   )r?   r   containsr   rT  r)   r*   ry  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<0    Y  tjtdd  |jddgd W d    n1 sz0    Y  tjtdd |jdgd W d    n1 s0    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   rO  rS   r[   r    r    r!   test_sortlevel_na_position6  s    z$TestIndex.test_sortlevel_na_positionzperiods, expected_results   r   c                 C   s.   t g d}||}t |}t|| d S )N)r   r  r   (   rL   )r   diffr/   r5   )r   rI   expected_resultsrO  rS   r[   r    r    r!   test_index_diff=  s    

zTestIndex.test_index_diffzdecimals, expected_resultsrQ  r  rr   )g333333?gffffff@g      @)gGz?g@gGz@c                 C   s.   t g d}||}t |}t|| d S )N)gX9v?g(\@gS@)r   roundr/   r5   )r   Zdecimalsr  rO  rS   r[   r    r    r!   test_index_roundM  s    

zTestIndex.test_index_round)rT  
__module____qualname__r)   fixturer   r"   r%   r   parametrizer-   r1   r8   rD   r   r   r   rT   rU   r]   rf   rk   r3   ri   rh   r   rA   rR  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   r$  r(  r,  r1  r6  r8  r9  r:  r>  rA  rF  r4   r   rL  rP  rV  rZ  r[  r^  rQ   r_  td
skip_if_nora  rd  re  rg  rl  rs  r   r   rz  r~  r  r  r   r  r  r  r  r  r  r@   r4  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ed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"   b  s    zTestMixedIntIndex.simple_indexc                 C   s>   |}t jtdd |  W d    n1 s00    Y  d S Nz'>|<' not supportedr'   )r)   r*   r~   argsortr   r"   r#   r    r    r!   test_argsortf  s    zTestMixedIntIndex.test_argsortc                 C   s@   |}t jtdd t| W d    n1 s20    Y  d S r  )r)   r*   r~   r3   r  r  r    r    r!   test_numpy_argsortk  s    z$TestMixedIntIndex.test_numpy_argsortc                 C   s   |}t ||ddd}t ||dd}||us2J t|| |jdksLJ |jdksZJ td|d}td|d d	 d}|| }|jjdksJ 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_namep  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sNJ |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   r?  r/   r5   )r   rO  r[   rS   r    r    r!   test_unique_na  s    z TestMixedIntIndex.test_unique_nac                 C   s4   |}|  |j  ksJ | |j ks0J d S r.   )r'  r@   r&  r  r    r    r!   r(    s    z%TestMixedIntIndex.test_logical_compatri  r&  r'  r>   Ncategoryzvals,expectedr   r  rQ  r  rr   r  rs   rt   ru   c                 C   s4   t ||d}|j|d}t ||d}t|| d S )Nr=   rh  )r   dropnar/   r5   )r   ri  r>   rw   r[   r#   rS   r    r    r!   test_dropna  s    zTestMixedIntIndex.test_dropnarH  )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  rR  r  c                 C   s   |j |d}t|| d S )Nrh  )r  r/   r5   )r   ri  r#   r[   rS   r    r    r!   test_dropna_dt_like  s    z%TestMixedIntIndex.test_dropna_dt_likec                 C   sJ   d}t jt|d$ tg djdd W d    n1 s<0    Y  d S )Nzinvalid how option: xxxr'   r   r  rh  )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   rR  r  )r  rR  r  r   c                 C   s<   |j du sJ |jdu sJ |jdu s*J |jdu s8J d S NFrB  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   sn   t g d|d}|r@|jdu s"J |jdu s0J |jdu sjJ n*|jdu sNJ |jdu s\J |jdu sjJ d S )N)Nro   ro   r=   F)r   r  rC  rD  )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 s4J d S )Nr   r   ra   rG   r  0)r   r   r   rS  )r   Zframe_or_seriesr#   rS   r    r    r!   test_int_name_format  s    z&TestMixedIntIndex.test_int_name_formatc                 C   sT   t dd tdD }d}tjt|d t| W d    n1 sF0    Y  d S )Nc                 S   s   g | ]}t |qS r    )r   rq  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)"rT  r  r  r)   r  r   r"   r  r  r  r  r  r(  r   r  rQ   r3   rh   r  r   rA   rR  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   r]  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_uint642  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)rT  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}   rT  )r  r#   rm  r    r    r!   test_generated_op_namesD  s    
r  rl   ro   )rj   r   r   r   )rj   r>   r   c                 C   s<   t jtdd | dd W d    n1 s.0    Y  d S )Nzunexpected keyword argumentr'   r   )r   )r)   r*   r~   )rl   r    r    r!   ,test_index_subclass_constructor_wrong_kwargsc  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 sD0    Y  t jt| d, ttjg d	d
dddd W d    n1 s0    Y  t jt| d" tdddddd W d    n1 s0    Y  t jt| d" tg dddd W d    n1 s0    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_fastpathv  s    ::0r  c                  C   sN   t g d} tjtdd  | d d d f  W d    n1 s@0    Y  d S )N)r   ro   r^   rG   r&   r'   )r   r)   r*   r+   )rO  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 sH0    Y  t|dd}tjt|d t|| d W d    n1 s0    Y  tdt	d}tjt|d* t j
ddggd | d|_W d    n1 s0    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     r   r   rH   rv  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  r%  c                 C   sL   t tjg}tdg}| ||}t|| | || }t|| d S r  )r   r3   rh   r4   r/   rB   r;   )r%  rO  r[   rS   r    r    r!   test_nan_comparison_same_object  s    
r  r`  c                  C   sF   dd l } tdgddggt| |  d}|jr8J |jrBJ d S )Nr   ro   r^   rG   r=   )r`  r   rA   Z
ArrowDtypelist_r   rC  rD  )parO  r    r    r!   #test_is_monotonic_pyarrow_list_type  s    &
r"  )Dcollectionsr   r   	functoolsr   rM  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  r"  r    r    r    r!   <module>   s   4	          ; 34






