Straddle Python 2.6, 2.7, 3.3, 3.4.
This commit is contained in:
parent
3d5460dd90
commit
a2abc947c0
10 changed files with 123 additions and 123 deletions
|
@ -49,17 +49,17 @@ class DescriptorDatabaseTest(unittest.TestCase):
|
|||
factory_test2_pb2.DESCRIPTOR.serialized_pb)
|
||||
db.Add(file_desc_proto)
|
||||
|
||||
self.assertEquals(file_desc_proto, db.FindFileByName(
|
||||
self.assertEqual(file_desc_proto, db.FindFileByName(
|
||||
'google/protobuf/internal/factory_test2.proto'))
|
||||
self.assertEquals(file_desc_proto, db.FindFileContainingSymbol(
|
||||
self.assertEqual(file_desc_proto, db.FindFileContainingSymbol(
|
||||
'google.protobuf.python.internal.Factory2Message'))
|
||||
self.assertEquals(file_desc_proto, db.FindFileContainingSymbol(
|
||||
self.assertEqual(file_desc_proto, db.FindFileContainingSymbol(
|
||||
'google.protobuf.python.internal.Factory2Message.NestedFactory2Message'))
|
||||
self.assertEquals(file_desc_proto, db.FindFileContainingSymbol(
|
||||
self.assertEqual(file_desc_proto, db.FindFileContainingSymbol(
|
||||
'google.protobuf.python.internal.Factory2Enum'))
|
||||
self.assertEquals(file_desc_proto, db.FindFileContainingSymbol(
|
||||
self.assertEqual(file_desc_proto, db.FindFileContainingSymbol(
|
||||
'google.protobuf.python.internal.Factory2Message.NestedFactory2Enum'))
|
||||
self.assertEquals(file_desc_proto, db.FindFileContainingSymbol(
|
||||
self.assertEqual(file_desc_proto, db.FindFileContainingSymbol(
|
||||
'google.protobuf.python.internal.MessageWithNestedEnumOnly.NestedEnum'))
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
|
|
@ -65,15 +65,15 @@ class DescriptorPoolTest(unittest.TestCase):
|
|||
name1 = 'google/protobuf/internal/factory_test1.proto'
|
||||
file_desc1 = self.pool.FindFileByName(name1)
|
||||
self.assertIsInstance(file_desc1, descriptor.FileDescriptor)
|
||||
self.assertEquals(name1, file_desc1.name)
|
||||
self.assertEquals('google.protobuf.python.internal', file_desc1.package)
|
||||
self.assertEqual(name1, file_desc1.name)
|
||||
self.assertEqual('google.protobuf.python.internal', file_desc1.package)
|
||||
self.assertIn('Factory1Message', file_desc1.message_types_by_name)
|
||||
|
||||
name2 = 'google/protobuf/internal/factory_test2.proto'
|
||||
file_desc2 = self.pool.FindFileByName(name2)
|
||||
self.assertIsInstance(file_desc2, descriptor.FileDescriptor)
|
||||
self.assertEquals(name2, file_desc2.name)
|
||||
self.assertEquals('google.protobuf.python.internal', file_desc2.package)
|
||||
self.assertEqual(name2, file_desc2.name)
|
||||
self.assertEqual('google.protobuf.python.internal', file_desc2.package)
|
||||
self.assertIn('Factory2Message', file_desc2.message_types_by_name)
|
||||
|
||||
def testFindFileByNameFailure(self):
|
||||
|
@ -84,17 +84,17 @@ class DescriptorPoolTest(unittest.TestCase):
|
|||
file_desc1 = self.pool.FindFileContainingSymbol(
|
||||
'google.protobuf.python.internal.Factory1Message')
|
||||
self.assertIsInstance(file_desc1, descriptor.FileDescriptor)
|
||||
self.assertEquals('google/protobuf/internal/factory_test1.proto',
|
||||
self.assertEqual('google/protobuf/internal/factory_test1.proto',
|
||||
file_desc1.name)
|
||||
self.assertEquals('google.protobuf.python.internal', file_desc1.package)
|
||||
self.assertEqual('google.protobuf.python.internal', file_desc1.package)
|
||||
self.assertIn('Factory1Message', file_desc1.message_types_by_name)
|
||||
|
||||
file_desc2 = self.pool.FindFileContainingSymbol(
|
||||
'google.protobuf.python.internal.Factory2Message')
|
||||
self.assertIsInstance(file_desc2, descriptor.FileDescriptor)
|
||||
self.assertEquals('google/protobuf/internal/factory_test2.proto',
|
||||
self.assertEqual('google/protobuf/internal/factory_test2.proto',
|
||||
file_desc2.name)
|
||||
self.assertEquals('google.protobuf.python.internal', file_desc2.package)
|
||||
self.assertEqual('google.protobuf.python.internal', file_desc2.package)
|
||||
self.assertIn('Factory2Message', file_desc2.message_types_by_name)
|
||||
|
||||
def testFindFileContainingSymbolFailure(self):
|
||||
|
@ -105,72 +105,72 @@ class DescriptorPoolTest(unittest.TestCase):
|
|||
msg1 = self.pool.FindMessageTypeByName(
|
||||
'google.protobuf.python.internal.Factory1Message')
|
||||
self.assertIsInstance(msg1, descriptor.Descriptor)
|
||||
self.assertEquals('Factory1Message', msg1.name)
|
||||
self.assertEquals('google.protobuf.python.internal.Factory1Message',
|
||||
self.assertEqual('Factory1Message', msg1.name)
|
||||
self.assertEqual('google.protobuf.python.internal.Factory1Message',
|
||||
msg1.full_name)
|
||||
self.assertEquals(None, msg1.containing_type)
|
||||
self.assertEqual(None, msg1.containing_type)
|
||||
|
||||
nested_msg1 = msg1.nested_types[0]
|
||||
self.assertEquals('NestedFactory1Message', nested_msg1.name)
|
||||
self.assertEquals(msg1, nested_msg1.containing_type)
|
||||
self.assertEqual('NestedFactory1Message', nested_msg1.name)
|
||||
self.assertEqual(msg1, nested_msg1.containing_type)
|
||||
|
||||
nested_enum1 = msg1.enum_types[0]
|
||||
self.assertEquals('NestedFactory1Enum', nested_enum1.name)
|
||||
self.assertEquals(msg1, nested_enum1.containing_type)
|
||||
self.assertEqual('NestedFactory1Enum', nested_enum1.name)
|
||||
self.assertEqual(msg1, nested_enum1.containing_type)
|
||||
|
||||
self.assertEquals(nested_msg1, msg1.fields_by_name[
|
||||
self.assertEqual(nested_msg1, msg1.fields_by_name[
|
||||
'nested_factory_1_message'].message_type)
|
||||
self.assertEquals(nested_enum1, msg1.fields_by_name[
|
||||
self.assertEqual(nested_enum1, msg1.fields_by_name[
|
||||
'nested_factory_1_enum'].enum_type)
|
||||
|
||||
msg2 = self.pool.FindMessageTypeByName(
|
||||
'google.protobuf.python.internal.Factory2Message')
|
||||
self.assertIsInstance(msg2, descriptor.Descriptor)
|
||||
self.assertEquals('Factory2Message', msg2.name)
|
||||
self.assertEquals('google.protobuf.python.internal.Factory2Message',
|
||||
self.assertEqual('Factory2Message', msg2.name)
|
||||
self.assertEqual('google.protobuf.python.internal.Factory2Message',
|
||||
msg2.full_name)
|
||||
self.assertIsNone(msg2.containing_type)
|
||||
|
||||
nested_msg2 = msg2.nested_types[0]
|
||||
self.assertEquals('NestedFactory2Message', nested_msg2.name)
|
||||
self.assertEquals(msg2, nested_msg2.containing_type)
|
||||
self.assertEqual('NestedFactory2Message', nested_msg2.name)
|
||||
self.assertEqual(msg2, nested_msg2.containing_type)
|
||||
|
||||
nested_enum2 = msg2.enum_types[0]
|
||||
self.assertEquals('NestedFactory2Enum', nested_enum2.name)
|
||||
self.assertEquals(msg2, nested_enum2.containing_type)
|
||||
self.assertEqual('NestedFactory2Enum', nested_enum2.name)
|
||||
self.assertEqual(msg2, nested_enum2.containing_type)
|
||||
|
||||
self.assertEquals(nested_msg2, msg2.fields_by_name[
|
||||
self.assertEqual(nested_msg2, msg2.fields_by_name[
|
||||
'nested_factory_2_message'].message_type)
|
||||
self.assertEquals(nested_enum2, msg2.fields_by_name[
|
||||
self.assertEqual(nested_enum2, msg2.fields_by_name[
|
||||
'nested_factory_2_enum'].enum_type)
|
||||
|
||||
self.assertTrue(msg2.fields_by_name['int_with_default'].has_default_value)
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
1776, msg2.fields_by_name['int_with_default'].default_value)
|
||||
|
||||
self.assertTrue(
|
||||
msg2.fields_by_name['double_with_default'].has_default_value)
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
9.99, msg2.fields_by_name['double_with_default'].default_value)
|
||||
|
||||
self.assertTrue(
|
||||
msg2.fields_by_name['string_with_default'].has_default_value)
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
'hello world', msg2.fields_by_name['string_with_default'].default_value)
|
||||
|
||||
self.assertTrue(msg2.fields_by_name['bool_with_default'].has_default_value)
|
||||
self.assertFalse(msg2.fields_by_name['bool_with_default'].default_value)
|
||||
|
||||
self.assertTrue(msg2.fields_by_name['enum_with_default'].has_default_value)
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
1, msg2.fields_by_name['enum_with_default'].default_value)
|
||||
|
||||
msg3 = self.pool.FindMessageTypeByName(
|
||||
'google.protobuf.python.internal.Factory2Message.NestedFactory2Message')
|
||||
self.assertEquals(nested_msg2, msg3)
|
||||
self.assertEqual(nested_msg2, msg3)
|
||||
|
||||
self.assertTrue(msg2.fields_by_name['bytes_with_default'].has_default_value)
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
b'a\xfb\x00c',
|
||||
msg2.fields_by_name['bytes_with_default'].default_value)
|
||||
|
||||
|
@ -190,29 +190,29 @@ class DescriptorPoolTest(unittest.TestCase):
|
|||
enum1 = self.pool.FindEnumTypeByName(
|
||||
'google.protobuf.python.internal.Factory1Enum')
|
||||
self.assertIsInstance(enum1, descriptor.EnumDescriptor)
|
||||
self.assertEquals(0, enum1.values_by_name['FACTORY_1_VALUE_0'].number)
|
||||
self.assertEquals(1, enum1.values_by_name['FACTORY_1_VALUE_1'].number)
|
||||
self.assertEqual(0, enum1.values_by_name['FACTORY_1_VALUE_0'].number)
|
||||
self.assertEqual(1, enum1.values_by_name['FACTORY_1_VALUE_1'].number)
|
||||
|
||||
nested_enum1 = self.pool.FindEnumTypeByName(
|
||||
'google.protobuf.python.internal.Factory1Message.NestedFactory1Enum')
|
||||
self.assertIsInstance(nested_enum1, descriptor.EnumDescriptor)
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
0, nested_enum1.values_by_name['NESTED_FACTORY_1_VALUE_0'].number)
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
1, nested_enum1.values_by_name['NESTED_FACTORY_1_VALUE_1'].number)
|
||||
|
||||
enum2 = self.pool.FindEnumTypeByName(
|
||||
'google.protobuf.python.internal.Factory2Enum')
|
||||
self.assertIsInstance(enum2, descriptor.EnumDescriptor)
|
||||
self.assertEquals(0, enum2.values_by_name['FACTORY_2_VALUE_0'].number)
|
||||
self.assertEquals(1, enum2.values_by_name['FACTORY_2_VALUE_1'].number)
|
||||
self.assertEqual(0, enum2.values_by_name['FACTORY_2_VALUE_0'].number)
|
||||
self.assertEqual(1, enum2.values_by_name['FACTORY_2_VALUE_1'].number)
|
||||
|
||||
nested_enum2 = self.pool.FindEnumTypeByName(
|
||||
'google.protobuf.python.internal.Factory2Message.NestedFactory2Enum')
|
||||
self.assertIsInstance(nested_enum2, descriptor.EnumDescriptor)
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
0, nested_enum2.values_by_name['NESTED_FACTORY_2_VALUE_0'].number)
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
1, nested_enum2.values_by_name['NESTED_FACTORY_2_VALUE_1'].number)
|
||||
|
||||
def testFindEnumTypeByNameFailure(self):
|
||||
|
@ -274,8 +274,8 @@ class ProtoFile(object):
|
|||
|
||||
def CheckFile(self, test, pool):
|
||||
file_desc = pool.FindFileByName(self.name)
|
||||
test.assertEquals(self.name, file_desc.name)
|
||||
test.assertEquals(self.package, file_desc.package)
|
||||
test.assertEqual(self.name, file_desc.name)
|
||||
test.assertEqual(self.package, file_desc.package)
|
||||
dependencies_names = [f.name for f in file_desc.dependencies]
|
||||
test.assertEqual(self.dependencies, dependencies_names)
|
||||
for name, msg_type in self.messages.items():
|
||||
|
@ -430,7 +430,7 @@ class AddDescriptorTest(unittest.TestCase):
|
|||
def _TestMessage(self, prefix):
|
||||
pool = descriptor_pool.DescriptorPool()
|
||||
pool.AddDescriptor(unittest_pb2.TestAllTypes.DESCRIPTOR)
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
'protobuf_unittest.TestAllTypes',
|
||||
pool.FindMessageTypeByName(
|
||||
prefix + 'protobuf_unittest.TestAllTypes').full_name)
|
||||
|
@ -441,18 +441,18 @@ class AddDescriptorTest(unittest.TestCase):
|
|||
prefix + 'protobuf_unittest.TestAllTypes.NestedMessage')
|
||||
|
||||
pool.AddDescriptor(unittest_pb2.TestAllTypes.NestedMessage.DESCRIPTOR)
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
'protobuf_unittest.TestAllTypes.NestedMessage',
|
||||
pool.FindMessageTypeByName(
|
||||
prefix + 'protobuf_unittest.TestAllTypes.NestedMessage').full_name)
|
||||
|
||||
# Files are implicitly also indexed when messages are added.
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
'google/protobuf/unittest.proto',
|
||||
pool.FindFileByName(
|
||||
'google/protobuf/unittest.proto').name)
|
||||
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
'google/protobuf/unittest.proto',
|
||||
pool.FindFileContainingSymbol(
|
||||
prefix + 'protobuf_unittest.TestAllTypes.NestedMessage').name)
|
||||
|
@ -464,7 +464,7 @@ class AddDescriptorTest(unittest.TestCase):
|
|||
def _TestEnum(self, prefix):
|
||||
pool = descriptor_pool.DescriptorPool()
|
||||
pool.AddEnumDescriptor(unittest_pb2.ForeignEnum.DESCRIPTOR)
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
'protobuf_unittest.ForeignEnum',
|
||||
pool.FindEnumTypeByName(
|
||||
prefix + 'protobuf_unittest.ForeignEnum').full_name)
|
||||
|
@ -475,18 +475,18 @@ class AddDescriptorTest(unittest.TestCase):
|
|||
prefix + 'protobuf_unittest.ForeignEnum.NestedEnum')
|
||||
|
||||
pool.AddEnumDescriptor(unittest_pb2.TestAllTypes.NestedEnum.DESCRIPTOR)
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
'protobuf_unittest.TestAllTypes.NestedEnum',
|
||||
pool.FindEnumTypeByName(
|
||||
prefix + 'protobuf_unittest.TestAllTypes.NestedEnum').full_name)
|
||||
|
||||
# Files are implicitly also indexed when enums are added.
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
'google/protobuf/unittest.proto',
|
||||
pool.FindFileByName(
|
||||
'google/protobuf/unittest.proto').name)
|
||||
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
'google/protobuf/unittest.proto',
|
||||
pool.FindFileContainingSymbol(
|
||||
prefix + 'protobuf_unittest.TestAllTypes.NestedEnum').name)
|
||||
|
@ -498,7 +498,7 @@ class AddDescriptorTest(unittest.TestCase):
|
|||
def testFile(self):
|
||||
pool = descriptor_pool.DescriptorPool()
|
||||
pool.AddFileDescriptor(unittest_pb2.DESCRIPTOR)
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
'google/protobuf/unittest.proto',
|
||||
pool.FindFileByName(
|
||||
'google/protobuf/unittest.proto').name)
|
||||
|
|
|
@ -673,7 +673,7 @@ class MakeDescriptorTest(unittest.TestCase):
|
|||
reformed_descriptor = descriptor.MakeDescriptor(descriptor_proto)
|
||||
|
||||
options = reformed_descriptor.GetOptions()
|
||||
self.assertEquals(101,
|
||||
self.assertEqual(101,
|
||||
options.Extensions[unittest_custom_options_pb2.msgopt].i)
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
|
|
@ -154,7 +154,7 @@ class GeneratorTest(unittest.TestCase):
|
|||
# extension and for its value to be set to -789.
|
||||
|
||||
def testNestedTypes(self):
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
set(unittest_pb2.TestAllTypes.DESCRIPTOR.nested_types),
|
||||
set([
|
||||
unittest_pb2.TestAllTypes.NestedMessage.DESCRIPTOR,
|
||||
|
@ -306,36 +306,36 @@ class SymbolDatabaseRegistrationTest(unittest.TestCase):
|
|||
"""Checks that messages, enums and files are correctly registered."""
|
||||
|
||||
def testGetSymbol(self):
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
unittest_pb2.TestAllTypes, symbol_database.Default().GetSymbol(
|
||||
'protobuf_unittest.TestAllTypes'))
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
unittest_pb2.TestAllTypes.NestedMessage,
|
||||
symbol_database.Default().GetSymbol(
|
||||
'protobuf_unittest.TestAllTypes.NestedMessage'))
|
||||
with self.assertRaises(KeyError):
|
||||
symbol_database.Default().GetSymbol('protobuf_unittest.NestedMessage')
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
unittest_pb2.TestAllTypes.OptionalGroup,
|
||||
symbol_database.Default().GetSymbol(
|
||||
'protobuf_unittest.TestAllTypes.OptionalGroup'))
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
unittest_pb2.TestAllTypes.RepeatedGroup,
|
||||
symbol_database.Default().GetSymbol(
|
||||
'protobuf_unittest.TestAllTypes.RepeatedGroup'))
|
||||
|
||||
def testEnums(self):
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
'protobuf_unittest.ForeignEnum',
|
||||
symbol_database.Default().pool.FindEnumTypeByName(
|
||||
'protobuf_unittest.ForeignEnum').full_name)
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
'protobuf_unittest.TestAllTypes.NestedEnum',
|
||||
symbol_database.Default().pool.FindEnumTypeByName(
|
||||
'protobuf_unittest.TestAllTypes.NestedEnum').full_name)
|
||||
|
||||
def testFindFileByName(self):
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
'google/protobuf/unittest.proto',
|
||||
symbol_database.Default().pool.FindFileByName(
|
||||
'google/protobuf/unittest.proto').name)
|
||||
|
|
|
@ -82,9 +82,9 @@ class MessageFactoryTest(unittest.TestCase):
|
|||
serialized = msg.SerializeToString()
|
||||
converted = factory_test2_pb2.Factory2Message.FromString(serialized)
|
||||
reserialized = converted.SerializeToString()
|
||||
self.assertEquals(serialized, reserialized)
|
||||
self.assertEqual(serialized, reserialized)
|
||||
result = cls.FromString(reserialized)
|
||||
self.assertEquals(msg, result)
|
||||
self.assertEqual(msg, result)
|
||||
|
||||
def testGetPrototype(self):
|
||||
db = descriptor_database.DescriptorDatabase()
|
||||
|
@ -125,8 +125,8 @@ class MessageFactoryTest(unittest.TestCase):
|
|||
'google.protobuf.python.internal.another_field']
|
||||
msg1.Extensions[ext1] = 'test1'
|
||||
msg1.Extensions[ext2] = 'test2'
|
||||
self.assertEquals('test1', msg1.Extensions[ext1])
|
||||
self.assertEquals('test2', msg1.Extensions[ext2])
|
||||
self.assertEqual('test1', msg1.Extensions[ext1])
|
||||
self.assertEqual('test2', msg1.Extensions[ext2])
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
|
|
@ -98,7 +98,7 @@ class MessageTest(unittest.TestCase):
|
|||
golden_message.ParseFromString(golden_data)
|
||||
all_set = unittest_pb2.TestAllExtensions()
|
||||
test_util.SetAllExtensions(all_set)
|
||||
self.assertEquals(all_set, golden_message)
|
||||
self.assertEqual(all_set, golden_message)
|
||||
self.assertEqual(golden_data, golden_message.SerializeToString())
|
||||
golden_copy = copy.deepcopy(golden_message)
|
||||
self.assertEqual(golden_data, golden_copy.SerializeToString())
|
||||
|
@ -109,7 +109,7 @@ class MessageTest(unittest.TestCase):
|
|||
golden_message.ParseFromString(golden_data)
|
||||
all_set = unittest_pb2.TestPackedTypes()
|
||||
test_util.SetAllPackedFields(all_set)
|
||||
self.assertEquals(all_set, golden_message)
|
||||
self.assertEqual(all_set, golden_message)
|
||||
self.assertEqual(golden_data, all_set.SerializeToString())
|
||||
golden_copy = copy.deepcopy(golden_message)
|
||||
self.assertEqual(golden_data, golden_copy.SerializeToString())
|
||||
|
@ -120,7 +120,7 @@ class MessageTest(unittest.TestCase):
|
|||
golden_message.ParseFromString(golden_data)
|
||||
all_set = unittest_pb2.TestPackedExtensions()
|
||||
test_util.SetAllPackedExtensions(all_set)
|
||||
self.assertEquals(all_set, golden_message)
|
||||
self.assertEqual(all_set, golden_message)
|
||||
self.assertEqual(golden_data, all_set.SerializeToString())
|
||||
golden_copy = copy.deepcopy(golden_message)
|
||||
self.assertEqual(golden_data, golden_copy.SerializeToString())
|
||||
|
@ -132,7 +132,7 @@ class MessageTest(unittest.TestCase):
|
|||
pickled_message = pickle.dumps(golden_message)
|
||||
|
||||
unpickled_message = pickle.loads(pickled_message)
|
||||
self.assertEquals(unpickled_message, golden_message)
|
||||
self.assertEqual(unpickled_message, golden_message)
|
||||
|
||||
|
||||
def testPickleIncompleteProto(self):
|
||||
|
@ -140,8 +140,8 @@ class MessageTest(unittest.TestCase):
|
|||
pickled_message = pickle.dumps(golden_message)
|
||||
|
||||
unpickled_message = pickle.loads(pickled_message)
|
||||
self.assertEquals(unpickled_message, golden_message)
|
||||
self.assertEquals(unpickled_message.a, 1)
|
||||
self.assertEqual(unpickled_message, golden_message)
|
||||
self.assertEqual(unpickled_message.a, 1)
|
||||
# This is still an incomplete proto - so serializing should fail
|
||||
self.assertRaises(message.EncodeError, unpickled_message.SerializeToString)
|
||||
|
||||
|
@ -344,11 +344,11 @@ class MessageTest(unittest.TestCase):
|
|||
msg.repeated_nested_message.add(bb=3)
|
||||
msg.repeated_nested_message.add(bb=4)
|
||||
|
||||
self.assertEquals([1, 2, 3, 4],
|
||||
self.assertEqual([1, 2, 3, 4],
|
||||
[m.bb for m in msg.repeated_nested_message])
|
||||
self.assertEquals([4, 3, 2, 1],
|
||||
self.assertEqual([4, 3, 2, 1],
|
||||
[m.bb for m in reversed(msg.repeated_nested_message)])
|
||||
self.assertEquals([4, 3, 2, 1],
|
||||
self.assertEqual([4, 3, 2, 1],
|
||||
[m.bb for m in msg.repeated_nested_message[::-1]])
|
||||
|
||||
def testSortingRepeatedScalarFieldsDefaultComparator(self):
|
||||
|
|
|
@ -141,10 +141,10 @@ class ReflectionTest(unittest.TestCase):
|
|||
repeated_bool=[True, False, False],
|
||||
repeated_string=["optional_string"])
|
||||
|
||||
self.assertEquals([1, 2, 3, 4], list(proto.repeated_int32))
|
||||
self.assertEquals([1.23, 54.321], list(proto.repeated_double))
|
||||
self.assertEquals([True, False, False], list(proto.repeated_bool))
|
||||
self.assertEquals(["optional_string"], list(proto.repeated_string))
|
||||
self.assertEqual([1, 2, 3, 4], list(proto.repeated_int32))
|
||||
self.assertEqual([1.23, 54.321], list(proto.repeated_double))
|
||||
self.assertEqual([True, False, False], list(proto.repeated_bool))
|
||||
self.assertEqual(["optional_string"], list(proto.repeated_string))
|
||||
|
||||
def testRepeatedCompositeConstructor(self):
|
||||
# Constructor with only repeated composite types should succeed.
|
||||
|
@ -163,18 +163,18 @@ class ReflectionTest(unittest.TestCase):
|
|||
unittest_pb2.TestAllTypes.RepeatedGroup(a=1),
|
||||
unittest_pb2.TestAllTypes.RepeatedGroup(a=2)])
|
||||
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
[unittest_pb2.TestAllTypes.NestedMessage(
|
||||
bb=unittest_pb2.TestAllTypes.FOO),
|
||||
unittest_pb2.TestAllTypes.NestedMessage(
|
||||
bb=unittest_pb2.TestAllTypes.BAR)],
|
||||
list(proto.repeated_nested_message))
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
[unittest_pb2.ForeignMessage(c=-43),
|
||||
unittest_pb2.ForeignMessage(c=45324),
|
||||
unittest_pb2.ForeignMessage(c=12)],
|
||||
list(proto.repeated_foreign_message))
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
[unittest_pb2.TestAllTypes.RepeatedGroup(),
|
||||
unittest_pb2.TestAllTypes.RepeatedGroup(a=1),
|
||||
unittest_pb2.TestAllTypes.RepeatedGroup(a=2)],
|
||||
|
@ -199,15 +199,15 @@ class ReflectionTest(unittest.TestCase):
|
|||
|
||||
self.assertEqual(24, proto.optional_int32)
|
||||
self.assertEqual('optional_string', proto.optional_string)
|
||||
self.assertEquals([1.23, 54.321], list(proto.repeated_double))
|
||||
self.assertEquals([True, False, False], list(proto.repeated_bool))
|
||||
self.assertEquals(
|
||||
self.assertEqual([1.23, 54.321], list(proto.repeated_double))
|
||||
self.assertEqual([True, False, False], list(proto.repeated_bool))
|
||||
self.assertEqual(
|
||||
[unittest_pb2.TestAllTypes.NestedMessage(
|
||||
bb=unittest_pb2.TestAllTypes.FOO),
|
||||
unittest_pb2.TestAllTypes.NestedMessage(
|
||||
bb=unittest_pb2.TestAllTypes.BAR)],
|
||||
list(proto.repeated_nested_message))
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
[unittest_pb2.ForeignMessage(c=-43),
|
||||
unittest_pb2.ForeignMessage(c=45324),
|
||||
unittest_pb2.ForeignMessage(c=12)],
|
||||
|
@ -235,18 +235,18 @@ class ReflectionTest(unittest.TestCase):
|
|||
|
||||
def testConstructorInvalidatesCachedByteSize(self):
|
||||
message = unittest_pb2.TestAllTypes(optional_int32 = 12)
|
||||
self.assertEquals(2, message.ByteSize())
|
||||
self.assertEqual(2, message.ByteSize())
|
||||
|
||||
message = unittest_pb2.TestAllTypes(
|
||||
optional_nested_message = unittest_pb2.TestAllTypes.NestedMessage())
|
||||
self.assertEquals(3, message.ByteSize())
|
||||
self.assertEqual(3, message.ByteSize())
|
||||
|
||||
message = unittest_pb2.TestAllTypes(repeated_int32 = [12])
|
||||
self.assertEquals(3, message.ByteSize())
|
||||
self.assertEqual(3, message.ByteSize())
|
||||
|
||||
message = unittest_pb2.TestAllTypes(
|
||||
repeated_nested_message = [unittest_pb2.TestAllTypes.NestedMessage()])
|
||||
self.assertEquals(3, message.ByteSize())
|
||||
self.assertEqual(3, message.ByteSize())
|
||||
|
||||
def testSimpleHasBits(self):
|
||||
# Test a scalar.
|
||||
|
@ -1188,7 +1188,7 @@ class ReflectionTest(unittest.TestCase):
|
|||
self.assertTrue(1 in unittest_pb2.TestAllExtensions._extensions_by_number)
|
||||
# Make sure extensions haven't been registered into types that shouldn't
|
||||
# have any.
|
||||
self.assertEquals(0, len(unittest_pb2.TestAllTypes._extensions_by_name))
|
||||
self.assertEqual(0, len(unittest_pb2.TestAllTypes._extensions_by_name))
|
||||
|
||||
# If message A directly contains message B, and
|
||||
# a.HasField('b') is currently False, then mutating any
|
||||
|
@ -1512,18 +1512,18 @@ class ReflectionTest(unittest.TestCase):
|
|||
test_util.SetAllNonLazyFields(proto)
|
||||
# Clear the message.
|
||||
proto.Clear()
|
||||
self.assertEquals(proto.ByteSize(), 0)
|
||||
self.assertEqual(proto.ByteSize(), 0)
|
||||
empty_proto = unittest_pb2.TestAllTypes()
|
||||
self.assertEquals(proto, empty_proto)
|
||||
self.assertEqual(proto, empty_proto)
|
||||
|
||||
# Test if extensions which were set are cleared.
|
||||
proto = unittest_pb2.TestAllExtensions()
|
||||
test_util.SetAllExtensions(proto)
|
||||
# Clear the message.
|
||||
proto.Clear()
|
||||
self.assertEquals(proto.ByteSize(), 0)
|
||||
self.assertEqual(proto.ByteSize(), 0)
|
||||
empty_proto = unittest_pb2.TestAllExtensions()
|
||||
self.assertEquals(proto, empty_proto)
|
||||
self.assertEqual(proto, empty_proto)
|
||||
|
||||
def testDisconnectingBeforeClear(self):
|
||||
proto = unittest_pb2.TestAllTypes()
|
||||
|
@ -2481,7 +2481,7 @@ class SerializationTest(unittest.TestCase):
|
|||
# Check that the message parsed well.
|
||||
extension_message1 = unittest_mset_pb2.TestMessageSetExtension1
|
||||
extension1 = extension_message1.message_set_extension
|
||||
self.assertEquals(12345, proto.Extensions[extension1].i)
|
||||
self.assertEqual(12345, proto.Extensions[extension1].i)
|
||||
|
||||
def testUnknownFields(self):
|
||||
proto = unittest_pb2.TestAllTypes()
|
||||
|
|
|
@ -65,53 +65,53 @@ class SymbolDatabaseTest(unittest.TestCase):
|
|||
messages['protobuf_unittest.TestAllTypes'])
|
||||
|
||||
def testGetSymbol(self):
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
unittest_pb2.TestAllTypes, self._Database().GetSymbol(
|
||||
'protobuf_unittest.TestAllTypes'))
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
unittest_pb2.TestAllTypes.NestedMessage, self._Database().GetSymbol(
|
||||
'protobuf_unittest.TestAllTypes.NestedMessage'))
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
unittest_pb2.TestAllTypes.OptionalGroup, self._Database().GetSymbol(
|
||||
'protobuf_unittest.TestAllTypes.OptionalGroup'))
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
unittest_pb2.TestAllTypes.RepeatedGroup, self._Database().GetSymbol(
|
||||
'protobuf_unittest.TestAllTypes.RepeatedGroup'))
|
||||
|
||||
def testEnums(self):
|
||||
# Check registration of types in the pool.
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
'protobuf_unittest.ForeignEnum',
|
||||
self._Database().pool.FindEnumTypeByName(
|
||||
'protobuf_unittest.ForeignEnum').full_name)
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
'protobuf_unittest.TestAllTypes.NestedEnum',
|
||||
self._Database().pool.FindEnumTypeByName(
|
||||
'protobuf_unittest.TestAllTypes.NestedEnum').full_name)
|
||||
|
||||
def testFindMessageTypeByName(self):
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
'protobuf_unittest.TestAllTypes',
|
||||
self._Database().pool.FindMessageTypeByName(
|
||||
'protobuf_unittest.TestAllTypes').full_name)
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
'protobuf_unittest.TestAllTypes.NestedMessage',
|
||||
self._Database().pool.FindMessageTypeByName(
|
||||
'protobuf_unittest.TestAllTypes.NestedMessage').full_name)
|
||||
|
||||
def testFindFindContainingSymbol(self):
|
||||
# Lookup based on either enum or message.
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
'google/protobuf/unittest.proto',
|
||||
self._Database().pool.FindFileContainingSymbol(
|
||||
'protobuf_unittest.TestAllTypes.NestedEnum').name)
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
'google/protobuf/unittest.proto',
|
||||
self._Database().pool.FindFileContainingSymbol(
|
||||
'protobuf_unittest.TestAllTypes').name)
|
||||
|
||||
def testFindFileByName(self):
|
||||
self.assertEquals(
|
||||
self.assertEqual(
|
||||
'google/protobuf/unittest.proto',
|
||||
self._Database().pool.FindFileByName(
|
||||
'google/protobuf/unittest.proto').name)
|
||||
|
|
|
@ -54,15 +54,15 @@ TEST_VALUES = [
|
|||
class TextEncodingTestCase(unittest.TestCase):
|
||||
def testCEscape(self):
|
||||
for escaped, escaped_utf8, unescaped in TEST_VALUES:
|
||||
self.assertEquals(escaped,
|
||||
self.assertEqual(escaped,
|
||||
text_encoding.CEscape(unescaped, as_utf8=False))
|
||||
self.assertEquals(escaped_utf8,
|
||||
self.assertEqual(escaped_utf8,
|
||||
text_encoding.CEscape(unescaped, as_utf8=True))
|
||||
|
||||
def testCUnescape(self):
|
||||
for escaped, escaped_utf8, unescaped in TEST_VALUES:
|
||||
self.assertEquals(unescaped, text_encoding.CUnescape(escaped))
|
||||
self.assertEquals(unescaped, text_encoding.CUnescape(escaped_utf8))
|
||||
self.assertEqual(unescaped, text_encoding.CUnescape(escaped))
|
||||
self.assertEqual(unescaped, text_encoding.CUnescape(escaped_utf8))
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
|
|
@ -233,7 +233,7 @@ class TextFormatTest(unittest.TestCase):
|
|||
parsed_message = unittest_pb2.TestAllTypes()
|
||||
r = text_format.Parse(wire_text, parsed_message)
|
||||
self.assertIs(r, parsed_message)
|
||||
self.assertEquals(message, parsed_message)
|
||||
self.assertEqual(message, parsed_message)
|
||||
|
||||
# Test as_utf8 = True.
|
||||
wire_text = text_format.MessageToString(
|
||||
|
@ -241,7 +241,7 @@ class TextFormatTest(unittest.TestCase):
|
|||
parsed_message = unittest_pb2.TestAllTypes()
|
||||
r = text_format.Parse(wire_text, parsed_message)
|
||||
self.assertIs(r, parsed_message)
|
||||
self.assertEquals(message, parsed_message,
|
||||
self.assertEqual(message, parsed_message,
|
||||
'\n%s != %s' % (message, parsed_message))
|
||||
|
||||
def testPrintRawUtf8String(self):
|
||||
|
@ -251,7 +251,7 @@ class TextFormatTest(unittest.TestCase):
|
|||
self.CompareToGoldenText(text, 'repeated_string: "\303\274\352\234\237"\n')
|
||||
parsed_message = unittest_pb2.TestAllTypes()
|
||||
text_format.Parse(text, parsed_message)
|
||||
self.assertEquals(message, parsed_message,
|
||||
self.assertEqual(message, parsed_message,
|
||||
'\n%s != %s' % (message, parsed_message))
|
||||
|
||||
def testPrintFloatFormat(self):
|
||||
|
@ -309,7 +309,7 @@ class TextFormatTest(unittest.TestCase):
|
|||
|
||||
message = unittest_pb2.TestAllTypes()
|
||||
test_util.SetAllFields(message)
|
||||
self.assertEquals(message, parsed_message)
|
||||
self.assertEqual(message, parsed_message)
|
||||
|
||||
def testParseGoldenExtensions(self):
|
||||
golden_text = '\n'.join(self.ReadGolden(
|
||||
|
@ -319,7 +319,7 @@ class TextFormatTest(unittest.TestCase):
|
|||
|
||||
message = unittest_pb2.TestAllExtensions()
|
||||
test_util.SetAllExtensions(message)
|
||||
self.assertEquals(message, parsed_message)
|
||||
self.assertEqual(message, parsed_message)
|
||||
|
||||
def testParseAllFields(self):
|
||||
message = unittest_pb2.TestAllTypes()
|
||||
|
@ -360,8 +360,8 @@ class TextFormatTest(unittest.TestCase):
|
|||
text_format.Parse(text, message)
|
||||
ext1 = unittest_mset_pb2.TestMessageSetExtension1.message_set_extension
|
||||
ext2 = unittest_mset_pb2.TestMessageSetExtension2.message_set_extension
|
||||
self.assertEquals(23, message.message_set.Extensions[ext1].i)
|
||||
self.assertEquals('foo', message.message_set.Extensions[ext2].str)
|
||||
self.assertEqual(23, message.message_set.Extensions[ext1].i)
|
||||
self.assertEqual('foo', message.message_set.Extensions[ext2].str)
|
||||
|
||||
def testParseExotic(self):
|
||||
message = unittest_pb2.TestAllTypes()
|
||||
|
@ -408,7 +408,7 @@ class TextFormatTest(unittest.TestCase):
|
|||
message = unittest_pb2.TestAllTypes()
|
||||
text = ''
|
||||
text_format.Parse(text, message)
|
||||
self.assertEquals(unittest_pb2.TestAllTypes(), message)
|
||||
self.assertEqual(unittest_pb2.TestAllTypes(), message)
|
||||
|
||||
def testParseInvalidUtf8(self):
|
||||
message = unittest_pb2.TestAllTypes()
|
||||
|
@ -581,7 +581,7 @@ class TextFormatTest(unittest.TestCase):
|
|||
|
||||
message = unittest_pb2.TestAllTypes()
|
||||
test_util.SetAllFields(message)
|
||||
self.assertEquals(message, parsed_message)
|
||||
self.assertEqual(message, parsed_message)
|
||||
|
||||
def testMergeLinesGolden(self):
|
||||
opened = self.ReadGolden('text_format_unittest_data.txt')
|
||||
|
|
Loading…
Add table
Reference in a new issue