2222
2323from google .cloud import storage
2424
25- import pytest
26-
2725import imggen_canny_ctrl_type_with_txt_img
2826import imggen_inpainting_insert_mask_with_txt_img
2927import imggen_inpainting_insert_with_txt_img
30- import imggen_inpainting_removal_mask_with_txt_img
31- import imggen_inpainting_removal_with_txt_img
32- import imggen_mask_free_edit_with_txt_img
3328import imggen_outpainting_with_txt_img
3429import imggen_product_background_mask_with_txt_img
3530import imggen_product_background_with_txt_img
4136import imggen_virtual_try_on_with_txt_img
4237import imggen_with_txt
4338
39+ import pytest
40+
4441os .environ ["GOOGLE_GENAI_USE_ENTERPRISE" ] = "True"
4542os .environ ["GOOGLE_CLOUD_LOCATION" ] = "us-central1"
4643# The project name is included in the CICD pipeline
@@ -81,18 +78,6 @@ def test_img_edit_inpainting_insert() -> None:
8178 assert response
8279
8380
84- def test_img_edit_inpainting_removal_mask () -> None :
85- OUTPUT_FILE = os .path .join (RESOURCES , "fruit_edit.png" )
86- response = imggen_inpainting_removal_mask_with_txt_img .edit_inpainting_removal_mask (OUTPUT_FILE )
87- assert response
88-
89-
90- def test_img_edit_inpainting_removal () -> None :
91- OUTPUT_FILE = os .path .join (RESOURCES , "fruit_edit.png" )
92- response = imggen_inpainting_removal_with_txt_img .edit_inpainting_removal (OUTPUT_FILE )
93- assert response
94-
95-
9681def test_img_edit_product_background_mask () -> None :
9782 OUTPUT_FILE = os .path .join (RESOURCES , "suitcase_edit.png" )
9883 response = imggen_product_background_mask_with_txt_img .edit_product_background_mask (OUTPUT_FILE )
@@ -111,12 +96,6 @@ def test_img_edit_outpainting() -> None:
11196 assert response
11297
11398
114- def test_img_edit_mask_free () -> None :
115- OUTPUT_FILE = os .path .join (RESOURCES , "latte_edit.png" )
116- response = imggen_mask_free_edit_with_txt_img .edit_mask_free (OUTPUT_FILE )
117- assert response
118-
119-
12099def test_img_customization_subject (output_gcs_uri : str ) -> None :
121100 response = imggen_subj_refer_ctrl_refer_with_txt_imgs .subject_customization (
122101 output_gcs_uri = output_gcs_uri
0 commit comments