nan
489003c2cc
- Introduced color_image.py for preparing standalone color images for tests. - Added hdr_image.py to create HDR textures for testing. - Implemented test_image_depth_panorama.py to evaluate panorama topology, UVs, and depth discontinuities. - Created test_shadeless.py to validate the opacity and HDR color endpoints of the Shadeless group. - Developed moge3_export.py for exporting dynamic MoGe-3 backbone and ONNX graphs. - Added material.py to inherit material slots onto generated geometry. - Built image_depth_panorama.py to construct a quad panorama with depth separation and sampling.
120 lines
3.8 KiB
Python
120 lines
3.8 KiB
Python
from types import SimpleNamespace
|
||
from unittest.mock import patch
|
||
from tests.support.blender import BlenderTestCase
|
||
|
||
|
||
class ImageMenuTest(BlenderTestCase):
|
||
def test_image_menu_keeps_ai_features_clickable_and_offers_models(self):
|
||
calls = []
|
||
labels = {}
|
||
|
||
class Layout:
|
||
operator_context = ""
|
||
enabled = True
|
||
|
||
def operator(self, identifier, **options):
|
||
calls.append((identifier, self.enabled))
|
||
labels[identifier] = options.get("text")
|
||
return SimpleNamespace()
|
||
|
||
def separator(self):
|
||
calls.append(("separator", self.enabled))
|
||
|
||
menu = self.anyimage.AnyImageImageMenu()
|
||
menu.layout = Layout()
|
||
with patch(
|
||
"anyimage.properties.ai_status",
|
||
return_value={
|
||
"environment_ready": True,
|
||
"missing_models": ("BEN2_BASE",),
|
||
"ready": False,
|
||
},
|
||
):
|
||
menu.draw(
|
||
SimpleNamespace(
|
||
object=SimpleNamespace(
|
||
type="EMPTY",
|
||
empty_display_type="IMAGE",
|
||
data=SimpleNamespace(size=(640, 480)),
|
||
)
|
||
)
|
||
)
|
||
|
||
self.assertEqual(
|
||
calls,
|
||
[
|
||
("anyimage.convert_to_plane", True),
|
||
("anyimage.convert_to_depth_plane", True),
|
||
("anyimage.convert_to_relief_plane", True),
|
||
("anyimage.convert_to_panorama", True),
|
||
("separator", True),
|
||
("anyimage.remove_image_background", True),
|
||
("anyimage.upscale_image", True),
|
||
("separator", True),
|
||
("anyimage.setup_ai_environment", True),
|
||
],
|
||
)
|
||
self.assertEqual(
|
||
labels["anyimage.setup_ai_environment"],
|
||
"Download Required Models…",
|
||
)
|
||
|
||
self.assertEqual(
|
||
labels["anyimage.upscale_image"],
|
||
"Upscale (1280 × 960)",
|
||
)
|
||
|
||
|
||
def test_image_menu_opens_environment_settings_when_ai_is_ready(self):
|
||
calls = []
|
||
|
||
class Layout:
|
||
operator_context = ""
|
||
enabled = True
|
||
|
||
def operator(self, identifier, **_options):
|
||
calls.append((identifier, self.enabled))
|
||
return SimpleNamespace()
|
||
|
||
def separator(self):
|
||
calls.append(("separator", self.enabled))
|
||
|
||
menu = self.anyimage.AnyImageImageMenu()
|
||
menu.layout = Layout()
|
||
with patch(
|
||
"anyimage.properties.ai_status",
|
||
return_value={
|
||
"environment_ready": True,
|
||
"missing_models": (),
|
||
"ready": True,
|
||
},
|
||
):
|
||
menu.draw(None)
|
||
|
||
self.assertIn(("anyimage.open_ai_environment_settings", True), calls)
|
||
self.assertNotIn(("anyimage.setup_ai_environment", True), calls)
|
||
|
||
|
||
def test_image_operators_stay_clickable_until_server_is_busy(self):
|
||
context = SimpleNamespace(
|
||
object=SimpleNamespace(type="EMPTY", empty_display_type="IMAGE", data=SimpleNamespace()),
|
||
)
|
||
operators = (
|
||
self.convert_to_plane.ConvertToDepthPlane,
|
||
self.convert_to_plane.ConvertToReliefPlane,
|
||
self.remove_background.RemoveImageBackground,
|
||
)
|
||
|
||
for busy, expected in ((False, True), (True, False)):
|
||
with (
|
||
self.subTest(busy=busy),
|
||
patch.object(
|
||
self.anyimage.runtime,
|
||
"server_busy",
|
||
return_value=busy,
|
||
),
|
||
):
|
||
for operator in operators:
|
||
with self.subTest(operator=operator.__name__):
|
||
self.assertEqual(operator.poll(context), expected)
|