nan
4c902ed7db
- Implemented `image_plane.py` to build the Image Plane Geometry Nodes group with customizable parameters for geometry, subdivide, and thickness. - Created `image_plane_validation.py` to validate the functionality of the image plane, ensuring correct geometry generation and material assignment. - Developed `material_layers.py` to construct shader node groups for image materials, including depth and normal handling. - Authored `plane_layouts.py` to define node arrangements for image geometry assets. - Introduced `shadeless.py` to create a shadeless shader node group that displays color without scene lighting while preserving shadows and reflections.
100 lines
4.0 KiB
Python
100 lines
4.0 KiB
Python
import importlib.util
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
|
|
PROJECT_ROOT = Path(__file__).resolve().parents[1]
|
|
|
|
|
|
def load_module(name, relative_path):
|
|
specification = importlib.util.spec_from_file_location(
|
|
name,
|
|
PROJECT_ROOT / relative_path,
|
|
)
|
|
module = importlib.util.module_from_spec(specification)
|
|
specification.loader.exec_module(module)
|
|
return module
|
|
|
|
|
|
class ModelCatalogTest(unittest.TestCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
cls.models = load_module(
|
|
"anyimage_server_models",
|
|
"src/anyimage/server/model_catalog.py",
|
|
)
|
|
|
|
|
|
def test_unknown_model_is_rejected(self):
|
|
with self.assertRaisesRegex(ValueError, "Unknown model"):
|
|
self.models.get_downloadable_model("UNKNOWN")
|
|
|
|
def test_ben2_download_only_includes_required_hugging_face_files(self):
|
|
model = self.models.get_downloadable_model("BEN2_BASE")
|
|
|
|
self.assertEqual(model.repository, "onnx-community/BEN2-ONNX")
|
|
self.assertEqual(model.license_name, "Apache 2.0")
|
|
self.assertEqual(model.download_patterns, ("onnx/model_fp16.onnx",))
|
|
self.assertEqual(model.ready_patterns, ("onnx/model_fp16.onnx",))
|
|
self.assertEqual(model.r2_files[0][0], "onnx/model_fp16.onnx")
|
|
|
|
|
|
def test_ben2_model_directory_uses_repository_name(self):
|
|
self.assertEqual(
|
|
self.models.model_directory_name("BEN2_BASE"),
|
|
"BEN2-ONNX",
|
|
)
|
|
|
|
def test_upscale_models_use_verified_onnx_artifacts(self):
|
|
expected = {
|
|
"REALESRGAN_GENERAL_WDN_X4V3": ("realesr-general-wdn-x4v3.onnx", 4867419, "b021a837"),
|
|
"SAFMN_REAL_X4": ("safmn-real-x4.onnx", 22657477, "8f13867b"),
|
|
"REALESRGAN_X4PLUS": ("realesrgan-x4plus.onnx", 67051618, "4ed6a45a"),
|
|
}
|
|
self.assertEqual(set(self.models.UPSCALE_MODELS), set(expected) | {"PISA_SR"})
|
|
for key, (filename, size, checksum_prefix) in expected.items():
|
|
with self.subTest(model=key):
|
|
model = self.models.get_downloadable_model(key)
|
|
self.assertEqual(model.download_patterns, (filename,))
|
|
self.assertEqual(model.ready_patterns, (filename,))
|
|
self.assertEqual(model.r2_files[0][:2], (filename, size))
|
|
self.assertTrue(model.r2_files[0][2].startswith(checksum_prefix))
|
|
self.assertEqual(self.models.model_record(model)["family"], "upscale")
|
|
wdn = self.models.UPSCALE_MODELS["REALESRGAN_GENERAL_WDN_X4V3"]
|
|
self.assertEqual(wdn.repository, "xinntao/Real-ESRGAN")
|
|
self.assertEqual(wdn.huggingface_repository, "")
|
|
self.assertEqual(
|
|
self.models.UPSCALE_MODELS["SAFMN_REAL_X4"].huggingface_repository, ""
|
|
)
|
|
|
|
|
|
def test_moge2_models_use_official_onnx_models(self):
|
|
for key, variant in (("MOGE2_VITS_NORMAL", "vits"),
|
|
("MOGE2_VITB_NORMAL", "vitb"),
|
|
("MOGE2_VITL_NORMAL", "vitl")):
|
|
with self.subTest(key=key):
|
|
model = self.models.get_downloadable_model(key)
|
|
self.assertEqual(
|
|
model.repository,
|
|
f"Ruicheng/moge-2-{variant}-normal-onnx",
|
|
)
|
|
self.assertEqual(model.license_name, "MIT")
|
|
self.assertEqual(model.download_patterns, ("model.onnx",))
|
|
self.assertEqual(model.ready_patterns, ("model.onnx",))
|
|
self.assertEqual(model.r2_directory, model.repository.split("/")[-1])
|
|
self.assertEqual(model.r2_files[0][0], "model.onnx")
|
|
|
|
def test_moge2_models_match_uploaded_r2_variants(self):
|
|
self.assertEqual(
|
|
list(self.models.MOGE2_MODELS),
|
|
["MOGE2_VITS_NORMAL", "MOGE2_VITB_NORMAL", "MOGE2_VITL_NORMAL"],
|
|
)
|
|
self.assertEqual(
|
|
[model.r2_directory for model in self.models.MOGE2_MODELS.values()],
|
|
[
|
|
"moge-2-vits-normal-onnx",
|
|
"moge-2-vitb-normal-onnx",
|
|
"moge-2-vitl-normal-onnx",
|
|
],
|
|
)
|