From 1625efc870f1a8b7c6e6382277ddbb245f91a294 Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Thu, 10 Jun 2021 18:24:34 +0100 Subject: IVGCVSW-5963 'Move unit tests to new framework' * Used doctest in ArmNN unit tests Signed-off-by: Sadik Armagan Change-Id: Ia9cf5fc72775878885c5f864abf2c56b3a935f1a --- src/armnnTfLiteParser/test/LoadModel.cpp | 76 ++++++++++++++++---------------- 1 file changed, 38 insertions(+), 38 deletions(-) (limited to 'src/armnnTfLiteParser/test/LoadModel.cpp') diff --git a/src/armnnTfLiteParser/test/LoadModel.cpp b/src/armnnTfLiteParser/test/LoadModel.cpp index 1afb5f12e5..e09de68c72 100644 --- a/src/armnnTfLiteParser/test/LoadModel.cpp +++ b/src/armnnTfLiteParser/test/LoadModel.cpp @@ -2,7 +2,7 @@ // Copyright © 2017 Arm Ltd. All rights reserved. // SPDX-License-Identifier: MIT // -#include + #include "ParserFlatbuffersFixture.hpp" #include "../TfLiteParser.hpp" @@ -13,8 +13,8 @@ using ModelPtr = TfLiteParserImpl::ModelPtr; using SubgraphPtr = TfLiteParserImpl::SubgraphPtr; using OperatorPtr = TfLiteParserImpl::OperatorPtr; -BOOST_AUTO_TEST_SUITE(TensorflowLiteParser) - +TEST_SUITE("TensorflowLiteParser_LoadModel") +{ struct LoadModelFixture : public ParserFlatbuffersFixture { explicit LoadModelFixture() @@ -137,53 +137,53 @@ struct LoadModelFixture : public ParserFlatbuffersFixture const std::vector& opcodes, size_t subgraphs, const std::string desc, size_t buffers) { - BOOST_CHECK(model); - BOOST_CHECK_EQUAL(version, model->version); - BOOST_CHECK_EQUAL(opcodeSize, model->operator_codes.size()); + CHECK(model); + CHECK_EQ(version, model->version); + CHECK_EQ(opcodeSize, model->operator_codes.size()); CheckBuiltinOperators(opcodes, model->operator_codes); - BOOST_CHECK_EQUAL(subgraphs, model->subgraphs.size()); - BOOST_CHECK_EQUAL(desc, model->description); - BOOST_CHECK_EQUAL(buffers, model->buffers.size()); + CHECK_EQ(subgraphs, model->subgraphs.size()); + CHECK_EQ(desc, model->description); + CHECK_EQ(buffers, model->buffers.size()); } void CheckBuiltinOperators(const std::vector& expectedOperators, const std::vector>& result) { - BOOST_CHECK_EQUAL(expectedOperators.size(), result.size()); + CHECK_EQ(expectedOperators.size(), result.size()); for (size_t i = 0; i < expectedOperators.size(); i++) { - BOOST_CHECK_EQUAL(expectedOperators[i], result[i]->builtin_code); + CHECK_EQ(expectedOperators[i], result[i]->builtin_code); } } void CheckSubgraph(const SubgraphPtr& subgraph, size_t tensors, const std::vector& inputs, const std::vector& outputs, size_t operators, const std::string& name) { - BOOST_CHECK(subgraph); - BOOST_CHECK_EQUAL(tensors, subgraph->tensors.size()); - BOOST_CHECK_EQUAL_COLLECTIONS(inputs.begin(), inputs.end(), subgraph->inputs.begin(), subgraph->inputs.end()); - BOOST_CHECK_EQUAL_COLLECTIONS(outputs.begin(), outputs.end(), - subgraph->outputs.begin(), subgraph->outputs.end()); - BOOST_CHECK_EQUAL(operators, subgraph->operators.size()); - BOOST_CHECK_EQUAL(name, subgraph->name); + CHECK(subgraph); + CHECK_EQ(tensors, subgraph->tensors.size()); + CHECK(std::equal(inputs.begin(), inputs.end(), subgraph->inputs.begin(), subgraph->inputs.end())); + CHECK(std::equal(outputs.begin(), outputs.end(), + subgraph->outputs.begin(), subgraph->outputs.end())); + CHECK_EQ(operators, subgraph->operators.size()); + CHECK_EQ(name, subgraph->name); } void CheckOperator(const OperatorPtr& operatorPtr, uint32_t opcode, const std::vector& inputs, const std::vector& outputs, tflite::BuiltinOptions optionType, tflite::CustomOptionsFormat custom_options_format) { - BOOST_CHECK(operatorPtr); - BOOST_CHECK_EQUAL(opcode, operatorPtr->opcode_index); - BOOST_CHECK_EQUAL_COLLECTIONS(inputs.begin(), inputs.end(), - operatorPtr->inputs.begin(), operatorPtr->inputs.end()); - BOOST_CHECK_EQUAL_COLLECTIONS(outputs.begin(), outputs.end(), - operatorPtr->outputs.begin(), operatorPtr->outputs.end()); - BOOST_CHECK_EQUAL(optionType, operatorPtr->builtin_options.type); - BOOST_CHECK_EQUAL(custom_options_format, operatorPtr->custom_options_format); + CHECK(operatorPtr); + CHECK_EQ(opcode, operatorPtr->opcode_index); + CHECK(std::equal(inputs.begin(), inputs.end(), + operatorPtr->inputs.begin(), operatorPtr->inputs.end())); + CHECK(std::equal(outputs.begin(), outputs.end(), + operatorPtr->outputs.begin(), operatorPtr->outputs.end())); + CHECK_EQ(optionType, operatorPtr->builtin_options.type); + CHECK_EQ(custom_options_format, operatorPtr->custom_options_format); } }; -BOOST_FIXTURE_TEST_CASE(LoadModelFromBinary, LoadModelFixture) +TEST_CASE_FIXTURE(LoadModelFixture, "LoadModelFromBinary") { TfLiteParserImpl::ModelPtr model = TfLiteParserImpl::LoadModelFromBinary(m_GraphBinary.data(), m_GraphBinary.size()); @@ -197,14 +197,14 @@ BOOST_FIXTURE_TEST_CASE(LoadModelFromBinary, LoadModelFixture) tflite::CustomOptionsFormat_FLEXBUFFERS); } -BOOST_FIXTURE_TEST_CASE(LoadModelFromFile, LoadModelFixture) +TEST_CASE_FIXTURE(LoadModelFixture, "LoadModelFromFile") { using namespace fs; fs::path fname = armnnUtils::Filesystem::NamedTempFile("Armnn-tfLite-LoadModelFromFile-TempFile.csv"); bool saved = flatbuffers::SaveFile(fname.c_str(), reinterpret_cast(m_GraphBinary.data()), m_GraphBinary.size(), true); - BOOST_CHECK_MESSAGE(saved, "Cannot save test file"); + CHECK_MESSAGE(saved, "Cannot save test file"); TfLiteParserImpl::ModelPtr model = TfLiteParserImpl::LoadModelFromFile(fname.c_str()); CheckModel(model, 3, 2, { tflite::BuiltinOperator_AVERAGE_POOL_2D, tflite::BuiltinOperator_CONV_2D }, @@ -218,26 +218,26 @@ BOOST_FIXTURE_TEST_CASE(LoadModelFromFile, LoadModelFixture) remove(fname); } -BOOST_AUTO_TEST_CASE(LoadNullBinary) +TEST_CASE("LoadNullBinary") { - BOOST_CHECK_THROW(TfLiteParserImpl::LoadModelFromBinary(nullptr, 0), armnn::InvalidArgumentException); + CHECK_THROWS_AS(TfLiteParserImpl::LoadModelFromBinary(nullptr, 0), armnn::InvalidArgumentException); } -BOOST_AUTO_TEST_CASE(LoadInvalidBinary) +TEST_CASE("LoadInvalidBinary") { std::string testData = "invalid data"; - BOOST_CHECK_THROW(TfLiteParserImpl::LoadModelFromBinary(reinterpret_cast(&testData), + CHECK_THROWS_AS(TfLiteParserImpl::LoadModelFromBinary(reinterpret_cast(&testData), testData.length()), armnn::ParseException); } -BOOST_AUTO_TEST_CASE(LoadFileNotFound) +TEST_CASE("LoadFileNotFound") { - BOOST_CHECK_THROW(TfLiteParserImpl::LoadModelFromFile("invalidfile.tflite"), armnn::FileNotFoundException); + CHECK_THROWS_AS(TfLiteParserImpl::LoadModelFromFile("invalidfile.tflite"), armnn::FileNotFoundException); } -BOOST_AUTO_TEST_CASE(LoadNullPtrFile) +TEST_CASE("LoadNullPtrFile") { - BOOST_CHECK_THROW(TfLiteParserImpl::LoadModelFromFile(nullptr), armnn::InvalidArgumentException); + CHECK_THROWS_AS(TfLiteParserImpl::LoadModelFromFile(nullptr), armnn::InvalidArgumentException); } -BOOST_AUTO_TEST_SUITE_END() +} -- cgit v1.2.1