From 1625efc870f1a8b7c6e6382277ddbb245f91a294 Mon Sep 17 00:00:00 2001
From: Sadik Armagan <sadik.armagan@arm.com>
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 <sadik.armagan@arm.com>
Change-Id: Ia9cf5fc72775878885c5f864abf2c56b3a935f1a
---
 src/armnnOnnxParser/test/GetInputsOutputs.cpp | 43 +++++++++++++--------------
 1 file changed, 20 insertions(+), 23 deletions(-)

(limited to 'src/armnnOnnxParser/test/GetInputsOutputs.cpp')

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()
+}
-- 
cgit v1.2.1