3
���h� � @ s d dl mZ d dlmZmZ d dlmZmZ d dlmZ d dlm Z d dl
Z d dl Z d dl
Zd dlmZmZmZ d dlmZ d d lmZmZmZ d d
lmZmZ d d l m Z m Z d d
l m Z m!Z! G dd� de"�Z#G dd� de e#�Z$G dd� de e#�Z%G dd� de �Z&dS )� )�
namedtuple)�copy�deepcopy)�datetime� timedelta)�dedent)� LooseVersionN)�Variable�Dataset� DataArray)�indexing)�
Coordinate� as_variable�as_compatible_data)�PandasIndexAdapter�LazilyIndexedArray)�PY3� OrderedDict� )�TestCase�source_ndarrayc @ s. e Zd Zdd� Zdd� Zdd� ZdJdd �Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Zd4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z d>d?� Z!d@dA� Z"dBdC� Z#dDdE� Z$dFdG� Z%dHdI� Z&dS )K�VariableSubclassTestCasesc C s� dt jd� }| jdg|ddi�}| j|jd� | j|j|� | j|jt� | j|j d � | j|j
d� | j|j d� | j|j d� | jt
|�d� | j|jddi� d S )
Ng �?�
�time�foo�bar�P r )r )r )�np�arange�cls� assertEqual�dims�assertArrayEqual�values�dtype�float�shape�size�nbytes�ndim�len�attrs)�self�data�v� r/ �9/tmp/pip-build-5_djhm0z/xray/xarray/test/test_variable.py�test_properties s z)VariableSubclassTestCases.test_propertiesc C sp | j dgdtjd� �}| j|ji � ddi}||_| j|j|� | j|jt� d|jd< | j|jd d� d S )Nr g �?r r r Zbaz)r r r r r+ �assertIsInstancer )r, r. r+ r/ r/ r0 �
test_attrs" s
z$VariableSubclassTestCases.test_attrsc C s: | j dgtjjd��}|ddi }|d }| j||� d S )N�x� r )r r �random�randn�assertVariableIdentical)r, r. �actual�expectedr/ r/ r0 �test_getitem_dict, s z+VariableSubclassTestCases.test_getitem_dictNc C s� | j |d jf � | j |d jd� | j |d jd� | j|j|j� �� | j|j|j� �� | j |jd |� | j |d j|� |dkr�| j t |jd �t |�� | j t |d j�t |�� n(| j |jd j
|� | j |d jj
|� dS )zjGiven a 1-dimensional variable, verify that the variable is indexed
like a numpy.ndarray.
r r N) r r& r) r' �
assertTrue�equalsr � identicalr# �typer$ )r, �variableZexpected_value0Zexpected_dtyper/ r/ r0 �assertIndexedLikeNDArray2 s z2VariableSubclassTestCases.assertIndexedLikeNDArrayc C sJ xDdt jft jd�t jfgD ]&\}}| jdg|g�}| j|||� q W d S )Nr r4 )r �int_Zint32r rA )r, �valuer$ r4 r/ r/ r0 �test_index_0d_intI s
z+VariableSubclassTestCases.test_index_0d_intc C sJ xDdt jft jd�t jfgD ]&\}}| jdg|g�}| j|||� q W d S )Ng �?r4 )r Zfloat_�float32r rA )r, rC r$ r4 r/ r/ r0 �test_index_0d_floatO s
z-VariableSubclassTestCases.test_index_0d_floatc C sT xNdt jtrdnd�fdt jd�fgD ]&\}}| jdg|g�}| j|||� q&W d S )Nr �U3�S3r4 )r r$ r r rA )r, rC r$ r4 r/ r/ r0 �test_index_0d_stringU s z.VariableSubclassTestCases.test_index_0d_stringc C s� t ddd�}| jdg|g�}| j|tj|�� | jdgtj|�g�}| j|tj|�d� | jdgtj|g��}| j|tj|�d� d S )Ni� r r4 zdatetime64[ns])r r rA r �
datetime64�pdZ
DatetimeIndex)r, �dr4 r/ r/ r0 �test_index_0d_datetime[ s z0VariableSubclassTestCases.test_index_0d_datetimec C sb t dd�}| jdgtj|�g�}| j|tj|�d� | jdgtj|g��}| j|tj|�d� d S )Nr )�hoursr4 ztimedelta64[ns])r r r � timedelta64rA rK � to_timedelta)r, �tdr4 r/ r/ r0 �test_index_0d_timedelta64f s
z3VariableSubclassTestCases.test_index_0d_timedelta64c C s. t jdd�}| jdg|g�}| j||d � d S )NZNaT�nsr4 )r rJ r rA )r, rL r4 r/ r/ r0 �test_index_0d_not_a_timeo s z2VariableSubclassTestCases.test_index_0d_not_a_timec C s6 G dd� dt �}|d�}| jd|g�}| j||� d S )Nc @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
zKVariableSubclassTestCases.test_index_0d_object.<locals>.HashableItemWrapperc S s
|| _ d S )N)�item)r, rU r/ r/ r0 �__init__w s zTVariableSubclassTestCases.test_index_0d_object.<locals>.HashableItemWrapper.__init__c S s | j |j kS )N)rU )r, �otherr/ r/ r0 �__eq__z s zRVariableSubclassTestCases.test_index_0d_object.<locals>.HashableItemWrapper.__eq__c S s
t | j�S )N)�hashrU )r, r/ r/ r0 �__hash__} s zTVariableSubclassTestCases.test_index_0d_object.<locals>.HashableItemWrapper.__hash__c S s dt | �j| jf S )Nz %s(item=%r))r? �__name__rU )r, r/ r/ r0 �__repr__� s zTVariableSubclassTestCases.test_index_0d_object.<locals>.HashableItemWrapper.__repr__N)r[ �
__module__� __qualname__rV rX rZ r\ r/ r/ r/ r0 �HashableItemWrapperv s r_ r � � r4 )r r` ra )�objectr rA )r, r_ rU r4 r/ r/ r0 �test_index_0d_objectt s
z.VariableSubclassTestCases.test_index_0d_objectc s� t jddd�}x�||j|j� gD ]�}| jd|�� xp� fdd�td�D �� fdd�td�D �� fdd�td�D �gD ],}tj|d�}| j� j |j � | j
� |� qrW q W d S ) Nz
2011-09-01r )�periods�tc s g | ] }� | �qS r/ r/ )�.0�i)r: r/ r0 �
<listcomp>� s zLVariableSubclassTestCases.test_index_and_concat_datetime.<locals>.<listcomp>c s g | ]}� ||d � �qS )r r/ )rf rg )r: r/ r0 rh � s c s g | ]}� |g �qS r/ r/ )rf rg )r: r/ r0 rh � s ) rK �
date_ranger# �
to_pydatetimer �ranger �concatr r$ r" )r, ri �dates�timesr9 r/ )r: r0 � test_index_and_concat_datetime� s z8VariableSubclassTestCases.test_index_and_concat_datetimec C s8 | j dtjddd��}tjdd�}| j|d j|� d S )Nr z
2000-01-01r5 )rd z2000-01-01T00ZrS r )r rK ri r rJ r r# )r, r4 r: r/ r/ r0 �test_0d_time_data� s z+VariableSubclassTestCases.test_0d_time_datac C s� t jddd�}x�|df|jdf|jjd�df|j� dfgD ]n\}}| jdg|�}| j|jtjd�� | j |j|j� | j|jjtjd�� t
|j�t
|�k}||ks:t �q:W d S ) Nz
2000-01-01ra )rd Tz
datetime64[s]Fre zdatetime64[ns]) rK ri r# �astyperj r r r$ r r" r �AssertionError)r, rn r# �preserve_sourcer. � same_sourcer/ r/ r0 �test_datetime64_conversion� s z4VariableSubclassTestCases.test_datetime64_conversionc C s� t jddd�}x�|df|jdf|jjd�df|j� dfgD ]n\}}| jdg|�}| j|jtjd�� | j |j|j� | j|jjtjd�� t
|j�t
|�k}||ks:t �q:W d S ) Nr ra )�startrd Tztimedelta64[s]Fre ztimedelta64[ns]) rK Ztimedelta_ranger# rq Zto_pytimedeltar r r$ r r" r rr )r, rn r# rs r. rt r/ r/ r0 �test_timedelta64_conversion� s z5VariableSubclassTestCases.test_timedelta64_conversionc C s6 t jd�jt�jt�}| jd|�}| j|j|j� d S )Nr5 r4 )r r rq �strrb r r r$ )r, r- r9 r/ r/ r0 �test_object_conversion� s z0VariableSubclassTestCases.test_object_conversionc C sp | j dgtjdddgdddgd��}| j||dddg � | j dgtjdddg��}| j|d j|jd � d S )Nr4 r r r` ra )�index)r rK ZSeriesr8 �Indexr r# )r, r. r/ r/ r0 �test_pandas_data� s $z*VariableSubclassTestCases.test_pandas_datac C sH | j dgtjdddd��}| j|d tjddd�� dt|�ksDt�d S ) Nr4 �2000� �B)rv rd �freqr )r� zPeriod('2000-01-03', 'B'))r rK Z period_ranger ZPeriod�reprrr )r, r. r/ r/ r0 �test_pandas_period_index� s z2VariableSubclassTestCases.test_pandas_period_indexc C sf dt jd� }t jd�}| jdg|�}| j||
� | j|t|�� | j| j| � | j||d � | j|d| � | j||d � | j|dkj|dk� | jd|kjd|k� | j|d j|d � | jd| jd| � | j|| j|d � | j|| j|d � | j|| |d � | j|| d| � | jdg|ddi�}| j||
� | j|| d| � | jdg|d d
i�}| j|| | jdg|| �� | j|| j|| � | j|d | d | j|d | d | � | jt |
j
� | jt |
jj
� | jt d| j
� | jt d| jj
� | j |
t � | j
|
t� | j d| t � | j
d| t� d S ) Ng �?r5 r4 r r r` �unitsZmetersr r )r r �onesr r8 �absr" r# r r% r$ r2 r �assertNotIsInstancer
)r, r4 �yr. �v2�wr/ r/ r0 � test_1d_math� s@
.z&VariableSubclassTestCases.test_1d_mathc C sJ t jd�}| jdg|�}|j� }tf d�}| j||� | jt|�t� d S )Nr5 r4 r )r r r �sumr r8 �assertIsr? )r, r4 r. r9 r: r/ r/ r0 �test_1d_reduce� s
z(VariableSubclassTestCases.test_1d_reducec C s� t jd�}| jdg|�}| jt j|�|� | j|jt�|jt�� | j|j� |� | j|j dd�| jd|j dd��� | jt j
|�| jdgt j
|��� | j t j
|�t � | j
t j
|�t� d S )Nr5 r4 r` ra )r r r r" �asarrayrq r% r8 ZargsortZclip�sinr2 r r� r
)r, r4 r. r/ r/ r0 �test_array_interface� s
$"z.VariableSubclassTestCases.test_array_interfacec c sf x`t d�dtjd� dtjdtjd� tjddd�tjdddgtd�gD ]}| jd |�|fV qHW d S )
Nra g �?)r$ z
2000-01-01)rd �a�b�cr4 ) rk r r rE rK ri �arrayrb r )r, r- r/ r/ r0 �example_1d_objects s z,VariableSubclassTestCases.example_1d_objectsc C s� x|| j � D ]p\}}| j|jtj|�� | jtj|�tj|�� | j|d jtj|�d � | jtj|d �tj|�d � q
W d S )Nr )r� r" r# r r� r )r, r. r- r/ r/ r0 �test___array__ s
z(VariableSubclassTestCases.test___array__c C s� x�| j � D ]�\}}|j� }| j|j|�� | j|j|�� | j|d j|d �� | j|d j|d �� | j|d d� j|d d� �� | j|d d� j|d d� �� q
W d S )Nr r` )r� r r< r= r> )r, r. �_r� r/ r/ r0 �test_equals_all_dtypes s z0VariableSubclassTestCases.test_equals_all_dtypesc C sV | j dddg �}x>| j� D ]2\}}d|k}| j||� d|k }| j||� q W d S )Nr4 ra F�z)r r� r8 )r, r: r. r� r9 r/ r/ r0 �test_eq_all_dtypes s
z,VariableSubclassTestCases.test_eq_all_dtypesc C s� | j dtd�ddiddi�}xf|j|d |j� |jtd �d�|jddi�|jd d
�|jd d
�gD ] }| j||� | j |j
|j
� q`W d S ) Nr4 ra r r r r` .)r4 T)�deepF) r rk �T�squeeze�isel�slice� expand_dimsr r8 r �encoding)r, r: r9 r/ r/ r0 �test_encoding_preserved* s
z1VariableSubclassTestCases.test_encoding_preservedc C sN t jd�}t jdd�}| jdg|�}| jdg|�}| jtddgt j||g��tj||gd�� | jtddgt j||g��tj||fd�� | jtddgt j||g��tj||fd�� | jtd�� tj|tdg|�gd� W d Q R X tj||gt ddd�t d dd�gdd
�}tdt j||g�j
d d ��}| j||� td
dgt j j d��}| j|tj|d d� |dd � gd
�� | j|tj|d d� |dd� |dd � gd
�� | j|tj|d d � |d d � gd
�� | j|tj|d d �d d�f |d d �dd �f gd�� | jtd��2 tj|d d �df |d d �d d �f gd� W d Q R X d S )Nr5 r r� r� zinconsistent dimensionsr� r r` r )Z positions�dim�F)�orderr r4 � � zall input arrays must have)r r� ) r r r r8 r r� rl �assertRaisesRegexp�
ValueErrorrk Zravelr6 )r, r4 r� r. r� r9 r: r/ r/ r0 � test_concat6 s.
" (2(8z%VariableSubclassTestCases.test_concatc C s� | j dtjd�ddi�}| j dtjd��}| j dtjtjd�tjd�g��}| j|tj||gd�� d|jd< | j|tj||gd�� d|jd< d|jd< | j|tj||gd�� d S )Nr� r5 r r r` ) r r r r� Z concatenater8 r rl r+ )r, r. r� r: r/ r/ r0 �test_concat_attrsR s "
z+VariableSubclassTestCases.test_concat_attrsc C sv xpdD ]h}| j dtjdg|d��}| j dtjdg|d��}tj||gd�}tdtjddg|d��}| j||� qW d S )N�S�UZanimalZhorse)r$ Zaardvark)r� r� )r r r� r rl ZassertVariableEqual)r, �kindr4 r� r9 r: r/ r/ r0 �test_concat_fixed_len_str^ s
z3VariableSubclassTestCases.test_concat_fixed_len_strc C s� | j ddddg�}| j dddg�}tj||gdd�}tdtjd�jt�jt��}| j||� | j |j
t� | j t |j d �t� d S )
Nr4 �0�1�2�3�4)r� r5 r )
r r rl r r rq rx rb r8 r r$ r? r# )r, r� r� r9 r: r/ r/ r0 �test_concat_number_stringsh s z4VariableSubclassTestCases.test_concat_number_stringsc C s� | j ddtjd� ddi�}x�d D ]�}|j|d�}| jt|�t|�� | j||� | j|j|j� | j t kr"|r�| j
t |j �t |j �� q"| jt |j �t |j �� q"W | j|t|�� d S )
Nr4 g �?r r r TF)r� )TF)
r r r r r� r? r8 r r$ r � assertIsNotr r# )r, r. r� r� r/ r/ r0 � test_copyr s
z#VariableSubclassTestCases.test_copyc C s� | j dtjd�dtjd� ddi�}| j dtjd�ddi�}| j|j|� | j dtjd� ddi�}| j|j|� | j dtjdtjd�d ��}| jt|�|� d S )Nr4 ra y �?r r r` ) r r r r8 �real�imag�sqrt�assertVariableAllCloser� )r, r. Z expected_reZ expected_imZ expected_absr/ r/ r0 �test_real_and_imag� s & z,VariableSubclassTestCases.test_real_and_imagc C s2 | j dddtjg�}tf d�}| j|j� |� d S )Nr4 r y @g �?y �?y �? �?)r r �nanr r� �mean)r, r. r: r/ r/ r0 �test_aggregate_complex� s
z0VariableSubclassTestCases.test_aggregate_complexc C s: t jtjddd��}| jd|�}t|� |jdks6t�d S )Nr Zint64)r$ r4 )rK Z Categoricalr r r �printr$ rr )r, r- r. r/ r/ r0 � test_pandas_cateogrical_dtype� s z7VariableSubclassTestCases.test_pandas_cateogrical_dtypec C sL t jdtjd�ddd�}| jd|�}t|� dt|j�krH|jdksHt�d S )Nz
2000-01-01zAmerica/New_Yorkr Z1h)rv �tzrd r� r4 rb ) rK ri �pytz�timezoner r� rx r$ rr )r, r- r. r/ r/ r0 � test_pandas_datetime64_with_tz� s
z8VariableSubclassTestCases.test_pandas_datetime64_with_tzc C sR t jjtd�ddgg�}| jd|�}| jtf d�|d � | j||d d � � d S )N�abcr r r4 r� )r� r )rK Z
MultiIndexZ from_product�listr r8 r )r, �idxr. r/ r/ r0 �test_multiindex� s z)VariableSubclassTestCases.test_multiindex)N)'r[ r] r^ r1 r3 r; rA rD rF rI rM rR rT rc ro rp ru rw ry r| r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r/ r/ r/ r0 r sH
*
r c @ sL e Zd Zee�Zdd� Zdd� Zdd� Zdd� Z d d
� Z
d d � Z d
d� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Z d/d0� Z d1d2� Z d3d4� Z d5d6� Z d7d8� Z!d9d:� Z"d;d<� Z#d=d>� Z$d?d@� Z%dAdB� Z&dCdD� Z'dEdF� Z(dGdH� Z)dIdJ� Z*dKdL� Z+dMdN� Z,dOS )P� TestVariablec C s t jjd�jt j�| _d S )Nr ra )r ra )r r6 rq Zfloat64rL )r, r/ r/ r0 �setUp� s zTestVariable.setUpc
C s� t ddg| j�}| j|j| j� | j|j| j� | jt|j�| j� | jt�� t j
j
d�|_W d Q R X t j
j
d�}||_| jt|j�|� t j
j
d�}||_| jt|j�|� d S )Nr r4 r5 r ra )r ra )r ra ) r rL r" r- r# r� r � assertRaisesr� r r6 )r, r. Zd2Zd3r/ r/ r0 �test_data_and_values� s z!TestVariable.test_data_and_valuesc C sZ t g tjd��}| j|j� d� | jt|j� �t� tdtj d��}| jd|j
d�� d S )Ng r r4 r5 r` ) r r rE r rU r� r? r% r
r Z searchsorted)r, r. r/ r/ r0 �test_numpy_same_methods� s
z$TestVariable.test_numpy_same_methodsc C sz t jdd�}xht jd�tjd�tddd�gD ]F}tg |�}| j|jt jd�� | j|j|� | j|jjt jd�� q,W d S )Nz2000-01-01T00:00:00ZrS z2000-01-01T00Zz
2000-01-01T00i� r zdatetime64[ns]) r rJ rK � Timestampr r r r$ r# )r, r: r# r. r/ r/ r0 �!test_datetime64_conversion_scalar� s
z.TestVariable.test_datetime64_conversion_scalarc C sz t jdd�}xht jdd�tjd�tdd �gD ]F}tg |�}| j|jt jd
�� | j|j|� | j|jjt jd
�� q,W d S )N� �< r � rS r �Dz1 day)�daysztimedelta64[ns]i� i�Q i ʚ;l �"R: ) r rO rK Z Timedeltar r r r$ r# )r, r: r# r. r/ r/ r0 �"test_timedelta64_conversion_scalar� s
z/TestVariable.test_timedelta64_conversion_scalarc C sp t g d�}| j|jtjd�� | j|jd� t g tjd��}| j|jtjd�� | j|jtrftdd�nd� d S )Nr rG rH �ascii)r r r$ r r# �string_r �bytes)r, r. r/ r/ r0 � test_0d_str� s
zTestVariable.test_0d_strc C s> t g tjd��}| j|jtjd�� | j|jtjdd�� d S )Nz
2000-01-01zdatetime64[ns]z2000-01-01T00ZrS )r rK r� r r$ r r# rJ )r, r. r/ r/ r0 �test_0d_datetime� s z TestVariable.test_0d_datetimec C sX xRt jd�tjdd�gD ]8}tg |�}| j|jtjd�� | j|jtjdd�� qW d S ) NZ1sr �sztimedelta64[ns]r r� rS i ʚ;)rK rP r rO r r r$ r# )r, rQ r. r/ r/ r0 �test_0d_timedelta� s
z TestVariable.test_0d_timedeltac C s( t jjdd�}t j|d < td
|ddddgd�d �}td|ddddgd�d �}| j|j|�� | j|j|�� td|d �}| j|j|�� td|d �}| j|j|�� | j|j|�� t |�}t jjdd�|j
d d �< | j|j|�� | j|jd �� | j|j|�� | j|jd �� | j|j|�� d S )Nr ra r �dim1�dim2r r` )Zatt1Zatt2)r- r+ �dim3)r- )r r )r� r� )r� r� )r� r� )r� r� ) r r6 Zrandr� r r< r= r> � assertFalser r# )r, rL �v1r� �v3�v4Zv5r/ r/ r0 �test_equals_and_identical� s(
z&TestVariable.test_equals_and_identicalc C s� t f tj�}t dtjtjg�}| j|j|�� | j|j|�� | j|j|�� t dtjg�}| j|j|�� | j|j|�� | j|j|�� | j|jd �� t dtjgd �}| j|j|�� d S )Nr4 ra )r r r� r< Zbroadcast_equalsr� r= r> )r, r� r� r� r� r/ r/ r0 �test_broadcast_equals s z"TestVariable.test_broadcast_equalsc C sB t jd�}td|�}| j|t|�� td|i�}| j|t|d �� | j|d t� | jt|d �t� | jt|d dd�t� t dd�}||j
|j �}| j|t|�� |j |j
f}| j|t|�� | j t
d�� tt|�� W d Q R X | j t
d�� t|� W d Q R X t|dd �}| j||� td
�}tg d
�}| j||� d S ) Nr r4 F)�strict� FakeVariablez values dimszcannot convert argzcannot infer .+ dimensions)�keyr )r r r r8 r r
r� r2 r r r# r! r� � TypeError�tuple)r, r- r: Zdsr� Z fake_xarrayZ xarray_tupler9 r/ r/ r0 �test_as_variable s,
z TestVariable.test_as_variablec C sD t ddgdddgdddggd d
i�}td �j� }| j|t|�� d S ) Nr r4 r r` ra � r5 r� r r z�
<xarray.Variable (time: 2, x: 3)>
array([[1, 2, 3],
[4, 5, 6]])
Attributes:
foo: bar
)r r �stripr r� )r, r. r: r/ r/ r0 � test_repr> s $
zTestVariable.test_reprc C s6 t dttjd���}| jdt|�� | j|jt� d S )Nr4 g jAz200000 values with dtype)r r r r ZassertInr� r2 �_data)r, r. r/ r/ r0 �test_repr_lazy_dataI s z TestVariable.test_repr_lazy_datac C s| t jjd �}tddg|�}| j||d d � � | j||d � | jtdg|d �|d � | jtdg|d d �df �|d d �df � | jtddg|d d�d d�f �|d d�d d�f � tdgt jd��}tdgt jd��}| j|||j � | j||| � | j|d d� ||dk � | j|d d �dd �f |d d �|dkf � | j|d d�dd �f ||dk |dkf � | j|d d�d d�f ||d d� |d d� f � | j|d d�d d�f |td�td�f � x.t|�D ]"\}}| jtdg|| �|� �q�W | jt d �� t
tg d�� W d Q R X d|jd d �< | j t j |jdk�� d
|td�td�f< | j
|jt jd �� d S )
Nr � r4 r� .r ra r` ziteration over a 0-dr )r r� )r r� )r r6 r r8 r r# rk � enumerater� r� �iterr< �allr" r� )r, r- r. r4 r� �nrU r/ r/ r0 �
test_itemsN s2 , ,,4, zTestVariable.test_itemsc
C s� t ddg| j�}| j|jtd �d�|� | j|jdd�|d � | j|jtdd�d�|d d� � | j|jdd�|d d �df � | jtd�� |jdd� W d Q R X d S ) Nr r4 )r r ra )r4 z do not exist)Z not_a_dim)r rL r8 r� r� r� r� )r, r. r/ r/ r0 � test_iseln s " zTestVariable.test_iselc C sD t g tjd��}| j|f |� t g tjd��}| j|f |� d S )NZasdf)r r r� r8 Zunicode_)r, r. r/ r/ r0 �test_index_0d_numpy_stringw s z'TestVariable.test_index_0d_numpy_stringc C s. t ddg�d f }t f d�}| j||� d S )Nr4 Ztmaxr )r r8 )r, r9 r: r/ r/ r0 �test_indexing_0d_unicode s
z%TestVariable.test_indexing_0d_unicodec C s| t ddddddg�}| j||jdd�� | j||jdd�� t dtjddddg�}| j||jdd�� t dtjtjdddg�}| j||jdd�� t dddddtjg�}| j||jdd�� t dtjgd �}| j||jdd�� | j||jd d�� | jtd
�� |jdd � W d Q R X t ddddddgd d
i�}| j||jdd�� t dtjddddgd d
i�}| j||jdd�� d S )Nr4 r r` ra r� r5 r )r4 r� � dimension)r� r r ���)r r8 �shiftr� r r� r� r� )r, r. r: r/ r/ r0 �
test_shift� s$ zTestVariable.test_shiftc C sL t dddgddgg�}t d tjtjgtjdgg�}| j||jddd�� d S )
Nr4 r� r r` ra r� )r4 r� )r4 r� )r4 r� )r r r� r8 r� )r, r. r: r/ r/ r0 � test_shift2d� s zTestVariable.test_shift2dc
C s� t ddddddg�}| j||jdd�� | j||jdd�� t ddddddg�}| j||jdd�� | j||jd d�� | j||jd d�� t ddddddg�}| j||jdd�� | j||jd
d�� | jtd
�� |jdd � W d Q R X d S )Nr4 r r` ra r� r5 r )r4 r� r� )r� ������)r r8 �rollr� r� r� )r, r. r: r/ r/ r0 � test_roll� s zTestVariable.test_rollc C sh t d tjjdd��}xNdD ]F\}}x<dD ]4}tj|j||d
�}|jf ||i�j}| j||� q(W qW d S )Nr4 r� r5 r� r r ra � r� )�axis)r4 r� �r r4 �r r� )r r r )r r r r r� )r r r6 r7 r r# r" )r, r. r r� r� r: r9 r/ r/ r0 �test_roll_consistency� s
z"TestVariable.test_roll_consistencyc C s� t ddg| j�}t ddg| jj�}| j||j� � | j|j� |j� tjjdddd�}t ddd d
g|�}t d
dd dgtjd |��}| j |j
d
� | j||jd
dd d�� | j||jddd d
�� t dd d
dgtjd |��}| j||jddd d
�� d S )Nr r4 r` ra r� r5 r� r� r� rL z
abcd->dbcaz
abcd->bcda)r5 ra r� r` ) r rL r� r8 � transposer r6 r7 �einsumr r& )r, r. r� r4 r� Zw2Zw3r/ r/ r0 �test_transpose� s zTestVariable.test_transposec C s t ddgdgg�}| jt g d�|j� � | jt dgdg�|jd�� | jt dgdg�|jdg�� | jt dgdg�|jd�� | jt g d�|jddg�� t ddgddgg�}| jt dgddg�|j� � | jt dgddg�|jd�� | jtd�� |jd� W d Q R X d S )Nr4 r� r r` zcannot select a dimension)r r8 r� r� r� )r, r. r/ r/ r0 � test_squeeze� s zTestVariable.test_squeezec C s� t dddgtjjddd��}| j|jd�d� | j|jdg�d � | j|jddg�d � | j|jdddg�d
� | jtd �� |jd
� W d Q R X d S )Nr4 r� r� r` ra r� r r znot found in array dimZfoobar)r )r r )r` r r )r r r6 r7 r Z get_axis_numr� r� )r, r. r/ r/ r0 �test_get_axis_num� s z TestVariable.test_get_axis_numc C s� t dgddg�}|jddg�}t ddgdgdgg�}| j||� |jddg�}| j||j� |jtd d
g��}t ddgddgddgg�}| j||� t dgddg�}|jd�}|}| j||� | jtd�� |jdg� W d Q R X d S ) Nr4 r r r� r` r zmust be a supersetr� )r4 r` )r� r` )r r� r8 r� r r� r� )r, r. r9 r: r/ r/ r0 �test_expand_dims� s
z TestVariable.test_expand_dimsc C s� t ddgddgddggddi�}|jdd �}t d
ddddg|j�}| j||� |jdd �}t d|jj|j�}| j||� |jf d �}| j||� |jddd �jd d
�}t d|j|j�}| j||� d S )Nr4 r� r r r` ra r r )r� r� )�X�Yr r )r4 r� )r4 )r� r� )r4 )r� )r r )r �stackr+ r8 r- r� r )r, r. r9 r: r/ r/ r0 �
test_stack� s zTestVariable.test_stackc C sl t ddgddgddggddi�}| jtd �� |jdd � W d Q R X | jtd �� |jdd
� W d Q R X d S )Nr4 r� r r r` ra r r zinvalid existing dim�x1)r� zcannot create a new dim)r4 )r )r4 )r r� r� r )r, r. r/ r/ r0 �test_stack_errors s
z TestVariable.test_stack_errorsc C s� t dddddgddi�}|jtd d
g�d
�}t dddgddgg|j�}| j||� |jtddg�d
�}t ddgdgdgdgg|j�}| j||� |jtdg�d
�}t dddddg|j�}| j||� d S )Nr� r r r` ra r r r4 r� )r� r� )r4 r` )r� r` )r4 r� )r4 r� )r� r )r4 r� )r4 r� )r �unstackr r+ r8 )r, r. r9 r: r/ r/ r0 � test_unstack s zTestVariable.test_unstackc C s� t dddddg�}| jtd�� |jddid � W d Q R X | jtd
�� |jdd � W d Q R X | jtd �� |jdd
id � W d Q R X d S )Nr� r r r` ra zinvalid existing dimr4 r� )r zcannot create a new dim)r� zthe product of the new dimr5 )r� )r r� r� r r )r, r. r/ r/ r0 �test_unstack_errors" s z TestVariable.test_unstack_errorsc C sx t ddgddgddgg�}|jddid�}t ddg|j�}| j||� |jddid �}t ddg|jj�}| j||� d S )
Nr4 r� r r r` ra r� )r� )r4 )r r r- r8 r� )r, r. r9 r: r/ r/ r0 �test_unstack_2d+ s z TestVariable.test_unstack_2dc C sF t ddgddgddgg�}|jdd�jtd d
g�d�}| j||� d S ) Nr4 r� r r r` ra )r� )r4 r� )r4 r` )r� r` )r r r r r8 )r, r. r9 r/ r/ r0 � test_stack_unstack_consistency5 s z+TestVariable.test_stack_unstack_consistencyc
C s� t jjdd�}tddg|�}| j|| tddgt jd||��� | j||d tddgt jd||d ��� | j|d | tddgt jd|d |��� | j|d |d d �df tddgt jd |d |d d �df ��� t jjdd
d �}tdd d
g|�}| j|| tddd d
gt jd||��� | j|| tdd d
dgt jd||��� | j||d tddd d
gt jd||d ��� d S )Nr` ra r� r� z ab,ab->abr zab,b->abzb,ab->bazb,a->bar� r5 r� rL z ab,bcd->abcdz bcd,ab->bcdaz ab,cd->abcd)r r6 r7 r r8 r
)r, r4 r. r� r� r/ r/ r0 �test_broadcasting_math; s2
*z#TestVariable.test_broadcasting_mathc C s� t dgtjd��}t dgtjd��}t ddgtjd�jdd��}| jtd�� || W d Q R X | jtd�� || W d Q R X d S )Nr4 r r5 �d zmismatched lengthszduplicate dimensions)r r r Zreshaper� r� )r, r� r� r� r/ r/ r0 �test_broadcasting_failuresX s z'TestVariable.test_broadcasting_failuresc C s� t jd�}tdg|�}|}|d7 }| j||� | jt|j�|� | j|jt jd�d � | jtd�� |tdt jd��7 }W d Q R X d S )Nr5 r4 r zdimensions cannot changer� ) r r r r� r r# r" r� r� )r, r4 r. r� r/ r/ r0 �test_inplace_matha s
z TestVariable.test_inplace_mathc C s& t ddg| jddi�}| j|jtjd�t dg| jjdd��� | j|jtjdd�|jtjdd�� | j|jtjddg�t g | jjd d��� | j|jtj�t g | jj� �� | j|jtjd�jtjd�t g | jjdd�j� �� | j|jd�|jtjd�� | jt d �� |jddd
� W d Q R X d S ) Nr4 r� Zignored�
attributesr )r )r� r zcannot supply both)r� r )r r )
r rL r8 �reducer Zstdr� r� r� r� )r, r. r/ r/ r0 � test_reducen s zTestVariable.test_reducec C s8 t jddd�}tdg|�}tg d�}| j||j� � d S )Nr5 z>f4)r$ r4 )r r� r r8 r� )r, r- r. r: r/ r/ r0 �test_big_endian_reduce� s
z#TestVariable.test_big_endian_reducec C s� t dtjdtjddg��}| j|j� t g d�� | j|jdd�t g d�� | j|jdd�t g tj�� | jtj|�t g d�� | j|j� t g d�� | j|j� t g d�� ttj �d
k r�| j
t �� |j � W d Q R X n| j|j � t g d�� t ddddg�}| j|j
� t g d�� | j|jdd �t g d�� t d tjd
dd��}| j
t �� |jdd� W d Q R X | j|j� t g tjd��� d S )Nr4 r r` ra T)ZskipnaFr� g @z1.9)r� re z
2000-01-01)rd z
2000-01-03gUUUUUU�?)r r r� r� r8 r� �prod�varr � __version__r� �NotImplementedErrorZmedian�anyr� rK ri �maxr� )r, r. r/ r/ r0 �test_reduce_funcs� s& z TestVariable.test_reduce_funcsc C sz ddd�}t ddg| j|�}|j� }| jt|j�d� | j|jt� � |jdd�}| jt|j�t|�� | j|j|� d S ) N�testZtesting)r� Z long_namer4 r� r T)Z
keep_attrs)r rL r� r r* r+ r )r, �_attrsr. Zvmr/ r/ r0 �test_reduce_keep_attrs� s
z#TestVariable.test_reduce_keep_attrsc C s� t g d�}t dgdddtjg�j� }| j||� t dgtjdddtjgtd��}|j� }| j||� t dgd d
d g�j� }| j||� | j|jt � t dgddg�}t dd gdd tjgdddgg�jd �}| j||� d S )
Nra r4 r r` r� r� r� )r$ TFr� r )
r r r� �countr8 r� rb r r$ �int)r, r: r9 r. r/ r/ r0 �
test_count� s
&zTestVariable.test_countN)-r[ r] r^ � staticmethodr 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/ r0 r� � sP
r� c @ s4 e Zd Zee�Zdd� Zdd� Zdd� Zdd� Z d S )
�TestCoordinatec
C s& | j td�� tf d� W d Q R X d S )Nzmust be 1-dimensionalr )r� r� r
)r, r/ r/ r0 � test_init� s zTestCoordinate.test_initc C sB dt jd� }tdg|ddi�}| jtj|dd�j|j� �� d S )Ng �?r r r r )�name)r r r
r< rK r{ r> Zto_index)r, r- r. r/ r/ r0 �
test_to_index� s z TestCoordinate.test_to_indexc C s� t dtjd��}| jt|j�tj� | jt|j � | j
tjd�|� | jt|j j � | j |jt
� | jtd�� d|d d �< W d Q R X d S )Nr4 g @ra zcannot be modifiedr )r
r r r� r? r� �ndarrayr r% r$ r" r# r2 r r� r� )r, r4 r/ r/ r0 � test_data� s zTestCoordinate.test_datac C s: t ddg�}| j|jd� | jt�� d|_W d Q R X d S )Nr4 g $@r� )r
r r2 r� �AttributeError)r, Zcoordr/ r/ r0 � test_name� s zTestCoordinate.test_nameN)
r[ r] r^ r/ r
r r1 r3 r5 r7 r/ r/ r/ r0 r0 � s
r0 c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�TestAsCompatibleDatac C sp t jttjf}x\|D ]T}xNt jd�tjddd�tjddd�jgD ]$}||�}| j t
|�t
t |��� q@W qW d S )Nra z
2000-01-01)rd ) r r� r r r r rK ri r# r� r r )r, �typesre r- r4 r/ r/ r0 �test_unchanged_types� s
z)TestAsCompatibleData.test_unchanged_typesc C sn xhdddggt jdddgg�gD ]F}t|�}| jtj|�|� | jtjt|�� | jtj|�j |j � q W d S )Nr r r` )
rK Z DataFramer r" r r� r r4 r? r$ )r, Z input_arrayr9 r/ r/ r0 �test_converted_types� s
$z)TestAsCompatibleData.test_converted_typesc C s� t jjt jd��}t jd�}t|�}| j||� | jt jt�|j� t jjt jd�ddg dg d�}t jd�}t j |d< t|�}| j||� | jt jt
�|j� d S ) Nr5 r� FT)�maskg @r r� ) r �maZ MaskedArrayr r r" r r$ r- r� r% )r, �originalr: r9 r/ r/ r0 �test_masked_array� s
"
z&TestAsCompatibleData.test_masked_arrayc C sX t jd�}t|�}| j||� | jt jt|�� | jt jd�|j� t jt jd�g�}t|�}| jt j|�|� | jt jt|�� | jt jd�|j� t jt jdd�g�}t|�}| jt j|�|� | jt jt|�� | jt jd�|j� | j |t
t j|��� t jdd�}tt ddd��}| jt j|�|� | jt jt|�� | jt jd�|j� d S )Nz2000-01-01T00Zzdatetime64[ns]rS i� r ) r rJ r r r4 r? r$ r� r� r� r r )r, r: r9 r/ r/ r0 �
test_datetime s*
z"TestAsCompatibleData.test_datetimeN)r[ r] r^ r: r; r? r@ r/ r/ r/ r0 r8 � s
r8 )'� collectionsr r r r r �textwrapr Zdistutils.versionr Znumpyr r� ZpandasrK Zxarrayr r
r Z xarray.corer Zxarray.core.variabler
r r Zxarray.core.indexingr r Zxarray.core.pycompatr r � r r rb r r� r0 r8 r/ r/ r/ r0 �<module> s0 |