Function TEST

tensorflow/lite/models/smartreply/ops/extract_feature_test.cc:60–68

}

TEST(ExtractFeatureOpTest, RegularInput) {
  ExtractFeatureOpModel m({"<S>", "<S> Hi", "Hi", "Hi !", "!", "! <E>", "<E>"});
  m.Invoke();
  EXPECT_THAT(m.GetSignature(),
              ElementsAre(0, CalcFeature("<S> Hi"), CalcFeature("Hi"),
                          CalcFeature("Hi !"), CalcFeature("!"),
                          CalcFeature("! <E>"), 0));
  EXPECT_THAT(m.GetWeight(), ElementsAre(0, 2, 1, 2, 1, 2, 0));
}

TEST(ExtractFeatureOpTest, OneInput) {
  ExtractFeatureOpModel m({"Hi"});