
    q[ic	                     b    d Z ddlZddlZddlZddlmZ ddlmZ d Z	d Z
d Zd	 Zd
 Zd Zd Zy)z2Tests t_bytes methods in the _transformers module.    N   )_transformers)typesc                      t        j                  t        g d      dd      } | j                  t        g d      k(  sJ | j                  dk(  sJ y N)r   r   r   r   r   r   z	audio/pcm)data	mime_type)tt_blobbytesr   r	   blobs    c/mnt/data/GridWatch/venv/lib/python3.12/site-packages/google/genai/tests/transformers/test_blobs.pytest_blob_dictr      sM    	
&' 
$ 
e./	//	/	;	&&	&    c                      t        j                  t        j                  t	        g d      d            } | j
                  t	        g d      k(  sJ | j                  dk(  sJ y r   )r
   r   r   Blobr   r   r	   r   s    r   	test_blobr   %   sP    	
jje./;G
$ 
e./	//	/	;	&&	&r   c                 V   t        j                  |       }|j                  dd dk(  sJ |j                  dk(  sJ |j	                         j
                  }|J |j                  | j                  k(  sJ |j                  | j                  k(  sJ |j                  | j                  k(  sJ y N   
   s   JFIFz
image/jpeg)	r
   r   r   r	   as_image
_pil_imagesizemodeformat)
image_jpegr   round_trip_images      r   
test_imager    -   s    	
*	$	1RG	##	#	<	''	']]_//		%%	%			*//	11	1			*//	11	1		 	 J$5$5	55	5r   c                  l    t        j                  t        g d      d      } | j                         J y r   )r   r   r   r   r   s    r   test_not_imager"   9   s*    	12k	J$		  	 r   c                 ~   t        j                  |       }|j                  j                  dd dk(  sJ |j                  j                  dk(  sJ |j                         j                  }|J |j                  | j                  k(  sJ |j                  | j                  k(  sJ |j                  | j                  k(  sJ y r   )
r
   t_partinline_datar   r	   r   r   r   r   r   )r   partr   s      r   test_part_imager'   >   s    	
*	$					q	$	//	/				#	#|	33	3]]_//		%%	%			*//	11	1			*//	11	1		 	 J$5$5	55	5r   c                  R    t        j                  d      } | j                         J y )Nzhello world)r
   r$   r   )r&   s    r   test_part_not_imager)   J   s"    	
-	 $		  	 r   c                     t         j                  j                  ddd      } t        j                  |       }|j
                  dk(  sJ |j                  rt        |j                        dk(  sJ |j                  dd d	k(  sJ y )
NRGB)   r,   red)colorz	image/pngE   r      s   PNG)PILImagenewr
   pil_to_blobr	   r   len)imgr   s     r   &test_pil_to_blob_with_memory_pil_imager7   O   sn    		eV51#	
s	$	;	&&	&	s499~++	+	1Q:	%%	%r   )__doc__os	PIL.Imager1   pytest r   r
   r   r   r   r    r"   r'   r)   r7    r   r   <module>r>      s<   " 9 	   " ''	6!
	6!
&r   