53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
import unittest
|
|
from types import SimpleNamespace
|
|
from tests.anyimage.common.support import load_common_modules
|
|
|
|
class ModifierInputsTest(unittest.TestCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
cls.modules = load_common_modules()
|
|
|
|
def test_runtime_modifier_enum_input_uses_string_identifier(self):
|
|
enum_items = (
|
|
SimpleNamespace(identifier="Balloon"),
|
|
SimpleNamespace(identifier="Shell"),
|
|
)
|
|
value_property = SimpleNamespace(type="ENUM", enum_items=enum_items)
|
|
input_slot = SimpleNamespace(
|
|
value=None,
|
|
bl_rna=SimpleNamespace(properties={"value": value_property}),
|
|
)
|
|
modifier = SimpleNamespace(
|
|
properties=SimpleNamespace(
|
|
inputs=SimpleNamespace(Mode=input_slot),
|
|
)
|
|
)
|
|
|
|
self.modules.object.set_modifier_input(modifier, "Mode", 1)
|
|
|
|
self.assertEqual(input_slot.value, "Shell")
|
|
|
|
|
|
def test_modifier_inputs_support_rna_and_dictionary_values(self):
|
|
value_property = SimpleNamespace(type="FLOAT")
|
|
input_slot = SimpleNamespace(
|
|
value=None,
|
|
bl_rna=SimpleNamespace(properties={"value": value_property}),
|
|
)
|
|
runtime_modifier = SimpleNamespace(
|
|
properties=SimpleNamespace(
|
|
inputs=SimpleNamespace(Thickness=input_slot),
|
|
)
|
|
)
|
|
|
|
self.modules.object.set_modifier_input(runtime_modifier, "Thickness", 2.0)
|
|
|
|
self.assertEqual(input_slot.value, 2.0)
|
|
|
|
class DictionaryModifier(dict):
|
|
properties = None
|
|
|
|
dictionary_modifier = DictionaryModifier()
|
|
self.modules.object.set_modifier_input(dictionary_modifier, "Mode", 1)
|
|
self.assertEqual(dictionary_modifier["Mode"], 1)
|