aboutsummaryrefslogtreecommitdiff
path: root/src/armnnTfLiteParser/test/Softmax.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnnTfLiteParser/test/Softmax.cpp')
-rw-r--r--src/armnnTfLiteParser/test/Softmax.cpp9
1 files changed, 4 insertions, 5 deletions
diff --git a/src/armnnTfLiteParser/test/Softmax.cpp b/src/armnnTfLiteParser/test/Softmax.cpp
index c4d19f2ac0..11a2a0d84b 100644
--- a/src/armnnTfLiteParser/test/Softmax.cpp
+++ b/src/armnnTfLiteParser/test/Softmax.cpp
@@ -3,15 +3,14 @@
// SPDX-License-Identifier: MIT
//
-#include <boost/test/unit_test.hpp>
#include "ParserFlatbuffersFixture.hpp"
#include "../TfLiteParser.hpp"
#include <string>
#include <iostream>
-BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
-
+TEST_SUITE("TensorflowLiteParser_Softmax")
+{
struct SoftmaxFixture : public ParserFlatbuffersFixture
{
explicit SoftmaxFixture()
@@ -69,9 +68,9 @@ struct SoftmaxFixture : public ParserFlatbuffersFixture
}
};
-BOOST_FIXTURE_TEST_CASE(ParseSoftmaxLite, SoftmaxFixture)
+TEST_CASE_FIXTURE(SoftmaxFixture, "ParseSoftmaxLite")
{
RunTest<2, armnn::DataType::QAsymmU8>(0, { 0, 0, 100, 0, 0, 0, 0 }, { 0, 0, 255, 0, 0, 0, 0 });
}
-BOOST_AUTO_TEST_SUITE_END()
+}