aboutsummaryrefslogtreecommitdiff
path: root/test/FullyConnected.cpp
diff options
context:
space:
mode:
authorSadik Armagan <sadik.armagan@arm.com>2021-05-26 15:40:53 +0100
committerSadik Armagan <sadik.armagan@arm.com>2021-06-09 14:50:34 +0000
commit9150bff63a690caa743c471943afe509ebed1044 (patch)
treef98047d0a3a0e6cf06a4f34e0270a3cc7e3ee8bd /test/FullyConnected.cpp
parent07648f41d8b1fe9f532fa9c7e8e96a8e3651e59d (diff)
downloadandroid-nn-driver-9150bff63a690caa743c471943afe509ebed1044.tar.gz
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
Diffstat (limited to 'test/FullyConnected.cpp')
-rw-r--r--test/FullyConnected.cpp66
1 files changed, 33 insertions, 33 deletions
diff --git a/test/FullyConnected.cpp b/test/FullyConnected.cpp
index a68a5870..704de44a 100644
--- a/test/FullyConnected.cpp
+++ b/test/FullyConnected.cpp
@@ -6,12 +6,12 @@
#include "../1.0/HalPolicy.hpp"
-#include <boost/test/unit_test.hpp>
+#include <doctest/doctest.h>
#include <log/log.h>
-BOOST_AUTO_TEST_SUITE(FullyConnectedTests)
-
+TEST_SUITE("FullyConnectedTests")
+{
using namespace android::hardware;
using namespace driverTestHelpers;
using namespace armnn_driver;
@@ -19,7 +19,7 @@ using namespace armnn_driver;
using HalPolicy = hal_1_0::HalPolicy;
// Add our own test here since we fail the fc tests which Google supplies (because of non-const weights)
-BOOST_AUTO_TEST_CASE(FullyConnected)
+TEST_CASE("FullyConnected")
{
// this should ideally replicate fully_connected_float.model.cpp
// but that uses slightly weird dimensions which I don't think we need to support for now
@@ -83,10 +83,10 @@ BOOST_AUTO_TEST_CASE(FullyConnected)
}
// check the result
- BOOST_TEST(outdata[0] == 152);
+ CHECK(outdata[0] == 152);
}
-BOOST_AUTO_TEST_CASE(TestFullyConnected4dInput)
+TEST_CASE("TestFullyConnected4dInput")
{
auto driver = std::make_unique<ArmnnDriver>(DriverOptions(armnn::Compute::CpuRef));
@@ -165,17 +165,17 @@ BOOST_AUTO_TEST_CASE(TestFullyConnected4dInput)
}
// check the result
- BOOST_TEST(outdata[0] == 1);
- BOOST_TEST(outdata[1] == 2);
- BOOST_TEST(outdata[2] == 3);
- BOOST_TEST(outdata[3] == 4);
- BOOST_TEST(outdata[4] == 5);
- BOOST_TEST(outdata[5] == 6);
- BOOST_TEST(outdata[6] == 7);
- BOOST_TEST(outdata[7] == 8);
+ CHECK(outdata[0] == 1);
+ CHECK(outdata[1] == 2);
+ CHECK(outdata[2] == 3);
+ CHECK(outdata[3] == 4);
+ CHECK(outdata[4] == 5);
+ CHECK(outdata[5] == 6);
+ CHECK(outdata[6] == 7);
+ CHECK(outdata[7] == 8);
}
-BOOST_AUTO_TEST_CASE(TestFullyConnected4dInputReshape)
+TEST_CASE("TestFullyConnected4dInputReshape")
{
auto driver = std::make_unique<ArmnnDriver>(DriverOptions(armnn::Compute::CpuRef));
@@ -254,17 +254,17 @@ BOOST_AUTO_TEST_CASE(TestFullyConnected4dInputReshape)
}
// check the result
- BOOST_TEST(outdata[0] == 1);
- BOOST_TEST(outdata[1] == 2);
- BOOST_TEST(outdata[2] == 3);
- BOOST_TEST(outdata[3] == 4);
- BOOST_TEST(outdata[4] == 5);
- BOOST_TEST(outdata[5] == 6);
- BOOST_TEST(outdata[6] == 7);
- BOOST_TEST(outdata[7] == 8);
+ CHECK(outdata[0] == 1);
+ CHECK(outdata[1] == 2);
+ CHECK(outdata[2] == 3);
+ CHECK(outdata[3] == 4);
+ CHECK(outdata[4] == 5);
+ CHECK(outdata[5] == 6);
+ CHECK(outdata[6] == 7);
+ CHECK(outdata[7] == 8);
}
-BOOST_AUTO_TEST_CASE(TestFullyConnectedWeightsAsInput)
+TEST_CASE("TestFullyConnectedWeightsAsInput")
{
auto driver = std::make_unique<ArmnnDriver>(DriverOptions(armnn::Compute::CpuRef));
@@ -366,14 +366,14 @@ BOOST_AUTO_TEST_CASE(TestFullyConnectedWeightsAsInput)
}
// check the result
- BOOST_TEST(outdata[0] == 1);
- BOOST_TEST(outdata[1] == 2);
- BOOST_TEST(outdata[2] == 3);
- BOOST_TEST(outdata[3] == 4);
- BOOST_TEST(outdata[4] == 5);
- BOOST_TEST(outdata[5] == 6);
- BOOST_TEST(outdata[6] == 7);
- BOOST_TEST(outdata[7] == 8);
+ CHECK(outdata[0] == 1);
+ CHECK(outdata[1] == 2);
+ CHECK(outdata[2] == 3);
+ CHECK(outdata[3] == 4);
+ CHECK(outdata[4] == 5);
+ CHECK(outdata[5] == 6);
+ CHECK(outdata[6] == 7);
+ CHECK(outdata[7] == 8);
}
-BOOST_AUTO_TEST_SUITE_END()
+}