icrdr/AnyImage
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: Bump version to 0.12.13 and update documentation for color space handling

Browse Source
This commit is contained in:
马南
2026-08-19 13:42:29 +08:00
parent c115498b5e
commit 753f87da03
20 changed files with 189 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
tests/test_worker.py
View File

@@ -118,6 +118,7 @@ class WorkerTest(unittest.TestCase):
cls.surface_outputs = load_server_module("outputs.surface")
cls.debug_da3_outputs = load_server_module("outputs.debug_da3")
cls.debug_moge2_outputs = load_server_module("outputs.debug_moge2")
cls.common_job = load_server_module("jobs.common")
cls.depth_job = load_server_module("jobs.depth")
cls.surface_job = load_server_module("jobs.surface")
cls.debug_da3_job = load_server_module("jobs.debug_da3")
@@ -127,20 +128,39 @@ class WorkerTest(unittest.TestCase):
cls.foreground_job = load_server_module("jobs.foreground")
cls.download = load_server_module("model_download")
def test_surface_inference_progress_omits_single_frame_count(self):
def test_inference_progress_omits_single_frame_count(self):
self.assertEqual(
self.surface_job.inference_progress_message("MoGe-2", 0, 1),
self.common_job.frame_progress_message(
"Estimating MoGe-2 geometry", 0, 1
),
"Estimating MoGe-2 geometry",
)
self.assertEqual(
self.surface_job.inference_progress_message("DA3", 0, 1),
self.common_job.frame_progress_message(
"Estimating DA3 geometry", 0, 1
),
"Estimating DA3 geometry",
)
self.assertEqual(
self.surface_job.inference_progress_message("DA3", 1, 3),
self.common_job.frame_progress_message(
"Estimating DA3 geometry", 1, 3
),
"Estimating DA3 geometry (2/3)",
)
self.assertEqual(
self.common_job.frame_progress_message(
"Estimating MoGe-2 depth", 0, 1
),
"Estimating MoGe-2 depth",
)
self.assertEqual(
self.common_job.frame_progress_message(
"Estimating DA3 depth", 0, 1
),
"Estimating DA3 depth",
)
def test_surface_has_one_model_independent_builder(self):
self.assertTrue(callable(self.surface.build_surface))
self.assertTrue(callable(self.surface_outputs.write_surface_archive))