o
    ic	                     @   sp   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d Z	dd	 Z
d
d Zdd Zdd Zdd Zdd ZdS )z2Tests t_bytes methods in the _transformers module.    N   )_transformers)typesc                  C   s@   t tg ddd} | jtg dksJ | jdksJ d S N)r   r   r   r   r   r   z	audio/pcm)data	mime_type)tt_blobbytesr   r   blob r   ^/home/ubuntu/.local/lib/python3.10/site-packages/google/genai/tests/transformers/test_blobs.pytest_blob_dict   s   
r   c                  C   sD   t tjtg ddd} | jtg dksJ | jdks J d S r   )r   r	   r   Blobr
   r   r   r   r   r   r   	test_blob%   s
   r   c                 C   sx   t | }|jdd dksJ |jdksJ | j}|d us"J |j| jks*J |j| jks2J |j| jks:J d S N   
   s   JFIFz
image/jpeg)	r   r	   r   r   as_image
_pil_imagesizemodeformat)
image_jpegr   round_trip_imager   r   r   
test_image-   s   

r   c                  C   s*   t jtg ddd} |  d u sJ d S r   )r   r   r
   r   r   r   r   r   test_not_image9   s   r   c                 C   s|   t | }|jjdd dksJ |jjdksJ | j}|d us$J |j| jks,J |j| jks4J |j	| j	ks<J d S r   )
r   t_partinline_datar   r   r   r   r   r   r   )r   partr   r   r   r   test_part_image>   s   

r!   c                  C   s   t d} |  d u sJ d S )Nzhello world)r   r   r   )r    r   r   r   test_part_not_imageJ   s   
r"   c                  C   s\   t jjdddd} t| }|jdksJ |jrt|jdks!J |jdd d	ks,J d S )
NRGB)   r$   red)colorz	image/pngE   r      s   PNG)PILImagenewr   pil_to_blobr   r   len)imgr   r   r   r   &test_pil_to_blob_with_memory_pil_imageO   s
   
r/   )__doc__os	PIL.Imager)   pytest r   r   r   r   r   r   r   r!   r"   r/   r   r   r   r   <module>   s   	