
    e/                         d Z ddlZddlmZmZmZmZmZm	Z	 ddl
mZ  G d de          Z G d de          Z G d	 d
e          Z G d de          ZdS )z
Unit tests of Raster elements
    N)HSVRGBCurveImageQuadMeshRaster)ComparisonTestCasec                   ,    e Zd Zd Zd Zd Zd Zd ZdS )
TestRasterc                 <    t          j        ddg          | _        d S Nr         )         nparrayarray1selfs    Clib/python3.11/site-packages/holoviews/tests/element/test_raster.pysetUpzTestRaster.setUp       h	9566    c                 .    t          | j                   d S N)r   r   r   s    r   test_raster_initzTestRaster.test_raster_init   s    t{r   c                 f    t          | j                  }|                     |d         d           d S )N)r   r   r   )r   r   assertEqualr   rasters     r   test_raster_indexzTestRaster.test_raster_index   s1    $$q)))))r   c           	          t          | j                  }|                     |                    d          t	          t          j        g d          dgdg                     d S )Nr   )y))r   r   )r   r   )r   r   xz)kdimsvdims)r   r   r"   sampler   r   r   r#   s     r   test_raster_samplezTestRaster.test_raster_sample   sq    $$++rx(@(@(@AA&)U3%9 9 9	: 	: 	: 	: 	:r   c                 P   t           j                            dd          dz
  }t           j                            |dk    |          }t          |                              d          }|                     |t          j        |          t          j	        |          f           d S )N
         ?r   r   )
r   randomrandmamasked_wherer   ranger"   minmax)r   arrrranges      r   test_raster_range_maskedz#TestRaster.test_raster_range_masked   s    innR##C'e  a--""1%%"&++rvc{{!;<<<<<r   N)__name__
__module____qualname__r   r    r%   r-   r:    r   r   r   r      s_        7 7 7  * * *: : := = = = =r   r   c                   2    e Zd Zd Zd Zd Zd Zd Zd ZdS )TestRGBc                 R    t           j                            ddd          | _        d S Nr      )r   r   r   )r   r1   randint	rgb_arrayr   s    r   r   zTestRGB.setUp&   !    **1c9==r   c                 ~    t          | j                  }|                     t          |j                  d           d S )Nr   r   rE   r"   lenr+   r   rgbs     r   $test_construct_from_array_with_alphaz,TestRGB.test_construct_from_array_with_alpha)   s5    $.!!SY+++++r   c                     t          g dg d| j        f          }|                     t          |j                  d           d S )Nr   r   rH   rJ   s     r   $test_construct_from_tuple_with_alphaz,TestRGB.test_construct_from_tuple_with_alpha-   sD    999iii899SY+++++r   c                     t          g dg d| j        d          }|                     t          |j                  d           d S )N)r   r   r   )r(   r'   )RGBAr   rH   rJ   s     r   #test_construct_from_dict_with_alphaz+TestRGB.test_construct_from_dict_with_alpha1   sG    						XXYYSY+++++r   c                     t          | j                  j        }t           j        }t          ||          D ]\  }}||usJ ||k    sJ d S r   )r   rE   r+   zipr   
init_vdims	cls_vdimsics        r   $test_not_using_class_variables_vdimsz,TestRGB.test_not_using_class_variables_vdims5   s^    ((.
I	
I.. 	 	DAqA::::66666	 	r   c                    d}t          j        dd||z  dz                                ||d          }t          |          }t	          t          j        |d                             dk    sJ t	          t          j        |d                             dk    sJ t	          t          j        |d                             dk    sJ |j                            d          }t	          t          j        |d                             dk    sJ t	          t          j        |d                             dk    sJ t	          t          j        |d                             dk    sJ d S )	Nr   r   r   r   rP   rQ   rR   )rP   )r   linspacereshaper   sumisnanredimnodata)r   Nrgb_drK   rgb_ns        r   test_nodatazTestRGB.test_nodata<   sG   Aq!a%!),,44Q1==%jj28CH%%&&!++++28CH%%&&!++++28CH%%&&!++++	  1 %%28E#J''((A----28E#J''((A----28E#J''((A------r   N)	r;   r<   r=   r   rL   rN   rT   r\   rg   r>   r   r   r@   r@   $   sn        > > >, , ,, , ,, , ,  . . . . .r   r@   c                       e Zd Zd Zd ZdS )TestHSVc                 R    t           j                            ddd          | _        d S rB   )r   r1   rD   	hsv_arrayr   s    r   r   zTestHSV.setUpK   rF   r   c                     t          | j                  j        }t           j        }t          ||          D ]\  }}||usJ ||k    sJ d S r   )r   rk   r+   rV   rW   s        r   r\   z,TestHSV.test_not_using_class_variables_vdimsN   s^    T^,,2J	IJ	22  1zzzzAvvvvv r   N)r;   r<   r=   r   r\   r>   r   r   ri   ri   I   s2        > > >    r   ri   c                        e Zd Zd Zd Zd ZdS )TestQuadMeshc                 <    t          j        ddg          | _        d S r   r   r   s    r   r   zTestQuadMesh.setUpW   r   r   c                    t          | j        ddgdgdd          }t          |          }|                     |                    dd          t          j        g d	                     |                     |                    d
d          t          j        ddg                     |                     |                    dd          | j        d d d                    |                     |j        |j                   |                     |j        |j                   |                     |j	        |j	                   |                     |j
        |j
                   d S )Nabr[   rS   rR   )r*   r+   grouplabelr   F)gF]kSUտg        gF]kSU?r   g      пg      ?r   )flat)r   r   r   r"   dimension_valuesr   r   r*   r+   rs   rt   )r   imgqmeshs      r   test_cast_image_to_quadmeshz(TestQuadMesh.test_cast_image_to_quadmeshZ   s=   DKSz#cQTUUU//59928D]D]D];^;^___//59928UDM;R;RSSS///>>DDbD@QRRRci000ci000ci000ci00000r   c                    t          ddgddgt          j        ddgddgg          f          }|                                }t          j        g dg dg dg dg d	g d
g dg dg          }t          j        g d          }|                     |                                |           |                     |j                            ddg          |           d S )Nr   r   r   r   )r   r   r   r   )r   r   r   r   )r   r      r   )r   r      r   )r   r   r   r   )r   r   r   r   )r}   r|   r   r   )   r}   r   r   )	)      r   )r   r0   )r         ?)r0   r   )r0   r0   )r0   r   )r   r   )r   r0   )r   r   )r   r   r   trimeshr"   nodes)r   ry   r   	simplicesverticess        r   test_quadmesh_to_trimeshz%TestQuadMesh.test_quadmesh_to_trimeshe   s   1a&1a&"(QFQF3C*D*DEFF--//Hlll*ll*ll*ll*ll*ll*ll*ll, - -	 8 B B B C C 	)444,,aV44h?????r   N)r;   r<   r=   r   rz   r   r>   r   r   rn   rn   U   sF        7 7 7	1 	1 	1@ @ @ @ @r   rn   )__doc__numpyr   holoviews.elementr   r   r   r   r   r   holoviews.element.comparisonr	   r   r@   ri   rn   r>   r   r   <module>r      s         F F F F F F F F F F F F F F F F ; ; ; ; ; ;= = = = =# = = =2#. #. #. #. #.  #. #. #.J
 
 
 
 
  
 
 
@ @ @ @ @% @ @ @ @ @r   