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()
+}