add unittest for dataset and archs

2021-11-28 15:59:14 +08:00
parent 7dd860a881
commit 1d180efaf3
18 changed files with 227 additions and 2 deletions
--- a/setup.cfg
+++ b/setup.cfg
@@ -17,7 +17,7 @@ line_length = 120
 multi_line_output = 0
 known_standard_library = pkg_resources,setuptools
 known_first_party = realesrgan
-known_third_party = PIL,basicsr,cv2,numpy,torch,torchvision,tqdm
+known_third_party = PIL,basicsr,cv2,numpy,pytest,torch,torchvision,tqdm,yaml
 no_lines_before = STDLIB,LOCALFOLDER
 default_section = THIRDPARTY

@@ -25,3 +25,9 @@ default_section = THIRDPARTY
 skip = .git,./docs/build
 count =
 quiet-level = 3
+
+[aliases]
+test=pytest
+
+[tool:pytest]
+addopts=tests/