From d2a671b3cdf24411f8eb36f15f4b3b74cf2d0311 Mon Sep 17 00:00:00 2001 From: Ilya Lavrenov Date: Thu, 14 Oct 2021 12:38:42 +0300 Subject: [PATCH] FEM test (#7899) --- ngraph/test/frontend/shared/src/load_from.cpp | 20 +++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/ngraph/test/frontend/shared/src/load_from.cpp b/ngraph/test/frontend/shared/src/load_from.cpp index 083ea2c22e3..0c93f71e197 100644 --- a/ngraph/test/frontend/shared/src/load_from.cpp +++ b/ngraph/test/frontend/shared/src/load_from.cpp @@ -6,6 +6,7 @@ #include +#include "openvino/core/variant.hpp" #include "utils.hpp" using namespace ngraph; @@ -40,6 +41,25 @@ TEST_P(FrontEndLoadFromTest, testLoadFromFilePath) { ASSERT_NE(function, nullptr); } +TEST_P(FrontEndLoadFromTest, testLoadFromFilePathWithExplicitVariants) { + std::string model_path = FrontEndTestUtils::make_model_path(m_param.m_modelsPath + m_param.m_file); + std::vector frontends; + FrontEnd::Ptr fe; + ASSERT_NO_THROW(frontends = m_fem.get_available_front_ends()); + + std::vector> variants; + variants.emplace_back(ov::make_variant(model_path)); + ASSERT_NO_THROW(m_frontEnd = m_fem.load_by_model(variants)); + ASSERT_NE(m_frontEnd, nullptr); + + ASSERT_NO_THROW(m_inputModel = m_frontEnd->load(variants)); + ASSERT_NE(m_inputModel, nullptr); + + std::shared_ptr function; + ASSERT_NO_THROW(function = m_frontEnd->convert(m_inputModel)); + ASSERT_NE(function, nullptr); +} + TEST_P(FrontEndLoadFromTest, testLoadFromTwoFiles) { std::string model_path = FrontEndTestUtils::make_model_path(m_param.m_modelsPath + m_param.m_files[0]); std::string weights_path = FrontEndTestUtils::make_model_path(m_param.m_modelsPath + m_param.m_files[1]);