def test_serialize_feature(self, feature, ascii_protos):
serialized_features_tensor = tf_utils._serialize_feature(feature)
with tf.compat.v1.Session():
serialized_features = serialized_features_tensor.eval()
feature_proto = tf.train.Feature()
self.assertEqual(len(serialized_features), len(ascii_protos))
for ascii_proto, serialized_feature in zip(ascii_protos,
serialized_features):
feature_proto.ParseFromString(serialized_feature)
self.assertProtoEquals(ascii_proto, feature_proto)
def _serialize_feature(feature):
return tf_utils._serialize_feature(feature)
