diff options
Diffstat (limited to 'src/armnnUtils/test/TensorUtilsTest.cpp')
-rw-r--r-- | src/armnnUtils/test/TensorUtilsTest.cpp | 112 |
1 files changed, 56 insertions, 56 deletions
diff --git a/src/armnnUtils/test/TensorUtilsTest.cpp b/src/armnnUtils/test/TensorUtilsTest.cpp index d24740b762..6d5f719eb1 100644 --- a/src/armnnUtils/test/TensorUtilsTest.cpp +++ b/src/armnnUtils/test/TensorUtilsTest.cpp @@ -7,131 +7,131 @@ #include <armnnUtils/TensorUtils.hpp> -#include <boost/test/unit_test.hpp> +#include <doctest/doctest.h> using namespace armnn; using namespace armnnUtils; -BOOST_AUTO_TEST_SUITE(TensorUtilsSuite) - -BOOST_AUTO_TEST_CASE(ExpandDimsAxis0Test) +TEST_SUITE("TensorUtilsSuite") +{ +TEST_CASE("ExpandDimsAxis0Test") { armnn::TensorShape inputShape({ 2, 3, 4 }); // Expand dimension 0 armnn::TensorShape outputShape = ExpandDims(inputShape, 0); - BOOST_TEST(outputShape.GetNumDimensions() == 4); - BOOST_TEST(outputShape[0] == 1); - BOOST_TEST(outputShape[1] == 2); - BOOST_TEST(outputShape[2] == 3); - BOOST_TEST(outputShape[3] == 4); + CHECK(outputShape.GetNumDimensions() == 4); + CHECK(outputShape[0] == 1); + CHECK(outputShape[1] == 2); + CHECK(outputShape[2] == 3); + CHECK(outputShape[3] == 4); } -BOOST_AUTO_TEST_CASE(ExpandDimsAxis1Test) +TEST_CASE("ExpandDimsAxis1Test") { armnn::TensorShape inputShape({ 2, 3, 4 }); // Expand dimension 1 armnn::TensorShape outputShape = ExpandDims(inputShape, 1); - BOOST_TEST(outputShape.GetNumDimensions() == 4); - BOOST_TEST(outputShape[0] == 2); - BOOST_TEST(outputShape[1] == 1); - BOOST_TEST(outputShape[2] == 3); - BOOST_TEST(outputShape[3] == 4); + CHECK(outputShape.GetNumDimensions() == 4); + CHECK(outputShape[0] == 2); + CHECK(outputShape[1] == 1); + CHECK(outputShape[2] == 3); + CHECK(outputShape[3] == 4); } -BOOST_AUTO_TEST_CASE(ExpandDimsAxis2Test) +TEST_CASE("ExpandDimsAxis2Test") { armnn::TensorShape inputShape({ 2, 3, 4 }); // Expand dimension 2 armnn::TensorShape outputShape = ExpandDims(inputShape, 2); - BOOST_TEST(outputShape.GetNumDimensions() == 4); - BOOST_TEST(outputShape[0] == 2); - BOOST_TEST(outputShape[1] == 3); - BOOST_TEST(outputShape[2] == 1); - BOOST_TEST(outputShape[3] == 4); + CHECK(outputShape.GetNumDimensions() == 4); + CHECK(outputShape[0] == 2); + CHECK(outputShape[1] == 3); + CHECK(outputShape[2] == 1); + CHECK(outputShape[3] == 4); } -BOOST_AUTO_TEST_CASE(ExpandDimsAxis3Test) +TEST_CASE("ExpandDimsAxis3Test") { armnn::TensorShape inputShape({ 2, 3, 4 }); // Expand dimension 3 armnn::TensorShape outputShape = ExpandDims(inputShape, 3); - BOOST_TEST(outputShape.GetNumDimensions() == 4); - BOOST_TEST(outputShape[0] == 2); - BOOST_TEST(outputShape[1] == 3); - BOOST_TEST(outputShape[2] == 4); - BOOST_TEST(outputShape[3] == 1); + CHECK(outputShape.GetNumDimensions() == 4); + CHECK(outputShape[0] == 2); + CHECK(outputShape[1] == 3); + CHECK(outputShape[2] == 4); + CHECK(outputShape[3] == 1); } -BOOST_AUTO_TEST_CASE(ExpandDimsNegativeAxis1Test) +TEST_CASE("ExpandDimsNegativeAxis1Test") { armnn::TensorShape inputShape({ 2, 3, 4 }); // Expand dimension -1 armnn::TensorShape outputShape = ExpandDims(inputShape, -1); - BOOST_TEST(outputShape.GetNumDimensions() == 4); - BOOST_TEST(outputShape[0] == 2); - BOOST_TEST(outputShape[1] == 3); - BOOST_TEST(outputShape[2] == 4); - BOOST_TEST(outputShape[3] == 1); + CHECK(outputShape.GetNumDimensions() == 4); + CHECK(outputShape[0] == 2); + CHECK(outputShape[1] == 3); + CHECK(outputShape[2] == 4); + CHECK(outputShape[3] == 1); } -BOOST_AUTO_TEST_CASE(ExpandDimsNegativeAxis2Test) +TEST_CASE("ExpandDimsNegativeAxis2Test") { armnn::TensorShape inputShape({ 2, 3, 4 }); // Expand dimension -2 armnn::TensorShape outputShape = ExpandDims(inputShape, -2); - BOOST_TEST(outputShape.GetNumDimensions() == 4); - BOOST_TEST(outputShape[0] == 2); - BOOST_TEST(outputShape[1] == 3); - BOOST_TEST(outputShape[2] == 1); - BOOST_TEST(outputShape[3] == 4); + CHECK(outputShape.GetNumDimensions() == 4); + CHECK(outputShape[0] == 2); + CHECK(outputShape[1] == 3); + CHECK(outputShape[2] == 1); + CHECK(outputShape[3] == 4); } -BOOST_AUTO_TEST_CASE(ExpandDimsNegativeAxis3Test) +TEST_CASE("ExpandDimsNegativeAxis3Test") { armnn::TensorShape inputShape({ 2, 3, 4 }); // Expand dimension -3 armnn::TensorShape outputShape = ExpandDims(inputShape, -3); - BOOST_TEST(outputShape.GetNumDimensions() == 4); - BOOST_TEST(outputShape[0] == 2); - BOOST_TEST(outputShape[1] == 1); - BOOST_TEST(outputShape[2] == 3); - BOOST_TEST(outputShape[3] == 4); + CHECK(outputShape.GetNumDimensions() == 4); + CHECK(outputShape[0] == 2); + CHECK(outputShape[1] == 1); + CHECK(outputShape[2] == 3); + CHECK(outputShape[3] == 4); } -BOOST_AUTO_TEST_CASE(ExpandDimsNegativeAxis4Test) +TEST_CASE("ExpandDimsNegativeAxis4Test") { armnn::TensorShape inputShape({ 2, 3, 4 }); // Expand dimension -4 armnn::TensorShape outputShape = ExpandDims(inputShape, -4); - BOOST_TEST(outputShape.GetNumDimensions() == 4); - BOOST_TEST(outputShape[0] == 1); - BOOST_TEST(outputShape[1] == 2); - BOOST_TEST(outputShape[2] == 3); - BOOST_TEST(outputShape[3] == 4); + CHECK(outputShape.GetNumDimensions() == 4); + CHECK(outputShape[0] == 1); + CHECK(outputShape[1] == 2); + CHECK(outputShape[2] == 3); + CHECK(outputShape[3] == 4); } -BOOST_AUTO_TEST_CASE(ExpandDimsInvalidAxisTest) +TEST_CASE("ExpandDimsInvalidAxisTest") { armnn::TensorShape inputShape({ 2, 3, 4 }); // Invalid expand dimension 4 - BOOST_CHECK_THROW(ExpandDims(inputShape, 4), armnn::InvalidArgumentException); + CHECK_THROWS_AS(ExpandDims(inputShape, 4), armnn::InvalidArgumentException); } -BOOST_AUTO_TEST_CASE(ExpandDimsInvalidNegativeAxisTest) +TEST_CASE("ExpandDimsInvalidNegativeAxisTest") { armnn::TensorShape inputShape({ 2, 3, 4 }); // Invalid expand dimension -5 - BOOST_CHECK_THROW(ExpandDims(inputShape, -5), armnn::InvalidArgumentException); + CHECK_THROWS_AS(ExpandDims(inputShape, -5), armnn::InvalidArgumentException); } -BOOST_AUTO_TEST_SUITE_END() +} |