Function TEST

tensorflow/lite/models/smartreply/ops/predict_test.cc:108–117

};

TEST(PredictOpTest, AllLabelsAreValid) {
  PredictOpModel m({4}, {5}, {5, 2}, 2, 0.0001);
  m.SetInputSignature({1, 3, 7, 9});
  m.SetModelKey({1, 2, 4, 6, 7});
  m.SetModelLabel({11, 12, 11, 12, 11, 12, 11, 12, 11, 12});
  m.SetModelWeight({0.1, 0.2, 0.1, 0.2, 0.1, 0.2, 0.1, 0.2, 0.1, 0.2});
  m.Invoke();
  EXPECT_THAT(m.GetLabel(), ElementsAreArray({12, 11}));
  EXPECT_THAT(m.GetWeight(), ElementsAreArray(ArrayFloatNear({0.1, 0.05})));
}

TEST(PredictOpTest, MoreLabelsThanRequired) {
  PredictOpModel m({4}, {5}, {5, 2}, 1, 0.0001);