diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2021-06-10 18:24:34 +0100 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2021-06-11 10:33:16 +0000 |
commit | 1625efc870f1a8b7c6e6382277ddbb245f91a294 (patch) | |
tree | 39fbbaa15ed7eb81337b082c2d20b0af68b91c02 /src/armnnOnnxParser/test/GetInputsOutputs.cpp | |
parent | 958e0ba61e940a8d11955cf2a10f681c7c47e1fa (diff) | |
download | armnn-1625efc870f1a8b7c6e6382277ddbb245f91a294.tar.gz |
IVGCVSW-5963 'Move unit tests to new framework'
* Used doctest in ArmNN unit tests
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: Ia9cf5fc72775878885c5f864abf2c56b3a935f1a
Diffstat (limited to 'src/armnnOnnxParser/test/GetInputsOutputs.cpp')
-rw-r--r-- | src/armnnOnnxParser/test/GetInputsOutputs.cpp | 43 |
1 files changed, 20 insertions, 23 deletions
diff --git a/src/armnnOnnxParser/test/GetInputsOutputs.cpp b/src/armnnOnnxParser/test/GetInputsOutputs.cpp index 5bb3095cc7..ac06cba410 100644 --- a/src/armnnOnnxParser/test/GetInputsOutputs.cpp +++ b/src/armnnOnnxParser/test/GetInputsOutputs.cpp @@ -2,17 +2,16 @@ // Copyright © 2017 Arm Ltd. All rights reserved. // SPDX-License-Identifier: MIT // -#include <boost/test/unit_test.hpp> + #include "../OnnxParser.hpp" #include "ParserPrototxtFixture.hpp" #include <onnx/onnx.pb.h> #include "google/protobuf/stubs/logging.h" - using ModelPtr = std::unique_ptr<onnx::ModelProto>; -BOOST_AUTO_TEST_SUITE(OnnxParser) - +TEST_SUITE("OnnxParser_GetInputsOutputs") +{ struct GetInputsOutputsMainFixture : public armnnUtils::ParserPrototxtFixture<armnnOnnxParser::IOnnxParser> { explicit GetInputsOutputsMainFixture() @@ -66,21 +65,21 @@ struct GetInputsOutputsMainFixture : public armnnUtils::ParserPrototxtFixture<ar }; -BOOST_FIXTURE_TEST_CASE(GetInput, GetInputsOutputsMainFixture) +TEST_CASE_FIXTURE(GetInputsOutputsMainFixture, "GetInput") { ModelPtr model = armnnOnnxParser::OnnxParserImpl::LoadModelFromString(m_Prototext.c_str()); std::vector<std::string> tensors = armnnOnnxParser::OnnxParserImpl::GetInputs(model); - BOOST_CHECK_EQUAL(1, tensors.size()); - BOOST_CHECK_EQUAL("Input", tensors[0]); + CHECK_EQ(1, tensors.size()); + CHECK_EQ("Input", tensors[0]); } -BOOST_FIXTURE_TEST_CASE(GetOutput, GetInputsOutputsMainFixture) +TEST_CASE_FIXTURE(GetInputsOutputsMainFixture, "GetOutput") { ModelPtr model = armnnOnnxParser::OnnxParserImpl::LoadModelFromString(m_Prototext.c_str()); std::vector<std::string> tensors = armnnOnnxParser::OnnxParserImpl::GetOutputs(model); - BOOST_CHECK_EQUAL(1, tensors.size()); - BOOST_CHECK_EQUAL("Output", tensors[0]); + CHECK_EQ(1, tensors.size()); + CHECK_EQ("Output", tensors[0]); } struct GetEmptyInputsOutputsFixture : public armnnUtils::ParserPrototxtFixture<armnnOnnxParser::IOnnxParser> @@ -137,22 +136,22 @@ struct GetEmptyInputsOutputsFixture : public armnnUtils::ParserPrototxtFixture<a } }; -BOOST_FIXTURE_TEST_CASE(GetEmptyInputs, GetEmptyInputsOutputsFixture) +TEST_CASE_FIXTURE(GetEmptyInputsOutputsFixture, "GetEmptyInputs") { ModelPtr model = armnnOnnxParser::OnnxParserImpl::LoadModelFromString(m_Prototext.c_str()); std::vector<std::string> tensors = armnnOnnxParser::OnnxParserImpl::GetInputs(model); - BOOST_CHECK_EQUAL(0, tensors.size()); + CHECK_EQ(0, tensors.size()); } -BOOST_AUTO_TEST_CASE(GetInputsNullModel) +TEST_CASE("GetInputsNullModel") { - BOOST_CHECK_THROW(armnnOnnxParser::OnnxParserImpl::LoadModelFromString(""), armnn::InvalidArgumentException); + CHECK_THROWS_AS(armnnOnnxParser::OnnxParserImpl::LoadModelFromString(""), armnn::InvalidArgumentException); } -BOOST_AUTO_TEST_CASE(GetOutputsNullModel) +TEST_CASE("GetOutputsNullModel") { auto silencer = google::protobuf::LogSilencer(); //get rid of errors from protobuf - BOOST_CHECK_THROW(armnnOnnxParser::OnnxParserImpl::LoadModelFromString("nknnk"), armnn::ParseException); + CHECK_THROWS_AS(armnnOnnxParser::OnnxParserImpl::LoadModelFromString("nknnk"), armnn::ParseException); } struct GetInputsMultipleFixture : public armnnUtils::ParserPrototxtFixture<armnnOnnxParser::IOnnxParser> @@ -241,15 +240,13 @@ struct GetInputsMultipleFixture : public armnnUtils::ParserPrototxtFixture<armnn } }; -BOOST_FIXTURE_TEST_CASE(GetInputsMultipleInputs, GetInputsMultipleFixture) +TEST_CASE_FIXTURE(GetInputsMultipleFixture, "GetInputsMultipleInputs") { ModelPtr model = armnnOnnxParser::OnnxParserImpl::LoadModelFromString(m_Prototext.c_str()); std::vector<std::string> tensors = armnnOnnxParser::OnnxParserImpl::GetInputs(model); - BOOST_CHECK_EQUAL(2, tensors.size()); - BOOST_CHECK_EQUAL("Input0", tensors[0]); - BOOST_CHECK_EQUAL("Input1", tensors[1]); + CHECK_EQ(2, tensors.size()); + CHECK_EQ("Input0", tensors[0]); + CHECK_EQ("Input1", tensors[1]); } - - -BOOST_AUTO_TEST_SUITE_END() +} |