lucasrufkahr/Real-ESRGAN
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add unittest for model and utils

Browse Source
This commit is contained in:
Xintao
2021-11-28 19:54:19 +08:00
parent 1d180efaf3
commit 42110857ef
8 changed files with 407 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_dataset.py
View File

@@ -7,7 +7,7 @@ from realesrgan.data.realesrgan_paired_dataset import RealESRGANPairedDataset
def test_realesrgan_dataset():
with open('tests/data/demo_option_realesrgan_dataset.yml', mode='r') as f:
with open('tests/data/test_realesrgan_dataset.yml', mode='r') as f:
opt = yaml.load(f, Loader=yaml.FullLoader)
dataset = RealESRGANDataset(opt)
@@ -81,7 +81,7 @@ def test_realesrgan_dataset():
def test_realesrgan_paired_dataset():
with open('tests/data/demo_option_realesrgan_paired_dataset.yml', mode='r') as f:
with open('tests/data/test_realesrgan_paired_dataset.yml', mode='r') as f:
opt = yaml.load(f, Loader=yaml.FullLoader)
dataset = RealESRGANPairedDataset(opt)