Commit bbc13e9d authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): better unittests

parent 3ba4caac

test/restore/conftest.py

+5 −0
Original line number Diff line number Diff line
@@ -39,3 +39,8 @@ def q45_image(sample_image): 
        img_file = os.path.join(td, 'image.jpg')

        sample_image.save(img_file, quality=45)

        yield load_image(img_file)





@pytest.fixture()

def rgba_image():

    yield load_image(get_testfile('rgba_restore.png'), mode='RGBA', force_background=None)

test/restore/test_nafnet.py

+7 −0
Original line number Diff line number Diff line 
import pytest



from imgutils.data import grid_transparent

from imgutils.metrics import psnr

from imgutils.restore import restore_with_nafnet

from imgutils.restore.nafnet import _open_nafnet_model
@@ -20,3 +21,9 @@ class TestRestoreNafNet: 


    def test_restore_with_nafnet_q45(self, q45_image, clear_image):

        assert psnr(restore_with_nafnet(q45_image), clear_image) >= 40.0



    def test_restore_with_nafnet_rgba(self, rgba_image):

        assert rgba_image.mode == 'RGBA'

        restored_image = restore_with_nafnet(rgba_image)

        assert restored_image.mode == 'RGBA'

        assert psnr(grid_transparent(restored_image), grid_transparent(rgba_image), ) >= 35

test/restore/test_scunet.py

+7 −0
Original line number Diff line number Diff line 
import pytest



from imgutils.data import grid_transparent

from imgutils.metrics import psnr

from imgutils.restore import restore_with_scunet

from imgutils.restore.scunet import _open_scunet_model
@@ -23,3 +24,9 @@ class TestRestoreSCUNet: 


    def test_restore_with_scunet_gnoise(self, gaussian_noise_image, clear_image):

        assert psnr(restore_with_scunet(gaussian_noise_image), clear_image) >= 33



    def test_restore_with_scunet_rgba(self, rgba_image):

        assert rgba_image.mode == 'RGBA'

        restored_image = restore_with_scunet(rgba_image)

        assert restored_image.mode == 'RGBA'

        assert psnr(grid_transparent(restored_image), grid_transparent(rgba_image), ) >= 35

test/testfile/rgba_restore.png

0 → 100644
+662 KiB
Loading image diff...