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/armnnTfLiteParser/test/SplitV.cpp | 15 +++++++-------- 1 file changed, 7 insertions(+), 8 deletions(-) (limited to 'src/armnnTfLiteParser/test/SplitV.cpp') diff --git a/src/armnnTfLiteParser/test/SplitV.cpp b/src/armnnTfLiteParser/test/SplitV.cpp index 9541114d71..51b75faf07 100644 --- a/src/armnnTfLiteParser/test/SplitV.cpp +++ b/src/armnnTfLiteParser/test/SplitV.cpp @@ -3,15 +3,14 @@ // SPDX-License-Identifier: MIT // -#include #include "ParserFlatbuffersFixture.hpp" #include "../TfLiteParser.hpp" #include #include -BOOST_AUTO_TEST_SUITE(TensorflowLiteParser) - +TEST_SUITE("TensorflowLiteParser") +{ struct SplitVFixture : public ParserFlatbuffersFixture { explicit SplitVFixture(const std::string& inputShape, @@ -126,7 +125,7 @@ struct SimpleSplitVAxisOneFixture : SplitVFixture {} }; -BOOST_FIXTURE_TEST_CASE(ParseAxisOneSplitVTwo, SimpleSplitVAxisOneFixture) +TEST_CASE_FIXTURE(SimpleSplitVAxisOneFixture, "ParseAxisOneSplitVTwo") { RunTest<4, armnn::DataType::Float32>( 0, @@ -148,7 +147,7 @@ struct SimpleSplitVAxisTwoFixture : SplitVFixture {} }; -BOOST_FIXTURE_TEST_CASE(ParseAxisTwoSplitVTwo, SimpleSplitVAxisTwoFixture) +TEST_CASE_FIXTURE(SimpleSplitVAxisTwoFixture, "ParseAxisTwoSplitVTwo") { RunTest<4, armnn::DataType::Float32>( 0, @@ -170,7 +169,7 @@ struct SimpleSplitVAxisThreeFixture : SplitVFixture {} }; -BOOST_FIXTURE_TEST_CASE(ParseAxisThreeSplitVTwo, SimpleSplitVAxisThreeFixture) +TEST_CASE_FIXTURE(SimpleSplitVAxisThreeFixture, "ParseAxisThreeSplitVTwo") { RunTest<4, armnn::DataType::Float32>( 0, @@ -192,7 +191,7 @@ struct SimpleSplitVAxisFourFixture : SplitVFixture {} }; -BOOST_FIXTURE_TEST_CASE(ParseAxisFourSplitVTwo, SimpleSplitVAxisFourFixture) +TEST_CASE_FIXTURE(SimpleSplitVAxisFourFixture, "ParseAxisFourSplitVTwo") { RunTest<4, armnn::DataType::Float32>( 0, @@ -206,4 +205,4 @@ BOOST_FIXTURE_TEST_CASE(ParseAxisFourSplitVTwo, SimpleSplitVAxisFourFixture) {"outputTensor2", { 4.0f, 8.0f, 12.0f, 16.0f, 20.0f, 24.0f, 28.0f, 32.0f } } } ); } -BOOST_AUTO_TEST_SUITE_END() +} -- cgit v1.2.1