a
    `@                     @   sp   d Z ddlmZmZ ddlZddlmZmZ G dd deZ	G dd deZ
G d	d
 d
eZG dd deZdS )z'
Tests patching of supported libraries
    )TestCaseSkipTestN)hvPlotTabularhvPlotc                   @   s$   e Zd Zdd Zdd Zdd ZdS )TestPatchPandasc                 C   s   dd l }d S )Nr   )Zhvplot.pandas)selfhvplot r	   5lib/python3.9/site-packages/hvplot/tests/testpatch.pysetUp   s    zTestPatchPandas.setUpc                 C   s(   dd l }|g d}| |jt d S )Nr   r         )pandasSeriesassertIsInstancer   r   )r   pdseriesr	   r	   r
   test_pandas_series_patched   s    z*TestPatchPandas.test_pandas_series_patchedc                 C   s>   dd l }|jddgddgddggdd	gd
}| |jt d S Nr   r   r               xy)columns)r   	DataFramer   r   r   )r   r   dfr	   r	   r
   test_pandas_dataframe_patched   s    $z-TestPatchPandas.test_pandas_dataframe_patchedN)__name__
__module____qualname__r   r   r   r	   r	   r	   r
   r      s   r   c                   @   s$   e Zd Zdd Zdd Zdd ZdS )TestPatchDaskc                 C   s2   zdd l m} W n   tdY n0 dd l}d S )Nr   zDask not available)dask.dataframe	dataframer   Zhvplot.dask)r   ddr   r	   r	   r
   r      s
    zTestPatchDask.setUpc                 C   s@   dd l }dd lm} |g d}||d}| |jt d S )Nr   r   r   )r   r$   r%   r   from_pandasr   r   r   )r   r   r&   r   Zdseriesr	   r	   r
   test_dask_series_patched%   s
    z&TestPatchDask.test_dask_series_patchedc                 C   sV   dd l }dd lm} |jddgddgddggdd	gd
}||d}| |jt d S r   )r   r$   r%   r   r'   r   r   r   )r   r   r&   r   Zddfr	   r	   r
   test_dask_dataframe_patched,   s
    $z)TestPatchDask.test_dask_dataframe_patchedN)r    r!   r"   r   r(   r)   r	   r	   r	   r
   r#      s   r#   c                   @   s$   e Zd Zdd Zdd Zdd ZdS )TestPatchXArrayc                 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   6   s
    zTestPatchXArray.setUpc                 C   sF   dd l }tjdd}|j|tdtdddd}| |jt d S )Nr   d   r   r   r   r   ZcoordsZdims)	r+   nprandomrand	DataArrayranger   r   r   )r   r,   arrayxr_arrayr	   r	   r
   test_xarray_dataarray_patched=   s    z-TestPatchXArray.test_xarray_dataarray_patchedc                 C   sT   dd l }tjdd}|j|tdtdddd}|d|i}| |jt	 d S )Nr   r-   r.   r/   r0   z)
r+   r1   r2   r3   r4   r5   ZDatasetr   r   r   )r   r,   r6   r7   Zxr_dsr	   r	   r
   test_xarray_dataset_patchedC   s
    z+TestPatchXArray.test_xarray_dataset_patchedN)r    r!   r"   r   r8   r:   r	   r	   r	   r
   r*   4   s   r*   c                   @   s4   e Zd Zdd Zdd Zdd Zdd Zd	d
 ZdS )TestPatchStreamzc                 C   s.   zdd l }W n   tdY n0 dd l}d S )Nr   zstreamz not available)streamzr   Zhvplot.streamz)r   r<   r   r	   r	   r
   r   M   s
    zTestPatchStreamz.setUpc                 C   s$   ddl m} | }| |jt d S Nr   Random)streamz.dataframer?   r   r   r   r   r?   Z	random_dfr	   r	   r
   test_streamz_dataframe_patchedT   s    z/TestPatchStreamz.test_streamz_dataframe_patchedc                 C   s&   ddl m} | }| |jjt d S r=   )r@   r?   r   r   r   r   rA   r	   r	   r
   test_streamz_series_patchedY   s    z,TestPatchStreamz.test_streamz_series_patchedc                 C   s.   ddl m} | }| |d jt d S Nr   r>   r   )r@   r?   r   groupbysumr   r   rA   r	   r	   r
   test_streamz_dataframes_patched^   s    z0TestPatchStreamz.test_streamz_dataframes_patchedc                 C   s0   ddl m} | }| |d jjt d S rD   )r@   r?   r   rE   rF   r   r   r   rA   r	   r	   r
   test_streamz_seriess_patchedc   s    z-TestPatchStreamz.test_streamz_seriess_patchedN)r    r!   r"   r   rB   rC   rG   rH   r	   r	   r	   r
   r;   K   s
   r;   )__doc__Zunittestr   r   Znumpyr1   Zhvplot.plottingr   r   r   r#   r*   r;   r	   r	   r	   r
   <module>   s   