3
���h� � D @ sv d dl Z d dlZd dlmZ d dlZd dlmZ d dlm Z m
Z
m Z d dl m
Z
d dlmZ y d dlZW n ek
r� d dlZY nX yd dlZdZW n ek
r� dZY nX yd dlZdZW n ek
r� dZY nX yd dlZdZW n ek
�r dZY nX yd dlZdZW n ek
�r0 dZY nX yd dl Z dZ W n ek
�r\ dZ Y nX y&d dl Z d dl Z e j e j!d � dZ"W n ek
�r� dZ"Y nX yd dl#Z#dZ$W n ek
�r� dZ$Y nX d
d � Z%d d
� Z&dd� Z'dd� Z(dd� Z)dd� Z*dd� Z+dd� Z,dd� Z-d d � Z.d,d d!�Z/G d"d#� d#ej0�Z0G d$d%� d%e1�Z2G d&d'� d'e j3�Z4G d(d)� d)e5�Z6d*d+� Z7dS )-� N)�contextmanager)�assert_array_equal)�utils�nputils�ops)� as_variable)�PY3TF)�getc C s t r| S tjd�| �S )Nzrequires scipy)� has_scipy�unittest�skip)�test� r �4/tmp/pip-build-5_djhm0z/xray/xarray/test/__init__.py�requires_scipyB s r c C s t r| S tjd�| �S )Nzrequires pydap.client)� has_pydapr r )r
r r r �requires_pydapF s r c C s t r| S tjd�| �S )Nzrequires netCDF4)� has_netCDF4r r )r
r r r �requires_netCDF4J s r c C s t r| S tjd�| �S )Nzrequires h5netcdf)� has_h5netcdfr r )r
r r r �requires_h5netcdfN s r c C s t r| S tjd�| �S )Nzrequires pynio)� has_pynior r )r
r r r �requires_pynioR s r c C s t str | S tjd�| �S )Nzrequires scipy or netCDF4)r
r r r )r
r r r �requires_scipy_or_netCDF4V s r c C s t r| S tjd�| �S )Nz
requires dask)�has_daskr r )r
r r r �
requires_dask[ s r c C s t r| S tjd�| �S )Nzrequires matplotlib)�has_matplotlibr r )r
r r r �requires_matplotlib_ s r c C sB t jd }t jd }d}|dkr,|dkr,d}|s4| S tjd�| �S )z'
Test won't work in Python 2.6
r � F� � Tzerror on Python 2.6)�sys� version_infor r )r
�major�minorZpy26r r r �incompatible_2_6c s
r% c C s" | j jdkr tjjj| dd�S | S )N�Szutf-8�replace)�dtype�kind�np�coreZ defchararray�decode)�datar r r �decode_string_datap s r. ��h㈵��>�:�0�yE>c sn t dd� | |gD ��r&t| �} t|�}ddddg� t � fdd�| |gD ��rXtj| |�S tj| |||d�S d S ) Nc s s | ]}|j jd kV qdS )r& N)r( r) )�.0�arrr r r � <genexpr>w s z)data_allclose_or_equiv.<locals>.<genexpr>�M�m�O�Uc 3 s | ]}|j j� kV qd S )N)r( r) )r1 r2 )� exact_dtypesr r r3 { s )�rtol�atol)�anyr. r Z array_equivZallclose_or_equiv)Zarr1Zarr2r9 r: r )r8 r �data_allclose_or_equivv s r< c @ s� e Zd Zerd"dd�Zedd� �Zdd� Zdd � Zd#d d
�Z dd� Z
dd� Z dd� Z dd� Z
dd� Zd$dd�Zdd� Zd d � Zd d � Zd%d d!�ZdS )&�TestCaseNc C s | j |||�S )N)�assertCountEqual)�self�first�second�msgr r r �assertItemsEqual� s zTestCase.assertItemsEqualc # sX t jdd��B}t jd� � d V t|�dks0t�t� fdd�|D ��sJt�W d Q R X d S )NT)�record�alwaysr c 3 s | ]}� t |j�kV qd S )N)�str�message)r1 Zwi)rG r r r3 � s z'TestCase.assertWarns.<locals>.<genexpr>)�warnings�catch_warnings�filterwarnings�len�AssertionError�all)r? rG �wr )rG r � assertWarns� s
zTestCase.assertWarnsc C s t |�j|�st||f��d S )N)r �equalsrL )r? �v1�v2r r r �assertVariableEqual� s z TestCase.assertVariableEqualc C s t |�j|�st||f��d S )N)r � identicalrL )r? rQ rR r r r �assertVariableIdentical� s z TestCase.assertVariableIdentical��h㈵��>�:�0�yE>c C s< | j |j|j� t|j|j||d�}|s8t|j|jf��d S )N)r9 r: )� assertEqualZdimsr< �valuesrL )r? rQ rR r9 r: Zallcloser r r �assertVariableAllClose� s z TestCase.assertVariableAllClosec C s | j t|�j|�� d S )N)Z assertFalser rP )r? rQ rR r r r �assertVariableNotEqual� s z TestCase.assertVariableNotEqualc C s t ||� d S )N)r )r? �a1�a2r r r �assertArrayEqual� s zTestCase.assertArrayEqualc C s ||ks ||kr||ks t �d S )N)rL )r? r\ r] r r r rX � s zTestCase.assertEqualc C s |j |�st||f��d S )N)rP rL )r? �d1�d2r r r �assertDatasetEqual� s zTestCase.assertDatasetEqualc C s |j |�st||f��d S )N)rT rL )r? r_ r` r r r �assertDatasetIdentical� s z TestCase.assertDatasetIdenticalc C sd | j t|td�t|td�� | j|j|j� x2|D ]*}|j| }|j| }| j||||d� q2W d S )N)�key)r9 r: )rX �sortedrF rC �coords� variablesrZ )r? r_ r` r9 r: �krQ rR r r r �assertDatasetAllClose� s
z TestCase.assertDatasetAllClosec C sL | j t|j�t|j�� x.|jD ]$}|j| }|j| }| j||� q W d S )N)rX rd re rS )r? r_ r` rg rQ rR r r r �assertCoordinatesEqual� s
z TestCase.assertCoordinatesEqualc C s | j ||� | j||� d S )N)rS ri )r? �ar1�ar2r r r �assertDataArrayEqual� s z TestCase.assertDataArrayEqualc C s( | j |j|j� | j|j� |j� � d S )N)rX �namerb Z_to_temp_dataset)r? rj rk r r r �assertDataArrayIdentical� s
z!TestCase.assertDataArrayIdenticalc C s" | j ||||d� | j||� d S )N)r9 r: )rZ ri )r? rj rk r9 r: r r r �assertDataArrayAllClose� s z TestCase.assertDataArrayAllClose)N)rV rW )rV rW )rV rW )�__name__�
__module__� __qualname__r rC r rO rS rU rZ r[ r^ rX ra rb rh ri rl rn ro r r r r r= � s
r= c @ s e Zd ZdS )�UnexpectedDataAccessN)rp rq rr r r r r rs � s rs c @ s e Zd Zdd� Zdd� ZdS )�InaccessibleArrayc C s
|| _ d S )N)�array)r? ru r r r �__init__� s zInaccessibleArray.__init__c C s t d��d S )NzTried accessing data)rs )r? rc r r r � __getitem__� s z InaccessibleArray.__getitem__N)rp rq rr rv rw r r r r rt � s rt c @ s e Zd Zdd� ZdS )�
ReturnItemc C s |S )Nr )r? rc r r r rw � s zReturnItem.__getitem__N)rp rq rr rw r r r r rx � s rx c C s$ t | dtj| �j�}|dkr | }|S )z_Given an ndarray, return the base object which holds its memory, or the
object itself.
�baseN)�getattrr* Zasarrayry )ru ry r r r �source_ndarray� s r{ )r/ r0 )8r! rH �
contextlibr Znumpyr* Z
numpy.testingr Z xarray.corer r r Zxarray.core.variabler Zxarray.core.pycompatr Z unittest2r � ImportErrorZscipyr
Z pydap.clientZpydapr ZnetCDF4r Zh5netcdfr ZNior Z
dask.arrayZdaskZ set_optionsr r Z
matplotlibr r r r r r r r r r% r. r< r= � Exceptionrs Z NDArrayMixinrt �objectrx r{ r r r r �<module> s�
N |