chromium/components/segmentation_platform/embedder/default_model/android_home_module_ranker_unittest.cc

// Copyright 2024 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/segmentation_platform/embedder/default_model/android_home_module_ranker.h"

#include "components/segmentation_platform/embedder/default_model/default_model_test_base.h"
#include "components/segmentation_platform/public/constants.h"
#include "components/segmentation_platform/public/features.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace segmentation_platform {

class AndroidHomeModuleRankerTest : public DefaultModelTestBase { … };

TEST_F(AndroidHomeModuleRankerTest, InitAndFetchModel) { … }

TEST_F(AndroidHomeModuleRankerTest, ExecuteModelWithInputForAllModules) { … }

TEST_F(AndroidHomeModuleRankerTest,
       ExecuteModelWithInputForAllModulesWithEngagement) { … }

TEST_F(AndroidHomeModuleRankerTest,
       ExecuteModelWithInputForAllModulesWithImpressions) { … }

}  // namespace segmentation_platform