commit 1625efc870f1a8b7c6e6382277ddbb245f91a294
author Sadik Armagan <sadik.armagan@arm.com> Thu Jun 10 18:24:34 2021 +0100
committer Sadik Armagan <sadik.armagan@arm.com> Fri Jun 11 10:33:16 2021 +0000
tree 39fbbaa15ed7eb81337b082c2d20b0af68b91c02
parent 958e0ba61e940a8d11955cf2a10f681c7c47e1fa

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/CreateNetwork.cpp

diff --git a/src/armnnOnnxParser/test/CreateNetwork.cpp b/src/armnnOnnxParser/test/CreateNetwork.cpp
index 1e3e1b4..eeeaca8 100644
--- a/src/armnnOnnxParser/test/CreateNetwork.cpp
+++ b/src/armnnOnnxParser/test/CreateNetwork.cpp
@@ -3,13 +3,14 @@
 // SPDX-License-Identifier: MIT
 //
 
-#include <boost/test/unit_test.hpp>
 #include  "armnnOnnxParser/IOnnxParser.hpp"
+#include <doctest/doctest.h>
+
 #include "google/protobuf/stubs/logging.h"
 
-BOOST_AUTO_TEST_SUITE(OnnxParser)
-
-BOOST_AUTO_TEST_CASE(CreateNetworkFromString)
+TEST_SUITE("OnnxParser_CreateNetwork")
+{
+TEST_CASE("CreateNetworkFromString")
 {
   std::string TestModel = R"(
                           ir_version: 3
@@ -43,21 +44,21 @@
     armnnOnnxParser::IOnnxParserPtr parser(armnnOnnxParser::IOnnxParser::Create());
 
     armnn::INetworkPtr network = parser->CreateNetworkFromString(TestModel.c_str());
-    BOOST_TEST(network.get());
+    CHECK(network.get());
 }
 
-BOOST_AUTO_TEST_CASE(CreateNetworkFromStringWithNullptr)
+TEST_CASE("CreateNetworkFromStringWithNullptr")
 {
     armnnOnnxParser::IOnnxParserPtr parser(armnnOnnxParser::IOnnxParser::Create());
-    BOOST_CHECK_THROW(parser->CreateNetworkFromString(""), armnn::InvalidArgumentException );
+    CHECK_THROWS_AS(parser->CreateNetworkFromString(""), armnn::InvalidArgumentException );
 }
 
-BOOST_AUTO_TEST_CASE(CreateNetworkWithInvalidString)
+TEST_CASE("CreateNetworkWithInvalidString")
 {
     auto silencer = google::protobuf::LogSilencer(); //get rid of errors from protobuf
     armnnOnnxParser::IOnnxParserPtr parser(armnnOnnxParser::IOnnxParser::Create());
-    BOOST_CHECK_THROW(parser->CreateNetworkFromString( "I'm not a model so I should raise an error" ),
+    CHECK_THROWS_AS(parser->CreateNetworkFromString( "I'm not a model so I should raise an error" ),
                       armnn::ParseException );
 }
 
-BOOST_AUTO_TEST_SUITE_END()
+}