34 lines
882 B
C++
34 lines
882 B
C++
// Copyright (C) 2018-2021 Intel Corporation
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
#pragma once
|
|
|
|
#include <frontend_manager/frontend_manager.hpp>
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
using BasicTestParam = std::tuple<std::string, // FrontEnd name
|
|
std::string, // Base path to models
|
|
std::string>; // Model name
|
|
|
|
class FrontEndBasicTest : public ::testing::TestWithParam<BasicTestParam>
|
|
{
|
|
public:
|
|
std::string m_feName;
|
|
std::string m_pathToModels;
|
|
std::string m_modelFile;
|
|
ngraph::frontend::FrontEndManager m_fem;
|
|
ngraph::frontend::FrontEnd::Ptr m_frontEnd;
|
|
ngraph::frontend::InputModel::Ptr m_inputModel;
|
|
|
|
static std::string getTestCaseName(const testing::TestParamInfo<BasicTestParam>& obj);
|
|
|
|
void SetUp() override;
|
|
|
|
protected:
|
|
void initParamTest();
|
|
|
|
void doLoadFromFile();
|
|
};
|