TheRock/rocm-libraries/projects/hipdnn/data_sdk/tests/utilities/TestVisitor.cpp

// Copyright © Advanced Micro Devices, Inc., or its affiliates.
// SPDX-License-Identifier:  MIT

#include <gtest/gtest.h>

#include <hipdnn_data_sdk/utilities/Visitor.hpp>
#include <string>
#include <variant>

using namespace hipdnn_data_sdk::utilities;

TEST(TestVisitor, Correctness)
{
    using VariantType = std::variant<float, int, char, std::string>;
    auto visitor = hipdnn_data_sdk::utilities::Visitor{
        [](float) { return 0; }, [](int) { return 1; }, [](const auto&) { return 2; }};

    EXPECT_EQ(std::visit(visitor, VariantType{0.f}), 0);
    EXPECT_EQ(std::visit(visitor, VariantType{1}), 1);
    EXPECT_EQ(std::visit(visitor, VariantType{'c'}), 2);
    EXPECT_EQ(std::visit(visitor, VariantType{std::string{"abc"}}), 2);
}