def testAddValue(self):
"""Tests the AddValue function."""
data_type_definition = data_types.EnumerationDefinition(
u'enum', description=u'enumeration')
data_type_definition.AddValue(u'enum_value', 5)
with self.assertRaises(KeyError):
data_type_definition.AddValue(u'enum_value', 5)
def testAddValue(self):
"""Tests the AddValue function."""
data_type_definition = data_types.EnumerationDefinition(
'enum', description='enumeration')
data_type_definition.AddValue('enum_value', 5, aliases=['value5'])
with self.assertRaises(KeyError):
data_type_definition.AddValue('enum_value', 7, aliases=['value7'])
with self.assertRaises(KeyError):
data_type_definition.AddValue('myenum', 5, aliases=['value7'])
with self.assertRaises(KeyError):
data_type_definition.AddValue('myenum', 7, aliases=['value5'])
def testGetStructFormatString(self):
"""Tests the GetStructFormatString function."""
data_type_definition = data_types.EnumerationDefinition(
u'enum', description=u'enumeration')
struct_format_string = data_type_definition.GetStructFormatString()
self.assertIsNone(struct_format_string)
data_type_definition.size = 1
struct_format_string = data_type_definition.GetStructFormatString()
self.assertEqual(struct_format_string, u'B')
data_type_definition.size = 2
struct_format_string = data_type_definition.GetStructFormatString()
self.assertEqual(struct_format_string, u'H')
data_type_definition.size = 4
struct_format_string = data_type_definition.GetStructFormatString()
self.assertEqual(struct_format_string, u'I')
def testInitialize(self):
"""Tests the __init__ function."""
data_type_definition = data_types.EnumerationDefinition(
u'enum', description=u'enumeration')
self.assertIsNotNone(data_type_definition)