3
���h| � @ s` d dl Zd dlZd dlmZmZ d dlmZ ddl m
Z
m Z G dd� de
�Z G dd � d e
�Z
dS )
� N)�Dataset�Variable)�indexing� )�TestCase�
ReturnItemc @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )� TestIndexersc C s |j � }d||< |S )Nr )�copy)�self�x�i� r
�9/tmp/pip-build-5_djhm0z/xray/xarray/test/test_indexing.py� set_to_zero
s zTestIndexers.set_to_zeroc C s� t jjddddd�}t jd�}t� }x�|d d � |d |dd d �df |d |d d�ddf |d || |||f |d||f |d g
D ]B}tj||j�}| j|| || � | j| j ||�| j ||�� q�W | j
t d
�� tj|d d
� W d Q R X d S )N�
� � �
� � .r r � � � ztoo many indices).r ).r .).r r r r r )r r r ) �np�random�randn�aranger r Zexpanded_indexer�ndim�assertArrayEqualr �assertRaisesRegexp�
IndexError)r
r �y�Ir �jr
r
r �test_expanded_indexer s
8"z"TestIndexers.test_expanded_indexerc C s* t jjddddd�}t jd�}t� }�xX|d d � |d |d |d d� |dd � |dd� |d d� |d d d� |d d d� |dd d� |d d d� |ddd� |d d d� |d d �d d �f |d d �dd d �f |d d �dd d �df || |d d �|f |d|f |d d�d d �|f |d|d d �d d �df gD ]D}tj||j�}| j|| || � | j| j ||�| j ||�� �q8W t jdd
d�}�x4|||f |d d�d d�f df|||f |d d�dd
d�f df|d||f |dd d�d d�f df||d|f |d d�ddd
d�f d f||d d �|f |d d�d d �dd
d�f d f|dd d �|f |dd d �dd
d�f d f|dd d�||df |dd d�d d�d d�df d f|dd d �|d d �df |dd d �d d�d d �df d f|d d �d d �|d d �df |d d �d d �d d�d d �df d!f|d d �d d �||d d �f |d d �d d �d d�dd
d�f d"fg
D ]\\}}}tj||j�}| j
||| j� | j|| || � | j| j ||�| j ||�� �qfW | j t d�� tj|dk|j� W d Q R X | j t d��$ t
tjd| d| f|j�� W d Q R X d S )#Nr r r r r r r r r r � r � zonly supports 1dzinvalid subkeyg �?)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 r )r r r )r r r )r r r r )r r r r r )r r r r r r Zorthogonal_indexer�shaper r � assertEqualr �
ValueError�print) r
r r! r" r r# �zr* �kr
r
r �test_orthogonal_indexer s@
NNH4 "$&.*48@N z$TestIndexers.test_orthogonal_indexerc C sx t jd�}t|tj�st�|jdks(t�|j� dks8t�t jddg�}|jd ksTt�|d d
ksdt�|d d kstt�d S ) N�ar r r )r1 r )r1 r )r )r )r )r )r ) r Z_asarray_tuplesafe�
isinstancer Zndarray�AssertionErrorr �itemr* )r
�resr
r
r �test_asarray_tuplesafeF s
z#TestIndexers.test_asarray_tuplesafec C s\ t jdddg�}| jtd�� tj|dg� W d Q R X | jt�� tj|d� W d Q R X d S )Nr r r znot all values foundr )�pd�Indexr �KeyErrorr �convert_label_indexer� assertRaises)r
�indexr
r
r �test_convert_label_indexerQ s
z'TestIndexers.test_convert_label_indexerc C s< t jdddg�}| jt�� tj|tdd�� W d Q R X d S )NZ2001�2000Z2002)r7 Z to_datetimer; r9 r r: �slice)r
r< r
r
r �+test_convert_unsorted_datetime_index_raisesY s z8TestIndexers.test_convert_unsorted_datetime_index_raisesc sn t dddddgfi�� � fdd�}| jddi|d�� | jddi|tjd��� | jddi|tg d��� d S )Nr r r r c s t j� d| i�S )Nr )r Zremap_label_indexers)r )�datar
r � test_indexere s z<TestIndexers.test_remap_label_indexers.<locals>.test_indexerr )r r+ r Zint32r )r
rB r
)rA r �test_remap_label_indexersa s
z&TestIndexers.test_remap_label_indexersN)
�__name__�
__module__� __qualname__r r$ r0 r6 r= r@ rC r
r
r
r r s ) r c @ s e Zd Zdd� Zdd� ZdS )�
TestLazyArrayc C s t � }tjd�}|d d� |d d� |dd� |d d� |d d
� |dd � |d d
� |d d d� |dd d� |d dd� |d dd� |dd� |d d d� |ddd� |d dd� g}xJ|D ]B}x<|D ]4}|| | }tj||dd �}|| }| j||� q�W q�W d S )N�d r r r r r � r )�sizer( r( ���r( r( r( r( r( ���)r r r r Z slice_slicer ) r
r" r Zslicesr r# �expectedZ new_slice�actualr
r
r �test_slice_slicem s
FF
z TestLazyArray.test_slice_slicec
C s� t jtjjddd��}t j|�}t� }|d d � dd
|d d� ddddgtjd�dk g}x�|D ]�}x�|D ]�}x�|D ]�}tj||||f �}xl||||f |d d �||f | |d d �d d �|f d d �|f | gD ] } | j |j
| j
� | j || � q�W qrW qhW q^W d |d d � df|d d� d f|d d � dgfd dgdfdddgddgfdddg|d d� fg}xL|D ]D\}}tj|| | �}|| | } | j |j
| j
� | j || � �qjW d S )
Nr � rI r r r r r r r) )r r r( ) r ZNumpyIndexingAdapterr r ZrandZLazilyIndexedArrayr r Zasarrayr+ r* r )
r
r Zlazyr" Zindexersr r# r/ rM rN r
r
r �test_lazily_indexed_arrayz s(
2
, 8( z'TestLazyArray.test_lazily_indexed_arrayN)rD rE rF rO rQ r
r
r
r rG l s
rG )Znumpyr Zpandasr7 Zxarrayr r Z xarray.corer � r r r rG r
r
r
r �<module> s c |