U
    Cvfa                     @   s   d dl Zd dlZd dlZd dlmZmZ d dlm	Z	m
Z
 ejdejde
dgddd	 Zejd
gddd Zejd
gddd ZdS )    N)	DataArrayDataset)create_test_datarequires_dasknumpydask)Zmarks)paramsc                 C   s   | j S )N)param)request r   9/tmp/pip-unpacked-wheel-h316xyqg/xarray/tests/conftest.pybackend	   s    r      c              
   C   s*  | j dkrvttddgtjddfddgtjddfdtdtd	d
dfdtd	d
dfdddgftdd}n| j dkrttddgtjddfdgtjdfddgtjddfdtdtd	d
dfdtd	d
dfdddgftdd}n| j dkrt }nt	|dkr&|
 S |S )Nr   yx      time
   )z1z2r   g      ?ab)r   r   cr   )r   r   Zz3   r   )r	   r   dictnprandomZrandnZlinspaceranger   
ValueErrorchunk)r
   r   dsr   r   r   r!      sB    




r!   c                 C   s   | j dkr6tjdddd}ttjddt|dd	}| j d
krhtdtjdd
tjdddtjddgdd}| j dkrttt	dd
dddtdtddtdd}|dkr| S |dkr|S td S )Nr   z
2000-01-01Z1D   )freqZperiods)r   r"      )r   r   r   )r   )dimscoordsr   r   r   r$            r   )r%   Zrepeating_ints   abcZdefg)r   r   Zzyx)r&   r%   r   r   )r	   pdZ
date_ranger   r   r   r   nanZtileZarangeZreshapelistr    r   )r
   r   timesdar   r   r   r0   6   s(    


(
r0   )r   r   Zpandasr,   ZpytestZxarrayr   r   Zxarray.testsr   r   Zfixturer	   r   r!   r0   r   r   r   r   <module>   s   

'