
    e                     b    d dl 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 )	    N)ScatterTiles   )TestPlotlyPlotc                   >    e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
d	S )
TestScatterPlotc                    t          g d          }|                     |          }|                     |d         d         d         d           |                     |d         d         d         t          j        g d                     |                     |d         d         d         d           |                     |d	         d
         d         ddg           d S )N      r   datar   typescatterymodemarkerslayoutyaxisranger   r   )r   _get_plot_stateassertEqualnparrayselfr   states      Plib/python3.11/site-packages/holoviews/tests/plotting/plotly/test_scatterplot.pytest_scatter_statez"TestScatterPlot.test_scatter_state
   s    )))$$$$W--vq)&19===vq)#.0C0CDDDvq)&19===x1':QFCCCCC    c                    t          g d                              d          }|                     |          }|                     |d         d         d         t	          j        g d                     |                     |d         d         d         t	          j        g d                     |                     |d         d         d	         d
           |                     |d         d         d         ddg           |                     |d         d         d         ddg           |                     |d         d         d         d         d           |                     |d         d         d         d         d           d S )N)r   r   r   T)invert_axesr   r   xr   r   r   r   r   r   r   xaxisr   r   r   r   r   titletextr   optsr   r   r   r   r   s      r   test_scatter_invertedz%TestScatterPlot.test_scatter_inverted   sg   )))$$))d);;$$W--vq)#.0C0CDDDvq)#.0C0CDDDvq)&19===x1':QFCCCx1':QFCCCx1':6BCHHHx1':6BCHHHHHr   c                    t          g d                              d          }|                     |          }|                     |d         d         d         d         t	          j        g d                     |                     |d         d         d         d	         d           |                     |d         d         d         d
         d           d S )Nr
   r"   colorr   r   markerr,   r#   cmincmaxr   r'   r   s      r   test_scatter_color_mappedz)TestScatterPlot.test_scatter_color_mapped   s    )))$$)))44$$W--vq)(3G<bhyyy>Q>QRRRvq)(3F;Q???vq)(3F;Q?????r   c                     t          g d                              d          }|                     |          }|                     |d         d         d         d         t	          j        g d                     d S )Nr
   r   sizer   r   r-   r3   r'   r   s      r   test_scatter_sizez!TestScatterPlot.test_scatter_size$   st    )))$$))s)33$$W--vq)(3F;RXiii=P=PQQQQQr   c           	      ,   t          g dddg                              d          }|                     |          }|                     t	          j        |d         d         d         d         t	          j        g d	                    d
           d S )N)r   r   red)r   r   green)r   r   bluer   r,   vdimsr+   r   r   r-   r7   r8   r9   Tr   r(   r   r   r   array_equalr   r   s      r   test_scatter_colorsz#TestScatterPlot.test_scatter_colors)   s     
 
 
w! ! !!%G!4!4 	 $$W--fa(8(B7(K(*1I1I1I(J(JL LMQ	S 	S 	S 	S 	Sr   c           	      ,   t          g dddg                              d          }|                     |          }|                     t	          j        |d         d         d         d         t	          j        g d	                    d
           d S )N)r   r   square)r   r   circle)r   r   triangle-upr   r-   r:   r-   r   r   symbolrB   rC   rD   Tr=   r   s      r   test_scatter_markersz$TestScatterPlot.test_scatter_markers1   s     
 
 
x" " ""&$h$"7"7 	 $$W--fa(8(B8(L(*1T1T1T(U(UW WX\	^ 	^ 	^ 	^ 	^r   c                     t          g d                              ddg          }|                     |          }|                     |d         d         d         ddg           d S )N)r   r   )r   r   )r   r   r   r   selectedpointsr   r   rL   r   r(   r   r   r   s      r   test_scatter_selectedpointsz+TestScatterPlot.test_scatter_selectedpoints9   sz     
 
 
  41v4&& 	 $$W--vq)*:;aVDDDDDr   c                     t          g d                              d          }|                     |          }|                     |d         d         d         d           d S )Nr
   Fvisibler   r   rQ   rM   r   elementr   s      r   test_visiblezTestScatterPlot.test_visible@   sa    )))$$))%)88$$W--vq))4e<<<<<r   N)__name__
__module____qualname__r   r)   r0   r4   r?   rH   rN   rT    r   r   r   r      s        D D D	I 	I 	I@ @ @R R R
S S S^ ^ ^E E E= = = = =r   r   c                   8    e Zd Zd Zd Zd Zd Zd Zd Zd Z	dS )	TestMapboxScatterPlotc                    g d}g d}d}t          |          dz  }d}t          |          dz  }t          j        |g|g          \  }}|d         |d         }
}	t          j        ||          \  }}t          d          t          ||f          j                            ||          z  }|                     |          }|                     |d	         d
         d         d           |                     |d	         d
         d         |           |                     |d	         d
         d         |           |                     |d	         d
         d         d           |                     |d         d         d         |
|	d           |                     d|d         v            |                     d|d         v            d S )N)i- 逄 i@B )@9i{i)ii 	= g       @)r]   r\   r    )r"   r   r   r   r   scattermapboxlonlatr   r   r   mapboxcenter)ra   r`   r$   r   )	sumr   easting_northing_to_lon_latr   redimr   r   r   assertFalse)r   xsysx_rangex_centery_rangey_centerlon_centerslat_centers
lon_center
lat_centerlonslatsr   r   s                  r   r   z(TestMapboxScatterPlot.test_scatter_stateG   s   (((+++%w<<#%%w<<#%#(#DhZRZQ[#\#\ [!,QQJ
6r2>>
d))gr2h//5;;g;QQQ$$W--vq)&1?CCCvq)%0$777vq)%0$777vq)&19===(OH%h/J1W1W	
 	
 	

 	E(O3444E(O344444r   c                    t          d          t          g d                              d          z  }|                     |          }|                     |d         d         d         d         t          j        g d	                     |                     |d         d         d         d
         d           |                     |d         d         d         d         d           d S )Nr^   r
   r"   r+   r   r   r-   r,   r#   r.   r   r/   r   r   r   r(   r   r   r   r   r   s      r   r0   z/TestMapboxScatterPlot.test_scatter_color_mappeda   s    ))giii0055C5@@@$$W--vq)(3G<bhyyy>Q>QRRRvq)(3F;Q???vq)(3F;Q?????r   c                    t          d          t          g d                              d          z  }|                     |          }|                     |d         d         d         d         t          j        g d                     d S )	Nr^   r
   r   r2   r   r   r-   r3   ru   r   s      r   r4   z'TestMapboxScatterPlot.test_scatter_sizeh   s    ))giii005535???$$W--vq)(3F;RXiii=P=PQQQQQr   c           	      L   t          d          t          g dddg                              d          z  }|                     |          }|                     t          j        |d         d         d	         d         t          j        g d
                    d           d S )Nr^   r6   r   r,   r:   r+   r   r   r-   r<   Tr   r   r(   r   r   r   r>   r   r   s      r   r?   z)TestMapboxScatterPlot.test_scatter_colorsn   s    ))g '
 '
 '
w! ! !!%G!4!45 $$W--fa(8(B7(K(*1I1I1I(J(JL LMQ	S 	S 	S 	S 	Sr   c           	      L   t          d          t          g dddg                              d          z  }|                     |          }|                     t          j        |d         d         d         d	         t          j        g d
                    d           d S )Nr^   rA   r   r-   r:   rE   r   r   rF   rG   Trx   r   s      r   rH   z*TestMapboxScatterPlot.test_scatter_markersw   s    ))g '
 '
 '
x" " ""&$h$"7"78 $$W--Nfa *84<<<==? ?@D	F 	F 	F 	F 	Fr   c                     t          d          t          g d                              ddg          z  }|                     |          }|                     |d         d         d         ddg           d S )Nr^   rJ   r   r   rK   r   rL   r   r   r(   r   r   r   s      r   rN   z1TestMapboxScatterPlot.test_scatter_selectedpoints   s    ))g '
 '
 '
  41v4&&' $$W--vq)*:;aVDDDDDr   c                     t          d          t          g d                              d          z  }|                     |          }|                     |d         d         d         d           d S )Nr^   r
   FrP   r   r   rQ   r{   rR   s      r   rT   z"TestMapboxScatterPlot.test_visible   sl    ))giii0055e5DDD$$W--vq))4e<<<<<r   N)
rU   rV   rW   r   r0   r4   r?   rH   rN   rT   rX   r   r   rZ   rZ   F   s        5 5 54@ @ @R R RS S SF F FE E E= = = = =r   rZ   )	numpyr   holoviews.elementr   r   	test_plotr   r   rZ   rX   r   r   <module>r      s        , , , , , , , , % % % % % %;= ;= ;= ;= ;=n ;= ;= ;=|E= E= E= E= E=N E= E= E= E= E=r   