
    |LeD                    @   d dl mZ d dlZd dlmZmZmZmZ  ej	        g dg dg dgd          Z
 ej        d           Z ej        d	          Z ej        d
          Z ej        d          Z ej        d          Z ej        d          Zd Zd Zd Zd ZdS )    )annotationsN)addsaturateoversource)r        )  }i  }i  })            ~ i=<;:uint32dtyper   r	   r   r
   } }}c                    t                                           } t          | dd df<   t          j                            t          t           t                    |            t          | dd df<   t          j                            t          t           t                    |            t          | dd df<   t          j                            t          t           t                    |            d S )Nr      )	srccopyclearnptestingassert_equalr   clear_white	half_blueos    ?lib/python3.11/site-packages/datashader/tests/test_composite.pytest_sourcer      s    

AAa!eHJF3..222Aa!eHJF344a888Aa!eHJF3	22A66666    c                    t                                           } d| d<   t          j                            t          t           t                    |            t          j                            t          t           t                    |            t          j        g dg dg dg          } t          j                            t          t           t                    |            t          j        g dg dg dg          } t          j                            t          t           t                    |            t          j        g d	g d
g dg          } t          j                            t          t           t                    |            t          j        g dg dg dg          } t          j                            t          t           t                    |            d S )Nr   r      r	   r	   r	   )    l    l    )r   r   l   R r   r   r	   )r   l    } l   }  )r   r   l    r
   r
   r	   )l     y l    (x l    x )r   r   l   j5 r   r   r	   )l   * y l   *(Ux l    Tx )r   r   l   d4 )r   r   r   r   r   r   r   r   arraywhitebluer   half_purpler   s    r   	test_overr,      s   

AAdGJDe,,a000JDk22A666
6666666668 	9 	9A JDe,,a000
6666666668 	9 	9A JDdOOQ///
6666666668 	9 	9A JDi00!444
6666666668 	9 	9A JDk22A66666r   c                    t                                           } d| d<   t          j                            t          t           t                    |            t          j                            t          t           t                    |            t          j        g dg dg dg          } t          j                            t          t           t                    |            t          j        g dg dg dg          } t          j                            t          t           t                    |            t          j        g dg dg d	g          } t          j                            t          t           t                    |            t          j        g d
g dg dg          } t          j                            t          t           t                    |            d S )Nr   r!   r#   r%   )r   l    } l   }  )r   r   l    r&   l     u l    t l    t r   l     ~ l   o r'   l   > |u l   >|t l    |t )l   =     = z~    hn )r   r   r   r   r   r   r   r   r(   r)   r*   r   r+   r   s    r   test_addr3   3   s   

AAdGJCUOOQ///JC[111555
6666666668 	9 	9A JCUOOQ///
6666666668 	9 	9A JCTNNA...
6666666668 	9 	9A JCY//333
6666666668 	9 	9A JC[11155555r   c                    t                                           } d| d<   t          j                            t          t           t                    |            t          j                            t          t           t                    |            t          j        dt          d          } t          j                            t          t           t                    |            t          j        dt          d          } t          j                            t          t           t                    |            t          j        g dg dg dg          } t          j                            t          t           t                    |            t          j        g d	g d
g dg          } t          j                            t          t           t                    |            d S )Nr   r!   )   r5   r   r   )r
   r
   r$   r.   r/   )r   r   l    r0   )l   = ~ r1   r2   )r   r   r   r   r   r   r   r   fullr)   r*   r(   r   r+   r   s    r   test_saturater7   J   s|   

AAdGJHS%00!444JHS+66:::
X...AJHS%00!444
H---AJHS$//333
6666666668 	9 	9A JHS)44a888
6666666668 	9 	9A JHS+66:::::r   )
__future__r   numpyr   datashader.compositer   r   r   r   r(   r   r   r   r   r)   r*   r   r+   r   r,   r3   r7    r   r   <module>r<      s1   " " " " " "     < < < < < < < < < < < <bh4444444446=EG G G 		!bi
##	*ryBIj!!	bi
##7 7 77 7 7.6 6 6.; ; ; ; ;r   