o
    Yic	                     @   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   d/home/kim/smarthome/.venv/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_partZ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)PILZImagenewr   Zpil_to_blobr   r   len)Zimgr   r   r   r   &test_pil_to_blob_with_memory_pil_imageO   s
   
r+   )__doc__osZ	PIL.Imager(   Zpytest r   r   r   r   r   r   r   r    r!   r+   r   r   r   r   <module>   s   	