IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[TFLite] Moved tests to a separate file

Browse Source
This commit is contained in:
Georgy Krivoruchko 2023-11-24 01:20:10 -08:00 committed by GitHub
parent a8e9989383
commit 71836a959d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 55 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/frontends/tensorflow_lite/tests/convert_model.cpp
View File

@ -13,9 +13,6 @@ using TFLiteConvertModelTest = FrontEndConvertModelTest;
static const std::vector<std::string> models{ static const std::vector<std::string> models{
std::string("2in_2out/2in_2out.tflite"), std::string("2in_2out/2in_2out.tflite"),
std::string("downloads/pose_detector.tflite"), std::string("downloads/pose_detector.tflite"),
std::string("bad_header/zerolen.tflite"),
std::string("bad_header/wrong_len_3.tflite"),
std::string("bad_header/wrong_pos.tflite"),
}; };
INSTANTIATE_TEST_SUITE_P(TFLiteConvertModelTest, INSTANTIATE_TEST_SUITE_P(TFLiteConvertModelTest,

53
src/frontends/tensorflow_lite/tests/convert_unsupported.cpp Normal file
View File

@ -0,0 +1,53 @@
// Copyright (C) 2018-2023 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
//
#include "convert_model.hpp"
#include "tf_utils.hpp"
#include "utils.hpp"
using namespace std;
using namespace ov;
using namespace ov::frontend;
TEST(FrontEndConvertModelTest, test_zerolen) {
FrontEndManager fem;
FrontEnd::Ptr frontEnd;
InputModel::Ptr inputModel;
ASSERT_NO_THROW(frontEnd = fem.load_by_framework(TF_LITE_FE));
ASSERT_NE(frontEnd, nullptr);
auto model_filename =
FrontEndTestUtils::make_model_path(string(TEST_TENSORFLOW_LITE_MODELS_DIRNAME) + string("zerolen.tflite"));
ASSERT_NO_THROW(inputModel = frontEnd->load(model_filename));
ASSERT_NE(inputModel, nullptr);
shared_ptr<ov::Model> model;
ASSERT_THROW(model = frontEnd->convert(inputModel), std::exception);
}
TEST(FrontEndConvertModelTest, test_wrong_len) {
FrontEndManager fem;
FrontEnd::Ptr frontEnd;
InputModel::Ptr inputModel;
ASSERT_NO_THROW(frontEnd = fem.load_by_framework(TF_LITE_FE));
ASSERT_NE(frontEnd, nullptr);
auto model_filename = FrontEndTestUtils::make_model_path(string(TEST_TENSORFLOW_LITE_MODELS_DIRNAME) +
string("bad_header/wrong_len_3.tflite"));
ASSERT_NO_THROW(inputModel = frontEnd->load(model_filename));
ASSERT_NE(inputModel, nullptr);
shared_ptr<ov::Model> model;
ASSERT_THROW(model = frontEnd->convert(inputModel), std::exception);
}
TEST(FrontEndConvertModelTest, test_wrong_pos) {
FrontEndManager fem;
FrontEnd::Ptr frontEnd;
InputModel::Ptr inputModel;
ASSERT_NO_THROW(frontEnd = fem.load_by_framework(TF_LITE_FE));
ASSERT_NE(frontEnd, nullptr);
auto model_filename =
FrontEndTestUtils::make_model_path(string(TEST_TENSORFLOW_LITE_MODELS_DIRNAME) + string("wrong_pos.tflite"));
ASSERT_NO_THROW(inputModel = frontEnd->load(model_filename));
ASSERT_NE(inputModel, nullptr);
shared_ptr<ov::Model> model;
ASSERT_THROW(model = frontEnd->convert(inputModel), std::exception);
}

14
src/frontends/tests/frontend/shared/src/convert_model.cpp
View File

@ -38,12 +38,7 @@ void FrontEndConvertModelTest::doLoadFromFile() {
TEST_P(FrontEndConvertModelTest, test_convert_partially_equal_convert) { TEST_P(FrontEndConvertModelTest, test_convert_partially_equal_convert) {
ASSERT_NO_THROW(doLoadFromFile()); ASSERT_NO_THROW(doLoadFromFile());
std::shared_ptr<ov::Model> model_ref; std::shared_ptr<ov::Model> model_ref;
if (m_modelFile.find("/bad_header/") == std::string::npos) { ASSERT_NO_THROW(model_ref = m_frontEnd->convert(m_inputModel));
ASSERT_NO_THROW(model_ref = m_frontEnd->convert(m_inputModel));
} else {
ASSERT_THROW(model_ref = m_frontEnd->convert(m_inputModel), std::exception);
return;
}
ASSERT_NE(model_ref, nullptr); ASSERT_NE(model_ref, nullptr);
std::shared_ptr<ov::Model> model; std::shared_ptr<ov::Model> model;
ASSERT_NO_THROW(model = m_frontEnd->convert_partially(m_inputModel)); ASSERT_NO_THROW(model = m_frontEnd->convert_partially(m_inputModel));
@ -61,12 +56,7 @@ TEST_P(FrontEndConvertModelTest, test_convert_partially_equal_convert) {
TEST_P(FrontEndConvertModelTest, test_decode_convert_equal_convert) { TEST_P(FrontEndConvertModelTest, test_decode_convert_equal_convert) {
ASSERT_NO_THROW(doLoadFromFile()); ASSERT_NO_THROW(doLoadFromFile());
std::shared_ptr<ov::Model> model_ref; std::shared_ptr<ov::Model> model_ref;
if (m_modelFile.find("/bad_header/") == std::string::npos) { ASSERT_NO_THROW(model_ref = m_frontEnd->convert(m_inputModel));
ASSERT_NO_THROW(model_ref = m_frontEnd->convert(m_inputModel));
} else {
ASSERT_THROW(model_ref = m_frontEnd->convert(m_inputModel), std::exception);
return;
}
ASSERT_NE(model_ref, nullptr); ASSERT_NE(model_ref, nullptr);
std::shared_ptr<ov::Model> model; std::shared_ptr<ov::Model> model;
ASSERT_NO_THROW(model = m_frontEnd->decode(m_inputModel)); ASSERT_NO_THROW(model = m_frontEnd->decode(m_inputModel));