3
���h � @ sH d dl Zd dlZd dlmZ d dlmZ ddlm Z G dd� de �Z
dS )� N)�
formatting)�PY3� )�TestCasec @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�TestFormattingc C s� d t d�ffd
dt d�ffd dt d�ffd t d�t d �ffd
dt d�t d �ffddt d�t d �ffddt d�t d �ffdt d�t d �t d �ffddt d�t d �ffdt d�t d �t d �ffg
}x(|D ] \}}tj|d�}| j||� q�W d S )N� �
� 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 )�slicer Z _get_indexer_at_least_n_items� assertEqual)�self�cases�shape�expected�actual� r �;/tmp/pip-build-5_djhm0z/xray/xarray/test/test_formatting.py�!test_get_indexer_at_least_n_items s z0TestFormatting.test_get_indexer_at_least_n_itemsc
C sp t jd�jddd�}x2d
D ]*}tj||�}|jd |� }| j||� qW | jtd�� tj|d � W d Q R X d S ) N�d r r r
r �
�� zat least one itemr )r r r r r ) �np�arangeZreshaper Z
first_n_itemsZflatZassertItemsEqualZassertRaisesRegexp�
ValueError)r �array�nr r r r r �test_first_n_items s
z!TestFormatting.test_first_n_itemsc
C s� t jd�dft jd�dft jd�dft jd�dft jd�dft jd �d
ft jd�dfdd tr`d nd
fdtrndnd fddg }x&|D ] \}}tj|�}| j||� q�W d S )Nz
2000-01-01T12z2000-01-01T12:00:00z
2000-01-01�NaTz10 days 1 hourz10 days 01:00:00z-3 daysz-3 days +00:00:00z3 hoursz0 days 03:00:00�foo�'foo'zu'foo's foozb'foo'r �1� �?�1.0)r! r" )r r# )r$ r% )�pdZ TimestampZ Timedeltar r Z format_itemr )r r �itemr r r r r �test_format_item' s
z TestFormatting.test_format_itemc C s� t jd�t jdd� dft jd�t jdd� dft jd�t jdd � d
ftjd d d
d g�dftjdddg�dfdddgdfg}x,|D ]$\}}djtj|��}| j||� q�W d S )Nr r �Dz0 days 1 days 2 days 3 daysr �hz#00:00:00 03:00:00 06:00:00 09:00:00i� �msz100:00:00 00:00:00.500000 00:00:01 00:00:01.500000r Z0sZ1szNaT 00:00:00 00:00:01 NaTz 1 day 1 hourz1 dayz0 hoursz/1 days 01:00:00 1 days 00:00:00 0 days 00:00:00r
z1 2 3� ) r r Z timedelta64r& Z to_timedelta�joinr Z format_itemsr )r r r' r r r r r �test_format_items: s z TestFormatting.test_format_itemsc C s� t jtjd�d�}d}| j||� t jtjd�d�}d}| j||� t jtjd�d�}d}| j||� t jtjd �d
�}d }| j||� t jtjd �d�}d}| j||� t jtjd
�d�}d}| j||� d S )Nr r z
0 1 2 3 4 ...g Y@� z 0.0 1.0 ...r z0.0 ...r r z0 1 2g @r r z0 ...)r Zformat_array_flatr r r )r r r r r r �test_format_array_flatM s$ z%TestFormatting.test_format_array_flatc C s | j tjdd�d� d S )NZ
abcdefghij� zabcde...)r r Z pretty_print)r r r r �test_pretty_printf s z TestFormatting.test_pretty_printN) �__name__�
__module__� __qualname__r r r( r. r0 r2 r r r r r
s
r ) Znumpyr Zpandasr& Z xarray.corer Zxarray.core.pycompatr � r r r r r r �<module> s
|