nan
838af0ee96
- Added ai_host.py to handle AI tasks with FastAPI integration. - Introduced HostState and ModelCache classes for managing client connections and model loading. - Implemented job submission and cancellation endpoints in the AI host. - Enhanced tests for AI host functionality, including client tracking and job cancellation. - Updated existing tests to accommodate changes in the operator and command structure. - Refactored runtime commands to use request-based architecture instead of command-line arguments. - Improved model loading logic to handle CUDA memory management and reuse models efficiently.
44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
import importlib.util
|
|
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
from zipfile import ZipFile
|
|
|
|
|
|
PROJECT_ROOT = Path(__file__).resolve().parents[1]
|
|
|
|
|
|
def load_pack():
|
|
specification = importlib.util.spec_from_file_location(
|
|
"snapform_pack",
|
|
PROJECT_ROOT / "pack.py",
|
|
)
|
|
module = importlib.util.module_from_spec(specification)
|
|
specification.loader.exec_module(module)
|
|
return module
|
|
|
|
|
|
class PackTest(unittest.TestCase):
|
|
def test_builds_extension_root_layout(self):
|
|
pack = load_pack()
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
original_dist = pack.DIST_DIR
|
|
pack.DIST_DIR = Path(directory)
|
|
try:
|
|
output_path = pack.build_extension()
|
|
finally:
|
|
pack.DIST_DIR = original_dist
|
|
|
|
self.assertEqual(output_path.name, "SnapForm.v0.9.28.zip")
|
|
with ZipFile(output_path) as archive:
|
|
names = set(archive.namelist())
|
|
|
|
self.assertIn("__init__.py", names)
|
|
self.assertIn("blender_manifest.toml", names)
|
|
self.assertIn("image_depth.py", names)
|
|
self.assertNotIn("snapform_depth_plane.py", names)
|
|
self.assertIn("worker.py", names)
|
|
self.assertIn("ai_host.py", names)
|
|
self.assertNotIn("importer.py", names)
|
|
self.assertNotIn("snapform/__init__.py", names)
|