render/rendertest.py

@@ -5,7 +5,6 @@ import concurrent.futures
 import json
 import os
 import pathlib
-import re
 import shutil
 import subprocess
 
@@ -14,11 +13,17 @@ imagecompare = f'{ROOT}/imagecompare'
 WORKDIR = f'{ROOT}/work'
 REPORT_ROOT = f'{ROOT}' # FIXME should be workdir?
 
+# TODO load all saves from the save/ dir
 # TODO rename to <map>_<description>_<issue(opt)>
-saves = []
+saves = [
-for (_, _, files) in os.walk(os.path.join(ROOT, 'save')):
+	'brush2_01',
-	for file in files:
+	'brush_01',
-		saves.append(file.removesuffix('.sav').removeprefix('rendertest_'))
+	'brush_02',
+	'c0a0d_emissive',
+	'light_01',
+	'additive_cullback',
+	'c1a3_fan_material_669',
+]
 
 channels = {
 	'full': '',
@@ -37,12 +42,10 @@ def test_list(arg: str) -> [str]:
 	items = arg.split(',')
 	tests = []
 	for item in items:
-		r = re.compile(item)
+		# TODO regex?
-		for save in saves:
+		if not item in saves:
-			if r.match(save):
+			raise argparse.ArgumentTypeError(f'Test {item} is not available. Available tests are: {saves}')
-				tests.append(save)
+		tests.append(item)
-	if not tests:
-		raise argparse.ArgumentTypeError(f'No tests match {item}. Available tests are: {saves}')
 	return tests
 
 parser = argparse.ArgumentParser(description='Generate scripts and makefiles for rendertest')