nan
9c896213a5
- Updated frame extraction logic to allow for single frame extraction. - Integrated model readiness checks for BEN2 and DA3 models in relevant operators. - Added directory name attribute to model specifications for better organization. - Improved device selection logic for production and debug tasks. - Refactored preferences to expose device settings and ensure they reflect available hardware. - Enhanced runtime device detection and management, including support for CUDA and MPS. - Updated tests to cover new device management features and model directory handling. - Removed deprecated download model tests to streamline codebase.
75 lines
2.5 KiB
Python
75 lines
2.5 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 ModelTest(unittest.TestCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
cls.models = load_module(
|
|
"snapform_models",
|
|
"src/snapform/models.py",
|
|
)
|
|
|
|
def test_base_model_is_default_friendly_apache_model(self):
|
|
model = self.models.get_model("BASE")
|
|
|
|
self.assertEqual(model.repository, "depth-anything/DA3-BASE")
|
|
self.assertEqual(model.license_name, "Apache 2.0")
|
|
|
|
def test_large_model_exposes_noncommercial_license(self):
|
|
model = self.models.get_model("LARGE_1_1")
|
|
|
|
self.assertEqual(model.license_name, "CC BY-NC 4.0")
|
|
|
|
def test_giant_models_expose_gaussian_head(self):
|
|
for model_key in ("GIANT_1_1", "NESTED_GIANT_LARGE_1_1"):
|
|
with self.subTest(model_key=model_key):
|
|
self.assertTrue(self.models.get_model(model_key).supports_gs)
|
|
|
|
def test_unknown_model_is_rejected(self):
|
|
with self.assertRaisesRegex(ValueError, "Unknown DA3 model"):
|
|
self.models.get_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, "PramaLLC/BEN2")
|
|
self.assertEqual(model.license_name, "MIT")
|
|
self.assertEqual(
|
|
model.download_patterns,
|
|
("config.json", "model.safetensors"),
|
|
)
|
|
self.assertNotIn("BEN2_BASE", self.models.MODELS)
|
|
|
|
def test_da3_model_directories_share_the_da3_prefix(self):
|
|
for model_key, model in self.models.MODELS.items():
|
|
with self.subTest(model_key=model_key):
|
|
self.assertTrue(
|
|
model.directory_name.startswith("da3_"),
|
|
model.directory_name,
|
|
)
|
|
self.assertEqual(
|
|
self.models.model_directory_name(model_key),
|
|
model.directory_name,
|
|
)
|
|
|
|
def test_ben2_model_directory_uses_ben2_prefix(self):
|
|
self.assertEqual(
|
|
self.models.model_directory_name("BEN2_BASE"),
|
|
"ben2_base",
|
|
)
|