def test_basic(self):
package = ModuleStructure()
module = ModuleStructure()
module.name = "module"
submodule = ModuleStructure()
submodule.name = "submodule"
module.add_child(submodule)
package.add_child(module)
entry_point_1 = EntryPoint()
entry_point_1.type = "class"
entry_point_1.module = "module.submodule"
entry_point_1.name = "ClassA"
entry_point_2 = EntryPoint()
entry_point_2.type = "class"
entry_point_2.module = "module"
entry_point_2.name = "ClassB"
refiner = DataTypeRefiner(package, [entry_point_1, entry_point_2])
self.assertEqual(refiner.get_base_name("module.submodule.ClassA"), "ClassA")
self.assertIsNone(refiner.get_module_name(None))
self.assertEqual(refiner.get_module_name("module.submodule.ClassA"), "module.submodule")
self.assertIsNone(refiner.get_module_name("module.submodule2.ClassA"))
self.assertEqual(refiner.get_module_name("module.ClassB"), "module")
def test_all(self):
entry_point = EntryPoint()
entry_point.type = "constant"
entry_point.module = "module.a"
entry_point.name = "DATA"
self.assertEqual(entry_point.type, "constant")
self.assertEqual(entry_point.module, "module.a")
self.assertEqual(entry_point.name, "DATA")
def test_get_refined_data_type(self):
package = ModuleStructure()
module_1 = ModuleStructure()
module_1.name = "module_1"
submodule_1 = ModuleStructure()
submodule_1.name = "submodule_1"
module_1.add_child(submodule_1)
submodule_2 = ModuleStructure()
submodule_2.name = "submodule_2"
module_1.add_child(submodule_2)
package.add_child(module_1)
module_2 = ModuleStructure()
module_2.name = "module_2"
package.add_child(module_2)
entry_point_1 = EntryPoint()
entry_point_1.type = "class"
entry_point_1.module = "module_1.submodule_1"
entry_point_1.name = "ClassA"
entry_point_2 = EntryPoint()
entry_point_2.type = "class"
entry_point_2.module = "module_1.submodule_2"
entry_point_2.name = "ClassB"
entry_point_3 = EntryPoint()
entry_point_3.type = "class"
entry_point_3.module = "module_1"
entry_point_3.name = "ClassC"
entry_point_4 = EntryPoint()
entry_point_4.type = "class"
entry_point_4.module = "module_1"
entry_point_4.name = "ClassD"
entry_point_5 = EntryPoint()
entry_point_5.type = "class"
entry_point_5.module = "module_2"
entry_point_5.name = "ClassE"
refiner = DataTypeRefiner(package, [entry_point_1, entry_point_2, entry_point_3, entry_point_4, entry_point_5])
unknown_data_type = UnknownDataType()
refined_data_type = refiner.get_refined_data_type(unknown_data_type, "module_1")
self.assertEqual(refined_data_type.type(), 'UNKNOWN')
intermidiate_data_type = IntermidiateDataType("list of int")
refined_data_type = refiner.get_refined_data_type(intermidiate_data_type, "module_1")
self.assertEqual(refined_data_type.type(), 'BUILTIN')
self.assertEqual(refined_data_type.modifier(), "list")
self.assertEqual(refined_data_type.data_type(), "int")
intermidiate_data_type = IntermidiateDataType("dict")
refined_data_type = refiner.get_refined_data_type(intermidiate_data_type, "module_1")
self.assertEqual(refined_data_type.type(), 'MODIFIER')
self.assertEqual(refined_data_type.modifier(), "dict")
intermidiate_data_type = IntermidiateDataType("sequence ")
refined_data_type = refiner.get_refined_data_type(intermidiate_data_type, "module_1")
self.assertEqual(refined_data_type.type(), 'MODIFIER')
self.assertEqual(refined_data_type.modifier(), "list")
intermidiate_data_type = IntermidiateDataType("float")
refined_data_type = refiner.get_refined_data_type(intermidiate_data_type, "module_1")
self.assertEqual(refined_data_type.type(), 'BUILTIN')
self.assertFalse(refined_data_type.has_modifier())
self.assertEqual(refined_data_type.data_type(), "float")
intermidiate_data_type = IntermidiateDataType("string ")
refined_data_type = refiner.get_refined_data_type(intermidiate_data_type, "module_1")
self.assertEqual(refined_data_type.type(), 'BUILTIN')
self.assertFalse(refined_data_type.has_modifier())
self.assertEqual(refined_data_type.data_type(), "str")
intermidiate_data_type = IntermidiateDataType("module_1.ClassC")
refined_data_type = refiner.get_refined_data_type(intermidiate_data_type, "module_1")
self.assertEqual(refined_data_type.type(), 'CUSTOM')
self.assertFalse(refined_data_type.has_modifier())
self.assertEqual(refined_data_type.data_type(), "module_1.ClassC")
intermidiate_data_type = IntermidiateDataType("list of module_1.ClassC")
refined_data_type = refiner.get_refined_data_type(intermidiate_data_type, "module_1")
self.assertEqual(refined_data_type.type(), 'CUSTOM')
self.assertEqual(refined_data_type.modifier(), "list")
self.assertEqual(refined_data_type.data_type(), "module_1.ClassC")
intermidiate_data_type = IntermidiateDataType("module_1.ClassC")
refined_data_type = refiner.get_refined_data_type(intermidiate_data_type, "module_1")
self.assertEqual(refined_data_type.type(), 'CUSTOM')
self.assertEqual(refined_data_type.data_type(), "module_1.ClassC")
intermidiate_data_type = IntermidiateDataType("ClassC")
refined_data_type = refiner.get_refined_data_type(intermidiate_data_type, "module_1")
self.assertEqual(refined_data_type.type(), 'CUSTOM')
self.assertEqual(refined_data_type.data_type(), "module_1.ClassC")
intermidiate_data_type = IntermidiateDataType("ClassC")
refined_data_type = refiner.get_refined_data_type(intermidiate_data_type, "")
self.assertEqual(refined_data_type.type(), 'CUSTOM')
self.assertEqual(refined_data_type.data_type(), "module_1.ClassC")
def assert_equal_commutative(expect, actual):
def compare_func(expect, actual):
if expect["type"] != actual.type():
return False
if expect["modifier"] is None:
if actual.has_modifier():
return False
else:
if expect["modifier"] != actual.modifier():
return False
if expect["data_type"] is None:
if actual.has_modifier():
return False
else:
if expect["data_type"] != actual.data_type():
return False
if expect["modifier_add_info"] is None:
if actual.modifier_add_info():
return False
else:
if expect["modifier_add_info"] != actual.modifier_add_info():
return False
return True
self.assertEqual(len(expect), len(actual))
actual_copied = actual.copy()
for e in expect:
for a in actual_copied[:]:
if compare_func(e, a):
actual_copied.remove(a)
if len(actual_copied) > 0:
self.log("--- Expect ---")
for e in expect:
self.log("{}".format(e))
self.log("--- Actual ---")
for a in actual:
self.log("{}".format(a.to_string()))
self.assertEqual(len(actual_copied), 0)
intermidiate_data_type = IntermidiateDataType("(bpy_prop_collection of float)")
refined_data_type = refiner.get_refined_data_type(intermidiate_data_type, "module_1")
self.assertEqual(refined_data_type.type(), 'MIXIN')
expect = [{
"type": 'BUILTIN',
"modifier": "list",
"data_type": "float",
"modifier_add_info": None,
},
{
"type": 'BUILTIN',
"modifier": "dict",
"data_type": "float",
"modifier_add_info": {"dict_key": "str"},
},
{
"type": 'CUSTOM',
"modifier": None,
"data_type": "bpy_prop_collection",
"modifier_add_info": None,
}]
assert_equal_commutative(expect, refined_data_type.data_types())
intermidiate_data_type = IntermidiateDataType(" bpy_prop_collection of float")
refined_data_type = refiner.get_refined_data_type(intermidiate_data_type, "module_1")
self.assertEqual(refined_data_type.type(), 'MIXIN')
expect = [{
"type": 'BUILTIN',
"modifier": "list",
"data_type": "float",
"modifier_add_info": None,
},
{
"type": 'BUILTIN',
"modifier": "dict",
"data_type": "float",
"modifier_add_info": {"dict_key": "str"},
},
{
"type": 'CUSTOM',
"modifier": None,
"data_type": "bpy_prop_collection",
"modifier_add_info": None,
}]
assert_equal_commutative(expect, refined_data_type.data_types())
intermidiate_data_type = IntermidiateDataType("enum")
refined_data_type = refiner.get_refined_data_type(intermidiate_data_type, "module_1")
self.assertEqual(refined_data_type.type(), 'MIXIN')
expect = [{
"type": 'BUILTIN',
"modifier": None,
"data_type": "str",
"modifier_add_info": None,
},
{
"type": 'BUILTIN',
"modifier": None,
"data_type": "int",
"modifier_add_info": None,
}]
assert_equal_commutative(expect, refined_data_type.data_types())
intermidiate_data_type = IntermidiateDataType("int, float")
refined_data_type = refiner.get_refined_data_type(intermidiate_data_type, "module_1")
self.assertEqual(refined_data_type.type(), 'MIXIN')
expect = [{
"type": 'BUILTIN',
"modifier": None,
"data_type": "float",
"modifier_add_info": None,
},
{
"type": 'BUILTIN',
"modifier": None,
"data_type": "int",
"modifier_add_info": None,
}]
assert_equal_commutative(expect, refined_data_type.data_types())
intermidiate_data_type = IntermidiateDataType("int, module_1.ClassC")
refined_data_type = refiner.get_refined_data_type(intermidiate_data_type, "module_1")
self.assertEqual(refined_data_type.type(), 'MIXIN')
expect = [{
"type": 'BUILTIN',
"modifier": None,
"data_type": "int",
"modifier_add_info": None,
},
{
"type": 'CUSTOM',
"modifier": None,
"data_type": "module_1.ClassC",
"modifier_add_info": None,
}]
assert_equal_commutative(expect, refined_data_type.data_types())
intermidiate_data_type = IntermidiateDataType("module_2.ClassE, module_1.submodule_1.ClassA")
refined_data_type = refiner.get_refined_data_type(intermidiate_data_type, "module_1")
self.assertEqual(refined_data_type.type(), 'MIXIN')
expect = [{
"type": 'CUSTOM',
"modifier": None,
"data_type": "module_2.ClassE",
"modifier_add_info": None,
},
{
"type": 'CUSTOM',
"modifier": None,
"data_type": "module_1.submodule_1.ClassA",
"modifier_add_info": None,
}]
assert_equal_commutative(expect, refined_data_type.data_types())
intermidiate_data_type = IntermidiateDataType("ClassZ")
refined_data_type = refiner.get_refined_data_type(unknown_data_type, "module_1")
self.assertEqual(refined_data_type.type(), 'UNKNOWN')
intermidiate_data_type = IntermidiateDataType("module_1.ClassZ")
refined_data_type = refiner.get_refined_data_type(unknown_data_type, "module_1")
self.assertEqual(refined_data_type.type(), 'UNKNOWN')
def test_get_generation_data_type(self):
package = ModuleStructure()
module_1 = ModuleStructure()
module_1.name = "module_1"
submodule_1 = ModuleStructure()
submodule_1.name = "submodule_1"
module_1.add_child(submodule_1)
submodule_2 = ModuleStructure()
submodule_2.name = "submodule_2"
module_1.add_child(submodule_2)
package.add_child(module_1)
module_2 = ModuleStructure()
module_2.name = "module_2"
package.add_child(module_2)
entry_point_1 = EntryPoint()
entry_point_1.type = "class"
entry_point_1.module = "module_1.submodule_1"
entry_point_1.name = "ClassA"
entry_point_2 = EntryPoint()
entry_point_2.type = "class"
entry_point_2.module = "module_1.submodule_2"
entry_point_2.name = "ClassB"
entry_point_3 = EntryPoint()
entry_point_3.type = "class"
entry_point_3.module = "module_1"
entry_point_3.name = "ClassC"
entry_point_4 = EntryPoint()
entry_point_4.type = "class"
entry_point_4.module = "module_1"
entry_point_4.name = "ClassD"
entry_point_5 = EntryPoint()
entry_point_5.type = "class"
entry_point_5.module = "module_2"
entry_point_5.name = "ClassE"
refiner = DataTypeRefiner(package, [entry_point_1, entry_point_2, entry_point_3, entry_point_4, entry_point_5])
self.assertIsNone(refiner.get_generation_data_type(None, "module_2"))
self.assertEqual(refiner.get_generation_data_type("module_1.ClassC", None), "module_1.ClassC")
self.assertEqual(refiner.get_generation_data_type("module_1.submodule_1.ClassA", "module_2"),
"module_1.submodule_1.ClassA")
self.assertEqual(refiner.get_generation_data_type("module_1.ClassC", "module_1"),
"ClassC")
self.assertEqual(refiner.get_generation_data_type("module_1.submodule_1.ClassA", "module_1.submodule_2"),
"module_1.submodule_1.ClassA")
self.assertEqual(refiner.get_generation_data_type("module_1.ClassC", "module_1.submodule_1"),
"module_1.ClassC")
self.assertEqual(refiner.get_generation_data_type("module_1.submodule_1.ClassA", "module_1"),
"module_1.submodule_1.ClassA")