From 71836a959dff133daf10717eb069fcb2edefd7e0 Mon Sep 17 00:00:00 2001 From: Georgy Krivoruchko Date: Fri, 24 Nov 2023 01:20:10 -0800 Subject: [PATCH] [TFLite] Moved tests to a separate file --- .../tensorflow_lite/tests/convert_model.cpp | 3 -- .../tests/convert_unsupported.cpp | 53 +++++++++++++++++++ .../frontend/shared/src/convert_model.cpp | 14 +---- 3 files changed, 55 insertions(+), 15 deletions(-) create mode 100644 src/frontends/tensorflow_lite/tests/convert_unsupported.cpp diff --git a/src/frontends/tensorflow_lite/tests/convert_model.cpp b/src/frontends/tensorflow_lite/tests/convert_model.cpp index 9a444037ef3..14e6e3bae03 100644 --- a/src/frontends/tensorflow_lite/tests/convert_model.cpp +++ b/src/frontends/tensorflow_lite/tests/convert_model.cpp @@ -13,9 +13,6 @@ using TFLiteConvertModelTest = FrontEndConvertModelTest; static const std::vector models{ std::string("2in_2out/2in_2out.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, diff --git a/src/frontends/tensorflow_lite/tests/convert_unsupported.cpp b/src/frontends/tensorflow_lite/tests/convert_unsupported.cpp new file mode 100644 index 00000000000..c56fae2d1f0 --- /dev/null +++ b/src/frontends/tensorflow_lite/tests/convert_unsupported.cpp @@ -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 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 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 model; + ASSERT_THROW(model = frontEnd->convert(inputModel), std::exception); +} diff --git a/src/frontends/tests/frontend/shared/src/convert_model.cpp b/src/frontends/tests/frontend/shared/src/convert_model.cpp index e10464fe8ae..bc7c661950f 100644 --- a/src/frontends/tests/frontend/shared/src/convert_model.cpp +++ b/src/frontends/tests/frontend/shared/src/convert_model.cpp @@ -38,12 +38,7 @@ void FrontEndConvertModelTest::doLoadFromFile() { TEST_P(FrontEndConvertModelTest, test_convert_partially_equal_convert) { ASSERT_NO_THROW(doLoadFromFile()); std::shared_ptr model_ref; - if (m_modelFile.find("/bad_header/") == std::string::npos) { - ASSERT_NO_THROW(model_ref = m_frontEnd->convert(m_inputModel)); - } else { - ASSERT_THROW(model_ref = m_frontEnd->convert(m_inputModel), std::exception); - return; - } + ASSERT_NO_THROW(model_ref = m_frontEnd->convert(m_inputModel)); ASSERT_NE(model_ref, nullptr); std::shared_ptr model; 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) { ASSERT_NO_THROW(doLoadFromFile()); std::shared_ptr model_ref; - if (m_modelFile.find("/bad_header/") == std::string::npos) { - ASSERT_NO_THROW(model_ref = m_frontEnd->convert(m_inputModel)); - } else { - ASSERT_THROW(model_ref = m_frontEnd->convert(m_inputModel), std::exception); - return; - } + ASSERT_NO_THROW(model_ref = m_frontEnd->convert(m_inputModel)); ASSERT_NE(model_ref, nullptr); std::shared_ptr model; ASSERT_NO_THROW(model = m_frontEnd->decode(m_inputModel));