
    ]eC                         d Z ddlmZmZ ddlZddlmZmZ  G d de          Z	 G d de          Z
 G d	 d
e          Z G d de          Z G d de          ZdS )z'
Tests patching of supported libraries
    )TestCaseSkipTestN)hvPlotTabularhvPlotc                        e Zd Zd Zd Zd ZdS )TestPatchPandasc                     dd l }d S )Nr   )hvplot.pandas)selfhvplots     6lib/python3.11/site-packages/hvplot/tests/testpatch.pysetUpzTestPatchPandas.setUp   s        c                 |    dd l }|                    g d          }|                     |j        t                     d S Nr   r         )pandasSeriesassertIsInstancer   r   )r   pdseriess      r   test_pandas_series_patchedz*TestPatchPandas.test_pandas_series_patched   sA    999%%fm];;;;;r   c                     dd l }|                    ddgddgddggdd	g
          }|                     |j        t                     d S Nr   r   r               xy)columns)r   	DataFramer   r   r   )r   r   dfs      r   test_pandas_dataframe_patchedz-TestPatchPandas.test_pandas_dataframe_patched   sX    \\Aq6Aq6Aq62S#J\GGbi77777r   N)__name__
__module____qualname__r   r   r&    r   r   r   r      sA          < < <
8 8 8 8 8r   r   c                        e Zd Zd Zd Zd ZdS )TestPatchDaskc                 F    	 dd l m} n#  t          d          xY wdd l}d S )Nr   zDask not available)dask.dataframe	dataframer   hvplot.dask)r   ddr   s      r   r   zTestPatchDask.setUp   sF    	1'''''''	1/000s   	 c                     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   r1   r   dseriess        r   test_dask_series_patchedz&TestPatchDask.test_dask_series_patched%   sf    ######999%%..++gnm<<<<<r   c                     dd l }dd lm} |                    ddgddgddggdd	g
          }|                    |d          }|                     |j        t                     d S r   )r   r.   r/   r$   r3   r   r   r   )r   r   r1   r%   ddfs        r   test_dask_dataframe_patchedz)TestPatchDask.test_dask_dataframe_patched,   s}    ######\\Aq6Aq6Aq62S#J\GGnnR##cj-88888r   N)r'   r(   r)   r   r5   r8   r*   r   r   r,   r,      sA          = = =9 9 9 9 9r   r,   c                        e Zd Zd Zd Zd ZdS )TestPatchXArrayc                 B    	 dd l }n#  t          d          xY wdd l}d S )Nr   zXArray not available)xarrayr   hvplot.xarray)r   xrr   s      r   r   zTestPatchXArray.setUp6   @    	3	31222    c                     dd l }t          j                            dd          }|                    |t          d          t          d          dd          }|                     |j        t                     d S )Nr   d   r!   r"   r"   r!   coordsdims)	r<   nprandomrand	DataArrayranger   r   r   )r   r>   arrayxr_arrays       r   test_xarray_dataarray_patchedz-TestPatchXArray.test_xarray_dataarray_patched=   sm    	sC((<<E#JJU3ZZ.P.PWa<bbhov66666r   c                 &   dd l }t          j                            dd          }|                    |t          d          t          d          dd          }|                    d|i          }|                     |j        t                     d S )Nr   rB   rC   rD   rE   z)
r<   rH   rI   rJ   rK   rL   Datasetr   r   r   )r   r>   rM   rN   xr_dss        r   test_xarray_dataset_patchedz+TestPatchXArray.test_xarray_dataset_patchedC   s    	sC((<<E#JJU3ZZ.P.PWa<bb

C?++elF33333r   N)r'   r(   r)   r   rO   rT   r*   r   r   r:   r:   4   sA          7 7 74 4 4 4 4r   r:   c                   ,    e Zd Zd Zd Zd Zd Zd ZdS )TestPatchStreamzc                 B    	 dd l }n#  t          d          xY wdd l}d S )Nr   zstreamz not available)streamzr   hvplot.streamz)r   rX   r   s      r   r   zTestPatchStreamz.setUpM   s<    	4NNNN	42333r@   c                 f    ddl m}  |            }|                     |j        t                     d S Nr   Random)streamz.dataframer]   r   r   r   r   r]   	random_dfs      r   test_streamz_dataframe_patchedz/TestPatchStreamz.test_streamz_dataframe_patchedT   s>    ,,,,,,FHH	i.>>>>>r   c                 p    ddl m}  |            }|                     |j        j        t
                     d S r[   )r^   r]   r   r!   r   r   r_   s      r   test_streamz_series_patchedz,TestPatchStreamz.test_streamz_series_patchedY   s@    ,,,,,,FHH	ik0-@@@@@r   c                     ddl m}  |            }|                     |                    d                                          j        t                     d S Nr   r\   r!   )r^   r]   r   groupbysumr   r   r_   s      r   test_streamz_dataframes_patchedz0TestPatchStreamz.test_streamz_dataframes_patched^   sX    ,,,,,,FHH	i//4488::A=QQQQQr   c                     ddl m}  |            }|                     |                    d                                          j        j        t                     d S re   )r^   r]   r   rf   rg   r"   r   r   r_   s      r   test_streamz_seriess_patchedz-TestPatchStreamz.test_streamz_seriess_patchedc   s[    ,,,,,,FHH	i//4488::<C]SSSSSr   N)r'   r(   r)   r   ra   rc   rh   rj   r*   r   r   rV   rV   K   sj          ? ? ?
A A A
R R R
T T T T Tr   rV   c                   &    e Zd Zd Zd Zd Zd ZdS )TestPatchPolarsc                 B    	 dd l }n#  t          d          xY wdd l}d S )Nr   zPolars not available)polarsr   hvplot.polars)r   plr   s      r   r   zTestPatchPolars.setUpk   r?   r@   c                 |    dd l }|                    g d          }|                     |j        t                     d S r   )rn   r   r   r   r   )r   rp   pseriess      r   test_polars_series_patchedz*TestPatchPolars.test_polars_series_patchedr   sA    ))III&&gnm<<<<<r   c                     dd l }|                    g dg dd          }|                     |j        t                     d S Nr   )r   r   r   )r   r   r    rC   )rn   r$   r   r   r   )r   rp   pdfs      r   test_polars_dataframe_patchedz-TestPatchPolars.test_polars_dataframe_patchedw   sM    ll;;<<cj-88888r   c                     dd l }|                    g dg dd          }|                     |j        t                     d S ru   )rn   	LazyFramer   r   r   )r   rp   pldfs      r   test_polars_lazyframe_patchedz-TestPatchPolars.test_polars_lazyframe_patched|   sM    ||))))))<<==dk=99999r   N)r'   r(   r)   r   rs   rw   r{   r*   r   r   rl   rl   i   sP          = = =
9 9 9
: : : : :r   rl   )__doc__unittestr   r   numpyrH   hvplot.plottingr   r   r   r,   r:   rV   rl   r*   r   r   <module>r      s1    ( ' ' ' ' ' ' '     1 1 1 1 1 1 1 18 8 8 8 8h 8 8 8 9 9 9 9 9H 9 9 904 4 4 4 4h 4 4 4.T T T T Tx T T T<: : : : :h : : : : :r   