
    q[i                         U d Z ddlZddlZddlmZ ddlmZ dZej                  j                  ej                  j                  ej                  j                  e      d            Zej                  j                  e	      Zej                  j                  ej                  j                  ej                  j                  e      d
            Zej                  j                  e	      Zej                  j                  ej                  j                  ej                  j                  e      d            Zej                  j                  e	      Z ej(                  dd ej*                  e ej,                  e       ej.                  ej0                  j2                  ddddddi                   ej(                  dd ej*                  e ej,                  e       ej.                  ej0                  j4                                     ej(                  dd ej*                  e ej,                  ed       ej.                  ej0                  j6                                     ej(                  dd ej*                  e ej,                  ed       ej.                  ej0                  j8                                     ej(                  dd ej*                  e ej,                  e ej:                  e             ej.                  ej0                  j<                                    gZe ej(                     e!d <    ejD                  e e#       d!e"      Z$ejJ                  jL                  d#        Z'y)$zTests for segment_image.    N   )types   )pytest_helperzimage-segmentation-001z../data/google.png)locationz ../data/skateboard_stop_sign.jpgz!../data/segmentation_scribble.jpgtest_segment_foregroundz&only supported in the Vertex AI clientimage   {Gz?b   imagen_label_keysegment_image)modemax_predictionsconfidence_thresholdmask_dilationbinary_color_thresholdlabelsmodelsourceconfig)nameexception_if_mldev
parameterstest_segment_background)r   test_segment_promptzThe letter G)r
   prompttest_segment_semantic
skateboardtest_segment_interactive)r
   scribble_image
test_tablezmodels.segment_image)fileglobals_for_filetest_methodr$   c                   K   t        j                  | t              5  | j                  j                  j                  t        t        j                  t              t        j                  t        j                  j                  dddd             d {   }|j                  d   j                  j                  sJ t!        |j                        dk(  sJ |j                  d   j"                  d   j$                  dk(  sJ |j                  d   j"                  d   j&                  dkD  sJ 	 d d d        y 7 # 1 sw Y   y xY ww)	Nr	   r   r   r   )r   r   r   r   r   r   r   
foreground)r   r   
ValueErroraiomodelsr   SEGMENT_IMAGE_MODEL_LATESTr   SegmentImageSourceSOURCE_IMAGE2SegmentImageConfigSegmentMode
FOREGROUNDgenerated_masksmaskimage_byteslenr   labelscore)clientresponses     e/mnt/data/GridWatch/venv/lib/python3.12/site-packages/google/genai/tests/models/test_segment_image.pytest_segment_foreground_asyncr<      s'    ''
; ;ZZ&&44(''
 ''""--!%#%
 5  H ##A&++7777x''(A---##A&--a066,FFF##A&--a066:::#; ;; ;s0   EA8D;D9BD;0	E9D;;E E)(__doc__ospytest r   r   r-   pathabspathjoindirname__file__SOURCE_IMAGE_FILE_PATH1Image	from_fileSOURCE_IMAGE1SOURCE_IMAGE_FILE_PATH2r/   SCRIBBLE_IMAGE_FILE_PATHSCRIBBLE_IMAGETestTableItem_SegmentImageParametersr.   r0   r1   r2   
BACKGROUNDPROMPTSEMANTICScribbleImageINTERACTIVEr$   list__annotations__setupglobals
pytestmarkmarkasyncior<        r;   <module>r]      se  "  	   5 ''//GGLL*,@A  %%/F%G''//GGLL*,NO  %%/F%G77??GGLL*,OP  &&0H&I  M&C0500,+5++# ,5++&&11 !%)"')*O<
$  M&C0500,+5++# ,5++&&11
  M"C0500,+5++#% ,5++&&--	
  M$C0500,+5++## ,5++&&//	
  M'C0500,+5++#2u22H ,5++&&22	
yJ1
D,,- JV !]  	Y&	
 ; ;r\   