diff options
Diffstat (limited to 'src/armnnTfLiteParser/test/TfLiteParser.cpp')
-rw-r--r-- | src/armnnTfLiteParser/test/TfLiteParser.cpp | 16 |
1 files changed, 8 insertions, 8 deletions
diff --git a/src/armnnTfLiteParser/test/TfLiteParser.cpp b/src/armnnTfLiteParser/test/TfLiteParser.cpp index 36827c0586..53fe4a33e7 100644 --- a/src/armnnTfLiteParser/test/TfLiteParser.cpp +++ b/src/armnnTfLiteParser/test/TfLiteParser.cpp @@ -3,18 +3,18 @@ // SPDX-License-Identifier: MIT // -#include <boost/test/unit_test.hpp> #include "ParserFlatbuffersFixture.hpp" #include "../TfLiteParser.hpp" +#include <doctest/doctest.h> -BOOST_AUTO_TEST_SUITE(TensorflowLiteParser) - -BOOST_AUTO_TEST_CASE(ParseEmptyBinaryData) +TEST_SUITE("TensorflowLiteParser") +{ +TEST_CASE("ParseEmptyBinaryData") { ITfLiteParser::TfLiteParserOptions options; ITfLiteParserPtr m_Parser(ITfLiteParser::Create(armnn::Optional<ITfLiteParser::TfLiteParserOptions>(options))); // Should throw armnn::ParseException: Buffer doesn't conform to the expected Tensorflow Lite flatbuffers format. - BOOST_CHECK_THROW(m_Parser->CreateNetworkFromBinary({0}), armnn::ParseException); + CHECK_THROWS_AS(m_Parser->CreateNetworkFromBinary({0}), armnn::ParseException); } struct NoInputBindingsFixture : public ParserFlatbuffersFixture @@ -32,10 +32,10 @@ struct NoInputBindingsFixture : public ParserFlatbuffersFixture } }; -BOOST_FIXTURE_TEST_CASE( ParseBadInputBindings, NoInputBindingsFixture ) +TEST_CASE_FIXTURE(NoInputBindingsFixture, "ParseBadInputBindings") { // Should throw armnn::ParseException: No input binding found for subgraph:0 and name:inputTensor. - BOOST_CHECK_THROW( (RunTest<4, armnn::DataType::QAsymmU8>(0, { }, { 0 })), armnn::ParseException); + CHECK_THROWS_AS((RunTest<4, armnn::DataType::QAsymmU8>(0, { }, { 0 })), armnn::ParseException); } -BOOST_AUTO_TEST_SUITE_END() +} |