a
    `-                     @   sT   d dl mZ d dlZd dlZd dlZd dlm	Z	 G dd de	Z
G dd de	ZdS )    )SkipTestN)ComparisonTestCasec                   @   s   e Zd Zdd Zdd ZdS )TestPandasTransformsc                 C   s   dd l }d S )Nr   )Zhvplot.pandas)selfhvplot r   :lib/python3.9/site-packages/hvplot/tests/testtransforms.pysetUp   s    zTestPandasTransforms.setUpc                 C   sh   t tjdtjdd}tdd }|jj	ddt
|dd}| |jd j|d jd  d S )N2   )valueprobabilityr   d   r   )r   )xy
transforms)pdZ	DataFramenpZrandomZrandnZrandhvdimr   scatterdictassertEqualdatavalues)r   Zdemo_dfpercentr   r   r   r   test_pandas_transform   s     z*TestPandasTransforms.test_pandas_transformN)__name__
__module____qualname__r	   r   r   r   r   r   r   
   s   r   c                   @   s   e Zd Zdd Zdd ZdS )TestXArrayTransformsc                 C   s.   zdd l }W n   tdY n0 dd l}d S )Nr   zxarray not available)xarrayr   Zhvplot.xarray)r   xrr   r   r   r   r	      s
    zTestXArrayTransforms.setUpc                 C   s   dd l }tdddd}td}td}|j|||dddd}|jjtt	dd dd	}| 
|jjj|jd  d S )
Nr   <      
   )r   r   r   )ZcoordsZdimsname)r   )r   )r    r   ZarangeZreshapeZ	DataArrayr   Zimager   r   r   r   r   r   )r   r!   r   r   r   daZimgr   r   r   test_xarray_transform#   s    

z*TestXArrayTransforms.test_xarray_transformN)r   r   r   r	   r'   r   r   r   r   r      s   r   )Zunittestr   Z	holoviewsr   Znumpyr   Zpandasr   Zholoviews.element.comparisonr   r   r   r   r   r   r   <module>   s   