
    q[i                        U d Z ddlZddlmZ ddlmZ dZ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d ej                  e ej                  d ej                   e      g       ej"                  ddd                   ej                  dd ej                  e ej                  d ej                   e       ej                   e	       ej                   e
      g       ej"                  dddddej$                  j&                  ej(                  j*                  dddd i!
                   ej                  d"d ej                  e ej                  e ej                   e      g#       ej"                  dd$                   ej                  d%d ej                  e ej                  e ej                   e      g#       ej"                  dddddej$                  j&                  ej(                  j*                  dd&	                  gZeej                     ed'<    ej2                  e e       d(e)      Zej:                  j<                  d*        Zej:                  j<                  d+        Z y),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)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.jpgtest_product_recontextz&only supported in the Vertex AI clientOn a school deskproduct_imagepromptproduct_images   T
image/jpeg)number_of_imagesadd_watermarkoutput_mime_typemodelsourceconfig)nameexception_if_mldev
parameters!test_product_recontext_all_config    i9  FK   imagen_label_keyrecontext_image)
r   r   
base_stepsseedr   safety_filter_levelperson_generationoutput_compression_qualityenhance_promptlabelstest_virtual_try_onperson_imager   r   r   test_virtual_try_on_all_config)	r   r   r    r!   r   r"   r#   r$   r%   
test_tablezmodels.recontext_image)fileglobals_for_filetest_methodr,   c           
        K   t        j                  | t              5  | j                  j                  j                  t        t        j                  dt        j                  t              g      t        j                  dd             d {   }t        |j                        dk(  sJ |j                  d   j                  j                  sJ 	 d d d        y 7 M# 1 sw Y   y xY ww)	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responses     g/mnt/data/GridWatch/venv/lib/python3.12/site-packages/google/genai/tests/models/test_recontext_image.pytest_product_recontext_asyncrA      s     ''
; :ZZ&&66,))%"";
 )))
 7  H x(()Q...$$Q'--9999: :: :s0   C,A3C CAC 	C,C  C)%C,c           
        K   t        j                  | t              5  | j                  j                  j                  t        t        j                  t        t        j                  t              g      t        j                  dd             d {   }t        |j                        dk(  sJ |j                  d   j                  j                   sJ 	 d d d        y 7 M# 1 sw Y   y xY ww)Nr
   r(   r   r   r*   r   r   )r   r   r1   r2   r3   r   !VIRTUAL_TRY_ON_IMAGE_MODEL_LATESTr   r5   PERSON_IMAGEr6   PANTSr8   r9   r:   r;   r<   r=   s     r@   test_virtual_try_on_asyncrF      s     ''
; :ZZ&&66/))%""7
 )))
 7  H x(()Q...$$Q'--9999: :: :s0   C0A7C$C"AC$	C0"C$$C-)C0)!__doc__pytest r   r   r4   rC   Imager7   	BACKPACK2	BACKPACK3rD   rE   TestTableItem_RecontextImageParametersr5   r6   r8   SafetyFilterLevelBLOCK_MEDIUM_AND_ABOVEPersonGenerationALLOW_ADULTr,   list__annotations__setup__file__globals
pytestmarkmarkasynciorA   rF        r@   <module>r]      sD  " !   !I $8 !EKK>	 EKK>	 EKK>	 u{{8 	:	
  M%C25220-5--)&E&&Y?  .5--!""!-
$  M0C25220-5--)&E&&&/ 'E&&&/ 'E&&&/
  .5--!"!-#$)$;$;$R$R"'"8"8"D"D+-$*,=>!
 B  M"C25223-5--)&E&&U;  .5--!"!-
"  M-C25223-5--)&E&&&+  .5--!"!-#$)$;$;$R$R"'"8"8"D"D+-$

K_1
D,,- _@ !]  	Y(	
 : :& : :r\   