// Copyright (C) 2018-2026 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
//

#pragma once

#include <string>
#include <tuple>

#include "openvino/op/topk.hpp"
#include "shared_test_classes/base/ov_subgraph.hpp"

namespace ov {
namespace test {
typedef std::tuple<
        int64_t,                        // keepK
        int64_t,                        // axis
        ov::op::v1::TopK::Mode,         // mode
        ov::op::v1::TopK::SortType,     // sort
        ov::element::Type,              // Model type
        std::vector<InputShape>,        // Input shape
        std::string                     // Target device name
> TopKParams;

typedef std::tuple<
        int64_t,                        // keepK
        int64_t,                        // axis
        ov::op::v1::TopK::Mode,         // mode
        ov::op::v1::TopK::SortType,     // sort
        ov::element::Type,              // Model type
        std::vector<InputShape>,        // Input shape
        bool,                           // Stable
        std::string                     // Target device name
> TopK11Params;

class TopKLayerTest : public testing::WithParamInterface<TopKParams>,
                      virtual public ov::test::SubgraphBaseTest {
public:
    static std::string getTestCaseName(const testing::TestParamInfo<TopKParams>& obj);

protected:
    void SetUp() override;
};

class TopK11LayerTest : public testing::WithParamInterface<TopK11Params>,
                      virtual public ov::test::SubgraphBaseTest {
public:
    static std::string getTestCaseName(const testing::TestParamInfo<TopK11Params>& obj);

protected:
    void SetUp() override;
};
}  // namespace test
}  // namespace ov
