diff options
Diffstat (limited to 'src/armnnTfLiteParser/test/Concatenation.cpp')
| -rw-r--r-- | src/armnnTfLiteParser/test/Concatenation.cpp | 23 |
1 files changed, 11 insertions, 12 deletions
diff --git a/src/armnnTfLiteParser/test/Concatenation.cpp b/src/armnnTfLiteParser/test/Concatenation.cpp index 8e31a3edb0..2407794f0d 100644 --- a/src/armnnTfLiteParser/test/Concatenation.cpp +++ b/src/armnnTfLiteParser/test/Concatenation.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_Concatenation") +{ struct ConcatenationFixture : public ParserFlatbuffersFixture { explicit ConcatenationFixture(const std::string & inputShape1, @@ -98,7 +97,7 @@ struct ConcatenationFixtureNegativeDim : ConcatenationFixture "-3" ) {} }; -BOOST_FIXTURE_TEST_CASE(ParseConcatenationNegativeDim, ConcatenationFixtureNegativeDim) +TEST_CASE_FIXTURE(ConcatenationFixtureNegativeDim, "ParseConcatenationNegativeDim") { RunTest<4, armnn::DataType::QAsymmU8>( 0, @@ -112,7 +111,7 @@ struct ConcatenationFixtureNCHW : ConcatenationFixture ConcatenationFixtureNCHW() : ConcatenationFixture("[ 1, 1, 2, 2 ]", "[ 1, 1, 2, 2 ]", "[ 1, 2, 2, 2 ]", "1" ) {} }; -BOOST_FIXTURE_TEST_CASE(ParseConcatenationNCHW, ConcatenationFixtureNCHW) +TEST_CASE_FIXTURE(ConcatenationFixtureNCHW, "ParseConcatenationNCHW") { RunTest<4, armnn::DataType::QAsymmU8>( 0, @@ -126,7 +125,7 @@ struct ConcatenationFixtureNHWC : ConcatenationFixture ConcatenationFixtureNHWC() : ConcatenationFixture("[ 1, 1, 2, 2 ]", "[ 1, 1, 2, 2 ]", "[ 1, 1, 2, 4 ]", "3" ) {} }; -BOOST_FIXTURE_TEST_CASE(ParseConcatenationNHWC, ConcatenationFixtureNHWC) +TEST_CASE_FIXTURE(ConcatenationFixtureNHWC, "ParseConcatenationNHWC") { RunTest<4, armnn::DataType::QAsymmU8>( 0, @@ -140,7 +139,7 @@ struct ConcatenationFixtureDim1 : ConcatenationFixture ConcatenationFixtureDim1() : ConcatenationFixture("[ 1, 2, 3, 4 ]", "[ 1, 2, 3, 4 ]", "[ 1, 4, 3, 4 ]", "1" ) {} }; -BOOST_FIXTURE_TEST_CASE(ParseConcatenationDim1, ConcatenationFixtureDim1) +TEST_CASE_FIXTURE(ConcatenationFixtureDim1, "ParseConcatenationDim1") { RunTest<4, armnn::DataType::QAsymmU8>( 0, @@ -159,7 +158,7 @@ struct ConcatenationFixtureDim3 : ConcatenationFixture ConcatenationFixtureDim3() : ConcatenationFixture("[ 1, 2, 3, 4 ]", "[ 1, 2, 3, 4 ]", "[ 1, 2, 3, 8 ]", "3" ) {} }; -BOOST_FIXTURE_TEST_CASE(ParseConcatenationDim3, ConcatenationFixtureDim3) +TEST_CASE_FIXTURE(ConcatenationFixtureDim3, "ParseConcatenationDim3") { RunTest<4, armnn::DataType::QAsymmU8>( 0, @@ -194,7 +193,7 @@ struct ConcatenationFixture3DDim0 : ConcatenationFixture ConcatenationFixture3DDim0() : ConcatenationFixture("[ 1, 2, 3]", "[ 2, 2, 3]", "[ 3, 2, 3]", "0" ) {} }; -BOOST_FIXTURE_TEST_CASE(ParseConcatenation3DDim0, ConcatenationFixture3DDim0) +TEST_CASE_FIXTURE(ConcatenationFixture3DDim0, "ParseConcatenation3DDim0") { RunTest<3, armnn::DataType::QAsymmU8>( 0, @@ -211,7 +210,7 @@ struct ConcatenationFixture3DDim1 : ConcatenationFixture ConcatenationFixture3DDim1() : ConcatenationFixture("[ 1, 2, 3]", "[ 1, 4, 3]", "[ 1, 6, 3]", "1" ) {} }; -BOOST_FIXTURE_TEST_CASE(ParseConcatenation3DDim1, ConcatenationFixture3DDim1) +TEST_CASE_FIXTURE(ConcatenationFixture3DDim1, "ParseConcatenation3DDim1") { RunTest<3, armnn::DataType::QAsymmU8>( 0, @@ -228,7 +227,7 @@ struct ConcatenationFixture3DDim2 : ConcatenationFixture ConcatenationFixture3DDim2() : ConcatenationFixture("[ 1, 2, 3]", "[ 1, 2, 6]", "[ 1, 2, 9]", "2" ) {} }; -BOOST_FIXTURE_TEST_CASE(ParseConcatenation3DDim2, ConcatenationFixture3DDim2) +TEST_CASE_FIXTURE(ConcatenationFixture3DDim2, "ParseConcatenation3DDim2") { RunTest<3, armnn::DataType::QAsymmU8>( 0, @@ -240,4 +239,4 @@ BOOST_FIXTURE_TEST_CASE(ParseConcatenation3DDim2, ConcatenationFixture3DDim2) 3, 4, 5, 12, 13, 14, 15, 16, 17 } } }); } -BOOST_AUTO_TEST_SUITE_END() +} |