tests: adds remix test image checks + comparison images

Haidra-Org · Mar 19, 2024 · f98cdcd · f98cdcd
1 parent 3be8a54
commit f98cdcd
Show file tree

Hide file tree

Showing 7 changed files with 17 additions and 22 deletions.
diff --git a/images_expected/stable_cascade_image_remix_double.png b/images_expected/stable_cascade_image_remix_double.png
diff --git a/images_expected/stable_cascade_image_remix_double_weak.png b/images_expected/stable_cascade_image_remix_double_weak.png
diff --git a/images_expected/stable_cascade_image_remix_single.png b/images_expected/stable_cascade_image_remix_single.png
diff --git a/images_expected/stable_cascade_image_remix_triple.png b/images_expected/stable_cascade_image_remix_triple.png
diff --git a/images_expected/stable_cascade_image_to_image_n_iter_0.png b/images_expected/stable_cascade_image_to_image_n_iter_0.png
diff --git a/images_expected/stable_cascade_image_to_image_n_iter_1.png b/images_expected/stable_cascade_image_to_image_n_iter_1.png
diff --git a/tests/test_horde_inference_cascade.py b/tests/test_horde_inference_cascade.py
@@ -147,12 +147,7 @@ def test_cascade_image_to_image(
             image_result.image.save(f"images/{img_filename}", quality=100)
             img_pairs_to_check.append((f"images_expected/{img_filename}", image_result.image))
 
-        # assert check_single_inference_image_similarity(
-        #     "images_expected/stable_cascade_image_to_image.png",
-        #     "images/stable_cascade_text_to_image_n_iter_0.png",
-        # )
-
-        # assert check_list_inference_images_similarity(img_pairs_to_check)
+        assert check_list_inference_images_similarity(img_pairs_to_check)
 
     def test_cascade_image_remix_single(
         self,
@@ -187,10 +182,10 @@ def test_cascade_image_remix_single(
         img_filename = "stable_cascade_image_remix_single.png"
         pil_image.save(f"images/{img_filename}", quality=100)
 
-        # assert check_single_inference_image_similarity(
-        #     f"images_expected/{img_filename}",
-        #     pil_image,
-        # )
+        assert check_single_inference_image_similarity(
+            f"images_expected/{img_filename}",
+            pil_image,
+        )
 
     def test_cascade_image_remix_double(
         self,
@@ -230,10 +225,10 @@ def test_cascade_image_remix_double(
         img_filename = "stable_cascade_image_remix_double.png"
         pil_image.save(f"images/{img_filename}", quality=100)
 
-        # assert check_single_inference_image_similarity(
-        #     f"images_expected/{img_filename}",
-        #     pil_image,
-        # )
+        assert check_single_inference_image_similarity(
+            f"images_expected/{img_filename}",
+            pil_image,
+        )
 
     def test_cascade_image_remix_double_weak(
         self,
@@ -274,10 +269,10 @@ def test_cascade_image_remix_double_weak(
         img_filename = "stable_cascade_image_remix_double_weak.png"
         pil_image.save(f"images/{img_filename}", quality=100)
 
-        # assert check_single_inference_image_similarity(
-        #     f"images_expected/{img_filename}",
-        #     pil_image,
-        # )
+        assert check_single_inference_image_similarity(
+            f"images_expected/{img_filename}",
+            pil_image,
+        )
 
     def test_cascade_image_remix_triple(
         self,
@@ -320,7 +315,7 @@ def test_cascade_image_remix_triple(
         img_filename = "stable_cascade_image_remix_triple.png"
         pil_image.save(f"images/{img_filename}", quality=100)
 
-        # assert check_single_inference_image_similarity(
-        #     f"images_expected/{img_filename}",
-        #     pil_image,
-        # )
+        assert check_single_inference_image_similarity(
+            f"images_expected/{img_filename}",
+            pil_image,
+        )