3
���h� � @ sh d dl Zd dl mZmZ d dlmZ d dlmZmZm Z m
Z
d dl m Z
ddlmZ G dd � d e�ZdS )
� N)�array�nan)�ops)�first�last�count�mean)�interleaved_concat� )�TestCasec @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )�TestOpsc C sN t ttdtgtddtgdddtggtddd gtd
d tgtd ttggg�| _d S )
Ng @g @g @g @g "@g $@g *@g ,@g .@g 1@g 2@g 5@)r r �x)�self� r �4/tmp/pip-build-5_djhm0z/xray/xarray/test/test_ops.py�setUp
s
z
TestOps.setUpc
C s t tdddgtddtgdddtgg�t dddtgtdd dgg�t dddgdd
d gg�g}x>td d
ddddgd| �D ] \}}t| j|�}| j||� qtW | jd }t| jd dd�}| j||� | jd }t| jddd�}| j||� | jtd�� t| jd� W d Q R X d S )N�
� � � � � � �
� � � r r
� F)�axis�skipna.z
out of bounds���������).r r" )r r �zipr r
�assertArrayEqual�assertRaisesRegexp�
IndexError)r �expected_resultsr �expected�actualr r r �
test_first s&
zTestOps.test_firstc
C s t tdddgtddtgdddtgg�t dd dtgtdddgg�t d
d dgdddgg�g}x>td d
d
dddgd
| �D ] \}}t| j|�}| j||� qtW | jd }t| jd dd�}| j||� | jd }t| jddd�}| j||� | jtd�� t| jd� W d Q R X d S )Nr r r r � r r r r r r r r
r F)r r .z
out of boundsr r! r" r" r" ).r" r" )r r r# r r
r$ r% r& )r r' r r( r) r r r � test_last. s&
zTestOps.test_lastc C sD | j dt| j�� tdddgdddgg�}| j|t| jdd�� d S )N� r
r r )r r" )Z assertEqualr r
r r$ )r r( r r r �
test_countF s zTestOps.test_countc C s t jtt jt jg��st�d S )N)�np�isnanr r �AssertionError)r r r r �test_all_nan_arraysL s zTestOps.test_all_nan_arraysc C s� �x�t tjtjgD �]n}tjd�}| j|||g|g�� tjd�jdd�}tjd�jdddd�}|||�}tjdddddd d
d d d
g
�}| j||� |jddd�}|||dd�}| j|jdd�|� ||dgdggdd�}| j|j |� ||t
d�t
dd�gdd�}| j|j |� | j t �� |||dd� W d Q R X | j t �� |||dd� W d Q R X | j t �� ||ddgdd� W d Q R X qW d S )Nr r r r
�F)�orderr r � r r � r )r r" r" r" r! )
�interleaved_concat_numpyr �_interleaved_concat_slowr r/ Zaranger$ Zreshaper �T�sliceZ assertRaisesr& )r r r
Zarrays�indicesr) r( Zarrays2r r r �test_interleaved_concatO s.
z TestOps.test_interleaved_concatc C s� x�t tjtjgD ]�}tjdg�}tjdg�}|||gdgdgg�}tjddg�}| j||� tjtjgtd�}||||gdgdgdgg�}tjddtjgtd�}| j||� qW d S )N�aZbcr r
)Zdtyper ) r7 r r8 r r/ r r$ r �object)r r r= �br) r( �cr r r � test_interleaved_concat_dtypesn s z&TestOps.test_interleaved_concat_dtypesc C s� | j tjdgg�� | j tjddgdddgg�� | j tjtd�tdd�g�� | j tjtddd�g�� | jtjdgdgg�� | jtjdgddgg�� | jtjddgddgg�� | jtjddgddgg�� | jtjtd d d�g�� d S )Nr r
r r r6 g @)Z assertFalser Z _interleaved_indices_requiredr: Z
assertTrue)r r r r �!test_interleaved_indices_required} s z)TestOps.test_interleaved_indices_requiredN) �__name__�
__module__� __qualname__r r* r, r. r2 r< rA rB r r r r r s r )Znumpyr/ r r Z xarray.corer Zxarray.core.opsr r r r Zxarray.core.nputilsr r7 � r r r r r r �<module> s |