o
    Yi                     @   s  U d Z ddlZddlmZ ddlmZ dZdZejd	d
Zejdd
Z	ejdd
Z
ejdd
Zejdd
Zejddejeejdejedgdejddddddejddejeejdejedeje	deje
dgdejdddddejjejjdddd id!
ddejd"dejeejeejedgd#ejddd$ddejd%dejeejeejedgd#ejdddddejjejjddd&	ddgZeej ed'< ejee d(ed)Zejjd*d+ Zejjd,d- Z dS ).zTests for recontext_image.    N   )types   )pytest_helperz&imagen-product-recontext-preview-06-30zvirtual-try-on-001z0gs://genai-sdk-tests/inputs/images/backpack1.png)Zgcs_uriz0gs://genai-sdk-tests/inputs/images/backpack2.pngz0gs://genai-sdk-tests/inputs/images/backpack3.pngz*gs://genai-sdk-tests/inputs/images/man.jpgz,gs://genai-sdk-tests/inputs/images/pants.jpgZtest_product_recontextz&only supported in the Vertex AI clientOn a school deskZproduct_imagepromptproduct_images   T
image/jpeg)number_of_imagesadd_watermarkoutput_mime_typemodelsourceconfig)nameexception_if_mldev
parametersZ!test_product_recontext_all_config    i9  FK   Zimagen_label_keyrecontext_image)
r   r   
base_stepsseedr   safety_filter_levelperson_generationoutput_compression_qualityenhance_promptlabelsZtest_virtual_try_onZperson_imager
   r   r   Ztest_virtual_try_on_all_config)	r   r   r   r   r   r   r   r   r   
test_tablezmodels.recontext_image)fileZglobals_for_fileZtest_methodr#   c                    s   t | t7 | jjjttjdtj	t
dgdtjddddI d H }t|jdks-J |jd jjs6J W d    d S 1 sAw   Y  d S )	Nr   r   r   r   r   r"   r   r   )r   r   
ValueErroraiomodelsr   PRODUCT_RECONTEXT_MODEL_LATESTr   RecontextImageSourceProductImage	BACKPACK1RecontextImageConfiglengenerated_imagesimageimage_bytesclientresponse r4   h/home/kim/smarthome/.venv/lib/python3.10/site-packages/google/genai/tests/models/test_recontext_image.pytest_product_recontext_async   "   
"r6   c                    s   t | t7 | jjjttjt	tj
tdgdtjddddI d H }t|jdks-J |jd jjs6J W d    d S 1 sAw   Y  d S )Nr   r!   r   r   r"   r   r   )r   r   r%   r&   r'   r   !VIRTUAL_TRY_ON_IMAGE_MODEL_LATESTr   r)   PERSON_IMAGEr*   PANTSr,   r-   r.   r/   r0   r1   r4   r4   r5   test_virtual_try_on_async   r7   r;   )!__doc__Zpytest r   r   r(   r8   ZImager+   Z	BACKPACK2Z	BACKPACK3r9   r:   ZTestTableItemZ_RecontextImageParametersr)   r*   r,   ZSafetyFilterLevelZBLOCK_MEDIUM_AND_ABOVEZPersonGenerationZALLOW_ADULTr#   list__annotations__setup__file__globalsZ
pytestmarkmarkZasyncior6   r;   r4   r4   r4   r5   <module>   s   
!
`
