def testProto2ContainsProto(self):
pb = compare_test_pb2.Large()
pb.string_ = 'abc'
pb.float_ = 1.234
pb.small.strings.append('xyz')
pb2 = compare_test_pb2.Large()
pb2.small.strings.append('xyz')
compare.assertProto2Contains(self, pb2, pb)
def testNormalizesDoubles(self): pb1 = compare_test_pb2.Large() pb1.double_ = 1.2314352351231 pb2 = compare_test_pb2.Large() pb2.double_ = 1.2314352 self.assertNotEqual(pb1.double_, pb2.double_) compare.NormalizeNumberFields(pb1) compare.NormalizeNumberFields(pb2) self.assertEqual(pb1.double_, pb2.double_)
def testNormalizesFloats(self): pb1 = compare_test_pb2.Large() pb1.float_ = 1.2314352351231 pb2 = compare_test_pb2.Large() pb2.float_ = 1.231435 self.assertNotEqual(pb1.float_, pb2.float_) compare.NormalizeNumberFields(pb1) compare.NormalizeNumberFields(pb2) self.assertEqual(pb1.float_, pb2.float_)
def testAssertSameElements(self): a = compare_test_pb2.Large() a.string_ = 'abc' a.float_ = 1.234 a.int64s[:] = [4, 3, 2] b = compare_test_pb2.Large() b.CopyFrom(a) b.int64s[:] = [2, 4, 3] self.assertProto2SameElements(a, b)
def testNormalizesFloat(self):
pb1 = compare_test_pb2.Large()
pb1.double_ = 4.0
pb2 = compare_test_pb2.Large()
pb2.double_ = 4
compare.assertProto2Equal(self, pb1, pb2, normalize_numbers=True)
pb1 = compare_test_pb2.Medium()
pb1.floats.extend([4.0, 6.0])
pb2 = compare_test_pb2.Medium()
pb2.floats.extend([6, 4])
compare.assertProto2SameElements(self, pb1, pb2, normalize_numbers=True)
def testNormalizesRepeatedInts(self): pb = compare_test_pb2.Large() pb.int64s.extend([1, 400, 999999999999999]) compare.NormalizeNumberFields(pb) self.assertTrue(isinstance(pb.int64s[0], six.integer_types)) self.assertTrue(isinstance(pb.int64s[1], six.integer_types)) self.assertTrue(isinstance(pb.int64s[2], six.integer_types))
def testPrimitives(self):
googletest.TestCase.assertEqual(self, 0, compare.Proto2Cmp('a', 'a'))
googletest.TestCase.assertLess(self, 0, compare.Proto2Cmp('b', 'a'))
pb = compare_test_pb2.Large()
googletest.TestCase.assertEquals(self, cmp('a', pb), compare.Proto2Cmp('a', pb))
googletest.TestCase.assertEqual(self, cmp(pb, 'a'), compare.Proto2Cmp(pb, 'a'))
def testAssertEqualWithStringArg(self):
pb = compare_test_pb2.Large()
pb.string_ = 'abc'
pb.float_ = 1.234
self.assertProtoEqual("""
string_: 'abc'
float_: 1.234
""", pb)
def LargePbs(*args):
"""Converts ASCII string Large PBs to messages."""
pbs = []
for arg in args:
pb = compare_test_pb2.Large()
text_format.Merge(arg, pb)
pbs.append(pb)
return pbs
def testProto2ContainsString(self):
pb = compare_test_pb2.Large()
pb.string_ = 'abc'
pb.float_ = 1.234
pb.small.strings.append('xyz')
compare.assertProto2Contains(
self, """
small {
strings: "xyz"
}
""", pb)
def testNormalizesInts(self):
pb = compare_test_pb2.Large()
pb.int64_ = 4
compare.NormalizeNumberFields(pb)
self.assertTrue(isinstance(pb.int64_, six.integer_types))
pb.int64_ = 4
compare.NormalizeNumberFields(pb)
self.assertTrue(isinstance(pb.int64_, six.integer_types))
pb.int64_ = 9999999999999999
compare.NormalizeNumberFields(pb)
self.assertTrue(isinstance(pb.int64_, six.integer_types))
def testAssertSameElementsWithStringArg(self):
pb = compare_test_pb2.Large()
pb.string_ = 'abc'
pb.float_ = 1.234
pb.int64s.extend([7, 3, 5])
compare.assertProto2SameElements(
self, """
string_: 'abc'
float_: 1.234
int64s: 3
int64s: 7
int64s: 5
""", pb)
def testNormalizesInts(self):
pb = compare_test_pb2.Large()
pb.int64_ = 4
compare.NormalizeNumberFields(pb)
self.assertTrue(isinstance(pb.int64_, long))
pb.int64_ = 4L
compare.NormalizeNumberFields(pb)
self.assertTrue(isinstance(pb.int64_, long))
pb.int64_ = 9999999999999999L
compare.NormalizeNumberFields(pb)
self.assertTrue(isinstance(pb.int64_, long))
def testNormalizesFloat(self): pb1 = compare_test_pb2.Large() pb1.double_ = 4.0 pb2 = compare_test_pb2.Large() pb2.double_ = 4 compare.assertProtoEqual(self, pb1, pb2, normalize_numbers=True)
def testNormalizesNumbers(self): pb1 = compare_test_pb2.Large() pb1.int64_ = 4 pb2 = compare_test_pb2.Large() pb2.int64_ = 4 compare.assertProtoEqual(self, pb1, pb2)
def testNormalizesRepeatedFloats(self):
pb = compare_test_pb2.Large()
pb.medium.floats.extend([0.111111111, 0.111111])
compare.NormalizeNumberFields(pb)
for value in pb.medium.floats:
self.assertAlmostEqual(0.111111, value)
