Commit a2abc947 authored by Tres Seaver's avatar Tres Seaver

Straddle Python 2.6, 2.7, 3.3, 3.4.

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