chromium/components/autofill/core/browser/ml_model/autofill_model_encoder_unittest.cc

// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "components/autofill/core/browser/ml_model/autofill_model_encoder.h"

#include "base/base_paths.h"
#include "base/files/file_path.h"
#include "base/files/file_util.h"
#include "base/path_service.h"
#include "base/strings/string_split.h"
#include "components/autofill/core/browser/autofill_field.h"
#include "components/autofill/core/browser/autofill_form_test_utils.h"
#include "components/autofill/core/browser/autofill_test_utils.h"
#include "components/autofill/core/browser/form_structure.h"
#include "components/autofill/core/common/form_field_data.h"
#include "components/optimization_guide/proto/autofill_field_classification_model_metadata.pb.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace autofill {

TokenId;
ElementsAre;

class AutofillModelEncoderTest : public testing::Test { … };

TEST_F(AutofillModelEncoderTest, TokensMappedCorrectly) { … }

// Tests that words out of vocabulary return 1.
TEST_F(AutofillModelEncoderTest, WordOutOfVocab) { … }

// Tests that empty strings return 0 for padding.
TEST_F(AutofillModelEncoderTest, EmptyToken) { … }

TEST_F(AutofillModelEncoderTest, InputEncodedCorrectly) { … }

// If a field label has more than one consecutive whitespace, they
// should all be removed without any empty strings.
TEST_F(AutofillModelEncoderTest, InputHasMoreThanOneWhitespace) { … }

TEST_F(AutofillModelEncoderTest, ReplaceSpecialWithWhitespace) { … }

// If a field label has more words than the kAttributeOutputSequenceLength,
// only the first kOutputSequenceLength many words should be used and the
// rest are ignored.
TEST_F(AutofillModelEncoderTest, InputHasMoreWordsThanOutputSequenceLength) { … }

TEST_F(AutofillModelEncoderTest, AttributeEncodedCorrectly) { … }

TEST_F(AutofillModelEncoderTest, InputConstructedCorrectly) { … }

TEST_F(AutofillModelEncoderTest, FormEncodedCorrectly) { … }

}  // namespace autofill