3
���h� � @ s` d Z ddlZddlZddlmZmZmZmZm Z m
Z
ddl j Z
dd� Zdd� ZG dd � d �ZdS )
z common utilities � N)� DataFrame� Float64Index�
MultiIndex�Series� UInt64Index�
date_rangec s � fdd�t |�D �S )Nc s g | ]}� � |� ��qS � r )�.0�i)�prefixr �>/tmp/pip-build-5_djhm0z/pandas/pandas/tests/indexing/common.py�
<listcomp> s z_mklbl.<locals>.<listcomp>)�range)r �nr )r r �_mklbl
s r c C s t d �g| j }|||< t|�S )N)�slice�ndim�tuple)�obj�key�axis�axesr r r �_axify s r c @ s^ e Zd ZdZddhZdddddd d
d d h Zd
d� Zddd�Zddd�Zddd�Z d dd�Z
dS ) �Basez# indexing comprehensive base class Zseries�frame�intsZuints�labels�mixed�tsZfloats�emptyZts_revZmultic C s� t tjjd�tjddd�d�| _ttjjdd�tjddd�tjddd�d�| _t tjjd�t tjddd��d�| _
ttjjdd�t t ddd��t t ddd��d�| _ t tjjd�t
t ddd��d�| _ttjjdd�t
t ddd��t
t ddd��d�| _tjd dgddgg�tjd
d gd dgg�tjd
dgddgg�g}t tjjd�|d d�| _ttjjdd�|d |d d�| _t tjjd�td�d�| _ttjjdd�td�td�d�| _t tjjd�ddddgd�| _ttjjdd�ddddgd�| _t tjjd�tddd�d�| _ttjjdd�tddd�d�| _tddd�j dd�}t tjjd�|d�| _ ttjjdd�|d�| _ t� | _ t t d�| _!xL| j"D ]B}t#� }x(| j$D ] }t%| |� d|� ��||< �q�W t&| ||� �q�W d S )N� r � � )�index� � )r# �columns� � � � � �
� Zabcd�ABCD�nullZ20130101)ZperiodsF)Z ascending)Zdtype�_)'r �np�randomZrandZarangeZ series_intsr ZrandnZ
frame_intsr Z series_uintsr Z frame_uintsr Z
series_floatsZ frame_floatsr Z from_productZ series_multiZ frame_multi�listZ
series_labelsZ frame_labelsZ series_mixedZ frame_mixedr Z series_tsZframe_tsZ sort_valuesZ
series_ts_revZ frame_ts_revZ frame_empty�objectZ series_empty�_kinds�dict�_typs�getattr�setattr)�self�methodZm_idcesZ dates_rev�kind�d�typr r r � setup_method% sP ," zBase.setup_methodFc C s" |j }|rdd� |D �}tj|� S )zv
generate the indices
if values is True , use the axis values
is False, use the range
c s s | ]}t tt|���V qd S )N)r3 r �len)r �axr r r � <genexpr>o s z(Base.generate_indices.<locals>.<genexpr>)r � itertools�product)r: �f�valuesr r r r �generate_indicesg s zBase.generate_indicesc C s: |r|j | S |dkr |j| S |dks,t�|j| S dS )z% return the value for the location i ZiatZatN)rF Ziloc�AssertionError�loc)r: �namerE r
rF r r r � get_values s
zBase.get_valuec
C sx |d kr d S |j }tj|� }xV|D ]N}t||�| }|rD|j| }n |}xt|�D ]} |j| �}qRW tj||� q"W d S )N) r rC rD r8 rF �reversed� __getitem__�tmZassert_almost_equal)
r: rE �funcrF r Zindiciesr
�result�expected�ar r r � check_values s
zBase.check_valuesNc s� � �fdd�}|d kr | j }|d kr.ddg}n|dks:t�|g}xb| jD ]X}t| |�}xH|D ]@} x:|D ]2}
|
| j ksxt�||
} | | jk rf|| | |d� qfW q\W qHW d S )Nc sf t ||| �}yt|��j|� W n@ tttfk
r` } z � dk rNt|� �rNdS � W Y dd}~X nX dS )z compare equal for these 2 keys N)r r8 rM �
IndexError� TypeError�KeyError�
isinstance)r r r ZaxifiedZdetail)�failsr; r r �_eq� s
z Base.check_result.<locals>._eqr r' )r r r )r r' )r7 rH r5 r8 r ) r: r; r Ztypsr rX rY r< r= rA r> r r )rX r; r � check_result� s
zBase.check_result)F)F)F)NNN) �__name__�
__module__� __qualname__�__doc__r5 r7 r? rG rK rS rZ r r r r r s B
r )r^ rC Znumpyr1 Zpandasr r r r r r Zpandas._testingZ_testingrN r r r r r r r �<module> s
|