2222
2323from google .cloud import storage
2424
25- import imggen_canny_ctrl_type_with_txt_img
26- import imggen_inpainting_insert_mask_with_txt_img
27- import imggen_inpainting_insert_with_txt_img
28- import imggen_outpainting_with_txt_img
29- import imggen_product_background_mask_with_txt_img
30- import imggen_product_background_with_txt_img
3125import imggen_raw_reference_with_txt_img
3226import imggen_scribble_ctrl_type_with_txt_img
3327import imggen_style_reference_with_txt_img
@@ -66,36 +60,6 @@ def test_img_generation() -> None:
6660 assert response
6761
6862
69- def test_img_edit_inpainting_insert_with_mask () -> None :
70- OUTPUT_FILE = os .path .join (RESOURCES , "fruit_edit.png" )
71- response = imggen_inpainting_insert_mask_with_txt_img .edit_inpainting_insert_mask (OUTPUT_FILE )
72- assert response
73-
74-
75- def test_img_edit_inpainting_insert () -> None :
76- OUTPUT_FILE = os .path .join (RESOURCES , "fruit_edit.png" )
77- response = imggen_inpainting_insert_with_txt_img .edit_inpainting_insert (OUTPUT_FILE )
78- assert response
79-
80-
81- def test_img_edit_product_background_mask () -> None :
82- OUTPUT_FILE = os .path .join (RESOURCES , "suitcase_edit.png" )
83- response = imggen_product_background_mask_with_txt_img .edit_product_background_mask (OUTPUT_FILE )
84- assert response
85-
86-
87- def test_img_edit_product_background () -> None :
88- OUTPUT_FILE = os .path .join (RESOURCES , "suitcase_edit.png" )
89- response = imggen_product_background_with_txt_img .edit_product_background (OUTPUT_FILE )
90- assert response
91-
92-
93- def test_img_edit_outpainting () -> None :
94- OUTPUT_FILE = os .path .join (RESOURCES , "living_room_edit.png" )
95- response = imggen_outpainting_with_txt_img .edit_outpainting (OUTPUT_FILE )
96- assert response
97-
98-
9963def test_img_customization_subject (output_gcs_uri : str ) -> None :
10064 response = imggen_subj_refer_ctrl_refer_with_txt_imgs .subject_customization (
10165 output_gcs_uri = output_gcs_uri
@@ -118,11 +82,6 @@ def test_img_customization_scribble(output_gcs_uri: str) -> None:
11882 assert response
11983
12084
121- def test_img_customization_canny_edge (output_gcs_uri : str ) -> None :
122- response = imggen_canny_ctrl_type_with_txt_img .canny_edge_customization (output_gcs_uri = output_gcs_uri )
123- assert response
124-
125-
12685def test_img_virtual_try_on () -> None :
12786 OUTPUT_FILE = os .path .join (RESOURCES , "man_in_sweater.png" )
12887 response = imggen_virtual_try_on_with_txt_img .virtual_try_on (OUTPUT_FILE )
0 commit comments