Add image tests for newly added model tests

chaiNNer-org · Nov 20, 2023 · 4aabff3 · 4aabff3
1 parent 8700592
commit 4aabff3
Show file tree

Hide file tree

Showing 13 changed files with 15 additions and 0 deletions.
diff --git a/tests/images/outputs/16x16/2x-BIGOLDIES.png b/tests/images/outputs/16x16/2x-BIGOLDIES.png
diff --git a/tests/images/outputs/16x16/4x-NMKD-YandereNeo.png b/tests/images/outputs/16x16/4x-NMKD-YandereNeo.png
diff --git a/tests/images/outputs/16x16/8x-ESRGAN.png b/tests/images/outputs/16x16/8x-ESRGAN.png
diff --git a/tests/images/outputs/32x32/2x-BIGOLDIES.png b/tests/images/outputs/32x32/2x-BIGOLDIES.png
diff --git a/tests/images/outputs/32x32/4x-NMKD-YandereNeo.png b/tests/images/outputs/32x32/4x-NMKD-YandereNeo.png
diff --git a/tests/images/outputs/32x32/8x-ESRGAN.png b/tests/images/outputs/32x32/8x-ESRGAN.png
diff --git a/tests/images/outputs/64x64/2x-BIGOLDIES.png b/tests/images/outputs/64x64/2x-BIGOLDIES.png
diff --git a/tests/images/outputs/64x64/4x-NMKD-YandereNeo.png b/tests/images/outputs/64x64/4x-NMKD-YandereNeo.png
diff --git a/tests/images/outputs/64x64/8x-ESRGAN.png b/tests/images/outputs/64x64/8x-ESRGAN.png
diff --git a/tests/images/outputs/8x8/2x-BIGOLDIES.png b/tests/images/outputs/8x8/2x-BIGOLDIES.png
diff --git a/tests/images/outputs/8x8/4x-NMKD-YandereNeo.png b/tests/images/outputs/8x8/4x-NMKD-YandereNeo.png
diff --git a/tests/images/outputs/8x8/8x-ESRGAN.png b/tests/images/outputs/8x8/8x-ESRGAN.png
diff --git a/tests/test_ESRGAN.py b/tests/test_ESRGAN.py
@@ -25,6 +25,11 @@ def test_ESRGAN_community_2x(snapshot):
     model = ModelLoader().load_from_file(file.path)
     assert model == snapshot(exclude=disallowed_props)
     assert isinstance(model.model, RRDBNet)
+    assert_image_inference(
+        file,
+        model,
+        [TestImage.SR_8, TestImage.SR_16, TestImage.SR_32, TestImage.SR_64],
+    )
 
 
 def test_ESRGAN_community_4x(snapshot):
@@ -34,6 +39,11 @@ def test_ESRGAN_community_4x(snapshot):
     model = ModelLoader().load_from_file(file.path)
     assert model == snapshot(exclude=disallowed_props)
     assert isinstance(model.model, RRDBNet)
+    assert_image_inference(
+        file,
+        model,
+        [TestImage.SR_8, TestImage.SR_16, TestImage.SR_32, TestImage.SR_64],
+    )
 
 
 def test_ESRGAN_community_8x(snapshot):
@@ -43,6 +53,11 @@ def test_ESRGAN_community_8x(snapshot):
     model = ModelLoader().load_from_file(file.path)
     assert model == snapshot(exclude=disallowed_props)
     assert isinstance(model.model, RRDBNet)
+    assert_image_inference(
+        file,
+        model,
+        [TestImage.SR_8, TestImage.SR_16, TestImage.SR_32, TestImage.SR_64],
+    )
 
 
 def test_BSRGAN(snapshot):