3
���h� � @ s� d dl Zd dlZd dlZd dlmZ d dlmZ d dl m
Z
m Z m Z m
Z
mZmZ d dlmZmZ ddlmZmZmZmZmZ G dd � d e�ZdS )
� N)�deepcopy)�dedent)�align� broadcast�Dataset� DataArray�
Coordinate�Variable)� iteritems� OrderedDict� )�TestCase�
ReturnItem�source_ndarray�unittest�
requires_daskc @ s e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zed%d&� �Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� 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*dOdP� Z+dQdR� Z,dSdT� Z-dUdV� Z.dWdX� Z/dYdZ� Z0d[d\� Z1d]d^� Z2d_d`� Z3dadb� Z4dcdd� Z5dedf� Z6dgdh� Z7didj� Z8dkdl� Z9dmdn� Z:dodp� Z;dqdr� Z<dsdt� Z=dudv� Z>dwdx� Z?dydz� Z@d{d|� ZAd}d~� ZBeCjDd�d�d�� �ZEd�d�� ZFd�d�� ZGd�d�� ZHd�d�� ZId�d�� ZJd�d�� ZKd�d�� ZLd�d�� ZMd�d�� ZNd�d�� ZOd�d�� ZPd�d�� ZQd�d�� ZRd�d�� ZSd�d�� ZTd�d�� ZUd�d�� ZVd�d�� ZWd�d�� ZXd�d�� ZYd�d�� ZZd�d�� Z[d�d�� Z\d�d�� Z]d�d�� Z^d�d�� Z_d�d�� Z`d�d�� Zad�d�� Zbd�d�� Zcd�d�� Zdd�d�� Zed�S )��
TestDataArrayc C sb ddd�| _ tjjd �| _tddg| j�| _tddg| j| j �| _td| ji�| _| jd | _ d S )
NZvalue1iq )Zattr1Zattr2�
� �x�y�foo)r r )
�attrs�np�randomr r �v�var �ds�dv)�self� r �:/tmp/pip-build-5_djhm0z/xray/xarray/test/test_dataarray.py�setUp s zTestDataArray.setUpc C sX t ddgdddgdddggd d
i�}t|d tjd �id
d�}td�}| j|t|�� d S )N�timer r � � � � � r �bar�otherr Z my_variable)�namea <xarray.DataArray 'my_variable' (time: 2, x: 3)>
array([[1, 2, 3],
[4, 5, 6]])
Coordinates:
other int64 0
* time (time) int64 0 1
* x (x) int64 0 1 2
Attributes:
foo: bar)r r r �int64r � assertEqual�repr)r r Z
data_array�expectedr r r! � test_repr s
$ zTestDataArray.test_reprc C sB | j | jj| j� | j| jj| jj� x(d D ] }| jt| j|�t| j|�� q,W | jt| j�t| j�� | j | j| j� | j t
| jj �t
| j j �� x,t
| jj �D ] \}}| j|| j j | � q�W | jt�� | jj W d Q R X | j| j d j� tj� | jtd �� | j d
j� W d Q R X | jt�� | j| j_W d Q R X d S ) N�dims�dtype�shape�size�nbytes�ndimr r zmust be 1-dimensionalr )r1 r2 r3 r4 r5 r6 r )�assertVariableEqualr �variabler �assertArrayEqual�valuesr- �getattr�len�assertItemsEqual�list�coordsr r
� assertRaises�AttributeErrorZdataset�assertIsInstanceZto_index�pd�Index�assertRaisesRegexp�
ValueError)r �attr�kr r r r! �test_properties% s
z TestDataArray.test_propertiesc C sr t tjd��}|j� }tjd�|_| jtjd�|j� dtjd� |_| jdtjd� |j� | j|j|j� d S ) Nr% r& r$ )r% r& )r% r& )r% r& )r% r& )r% r& )r r �zeros�copy�onesr: r9 �data)r �array�actualr r r! �test_data_property7 s z TestDataArray.test_data_propertyc C sx | j }| j|jd� |j� }d|_| j|jd� | j||� ttddg��}d|_tdgddgidd�}| j||� d S )Nr r) r r% r )r+ )r r- r+ rK �assertDataArrayEqualr r �assertDataArrayIdentical)r �arr�copiedrO r/ r r r! � test_name@ s zTestDataArray.test_namec C s6 | j }| j|jd� | jtd�� d|_W d Q R X d S )Nr r zyou cannot assign�w�z)r r )rV rW )r r- r1 rE rA )r rS r r r! � test_dimsN s zTestDataArray.test_dimsc C sX ddi}d| j jd< | j|| j j� ddi}|| j _| j|| j j� | j|| j j� d S )Nr r) �bazr )r �encoding� assertEqualsZ assertIsNot)r r/ r r r! �
test_encodingU s zTestDataArray.test_encodingc C s@ t jjd�}t|�}td ddg|fi�d }| j||� t|ddgdddgg�}tddg|fdddgfddd d gfd��d }| j||� t|tjddgd d
�tjd d d gd d
�g�}td d g|fd ddgfd d!d"d#gfd ��d }| j||� ddgd$d%d&gg}t||d d g�}| j||� tjddgd
d
�tjd'd(d)gdd
�g}t||d d g�}| j||� ddgd*d+d,gd�}t||d d g�}| j||� d ddgfd d-d.d/gfg}t||�}| j||� t|t|��}| j||� td d g|fd ddgfd��d }t|d ddgid d g�}| j||� t|d d gd�}td d d g|fi�d }| j||� t|d d gdd�}tdd d g|fi�d }| j||� t|dd
�}tdddg|fi�d }| j||� t|d d gddid�}td d d g|ddifi�d }| j||� t|d d gddid�}td d d g|i ddifi�d }| j||� d S )0Nr$ r% �dim_0�dim_1�a�br )Nr] r^ r )r+ r )Nr r �A�B)r r )Nr )r1 r )r1 r+ r) )r1 r )r1 rZ )r$ r% ���������rc rd re rc rd re rc rd re rc rd re rc rd re rc rd re rc rd re )r r r r rR rC rD r )r rM rO r/ r? r r r! �test_constructor_ sb
z TestDataArray.test_constructorc C s� t jjdd�}| jtd�� t|dddggddg� W d Q R X | jtd�� t|ddddgid d
g� W d Q R X | jtd�� t|ddddgi� W d Q R X | jtd �� t|dd gd � W d Q R X d S )
Nr% r$ zcoords is not dict-liker r r r znot a subset of the .* dimr_ r` zis not a string)r1 )r r �randnrE rF r � TypeError)r rM r r r! �test_constructor_invalid� s "$ z&TestDataArray.test_constructor_invalidc C s� ddgddgg}t |ddgddgd�d d
gd d did
did�}t |�}| j||� t |j|j�}| j||� tj|tjddgd d�tjd d gd
d�d�}t |�}| j||� tj|d tjd d gd
d�d�}t |�}| j|d j d dd�|� tj
d|i�}t |�}t |g|jdd d
g�}| j||� t |ddgd d!gdd d"dgfd�d d
gd�}t |�}| j||� t |j|j�}| j||� t d
d
ddgfi�d
}t tjddgd
d��}| j||� t t d
ddg��}| j||� d S )#Ng�������?� r r$ r_ r` r )r r r r Zfoobarr) r r% )r? r1 r+ r rZ )r+ )�index�columns)rk T)�dropr] g �?)r r r_ rW )r? r1 g��������rc rd rc rd rc rd rc rd g �)
r rR r: r? rQ rC � DataFramerD �Series� reset_coordsZPanelr r )r rM r/ rO �frame�seriesZpanelr r r! �$test_constructor_from_self_described� sD z2TestDataArray.test_constructor_from_self_describedc C s, t d g dfi�d }td�}| j||� d S )Nr )r r rR )r r/ rO r r r! �test_constructor_from_0d� s z&TestDataArray.test_constructor_from_0dc C s� t tjd�ddidd�}|}|j� }| j|j|�� | j|j|�� |jd�}| j|j|�� | j|j|�� |jddi�}| j|j|�� | j|j|�� |j� }d|j d < | j|j|�� | j|j|�� |j� }dtjd
� f|d< | j|j|�� | j|j|�� |j
d d �}| j|j|�� | j|j|�� |j� }tj |d
< |j� }| j|j|�� | j|j|�� tj |d d �< | j|j|�� | j|j|�� |j� }d|d< | j|j|�� | j|j|�� d S )Ng @r_ �* r )r1 rY Zxxxr) r r' T)rm r i�� ) r r �arangerK �
assertTrue�equals� identical�rename� assertFalser rp �nan)r �origr/ rO r r r! �test_equals_and_identical� sD
z'TestDataArray.test_equals_and_identicalc C sV t tjd�ddidd�}| j|jtjd��� | j|jd�� | j|jdd i�� d S )
Ng @r_ ru r )r1 r' �{ r r$ )r r rv r{ rx ry �broadcast_equals)r r} r r r! �test_equals_failures s z"TestDataArray.test_equals_failuresc C s� t ddgddidd�}t ddgddddgfidd�}| j|j|�� | j|j|�� | j|j|�� | j|j|�� t dgddidd�}| j|j|�� | j|j|�� d S )Nr r r )r1 )r? r1 )r rw r� r{ rx ry )r r_ r` �cr r r! �test_broadcast_equals s z#TestDataArray.test_broadcast_equalsc C s� | j | j| jd � | jd }| jd }| j | jd |� | j | jd |� t� }x�|d d � |d ||j ||j || |||f ||jd
k ||jd k ||d k ||d
k|dkf g
D ]}| j| j| j| � q�W x�|d |d d �df |d d�d d�f ||jd d� ||jd d� ||d d� ||d d� |d d � f ||jdk ||jdk ||dk ||dk|dkf g D ] }| j| j| | j| � �q�W d S )Nr r r .r r r% r$ r& rc rc rc rc rc )rR r r r r: r8 r7 r )r r r �I�ir r r! � test_getitem s
2"(H<zTestDataArray.test_getitemc C s8 | j td�dd� }| j jtd�dd�}| j||� d S )Nr% r )r r )r �slice�iselrR )r rO r/ r r r! �test_getitem_dict0 s z TestDataArray.test_getitem_dictc
C s� t dgdggddgdgdddd gfd
d gfd
dgd d
ggfd�dd
gd�}| j||d d � � | j||d d �d d �f � | j||d � | j||d d�d d�f � | j||ddgdgf � |d }t dddddd d d��}| j||� |dd d �f }t dgddgddd
d gfd
d gfd�d
d�}| j||� |d d �df }t ddgddgddddd gfd dd d
gfd�dd�}| j||� d S )Nr r r r$ r% r& r r_ r` r r� �d�e)r r rW �x2�y2Zxy)r1 .r )r r )r rR )r r} rO r/ r r r! �test_getitem_coords5 s6
z!TestDataArray.test_getitem_coordsc C s4 t tjjd�dd�}tjtj|��}| j||� d S )Nr% �idr# )r1 )r% r% )r� r# )r r r �pickle�loads�dumpsrR )r rM � roundtrippedr r r! � test_pickleV s zTestDataArray.test_picklec C sx t tjd��}| j|j� |j� }| j|jd � |jd d�}| j|jd� |jdd�}| j|jd� | j|j� j� d S )Nr% r& r$ r )�chunks)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 rL Z assertIsNoner� �chunkr- �load)r Z unblockedZblockedr r r! �
test_chunk[ s zTestDataArray.test_chunkc C s� | j | jd | jjdd�� | j | j| jjtd �d�� | j | jd d� | jjtd�d�� | j | jd d�d d�f | jjtd�td�d�� d S )Nr )r r% r' )r r )rR r r� r� )r r r r! � test_iselk s
$zTestDataArray.test_iselc C s dt jtd��f| jd< | jd }| j||jtd �d�� | j|d |jdd�� | j|d d� |jtd�d�� | j|d d� |jd ddgd�� | j|d d �d d
�f |j| jd d
k d �� td�}| j|d |j|d�� | j|dg |jt||�d�� d S )
Nr �
abcdefghijr )r r r` r% r� r_ r& r )r )r rN r> r rR �selr� r )r �dar` r r r! �test_selr s
".zTestDataArray.test_selc C s� t tjjdd�ddddgfdtd�fg�}|jd d
gd �}|jd d
gdd�}| j||� tjdkr�|jddgd�}|jddgddd�}| j||� n,| j t
d�� |jddgddd� W d Q R X d S )Nr% r& r r r r$ r Zabcdr_ r` )r �abZba�pad)r �methodz0.17)r g�������?gffffff�?�backfill)r r� � tolerancer� ) r r r rg r> r� rR rC � __version__rE �NotImplementedError)r rM r/ rO r r r! �test_sel_method s
z TestDataArray.test_sel_methodc / s� d!}t jj|�}t|dddgd�}dd g}d d
g}|jd d �||f }|j||d d �� d � jksdt�� jd jt|�fks~t�t � fd
d�d"D ��s�t�� j
d#ks�t�|j||d�� d� jks�t�t j j � j
|� |jddgddgd dgd� t j j|jdgdgdgd�jj� |d$ j� � |jddgd� d%d
g}d&dg}|jd d �||f }|j||d�j� t j j � j
|� | j|j||d�|j||d�� | jtd�� |jddgddd gd� W d Q R X | jtd�� |jddgd� W d Q R X | jtd�� |jddgd � W d Q R X | jtd�� |jddd gtd �d� W d Q R X | jtd �� |jddd� W d Q R X | jtd �� |jddgddgdd � W d Q R X |jddgddgd d gd �� d� jk�s�t�d S )'Nr r' r( r# r r )r1 r r% r �
test_coord)r r �dimc 3 s | ]}|� kV qd S )Nr )�.0r )rO r r! � <genexpr>� s z1TestDataArray.test_isel_points.<locals>.<genexpr>)r r Zpointsr$ r& )r# r r )r# )r r z$All indexers must be the same lengthz dimension bad_key does not exist)Zbad_keyzIndexers must be integersg �?g������@)r z Indexers must be 1 dimensionalz Existing dimension names are notra rb )r r' r( )r# r r r� )r� r# )r r& r$ rc rd )r r r r: Z isel_pointsr? �AssertionErrorr3 r< �allr1 ZtestingZ assert_equal�TZassert_allclose�squeezerR rE rF rh r� )r r3 Znp_arrayr� r r r/ r )rO r! �test_isel_points� sZ """ z TestDataArray.test_isel_pointsc C s^ dt jtd��f| jd< | jd }| j|d d� |jd d� � | j|d |jd � | j|d |jddi � | j|d |jd � | j|d d� |jd ddg � | j|d d�d d
�f |jd ddgt jd
�f � | j|d d �d d
�f |jd d �| jd d
k f � d |jd d
�< | jt j|j d k�� d|jdt
d d
�i< | jt j|j dk�� d S )Nr r� r r% r� r r` .r_ r& r r �jr$ )r` .) r rN r> r rR �locrv rw r� r: r� )r r� r r r! �test_loc� s
4zTestDataArray.test_locc C s> t ddgddggd�}| j|jd d� | j|jd d� d S )Nr r TF)r? )r r- r� )r rM r r r! �test_loc_single_boolean� s z%TestDataArray.test_loc_single_booleanc C s� t jjtd�ddgg�}ttd�d|fg�}| j|jd
d�|jdd�� | j|jd d�|jd d�� | j|jd
gd�|jdgd�� | j|jddgd�|jddgd�� | j|jdd�|jt d �d�� d S )N�abcr r r( r r_ )r r� r$ )r_ r )r� r rc )r_ r )r_ r )r� r rc )
rC Z
MultiIndexZ from_productr> r �rangerR r� r� r� )r �idxrM r r r! �test_multiindex� s z TestDataArray.test_multiindexc C sP t jddd�}ttjdd�d|fg�}| j|d |j� | j|jd |j� d S )Nz
2000-01-01r )�periodsr � r# ztime.dayofyear)rC �
date_ranger r rv r9 r: r? )r �datesr� r r r! �test_time_components� s z"TestDataArray.test_time_componentsc C sl t dtjd d
gd��t dtjdddgd��g}ttjjdd�|dd �}| jdt|j�� | j ddgt
|j�� | j |d j |jd �� | j |d j |jd �� | j
d|j� | jd|j� | jd|j� | jt�� |jd W d Q R X | jt�� |jd W d Q R X td
�}t|j�}| j||� | jtd �� |d= W d Q R X | jtd �� |jd= W d Q R X d S )Nr r r$ r, r r r% r )r+ z^ Coordinates:
* x (x) int64 -1 -2
* y (y) int64 0 1 2z
cannot deleterc rd )r r rN r r rg r[ r< r? r- r> rw ry ZassertInZ assertNotInr@ �KeyErrorr r. rE rF )r r? r� r/ rO r r r! � test_coords� s,
zTestDataArray.test_coordsc C s� t ddgddgdddgfdd �dd
�}|jd }t ddgddddgfddgd �ddd �}| j||� |jd
= | j|jd
dd�|� dddgf|jd< t ddgddddgfddgd�ddd �}| j||� d S )Nr r r r$ r r_ r` r& )r r� rW )r1 )rW r� r )r1 r+ r� T)rm �x3)rW r� r )r r? rR rp )r r} rO r/ r r r! �test_coord_coords s
z TestDataArray.test_coord_coordsc C s t tjd�ddddgfdtd�fd�ddgd d
�}|j� }tddgtjd�fddddgfdtd�fd ��}| j||� |jd d
g�}| j||� |jd �}tddgtjd�fddddgfd�d
dtd�fi�}| j||� |jd g�}| j||� |jdd�}t tjd�ddgd d
�}| j||� |j� }|jddd� | j||� |jd dd�}t tjd�d
dtd�fiddgd d
�}| j||� | j t
d�� |jdd� W d Q R X | j t
d�� |jd dd� W d Q R X | j t
d�� |jd� W d Q R X | j t
d�� |jd� W d Q R X d S ) Nr% r& r r_ r` r� r )r) rY r )r1 r+ )r r) rY r) rY )r r) T)rm )rm �inplacezcannot reset coord)r� zcannot be found� not_foundzcannot remove index)r% r& )r% r& )r% r& )r% r& )r% r& ) r r rJ r� rp r �assertDatasetIdenticalrR rK rE rF )r rM rO r/ r r r! �test_reset_coords* sJ
z TestDataArray.test_reset_coordsc C s� t d�}|jdd�}t dddi�}| j||� t ddddgdd d
d
ddgfid d �}|jd�jd d
� d�}|j� }d ddddgf|jd< | j||� d S )Nr ru )r� r� r r$ r% r& r r )r1 c S s | j � S )N)�mean)r_ r r r! �<lambda>` s z2TestDataArray.test_assign_coords.<locals>.<lambda>)r� g �?g @r� )r Z
assign_coordsrR �groupbyrK r? )r rN rO r/ r r r! �test_assign_coordsY s &z TestDataArray.test_assign_coordsc C sv t dddgddddgfg�}t dddgddddgfg�}||jd< t dddgddtjddgfidd�}| j||� d S ) Nr r$ r% r r r& �rhs)r? r1 )r r? r r| rR )r �lhsr� r/ r r r! �test_coords_alignmente s
z#TestDataArray.test_coords_alignmentc C s� | j }| j d d�d d�f }| j|j|�|� |j� }tj|d d �< ||d d�d d�f< | j|j|�|� ttjjd�dgd�}ttj d�dd�}|j
|d�}| j||� d S )Nr$ r' r# )r1 �time2)r# ) r rR � reindex_likerK r r| r r rg rv �reindex)r r r) r/ r� rO r r r! � test_reindexn s zTestDataArray.test_reindexc C s� t ddgdd�}dddg}tjdkr\|j|d d
d �}t dtjtjgd|fgd �}| j||� td|i�}|j|d d
�}t ddtjgd|fgd �}| j ||� d S )Nr r r )r1 g�������?g �?g�������?z0.17r� g�������?)r r� r� )r? )r� g��������)
r rC r� r� r r| rR r r� r� )r r r rO r/ �altr r r! �test_reindex_method~ s
z!TestDataArray.test_reindex_methodc C st | j jd�}| j|j� | jjddi�� | j|jd� | j jddi�}| j|j� | jjddi�� | j|jd� d S )Nr) r )r rz r� �
to_datasetr r- r+ )r �renamedr r r! � test_rename� s zTestDataArray.test_renamec C s` t tjjd�ddtd�fid�}t |jtd�dtd�fd�dd�}|jddi�}| j||� d S )Nr% r r r� )r r )r1 ) r r r rg r> r: r� Z swap_dimsrR )r rN r/ rO r r r! �test_swap_dims� s z TestDataArray.test_swap_dimsc C s | j d }| j|| j� d S )Nr )r rR r )r r r r r! �test_dataset_getitem� s
z"TestDataArray.test_dataset_getitemc C s� | j tj| j�| j� | j | jjt�| jjt�� | j| jj � | jj � � | j| jj
dd�| jj
dd�� t | j �}tj
| j�|d d d �< | j|d tj
| j�� | j| jtj| j| j�� tddgtjd��}| j| jtj| j|�� d S ) Nr$ r% r r r r r )r r )r9 r �asarrayr r �astype�floatr r7 ZargsortZclipr r �sinrQ �maximumr rJ )r r/ r) r r r! �test_array_interface� s
z"TestDataArray.test_array_interfacec C s� t jjd�jdd�}t j||dk < t|t jd� t jd�gddg�}ttj|�t jd� t jd�gddg�}| j ||j� � | j | |j
� � d S )Nru r' r( r r r ) r r � RandomStaterg r| r rv rC ZisnullrR Znotnull)r r �originalr/ r r r! � test_is_null� s "
zTestDataArray.test_is_nullc C s� | j }| j}| j}| j||
� | j||d � | j|d| � | j||d| � | j|d| | � | j||d| � | j|d| | � | j||d| � | j|d| | � d S )Nr )r r r rQ )r r r r_ r r r! � test_math� s zTestDataArray.test_mathc C s� t td�dtd�fg�}t td�dtdd�fg�}t tjd�dddddgfg�}| j|| |� | jtd��( |jtd�d �|jtdd �d � W d Q R X d S )
Nr' r r r( r& r$ r% zno overlapping labels)r ) r r� r rL rR rE rF r� r� )r r_ r` r/ r r r! � test_math_automatic_alignment� s z+TestDataArray.test_math_automatic_alignmentc C s\ | j }| j}| j}|}|d7 }| j||� | j|j|� | j|j|� | jt|j�|� d S )Nr )r r r �assertIsr8 r9 r: r )r r r r_ r` r r r! �test_inplace_math_basics� s z&TestDataArray.test_inplace_math_basicsc C sx t td�dtd�fg�}t tdd�dtdd�fg�}| jtd�� ||7 }W d Q R X | jtd�� ||7 }W d Q R X d S )Nr' r r r( z not aligned)r r� rE rF )r r_ r` r r r! �%test_inplace_math_automatic_alignment� s z3TestDataArray.test_inplace_math_automatic_alignmentc C s� | j }| j|
jd� | j|d jd� | j||jd � jd � | j||jd� jd � | j|| jd� | j|d
jd� | j|d d jd� | j||d jd � d S )Nr r r) r )r r- r+ r� rz )r r_ r r r! �test_math_name� s z TestDataArray.test_math_namec C sn ddgdddgddgdddgdddggfdd
�}t tjjdd�|ddgd �}|d }t |jd |j�}| j||� d| }| j||� ||d }td
d� |j� D ��}t |j|jd |ddgd �}| j||� |d | }| j||� |d |d }t |jd! |jd$ dd%i�}| j||� |d d �d f |d d d �f }|d d �d f jd d �d f |d d d �f jd d d �f }t ||ddgd �}| j||� |d d d �f |d d �d f }| j|j |� ||j }t tj
d&�|j�}| j||� |j | }| j|j |� t ddgd'd(gddd�d�}|| }|d }d|jd< |jd= | j||� || }| j||� d S ))Nr r$ r� �cd�efr r r% i� )r r �latr� )r1 r c s s" | ]\}}|d kr||fV qdS )r� Nr )r� rH r r r r! r� s z6TestDataArray.test_math_with_coords.<locals>.<genexpr>r� r i+ )r r� r� r� rc rd rc rd re i���)r r )r r )r r )r r rc rc )rc rc )r r rc rc )rc rc i���)r$ r% rc rd ) r r r rg r: r? rR �dict�itemsr� rJ )r r? r} rO r/ Z
exp_coordsZ
exp_valuesr� r r r! �test_math_with_coords� sN
<
z#TestDataArray.test_math_with_coordsc C s� t td�ddd�}|d }t dtjd� dtd�fgdd�}| j||� |d |k }t dddgdtd�fgdd�}| j||� ||d k}| j||� d S ) Nr% r )r1 r+ r )r? r+ r FT)r r� r rv rR )r r} rO r/ r r r! �test_index_math( s z TestDataArray.test_index_mathc C s� t dtjd�fddtjd� fd�ddtjd� fdtdd�fd��}d|d }tddtjd� |jd d
�}| j||� |d |d }tdtjd� |j�}| j||� t dd tjd� fdd
tjd� fddtjd� fd��}|d |d }ttjd�|jd d
�}| j||� |d |d }| j||� |d j� }||d 8 }| j||� |j� }|d |d |d < t dtjd�fd|d j fd�|j�}| j
||� |j� }|d |d 8 < | j
||� d S )Nr r' r )�tmin�tmaxg �?r$ r% )r r� r� )r+ r� r r� )r� r� r rd ) r r rv r� r r? rR rL rK r: r� )r ZobsrO r/ �simr r r! �test_dataset_math8 s< z TestDataArray.test_dataset_mathc C sH t ddgddggddgddid�}|jddgd �jd
�}| j||� d S ) Nr r r$ r% r r r )r1 r )rW rW )r �stack�unstackrR )r r} rO r r r! �test_stack_unstackb s "z TestDataArray.test_stack_unstackc C sf t jtd�dddgdddgd��}|jddg�d }t|j� d d
�}t|d d �jd �}| j||� d S )
Nr% r_ r` r r )r r r r r r )r+ rW )r1 )rC rn r� Z set_indexr r� rR )r Zdf�sr/ rO r r r! � test_unstack_pandas_consistencyg s
z-TestDataArray.test_unstack_pandas_consistencyc C s | j | jjj� | jj� � d S )N)r7 r r8 Z transpose)r r r r! �test_transposep s z TestDataArray.test_transposec C s | j | jjj� | jj� � d S )N)r7 r r8 r� )r r r r! � test_squeezet s zTestDataArray.test_squeezec C s� t tjjdd�ddgd�}|j� }d|jd< |jd�}| j||� | jt �� |jd� W d Q R X | j
t d�� |jd � W d Q R X |j d �}| j
t d�� |jd � W d Q R X d S )
Nr$ r% r r )r1 rW z not foundzcannot be foundr ) r r r rg rK r? rm rR r@ rF rE rz )r r/ rS rO r� r r r! �test_drop_coordinatesw s
z#TestDataArray.test_drop_coordinatesc C sP t tjjdd�ddgd�}|jddgdd�}|d d �dd �f }| j||� d S ) Nr$ r% r r )r1 r r )r� )r r r rg rm rR )r rS rO r/ r r r! �test_drop_index_labels� s z$TestDataArray.test_drop_index_labelsc C s� t jjdd�}t j|d d d�df< t|ddgd�}|jd�}|dd d� }| j||� |jddd �}| j||� |jddd
�}| j||� |jdd d
�}|d d �dd �f }| j||� d S ) Nr& r$ r r_ r` )r1 r r� )�how)Zthreshr% )r r rg r| r ZdropnarR )r r rS rO r/ r r r! � test_dropna� s
zTestDataArray.test_dropnac C sv ddgdddgddgdddgdddggfdd
�}t dd dgdd dgg|ddgd �}|j� }t d d
di�}| j||� |jddg�}| j||� |jd�}t dd dg|d dd�d�}| j||� |jdg�}| j||� |jd�}t d d g|d d d�d�}| j||� | j| jjtjd�| jjtjd�� t dd tjgdd dgg|ddgd �}|j � }t dd
d i�}| j||� d S ) Nr r$ r� r� r� r r r% i� )r r r� r� r )r1 r� )r r� )r r� r' rc rd rc rd re i���rc re i���rd i���i���i���)
r r� rR r7 r �reducer r r| �count)r r? r} rO r/ r r r! � test_reduce� s. "
$zTestDataArray.test_reducec C sf | j j� }| jt|j�d� | j|jt� � | j jdd�}| jt|j�t| j�� | j|j| j� d S )Nr T)�
keep_attrs)r r� r- r< r r )r Zvmr r r! �test_reduce_keep_attrs� s
z$TestDataArray.test_reduce_keep_attrsc C s� t tjdtjdgdd�}|jd �}t d
dddgdd�}| j||� t td�dd�}|j|�}|j� }| j||� |jtd��}| j||� |j|d d� �}| j||� |j|d d� �}| j||� | jtd�� |jddi� W d Q R X | jt d�� |jdd g� W d Q R X t ddgd
d�}|j|�}t ddgddgddgddggdd�}| j||� |j� }x(||gD ] }dddddgf|j
d < �qnW |j d �jt dd gd d��}| j||� d S )Nr r% r )r1 r& r zfillna on a DataArrayr r$ r r` rc rc rc )r r ) r r r| ZfillnarR r� rK rE rh rF r? r� )r r_ rO r/ r` Z
fill_value�targetr r r! � test_fillna� s6
$ zTestDataArray.test_fillnac C s� xJt | jjd�| jjd��D ],\\}}\}}| j||� | j|d |� q W x0t | jjd�| j�D ]\\}}}| j||� qbW d S )Nr r r )�zipr r� r r- rR )r Zact_xZact_dv�exp_xZexp_ds�_Zexp_dvr r r! �test_groupby_iter� s * "z TestDataArray.test_groupby_iterc C sN | j j� }dtjdgd dg dgd �f|jd< dd |d |jd< |S )
Nr r_ � r� r` r r� r �d )r rK r rN r? )r r� r r r! �make_groupby_example_array� s
*z(TestDataArray.make_groupby_example_arrayc C s@ | j � jd�}tddddg�}| j||j� | jdt|�� d S )Nr� r_ r` r� r% )r r� r r7 Z unique_coordr- r< )r �groupedZexpected_uniquer r r! �test_groupby_properties� s z%TestDataArray.test_groupby_propertiesc C sx | j � }|jd }dd� }xXddd|gD ]H}xBd
D ]:}x4d D ],}|j||d�}|j||d �}| j||� q<W q2W q(W d S ) Nr c S s | S )Nr )r r r r! �identity s z;TestDataArray.test_groupby_apply_identity.<locals>.identityr r� FT)r� )�shortcut)FT)FT)r r? r� �applyrR ) r r/ r� r �gr r� r rO r r r! �test_groupby_apply_identity s
z)TestDataArray.test_groupby_apply_identityc
s� | j � � � jd�}ttdgtj| jd d �d d�f j� | jd d �dd �f j� | jd d �dd�f j� g�j�tdgtjdddg��d��d }| j ||j
tj�� | j ||j� � t � fd d
�t d�t dd �t dd�gD �dddggdg�}� d jd�j
tj�}| j ||� � d jd�j� }| j ||� td dgtj| jd d �d d�f jd
�| jd d �dd �f jd
�| jd d �dd�f jd
�g�jf| jd tdgtjdddg��d��d }| j ||j
tjd �� | j ||jd �� d S )Nr� r r r_ r` r� )r r� r c s g | ]}� d j | j� �qS )r )r: �sum)r� r� )rN r r! �
<listcomp> s z2TestDataArray.test_groupby_sum.<locals>.<listcomp>r r r )r r r� )r r� r r r rN r r r� �assertDataArrayAllCloser r r� r r )r r Zexpected_sum_allr/ rO Zexpected_sum_axis1r )rN r! �test_groupby_sum s2
" " $""z TestDataArray.test_groupby_sumc
C sn t ddtjtjddgddddddddgfidd�}|jd�j� }t ddd gddddgfgd
�}| j||� d S ) Nr �catr r_ r` r� )r? r1 r r$ )r? )r r r| r� r rR )r rN rO r/ r r r! �test_groupby_count0 s z TestDataArray.test_groupby_countz6needs to be fixed for shortcut=False, keep_attrs=Falsec C s� | j � }d|jd< xjdD ]b}x\d D ]T}td||f � |jd�jtj||d�}|jd�j� }|rjd|jd< | j||� q"W qW d S )
Nr) r TFzshortcut=%s, keep_attrs=%sr� )r r )TF)TF)r r �printr� r r r� rR )r rN r r rO r/ r r r! �test_groupby_reduce_attrs8 s
z'TestDataArray.test_groupby_reduce_attrsc C s� dd� }| j � }|jd�}|j� }tj|| jd d �d d�f �|| jd d �dd�f �|| jd d �dd �f �g�}ddg|f|d< |d }| j||j|�� d S ) Nc S s | t j| � S )N)r r� )r r r r! �centerH s z7TestDataArray.test_groupby_apply_center.<locals>.centerr� r r r r r )r r� r� r Zhstackr r r )r r rN r Z expected_dsZexp_dataZexpected_centeredr r r! �test_groupby_apply_centerG s
z'TestDataArray.test_groupby_apply_centerc C s. | j � }|jd�}|jtj�}| j||� d S )Nr� )r r� r r r� rQ )r rN r rO r r r! �test_groupby_apply_ndarrayV s
z(TestDataArray.test_groupby_apply_ndarrayc C sD dd� }| j � }|jd�}|j|�}|j� }||�}| j||� d S )Nc S s" | j d d | j d< d| jd< | S )Nr r$ ZlemonZfruit)r? r )r r r r! �change_metadata^ s
zJTestDataArray.test_groupby_apply_changes_metadata.<locals>.change_metadatar� )r r� r rK rQ )r r# rN r rO r/ r r r! �#test_groupby_apply_changes_metadata] s
z1TestDataArray.test_groupby_apply_changes_metadatac C sd | j � }x�dD ]�}|jd|d�}||jd }||jd }| j||� |jd | }| j||� |jd jd�}|| }|| }| j||� || }| j||� qW |jd�}|j� tjd� j d �}|t
t d�ddd d
gfg� }|jd�j� }| j ||� | j
td �� |d W d Q R X | j
td �� || W d Q R X | j
td
�� ||7 }W d Q R X d S )NTFr )r� �Xr� r% r_ r` r� zonly support binary opsr zin-place operations)TF)r r� r? rR r� r� r� r rv rz r r� r rE rh ) r rN r� r r/ rO r Z expected_aggZ
actual_aggr r r! �test_groupby_mathj s2
z TestDataArray.test_groupby_mathc C s t td�ddddddgfidd�}t dgdd�}|jd�| }t ddtjtjg|j�}| j||� t dgd d
idd �}|jd�| }dgd
gd tjgd f|jd < | j||� td
ddgfi�}|jd�| }td
dddtjtjgfi|j�}| j||� d S )Nr& r` r r r )r1 r r� r� r )r? r1 r$ r_ ) r r� r� r r| r? rR r r� )r rN r* rO r/ r r r! � test_groupby_math_not_aligned� s " " z+TestDataArray.test_groupby_math_not_alignedc C sj t tjjdd�dtd�fdtd�fd�ddgd�}x2dD ]*\}}|j|�jd d
� �}| j|j|� q8W d S )Nr' r% r r )r_ r` )r? r1 r_ r` c S s | j � S )N)r� )r r r r! r� � s z>TestDataArray.test_groupby_restore_dim_order.<locals>.<lambda>�r r �r r( �r r �r r* �r_ r �r_ r, �r r` �r` r. )r) r+ r- r/ ) r r r rg r� r� r r- r1 )r rN �byZ
expected_dims�resultr r r! � test_groupby_restore_dim_order� s z,TestDataArray.test_groupby_restore_dim_orderc C s t dddddgdd�}t dgd d gd dd
d �}t ddgd
dd gfg�}|j|�j� }| j||� t ddgd
dd gfg�}|j|�j� }| j||� t tjjdd�dd gd�}t |d
dg d
dd gid
d g�}|j|�j� }| j||� |jd�j� }|}| j||� d S )Nr r$ r% r& r' r )r1 r_ r` r� )r1 r+ r r )r r� �firstrR �lastr r rg )r rN r0 r/ rO r r r! �test_groupby_first_and_last� s z)TestDataArray.test_groupby_first_and_lastc C s� t jdddd�}ttjd�d|fg�}|jddd�}| j||� |jddd�}t|j� jd��}| j||� |jddtjd�}| j||� | j t
d �� |d
d d g jd
dd� W d Q R X d S )Nz
2000-01-01�6Hr )�freqr� r# )r� �24H)r� r zindex must be monotonicr$ r r �1D) rC r� r r rv �resamplerR � to_seriesr� rE rF )r �timesrN rO r/ r r r! �
test_resample� s zTestDataArray.test_resamplec C s� t jdddd�}ttjd�d|fg�}|jdddd�}td d
d gd|d d d
� fg�}| j||� |jd ddd�}t|j� jd dd
��}| j||� |jt �}tj
|d d�< |jdddd�}tdd
d gd|d d d
� fg�}| j||� |jddddd�}ttj
d
d gd|d d d
� fg�}| j||� t d|i�d }|jdddd�}t j dddg�}t|d|d d d
� fgdd�}| j||� d S )Nz
2000-01-01r6 r )r7 r� r# r9 r3 )r� r r r& � r8 )r r$ F)r� r �skipnar4 z
2000-01-01T18z
2000-01-02T18z
2000-01-03T06)r+ )
rC r� r r rv r: rR r; r� r� r| r Z to_datetime)r r< rN rO r/ Zexpected_timesr r r! �test_resample_first� s.
" z!TestDataArray.test_resample_firstc C sr t jdddd�}ttjd�d|fg�}tj|d< |jdddd �}ttjddgd|d d d
� fg�}| j||� d S ) Nz
2000-01-01r6 r )r7 r� r# r r9 F)r� r? r& )rC r� r r rL r| r: rR )r r< rN rO r/ r r r! �test_resample_skipna� s
"z"TestDataArray.test_resample_skipnac C s~ t jdddd�}ttjd�d|fg�}t jdddd�}|j|d�}x6d d
d d d
tjgD ] }|jdd|d�}| j||� qVW d S )Nz
2000-01-01r9 r' )r7 r� r# Z12Hr )r# r� Zmedianr r3 r4 )r ) rC r� r r rv r� r� r: rR )r r< rN Z
expected_timer/ r rO r r r! �test_resample_upsampling� s z&TestDataArray.test_resample_upsamplingc C sf dt jtd��f| jd< t| j| jd d� dd�\}}| j|| jd d� � | j|| jd d� � d S )Nr r� r' �inner)�join)r rN r> r r r rR )r Zdv1Zdv2r r r! �
test_align s zTestDataArray.test_alignc C sx t jd�}t jdd�}tt jjd�jt j�d|i�}tt jjd�jt j�d|i�}t||dd�\}}| j|jt j� d S ) N� r' �# r �outer)rD )rF )rF ) r rv r r r� Zfloat32r r- r2 )r Zx1r� r_ r` r� r� r r r! �test_align_dtype s
z TestDataArray.test_align_dtypec
C sN t ddgdddgfgdd�}t ddgdddgfgd d�}t||�\}}dddgfdddgfg}t ddgddgg|dd
�}t ddgddgg|d d
�}| j||� | j||� t tjjdd�ddgd �}t tjjdd�ddgd �}t||�\}}|}|j}| j||� | j||� t ddgddd
gfgd�}| jtd�� t||� W d Q R X d S )Nr r$ r_ r )r? r+ r` r% r& r )r+ )r1 r r )r? zcannot broadcastrc rd rc rd i����) r r rR r r rg r� rE rF ) r r r r� r� �expected_coordsZ expected_x2Z expected_y2rW r r r! �test_broadcast_arrays s$ z#TestDataArray.test_broadcast_arraysc C s� t dddgtjd �fi�}t|j|j|j�\}}}| j|j|� tjtj d�tj d�dd�\}}t
|ddgdd�}t
|ddgdd�}| j||� | j||� d S )
Nr_ r r r' r( Zij)Zindexing)r1 r+ )r' r( ) r r rL r r r r_ rR Zmeshgridrv r ) r r Zx_bcZy_bcZa_bcr% �Yr Zexp_yr r r! �test_broadcast_coordinates' s z(TestDataArray.test_broadcast_coordinatesc C sz t d�j� }tjd�}| j||� tjjd�}tjdddgdd�}t ||gd�}|j� }| j|j |� | j|j
|� | j|j
j d� tjjdd �}t |ddddgfd
d d gfgd
d�}|j� }| j|j |� | j|j
dddg� | j|j d d g� xNdD ]F}t
d�d t|�� }t tjj|� |d�}t |j� �}| j||� � q�W | jtd��" t tjjd d ddd��j� W d Q R X d S )Nru r% r_ r` r� r )r+ )r? r$ r r r r )r? r+ r& r' r� )r1 zcannot convert�r% �r% r& �r% r& r' )rN rO rP )r � to_pandasr rN r9 r rg rC rD r: rk r+ rl r> r< rR rE rF ) r rO r/ r: rk r� r3 r1 r� r r r! �test_to_pandas3 s0
z TestDataArray.test_to_pandasc C s. t tjjdd�ddddgfdtd�fgdd �}|j� }|j� d }| j|j|j� | j|j |j � | j|j
j|j
j� dd d
dgf|j d
< |j� j � }dgd dgd dgd |d
< |d
dg }|j� }| j|j|j� | j|j
j|j
j� | j|j
j|j
j� d |_ | jtd �� |j� W d Q R X d S )Nr% r& rb r r$ ra Zcdefr )r+ �C�unnamedrc rd re rc rd re )r r r rg r> r; � to_dataframer9 r: r+ rk r? �to_framerl rE rF )r rS r/ rO r r r! �test_to_dataframeU s$ " z TestDataArray.test_to_dataframec C s� t dddgddd�}|j� }| jdddg|j� | jdddg|jj� | jd|j� | jd|jj� |j� }|j� }| j |j
|�� d S )Nr r$ r% r )r1 r+ r ) r r; r9 r: rk r- r+ rU rV rw rx )r rS rr rq r/ r r r! �&test_to_pandas_name_matches_coordinatem s z4TestDataArray.test_to_pandas_name_matches_coordinatec C s� | j j� d }| j j� }| j|j|j� | j|jj|jj� | jd|j� | j| j t j
|�� d |_| j j d �}| j|t j
|�� d S )Nr ) r rU r; r9 r: rk r- r+ rR r Z from_seriesrz )r r/ rO Z expected_dar r r! �test_to_and_from_seriesz s
z%TestDataArray.test_to_and_from_seriesc C sL t td�stjd��tjtd�tjtd��d�}t|�}dt |�ksHt
�d S )N�CategoricalIndexz%requires pandas with CategoricalIndexr' Zaabbc)rk z'a') �hasattrrC r �SkipTestro r� rZ r> r r. r� )r r� rS r r r! � test_series_categorical_index� s
z+TestDataArray.test_series_categorical_indexc C sz t jjd�}|jdd�}t jj|dk |�}t|�}|j� }t|�}| j||� | j ||� |jdd�}| j
t |t jj �� | j|j
|j
� | j|j|jt j�� | j||jt j�� |jdd�}d |d< | j|d d � | j|d jd � | j
|j|jk� | j|t jj � x:d t jdd d
gD ]&} t| �}|j� }
| j|
t jj � �q W d } t| �} t| �}|j� }
| jt|
j
�| � d S )N�, r r )r4 g �?T)rK Fg $@r r& �4Zfour)r r )r r )r r )r r )r r r� Z
random_sample�maZ masked_wherer Zto_masked_arrayr9 rQ rw �
isinstanceZ MaskedArray�maskr: Zfilledr| r- �baserB r� r< ) r �rsr Zx_maskedr� Z
x_masked_2Zda_2Zda_masked_arrayZ masked_arrayr r` �Nr r r! �test_to_masked_array� s8 z"TestDataArray.test_to_masked_arrayc C s\ y dd l }W n tk
r* tjd��Y nX ttjd�jdd�dddgddifd tj d
dd �fgd d
did�}t
dddg�t
d dddg�g}|j � }| j ||� | j
|j|j� | j|j� |j� x8t|j� |�D ]&\}}| j
|j|j� | j ||j� q�W | j
|j|jd
� |jd�j� }| j
t|�d� | j
t|d �d� tj |�}| j ||� d S )Nr zcdms2 not installedr( r$ r% ZdistanceZunitsZmetersr# z
2000-01-01)r� r rY r )r+ r r z2000-1-1 0:0:0.0rd rd ) �cdms2� ImportErrorr r\ r r rv ZreshaperC r� r Zto_cdms2r9 r- r� r+ r= Z
getAxisIdsr1 r
Z getAxisListr: rY r ZgetAxisZasComponentTimer< �strZ
from_cdms2rR ) r rg r� rJ rO �axisZcoordZcomponent_timesr� r r r! �test_to_and_from_cdms2� s.
z$TestDataArray.test_to_and_from_cdms2c C s� t ddgdd�}| jtd�� |j� W d Q R X |jdd�}tddddgfi�}| j||� t ddgddd�}|j� }tddddgfi�}| j||� td dddgfi�}| jd
�� |jd �}W d Q R X | j||� d S ) Nr r$ r )r1 zunable to convert unnamedr )r+ )r1 r+ r) zorder of the arguments)r rE rF r� r r� Z assertWarns)r rT rO r/ �namedr r r! �test_to_dataset_whole� s z#TestDataArray.test_to_dataset_wholec
C s� t dddgdtd�fgddid�}ttdddg�ddid
�}|jd�}| j||� | jt�� |jdd d � W d Q R X |jdd
�}| j ||� t dddgdd�}ttdddg��}|jd�}| j||� d S )Nr r$ r% r r� r_ )r? r r` r� )r r )r+ )r� )r1 �0�1�2)r_ r )r` r$ )r� r% )rn r )ro r$ )rp r% )
r r> r r r� r� r@ rh Zto_arrayrR )r rN r/ rO r� r r r! �test_to_dataset_split� s
z#TestDataArray.test_to_dataset_splitc C s� t tjd�dddgd�}| jd|j� � | jd |jd
d �j� � |jd d
d
�j� }| jd|kpdd|k� t tjd�ddgd�}| jd|j� � d S )Nr& r% r$ r_ r` r� )r1 � zc = 0r )r� r )r` r� z b = 1, c = 0z c = 0, b = 1zb = [0])r& r% r$ )r& r )r r rL r- �_title_for_slicer� rw )r rN �titleZa2r r r! �test__title_for_slice� s z#TestDataArray.test__title_for_slicec C s^ t tjd��}dd |jd< dd |jd< d}|j|d�}| j|t|�� | j|jd�� d S )Nr& r_ r r` �P )�truncatez...) r r rL r? rs r- r< rw �endswith)r rN Zncharrt r r r! � test__title_for_slice_truncate s z,TestDataArray.test__title_for_slice_truncatec C sX | j d }|jd�}ttj|jdd�|d j|d jdd � gddg�}| j||� d S )Nr r r )rj r )r �diffr r r: rQ )r r� rO r/ r r r! �test_dataarray_diff_n1 s
z$TestDataArray.test_dataarray_diff_n1c C sV t tddd�dgtd�gd�}|jd }t dgd dgtdd�gdd �}|jd�}d S )
Nr r r$ �lonr )r1 r? r r )r1 r? r+ )r r� r? rz )r rS r| r/ rO r r r! �test_coordinate_diff s
z"TestDataArray.test_coordinate_diffc C sz t dddgdd�}|jdd�}t tjddgdd�}| j||� x6d D ].}t |j� j|��}|j|d�}| j||� qDW d S )
Nr r$ r% r )r1 )r r' r ���rd rc )r~ rd rc r r r$ r' )r �shiftr r| rR rQ )r rS rO r/ �offsetr r r! �
test_shift" s
zTestDataArray.test_shiftc C sL t dddgdd�}|jdd�}t dddgddddgfgd�}| j||� d S ) Nr r$ r% r )r1 )r r )r? )r ZrollrR )r rS rO r/ r r r! � test_roll- s zTestDataArray.test_rollc C s0 t d�}| j|jt d�� | j|jt d�� d S )Nr y @r$ y �? @)r rR �real�imag)r rN r r r! �test_real_and_imag3 s z TestDataArray.test_real_and_imagc C st t dddiddid�}| jtd�� d|_W d Q R X | jtd�� d|_W d Q R X | jtd�� d|_W d Q R X d S ) Nr �scalarr r r) )r? r zcannot set attrr$ )r rE rA r� r r* )r rN r r r! �test_setattr_raises8 s z!TestDataArray.test_setattr_raisesN)f�__name__�
__module__� __qualname__r" r0 rI rP rU rX r\ rf ri rs rt 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 �skipr r! r" r$ r& r' r2 r5 r= r@ rA rB rE rI rK rM rR rW rX rY r] rf rk rm rq ru ry r{ r} r� r� r� r� r r r r! r
s�
A,- !
< #/
4* %
! !
"
*
r )Znumpyr ZpandasrC r� rK r �textwrapr Zxarrayr r r r r r Zxarray.core.pycompatr
r rr r
r r r r r r r r r! �<module> s |