diff --git a/tests/images/outputs/16x16/2x-BIGOLDIES.png b/tests/images/outputs/16x16/2x-BIGOLDIES.png new file mode 100644 index 00000000..9b5ac4a8 Binary files /dev/null and b/tests/images/outputs/16x16/2x-BIGOLDIES.png differ diff --git a/tests/images/outputs/16x16/4x-NMKD-YandereNeo.png b/tests/images/outputs/16x16/4x-NMKD-YandereNeo.png new file mode 100644 index 00000000..39077070 Binary files /dev/null and b/tests/images/outputs/16x16/4x-NMKD-YandereNeo.png differ diff --git a/tests/images/outputs/16x16/8x-ESRGAN.png b/tests/images/outputs/16x16/8x-ESRGAN.png new file mode 100644 index 00000000..f2d8e4d0 Binary files /dev/null and b/tests/images/outputs/16x16/8x-ESRGAN.png differ diff --git a/tests/images/outputs/32x32/2x-BIGOLDIES.png b/tests/images/outputs/32x32/2x-BIGOLDIES.png new file mode 100644 index 00000000..1c7251aa Binary files /dev/null and b/tests/images/outputs/32x32/2x-BIGOLDIES.png differ diff --git a/tests/images/outputs/32x32/4x-NMKD-YandereNeo.png b/tests/images/outputs/32x32/4x-NMKD-YandereNeo.png new file mode 100644 index 00000000..4994bc22 Binary files /dev/null and b/tests/images/outputs/32x32/4x-NMKD-YandereNeo.png differ diff --git a/tests/images/outputs/32x32/8x-ESRGAN.png b/tests/images/outputs/32x32/8x-ESRGAN.png new file mode 100644 index 00000000..8f58b0a4 Binary files /dev/null and b/tests/images/outputs/32x32/8x-ESRGAN.png differ diff --git a/tests/images/outputs/64x64/2x-BIGOLDIES.png b/tests/images/outputs/64x64/2x-BIGOLDIES.png new file mode 100644 index 00000000..281c208d Binary files /dev/null and b/tests/images/outputs/64x64/2x-BIGOLDIES.png differ diff --git a/tests/images/outputs/64x64/4x-NMKD-YandereNeo.png b/tests/images/outputs/64x64/4x-NMKD-YandereNeo.png new file mode 100644 index 00000000..dcdc350e Binary files /dev/null and b/tests/images/outputs/64x64/4x-NMKD-YandereNeo.png differ diff --git a/tests/images/outputs/64x64/8x-ESRGAN.png b/tests/images/outputs/64x64/8x-ESRGAN.png new file mode 100644 index 00000000..46074506 Binary files /dev/null and b/tests/images/outputs/64x64/8x-ESRGAN.png differ diff --git a/tests/images/outputs/8x8/2x-BIGOLDIES.png b/tests/images/outputs/8x8/2x-BIGOLDIES.png new file mode 100644 index 00000000..f7788546 Binary files /dev/null and b/tests/images/outputs/8x8/2x-BIGOLDIES.png differ diff --git a/tests/images/outputs/8x8/4x-NMKD-YandereNeo.png b/tests/images/outputs/8x8/4x-NMKD-YandereNeo.png new file mode 100644 index 00000000..1926320b Binary files /dev/null and b/tests/images/outputs/8x8/4x-NMKD-YandereNeo.png differ diff --git a/tests/images/outputs/8x8/8x-ESRGAN.png b/tests/images/outputs/8x8/8x-ESRGAN.png new file mode 100644 index 00000000..9835bd47 Binary files /dev/null and b/tests/images/outputs/8x8/8x-ESRGAN.png differ diff --git a/tests/test_ESRGAN.py b/tests/test_ESRGAN.py index 95d7ee00..08cab5a2 100644 --- 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):