3
���h � @ s� d dl Zd dlZd dlmZmZ d dlmZ ddl m
Z
G dd� de
�Z G dd � d e
�Z G d
d � d e
�Z
G d d
� d
e
�ZG dd� de
�ZdS )� N)�ops�utils)� OrderedDict� )�TestCasec @ s e Zd Zdd� ZdS )�TestSafeCastToIndexc C s� t jddd�}tjd�}|tjdd� }xv||jft j|td�|jt�ft j|�|ft j|td�|jt�fgD ].\}}t j
|�}| j ||� | j |j
|j
� qnW d S )Nz
2000-01-01�
)Zperiods� r �D)�dtype)�pdZ
date_range�np�arangeZ timedelta64�values�Index�objectZastyper Zsafe_cast_to_indexZassertArrayEqual� assertEqualr )�self�dates�x�td�expected�array�actual� r �6/tmp/pip-build-5_djhm0z/xray/xarray/test/test_utils.py�test
s
"
zTestSafeCastToIndex.testN)�__name__�
__module__� __qualname__r r r r r r s r c @ s e Zd Zdd� ZdS )�TestArrayEquivc C s\ | j tjdtjdtd��� | j tjtjtjtjtd��� | jtjdtjdtd��� d S )Nr )r r )�
assertTruer Z array_equivr
r r �nan� assertFalse)r r r r �test_0d s
zTestArrayEquiv.test_0dN)r r r r$ r r r r r s r c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�TestDictionariesc C s& ddd�| _ ddd�| _ddi| _d S )N�A�B)�a�b�C)�cr) r( �Z)r �y�z)r r r r �setUp% s zTestDictionaries.setUpc C s� | j tjdd�� | j tjtjtj�� | j tjdtjd��� | j tjdgtjdg��� | j tjtjdg�dg�� | j tjtjd�dtjd� �� | jtjdtjd��� d S )Nr g � g �?) r! r Z
equivalentr
r" r r r# �zeros)r r r r �test_equivalent* s "z TestDictionaries.test_equivalentc C s t j| j| j� d S )N)r �update_safety_checkr r- )r r r r � test_safe3 s zTestDictionaries.test_safec
C s* | j t�� tj| j| j� W d Q R X d S )N)� assertRaises�
ValueErrorr r3 r r. )r r r r � test_unsafe7 s z TestDictionaries.test_unsafec C s8 | j dditj| j| j�� | j i tj| j| j�� d S )Nr) r' )r r Zordered_dict_intersectionr r- r. )r r r r � test_ordered_dict_intersection; s
z/TestDictionaries.test_ordered_dict_intersectionc C s> t � }d|d< tjdddg�|d< t � }tjdddg�|d< d|d< | jtj||�� dddg|d< | jtj||�� dddg|d< | jtj||�� d |d < | jtj||�� tj|d < tj|d < | jtj||�� tj|d < tj|d < | jtj||�� t |�}| jtj||�� dtj
d� |d< | jtj||�� d S )
Nr0 r( r � r) g �?g @g @r+ ) r r
r r! r Z
dict_equivr# r" �inf�dictr )r r r- r r r �test_dict_equiv@ s.
z TestDictionaries.test_dict_equivc C s� t j| j�}| jt�� d|d< W d Q R X | jt�� |d= W d Q R X | jt�� |j| j� W d Q R X | j|j | j� | j
t |�d� d S )N�barZfoor( � Frozen({'a': 'A', 'b': 'B'})� Frozen({'b': 'B', 'a': 'A'}))r> r? ) r ZFrozenr r5 � TypeError�AttributeError�updater- r �mapping�assertIn�repr)r r r r r � test_frozenY s
z TestDictionaries.test_frozenc C s@ dddd�}t j|�}| j|dddg� | jtt j� �d� d S ) Nr r9 r0 )r( r) r+ r( r) r+ zSortedKeysDict({}))r ZSortedKeysDict�assertItemsEqualr rE )r r r- r r r �test_sorted_keys_dicte s
z&TestDictionaries.test_sorted_keys_dictc C s� t jddd�d
dd��}| jd|� | jd|� | jd |� | j|d d� | j|d d� | j|d d� d|d< | j|d d� | j|jd d d� | jddd g|� d S ) Nr r )r r- �d r9 )r r. r r- r. i����)r �ChainMaprD r �mapsrG )r �mr r r �test_chain_mapl s z TestDictionaries.test_chain_mapN) r r r r/ r2 r4 r7 r8 r< rF rH rM r r r r r% $ s r% c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�Test_is_uniform_and_sortedc C s | j tjtjd��� d S )Nr )r! r �is_uniform_spacedr
r )r r r r �test_sorted_uniform| s z.Test_is_uniform_and_sorted.test_sorted_uniformc C s | j dtjdddg�� d S )NFr9 r �Y ���)r r rO )r r r r �test_sorted_not_uniform s z2Test_is_uniform_and_sorted.test_sorted_not_uniformc C s | j dtjdddg�� d S )NFr r0 ���)r r rO )r r r r �test_not_sorted_uniform� s z2Test_is_uniform_and_sorted.test_not_sorted_uniformc C s | j dtjdddg�� d S )NF� r rQ )r r rO )r r r r �test_not_sorted_not_uniform� s z6Test_is_uniform_and_sorted.test_not_sorted_not_uniformc C s | j tjddg�� d S )Nr g333333�?)r! r rO )r r r r �test_two_numbers� s z+Test_is_uniform_and_sorted.test_two_numbersc C s | j tjdddgdd�� d S )Nr g
ףp=
�?r9 g�������?)Zrtol)r! r rO )r r r r �test_relative_tolerance� s z2Test_is_uniform_and_sorted.test_relative_toleranceN) r r r rP rS rU rW rX rY r r r r rN z s rN c @ s e Zd Zdd� ZdS )�
Test_hashablec C sR x dD ]}| j tj|�� qW x.ddgd d
gd d igD ]}| jtj|�� q6W d S )NFr r9 r0 rV �fourr � Zseven�8� Zten�r9 �r0 rV )Fr r_ r` r[ )r! r Zhashabler# )r �vr r r �
test_hashable� s
zTest_hashable.test_hashableN)r r r rb r r r r rZ � s rZ )Znumpyr
Zpandasr Z xarray.corer r Zxarray.core.pycompatr � r r r r% rN rZ r r r r �<module> s V |