
    ]eN                         d dl mZ 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 d dlmZ  G d de          Z G d d	e          ZdS )
    )OrderedDictN)hvPlothvPlotTabular)StoreScatter)ComparisonTestCasec                   2    e Zd Zd Zd Zd Zd Zd Zd ZdS )TestOverridesc                 Z    dd l }t          j        ddgddgddggdd	g
          | _        d S )Nr                     xy)columns)hvplot.pandaspd	DataFramedfselfhvplots     :lib/python3.11/site-packages/hvplot/tests/testoverrides.pysetUpzTestOverrides.setUp   s?    ,AAA7#sLLL    c                 0   t          | j        dd          } |d          }t          j        d|d          }|                     |j                            d          d           |                     |j                            d          d           d S )	N*   widthheightr   r   bokehplotr"   r#   )r   r   r   lookup_optionsassertEqualoptionsget)r   r   curveoptss       r   test_define_default_optionsz)TestOverrides.test_define_default_options   s    twb<<<#GUF;;))'22B777))(33R88888r   c                 Z   t          | j        ddddi          }|                    d          }|                    d          }t	          j        d|d          }t	          j        d|d          }|                     |j                            d          d           |                     |j                            d	          d           | 	                    |j                            d          d           | 	                    |j                            d	          d           d S )
Ncustom_scatterr    r!   r   r$   r%   r&   r"   r#   )
r   r   r/   scatterr   r'   r(   r)   r*   assertNotEqual)r   r   r/   r0   custom_optsr,   s         r   test_define_custom_methodz'TestOverrides.test_define_custom_method   s   tw)9RSU;V;V(WXX...55..3.''*7NFKK#GWf==,00992>>>,00::B???DL,,W55r:::DL,,X66;;;;;r   c                 Z   t          | j        ddddi          }|                    d          }|                    d          }t	          j        d|d          }t	          j        d|d          }|                     |j                            d          d           |                     |j                            d	          d           | 	                    |j                            d          d           | 	                    |j                            d	          d           d S )
Nr0   r    r!   r   r$   r%   r&   r"   r#   )
r   r   r0   liner   r'   r(   r)   r*   r1   )r   r   r/   r+   r2   r,   s         r   test_define_customize_methodz*TestOverrides.test_define_customize_method$   s   twbB4O4O(PQQ#..c""*7NFKK#GUF;;,00992>>>,00::B???DL,,W55r:::DL,,X66;;;;;r   c                     t          | j        dddii          }|                     |                    d          t                     d S )Nr0   kindr5   r   r$   )r   r   assertIsInstancer0   r   r   s     r   'test_attempt_to_override_kind_on_methodz5TestOverrides.test_attempt_to_override_kind_on_method/   sG    twVV4D(EFFfnnsn33W=====r   c                 f    t          | j        d          }t          |j                  dk    sJ d S )Nzx>2)queryr   )r   r   len_datar   s     r   test_pandas_query_metadataz(TestOverrides.test_pandas_query_metadata3   s7    twe4446<  A%%%%%%r   N)	__name__
__module____qualname__r   r-   r3   r6   r:   r?    r   r   r
   r
      sq        M M M9 9 9	< 	< 	<	< 	< 	<> > >& & & & &r   r
   c                   &    e Zd Zd Zd Zd Zd ZdS )TestXArrayOverridesc                    t          dddgfdddgfdddgfg          }t          j        t          j        d                              d          |g d                              t          j        d	d
gdg                    | _        d S )Ntimer   r   latlon   )r   r   r   )rG   rH   rI   r   r   )dims)lat1)r   xr	DataArraynparangereshapeassign_coordsda_img_by_time)r   coordss     r   r   zTestXArrayOverrides.setUp:   s    v1v.A%!QQRR l29Q<<+?+?	+J+J+13I3I3IK KKX=46L!AeW4U4U4U LY LW LW 	r   c                 Z    t          | j        ddi          }|j        j        dk    sJ d S )NrG   r   iselr   r   rS   r>   ndimr   s     r    test_xarray_isel_scalar_metadataz4TestXArrayOverrides.test_xarray_isel_scalar_metadata@   s7    +61+>>>| A%%%%%%r   c                     t          | j        ddgi          }|j        j        dk    sJ t	          |j        j                  dk    sJ d S )NrG   r   rV   r   )r   rS   r>   rY   r=   rG   r   s     r   #test_xarray_isel_nonscalar_metadataz7TestXArrayOverrides.test_xarray_isel_nonscalar_metadataD   sW    +6A3-@@@| A%%%%6<$%%******r   c                 Z    t          | j        ddi          }|j        j        dk    sJ d S )NrG   r   )selr   rX   r   s     r   test_xarray_sel_metadataz,TestXArrayOverrides.test_xarray_sel_metadataI   s7    +&!===| A%%%%%%r   N)r@   rA   rB   r   rZ   r\   r_   rC   r   r   rE   rE   8   sS        W W W& & &+ + +
& & & & &r   rE   )collectionsr   numpyrO   pandasr   xarrayrM   hvplot.plottingr   r   	holoviewsr   r   holoviews.element.comparisonr   r
   rE   rC   r   r   <module>rg      s    # # # # # #             1 1 1 1 1 1 1 1 $ $ $ $ $ $ $ $ ; ; ; ; ; ;)& )& )& )& )&& )& )& )&X& & & & &, & & & & &r   