3
���h�k � @ s� d dl mZ d dlmZmZ d dlmZ d dlZd dlZd dlZd dl Z
d dl Z d dl j
jZd dlmZ d dlmZ d dlmZ d dlmZ d dlZd d lmZmZm Z m Z m Z m Z m Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z& d dl'j(Z)d d
l*m+Z+m,Z,m-Z-m.Z.m/Z/ d d l0m1Z1 G d d
� d
e1�Z2G dd� de1�Z3G dd� d�Z4e j5j6ddddddddddd d d d d d!d"d#d$d%d&g�d'd(� �Z7e j5j6d)e)j8� �d*d+� �Z9d,d-� Z:d.d/� Z;d0d1� Z<d2d3� Z=e
j>d4�e
j>d5�ej?d6d7�ej@d6�gZAe j5j6d8eA�e j5j6d9eA�d:d;� ��ZBdS )<� )� defaultdict)�datetime� timedelta)�StringION)� Timestamp)�np_datetime64_compat)�
async_mark)�ABCIndex)
�CategoricalIndex� DataFrame�
DatetimeIndex� Float64Index�
Int64Index� PeriodIndex�
RangeIndex�Series�TimedeltaIndex� UInt64Index�
date_range�isna� period_range)�Index�
MultiIndex�_get_combined_index� ensure_index�ensure_index_from_sequences)�Basec @ s e Zd ZeZed�dd�Zdd� Zejj ddgdd �d
d � �Z
ejj dd d
dgdd �dd� �Z dd� Z ejj ddgdd �dd� �Z
ejj ddgdd �dd� �Zejj dddg�ejj dejddd d d d �ejddd d �ejd!dd d �ejd"dd d �g�d#d$� ��Zejj d%ejddd d d&�dfejd!dd d �dfejd"dd d �dfg�d'd(� �Zd)d*� Zd+d,� Zejj d-ejd.�ejd/d0d1g�ed2d d3�jg�d4d5� �Zd6d7� Z ejj d8d9d:g�d;d<� �Z d=d>� Z ejj d?ej ej ej!fej"d@ej#fg�dAdB� �Z$ejj dCdDdEd dFd.gd fdGej!dHdIgdfdJdKdLej!gdMfg�dNdO� �Z%ejj dPdDdEd gejdDdEd g�ejdDdEd ge&dQ�dRdSdIgejdRdSdIge'dQ�g�dTdU� �Z(ejj dPdDdEd gdRdSdIgejdRdSdIg�ejdDdEd ge&dQ�ejdRdSdIge'dQ�g�dVdW� �Z)ejj dPdDdEd gejdDdEd ge&dQ�eje*dX�e*dY�g�e+dZdDdD�e+dZdDdE�gg�d[d\� �Z,ejj d]ddg�ejj dPeeje*dX�e*dY�g��ee+dZdDdD�e+dZdDdE�g�g�d^d_� ��Z-ejj d]ddg�ejj dPejej.dDd�ej.dDd�g�e/dD�e/dD�gg�d`da� ��Z0ejj dbdcddg�ejj deejej"g�dfdg� ��Z1ejj dbdcddg�ejj deejej2g�dhdi� ��Z3ejj djg e4g �dkdl� g D �g�ejj deee e5e6e7e"e2g�dmdn� ��Z8ejj doe9g dKdp�e9fe9e4g �dKdp�e9fe9dqdl� g D �dKdp�e9fe:dDdr�ej:fe;dDdEgdsdtggg g gdu�e;fg�dvdw� �Z<dxdy� Z=ejj dddd dzd{d|d}d~d
g dd �dd�� �Z>ejj dd�dej?d�ejj@d�d��d��d�d�gdd �d�d�� �ZAejj dd d{gdd �d�d�� �ZBd�d�� ZCejj d�ed/d0g�ed/d0d�g�d/d0d1gg�d�d�� �ZDd�d�� ZEd�d�� ZFd�d�� ZGd�d�� ZHejj ddgdd �d�d�� �ZId�d�� ZJd�d�� ZKejj ddgdd �d�d�� �ZLd�d�� ZMejj ddd dgdd �ejj d8ejNejOg�d�d�� ��ZPejj ddd dgdd �d�d�� �ZQejj ddgdd �d�d�� �ZRejj d�ed dFd.d�d�gdd��dfed dFd.d�d�gd�d��dfed dFd.d�d�g�dfg�d�d�� �ZSejj ddgdd �ejj d��d(�d)�d*g�d�d�� ��ZTejj d�edFd�d�d.d gdd��dfedFd�d�d.d gd�d��dfg�d�d�� �ZUejj d�edKdg�dKgfedKddJg�dJdKgfg�d�d�� �ZVd�d�� ZWejj@d�d��d�d�� �ZXd�d�� ZYejj ddgdd �d�d �ZZejj d�e[d��e[dăg�d�dƄ �Z\ejj@d�d��ejj d�e[d��e[dăg�d�dȄ ��Z]ejj deeje^e_g�ejj ddgdd �d�dʄ ��Z`ejj ddgdd �d�d̄ �Zaejj d�e_de_� g�ejj d�e_dЃe_� g�ejj dѐd+�d,�d-g�d�dӄ ���Zbd�dՄ Zcd�dׄ Zdd�dل Zed�dۄ Zfejj dbd�d�d�g�d�d� �Zgd�d� Zhejj d�d�d� d�d� g�d�d� �Ziejj d�d�d� d�d� g�d�d� �Zjejj d�e^d�dSd�gd�dE�d.gd�d�d�dEdS�d/d�ig�d�d� �Zkd�d� Zld�d�� Zmejj d��d0�d1g�d�d�� �Znejj ddgdd �ejj d��d2�d3g�d�d�� ��Zoejj ddgdd �d�d�� �Zpejj ddgdd ��d �d� �Zqejj ddgdd ��d�d� �Zr�d�d� Zsejj �d�d�dg��d �d
� �Ztejj@d�d��ejj �d�d�dg��d �d � ��Zu�d
�d� Zvejj �ded�dDej!g�edSdI�dg�fed�dDg�eej!dSdI�dg�fg��d�d� �Zw�d�d� Zx�d�d� Zy�d�d� Zz�d�d� Z{ejj �d�d4�d5�d6�d7�d8�d9gdgd ��d �d � �Z|ejj �d�d:�d;�d<�d=�d>�d?gdgd ��d �d � �Z}ejj �d�d@�dA�dB�dC�dD�dEgdgd ��d �d!� �Z~�d"�d#� Z�d$�d%� Z��d&�d'� Z��d(�d)� Z�ejj dPdD�dF�d+gd/d0d1gg��d,�d-� �Z��d.�d/� Z��d0�d1� Z�ejj �d2�d3�d4g��d5�d6� �Z��d7�d8� Z��d9�d:� Z�ejj �d;ddg�ejj �d<ej�dGd�d�dDdDgej�dQ��d=fej�dHd�d�dDdDgej�dQ��d>fejd�d�dDdDdEgej�dQ��d?fejd�d�dDdDdEgej�dQ��d@fg��dA�dB� ��Z��dC�dD� Z�ejj �dE�d=d�d�d.�dFgd�d.�dFgf�d?d�d�d.�dFgd�d.�dFgf�dGd�d�d.�dFgd�d.�dFgf�d=d�d�d.�dFgd�d.�dFgf�d?d�d�d.�dFgd�d.�dFgf�dGd�d�d.�dFgd�d.�dFgf�d=d��dH�dI�dJgd�dD�dKgf�d?d��dH�dI�dJgdDdE�dFgf�dGd��dH�dI�dJgd�dE�dFgf�d=dD�dH�dI�dJgd�dD�dKgf�d?dD�dH�dI�dJgdDdE�dFgf�dGdD�dH�dI�dJgd�dE�dFgf�d=�dH�dH�dI�dJgdĐdI�dJgf�d?�dH�dH�dI�dJg�dKdE�dLgf�dG�dH�dH�dI�dJgd�dE�dMgfg��dL�dM� �Z�ejj �dNe_e�e^ejg�ejj �dOe_e��dP�dP�dQg�dH�dQ�dQg�dQ�dR�dRggd�dE�dNgdĐdO�dPg�dQdE�dFgg����dS�dT� ��Z��dU�dV� Z�ejj �dW�d=�dKd�d�gf�d?�dF�dKdDgf�dG�dFd�d�gfg��dX�dY� �Z�ejj �dW�d=ej�dRd�dDdDgej�dQ�f�d?ejd�d�dD�dSgej�dQ�fg��dZ�d[� �Z��d\�d]� Z�ejj �d^e5e:e g��d_�d`� �Z��da�db� Z�ejj �dcd��d=�d?�dGg��dd�de� �Z�ejj �dcd��d=�d?�dGg��df�dg� �Z�ejj �dh�dT�dU�dVg��di�dj� �Z�ejj �dc�d=�d?�dGg��dk�dl� �Z��dm�dn� Z��do�dp� Z��dq�dr� Z��ds�dt� Z��du�dv� Z�ejj d8e&e'g��dw�dx� �Z�ejj d8e&e'g��dy�dz� �Z��d{�d|� Z�ejj d8e&e'g��d}�d~� �Z��d�d�� Z��d��d�� Z�ejj �d�ej�d�d��dW� �d�fej�d0�d��dX� �d�fej�d0d��dY� d0fej�d�d0�dZ� �d�fej�d��d��d[� �d�fej��d�d��d\� �d�fej��d�d��d]� �d�fej�d�d/�d^� �d�fej�d�d/�d_� �d�fej��d�d��d`� �d�fej��d�d��da� �d�fej��d�d��db� �d�fej�d��d��dc� �d�fej�d/d/�dd� �d�fej��d��d��de� �d�fej��d��d��df� �d�fg��d��d�� �Z�ejj ddd dgdd ��d��d�� �Z�ejj ddd dgdd �ejj �d�d�d�g�d�d�gg��d��d�� ��Z�ejj ddd dgdd ��d��d�� �Z��d��d�� Z��d��d�� Z�ejj �d�dFedDdEd g�fd dFd.gedDdEg�fg��d��d�� �Z�ejj dcd/d0�dggd/�dhd0g�did/d0gg�ejj �d��djd/gd/�dkgg��d��d�� ��Z�ejj �d��d�ej�dl�dm�dn�dog�d�e&f�dpgdQ�df�d�ej�dq�dr�ds�dtg�d�e&f�dugdQ�d�f�d�ej�dv�dw�dx�dy�dz�d{g�d�e&f�d|gdQ�d�fg��d��d�� �Z�ejj db�d��d��d��d�g��d��d�� �Z��d��d�� Z�ejj ddd dd}gdd ��d��d�� �Z�ejj dcd�d��d�gd�d��d�hg�ejj �d�e�d�d�d�d�g�ejddddg�feg �ejg e�dQ�fg��d��d�� ��Z��d��d�� Z��d��d�� Z�ejj �d�dĐd}g�ejj de�d�d�d�d�g�e dRdSdI�d+g�g��d��d�� ��Z��d��d�� Z�ejj �d�dR�dÐd�ej!g��dŐdƄ �Z�ejj d�g e^e�dQ�ejg �g��dǐdȄ �Z�ejj dcdDdEd dFgdRdSdI�d+gddddgd�d�d�d�gej�d�ddFd �g��dʐd˄ �Z�ejj ddgdd �ejj �d̐d~�dg��d͐d΄ ��Z��dϐdЄ Z�ejj dd�ddd d
dgdd ��dѐd҄ �Z�ejj �dc�dӐdԐd�g��ddׄ �Z�ejj dee�d.��e�jde;jÐd��d�g�e�dڐd�dJ�d܍g��dݐdބ �Z�ejj �d�d�ed/d0d1gd��d�g�d�gg�fded/d0d1gd��d�g�d�gg�fde;jÐd�d��d�ej!f�d�ej!ej!fg�fg��d�d� �ZŐd�d� ZƐd�d� Z�ejj �d�ee_�d��dfee�dF��dfg��d�d� �ZȐd�d� Zɐd�d� Zʐd�d� Zːd�d� Z̐d�d� Z͐d��d�� Z�ejj d�d��d�g�ejj �d�g ejg �dJdKdLgdLdKdJgejdJdKdLg�ejdLdKdJg�ej�d�d d3�jej�d�d d3�jσ g��d��d�� ��Z�ejj �d�g ejg �ejg ej�dQ�g��d��d�� �Z�ejj �d�ej5g �ej�fej g �ej fej"g �ej�fg��d��d�� �ZԐd �d� ZՐd�d� Z�ejj �de;jÐd��d�g�ejddg�fe;jÐd��d�g�ejddg�fg��d�d� �Zאd�d� Z�ejj de;jÐd��d��d�g�ed�d�d�g�g��d �d
� �Zِd �d � Z�ejj �d
ej�ej�g��d�d� �Z�ejj �d�ejd/�d�dg��dfejd/�d�dg�d� ��dfejd/�d�dg�d ��dfej�d�d�dg��dfej�d�d�dg�d� ��dfej�d�d�dg�d ��dfg��d �d � �Z�ejj �d�ej�d�d�dg��dfej�d�d�dg�d� ��d fej�d�d�dg�d ��d fg��d �d!� �Zߐd"�d#� Z�e� �d$�d%� �Z�d&�d'� Z�d�S (� � TestIndex)�returnc C s t td��S )NZabcde)r �list)�self� r! �@/tmp/pip-build-5_djhm0z/pandas/pandas/tests/indexes/test_base.py� create_index1 s zTestIndex.create_indexc C s& | j � }|d }|j|�dks"t�d S )Nr T)r# Z$_can_hold_identifiers_and_holds_name�AssertionError)r �index�keyr! r! r" �test_can_hold_identifiers4 s z#TestIndex.test_can_hold_identifiersr% r T)Zindirectc C sH t jt�� |d d d �f }W d Q R X |jdks4t�t|tj�sDt�d S )N� )�tm�assert_produces_warning�
FutureWarning�ndimr$ �
isinstance�np�ndarray)r r% � new_indexr! r! r" �
test_new_axis9 s zTestIndex.test_new_axis�intZuint�floatc C s |j td�}|jjdks t�d S )N)�dtype�i)�copyr2 r4 �kindr$ )r r% Z new_copy2r! r! r" �test_copy_and_deepcopyA s z TestIndex.test_copy_and_deepcopyc C s t j||� d S )N)r) �assert_contains_all)r r% r! r! r" �test_constructor_regularF s z"TestIndex.test_constructor_regular�stringc C s. t j|�}t|�}tj||� tj||� d S )N)r. �arrayr r) r9 �assert_index_equal)r r% �arrr0 r! r! r" �test_constructor_castingI s
z"TestIndex.test_constructor_castingc C s^ t j|�}t|ddd�}t|t�s&t�|jdks4t�tj||j� d|d< |d dksZt�d S )NT�name)r6 r@ ZSOMEBIGLONGSTRINGr ) r. r< r r- r$ r@ r) �assert_numpy_array_equal�values)r r% r> r0 r! r! r" �test_constructor_copyQ s
z TestIndex.test_constructor_copy� cast_as_objFz2015-01-01 10:00�D� z
US/EasternzGreen Eggs & Ham)�freq�periods�tzr@ )rG rH z1 daysz
2015-01-01c C s� |rt j|jt��}n
t j|�}tj||� t|t j�r�|j|jksHt �|r�|t j
dd�7 }t j|td�}|j t j
kszt �t|�t|�ks�t �d S )N�2 )� nanoseconds)r4 )�pdr �astype�objectr) r= r- r rI r$ Z Timedeltar4 r. �object_r )r rD r% �resultr! r! r" �"test_constructor_from_index_dtlikeb s
z,TestIndex.test_constructor_from_index_dtlikez index,has_tz)rG rH rI c C s4 t jt j|��}tj||� |r0|j|jks0t�d S )N)rL r r r) r= rI r$ )r r% Zhas_tzrP r! r! r" �#test_constructor_from_series_dtlike� s z-TestIndex.test_constructor_from_series_dtlikec C sD dddddg}t |dd�}ttj|��}t |dd�}tj||� d S )Nz1-1-1990z2-1-1990z3-1-1990z4-1-1990z5-1-1990�MS)rG )r r rL � to_datetimer) r= )r �dts�expected�srP r! r! r" �!test_constructor_from_series_freq� s
z+TestIndex.test_constructor_from_series_freqc C s� dddddg}t |dd�}tjtjjdd ��}||d
< t |d
dd�}|d
jtksXt�d
|_ t
j ||� tj |d
d �}t
j
|d
|� tj|d
�}|dks�t�d S ) Nz1-1-1990z2-1-1990z3-1-1990z4-1-1990z5-1-1990rS )rG � rF �date)r@ )r rL r r. �randomZrandr4 rN r$ r@ r) r= r �assert_series_equalZ
infer_freq)r rU rV �dfrP rG r! r! r" �'test_constructor_from_frame_series_freq� s z1TestIndex.test_constructor_from_frame_series_freqr< rY �a�b�cz
2000-01-01)rH c C s6 G dd� d�}t j|�}t j||��}tj||� d S )Nc @ s& e Zd Zdd� Zdejd�dd�ZdS )z:TestIndex.test_constructor_ndarray_like.<locals>.ArrayLikec S s
|| _ d S )N)r< )r r< r! r! r" �__init__� s zCTestIndex.test_constructor_ndarray_like.<locals>.ArrayLike.__init__N)r c S s | j S )N)r< )r r4 r! r! r" � __array__� s zDTestIndex.test_constructor_ndarray_like.<locals>.ArrayLike.__array__)N)�__name__�
__module__� __qualname__rb r. r/ rc r! r! r! r" � ArrayLike� s rg )rL r r) r= )r r< rg rV rP r! r! r" � test_constructor_ndarray_like� s
z'TestIndex.test_constructor_ndarray_likec C s, t jg}t|�}t|dd�}tj||� d S )Nr3 )r4 )r. �nanr
r r) r= )r �datarV rP r! r! r" � test_constructor_int_dtype_nan� s z(TestIndex.test_constructor_int_dtype_nanr4 �int64�uint64c C s6 t jg}d}tjt|d�� t||d� W d Q R X d S )Nzcannot convert)�match)r4 )r. ri �pytest�raises�
ValueErrorr )r r4 rj �msgr! r! r" �%test_constructor_int_dtype_nan_raises� s z/TestIndex.test_constructor_int_dtype_nan_raisesc C s< t jdddg�}t j|j�}t jdddg�}tj||� d S )N� r( rF )rL r r r< r) r= )r �serrP rV r! r! r" � test_constructor_no_pandas_array� s z*TestIndex.test_constructor_no_pandas_arrayzklass,dtype,na_valzdatetime64[ns]c C sP ||g}||�}|j |ks t�t|�}tj||� ttj|��}tj||� d S )N)r4 r$ r r) r= r. r< )r �klassr4 Zna_valZna_listrV rP r! r! r" � test_index_ctor_infer_nan_nat� s z'TestIndex.test_index_ctor_infer_nan_natz
vals,dtypert r( � g�������?g������@g @�A�B�C�objc C s* t ||d�}|j|j|�}tj||� d S )N)r@ )r Z _simple_newrB r) r= )r �valsr4 r% rP r! r! r" �test_constructor_simple_new� s z%TestIndex.test_constructor_simple_newr~ )r4 g �?g @c C s t |td�}t|t�st�d S )N)r4 )r r2 r- r r$ )r r~ r% r! r! r" � test_constructor_dtypes_to_int64 s z*TestIndex.test_constructor_dtypes_to_int64c C s t |td�}t|t�st�d S )N)r4 )r r3 r- r
r$ )r r~ r% r! r! r" �"test_constructor_dtypes_to_float64 s z,TestIndex.test_constructor_dtypes_to_float64z
2011-01-01z
2011-01-02i� c C s t |dd�}t|t�st�d S )N�category)r4 )r r- r
r$ )r r~ r% r! r! r" �&test_constructor_dtypes_to_categorical! s z0TestIndex.test_constructor_dtypes_to_categorical�
cast_indexc C sH |r.t |td�}t|t �s t�|jtksDt�nt |�}t|t�sDt�d S )N)r4 )r rN r- r$ r4 r )r r� r~ r% r! r! r" �#test_constructor_dtypes_to_datetime0 s z-TestIndex.test_constructor_dtypes_to_datetimec C sH |r.t |td�}t|t �s t�|jtksDt�nt |�}t|t�sDt�d S )N)r4 )r rN r- r$ r4 r )r r� r~ r% r! r! r" �$test_constructor_dtypes_to_timedeltaH s z.TestIndex.test_constructor_dtypes_to_timedelta�attrrB �asi8rw c C s t jddd�}t||�}|j|�}|j}|dkrBt j|�j|�}n |||d�}tj||� |dkrtt j|�j|�}n |||d�}tj||� |dkr�t jt |��j|�}n|t |�|d�}tj||� |dkr�t jt |��j|�}n|t |�|d�}tj||� d S )Nz
2011-01-01rY )rH r� )rI )r4 )
rL r �getattrZ tz_localizer4 r r) r= rM r )r Ztz_naive_fixturer� rw r% �argr4 rP r! r! r" � test_constructor_dtypes_datetimeY s(
z*TestIndex.test_constructor_dtypes_datetimec C s` t jddd�}|jd �}|j}t||�}|||d�}tj||� |t|�|d�}tj||� d S )Nz1 daysrY )rH )r4 )rL �timedelta_rangeZ
_with_freqr4 r� r) r= r )r r� rw r% r4 rB rP r! r! r" �!test_constructor_dtypes_timedelta~ s
z+TestIndex.test_constructor_dtypes_timedelta�valuec c s | ]
}|V qd S )Nr! )�.0�_r! r! r" � <genexpr>� s zTestIndex.<genexpr>c C s( ||�}t ||�st�t|� s$t�d S )N)r- r$ �len)r r� rw �emptyr! r! r" �test_constructor_empty� s z TestIndex.test_constructor_emptyz empty,klass)rG c c s | ]
}|V qd S )Nr! )r� r� r! r! r" r� � s )�stepZblueZred)�levels�codesc C s t ||�st�t|� s t�d S )N)r- r$ r� )r r� rw r! r! r" � test_constructor_empty_special� s z(TestIndex.test_constructor_empty_specialc C s> d}t jt|d��" ttjtj�jd gdd� W d Q R X d S )NzIThe elements provided in the data cannot all be casted to the dtype int64)rn rt rl )r4 )ro rp �
OverflowErrorr r. Ziinform �max)r rr r! r! r" � test_constructor_overflow_int64� s z)TestIndex.test_constructor_overflow_int64Zperiod�rangeZrepeatsr Ztuplesc C s |j d� d S )N�i8)�view)r r% r! r! r" �test_view_with_args� s z TestIndex.test_view_with_args�unicode� categoricalzgh-25464)�reason)Zmarks�boolr� c C s, d}t jt|d�� |jd� W d Q R X d S )Nz(Cannot change data-type for object array)rn r� )ro rp � TypeErrorr� )r r% rr r! r! r" �'test_view_with_args_object_array_raises� s z1TestIndex.test_view_with_args_object_array_raisesc C s6 |j d�}|jd� d|_|j d�}|jdks2t�d S )Nr� rY �foobar)rM �get_locr@ r$ )r r% Zcastedr! r! r" � test_astype� s
zTestIndex.test_astypec C s& t dddg�jt dddg��s"t�d S )Nr_ r` ra )r �equalsr$ )r r! r! r" �test_equals_object� s z TestIndex.test_equals_object�comp�dc C s t dddg�j|� st�d S )Nr_ r` ra )r r� r$ )r r� r! r! r" �test_not_equals_object� s z TestIndex.test_not_equals_objectc C s4 t d|ddg�}t td��jd|�}tj||� d S )Nr_ r` ra �abcrt )r r �insertr) r= )r �
nulls_fixturerV rP r! r! r" �test_insert_missing� s z TestIndex.test_insert_missingc C s@ t ddddgdd�}d}tjt|d�� |jd � W d Q R X d S )
Nr_ r` ra r� r% )r@ z/index 5 is out of bounds for axis 0 with size 4)rn rY )r ro rp �
IndexError�delete)r r% rr r! r! r" �test_delete_raises� s z TestIndex.test_delete_raisesc C s� t dddg�}t dddg�}|j|�s*t�|jd�}|j|�sBt�|j|� sRt�|jd�}|j|�sjt�t ddd g�}t d
d d gdd�}|j|� s�t�d S )
Nr_ r` ra �fooF)�
tupleize_cols)r_ r_ )r_ r` )r` r_ )r_ r_ )r_ r` )r` r_ )r � identicalr$ �renamer� )r �i1�i2�i3�i4r! r! r" �test_identical� s
zTestIndex.test_identicalc C s@ t td��}|j|�st�|j|j� j� j� j� �s8t�|jt td��� sPt�|j|j� � sdt�|j|jdd�� s|t�|j|d d � � s�t�|jtjtd��� s�t�|j|j� �s�t�|j� }d|_|j|�s�t�|j|�s�t�|jt |j �� �st�tjtdd��}t |dd�}t |dd�}|j|� �s<t�d S )N�
F)�deepZbobrt � )r6 )
r r� �is_r$ r� r6 r. r< r@ rB )r �indZind2r> Zind1r! r! r" �test_is_ s$ zTestIndex.test_is_c C sz |d }|j |�|kst�t|j |td� ��s4t�|d }|j |td� �|ksVt�|d j� }t|j |�t�svt�d S )Nr rt ���)�asofr$ r r Z
to_pydatetimer- r )r r% r� r! r! r" � test_asof% s zTestIndex.test_asofc C sB t jdddd�}td�}|jd�}||ks.t�t|t� s>t�d S )Nz
2010-01-01r( �m)rH rG z
2010-02-28z2010-02)rL r r r� r$ r- r )r r% rV rP r! r! r" �test_asof_datetime_partial1 s
z$TestIndex.test_asof_datetime_partialc sz t td�g�jjd�d � t� fdd�td�D ��}t tjjd�|d�}|j |j
d �}t dd �}||t|� ksvt �d S )
N�20130101r� r c s g | ]}� d | �qS )rJ r! )r� r5 )rW r! r" �
<listcomp>: s z:TestIndex.test_nanosecond_index_access.<locals>.<listcomp>�d )r% z"2013-01-01 00:00:00.000000050+0000�ns)
r r rB r� r r� r. r[ �randnr� r% r r$ )r �r�xZ first_valueZ expected_tsr! )rW r" � test_nanosecond_index_access8 s
z&TestIndex.test_nanosecond_index_accessc C s� t jt|�td�}d|ddd�< || }x&t|�D ]\}}|j|�|ks2t�q2W |t|� }x&t|�D ]\}}|j|�|ksft�qfW d S )N)r4 FrY � r( )r. Zonesr� r� � enumerater� r$ r )r r% Z
bool_indexZ sub_indexr5 �valr! r! r" �test_booleanindexE s zTestIndex.test_booleanindexc C s> | j � }|dddg }x"|D ]}|||j|� ks t�q W d S )Nrt r( rF )r# r� r$ )r r% �slr5 r! r! r" �
test_fancyS s
zTestIndex.test_fancyc C sB t jg |d�}t|�g �}|g j|�s,t�|| j|�s>t�d S )N)r4 )r. r< �typer� r$ )r r% r4 Z empty_arr� empty_indexr! r! r" �test_empty_fancyY s zTestIndex.test_empty_fancyc C sX t jg t jd�}t|�g �}|g j|�s.t�d}tjt|d�� || W d Q R X d S )N)r4 z=arrays used as indices must be of integer \(or boolean\) type)rn ) r. r< Zfloat_r� r� r$ ro rp r� )r r% Z
empty_farrr� rr r! r! r" �test_empty_fancy_raisesb s z!TestIndex.test_empty_fancy_raisesc C sl |d d� }|d d� }|j ||d�}|d kr>tj||j� � tj||�sNt�|j ||d�}||ksht�d S )N� r� )�sort)� intersectionr) r= � sort_values�
equalContentsr$ )r r% r� �first�second� intersect�interr! r! r" �test_intersectiono s zTestIndex.test_intersectionzindex2,keeps_name� � )r@ �otherc C sZ t dddddgdd�}t dddg�}|j||�}|r:d|_|j|jksJt�tj||� d S )Nrt r( rF ry rY r% )r@ )r r� r@ r$ r) r= )r �index2�
keeps_namer� �index1rV rP r! r! r" �#test_intersection_name_preservation| s z-TestIndex.test_intersection_name_preservationz$first_name,second_name,expected_nameNc C sD |dd� }|d d� }||_ ||_ |j||d�}|j |ks@t�d S )NrY r� r� )r� )r@ r� r$ ) r r% �
first_name� second_name�
expected_namer� r� r� r� r! r! r" �$test_intersection_name_preservation2� s z.TestIndex.test_intersection_name_preservation2c C s\ t dddddgdd�}t dddg�}|r.d|_|j||d�}|d krL|j� }tj||� d S ) NrY rF r( ry rt r% )r@ )r� )r r@ r� r� r) r= )r r� r� r� r� rV rP r! r! r" �test_intersection_monotonic� s z%TestIndex.test_intersection_monotoniczindex2,expected_arrc C sJ t ddddg�}t |dd�}|j||d�}|d kr:|j� }tj||� d S )Nrz r{ r| rN )r4 )r� )r r� r� r) r= )r r� Z expected_arrr� r� rV rP r! r! r" �*test_intersection_non_monotonic_non_unique� s z4TestIndex.test_intersection_non_monotonic_non_uniquec C sT t ddd�t ddd�g}t|td�}tdgtd�}|j||d�}t|�dksPt�d S ) Ni� r( � � )r4 �aa)r� r )r r rN r� r� r$ )r r� �dt_datesr� r� rP r! r! r" �test_intersect_str_dates� s
z"TestIndex.test_intersect_str_dateszNot implementedc C s: t jdddg�}t jdddg�}tj|j|dd�|� d S )Nra r_ r` T)r� )rL r r) r= r� )r �idxZsorted_r! r! r" �!test_intersection_equal_sort_true� s z+TestIndex.test_intersection_equal_sort_truec
C s� t ddgdd�}t ddgdd�}t dd gd
d�}|j|j||d �|d �}|j||d �j||d �}tj||� t ddgd d�}t g d
d�}t g dd�} |j|j| |d �|d �}|j||d �j| |d �}tj||� d S )Nrt r( r� )r@ rY r� r� rF ry r� )r� �j1�j2�j3)r �unionr) r= )
r r� r� r� r� r� rV r� r� r� r! r! r" �test_chained_union� s z TestIndex.test_chained_unionc C s^ |dd� }|d d� }|d d� }|j ||d�}|d krJtj||j� � tj||�sZt�d S )NrY r� r� )r� )r� r) r= r� r� r$ )r r% r� r� r� �
everythingr� r! r! r" �
test_union� s zTestIndex.test_union�slice_r c C sV t jdddg�}|| }tj|j|�|� tj|j|�|� tj|j|dd�|� d S )Nrt r r( F)r� )rL r r) r= r� )r r r� r� r! r! r" � test_union_sort_other_special� s
z'TestIndex.test_union_sort_other_specialc C sF t jdddg�}|| }|j|dd�}t jdddg�}tj||� d S )Nrt r r( T)r� )rL r r� r) r= )r r r� r� rP rV r! r! r" � test_union_sort_special_true� s
z&TestIndex.test_union_sort_special_truec C sh |dd� }|d d� }|d d� }||j �}|j||d�}|d krTtj||j� � tj||�sdt�d S )NrY r� r� )r� )rB r� r) r= r� r� r$ ) r r% rw r� r� r� r� ZcaserP r! r! r" �test_union_from_iterables s
z#TestIndex.test_union_from_iterablesc C st |dd� }|j ||d�}||k| ks,t�|j g |d�}||k| ksLt�tg �j ||d�}||k| kspt�d S )NrY r� )r� )r� r$ r )r r% r� r� r� r! r! r" �test_union_identity s z TestIndex.test_union_identity�
first_listZba� second_list�abz&first_name, second_name, expected_namec C s� t ||d�}t ||d�}|j||d�} t|�j|�}
|d krrt|�dkrrt|�dkrrt t|
�|d�} tj| | � n t |
|d�} tj| | �s�t�d S )N)r@ )r� r ) r r� �setr� �sortedr) r= r� r$ ) r r r r� r� r� r� r� r� r� r~ rV r! r! r" � test_union_name_preservation! s z&TestIndex.test_union_name_preservationc C s� | j � }tjddd�}|j|�}|j|�}|jtjkrDtj||�}ntj||jd��}t j
||�sft �t j
||�svt �t j ||� t j ||� t j ||� d S )Nz
2019-01-01r� )rH �O)
r# rL r r� r4 r. rO �appendrM r) r� r$ r9 )r r� r% �
date_indexZ first_catZ
second_catZappendedr! r! r" �test_union_dt_as_obj7 s
z TestIndex.test_union_dt_as_objc C s t j||jdd� �� d S )Nc S s | S )Nr! )r� r! r! r" �<lambda>K s z5TestIndex.test_map_identity_mapping.<locals>.<lambda>)r) r= �map)r r% r! r! r" �test_map_identity_mappingI s z#TestIndex.test_map_identity_mappingc C sn t jd�}t jd�jdd� �}tdd� |D ��}t j||� |jdd� �}tjdd� |D ��}t j||� d S )NrF c S s | fS )Nr! )r� r! r! r" r S s z0TestIndex.test_map_with_tuples.<locals>.<lambda>c S s g | ]
}|f�qS r! r! )r� r5 r! r! r" r� T s z2TestIndex.test_map_with_tuples.<locals>.<listcomp>c S s | | dkfS )Nrt r! )r� r! r! r" r Y s c S s g | ]}||d kf�qS )rt r! )r� r5 r! r! r" r� Z s )r) � makeIntIndexr r r= r � from_tuples)r r% rP rV r! r! r" �test_map_with_tuplesM s
z TestIndex.test_map_with_tuplesc C sB dddg}t jt|dddg��}|jdd� �}tj|t|�� d S ) Nr� �bar�bazrt r( rF c S s | d S )Nr r! )r� r! r! r" r b s z3TestIndex.test_map_with_tuples_mi.<locals>.<lambda>)r r �zipr r) r= r )r Z first_levelZ multi_indexZ
reduced_indexr! r! r" �test_map_with_tuples_mi] s
z!TestIndex.test_map_with_tuples_mi�
makeDateIndexZmakePeriodIndexZmakeTimedeltaIndexc C s: t t|�d�}tdgd �}|jdd� �}tj||� d S )Nr� rt c S s dS )Nrt r! )r� r! r! r" r k s zATestIndex.test_map_tseries_indices_return_index.<locals>.<lambda>)r� r) r r r= )r r� r% rV rP r! r! r" �%test_map_tseries_indices_return_indexe s z/TestIndex.test_map_tseries_indices_return_indexc C s: t jdddd�}ttd�dd�}t j||jdd� �� d S )N� �hZhourly)rG r@ )r@ c S s | j S )N)�hour)r� r! r! r" r q s zGTestIndex.test_map_tseries_indices_accsr_return_index.<locals>.<lambda>)r) r r r� r= r )r r
rV r! r! r" �+test_map_tseries_indices_accsr_return_indexn s z5TestIndex.test_map_tseries_indices_accsr_return_index�mapperc C s dd� t | |�D �S )Nc S s i | ]\}}||�qS r! r! )r� �er5 r! r! r" �
<dictcomp>v s z&TestIndex.<lambda>.<locals>.<dictcomp>)r )rB r% r! r! r" r v s zTestIndex.<lambda>c C s t j| |�S )N)rL r )rB r% r! r! r" r w s c C s: t dddg�}tjd�}|j||j|��}tj||� d S )Nr� r r rF )r r) r r rB r= )r r rV r% rP r! r! r" �test_map_dictlike_simples s
z"TestIndex.test_map_dictlike_simplec C s dd� t | |�D �S )Nc S s i | ]\}}||�qS r! r! )r� r r5 r! r! r" r! � s z&TestIndex.<lambda>.<locals>.<dictcomp>)r )rB r% r! r! r" r � s c C s t j| |�S )N)rL r )rB r% r! r! r" r � s c C s^ t |t�rd S |jsd S |jr(tg �}nttjt|�dd��}|j|||��}t j
||� d S )Nr rt r� ) r- r
� is_uniquer� r r. �aranger� r r) r= )r r% r rV rP r! r! r" �test_map_dictlike� s
zTestIndex.test_map_dictliker� r )r% c C s4 t dtjdg�}t dddg�j|�}tj||� d S )Ng @r� r( rt r )r r. ri r r) r= )r r rV rP r! r! r" �)test_map_with_non_function_missing_values� s z3TestIndex.test_map_with_non_function_missing_valuesc C s@ t dtjdtjdg�}|jdd� dd�}|d }tj||� d S ) Ng �?rF rY c S s | d S )Nr( r! )r� r! r! r" r � s z1TestIndex.test_map_na_exclusion.<locals>.<lambda>�ignore)Z na_actionr( )r r. ri r r) r= )r r% rP rV r! r! r" �test_map_na_exclusion� s z TestIndex.test_map_na_exclusionc C sJ t dddg�}tdd� �}d|d< |j|�}t dddg�}tj||� d S )Nrt r( rF c S s dS )N�blankr! r! r! r! r" r � s z0TestIndex.test_map_defaultdict.<locals>.<lambda>�stuffr) )r r r r) r= )r r% Z default_dictrP rV r! r! r" �test_map_defaultdict� s
z TestIndex.test_map_defaultdictz
name,expectedr c C s: t g dd�}t dddg|d�}|j|�}|j|ks6t�d S )Nr� )r@ rt r( rF )r r r@ r$ )r r@ rV �left�rightrP r! r! r" � test_append_empty_preserve_name� s
z)TestIndex.test_append_empty_preserve_namezsecond_name,expectedr@ c C sx |dd� }|d d� }|dd� }d|_ ||_ |j||d�}tj||�sNt�|d krf|j d kstt�n|j |kstt�d S )NrY r� r� r@ )r� )r@ �
differencer) r� r$ ) r r% r� rV r� r� r� ZanswerrP r! r! r" �!test_difference_name_preservation� s z+TestIndex.test_difference_name_preservationc C sB |dd� }d|_ |jg |�}tj||�s.t�|j |j ks>t�d S )NrY r� r@ )r@ r/ r) r� r$ )r r% r� r� rP r! r! r" �test_difference_empty_arg� s
z#TestIndex.test_difference_empty_argc C sB |dd� }d|_ |j||�}t|�dks.t�|j |j ks>t�d S )NrY r� r@ r )r@ r/ r� r$ )r r% r� r� rP r! r! r" �test_difference_identity� s
z"TestIndex.test_difference_identityc C sP |dd� }|d d� }|j ||�}|dd� }|d kr@|j� }tj||� d S )NrY r� r� )r/ r� r) r= )r r% r� r� r� rP rV r! r! r" �test_difference_sort� s z TestIndex.test_difference_sortc C s� t ddddgdd�}t ddddg�}|j||d�}t ddg�}tj||�sNt�|jd ks\t�|d krl|j� }tj||� ||A }tj||�s�t�|jd ks�t�d S ) NrY r( rF ry r� )r@ rt )r� )r �symmetric_differencer) r� r$ r@ r� r= )r r� r� r� rP rV r! r! r" �test_symmetric_difference� s z#TestIndex.test_symmetric_difference�opnamer/ r4 c C s� t jdt jd�dg�}t jdt jd�dg�}tj||�}||�}t jdt jd�dt jd�g�}|dkrr|d d� }tj||� tj||dd�}||�}tj||� d S ) NrF �2000rt r( �1999r/ F)r� )rL r r �operator� methodcallerr) r= )r r6 r_ r` �oprP rV r! r! r" � test_difference_incomparable� s z&TestIndex.test_difference_incomparablec C sb t jdt jd�dg�}t jdt jd�dg�}tj||dd�}tjtdd �� ||� W d Q R X d S )
NrF r7 rt r( r8 T)r� zCannot compare)rn )rL r r r9 r: ro rp r� )r r6 r_ r` r; r! r! r" �!test_difference_incomparable_true s
z+TestIndex.test_difference_incomparable_truec C sx t jtdddgdddg��}t jdd g�}|j||d�}t jd
d d g�}|d krX|j� }tj||� tj||�stt�d S )
Nr� r r rt r( rF )r� )r� rt )r rF )r r( )r rF )r rF ) r r r r4 r� r) r= r� r$ )r r� r� r� rP rV r! r! r" � test_symmetric_difference_mi s z&TestIndex.test_symmetric_difference_mizindex2,expectedg c C s@ t dtjddg�}|j||d�}|d kr0|j� }tj||� d S )Nrt r( rF )r� )r r. ri r4 r� r) r= )r r� rV r� r� rP r! r! r" �!test_symmetric_difference_missing& s
z+TestIndex.test_symmetric_difference_missingc C s� t ddddgdd�}tjddddg�}t ddg�}|j||d�}tj||�sPt�|jdks^t�|j|d |d
�}tj||�s~t�|jd ks�t�d S ) Nrt r( rF ry r� )r@ rY )r� �new_name)Z result_namer� )r r. r< r4 r) r� r$ r@ )r r� r� r� rV rP r! r! r" �#test_symmetric_difference_non_index7 s z-TestIndex.test_symmetric_difference_non_indexc C s2 |j s
d S |j||d�}|j|�}tj||� d S )N)r� )r# r/ �dropr) r= )r r% r� rP rV r! r! r" �test_difference_typeC s
z TestIndex.test_difference_typec C s8 |j s
d S |j|j|��}|j||d�}tj||� d S )N)r� )r# r� rB r/ r) r= )r r% r� r� Zdiffr! r! r" � test_intersection_differenceM s
z&TestIndex.test_intersection_differencec C s* | j � }tjt�� |j� W d Q R X d S )N)r# r) r* r+ Zis_mixed)r r% r! r! r" �test_is_mixed_deprecatedX s z"TestIndex.test_is_mixed_deprecatedzindex, expectedc C s |j � |kst�d S )N)Z
is_numericr$ )r r% rV r! r! r" �test_is_numeric^ s
zTestIndex.test_is_numericc C s |j � |kst�d S )N)Z is_objectr$ )r r% rV r! r! r" �test_is_objectm s
zTestIndex.test_is_objectc C s |j |kst�d S )N)Z is_all_datesr$ )r r% rV r! r! r" �test_is_all_dates| s
zTestIndex.test_is_all_datesc C s | j tj|� d S )N)�_check_method_worksr �_summary)r r% r! r! r" � test_summary� s zTestIndex.test_summaryc C s4 t ddgdd�}|j� }d|ks$t�d|ks0t�d S )Nz {other}%sz ~:{range}:0rz )r@ )r rJ r$ )r r� rP r! r! r" �test_summary_bug� s zTestIndex.test_summary_bugc C s( t ddg�}ddg}|j� |ks$t�d S )NZ aaaaaaaaar` )r �formatr$ )r r� rV r! r! r" �$test_format_different_scalar_lengths� s z.TestIndex.test_format_different_scalar_lengthsc C sR t j� }t|�jd�sBt|g�}|j� }t|d �g}||ksBt�tg �j� d S )NZ000r )r �now�str�endswithr rM r$ )r rO r% � formattedrV r! r! r" �test_format_bug� s
zTestIndex.test_format_bugy @g @c C sf t |�}|j|� t|�}|j� }t|d �t|d �t|d �dg}||ksRt�|d |ksbt�d S )Nr rt r( �NaNrF )r r r rM rP r$ )r r~ r� r% rR rV r! r! r" �test_format_missing� s
$ z TestIndex.test_format_missingc C s. t dddd�}|jdd�}|d dks*t�d S )Nz2011-01-01 04:00:00r� Z something)rH r@ T)r@ r )r rM r$ )r �datesrR r! r! r" � test_format_with_name_time_info� s z)TestIndex.test_format_with_name_time_infoc C sN t tddd�tdddd�g�}|j� }ddg}t|�dks>t�||ksJt�d S )Ni� r( r� � z2012-02-07 00:00:00z2012-02-07 23:00:00)r r rM r� r$ )r �trP rV r! r! r" � test_format_datetime_with_time� s
z(TestIndex.test_format_datetime_with_timer; �any�allc C s* | j � }t||�� t|j|�� ks&t�d S )N)r# r� rB r$ )r r; r% r! r! r" �test_logical_compat� s z TestIndex.test_logical_compatc C s ||� d S )Nr! )r �methodr% r! r! r" rI � s z TestIndex._check_method_worksc C sP t dddddg�}t dddg�}|j|�}tjdddgtjd�}tj||� d S ) Nrt r( rF ry rY r� )r4 r� )r � get_indexerr. r< �intpr) �assert_almost_equal)r r� r� Zr1Ze1r! r! r" �test_get_indexer� s
zTestIndex.test_get_indexer�reversezexpected,method�padZffill�backfillZbfillc C s^ t dddddg�}t dddg�}|r@|d d d� }|d d d � }|j||d�}tj||� d S )
Nrt r( rF ry rY r� )r^ r� r� )r r_ r) ra )r rc rV r^ r� r� rP r! r! r" �test_get_indexer_methods� s z"TestIndex.test_get_indexer_methodsc C sj t tjd��}tjtdd�� |jddgdd� W d Q R X tjtdd�� |jddgdd� W d Q R X d S ) Nr� ztolerance argument)rn rt r )� tolerancezlimit argument)�limit)r r. r$ ro rp rq r_ )r r% r! r! r" �test_get_indexer_invalid� s
z"TestIndex.test_get_indexer_invalidz$method, tolerance, indexer, expectedr� �nearestg�������?g������?g !@� c C s: t tjd��}|j|||d�}tj|tj|tjd�� d S )Nr� )r^ rg )r4 )r r. r$ r_ r) rA r< r` )r r^ rg ZindexerrV r% �actualr! r! r" �test_get_indexer_nearest� s z"TestIndex.test_get_indexer_nearest�listtypeztolerance, expectedg333333�?g�������?g �?c C sD t tjd��}|jdddgd||�d�}tj|tj|tjd�� d S )Nr� g�������?g������?g !@rj )r^ rg )r4 )r r. r$ r_ r) rA r< r` )r rg rV rn r% rl r! r! r" �+test_get_indexer_nearest_listlike_tolerance s
z5TestIndex.test_get_indexer_nearest_listlike_tolerancec C st t tjd��}tjtdd�� |jddgddd� W d Q R X tjtdd�� |jddgddd d
gd � W d Q R X d S ) Nr� zlimit argument)rn rt r rj )r^ rh ztolerance size must matchr( rF )r^ rg )r r. r$ ro rp rq r_ )r r% r! r! r" � test_get_indexer_nearest_error# s
z(TestIndex.test_get_indexer_nearest_errorzmethod,expectedc C sz t tjd��d d d � }|jdddg|d�}tj|tjdddgtjd�� |jd d
d g|d�}tj|tj|tjd�� d S )
Nr� rt r rY r� )r^ ry )r4 g�������?g������?g !@r� )r r. r$ r_ r) rA r< r` )r r^ rV r% rl r! r! r" �#test_get_indexer_nearest_decreasing+ s
z-TestIndex.test_get_indexer_nearest_decreasingc C s4 t jddg�}|jddddg|d�}tj||� d S )Nr` ra r_ r� )r^ )rL r r_ r) rA )r r^ rV r% rl r! r! r" �test_get_indexer_strings8 s z"TestIndex.test_get_indexer_stringsc C s� t jddg�}d}tjt|d�� |jddddgdd� W d Q R X tjt|d�� |jddddgd d
d � W d Q R X tjt|d��& |jddddgd d
d
d
d
gd � W d Q R X d S ) Nr` ra z4unsupported operand type\(s\) for -: 'str' and 'str')rn r_ r� rj )r^ rd r( )r^ rg )rL r ro rp r� r_ )r r% rr r! r! r" � test_get_indexer_strings_raisesE s "z)TestIndex.test_get_indexer_strings_raises� idx_classc C sB |t d��}|jdddg�}tjdddgtjd�}tj||� d S ) Nry TFrt )r4 r� r� r� )r r_ r. r< r` r) rA )r rt Z
numeric_indexrP rV r! r! r" �-test_get_indexer_numeric_index_boolean_targetT s z7TestIndex.test_get_indexer_numeric_index_boolean_targetc C sb ||kr d S t j||gtd�}tj|td�}|j||dg�}t jdddgt jd�}tj||� d S )N)r4 �Unknownr rt r� ) r. r< rN rL r r_ r` r) rA )r Zunique_nulls_fixtureZunique_nulls_fixture2r> r% rP rV r! r! r" � test_get_indexer_with_NA_values] s z)TestIndex.test_get_indexer_with_NA_valuesr^ c C sF t jdddg�}|jd|d�dks&t�|rB|jd|dd�dksBt�d S )Nr rt r( )r^ )r^ rg )rL r r� r$ )r r^ r% r! r! r" � test_get_locm s zTestIndex.test_get_locc C sN t jdddg�}|rd}nd}tjt|d�� |jddg|d� W d Q R X d S )Nr rt r( znot supported betweenz invalid key)rn )r^ )rL r ro rp r� r� )r r^ r% rr r! r! r" � test_get_loc_raises_bad_labelu s z'TestIndex.test_get_loc_raises_bad_labelz
method,locc C s@ t jdddg�}|jd|�|ks$t�|jd|dd�|ks<t�d S )Nr rt r( g�������?)rg )rL r r� r$ )r r^ �locr% r! r! r" �test_get_loc_tolerance� s z TestIndex.test_get_loc_tolerancec C s> t jdddg�}tjtdd�� |jd|dd� W d Q R X d S ) Nr rt r( z1.1)rn g�������?g�������?)rg )rL r ro rp �KeyErrorr� )r r^ r% r! r! r" �%test_get_loc_outside_tolerance_raises� s z/TestIndex.test_get_loc_outside_tolerance_raisesc C s> t jdddg�}tjtdd�� |jdddd � W d Q R X d S )
Nr rt r( zmust be numeric)rn g�������?rj �invalid)rg )rL r ro rp rq r� )r r% r! r! r" �!test_get_loc_bad_tolerance_raises� s z+TestIndex.test_get_loc_bad_tolerance_raisesc C s< t jdddg�}tjtdd�� |jddd� W d Q R X d S )Nr rt r( ztolerance .* valid if)rn g�������?)rg )rL r ro rp rq r� )r r% r! r! r" �'test_get_loc_tolerance_no_method_raises� s z1TestIndex.test_get_loc_tolerance_no_method_raisesc C sB t jdddg�}tjtdd�� |jddddgd� W d Q R X d S ) Nr rt r( ztolerance size must match)rn g�������?rj )rg )rL r ro rp rq r� )r r% r! r! r" �&test_get_loc_raises_missized_tolerance� s z0TestIndex.test_get_loc_raises_missized_tolerancec C s: t jddg�}tjtdd�� |jddd� W d Q R X d S )Nr_ ra zunsupported operand type)rn rj )r^ )rL r ro rp r� r� )r r% r! r! r" �"test_get_loc_raises_object_nearest� s z,TestIndex.test_get_loc_raises_object_nearestc C s< t jddg�}tjtdd�� |jdddd� W d Q R X d S )Nr_ ra zunsupported operand type)rn rd r~ )r^ rg )rL r ro rp r� r� )r r% r! r! r" �$test_get_loc_raises_object_tolerance� s z.TestIndex.test_get_loc_raises_object_tolerancec
C s� t tjddddddddg|d ��}t|�}|jdd
�d|fksBt�|jd d
�d |fksZt�|jd d �dksnt�|jdd�d |fks�t�|jd d
�dks�t�|jdd
�dks�t�|d d d� }|jd d�dks�t�|jdd �dks�t�d S )Nr rt r( rY r� r� r� r� )r4 )�startrF rk )�end)rF r� )r r� )r r� r� )r( r� )r( rY )r r. r< r� �
slice_locsr$ )r r4 r% �nr� r! r! r" �test_slice_locs� s "zTestIndex.test_slice_locsc
C s� t tjddddddddg|d ��}t|�}|jd
d �d |fksBt�|jd
d�dksVt�|d d d� }|jdd�dksxt�|jdd�d|fks�t�d S )Nr rt r( rY r� r� r� r� )r4 g @g $@rF g @g %@rk g !@g �?)rF rk r� )r( r� r� )r r. r< r� r� r$ )r r4 r% r� r� r! r! r" �test_slice_float_locs� s "z TestIndex.test_slice_float_locsc C s� t ddddddg�}|jdd�d ks(t�|jdd�d ks<t�|jdd�d
ksPt�|jdd�dksdt�|d d d� }|jdd�dks�t�|jdd�dks�t�|jdd�dks�t�|jdd�dks�t�d S )Nr_ r` ra r� r r� )r� ry r( rt )r r� )r r� )r ry )r( r� r� )r r� )r r� )r ry )r( r� )r r� r$ )r r% r� r! r! r" �test_slice_locs_dup� s z TestIndex.test_slice_locs_dupc C s| t tjddddg|d��}|jdd�d ks.t�|jdd�d
ksBt�|d d d � }|jdd�d ksdt�|jdd�d
ksxt�d S )Nr� � � )r4 rt rF r� �
)rt rF )rt rF r� )rt rF )rt rF )r r. r< r� r$ )r r4 r% r� r! r! r" �test_slice_locs_dup_numeric� s z%TestIndex.test_slice_locs_dup_numericc C sd t tjddg�}|jd�dks"t�|jtj�dks6t�t dtjtjddg�}|jtj�dks`t�d S ) Nrt r( rF r rY )rt rF )r rF )rt rY )r r. ri r� r$ )r r% r! r! r" �test_slice_locs_na� s
z TestIndex.test_slice_locs_nac C s` t tjddg�}tjtdd�� |jdd� W d Q R X tjtdd�� |jdd� W d Q R X d S )Nrt r( � )rn g �?)r� )r� )r r. ri ro rp r| r� )r r% r! r! r" �test_slice_locs_na_raises� s
z#TestIndex.test_slice_locs_na_raiseszin_slice,expectedZyxdcb�yr� ZybZydb�zZycr� ZdcbZyxc C sR t td��}|j|j|j|j�\}}||||j� }tj t|��}tj||� d S )NZbcdxy) r r r� r� �stopr� rL r) r= )r Zin_slicerV r% Zs_startZs_stoprP r! r! r" � test_slice_locs_negative_step� s
z'TestIndex.test_slice_locs_negative_stepc C sx t |�}|ttdd�� }|j|�}|ttd��ttd|�� }tj||� |j|d �}|dd � }tj||� d S )NrY r� r rt )r� r r� rB r) r= )r r% r� rB �droppedrV r! r! r" �test_drop_by_str_label s
z TestIndex.test_drop_by_str_label�keys�1c C s( t jtdd�� |j|� W d Q R X d S )Nr� )rn )ro rp r| rB )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 |�}|ttdd�� }|j� dg }|j|dd�}|ttd��ttd|�� }tj||� |jddgdd�}|tt|�� }tj||� d S )NrY r� r� r' )�errorsr )r� r r� �tolistrB r) r= )r r% r� rB �mixedr� rV r! r! r" �$test_drop_by_str_label_errors_ignore s z.TestIndex.test_drop_by_str_label_errors_ignorec C s4 t dddg�}|jd�}t ddg�}tj||� d S )Nrt r( rF )r rB r) r= )r r% r� rV r! r! r" � test_drop_by_numeric_label_loc) s
z(TestIndex.test_drop_by_numeric_label_locc
C s: t dddg�}tjtdd�� |jddg� W d Q R X d S )Nrt r( rF r� )rn ry )r ro rp r| rB )r r% r! r! r" �.test_drop_by_numeric_label_raises_missing_keys1 s z8TestIndex.test_drop_by_numeric_label_raises_missing_keysz key,expectedc C s, t dddg�}|j|dd�}tj||� d S )Nrt r( rF r' )r� )r rB r) r= )r r& rV r% r� r! r! r" �(test_drop_by_numeric_label_errors_ignore6 s z2TestIndex.test_drop_by_numeric_label_errors_ignore�to_dropc
C s� t j|�}t jdg�}|j|�}tj||� |j|d �}x0|d |d gfD ]}|j|�}tj||� qNW |j|d �}dtj|d j� �� d�}x>|d |d gfD ](}tj t
|d�� |j|� W d Q R X q�W d S )Nr` r rt z\"\[z\] not found in axis\")rn ) rL r rB r) r= �re�escape�__repr__ro rp r| ) r rB r� r% rV rP �removedZdrop_merr r! r! r" �test_drop_tuple? s
zTestIndex.test_drop_tuplezmethod,expected,sortr� �num�let�a1r� c C s� t tjd d d
dgdtfdgd��}t tjddddddgdtfdgd��}t||�||d
�}|jdksht�t |�}tj||� d S )Nrt rz r( r{ r� r� r� )r4 r| )r� )rt rz )r( rz )rt r{ )r( r{ )r� r� )rt rz )r( rz )rt r{ )r( r{ )rt r| )r( r| )r� r� ) r r. r< r2 r� r, r$ r) r= )r r^ rV r� r� r� rP r! r! r" �test_tuple_union_bugW s
z TestIndex.test_tuple_union_bug�is_monotonic_increasing�is_monotonic_decreasing�!_is_strictly_monotonic_increasing�!_is_strictly_monotonic_decreasingc C s& t dtj� dg�}t||� s"t�d S )NrY r� )r r rO r� r$ )r r� r% r! r! r" � test_is_monotonic_incomparable� s
z(TestIndex.test_is_monotonic_incomparablec C sT | j � }tjdddg�}tjt�� |j||d d� W d Q R X |d dksPt�d S )Nrt r( rF �P )r# r. r< r) r* r+ Z set_valuer$ )r r� r> r! r! r" �test_set_value_deprecated� s
z#TestIndex.test_set_value_deprecatedc C s� t jjd�}|d }tjtdd��( tjt�� |j ||� W d Q R X W d Q R X tjt�� |j t
||d�|�}W d Q R X tj ||d � d S )Nr� �C zhas no attribute '_values')rn )r% ) r. r[ r� ro rp �AttributeErrorr) r* r+ � get_valuer ra )r r% rB r� rP r! r! r" �test_get_value� s zTestIndex.test_get_valueZquuxzindex,expectedZquxc C s |j |�}tj||� d S )N)�isinr) rA )r rB r% rV rP r! r! r" � test_isin� s
zTestIndex.test_isinc C s� t |t�rPt |t�rPtj|�rPtj|�rPtjtd|g�j|g�tj ddg�� nV||kr�tjtd|g�j|g�tj ddg�� n&tjtd|g�j|g�tj ddg�� d S )Nr_ FT)
r- r3 �math�isnanr) rA r r� r. r< )r r� Znulls_fixture2r! r! r" �test_isin_nan_common_object� s
z%TestIndex.test_isin_nan_common_objectc C s| |t jkrtjd� |t jkr(tjd� tjtd|g�j t
j g�t
j ddg�� tjtd|g�j t jg�t
j ddg�� d S )Nz'pd.NaT not compatible with Float64Indexz!Float64Index cannot contain pd.NAg �?FT)
rL �NaTro �skipZNA�xfailr) rA r
r� r. ri r< )r r� r! r! r" � test_isin_nan_common_float64� s
$z&TestIndex.test_isin_nan_common_float64�levelc C s^ |j � dd � dg }tjddddg�}tj||j||d�� d|_tj||j|dd�� d S )Nr( Z nonexistingFT)r� r� ���)r� r. r< r) rA r� r@ )r r� r% rB rV r! r! r" �test_isin_level_kwarg� s
z TestIndex.test_isin_level_kwargc C sJ xDd|j |j d gD ],}tjtdd�� |jg |d� W d Q R X qW d S )Nr� rt zToo many levels)rn )r� )Znlevelsro rp r� r� )r r% r� r! r! r" �&test_isin_level_kwarg_bad_level_raises� s z0TestIndex.test_isin_level_kwarg_bad_level_raises�labelr� Zxyzzyc C sv t |t�r4|jddg|jdd � �}d|� d�}n|jd�}d|� d�}tjt|d�� |jg |d � W d Q R X d S )
Nr� r r( z'Level z not found'zRequested level \(z$\) does not match index name \(foo\))rn )r� )r- r r� �namesro rp r| r� )r r� r% rr r! r! r" �&test_isin_level_kwarg_bad_label_raises� s
z0TestIndex.test_isin_level_kwarg_bad_label_raisesc C s4 t ddg�}tjddg�}|j|�}tj||� d S )Nr_ r` F)r r. r< r� r) rA )r r� r% rV rP r! r! r" �test_isin_empty s
zTestIndex.test_isin_emptyz
2018-01-01c C s6 t |�}||k}tjddddgtd�}tj||� d S )NT)r4 )r r. r< r� r) rA )r rB r% rP rV r! r! r" �test_boolean_cmp s zTestIndex.test_boolean_cmpz
name,levelc C s, |j � }|r||_|j|�}tj||� d S )N)r6 r@ Zget_level_valuesr) r= )r r% r@ r� rV rP r! r! r" �test_get_level_values# s
z TestIndex.test_get_level_valuesc C s, t ddgdd�}|j|dd � jks(t�d S )Nr_ r` Zasdf)r@ rt )r r@ r$ )r r% r! r! r" �test_slice_keep_name- s z TestIndex.test_slice_keep_namec C s |j ||d�}||kst�d S )N)�how)�joinr$ )r r% Z join_typeZjoinedr! r! r" �test_join_self1 s zTestIndex.test_join_self�strip�rstrip�lstripc sF t ddddg�}t � fdd�|jD ��}t|j� �� }tj||� d S )Nz jackzjill z jesse Zfrankc s g | ]}t t� �|��qS r! )r� rP )r� r� )r^ r! r" r� > s z0TestIndex.test_str_attribute.<locals>.<listcomp>)r rB r� rP r) r= )r r^ r% rV rP r! )r^ r" �test_str_attribute: s z TestIndex.test_str_attributer� �3r7 Z2010)r� r� rG c C s* t jtdd�� |jjd� W d Q R X d S )Nzonly use .str accessor)rn r( )ro rp r� rP �repeat)r r% r! r! r" �test_str_attribute_raisesC s
z#TestIndex.test_str_attribute_raiseszexpand,expectedr �fc C s@ t dddg�}|d k r&|jj|d�}n
|jj� }tj||� d S )Nza b czd er� )�expand)r rP �splitr) r= )r r� rV r% rP r! r! r" �test_str_splitP s
zTestIndex.test_str_splitc C sN t ddddg�}|jjd�}tjddddg�}tj||� t|tj�sJt �d S )Nr� �a2�b1�b2r_ TF)
r rP �
startswithr. r< r) rA r- r/ r$ )r r% rP rV r! r! r" �test_str_bool_returnf s
z TestIndex.test_str_bool_returnc C sV t ddddg�}ttd�|d�}||jjjd� }ttd�ddgd�}tj||� d S ) Nr� r� r� r� ry )r% r_ r( )r r r� r% rP r� r) r\ )r r% rW rP rV r! r! r" � test_str_bool_series_indexingo s
z'TestIndex.test_str_bool_series_indexingZabcdc C s dt |�k}||kst�d S )NrP )�dirr$ )r r% rV rP r! r! r" �test_tab_completionw s z TestIndex.test_tab_completionc C sd t ddddddg�}|dd� jtj ddgtjd��s:t�|ddg jtj ddgtjd��s`t�d S ) Nrt r( rF r_ r` ra )r4 r )r r� rL r. rO r$ )r r% r! r! r" �!test_indexing_doesnt_change_class s &z+TestIndex.test_indexing_doesnt_change_classc C sz t tjjd��}tjd�}tjt�� |j|dd�}W d Q R X tjt�� |j t
�j |j t
��}W d Q R X tj ||� d S )N� r� �outer)r� )
r r. r[ Z permutationr) r r* �RuntimeWarningr� rM rN r� r= )r Z
left_indexZ right_indexrP rV r! r! r" �test_outer_join_sort� s
z TestIndex.test_outer_join_sortc C sR t tjtd�td�g�}|jdddg�}t |d |d |d g�}tj||� d S )Nr� Z20130102rt r r� r� )r rL r� r �taker) r= )r r% rP rV r! r! r" � test_nan_first_take_datetime� s z&TestIndex.test_nan_first_take_datetimec C s� t jtd�dd�}|jtjdddg��}t jtd�dd�}tj||� |jtjdddg�dd�}t jd d
tjgdd�}tj||� |jtjdddg�d dd �}t jd d
d
gdd�}tj||� d S )N�ABC�xxx)r@ rt r ZBACT)�
fill_valuer{ rz F)Z
allow_fillr� r| r� r� r� ) rL r r r� r. r< r) r= ri )r r% rP rV r! r! r" �test_take_fill_value� s z TestIndex.test_take_fill_valuec C s� t jtd�dd�}d}tjt|d�� |jtjddd g�d d
� W d Q R X tjt|d�� |jtjddd
g�d d
� W d Q R X d S )Nr� r� )r@ zJWhen allow_fill=True and fill_value is not None, all indices must be >= -1)rn rt r r( T)r� rY r� ���) rL r r ro rp rq r� r. r< )r r% rr r! r! r" � test_take_fill_value_none_raises� s $z*TestIndex.test_take_fill_value_none_raisesc C sD t jtd�dd�}tjtdd�� |jtjddg�� W d Q R X d S ) Nr� r� )r@ z
out of bounds)rn rt rY r� ) rL r r ro rp r� r� r. r< )r r% r! r! r" �test_take_bad_bounds_raises� s z%TestIndex.test_take_bad_bounds_raises�labelsr� c C s2 t jdddg�}||_|j|�d j|ks.t�d S )Nr rt r( )rL r r@ �reindexr$ )r r@ r� r% r! r! r" �8test_reindex_preserves_name_if_target_is_list_or_ndarray� s zBTestIndex.test_reindex_preserves_name_if_target_is_list_or_ndarrayc C s. t jtd��}|j|�d jjtjks*t�d S )Nr� r ) rL r r r� r4 r� r. rO 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 jtd��}|j|�d jj|ks(t�d S )Nr� r )rL r r r� r4 r� r$ )r r� r4 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 sl t jtd��}|jt jt jg �t jg �gg g g��d }|jd jj t
j ksPt �|jd jj t
j
ksht �d S )Nr� r rt )rL r r r� r r r
r� r4 r� r. rl r$ �float64)r r% rP r! r! r" �-test_reindex_no_type_preserve_target_empty_mi� s z7TestIndex.test_reindex_no_type_preserve_target_empty_mic C sV t td��}|jtjdddddg��}tj ddg�tj dddg�d�}tj||� d S )NrY rt r( r rF ry )rt r( )r r� �groupbyr. r< rL r) Zassert_dict_equal)r r% rP rV r! r! r" � test_groupby� s zTestIndex.test_groupbyz mi,expectedc C s0 t jtd�ddgd�}|j|k}tj||� d S )Nza,b,c
1,2,3
4,5,6r rt )� index_col)rL �read_csvr r% r) rA )r �mirV r] rP r! r! r" �test_equals_op_multiindex� s
z#TestIndex.test_equals_op_multiindexc C s@ t jtd�ddgd�}|j|jk}tjddg�}tj||� d S )Nza,b,c
1,2,3
4,5,6r rt )r� T)rL r� r r% r. r< r) rA )r r] rP rV r! r! r" �"test_equals_op_multiindex_identify s z,TestIndex.test_equals_op_multiindex_identifyc C s> t jtd�ddgd�}tjtdd�� |j|k W d Q R X d S )Nza,b,c
1,2,3
4,5,6r rt )r� zLengths must match)rn )rL r� r ro rp rq r% )r r% r] r! r! r" �+test_equals_op_mismatched_multiindex_raises s z5TestIndex.test_equals_op_mismatched_multiindex_raisesc C sF t jd d d
g�}tddd g�}||k}tjd
d
d
g�}tj||� d S )Nrt r( ry rY rk r� r� r r F)rt r( )ry rY )rk r� )r r r r. r< r) rA )r r� r% rP rV r! r! r" �&test_equals_op_index_vs_mi_same_length s
z0TestIndex.test_equals_op_index_vs_mi_same_length�dt_convc C s* t jddgdd�}|j||�jks&t�d S )Nz01:02:03z01:02:04r� )r@ )rL r r@ r$ )r r� r% r! r! r" �!test_dt_conversion_preserves_name s z+TestIndex.test_dt_conversion_preserves_nameZbb�cccz)Index(['a', 'bb', 'ccc'], dtype='object')z�Index(['a', 'bb', 'ccc', 'a', 'bb', 'ccc', 'a', 'bb', 'ccc', 'a', 'bb', 'ccc',
'a', 'bb', 'ccc', 'a', 'bb', 'ccc', 'a', 'bb', 'ccc', 'a', 'bb', 'ccc',
'a', 'bb', 'ccc', 'a', 'bb', 'ccc'],
dtype='object')r� z�Index(['a', 'bb', 'ccc', 'a', 'bb', 'ccc', 'a', 'bb', 'ccc', 'a',
...
'ccc', 'a', 'bb', 'ccc', 'a', 'bb', 'ccc', 'a', 'bb', 'ccc'],
dtype='object', length=300)u あu いいu うううu5 Index(['あ', 'いい', 'ううう'], dtype='object')uW Index(['あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう',
'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう',
'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう'],
dtype='object')u Index(['あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ',
...
'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう'],
dtype='object', length=300)c C s t |�}||kst�d S )N)�reprr$ )r r% rV rP r! r! r" �test_string_index_repr# s =z TestIndex.test_string_index_repru^ Index(['あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう',
'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう',
'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう',
'あ', 'いい', 'ううう'],
dtype='object')u Index(['あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう',
'あ',
...
'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい',
'ううう'],
dtype='object', length=300)c C s0 t jdd�� t|�}||ks"t�W d Q R X d S )Nz display.unicode.east_asian_widthT)�cf�option_contextr r$ )r r% rV rP r! r! r" �*test_string_index_repr_with_unicode_optionc s )z4TestIndex.test_string_index_repr_with_unicode_optionc C s4 t jdddg�}tjtdd�� d|_W d Q R X d S )Nrt r( rF zCan't set attribute)rn F)rL r ro rp r� r# )r r% r! r! r" �#test_cached_properties_not_settable� s z-TestIndex.test_cached_properties_not_settablec � s� t jddd� ddlm} d}|j|�I d H dd l}|jdk rLtjd �}ntjt dd �}|�, |d
�� t
|j j d d �� W d Q R X W d Q R X d S )
NZIPythonz6.0.0)Z
minversionr )�provisionalcompleterz+import pandas as pd; idx = pd.Index([1, 2])z0.16.0F)Zcheck_stacklevelr' zidx.ry )
ro Z importorskipZIPython.core.completerr Zrun_code�jedi� __version__r) r* �DeprecationWarningr Z CompleterZ completions)r �ipr �coder �warningr! r! r" �test_tab_complete_warning� s
z#TestIndex.test_tab_complete_warningc C sR t |tj�r|jd� n6dt|�j� d�}tjt|d�� |jd� W d Q R X d S )Nrt �'z$' object has no attribute 'contains')rn ) r- rL �
IntervalIndex�containsr� rd ro rp r� )r r% rr r! r! r" � test_contains_method_removed� s
z&TestIndex.test_contains_method_removed)rz rz rz )rz r{ N)Nr{ N)rz r{ N)Nr{ N)rz NNr� r� )r� r� )r N)NN)r@ r@ )r; F)r� F)r� F)r2 T)r F)r3 T)r; T)r� T)r� F)r2 F)r F)r3 F)r; F)r� F)r� F)r2 F)r T)r3 Fy @ @r� r� r� r� r� r� r� r� r� r� r� r� r� )rd rt )re r( )rj rt r� r� r� r� r� r� ���r� r� r� ���r� r� r� r� r� )ra r� )ra r� )ra r� )ra r� )ra r� )rt rz )r( rz )rt r{ )r( r{ )r� r� )rt rz )rt r{ )r( rz )r( r{ )r� r� )rt rz )rt r{ )rt r| )r( rz )r( r{ )r( r| )r� r� r� )Nr )r_ r_ )r� r� )r r� )r_ r` ra )rt r( )ry rY )rt r( )ry r� )rt r( )ry rY )rk r� )�rd re rf r �_holderr# r' ro �mark� parametrizer1 r8 r: r? rC rL r r� r rQ rR rX r^ r. r$ r< rB rh rk rs rv r
r� ri r r� rx r r2 r3 r� r� r r r� r� Z timedelta64r r� r� r r� �iterr r r
r� r r r r� r� r� �paramr� r� r� r� r� r� r� r� r� r� r� r� r� r� �int_Zbool_r� r� r� r� r� r� r� r� r� r� r� �slicer r r r r r r
r r r r r r r" r% r& r( r+ r. r0 r1 r2 r3 r5 r< r= r> r? rA rC rD rE rF rG rH rK rL rN rS rU rW rZ r] rI rb r` rf ri rm �tupler ro rp rq rr rs ru rw rx ry r{ r} r r� r� r� r� r� r� r� r� r� r� Z
IndexSlicer� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rN r� r� r� r� r� r� r� r) r r r� r� r� r� r� r� r� r� r� r� r� r� r� rl r� Z
datetime64r� r� r� r� r� r� r� rT � to_timedeltar r r r r r r r! r! r! r" r . s�
$ * $ , .
" &2 "
$
6
$ &
0
.
$ 2
$$4 ,: D 0 " ** "(
, $
,$ :". "( $ * ,
"
& (
$ 6 4 & 4 0 $
r c
@ s� e Zd ZeZejddddddggdgd�d d
� �Zed �d d
�Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zejjddd g�ejjd d ed g�ejjd ddd!gddd!gfd"d#d$gd"d#d$gfd"d#ejd$gd"d#d$gfd%d&d'gd%d&d'gfd%ejd&d'gd%d&d'gfg�d(d)� ���Zejjddd g�ejjd*ejd+d,d-g�ejd+d,d-g�fejd+d,d-ejg�ejd+d,d-g�fejd.d/d0g�ejd.d/d0g�fejejd.d/d0ejg�ejd.d/d0g�fejd1d2d3gd4d5�ejd1d2d3gd4d5�fejd1d2d6d3gd4d5�ejd1d2d3gd4d5�fg�d7d8� ��Zd9d:� Zd;d<� Z ejjd=ejejg�ejejdg�ejddejg�ejddejg�ej d6g�ej d6d>g�ej d>d6d?g�ej d@d6g�g�dAdB� �Z dCdD� Z ejjdEe!e"g�dFdG� �Z#dHdI� Z$dJdK� Z%dLdM� Z&ejj'dN�dOdP� �Z(d S )Q�TestMixedIntIndexr r_ rt r` r( ra Z
mixedIndex)�params�idsc C s
t |j�S )N)r r )r �requestr! r! r" r% � s zTestMixedIntIndex.index)r c C s t ddddddg�S )Nr r_ rt r` r( ra )r )r r! r! r" r# � s z TestMixedIntIndex.create_indexc C s. | j � }tjtdd�� |j� W d Q R X d S )Nz'>|<' not supported)rn )r# ro rp r� �argsort)r r% r! r! r" � test_argsort� s z TestMixedIntIndex.test_argsortc C s0 | j � }tjtdd�� tj|� W d Q R X d S )Nz'>|<' not supported)rn )r# ro rp r� r. r# )r r% r! r! r" �test_numpy_argsort� s z$TestMixedIntIndex.test_numpy_argsortc C s� | j � }t|�|ddd�}t|�|dd�}||k s6t�tj||� |jdksPt�|jdks^t�td|d�}td|d d
� d�}|| }|jjdks�t�d S ) NTZmario)r6 r@ F)r6 r( )r% rF rt r� )r# r� r$ r) r= r@ r r% )r r% r� r� �s1�s2Zs3r! r! r" �test_copy_name� s z TestMixedIntIndex.test_copy_namec C s� t jddgdd�}|j� }tj||� |jdd�}tj||dd� |jdksPt�|jdks^t�|jdgd�}tj||dd� |jdks�t�|jdgks�t�|jdks�t�|jdgks�t�d S ) Nrt r( ZMyName)r@ ZNewNameF)Z check_names)r� )rL r r6 r) r= r@ r$ r� )r r% r� r� Zindex3r! r! r" �test_copy_name2� s z!TestMixedIntIndex.test_copy_name2c C sF t jdtjddgdd�}t jdtjdgdd�}|j� }tj||� d S )Nr( rt Zmy_index)r@ )rL r r. ri �uniquer) r= )r r� rV rP r! r! r" �test_unique_na� s z TestMixedIntIndex.test_unique_nac C s8 | j � }|j� |jj� ks t�|j� |jj� ks4t�d S )N)r# r\ rB r$ r[ )r r% r! r! r" r] � s z%TestMixedIntIndex.test_logical_compatr� r[ r\ r4 Nr� z
vals,expectedrF g �?g @g @rz r{ r| c C s8 t j||d�}|j|d�}t j||d�}tj||� d S )N)r4 )r� )rL r �dropnar) r= )r r� r4 r~ rV r% rP r! r! r" � test_dropna s z TestMixedIntIndex.test_dropnazindex,expectedz
2011-01-01z
2011-01-02z
2011-01-03z1 daysz2 daysz3 daysz2012-02z2012-04z2012-05�M)rG r� c C s |j |d�}tj||� d S )N)r� )r, r) r= )r r� r% rV rP r! r! r" �test_dropna_dt_like s z%TestMixedIntIndex.test_dropna_dt_likec C s: d}t jt|d�� tjdddg�jdd� W d Q R X d S )Nzinvalid how option: xxx)rn rt r( rF r� )r� )ro rp rq rL r r, )r rr r! r! r" � test_dropna_invalid_how_raises6 s z0TestMixedIntIndex.test_dropna_invalid_how_raisesc C s t g �}tg �}tj||� d S )N)r r r) r= )r rP rV r! r! r" �test_get_combined_index; s z)TestMixedIntIndex.test_get_combined_indexr% z
2000-01-01z
2000-01-02z1 dayc C s< |j dkst�|jdks t�|jdks*t�|jdks8t�d S )NF)r� r$ r� r� r� )r r% r! r! r" �test_is_monotonic_na@ s z&TestMixedIntIndex.test_is_monotonic_nac C sL t jdd��6 ttjtjd���}t|�dk s2t�d|ks>t�W d Q R X d S )Nzdisplay.max_seq_itemsr� i� �� z...) r r r rL r r. r$ r� r$ )r rP r! r! r" �test_repr_summaryS s z#TestMixedIntIndex.test_repr_summaryrw c C s: t dddgdd�}|ttd��|d�}dt|�ks6t�d S ) Nr_ r` ra r )r@ rF )r% �0)r r r� r r$ )r rw r% rP r! r! r" �test_int_name_formatY s z&TestMixedIntIndex.test_int_name_formatc C s@ t dd� td�D ��}d}tjt|d�� t|� W d Q R X d S )Nc S s g | ] }t |��qS r! )rP )r� r� r! r! r" r� a s z>TestMixedIntIndex.test_str_to_bytes_raises.<locals>.<listcomp>r� z2^'str' object cannot be interpreted as an integer$)rn )r r� ro rp r� �bytes)r r% rr r! r! r" �test_str_to_bytes_raises_ s z*TestMixedIntIndex.test_str_to_bytes_raisesc C sX t ddd�t ddd�g}t|td�}tdgtd�}|j|�}tg td�}tj||� d S )Ni� r( r� r� )r4 r� )r r rN r� r) r= )r r� r� r� rP rV r! r! r" r� f s
z*TestMixedIntIndex.test_intersect_str_datesc C sN t ddtjgtd�}|j� }dddg}||ks2t�t|�}d}||ksJt�d S )NTF)r4 �True�FalserT z)Index([True, False, nan], dtype='object'))r r. ri rN rM r$ r )r r> Zexp1Zout1Zexp2Zout2r! r! r" �test_index_repr_bool_nanp s
z*TestMixedIntIndex.test_index_repr_bool_nanz2ignore:elementwise comparison failed:FutureWarningc C s: t ddd gdd�}|d
k}tjdddg�}tj||� d S ) Nr_ r` ra F)r� T)r_ r` )r` ra )ra r_ )ra r_ )r r. r< r) rA )r r� rP rV r! r! r" �test_index_with_tuple_bool{ s z,TestMixedIntIndex.test_index_with_tuple_bool))rd re rf r r ro Zfixturer% r# r$ r% r( r) r+ r] r r rN r. ri r- rL r r� r r r/ r0 r1 rT r r2 r4 r r r6 r8 r� r; �filterwarningsr< r! r! r! r" r � sf &(
$
r c @ s� e Zd Zejjddddggdedddg�fdddggdgedddgdd�fddgd d
ggdedgd d
ggd d gd dgg�fddgd d
ggd d
gedgd d
ggd d gd dggd d
gd�fg�dd� �Zdd� Z dS )�TestIndexUtilszdata, names, expectedrt r( rF Nr@ )r@ r_ ra r� r ZL1ZL2)r� c C s t ||�}tj||� d S )N)r r) r= )r rj r� rV rP r! r! r" � test_ensure_index_from_sequences� s
z/TestIndexUtils.test_ensure_index_from_sequencesc C s` t jdddd�t jdddd�t jdddd�t jdd d
d�g}t|�}t|td �}tj||� d S ) Nr rt r, )�closedr( r- rF Zneitherry Zboth)r4 )rL ZIntervalr r rN r) r= )r Z intervalsrP rV r! r! r" �(test_ensure_index_mixed_closed_intervals� s z7TestIndexUtils.test_ensure_index_mixed_closed_intervals)
rd re rf ro r r r r r? rA r! r! r! r" r> � s 6r> r6 �eq�ne�le�lt�ge�gt�addZradd�sub�rsub�mulZrmul�truedivZrtruediv�floordivZ rfloordiv�powZrpow�mod�divmodc C s> t |t�r| dkrd S d| � d�} t|| �}|j| ks:t�d S )NrJ �__)r- r r� rd r$ )r6 r% r^ r! r! r" �test_generated_op_names� s
rR � index_makerc
C s( t jtdd�� | dd� W d Q R X d S )Nzunexpected keyword argument)rn r )r� )ro rp r� )rS r! r! r" �,test_index_subclass_constructor_wrong_kwargs� s rT c ! C s� d} t jt| d��$ tjtjddgtd�ddd� W d Q R X t jt| d��& tjtjd d
d gd d�ddd� W d Q R X t jt| d�� tj d
dd
ddd� W d Q R X t jt| d�� tj
dddgddd� W d Q R X d S )Nz [Uu]nexpected keyword argument)rn r_ r` )r4 �testT)r@ Zfastpathrt r( rF rl r rY ra ) ro rp r� rL r r. r< rN r r r
)rr r! r! r" �test_deprecated_fastpath� s (* rV c
C sF t jddddg�} tjt�� | d d �d f jdks8t�W d Q R X d S )Nr rt r( rF ry )ry rt )rL r r) r* r+ �shaper$ )r� r! r! r" �test_shape_of_invalid_index� s rX c - C s: d} t jd�jddd�}tjt| d�� tj|� W d Q R X tjt| d�� tj|j t j
�� W d Q R X tjt| d�� tj |j t j �� W d Q R X tjt| d�� tj
|j t j�� W d Q R X tj|jdd��}tjt| d�� tj|� W d Q R X tjdtd��}tjt| d�� t jddggd �|_W d Q R X d S )Nz Index data must be 1-dimensionalrk r( )rn ry r rF )r. r$ Zreshapero rp rq rL r r
rM r� r rl r rm r r r� r< r% )rr r> r] ru r! r! r" �test_validate_1d_input� s rY c
C sr t d d d�}t| tj�rDd}tjt|d�� | j|d� W d Q R X n*d}tjt|d�� | j|d� W d Q R X d S )Nr� zClabel-based slicing with step!=1 is not supported for IntervalIndex)rn rz z7'>=' not supported between instances of 'str' and 'int') r r- rL r ro rp rq Z_convert_slice_indexerr� )r% r& rr r! r! r" � test_convert_almost_null_slice
s rZ ztimedelta64[ns]zdatetime64[ns]r� z
Asia/Tokyo�ldtype�rdtypec s� t jdt jd� d�� � fdd�}|| �}||�}|j|�}| |kr�t jddd d
dd gd t jd �}t jg t jd �}tj|d |� tj|d |jt j �� n6t jdgd
t jd �}tj|d |� tj|d |� d S )Ni r� r� rF r( c s* | t d kr tj� �j| �S tj� | d�S )Nrt )r4 r� )�
dtlike_dtypesrL r rM r )r4 )r~ r! r" � construct'
s z:test_get_indexer_non_unique_wrong_dtype.<locals>.constructr rt ry rY )r4 r� i ʚ;l qa
r� )
r. Ztiler$ Zget_indexer_non_uniquer< r` r) rA rM rl ) r[ r\ r^ r, r- rP Zex1Zex2Z
no_matchesr! )r~ r" �'test_get_indexer_non_unique_wrong_dtype!
s
r_ )C� collectionsr r r �ior r� r9 r� Znumpyr. ro Zpandas._config.configZ_config�configr Zpandas._libs.tslibr Zpandas.compat.numpyr Z pandas.util._test_decoratorsr Zpandas.core.dtypes.genericr ZpandasrL r
r r r
r r r r r r r r r Zpandas._testingZ_testingr) Zpandas.core.indexes.apir r r r r Zpandas.tests.indexes.commonr r r r> r r rR Z index_subclass_makers_generatorrT rV rX rY rZ r4 ZDatetimeTZDtypeZ PeriodDtyper] r_ r! r! r! r" �<module> s� <
R#
|