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/armnnOnnxParser/test/CreateNetwork.cpp | 21 +++++++++++---------- 1 file changed, 11 insertions(+), 10 deletions(-) (limited to 'src/armnnOnnxParser/test/CreateNetwork.cpp') 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 #include "armnnOnnxParser/IOnnxParser.hpp" -#include "google/protobuf/stubs/logging.h" +#include -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() +} -- cgit v1.2.1