3
���hN � @ s: d dl Zd dljZd dlZddlmZ G dd� de�Z dS )� N� )�TestCasec @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�TestOpsc C sd t |�t |�ks$t|�t|�ks$t�y|j|�s<t||f��W n tk
r^ | j||� Y nX d S )N)�type�float�AssertionErrorZ identical�AttributeErrorZassertArrayEqual)�self�a�b� r �7/tmp/pip-build-5_djhm0z/xray/xarray/test/test_ufuncs.py�assertIdentical
s
$zTestOps.assertIdenticalc C sj dt jd�tjdgddg�tjddgdd�tjddddgfi�g}x"|D ]}| j|d tj|�� qHW d S )Nr � �x)�dims�yr ) �np�zeros�xr�Variable� DataArray�Datasetr �xu�cos)r �argsr
r r r
�
test_unary s
zTestOps.test_unaryc C s� dt jd�tjdgddg�tjddgdd�tjddddgfi�g}x�t|�D ]�\}}x�||d � D ]t}| j|d tj ||d �� | j|d tj ||d �� | j|d tj |d |�� | j|d tj |d |�� qbW qLW d S )Nr r r )r r r )
r r r r r r � enumerater r �maximum)r r �n�t1�t2r r r
� test_binary s zTestOps.test_binaryc
C s& t jdddddgfidddddgfi�}|jd�}|jd�}|d jd�}| j|tj||�� | j|tj||�� | j|tj||�� | j|tj||�� | j|tj||d �� | j|tj|d |�� | j|jtj||j�� | j|jtj|j|�� | jt d�� tj|jj
|� W d Q R X d S )Nr
r r �cr zonly support binary ops) r r �groupbyZmeanr r r r
ZassertRaisesRegexp� TypeError�variable)r ZdsZ
ds_groupedZ
group_meanZ arr_groupedr r r
� test_groupby' s (
zTestOps.test_groupbyN)�__name__�
__module__� __qualname__r r r" r' r r r r
r s
r )
Znumpyr Z
xarray.ufuncsZufuncsr Zxarrayr � r r r r r r
�<module> s
|