def shaped_batch_test_case():
test_case = test_example_pb2.TestCase()
value = test_case.values.add()
value.double_value.append(23.5)
value.bool_value.append(True)
value = test_case.values.add()
value.double_value.append(44.0)
value.bool_value.append(False)
value = test_case.values.add()
value.double_value.append(3.14159)
value.bool_value.append(True)
value = test_case.values.add()
value.double_value.append(1.414)
value.bool_value.append(True)
value = test_case.values.add()
value.double_value.append(-32.2)
value.bool_value.append(False)
value = test_case.values.add()
value.double_value.append(0.0001)
value.bool_value.append(True)
test_case.shapes.append(3)
test_case.shapes.append(2)
for _ in range(12):
test_case.sizes.append(1)
field = test_case.fields.add()
field.name = "double_value"
field.dtype = types_pb2.DT_DOUBLE
field.value.double_value.append(23.5)
field.value.double_value.append(44.0)
field.value.double_value.append(3.14159)
field.value.double_value.append(1.414)
field.value.double_value.append(-32.2)
field.value.double_value.append(0.0001)
field = test_case.fields.add()
field.name = "bool_value"
field.dtype = types_pb2.DT_BOOL
field.value.bool_value.append(True)
field.value.bool_value.append(False)
field.value.bool_value.append(True)
field.value.bool_value.append(True)
field.value.bool_value.append(False)
field.value.bool_value.append(True)
return test_case
def promote_unsigned_test_case():
test_case = test_example_pb2.TestCase()
primitive = test_case.primitive.add()
primitive.fixed32_value.append(4294967295)
primitive.uint32_value.append(4294967295)
test_case.shape.append(1)
test_case.sizes.append(1)
field = test_case.field.add()
field.name = "fixed32_value"
field.dtype = types_pb2.DT_INT64
field.expected.int64_value.append(4294967295)
test_case.sizes.append(1)
field = test_case.field.add()
field.name = "uint32_value"
field.dtype = types_pb2.DT_INT64
field.expected.int64_value.append(4294967295)
# Comes from an explicitly-specified default
test_case.sizes.append(0)
field = test_case.field.add()
field.name = "uint32_default"
field.dtype = types_pb2.DT_INT64
field.expected.int64_value.append(4294967295)
return test_case
def defaults_test_case():
test_case = test_example_pb2.TestCase()
test_case.primitive.add(
) # No fields specified, so we get all defaults.
test_case.shape.append(1)
test_case.sizes.append(0)
field = test_case.field.add()
field.name = "double_default"
field.dtype = types_pb2.DT_DOUBLE
field.expected.double_value.append(1.0)
test_case.sizes.append(0)
field = test_case.field.add()
field.name = "float_default"
field.dtype = types_pb2.DT_FLOAT
field.expected.float_value.append(2.0)
test_case.sizes.append(0)
field = test_case.field.add()
field.name = "int64_default"
field.dtype = types_pb2.DT_INT64
field.expected.int64_value.append(3)
test_case.sizes.append(0)
field = test_case.field.add()
field.name = "sfixed64_default"
field.dtype = types_pb2.DT_INT64
field.expected.int64_value.append(11)
test_case.sizes.append(0)
field = test_case.field.add()
field.name = "sint64_default"
field.dtype = types_pb2.DT_INT64
field.expected.int64_value.append(13)
test_case.sizes.append(0)
field = test_case.field.add()
field.name = "uint64_default"
field.dtype = types_pb2.DT_INT64
field.expected.int64_value.append(4)
test_case.sizes.append(0)
field = test_case.field.add()
field.name = "fixed64_default"
field.dtype = types_pb2.DT_INT64
field.expected.int64_value.append(6)
test_case.sizes.append(0)
field = test_case.field.add()
field.name = "int32_default"
field.dtype = types_pb2.DT_INT32
field.expected.int32_value.append(5)
test_case.sizes.append(0)
field = test_case.field.add()
field.name = "sfixed32_default"
field.dtype = types_pb2.DT_INT32
field.expected.int32_value.append(10)
test_case.sizes.append(0)
field = test_case.field.add()
field.name = "sint32_default"
field.dtype = types_pb2.DT_INT32
field.expected.int32_value.append(12)
test_case.sizes.append(0)
field = test_case.field.add()
field.name = "uint32_default"
field.dtype = types_pb2.DT_INT32
field.expected.int32_value.append(-1)
test_case.sizes.append(0)
field = test_case.field.add()
field.name = "fixed32_default"
field.dtype = types_pb2.DT_INT32
field.expected.int32_value.append(7)
test_case.sizes.append(0)
field = test_case.field.add()
field.name = "bool_default"
field.dtype = types_pb2.DT_BOOL
field.expected.bool_value.append(True)
test_case.sizes.append(0)
field = test_case.field.add()
field.name = "string_default"
field.dtype = types_pb2.DT_STRING
field.expected.string_value.append("a")
test_case.sizes.append(0)
field = test_case.field.add()
field.name = "bytes_default"
field.dtype = types_pb2.DT_STRING
field.expected.string_value.append("a longer default string")
return test_case
def minmax_test_case():
test_case = test_example_pb2.TestCase()
primitive = test_case.primitive.add()
primitive.double_value.append(-1.7976931348623158e+308)
primitive.double_value.append(2.2250738585072014e-308)
primitive.double_value.append(1.7976931348623158e+308)
primitive.float_value.append(-3.402823466e+38)
primitive.float_value.append(1.175494351e-38)
primitive.float_value.append(3.402823466e+38)
primitive.int64_value.append(-9223372036854775808)
primitive.int64_value.append(9223372036854775807)
primitive.sfixed64_value.append(-9223372036854775808)
primitive.sfixed64_value.append(9223372036854775807)
primitive.sint64_value.append(-9223372036854775808)
primitive.sint64_value.append(9223372036854775807)
primitive.uint64_value.append(0)
primitive.uint64_value.append(18446744073709551615)
primitive.fixed64_value.append(0)
primitive.fixed64_value.append(18446744073709551615)
primitive.int32_value.append(-2147483648)
primitive.int32_value.append(2147483647)
primitive.sfixed32_value.append(-2147483648)
primitive.sfixed32_value.append(2147483647)
primitive.sint32_value.append(-2147483648)
primitive.sint32_value.append(2147483647)
primitive.uint32_value.append(0)
primitive.uint32_value.append(4294967295)
primitive.fixed32_value.append(0)
primitive.fixed32_value.append(4294967295)
primitive.bool_value.append(False)
primitive.bool_value.append(True)
primitive.string_value.append("")
primitive.string_value.append("I refer to the infinite.")
test_case.shape.append(1)
test_case.sizes.append(3)
field = test_case.field.add()
field.name = "double_value"
field.dtype = types_pb2.DT_DOUBLE
field.expected.double_value.append(-1.7976931348623158e+308)
field.expected.double_value.append(2.2250738585072014e-308)
field.expected.double_value.append(1.7976931348623158e+308)
test_case.sizes.append(3)
field = test_case.field.add()
field.name = "float_value"
field.dtype = types_pb2.DT_FLOAT
field.expected.float_value.append(-3.402823466e+38)
field.expected.float_value.append(1.175494351e-38)
field.expected.float_value.append(3.402823466e+38)
test_case.sizes.append(2)
field = test_case.field.add()
field.name = "int64_value"
field.dtype = types_pb2.DT_INT64
field.expected.int64_value.append(-9223372036854775808)
field.expected.int64_value.append(9223372036854775807)
test_case.sizes.append(2)
field = test_case.field.add()
field.name = "sfixed64_value"
field.dtype = types_pb2.DT_INT64
field.expected.int64_value.append(-9223372036854775808)
field.expected.int64_value.append(9223372036854775807)
test_case.sizes.append(2)
field = test_case.field.add()
field.name = "sint64_value"
field.dtype = types_pb2.DT_INT64
field.expected.int64_value.append(-9223372036854775808)
field.expected.int64_value.append(9223372036854775807)
test_case.sizes.append(2)
field = test_case.field.add()
field.name = "uint64_value"
field.dtype = types_pb2.DT_INT64
field.expected.int64_value.append(0)
field.expected.int64_value.append(-1)
test_case.sizes.append(2)
field = test_case.field.add()
field.name = "fixed64_value"
field.dtype = types_pb2.DT_INT64
field.expected.int64_value.append(0)
field.expected.int64_value.append(-1)
test_case.sizes.append(2)
field = test_case.field.add()
field.name = "int32_value"
field.dtype = types_pb2.DT_INT32
field.expected.int32_value.append(-2147483648)
field.expected.int32_value.append(2147483647)
test_case.sizes.append(2)
field = test_case.field.add()
field.name = "sfixed32_value"
field.dtype = types_pb2.DT_INT32
field.expected.int32_value.append(-2147483648)
field.expected.int32_value.append(2147483647)
test_case.sizes.append(2)
field = test_case.field.add()
field.name = "sint32_value"
field.dtype = types_pb2.DT_INT32
field.expected.int32_value.append(-2147483648)
field.expected.int32_value.append(2147483647)
test_case.sizes.append(2)
field = test_case.field.add()
field.name = "uint32_value"
field.dtype = types_pb2.DT_INT32
field.expected.int32_value.append(0)
field.expected.int32_value.append(-1)
test_case.sizes.append(2)
field = test_case.field.add()
field.name = "fixed32_value"
field.dtype = types_pb2.DT_INT32
field.expected.int32_value.append(0)
field.expected.int32_value.append(-1)
test_case.sizes.append(2)
field = test_case.field.add()
field.name = "bool_value"
field.dtype = types_pb2.DT_BOOL
field.expected.bool_value.append(False)
field.expected.bool_value.append(True)
test_case.sizes.append(2)
field = test_case.field.add()
field.name = "string_value"
field.dtype = types_pb2.DT_STRING
field.expected.string_value.append("")
field.expected.string_value.append("I refer to the infinite.")
return test_case
def testSanitizerGood(self):
with open(FLAGS.message_text_file, 'r') as fp:
case = text_format.Parse(fp.read(), test_example_pb2.TestCase())
def testBinaryDisordered(self):
with open(FLAGS.message_text_file, 'r') as fp:
case = text_format.Parse(fp.read(), test_example_pb2.TestCase())
def testRoundtripPacked(self):
with open(FLAGS.message_text_file, 'r') as fp:
case = text_format.Parse(fp.read(), test_example_pb2.TestCase())
