commit 51a9e54848c034cac33ce148351a824202f8e166
author Tony Mak <tonymak@google.com> Fri Nov 02 13:36:22 2018 +0000
committer Tony Mak <tonymak@google.com> Fri Nov 02 14:39:45 2018 +0000
tree 0a6c6626212dc3ba7a4ec35b35f3a65e87cc67ce
parent f2eef8b7f1ea74cf88761d38c407b1baff680458

Export libtextclassifier to Android (generated by the export script)

Test: Compile and boot

Change-Id: I0433e6fb549ba0b32bc55933b3c11562e61a0b4d

utils/tflite/text_encoder_test.cc

diff --git a/utils/tflite/text_encoder_test.cc b/utils/tflite/text_encoder_test.cc
index d1892c7..0b6ff71 100644
--- a/utils/tflite/text_encoder_test.cc
+++ b/utils/tflite/text_encoder_test.cc
@@ -55,6 +55,9 @@
   std::vector<int> GetOutputEncoding() {
     return ExtractVector<int>(output_encoding_);
   }
+  std::vector<int> GetOutputPositions() {
+    return ExtractVector<int>(output_positions_);
+  }
   std::vector<int> GetOutputAttributeInt32() {
     return ExtractVector<int>(output_attributes_int32_);
   }
@@ -71,6 +74,7 @@
   int input_attributes_float_;
 
   int output_encoding_;
+  int output_positions_;
   int output_length_;
   int output_attributes_int32_;
   int output_attributes_float_;
@@ -86,6 +90,7 @@
   input_attributes_float_ = AddInput(tflite::TensorType_FLOAT32);
 
   output_encoding_ = AddOutput(tflite::TensorType_INT32);
+  output_positions_ = AddOutput(tflite::TensorType_INT32);
   output_length_ = AddOutput(tflite::TensorType_INT32);
   output_attributes_int32_ = AddOutput(tflite::TensorType_INT32);
   output_attributes_float_ = AddOutput(tflite::TensorType_FLOAT32);
@@ -113,6 +118,8 @@
   EXPECT_EQ(m.GetEncodedLength(), 5);
   EXPECT_THAT(m.GetOutputEncoding(),
               testing::ElementsAre(1, 90, 547, 58, 2, 2, 2, 2, 2, 2));
+  EXPECT_THAT(m.GetOutputPositions(),
+              testing::ElementsAre(0, 1, 2, 3, 4, 10, 10, 10, 10, 10));
   EXPECT_THAT(m.GetOutputAttributeInt32(),
               testing::ElementsAre(7, 7, 7, 7, 7, 7, 7, 7, 7, 7));
   EXPECT_THAT(
@@ -130,6 +137,8 @@
   EXPECT_EQ(m.GetEncodedLength(), 10);
   EXPECT_THAT(m.GetOutputEncoding(),
               testing::ElementsAre(547, 58, 2, 1, 862, 2, 1, 1919, 19, 2));
+  EXPECT_THAT(m.GetOutputPositions(),
+              testing::ElementsAre(2, 3, 4, 0, 1, 2, 0, 1, 2, 3));
   EXPECT_THAT(m.GetOutputAttributeInt32(),
               testing::ElementsAre(1, 1, 1, 2, 2, 2, 3, 3, 3, 3));
   EXPECT_THAT(
@@ -147,6 +156,8 @@
   EXPECT_EQ(m.GetEncodedLength(), 9);
   EXPECT_THAT(m.GetOutputEncoding(),
               testing::ElementsAre(862, 2, 1, 1919, 19, 2, 1, 862, 2));
+  EXPECT_THAT(m.GetOutputPositions(),
+              testing::ElementsAre(1, 2, 0, 1, 2, 3, 0, 1, 2));
   EXPECT_THAT(m.GetOutputAttributeInt32(),
               testing::ElementsAre(2, 2, 3, 3, 3, 3, 4, 4, 4));
   EXPECT_THAT(