a
    ~¤b@  ã                   @   s4   d dl Zd dlmZ ddlmZ G dd„ deƒZdS )é    N)ÚSurfaceé   )ÚTestPlotlyPlotc                   @   s$   e Zd Zdd„ Zdd„ Zdd„ ZdS )ÚTestSurfacePlotc                 C   sL  t g d¢ddgt g d¢g d¢g¡fƒ}|  |¡}|  |d d d d¡ |  |d d d	 t g d¢¡¡ |  |d d d
 t ddg¡¡ |  |d d d t g d¢g d¢g¡¡ |  |d d d d¡ |  |d d d d¡ |  |d d d d ddg¡ |  |d d d d ddg¡ |  |d d d d ddg¡ d S )N©r   é   é   r   r   ©r   r   r   ©r   r   é   ÚdataÚtypeZsurfaceÚxÚyÚzZcminZcmaxr   ZlayoutZsceneZxaxisÚrangeg      à?g      @Zyaxisg      à¿g      ø?Zzaxis)r   ÚnpÚarrayÚ_get_plot_stateÚassertEqual)ÚselfÚimgÚstate© r   úNlib/python3.9/site-packages/holoviews/tests/plotting/plotly/testsurfaceplot.pyÚtest_surface_state
   s    (
""*  z"TestSurfacePlot.test_surface_statec                 C   sŒ   t g d¢ddgt g d¢g d¢g¡fƒ}|jdd |  |¡}|d d }|  |d	 ¡ |jd
d |  |¡}|d d }|  |d	 ¡ d S )Nr   r   r   r	   r
   T)Zcolorbarr   Z	showscaleF)r   r   r   Zoptsr   Z
assertTrueZassertFalse)r   r   r   Ztracer   r   r   Útest_surface_colorbar   s    (

z%TestSurfacePlot.test_surface_colorbarc                 C   sV   t g d¢ddgt g d¢g d¢g¡fƒjdd}|  |¡}|  |d d d	 d¡ d S )
Nr   r   r   r	   r
   F)Úvisibler   r   )r   r   r   Úoptionsr   r   )r   Úelementr   r   r   r   Útest_visible#   s    "ÿþ
zTestSurfacePlot.test_visibleN)Ú__name__Ú
__module__Ú__qualname__r   r   r    r   r   r   r   r      s   r   )Znumpyr   Zholoviews.elementr   Ztestplotr   r   r   r   r   r   Ú<module>   s   