Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexes/__pycache__/test_base.cpython-37.pyc
Назад
B �5�g$� � @ s� d dl mZ d dlmZmZ d dlmZ d dlZd dlZd dlZ d dl Z d dl m Z m Z d dlmZ d dlZd dlmZmZmZmZmZmZmZmZmZmZm Z m Z m Z m Z m Z d dl!m"Z# d dl$m%Z%m&Z&m'Z'm(Z(m)Z) d d l*m+Z+ G d d � d e+�Z,G d d � d e+�Z-G dd� d�Z.e j/�0ddddddddddddd d d d d d!d"d#d$g�d%d&� �Z1e j/�0d'e#�2� �d(d)� �Z3e j/�4d*�d+d,� �Z5d-d.� Z6d/d0� Z7e j/�0d1e%i gei gei gei gei ged2d3igg�d4d5� �Z8d6d7� Z9d8d9� Z:dS ):� )� defaultdict)�datetime� timedelta)�StringION)�IS64�np_datetime64_compat)� async_mark)�CategoricalIndex� DataFrame� DatetimeIndex� Float64Index� Int64Index� IntervalIndex� PeriodIndex� RangeIndex�Series�TimedeltaIndex� Timestamp� UInt64Index� date_range�isna� period_range)�Index� MultiIndex�_get_combined_index� ensure_index�ensure_index_from_sequences)�Basec @ sf e Zd ZeZejed�dd��Zdd� Zej j ddgdd �d d � �Z d d � Z ej j ddgdd �dd� �Z ej j ddgdd �dd� �Zej � dddg�ej � dedddddd�edddd�ejd ddd�ed ddd�g�d d � ��Zej � d eddddd!�dfejd ddd�dfed ddd�dfg�d"d#� �Zd$d%� Zd&d'� Zej � d(e�d)�e�d*d+d,g�ed-dd.�jg�d/d0� �Zd1d2� Z ej � d3d4d5g�d6d7� �Z d8d9� Z ej � d:e ej ej!fe"d;ej#fg�d<d=� �Z$ej � d>d?d@ddAd)gdBfdCej!dDdEgdFfdGdHdIej!gdJfg�dKdL� �Z%ej � dMd?d@dge�d?d@dg�ejd?d@dge&dN�dOdPdEgejdOdPdEge'dN�g�dQdR� �Z(ej � dMd?d@dgdOdPdEge�dOdPdEg�ejd?d@dge&dN�ejdOdPdEge'dN�g�dSdT� �Z)ej � dMd?d@dgejd?d@dge&dN�e�e*dU�e*dV�g�e+dWd?d?�e+dWd?d@�gg�dXdY� �Z,ej � dZddg�ej � dMee�e*dU�e*dV�g��ee+dWd?d?�e+dWd?d@�g�g�d[d\� ��Z-ej � dZddg�ej � dMe�e�.d?d�e�.d?d�g�e/d?�e/d?�gg�d]d^� ��Z0ej �1d_�ej � d`dadbg�ej � dcee"g�ddde� ���Z2ej � d`dadbg�ej � dcee3g�dfdg� ��Z4ej � dhg e5g �didj� dkD �g�ej � dcee e6e7e8e"e3g�dldm� ��Z9ej � dne:g dHdo�e:fe:e5g �dHdo�e:fe:dpdj� dkD �dHdo�e:fe;d?dq�e;fe<d?d@gdrdsggg g gdt�e<fg�dudv� �Z=dwdx� Z>ej j dddFdBdydzd{d|d}d~g dd �dd�� �Z?ej j dd�dej@d�ej jAd�d��d��d�d�gdd �d�d�� �ZBej j ddBdzgdd �d�d�� �ZCd�d�� ZDej � d�ed*d+g�ed*d+d�g�d*d+d,gg�d�d�� �ZEd�d�� ZFd�d�� ZGd�d�� ZHd�d�� ZIej j ddgdd �d�d�� �ZJd�d�� ZKd�d�� ZLej j ddgdd �d�d�� �ZMd�d�� ZNej j dddBdFgdd �ej � d3ejOejPg�d�d�� ��ZQej j dddBdFgdd �d�d�� �ZRd�d�� ZSd�d�� ZTd�d�� ZUej � d`d�d�d�g�d�d�� �ZVd�d�� ZWej � d�d�d�� d�d�� g�d�d�� �ZXej � d�d�d�� d�d�� g�d�d�� �ZYej � d�eZd�dPd�gd�d@d�gdÍd�dPd�dĜg�d�dƄ �Z[d�dȄ Z\d�dʄ Z]ej � d�d�d�g�d�dτ �Z^d�dф Z_ej j d�d�d�d�d�d�d�gdgd �d�dڄ �Z`ej j d�d�d�d�d�d�d�gdgd �d�d� �Zaej j d�d�d�d�d�d�d�gdgd �d�d� �Zbd�d� Zcd�d� Zdd�d� Zed�d� Zfej � dMd?d�d�gd*d+d,gg�d�d� �Zgd�d� Zhd�d� Ziej � d�d�d�g�d�d�� �Zjej j dddBdFgdd �d�d�� �Zkej j dddBdFgdd �ej � d�d�d�gd�d�gg�d�d�� ��Zlej j dddBdFgdd ��d �d� �Zm�d�d� Zn�d�d� Zoej � �ddAed?d@dg�fddAd)ged?d@g�fg��d�d� �Zpej � dad*d+�d gd*�d d+g�d d*d+gg�ej � �d �d d*gd*�d gg��d �d � ��Zq�d �d� Zrej � d`�d�d�d�dg��d�d� �Zs�d�d� Ztej � dad�d��dgd�d��dhg�ej � �de�dd�d�d�g�e�ddddg�feg �ejg eudN�fg��d�d� ��Zv�d �d � Zw�d �d � Zxej � �d d�d�g�ej � de�dd�d�d�g�e dOdPdEd�g�g��d!�d"� ��Zy�d#�d$� Zzej � �d%dO�d&�d'ej!g��d(�d)� �Z{ej � d�g eZe|dN�e�g �g��d*�d+� �Z}ej � dad?d@ddAgdOdPdEd�gddddgd�d�d��dge�d,ddAd�g��d-�d.� �Z~ej j ddgdd �ej � �d/�d0�d1g��d2�d3� ��Z�d4�d5� Z�ej j dd�dddBd~dFgdd ��d6�d7� �Z�ej � �d8�d9�d:�d;g��d<�d=� �Z�ej � dee�d)��e����d>�e<���d?�d@g�e�dA�dBdG�dC�g��dD�dE� �Z�ej � �dF�dGed*d+d,gd��dHg�dIgg�fded*d+d,gd��dHg�dIgg�fde<���dJd��dHej!f�dIej!ej!fg�fg��dK�dL� �Z��dM�dN� Z��dO�dP� Z�ej � �dee��dQ��dfee�dA��dfg��dR�dS� �Z��dT�dU� Z��dV�dW� Z��dX�dY� Z��dZ�d[� Z��d\�d]� Z��d^�d_� Z�ej � �d`�dG�d&g�ej � �dag e�g �dGdHdIgdIdHdGge�dGdHdIg�e�dIdHdGg�e�dbdd.�je�dbdd.���� g��dc�dd� ��Z�ej � �dag e�g �ejg ej�dN�g��de�df� �Z�ej � �dge6g �ej�fe g �ej fe"g �ej�fg��dh�di� �Z��dj�dk� Z��dl�dm� Z�ej � �dne<���do�dpg�e�ddg�fe<���do�dqg�e�ddg�fg��dr�ds� �Z��dt�du� Z�ej � de<���do�dp�dvg�ed�d�d�g�g��dw�dx� �Z��dy�dz� Z�ej � �d{ej�ej�g��d|�d}� �Z��d~�d� Z�e�� �d��d�� �Z��d��d�� Z��d��d�� Z��dGS (� � TestIndex)�returnc C s | � td��S )NZabcde)� _index_cls�list)�self� r# �T/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexes/test_base.py� simple_index3 s zTestIndex.simple_indexc C s" |}|d }|� |�dks t�d S )Nr T)Z$_can_hold_identifiers_and_holds_name�AssertionError)r" r% �index�keyr# r# r$ �test_can_hold_identifiers7 s z#TestIndex.test_can_hold_identifiersr' r T)Zindirectc C sH t �t�� |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_axis< s zTestIndex.test_new_axisc C s t �||� d S )N)r+ �assert_contains_all)r" r' r# r# r$ �test_constructor_regularD s z"TestIndex.test_constructor_regular�stringc C s. t �|�}t|�}t�||� t�||� d S )N)r0 �arrayr r+ r4 �assert_index_equal)r" r' �arrr2 r# r# r$ �test_constructor_castingG s z"TestIndex.test_constructor_castingc C s^ t �|�}t|ddd�}t|t�s&t�|jdks4t�t�||j� d|d< |d dksZt�d S )NT�name)�copyr; ZSOMEBIGLONGSTRINGr ) r0 r7 r r/ r& r; r+ �assert_numpy_array_equal�values)r" r' r9 r2 r# r# r$ �test_constructor_copyO s z TestIndex.test_constructor_copy� cast_as_objFz2015-01-01 10:00�D� z US/EasternzGreen Eggs & Ham)�freq�periods�tzr; )rC rD z1 daysz 2015-01-01c C s� |rt |�t��}nt |�}t�||� t|t�r�|j|jksBt�|r�|t j dd�7 }t |td�}|j t j ksrt�t|�t|�ks�t�d S )N�2 )� nanoseconds)�dtype)r �astype�objectr+ r8 r/ r rE r& �pdZ TimedeltarH r0 �object_r! )r" r@ r' �resultr# r# r$ �"test_constructor_from_index_dtlike^ s z,TestIndex.test_constructor_from_index_dtlikez index,has_tz)rC rD rE c C s0 t t|��}t�||� |r,|j|jks,t�d S )N)r r r+ r8 rE r& )r" r' Zhas_tzrM r# r# r$ �#test_constructor_from_series_dtlike� s z-TestIndex.test_constructor_from_series_dtlikec C sD dddddg}t |dd�}tt�|��}t |dd�}t�||� d S )Nz1-1-1990z2-1-1990z3-1-1990z4-1-1990z5-1-1990�MS)rC )r r rK � to_datetimer+ r8 )r" �dts�expected�srM r# r# r$ �!test_constructor_from_series_freq� s z+TestIndex.test_constructor_from_series_freqc C s� dddddg}t |dd�}ttj�dd ��}||d < t |d dd�}|d jtksVt�d |_t � ||� t |d d �}t � |d |� t �|d �}|dks�t�d S ) Nz1-1-1990z2-1-1990z3-1-1990z4-1-1990z5-1-1990rP )rC � rB �date)r; )r r r0 �randomZrandrH rJ r& r; r+ r8 r �assert_series_equalrK Z infer_freq)r" rR rS �dfrM rC r# r# r$ �'test_constructor_from_frame_series_freq� s z1TestIndex.test_constructor_from_frame_series_freqr7 rV �a�b�cz 2000-01-01)rD c C s2 G dd� d�}t |�}t ||��}t�||� d S )Nc @ s& e Zd Zdd� Zdejd�dd�ZdS )z:TestIndex.test_constructor_ndarray_like.<locals>.ArrayLikec S s || _ d S )N)r7 )r" r7 r# r# r$ �__init__� s zCTestIndex.test_constructor_ndarray_like.<locals>.ArrayLike.__init__N)r c S s | j S )N)r7 )r" rH r# r# r$ � __array__� s zDTestIndex.test_constructor_ndarray_like.<locals>.ArrayLike.__array__)N)�__name__� __module__� __qualname__r_ r0 r1 r` r# r# r# r$ � ArrayLike� s rd )r r+ r8 )r" r7 rd rS rM r# r# r$ � test_constructor_ndarray_like� s z'TestIndex.test_constructor_ndarray_likec C s, t jg}t|�}t|dd�}t�||� d S )N�float)rH )r0 �nanr r r+ r8 )r" �datarS rM r# r# r$ � test_constructor_int_dtype_nan� s z(TestIndex.test_constructor_int_dtype_nanrH �int64�uint64c C s6 t jg}d}tjt|d�� t||d� W d Q R X d S )Nzcannot convert)�match)rH )r0 rg �pytest�raises� ValueErrorr )r" rH rh �msgr# r# r$ �%test_constructor_int_dtype_nan_raises� s z/TestIndex.test_constructor_int_dtype_nan_raisesc C s6 t dddg�}t|j�}tdddg�}t�||� d S )N� r* rB )r r r7 r+ r8 )r" �serrM rS 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|�}t�||� tt�|��}t�||� d S )N)rH r& r r+ r8 r0 r7 )r" �klassrH Zna_valZna_listrS rM r# r# r$ � test_index_ctor_infer_nan_nat� s z'TestIndex.test_index_ctor_infer_nan_natz vals,dtyperr r* � �intg�������?g������@g @rf �A�B�C�objc C s* t ||d�}|�|j|�}t�||� d S )N)r; )r Z _simple_newr> r+ r8 )r" �valsrH r' rM r# r# r$ �test_constructor_simple_new� s z%TestIndex.test_constructor_simple_newr} )rH g �?g @c C s t |td�}t|t�st�d S )N)rH )r rx 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)rH )r rf 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)rH )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)rH )r rJ r/ r& rH r )r" r� r} r' r# r# r$ �#test_constructor_dtypes_to_datetime, 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)rH )r rJ r/ r& rH r )r" r� r} r' r# r# r$ �$test_constructor_dtypes_to_timedeltaD s z.TestIndex.test_constructor_dtypes_to_timedeltaz+ignore:Passing keywords other:FutureWarning�attrr> �asi8ru c C s8 t ddd�}t||�}|�|�}|j}|d kr2d nt}|dkrNt|��|�}n |||d�}t�||� |dkr�t�|�� t|�� |�}W d Q R X n |||d�}t�||� |dkr�tt |���|�}n|t |�|d�}t�||� |dk�rt�|�� tt |��� |�}W d Q R X n|t |�|d�}t�||� d S )Nz 2011-01-01rV )rD r� )rE )rH ) r �getattrZ tz_localizerH r- r r+ r8 r, rI r! ) r" Ztz_naive_fixturer� ru r' �argrH �warnrM r# r# r$ � test_constructor_dtypes_datetimeU s. z*TestIndex.test_constructor_dtypes_datetimec C s` t jddd�}|�d �}|j}t||�}|||d�}t�||� |t|�|d�}t�||� d S )Nz1 daysrV )rD )rH )rK �timedelta_rangeZ _with_freqrH r� r+ r8 r! )r" r� ru r' rH r> rM 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>r# c C s& ||�}t ||�st�t|�r"t�d S )N)r/ r& �len)r" r� ru �emptyr# r# r$ �test_constructor_empty� s z TestIndex.test_constructor_emptyz empty,klass)rC c c s | ] }|V qd S )Nr# )r� r� r# r# r$ r� � s )�step�blue�red)�levels�codesc C s t ||�st�t|�rt�d S )N)r/ r& r� )r" r� ru r# r# r$ � test_constructor_empty_special� s z(TestIndex.test_constructor_empty_specialc C s> d}t jt|d��" tt�tj�jd gdd� W d Q R X d S )NzIThe elements provided in the data cannot all be casted to the dtype int64)rl rr rj )rH )rm rn � OverflowErrorr r0 �iinfork �max)r" rp r# r# r$ � test_constructor_overflow_int64� s z)TestIndex.test_constructor_overflow_int64Zperiod�rangeZrepeatsr ZtuplesZuintc C s |� 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�� |�d� W d Q R X d S )Nz(Cannot change data-type for object array)rl r� )rm rn � TypeErrorr� )r" r' rp r# r# r$ �'test_view_with_args_object_array_raises� s z1TestIndex.test_view_with_args_object_array_raisesc C s6 |� d�}|�d� d|_|� d�}|jdks2t�d S )Nr� rV �foobar)rI �get_locr; r& )r" r' Zcastedr# r# r$ � test_astype� s zTestIndex.test_astypec C s& t dddg��t dddg��s"t�d S )Nr\ r] r^ )r �equalsr& )r" r# r# r$ �test_equals_object� s z TestIndex.test_equals_object�comp�dc C s t dddg��|�rt�d S )Nr\ r] r^ )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���d|�}t�||� d S )Nr\ r] r^ �abcrr )r r! �insertr+ r8 )r" � nulls_fixturerS rM r# r# r$ �test_insert_missing� s z TestIndex.test_insert_missingc C s@ t ddddgdd�}d}tjt|d�� |�d � W d Q R X d S ) Nr\ r] r^ r� r' )r; z/index 5 is out of bounds for axis 0 with size 4)rl rV )r rm rn � IndexError�delete)r" r' rp r# r# r$ �test_delete_raises� s z TestIndex.test_delete_raisesc C s� t dddg�}t dddg�}|�|�s*t�|�d�}|�|�sBt�|�|�rPt�|�d�}|�|�sht�t dddg�}t dddgdd �}|�|�r�t�d S ) Nr\ r] r^ �foo)r\ r\ )r\ r] )r] r\ F)� tupleize_cols)r � identicalr& �renamer� )r" �i1�i2Zi3�i4r# r# r$ �test_identical� s zTestIndex.test_identicalc C s0 t td��}|�|�st�|�|�� �� �� �� �s8t�|�t td���rNt�|�|�� �r`t�|�|jdd��rvt�|�|d d � �r�t�|�t�td���r�t�|�|�� �s�t�|�� }d|_|�|�s�t�|�|�s�t�|�t |j ��r�t�t�tdd��}t |dd�}t |dd�}|�|��r,t�d S )N� F)�deepZbobrr � )r< ) r r� �is_r& r� r< r0 r7 r; r> )r" �indZind2r9 Zind1r# r# r$ �test_is_ s$ zTestIndex.test_is_c C sz |d }|� |�|kst�t|� |td� ��s4t�|d }|� |td� �|ksVt�|d �� }t|� |�t�svt�d S )Nr rr ���)�asofr& r r Z to_pydatetimer/ r )r" r' r� r# r# r$ � test_asof' s zTestIndex.test_asofc C sj t dddg�}t ddg�}d}tjt|d�� |�|� W d Q R X tjt|d�� |�|� W d Q R X d S )Nrr r* rB TFz#'<' not supported between instances)rl )r rm rn r� r� )r" �left�rightrp r# r# r$ � test_asof_numeric_vs_bool_raises3 s z*TestIndex.test_asof_numeric_vs_bool_raisesc sj t d�j� t� fdd�td�D ��}ttj�d�|d�}|�|j d �}t dd�}||t |� ksft �d S ) N�20130101c s g | ]}� d | �qS )rF r# )r� �i)rT r# r$ � <listcomp>A s z?TestIndex.test_asof_nanosecond_index_access.<locals>.<listcomp>�d )r' r z"2013-01-01 00:00:00.000000050+0000�ns) r r� r r� r r0 rX Zrandnr� r' r r& )r" �rrs Z first_valueZ expected_tsr# )rT r$ �!test_asof_nanosecond_index_access? s z+TestIndex.test_asof_nanosecond_index_accessc C s� t jt|�td�}d|ddd�< || }x&t|�D ]\}}|�|�|ks2t�q2W |t|� }x&t|�D ]\}}|�|�|ksft�qfW d S )N)rH FrV � r* )r0 Zonesr� r� � enumerater� r& r! )r" r' Z bool_indexZ sub_indexr� �valr# r# r$ �test_booleanindexL s zTestIndex.test_booleanindexc C s: |}|dddg }x"|D ]}|||� |� kst�qW d S )Nrr r* rB )r� r& )r" r% r' �slr� r# r# r$ � test_fancyZ s zTestIndex.test_fancyc C sB t jg |d�}t|�g �}|g �|�s,t�|| �|�s>t�d S )N)rH )r0 r7 �typer� r& )r" r' rH Z empty_arr� empty_indexr# r# r$ �test_empty_fancy` s zTestIndex.test_empty_fancyc C sX t jg t jd�}t|�g �}|g �|�s.t�d}tjt|d�� || W d Q R X d S )N)rH z=arrays used as indices must be of integer \(or boolean\) type)rl ) r0 r7 Zfloat_r� r� r& rm rn r� )r" r' Z empty_farrr� rp r# r# r$ �test_empty_fancy_raisesi s z!TestIndex.test_empty_fancy_raisesc C s~ |}t ddd�}|�|�}|�|�}t�||�d��}t�||�sFt�t�||�sVt�t�||� t�||� t�||� d S )Nz 2019-01-01r� )rD �O) r �unionr0 �appendrI r+ Z equalContentsr& r4 )r" �sortr% r' � date_indexZ first_catZ second_catZappendedr# r# r$ �test_union_dt_as_objv s z TestIndex.test_union_dt_as_objc C sn t �d�}t �d��dd� �}tdd� |D ��}t �||� |�dd� �}t�dd� |D ��}t �||� d S )NrB c S s | fS )Nr# )�xr# r# r$ �<lambda>� � z0TestIndex.test_map_with_tuples.<locals>.<lambda>c S s g | ] }|f�qS r# r# )r� r� r# r# r$ r� � s z2TestIndex.test_map_with_tuples.<locals>.<listcomp>c S s | | dkfS )Nrr r# )r� r# r# r$ r� � r� c S s g | ]}||d kf�qS )rr r# )r� r� r# r# r$ r� � s )r+ � makeIntIndex�mapr r8 r � from_tuples)r" r' rM rS r# r# r$ �test_map_with_tuples� s z TestIndex.test_map_with_tuplesc C sB dddg}t �t|dddg��}|�dd� �}t�|t|�� d S ) Nr� �bar�bazrr r* rB 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+ r8 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 �}|�dd� �}t�||� d S )Nr� rr c S s dS )Nrr r# )r� r# r# r$ r� � r� zATestIndex.test_map_tseries_indices_return_index.<locals>.<lambda>)r� r+ r r� r8 )r" r� r' rS rM r# r# r$ �%test_map_tseries_indices_return_index� s z/TestIndex.test_map_tseries_indices_return_indexc C s: t jdddd�}ttd�dd�}t �||�dd� �� d S )N� �hZhourly)rC r; )r; c S s | j S )N)�hour)r� r# r# r$ r� � r� zGTestIndex.test_map_tseries_indices_accsr_return_index.<locals>.<lambda>)r+ r� r r� r8 r� )r" r� rS r# r# r$ �+test_map_tseries_indices_accsr_return_index� s z5TestIndex.test_map_tseries_indices_accsr_return_index�mapperc C s dd� t | |�D �S )Nc S s i | ]\}}||�qS r# r# )r� �er� r# r# r$ � <dictcomp>� s z&TestIndex.<lambda>.<locals>.<dictcomp>)r� )r> r' r# r# r$ r� � r� zTestIndex.<lambda>c C s t | |�S )N)r )r> r' r# r# r$ r� � r� c C s: t dddg�}t�d�}|�||j|��}t�||� d S )Nr� r� r� rB )r r+ r� r� r> r8 )r" r� rS r' rM r# r# r$ �test_map_dictlike_simple� s z"TestIndex.test_map_dictlike_simplec C s dd� t | |�D �S )Nc S s i | ]\}}||�qS r# r# )r� r� r� r# r# r$ r� � s z&TestIndex.<lambda>.<locals>.<dictcomp>)r� )r> r' r# r# r$ r� � r� c C s t | |�S )N)r )r> r' r# r# r$ r� � r� c C s^ t |t�rd S |jsd S |jr(tg �}ntt�t|�dd��}|�|||��}t � ||� d S )Nr r� ) r/ r � is_uniquer� r r0 �aranger� r� r+ r8 )r" r' r� rS rM r# r# r$ �test_map_dictlike� s zTestIndex.test_map_dictliker� r� r r� )r' )r r* r� c C s4 t dtjdg�}t dddg��|�}t�||� d S )Ng @r� r* rr r )r r0 rg r� r+ r8 )r" r� rS rM 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 }t�||� d S ) Ng �?rB rV c S s | d S )Nr* r# )r� r# r# r$ r� � r� z1TestIndex.test_map_na_exclusion.<locals>.<lambda>�ignore)Z na_actionr* )r r0 rg r� r+ r8 )r" r' rM rS r# r# r$ �test_map_na_exclusion� s z TestIndex.test_map_na_exclusionc C sJ t dddg�}tdd� �}d|d< |�|�}t dddg�}t�||� d S )Nrr r* rB c S s dS )N�blankr# r# r# r# r$ r� � r� z0TestIndex.test_map_defaultdict.<locals>.<lambda>�stuffr� )r r r� r+ r8 )r" r' Z default_dictrM rS r# r# r$ �test_map_defaultdict� s z TestIndex.test_map_defaultdictz name,expected)r� r� )r� Nc C s: t g dd�}t dddg|d�}|�|�}|j|ks6t�d S )Nr� )r; rr r* rB )r r� r; r& )r" r; rS r� r� rM r# r# r$ � test_append_empty_preserve_name� s z)TestIndex.test_append_empty_preserve_namec C s& |}t �t�� |�� W d Q R X d S )N)r+ r, r- Zis_mixed)r" r% r' r# r# r$ �test_is_mixed_deprecated� s z"TestIndex.test_is_mixed_deprecatedzindex, expected)r6 F)r� F)r� F)rx T)r F)rf Tc C s |� � |kst�d S )N)Z is_numericr& )r" r' rS r# r# r$ �test_is_numeric� s zTestIndex.test_is_numeric)r6 T)r� T)rx F)rf Fc C s |� � |kst�d S )N)Z is_objectr& )r" r' rS r# r# r$ �test_is_object s zTestIndex.test_is_object)r Tc C s( t �t�� |j|kst�W d Q R X d S )N)r+ r, r- Z is_all_datesr& )r" r' rS r# r# r$ �test_is_all_dates s zTestIndex.test_is_all_datesc C s |� � d S )N)�_summary)r" r' r# r# r$ � test_summary( s zTestIndex.test_summaryc C s4 t ddgdd�}|�� }d|ks$t�d|ks0t�d S )Nz {other}%sz ~:{range}:0ry )r; )r r r& )r" r� rM r# r# r$ �test_summary_bug+ s zTestIndex.test_summary_bugc C s( t ddg�}ddg}|�� |ks$t�d S )NZ aaaaaaaaar] )r �formatr& )r" �idxrS r# r# r$ �$test_format_different_scalar_lengths3 s z.TestIndex.test_format_different_scalar_lengthsc C sR t �� }t|��d�sBt|g�}|�� }t|d �g}||ksBt�tg ��� d S )NZ000r )r �now�str�endswithr r r& )r" r r' � formattedrS r# r# r$ �test_format_bug9 s zTestIndex.test_format_bugy @ @g @c C sf t |�}|�|� t|�}|�� }t|d �t|d �t|d �dg}||ksRt�|d |ksbt�d S )Nr rr r* �NaNrB )r! r� r r r r& )r" r} r� r' r rS r# r# r$ �test_format_missingG s $ z TestIndex.test_format_missingc C s. t dddd�}|jdd�}|d dks*t�d S )Nz2011-01-01 04:00:00r� Z something)rD r; T)r; r )r r r& )r" �datesr r# r# r$ � test_format_with_name_time_infoT s z)TestIndex.test_format_with_name_time_infoc C sN t tddd�tdddd�g�}|�� }ddg}t|�dks>t�||ksJt�d S )Ni� r* � � z2012-02-07 00:00:00z2012-02-07 23:00:00)r r r r� r& )r" �trM rS r# r# r$ � test_format_datetime_with_time[ s z(TestIndex.test_format_datetime_with_time�op�any�allc C s& |}t ||�� t |j|�� ks"t�d S )N)r� r> r& )r" r r% r' r# r# r$ �test_logical_compatc s z TestIndex.test_logical_compatc C sx t |�}|ttdd�� }|�|�}|ttd��ttd|�� }t�||� |�|d �}|dd � }t�||� d S )NrV r� r rr )r� r! r� �dropr+ r8 )r" r' �nr �droppedrS r# r# r$ �test_drop_by_str_labelh s z TestIndex.test_drop_by_str_label�keysr� �1c C s( t jtdd�� |�|� W d Q R X d S )N� )rl )rm rn �KeyErrorr )r" r' r! r# r# r$ �*test_drop_by_str_label_raises_missing_keysu s z4TestIndex.test_drop_by_str_label_raises_missing_keysc C s� t |�}|ttdd�� }|�� dg }|j|dd�}|ttd��ttd|�� }t�||� |jddgdd�}|tt|�� }t�||� d S )NrV r� r� r� )�errorsr� )r� r! r� �tolistr r+ r8 )r" r' r r �mixedr rS r# r# r$ �$test_drop_by_str_label_errors_ignore{ s z.TestIndex.test_drop_by_str_label_errors_ignorec C s4 t dddg�}|�d�}t ddg�}t�||� d S )Nrr r* rB )r r r+ r8 )r" r' r rS r# r# r$ � test_drop_by_numeric_label_loc� s z(TestIndex.test_drop_by_numeric_label_locc C s: t dddg�}tjtdd�� |�ddg� W d Q R X d S )Nrr r* rB r# )rl rw )r rm rn r$ r )r" r' r# r# r$ �.test_drop_by_numeric_label_raises_missing_keys� s z8TestIndex.test_drop_by_numeric_label_raises_missing_keysz key,expectedc C s, t dddg�}|j|dd�}t�||� d S )Nrr r* rB r� )r& )r r r+ r8 )r" r( rS r' r r# r# r$ �(test_drop_by_numeric_label_errors_ignore� s z2TestIndex.test_drop_by_numeric_label_errors_ignore)r^ r� �to_dropc C s� t |�}t dg�}|�|�}t�||� |�|d �}x0|d |d gfD ]}|�|�}t�||� qJW |�|d �}dt�|d �� �� d�}x>|d |d gfD ](}tjt |d�� |�|� W d Q R X q�W d S )Nr] r rr z\"\[z\] not found in axis\")rl ) r r r+ r8 �re�escape�__repr__rm rn r$ ) r" r> r- r' rS rM �removedZdrop_merp r# r# r$ �test_drop_tuple� s zTestIndex.test_drop_tuplec C sj t |�dkst|t�rd S t|t�r2ts2t�d� |�� �d�}|dd � }|� |d �}t � ||� d S )Nr z=Cannot test IntervalIndex with int64 dtype on 32 bit platformr* ) r� r/ r r r rm �skip�unique�repeatr r+ r8 )r" r' rS rM r# r# r$ �"test_drop_with_duplicates_in_index� s z,TestIndex.test_drop_with_duplicates_in_index�is_monotonic_increasing�is_monotonic_decreasing�!_is_strictly_monotonic_increasing�!_is_strictly_monotonic_decreasingc C s$ t dt�� dg�}t||�r t�d S )NrV r )r r r r� r& )r" r� r' r# r# r$ � test_is_monotonic_incomparable� s z(TestIndex.test_is_monotonic_incomparablec C sP |}t �dddg�}t�t�� |�||d d� W d Q R X |d dksLt�d S )Nrr r* rB �P )r0 r7 r+ r, r- � set_valuer& )r" r% r r9 r# r# r$ �test_set_value_deprecated� s z#TestIndex.test_set_value_deprecatedZquuxzindex,expectedZquxc C s |� |�}t�||� d S )N)�isinr+ r= )r" r> r' rS rM r# r# r$ � test_isin� s zTestIndex.test_isinc C s� t |t�rPt |t�rPt�|�rPt�|�rPt�td|g��|g�t� ddg�� nV||kr�t�td|g��|g�t� ddg�� n&t�td|g��|g�t� ddg�� d S )Nr\ FT) r/ rf �math�isnanr+ r= r r? r0 r7 )r" �requestr� Znulls_fixture2r# r# r$ �test_isin_nan_common_object� s z%TestIndex.test_isin_nan_common_objectc C s� |t jkrt�d� |t jkr4|j�tjjdd�� t � t d|g�� t jg�t �ddg�� t � t d|g�� t jg�t �ddg�� d S )Nz'pd.NaT not compatible with Float64Indexz!Float64Index cannot contain pd.NA)r� g �?FT)rK �NaTrm r3 ZNA�nodeZ add_marker�mark�xfailr+ r= r r? r0 rg r7 )r" rC r� r# r# r$ � test_isin_nan_common_float64� s $z&TestIndex.test_isin_nan_common_float64�levelc C s^ |� � dd � dg }t�ddddg�}t�||j||d�� d|_t�||j|dd�� d S )N���Z nonexistingFT)rJ r� )r' r0 r7 r+ r= r? r; )r" rJ r' r> rS 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� rr zToo many levels)rl )rJ )Znlevelsrm rn r� r? )r" r' rJ 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|�ddg|jdd � �}d|� d�}n|�d�}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\))rl )rJ )r/ r r� �namesrm rn r$ r? )r" rN r' rp r# r# r$ �&test_isin_level_kwarg_bad_label_raises) s z0TestIndex.test_isin_level_kwarg_bad_label_raisesc C s4 t ddg�}t�ddg�}|�|�}t�||� d S )Nr\ r] F)r r0 r7 r? r+ r= )r" r� r' rS rM r# r# r$ �test_isin_empty4 s zTestIndex.test_isin_emptyz 2018-01-01c C s6 t |�}||k}tjddddgtd�}t�||� d S )NT)rH )r r0 r7 r� r+ r= )r" r> r' rM rS r# r# r$ �test_boolean_cmp= s zTestIndex.test_boolean_cmpz name,level)Nr )r\ r\ c C s, |� � }|r||_|�|�}t�||� d S )N)r< r; Zget_level_valuesr+ r8 )r" r' r; rJ rS rM r# r# r$ �test_get_level_valuesN s z TestIndex.test_get_level_valuesc C s, t ddgdd�}|j|dd � jks(t�d S )Nr\ r] Zasdf)r; rr )r r; r& )r" r' r# r# r$ �test_slice_keep_nameX 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_self\ s zTestIndex.test_join_self�method�strip�rstrip�lstripc sF t ddddg�}t � fdd�|jD ��}t|j� �� }t�||� d S )Nz jackzjill z jesse Zfrankc s g | ]}t t� �|��qS r# )r� r )r� r� )rX r# r$ r� i s z0TestIndex.test_str_attribute.<locals>.<listcomp>)r r> r� r r+ r8 )r" rX r' rS rM r# )rX r$ �test_str_attributee s z TestIndex.test_str_attributer� )r� r" )r� �3�2000Z2010)�start�endrC c C s* t jtdd�� |j�d� W d Q R X d S )Nzonly use .str accessor)rl r* )rm rn �AttributeErrorr r5 )r" r' r# r# r$ �test_str_attribute_raisesn s z#TestIndex.test_str_attribute_raiseszexpand,expectedNr� �f)r\ r] r^ c C s@ t dddg�}|d k r&|jj|d�}n |j�� }t�||� d S )Nza b czd erc )�expand)r r �splitr+ r8 )r" rd rS r' rM r# r# r$ �test_str_split{ s zTestIndex.test_str_splitc C sN t ddddg�}|j�d�}t�ddddg�}t�||� t|tj�sJt �d S )N�a1�a2�b1�b2r\ TF) r r � startswithr0 r7 r+ r= r/ r1 r& )r" r' rM rS r# r# r$ �test_str_bool_return� s z TestIndex.test_str_bool_returnc C sV t ddddg�}ttd�|d�}||jj�d� }ttd�ddgd�}t�||� d S ) Nrg rh ri rj rw )r' r\ r* )r r r� r' r rk r+ rY )r" r' rT rM rS r# r# r$ � test_str_bool_series_indexing� s z'TestIndex.test_str_bool_series_indexing�abcdc C s dt |�k}||kst�d S )Nr )�dirr& )r" r' rS rM r# r# r$ �test_tab_completion� s z TestIndex.test_tab_completionc C s` t ddddddg�}|dd� �t ddgtjd��s8t�|ddg �t ddgtjd��s\t�d S ) Nrr r* rB r\ r] r^ )rH r )r r� r0 rL r& )r" r' r# r# r$ �!test_indexing_doesnt_change_class� s $z+TestIndex.test_indexing_doesnt_change_classc C sz t tj�d��}t�d�}t�t�� |j|dd�}W d Q R X t�t�� |� t �� |� t ��}W d Q R X t� ||� d S )N� r� �outer)rU ) r r0 rX Z permutationr+ r� r, �RuntimeWarningrV rI rJ r� r8 )r" Z left_indexZ right_indexrM rS r# r# r$ �test_outer_join_sort� s z TestIndex.test_outer_join_sortc C sR t tjtd�td�g�}|�dddg�}t |d |d |d g�}t�||� d S )Nr� Z20130102r� r rr )r rK rE r �taker+ r8 )r" r' rM rS r# r# r$ � test_nan_first_take_datetime� s z&TestIndex.test_nan_first_take_datetimec C s� t td�dd�}|�t�dddg��}t td�dd�}t�||� |jt�dddg�dd �}t d d tjgdd�}t�||� |jt�dddg�d dd �}t d d dgdd�}t�||� d S )N�ABC�xxx)r; rr r r� ZBACT)� fill_valuerz ry F)Z allow_fillrz r{ )r r! rv r0 r7 r+ r8 rg )r" r' rM rS r# r# r$ �test_take_fill_value� s z TestIndex.test_take_fill_valuec C s� t td�dd�}d}tjt|d�� |jt�dddg�d d � W d Q R X tjt|d�� |jt�ddd g�d d � W d Q R X d S ) Nrx ry )r; zJWhen allow_fill=True and fill_value is not None, all indices must be >= -1)rl rr r rK T)rz ���)r r! rm rn ro rv r0 r7 )r" r' rp r# r# r$ � test_take_fill_value_none_raises� s $z*TestIndex.test_take_fill_value_none_raisesc C sB t td�dd�}tjtdd�� |�t�ddg�� W d Q R X d S )Nrx ry )r; z out of bounds)rl rr r| )r r! rm rn r� rv r0 r7 )r" r' r# r# r$ �test_take_bad_bounds_raises� s z%TestIndex.test_take_bad_bounds_raisesr; �labelsr� c C s0 t dddg�}||_|�|�d j|ks,t�d S )Nr rr r* )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 td��}|�|�d jjtjks(t�d S )Nr� r )r r! r� rH r� r0 rL 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 td��}|�|�d jj|ks&t�d S )Nr� r )r r! r� rH r� r& )r" r rH 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 sd t td��}|�ttg �tg �gg g g��d }|jd jjt j ksHt �|jd jjt j ks`t �d S )Nr� r rr ) r r! r� r r r r� rH r� r0 rj r& �float64)r" r' rM r# r# r$ �-test_reindex_no_type_preserve_target_empty_mi s z7TestIndex.test_reindex_no_type_preserve_target_empty_mic C sR t td��}|�t�dddddg��}t ddg�t dddg�d�}t�||� d S )NrV rr r* r rB rw )rr r* )r r� �groupbyr0 r7 r+ Zassert_dict_equal)r" r' rM rS r# r# r$ � test_groupby s zTestIndex.test_groupbyz mi,expected)rr r* )rw rV )rw � c C s0 t jtd�ddgd�}|j|k}t�||� d S )Nza,b,c 1,2,3 4,5,6r rr )� index_col)rK �read_csvr r' r+ r= )r" �mirS rZ rM r# r# r$ �test_equals_op_multiindex s z#TestIndex.test_equals_op_multiindexc C s@ t jtd�ddgd�}|j|jk}t�ddg�}t�||� d S )Nza,b,c 1,2,3 4,5,6r rr )r� T)rK r� r r' r0 r7 r+ r= )r" rZ rM rS r# r# r$ �"test_equals_op_multiindex_identify, s z,TestIndex.test_equals_op_multiindex_identify)� � c C s> t jtd�ddgd�}tjtdd�� |j|k W d Q R X d S )Nza,b,c 1,2,3 4,5,6r rr )r� zLengths must match)rl )rK r� r rm rn ro r' )r" r' rZ r# r# r$ �+test_equals_op_mismatched_multiindex_raises3 s z5TestIndex.test_equals_op_mismatched_multiindex_raisesc C sF t �dddg�}tdddg�}||k}t�dddg�}t�||� d S )N)rr r* )rw rV )r� r� r� r� r� F)r r� r r0 r7 r+ r= )r" r� r' rM rS 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 ddgdd�}|j||�jks$t�d S )Nz01:02:03z01:02:04rN )r; )r r; r& )r" r� r' r# r# r$ �!test_dt_conversion_preserves_nameH s z+TestIndex.test_dt_conversion_preserves_namec C s2 t dddg�}tjtdd�� d|_W d Q R X d S )Nrr r* rB zCan't set attribute)rl F)r rm rn ra r� )r" r' r# r# r$ �#test_cached_properties_not_settableN s z-TestIndex.test_cached_properties_not_settablec � sn t jddd� ddlm} d}|�|�I d H t�d ��, |d�� t|j� dd �� 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])r� zidx.rw ) rm Z importorskipZIPython.core.completerr� Zrun_coder+ r, r! Z CompleterZ completions)r" �ipr� �coder# r# r$ �test_tab_complete_warningS s z#TestIndex.test_tab_complete_warningc C sP t |t�r|�d� n6dt|�j� d�}tjt|d�� |�d� W d Q R X d S )Nrr �'z$' object has no attribute 'contains')rl )r/ r �containsr� ra rm rn ra )r" r' rp r# r# r$ � test_contains_method_removedb s z&TestIndex.test_contains_method_removedc C s t dddddg�}tjtdd�� |jdd � W d Q R X tjtd d�� |jd d gd � W d Q R X tjtd d�� |jdgd � W d Q R X t dddddg�}|jd gd �}t�|d |� t dddddg�}|jd d �}t�|d |� t dddddg�}|jdd �}t�|d |� d S )NrV rw rB r* rr z(ascending must be a single bool value or)rl �True)Z ascendingz3ascending must be a list of bool values of length 1Tz ascending must be a bool valuer F)r rm rn � ExceptionZ sortlevelr+ r8 )r" r' rS rM r# r# r$ �test_sortlevelk s" zTestIndex.test_sortlevel)�ra rb rc r r rm �fixturer% r) rG � parametrizer3 r5 r: r? r rK r� r rN rO rU r[ r0 r� r7 r> re ri rq rt r r� rg r rE rv r~ rx rf r r� r r r� r� Z timedelta64r r� �filterwarningsr� r r� �iterr r r r� r r r r� r� r� �paramrH 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� r� r r r r r r r r r r r r r r r r% r) r* r+ r, r2 r6 r; r> r� r@ rD rI rL rM rP rJ rQ rR rS rT rW r\ r� r+ r� r� rb rf rl rm r! rp rq ru rw r{ r} r~ r' r� rj r� Z datetime64r� r� r� r� r� r� r� rQ � to_timedeltar� r� r r� r� r� r# r# r# r$ r 0 sZ $ * ) , . . & $$ :". "( " * , & ( $" 6 4 $ 4 0 $ r c @ s� e Zd ZeZejed�dd��Zejdddddd ggd gd �d d � �Zdd� Z dd� Z dd� Z dd� Z dd� Z dd� Zej�ddd g�ej�d d ed g�ej�d 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)� ���Zej�ddd g�ej�d*ed+d,d-g�ed+d,d-g�fed+d,d-ejg�ed+d,d-g�fed.d/d0g�ed.d/d0g�feejd.d/d0ejg�ed.d/d0g�fed1d2d3gd4d5�ed1d2d3gd4d5�fed1d2d6d3gd4d5�ed1d2d3gd4d5�fg�d7d8� ��Zd9d:� Zej�d;eejg�eejdg�eddejg�eddejg�e� d6g�e� d6d<g�e� d<d6d=g�e� d>d6g�g�d?d@� �Z ej�dAe e g�dBdC� �Z!dDdE� Z"ej�#dF�dGdH� �Z$d S )I�TestMixedIntIndex)r c C s | � ddddddg�S )Nr r\ rr r] r* r^ )r )r" r# r# r$ r% � s z TestMixedIntIndex.simple_indexr r\ rr r] r* r^ Z mixedIndex)�paramsZidsc C s t |j�S )N)r r� )r" rC r# r# r$ r' � s zTestMixedIntIndex.indexc C s* |}t jtdd�� |�� W d Q R X d S )Nz'>|<' not supported)rl )rm rn r� �argsort)r" r% r' r# r# r$ � test_argsort� s z TestMixedIntIndex.test_argsortc C s, |}t jtdd�� t�|� W d Q R X d S )Nz'>|<' not supported)rl )rm rn r� r0 r� )r" r% r' r# r# r$ �test_numpy_argsort� s z$TestMixedIntIndex.test_numpy_argsortc C s� |}t |�|ddd�}t |�|dd�}||k s2t�t�||� |jdksLt�|jdksZt�td|d�}td|d d � d�}|| }|jjdks�t�d S ) NTZmario)r< r; F)r< r* )r' rB r� )r� r& r+ r8 r; r r' )r" r% r' �first�second�s1�s2Zs3r# r# r$ �test_copy_name� s z TestMixedIntIndex.test_copy_namec C s� t ddgdd�}|�� }t�||� |jdd�}tj||dd� |jdksNt�|jdks\t�|jdgd�}tj||dd� |jdks�t�|jdgks�t�|jdks�t�|jdgks�t�d S ) Nrr r* ZMyName)r; ZNewNameF)Z check_names)rO )r r< r+ r8 r; r& rO )r" r' Zindex1�index2Zindex3r# r# r$ �test_copy_name2� s z!TestMixedIntIndex.test_copy_name2c C sB t dtjddgdd�}t dtjdgdd�}|�� }t�||� d S )Nr* rr Zmy_index)r; )r r0 rg r4 r+ r8 )r" r rS rM r# r# r$ �test_unique_na� s z TestMixedIntIndex.test_unique_nac C s4 |}|� � |j� � kst�|�� |j�� ks0t�d S )N)r r> r& r )r" r% r' r# r# r$ r � s z%TestMixedIntIndex.test_logical_compatrU r r rH Nr� z vals,expectedrB g �?g @g @ry rz r{ c C s4 t ||d�}|j|d�}t ||d�}t�||� d S )N)rH )rU )r �dropnar+ r8 )r" rU rH r} rS r' rM 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)rC rE c C s |j |d�}t�||� d S )N)rU )r� r+ r8 )r" rU r' rS rM r# r# r$ �test_dropna_dt_like� s z%TestMixedIntIndex.test_dropna_dt_likec C s8 d}t jt|d�� tdddg�jdd� W d Q R X d S )Nzinvalid how option: xxx)rl rr r* rB ry )rU )rm rn ro r r� )r" rp r# r# r$ � test_dropna_invalid_how_raises s z0TestMixedIntIndex.test_dropna_invalid_how_raisesr' z 2000-01-01z 2000-01-02z1 dayc C s< |j dkst�|jdks t�|jdks*t�|jdks8t�d S )NF)r7 r& r8 r9 r: )r" r' r# r# r$ �test_is_monotonic_na s z&TestMixedIntIndex.test_is_monotonic_naru c C s: t dddgdd�}|ttd��|d�}dt|�ks6t�d S ) Nr\ r] r^ r )r; rB )r' �0)r r! r� �reprr& )r" ru r' rM r# r# r$ �test_int_name_format 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# )r )r� r� r# r# r$ r� ' s z>TestMixedIntIndex.test_str_to_bytes_raises.<locals>.<listcomp>r� z2^'str' object cannot be interpreted as an integer$)rl )r r� rm rn r� �bytes)r" r' rp r# r# r$ �test_str_to_bytes_raises% s z*TestMixedIntIndex.test_str_to_bytes_raisesz2ignore:elementwise comparison failed:FutureWarningc C s: t dddgdd�}|dk}t�dddg�}t�||� d S )N)r\ r] )r] r^ )r^ r\ F)r� T)r r0 r7 r+ r= )r" r rM rS r# r# r$ �test_index_with_tuple_bool, s z,TestMixedIntIndex.test_index_with_tuple_bool)%ra rb rc r r rm r� r% r' r� r� r� r� r� r rG r� rJ r0 rg r� r rK rE r r r� r� rQ r� r� r r r� r� r� r� r# r# r# r$ r� � s` &( " r� c @ s� e Zd Zej�ddddggdedddg�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 dd� Z dd� Z dS )�TestIndexUtilszdata, names, expectedrr r* rB Nr; )r; r\ r^ r� r ZL1ZL2)rO c C s t ||�}t�||� d S )N)r r+ r8 )r" rh rO rS rM r# r# r$ � test_ensure_index_from_sequences8 s z/TestIndexUtils.test_ensure_index_from_sequencesc C s` t jdddd�t jdddd�t jdddd�t jdd d d�g}t|�}t|td �}t�||� d S ) Nr rr r� )�closedr* r� rB Zneitherrw Zboth)rH )rK ZIntervalr r rJ r+ r8 )r" Z intervalsrM rS r# r# r$ �(test_ensure_index_mixed_closed_intervalsM s z7TestIndexUtils.test_ensure_index_mixed_closed_intervalsc C sF dt �t j�jg}t|�}t|�|ks*t�t|dd�}t� ||� d S )Nr rk )rH ) r0 r� rk r� r r! r& r r+ r8 )r" r> rM rS r# r# r$ �test_ensure_index_uint64Y s z'TestIndexUtils.test_ensure_index_uint64c C s t g �}tg �}t�||� d S )N)r r r+ r8 )r" rM rS r# r# r$ �test_get_combined_indexe s z&TestIndexUtils.test_get_combined_index) ra rb rc rm rG r� r r r� r� r� r� r# r# r# r$ r� 7 s 6 r� �opname�eq�ne�le�lt�ge�gt�addZradd�subZrsub�mulZrmul�truedivZrtruediv�floordivZ rfloordiv�powZrpow�mod�divmodc C s( d| � d�} t || �}|j| ks$t�d S )N�__)r� ra r& )r� r' rX r# r# r$ �test_generated_op_namesk s r� � index_makerc C s( t jtdd�� | dd� W d Q R X d S )Nzunexpected keyword argument)rl r� )r� )rm rn r� )r� r# r# r$ �,test_index_subclass_constructor_wrong_kwargs� s r� z+ignore:Passing keywords other:FutureWarningc C s� d} t jt| d��" ttjddgtd�ddd� W d Q R X t jt| d��$ ttjd d d gd d�ddd� W d Q R X t jt| d�� td dd ddd� W d Q R X t jt| d�� t dddgddd� W d Q R X d S )Nz [Uu]nexpected keyword argument)rl r\ r] )rH �testT)r; Zfastpathrr r* rB rj r rV r^ ) rm rn r� r r0 r7 rJ r r r )rp r# r# r$ �test_deprecated_fastpath� s &( r� c C sD t ddddg�} t�t�� | d d �d f jdks6t�W d Q R X d S )Nr rr r* rB )rw rr )r r+ r, r- �shaper& )r r# r# r$ �test_shape_of_invalid_index� s r� c C s, d} t �d��ddd�}tjt| d�� t|� W d Q R X tjt| d�� t|�t j �� W d Q R X tjt| d�� t |�t j �� W d Q R X tjt| d�� t |�t j �� W d Q R X t|�dd��}tjt| d�� t|� W d Q R X tdtd��}tjt| d�� t �ddggd �|_W d Q R X d S )Nz Index data must be 1-dimensionalr� r* )rl rw r rB )r0 r� Zreshaperm rn ro r r rI r� r rj r rk r r r� r7 r' )rp r9 rZ rs r# r# r$ �test_validate_1d_input� s r� zklass, extra_kwargsrC �Yc C s: | t t�dd��f|�}| tdd�f|�}t�||� d S )Ni� i� )� memoryviewr0 r� r� r+ r8 )ru � extra_kwargsrM rS r# r# r$ � test_construct_from_memoryview� s r� c C s^ t ddddg�} d}tjt|d�� | �dd �}W d Q R X t ddddgdd�}t�||� d S ) Nrr r* rB rw zsIn a future version of pandas all arguments of Index.set_names except for the argument 'names' will be keyword-only)rl Zquarter)r; )r r+ r, r- Z set_namesr8 )r rp rM rS r# r# r$ �)test_index_set_names_pos_args_deprecation� s r� c C s` t ddddg�} d}tjt|d�� | �d� | �d�}W d Q R X t dddg�}t�||� d S )Nrr r* rB zYIn a future version of pandas all arguments of Index.drop_duplicates will be keyword-only)rl �last)r r+ r, r- Zdrop_duplicatesr8 )r rp rM rS r# r# r$ �)test_drop_duplicates_pos_args_deprecation� s r� );� collectionsr r r �ior rA r. �numpyr0 rm Z pandas.compatr r Z pandas.util._test_decoratorsr ZpandasrK r r 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� rG r� r� Z index_subclass_makers_generatorr� r� r� r� r� r� r� r� r# r# r# r$ �<module> s~ D _ 34
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка