def test_valid_pack(self): """Test packing valid enum values.""" field = ('a', 'b', SimpleEnum) # signed byte: -128, 127 out = ElementEnum.valid(field) self.assertTrue(out) elem = ElementEnum(field) test_values = [ ({ 'a': 0 }, b'\x00'), ({ 'a': 2 }, b'\x02'), ({ 'a': SimpleEnum.one }, b'\x01'), ({ 'a': SimpleEnum.two }, b'\x02'), ] for (in_val, out_val) in test_values: with self.subTest((out_val, in_val)): # pylint: disable=no-member ret = elem.pack(in_val) self.assertEqual(ret, out_val)
def test_invalid_pack(self): """Test packing invalid enum values.""" field = ('a', 'b', SimpleEnum) # signed byte: -128, 127 out = ElementEnum.valid(field) self.assertTrue(out) elem = ElementEnum(field) test_values = [ {'a': -1}, {'a': 3}, {'a': 'three'}, {'a': 'ONE'}, ] msg = '{} is not a valid {}' for val in test_values: with self.subTest(val): # pylint: disable=no-member with self.assertRaises(ValueError) as cm: elem.pack(val) self.assertEqual(str(cm.exception), msg.format(val['a'], 'SimpleEnum'))
def test_valid_pack(self): """Test packing valid enum values.""" field = ('a', 'b', SimpleEnum) # signed byte: -128, 127 out = ElementEnum.valid(field) self.assertTrue(out) elem = ElementEnum(field) test_values = [ ({'a': 0}, b'\x00'), ({'a': 2}, b'\x02'), ({'a': SimpleEnum.one}, b'\x01'), ({'a': SimpleEnum.two}, b'\x02'), ({'a': 'two'}, b'\x02'), ] for (in_val, out_val) in test_values: with self.subTest((out_val, in_val)): # pylint: disable=no-member ret = elem.pack(in_val) self.assertEqual(ret, out_val)