def _to_proto(self):
pb = dataset_options_pb2.MapVectorization()
if self.enabled is not None:
pb.enabled = self.enabled
if self.use_choose_fastest is not None:
pb.use_choose_fastest = self.use_choose_fastest
return pb
def testProtoOptionsDefaultValuesRoundTrip(self):
pb = dataset_options_pb2.Options()
options = dataset_ops.Options()
options._from_proto(pb)
result = options._to_proto()
expected_pb = dataset_options_pb2.Options()
expected_pb.distribute_options.CopyFrom(
dataset_options_pb2.DistributeOptions())
expected_pb.optimization_options.CopyFrom(
dataset_options_pb2.OptimizationOptions())
expected_pb.optimization_options.map_vectorization.CopyFrom(
dataset_options_pb2.MapVectorization())
expected_pb.threading_options.CopyFrom(
dataset_options_pb2.ThreadingOptions())
self.assertProtoEquals(expected_pb, result)
