| Index: components/translate/core/browser/ranker_model_unittest.cc
|
| diff --git a/components/translate/core/browser/ranker_model_unittest.cc b/components/translate/core/browser/ranker_model_unittest.cc
|
| deleted file mode 100644
|
| index 83bab0b38e60e721e7ac3fdc9865f24e31dcce24..0000000000000000000000000000000000000000
|
| --- a/components/translate/core/browser/ranker_model_unittest.cc
|
| +++ /dev/null
|
| @@ -1,76 +0,0 @@
|
| -// Copyright 2017 The Chromium Authors. All rights reserved.
|
| -// Use of this source code is governed by a BSD-style license that can be
|
| -// found in the LICENSE file.
|
| -
|
| -#include "components/translate/core/browser/ranker_model.h"
|
| -
|
| -#include <memory>
|
| -
|
| -#include "base/memory/ptr_util.h"
|
| -#include "base/time/time.h"
|
| -#include "components/translate/core/browser/proto/ranker_model.pb.h"
|
| -#include "testing/gtest/include/gtest/gtest.h"
|
| -
|
| -namespace {
|
| -
|
| -using chrome_intelligence::RankerModel;
|
| -
|
| -const char kModelURL[] = "https://some.url.net/model";
|
| -
|
| -int64_t InSeconds(const base::Time t) {
|
| - return (t - base::Time()).InSeconds();
|
| -}
|
| -
|
| -std::unique_ptr<RankerModel> NewModel(const std::string& model_url,
|
| - base::Time last_modified,
|
| - base::TimeDelta cache_duration) {
|
| - std::unique_ptr<RankerModel> model = base::MakeUnique<RankerModel>();
|
| - auto* metadata = model->mutable_proto()->mutable_metadata();
|
| - if (!model_url.empty())
|
| - metadata->set_source(model_url);
|
| - if (!last_modified.is_null())
|
| - metadata->set_last_modified_sec(InSeconds(last_modified));
|
| - if (!cache_duration.is_zero())
|
| - metadata->set_cache_duration_sec(cache_duration.InSeconds());
|
| -
|
| - auto* translate = model->mutable_proto()->mutable_translate();
|
| - translate->set_version(1);
|
| -
|
| - auto* logit = translate->mutable_logistic_regression_model();
|
| - logit->set_bias(0.1f);
|
| - logit->set_accept_ratio_weight(0.2f);
|
| - logit->set_decline_ratio_weight(0.3f);
|
| - logit->set_ignore_ratio_weight(0.4f);
|
| - return model;
|
| -}
|
| -
|
| -} // namespace
|
| -
|
| -TEST(RankerModelTest, Serialization) {
|
| - base::Time last_modified = base::Time::Now();
|
| - base::TimeDelta cache_duration = base::TimeDelta::FromDays(3);
|
| - std::unique_ptr<RankerModel> original_model =
|
| - NewModel(kModelURL, last_modified, cache_duration);
|
| - std::string original_model_str = original_model->SerializeAsString();
|
| - std::unique_ptr<RankerModel> serialized_model =
|
| - RankerModel::FromString(original_model_str);
|
| - std::string serialized_model_str = serialized_model->SerializeAsString();
|
| -
|
| - EXPECT_EQ(serialized_model_str, original_model_str);
|
| - EXPECT_EQ(serialized_model->GetSourceURL(), kModelURL);
|
| - EXPECT_EQ(serialized_model->proto().metadata().last_modified_sec(),
|
| - InSeconds(last_modified));
|
| - EXPECT_EQ(serialized_model->proto().metadata().cache_duration_sec(),
|
| - cache_duration.InSeconds());
|
| -}
|
| -
|
| -TEST(RankerModelTest, IsExpired) {
|
| - base::Time today = base::Time::Now();
|
| - base::TimeDelta days_15 = base::TimeDelta::FromDays(15);
|
| - base::TimeDelta days_30 = base::TimeDelta::FromDays(30);
|
| - base::TimeDelta days_60 = base::TimeDelta::FromDays(60);
|
| -
|
| - EXPECT_FALSE(NewModel(kModelURL, today, days_30)->IsExpired());
|
| - EXPECT_FALSE(NewModel(kModelURL, today - days_15, days_30)->IsExpired());
|
| - EXPECT_TRUE(NewModel(kModelURL, today - days_60, days_30)->IsExpired());
|
| -}
|
|
|