aboutsummaryrefslogtreecommitdiff
path: root/src/armnnOnnxParser/test/CreateNetwork.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnnOnnxParser/test/CreateNetwork.cpp')
-rw-r--r--src/armnnOnnxParser/test/CreateNetwork.cpp21
1 files changed, 11 insertions, 10 deletions
diff --git a/src/armnnOnnxParser/test/CreateNetwork.cpp b/src/armnnOnnxParser/test/CreateNetwork.cpp
index 1e3e1b4fdd..eeeaca85dc 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 "google/protobuf/stubs/logging.h"
+#include <doctest/doctest.h>
-BOOST_AUTO_TEST_SUITE(OnnxParser)
+#include "google/protobuf/stubs/logging.h"
-BOOST_AUTO_TEST_CASE(CreateNetworkFromString)
+TEST_SUITE("OnnxParser_CreateNetwork")
+{
+TEST_CASE("CreateNetworkFromString")
{
std::string TestModel = R"(
ir_version: 3
@@ -43,21 +44,21 @@ BOOST_AUTO_TEST_CASE(CreateNetworkFromString)
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()
+}