HOME


sh-3ll 1.0
DIR:/usr/local/lib/python3.6/site-packages/xarray/test/__pycache__/
Upload File :
Current File : //usr/local/lib/python3.6/site-packages/xarray/test/__pycache__/test_indexing.cpython-36.pyc
3

���h|�@s`ddlZddlZddlmZmZddlmZddl	m
Z
mZGdd�de
�ZGdd	�d	e
�Z
dS)
�N)�Dataset�Variable)�indexing�)�TestCase�
ReturnItemc@sDeZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dS)�TestIndexerscCs|j�}d||<|S)Nr)�copy)�self�x�i�r
�9/tmp/pip-build-5_djhm0z/xray/xarray/test/test_indexing.py�set_to_zero
szTestIndexers.set_to_zerocCs�tjjddddd�}tjd�}t�}x�|dd�|d|ddd�df|d|dd�ddf|d|||||f|d||f|dg
D]B}tj||j�}|j||||�|j|j	||�|j	||��q�W|j
td
��tj|dd
�WdQRXdS)N�
���
��.rr���ztoo many indices).r).r.).rrrrr)rrr)�np�random�randn�arangerrZexpanded_indexer�ndim�assertArrayEqualr�assertRaisesRegexp�
IndexError)r
r�y�Ir�jr
r
r�test_expanded_indexers
8"z"TestIndexers.test_expanded_indexerc	Cs*tjjddddd�}tjd�}t�}�xX|dd�|d|d|dd�|dd�|dd�|d	d�|ddd�|ddd�|ddd�|dd	d�|ddd�|dd	d�|dd	�dd�f|dd	�ddd�f|dd	�ddd�df|||dd�|f|d|f|dd�dd	�|f|d|dd�dd�dfgD]D}tj||j�}|j||||�|j|j	||�|j	||���q8Wtjdd
d�}�x4|||f|dd�dd�fdf|||f|dd�dd
d�fdf|d||f|ddd�dd�fdf||d|f|dd�ddd
d�fdf||dd�|f|dd�dd�dd
d�fdf|ddd�|f|ddd�dd
d�fdf|ddd�||df|ddd�dd�dd�dfdf|ddd�|dd�df|ddd�dd�dd�dfd f|dd�dd�|dd�df|dd�dd�dd�dd�dfd!f|dd�dd�||dd�f|dd�dd�dd�dd
d�fd"fg
D]\\}}}tj||j�}|j
|||j�|j||||�|j|j	||�|j	||���qfW|jtd��tj|dk|j�WdQRX|jtd��$t
tjd|d|f|j��WdQRXdS)#Nrrrrrrrrrr�r�zonly supports 1dzinvalid subkeyg�?)rr���������r)r)r(r))rrrrr)rrrrr)rrrr)rrrr)rrrrr)rrrr)rrr)rrr)rrrr)rrrrr)rrrrrrZorthogonal_indexer�shaperr�assertEqualr�
ValueError�print)	r
rr!r"rr#�zr*�kr
r
r�test_orthogonal_indexers@
NNH4 "$&.*48@Nz$TestIndexers.test_orthogonal_indexercCsxtjd�}t|tj�st�|jdks(t�|j�dks8t�tjddg�}|jd	ksTt�|dd
ksdt�|ddkstt�dS)N�arrr)r1r)r1r)r)r)r)r)r)	rZ_asarray_tuplesafe�
isinstancerZndarray�AssertionErrorr�itemr*)r
�resr
r
r�test_asarray_tuplesafeFs
z#TestIndexers.test_asarray_tuplesafecCs\tjdddg�}|jtd��tj|dg�WdQRX|jt��tj|d�WdQRXdS)Nrrrznot all values foundr)�pd�Indexr�KeyErrorr�convert_label_indexer�assertRaises)r
�indexr
r
r�test_convert_label_indexerQs
z'TestIndexers.test_convert_label_indexercCs<tjdddg�}|jt��tj|tdd��WdQRXdS)NZ2001�2000Z2002)r7Zto_datetimer;r9rr:�slice)r
r<r
r
r�+test_convert_unsorted_datetime_index_raisesYsz8TestIndexers.test_convert_unsorted_datetime_index_raisescsntdddddgfi���fdd�}|jddi|d��|jddi|tjd���|jddi|tgd���dS)Nrrrrcstj�d|i�S)Nr)rZremap_label_indexers)r)�datar
r�test_indexeresz<TestIndexers.test_remap_label_indexers.<locals>.test_indexerr)rr+rZint32r)r
rBr
)rAr�test_remap_label_indexersas
z&TestIndexers.test_remap_label_indexersN)
�__name__�
__module__�__qualname__rr$r0r6r=r@rCr
r
r
rr	s)rc@seZdZdd�Zdd�ZdS)�
TestLazyArrayc	Cst�}tjd�}|dd�|dd�|dd�|dd�|dd
�|dd�|dd
�|ddd�|ddd�|ddd�|ddd�|dd�|ddd�|ddd�|ddd�g}xJ|D]B}x<|D]4}|||}tj||dd	�}||}|j||�q�Wq�WdS)N�drrrrr�r)�sizer(r(���r(r(r(r(r(���)rrrrZslice_slicer)	r
r"rZslicesrr#�expectedZ	new_slice�actualr
r
r�test_slice_slicems
FF

zTestLazyArray.test_slice_slicec

Cs�tjtjjddd��}tj|�}t�}|dd�dd
|dd�ddddgtjd�dkg}x�|D]�}x�|D]�}x�|D]�}tj||||f�}xl||||f|dd�||f||dd�dd�|fdd�|f|gD] }	|j	|j
|	j
�|j||	�q�WqrWqhWq^Wd|dd�df|dd�df|dd	�dgfd	dgdfdddgddgfdddg|dd�fg}xL|D]D\}}tj|||�}|||}	|j	|j
|	j
�|j||	��qjWdS)
Nr�rIrrrrrrr))rrr()rZNumpyIndexingAdapterrrZrandZLazilyIndexedArrayrrZasarrayr+r*r)
r
rZlazyr"Zindexersrr#r/rMrNr
r
r�test_lazily_indexed_arrayzs(
2


,8(z'TestLazyArray.test_lazily_indexed_arrayN)rDrErFrOrQr
r
r
rrGls
rG)ZnumpyrZpandasr7ZxarrayrrZxarray.corer�rrrrGr
r
r
r�<module>sc