ABX/tests/test_prop_factory.py

160 lines
5.2 KiB
Python
Raw Normal View History

2021-05-29 05:55:12 +00:00
# test_prop_factory
"""
Test custom Blender property-group factory module.
"""
import unittest, os, textwrap, io
# This is the most ridiculous work-around, but it seems to be necessary to
# get Python 3 to import the modules for testing
import sys
print("__file__ = ", __file__)
sys.path.append(os.path.normpath(os.path.join(__file__, '..', '..')))
import yaml
import bpy
from abx import prop_factory
class TestPropertyGroupFactory(unittest.TestCase):
SIMPLE = textwrap.dedent("""\
---
my_custom_property_group:
- code: prop1
name: prop1
type: str
maxlen: 20
default: default
desc: My test property.
""")
SAMPLER = textwrap.dedent("""\
---
sampler:
- code: str_prop
name: MyString
type: str
maxlen: 20
default: str_prop
desc: A string property example.
- code: int_prop
name: MyInteger
type: int
min: 0
max: 100
soft_min: 0
soft_max: 10
default: 1
desc: An integer property example.
- code: float_prop
name: MyFloat
type: float
min: -10.0
max: 10.0
soft_min: 0.0
soft_max: 1.0
precision: 3
unit: 'LENGTH'
default: 1.0
desc: A float property example.
- code: bool_prop
name: MyBoolean
type: bool
default: False
desc: A boolean property example.
- code: enum_prop1
name: MyEnum1
type: enum
items_from: enum1_list
default: 'unknown'
desc: An enumerated property using a list reference.
- code: enum_prop2
name: MyEnum2
type: enum
items:
- op1
- op2
- op3
default: op1
desc: An enumerated property using a direct list of strings.
- code: enum_prop3
name: MyEnum3
items:
- op1
- ('op2', 'Option2')
- ('op3', 'Option3', 'Option Three')
default: op1
desc: An enumerated property using direct list with mixed types.
2021-05-29 05:55:12 +00:00
enum1_list:
- unknown
- option1
- option2
""")
def assertHasAttr(self, ob, attr):
self.assertTrue(hasattr(ob,attr),
msg="Object %s has no attribute %s." % (ob, attr))
def test_creating_simple_yaml_example(self):
schema = yaml.safe_load(io.StringIO(self.SIMPLE))
cpg = prop_factory.PropertyGroupFactory(
'my_custom_property_group', schema)
self.assertHasAttr(cpg, 'prop1')
def test_attach_simple_yaml_to_scene(self):
schema = yaml.safe_load(io.StringIO(self.SIMPLE))
cpg = prop_factory.PropertyGroupFactory(
'my_custom_property_group', schema)
bpy.types.Scene.my_custom_property_group = bpy.props.PointerProperty(type=cpg)
self.assertEqual(
bpy.context.scene.my_custom_property_group.prop1,
'default')
def test_creating_and_attaching_complex_sampler(self):
2021-05-29 05:55:12 +00:00
schema = yaml.safe_load(io.StringIO(self.SAMPLER))
sampler = prop_factory.PropertyGroupFactory(
'sampler', schema)
bpy.types.Scene.sampler = bpy.props.PointerProperty(type=sampler)
self.assertEqual(bpy.context.scene.sampler.str_prop, 'str_prop')
self.assertEqual(bpy.context.scene.sampler.int_prop, 1)
self.assertEqual(bpy.context.scene.sampler.float_prop, 1.0)
self.assertEqual(bpy.context.scene.sampler.bool_prop, False)
self.assertEqual(bpy.context.scene.sampler.enum_prop1, 'unknown')
self.assertEqual(bpy.context.scene.sampler.enum_prop2, 'op1')
self.assertEqual(bpy.context.scene.sampler.enum_prop3, 'op1')
2021-05-29 05:55:12 +00:00
bpy.context.scene.sampler.str_prop = 'my string'
bpy.context.scene.sampler.int_prop = 2
bpy.context.scene.sampler.float_prop = 0.7
bpy.context.scene.sampler.bool_prop = True
bpy.context.scene.sampler.enum_prop1 = 'option1'
bpy.context.scene.sampler.enum_prop2 = 'op2'
bpy.context.scene.sampler.enum_prop3 = 'op3'
self.assertEqual(bpy.context.scene.sampler.str_prop, 'my string')
self.assertEqual(bpy.context.scene.sampler.int_prop, 2)
self.assertAlmostEqual(bpy.context.scene.sampler.float_prop, 0.7)
self.assertEqual(bpy.context.scene.sampler.bool_prop, True)
self.assertEqual(bpy.context.scene.sampler.enum_prop1, 'option1')
self.assertEqual(bpy.context.scene.sampler.enum_prop2, 'op2')
self.assertEqual(bpy.context.scene.sampler.enum_prop3, 'op3')
2021-05-29 05:55:12 +00:00