commit 9150bff63a690caa743c471943afe509ebed1044
author Sadik Armagan <sadik.armagan@arm.com> Wed May 26 15:40:53 2021 +0100
committer Sadik Armagan <sadik.armagan@arm.com> Wed Jun 09 14:50:34 2021 +0000
tree f98047d0a3a0e6cf06a4f34e0270a3cc7e3ee8bd
parent 07648f41d8b1fe9f532fa9c7e8e96a8e3651e59d

IVGCVSW-4618 'Transition Units Test Suites'

* Used doctest in android-nn-driver unit tests.

Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I9b5d4dfd77d53c7ebee7f8c43628a1d6ff74d1a3

test/Dilation.hpp

diff --git a/test/Dilation.hpp b/test/Dilation.hpp
index d0189c9..a05dba4 100644
--- a/test/Dilation.hpp
+++ b/test/Dilation.hpp
@@ -10,14 +10,11 @@
 #include <armnn/LayerVisitorBase.hpp>
 #include <armnn/utility/IgnoreUnused.hpp>
 
-#include <boost/test/unit_test.hpp>
+#include <doctest/doctest.h>
 
 #include <numeric>
 
-BOOST_AUTO_TEST_SUITE(DilationTests)
-
 using namespace armnn;
-using namespace boost;
 using namespace driverTestHelpers;
 
 struct DilationTestOptions
@@ -82,8 +79,8 @@
     template<typename ConvolutionDescriptor>
     void CheckDilationParams(const ConvolutionDescriptor& descriptor)
     {
-        BOOST_CHECK_EQUAL(descriptor.m_DilationX, m_ExpectedDilationX);
-        BOOST_CHECK_EQUAL(descriptor.m_DilationY, m_ExpectedDilationY);
+        CHECK_EQ(descriptor.m_DilationX, m_ExpectedDilationX);
+        CHECK_EQ(descriptor.m_DilationY, m_ExpectedDilationY);
     }
 };
 
@@ -169,11 +166,9 @@
     data.m_OutputSlotForOperand = std::vector<IOutputSlot*>(model.operands.size(), nullptr);
 
     bool ok = HalPolicy::ConvertOperation(model.operations[0], model, data);
-    BOOST_CHECK(ok);
+    CHECK(ok);
 
     // check if dilation params are as expected
     DilationTestVisitor visitor = options.m_HasDilation ? DilationTestVisitor(2, 2) : DilationTestVisitor();
     data.m_Network->Accept(visitor);
 }
-
-BOOST_AUTO_TEST_SUITE_END()