3636import imggen_raw_reference_with_txt_img
3737import imggen_scribble_ctrl_type_with_txt_img
3838import imggen_style_reference_with_txt_img
39- import imggen_subj_refer_ctrl_refer_with_txt_imgs
40- import imggen_upscale_with_img
4139import imggen_virtual_try_on_with_txt_img
42- import imggen_with_txt
4340
4441os .environ ["GOOGLE_GENAI_USE_ENTERPRISE" ] = "True"
4542os .environ ["GOOGLE_CLOUD_LOCATION" ] = "us-central1"
@@ -63,12 +60,6 @@ def output_gcs_uri() -> str:
6360 blob .delete ()
6461
6562
66- def test_img_generation () -> None :
67- OUTPUT_FILE = os .path .join (RESOURCES , "dog_newspaper.png" )
68- response = imggen_with_txt .generate_images (OUTPUT_FILE )
69- assert response
70-
71-
7263def test_img_edit_inpainting_insert_with_mask () -> None :
7364 OUTPUT_FILE = os .path .join (RESOURCES , "fruit_edit.png" )
7465 response = imggen_inpainting_insert_mask_with_txt_img .edit_inpainting_insert_mask (OUTPUT_FILE )
@@ -117,13 +108,6 @@ def test_img_edit_mask_free() -> None:
117108 assert response
118109
119110
120- def test_img_customization_subject (output_gcs_uri : str ) -> None :
121- response = imggen_subj_refer_ctrl_refer_with_txt_imgs .subject_customization (
122- output_gcs_uri = output_gcs_uri
123- )
124- assert response
125-
126-
127111def test_img_customization_style (output_gcs_uri : str ) -> None :
128112 response = imggen_style_reference_with_txt_img .style_customization (output_gcs_uri = output_gcs_uri )
129113 assert response
@@ -148,9 +132,3 @@ def test_img_virtual_try_on() -> None:
148132 OUTPUT_FILE = os .path .join (RESOURCES , "man_in_sweater.png" )
149133 response = imggen_virtual_try_on_with_txt_img .virtual_try_on (OUTPUT_FILE )
150134 assert response
151-
152-
153- def test_img_upscale () -> None :
154- OUTPUT_FILE = os .path .join (RESOURCES , "dog_newspaper.png" )
155- response = imggen_upscale_with_img .upscale_images (OUTPUT_FILE )
156- assert response
0 commit comments