diff options
author | Mike Kelly <mike.kelly@arm.com> | 2021-10-14 12:35:58 +0100 |
---|---|---|
committer | Mike Kelly <mike.kelly@arm.com> | 2021-10-14 12:35:58 +0100 |
commit | e2d611e4502fb5dce8b8a398ccfc8d7ef29da96b (patch) | |
tree | 79d9178420924d57ed0ab23db46ff57159279886 /test/FullyConnected.cpp | |
parent | 4bd8f7c96e01e081276c376fe5c34018e7b70d17 (diff) | |
download | android-nn-driver-e2d611e4502fb5dce8b8a398ccfc8d7ef29da96b.tar.gz |
IVGCVSW-6428 Remove asserts
* Changed asserts to check for errors and return appropriate values or
throw exceptions
* Changed unit tests to use Doctest's long macro names as the short
macro names clashed with Android's Logging macros
* Removed unused #includes
* Clarified ambiguous #includes
Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Change-Id: Ice92a37590df727fd581d3be5ff2716665f26a13
Diffstat (limited to 'test/FullyConnected.cpp')
-rw-r--r-- | test/FullyConnected.cpp | 67 |
1 files changed, 32 insertions, 35 deletions
diff --git a/test/FullyConnected.cpp b/test/FullyConnected.cpp index 704de44a..4717357b 100644 --- a/test/FullyConnected.cpp +++ b/test/FullyConnected.cpp @@ -1,16 +1,13 @@ // -// Copyright © 2017 Arm Ltd. All rights reserved. +// Copyright © 2017 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // -#include "DriverTestHelpers.hpp" - -#include "../1.0/HalPolicy.hpp" -#include <doctest/doctest.h> +#include "DriverTestHelpers.hpp" #include <log/log.h> -TEST_SUITE("FullyConnectedTests") +DOCTEST_TEST_SUITE("FullyConnectedTests") { using namespace android::hardware; using namespace driverTestHelpers; @@ -19,7 +16,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) -TEST_CASE("FullyConnected") +DOCTEST_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 +80,10 @@ TEST_CASE("FullyConnected") } // check the result - CHECK(outdata[0] == 152); + DOCTEST_CHECK(outdata[0] == 152); } -TEST_CASE("TestFullyConnected4dInput") +DOCTEST_TEST_CASE("TestFullyConnected4dInput") { auto driver = std::make_unique<ArmnnDriver>(DriverOptions(armnn::Compute::CpuRef)); @@ -165,17 +162,17 @@ TEST_CASE("TestFullyConnected4dInput") } // check the result - 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); + DOCTEST_CHECK(outdata[0] == 1); + DOCTEST_CHECK(outdata[1] == 2); + DOCTEST_CHECK(outdata[2] == 3); + DOCTEST_CHECK(outdata[3] == 4); + DOCTEST_CHECK(outdata[4] == 5); + DOCTEST_CHECK(outdata[5] == 6); + DOCTEST_CHECK(outdata[6] == 7); + DOCTEST_CHECK(outdata[7] == 8); } -TEST_CASE("TestFullyConnected4dInputReshape") +DOCTEST_TEST_CASE("TestFullyConnected4dInputReshape") { auto driver = std::make_unique<ArmnnDriver>(DriverOptions(armnn::Compute::CpuRef)); @@ -254,17 +251,17 @@ TEST_CASE("TestFullyConnected4dInputReshape") } // check the result - 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); + DOCTEST_CHECK(outdata[0] == 1); + DOCTEST_CHECK(outdata[1] == 2); + DOCTEST_CHECK(outdata[2] == 3); + DOCTEST_CHECK(outdata[3] == 4); + DOCTEST_CHECK(outdata[4] == 5); + DOCTEST_CHECK(outdata[5] == 6); + DOCTEST_CHECK(outdata[6] == 7); + DOCTEST_CHECK(outdata[7] == 8); } -TEST_CASE("TestFullyConnectedWeightsAsInput") +DOCTEST_TEST_CASE("TestFullyConnectedWeightsAsInput") { auto driver = std::make_unique<ArmnnDriver>(DriverOptions(armnn::Compute::CpuRef)); @@ -366,14 +363,14 @@ TEST_CASE("TestFullyConnectedWeightsAsInput") } // check the result - 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); + DOCTEST_CHECK(outdata[0] == 1); + DOCTEST_CHECK(outdata[1] == 2); + DOCTEST_CHECK(outdata[2] == 3); + DOCTEST_CHECK(outdata[3] == 4); + DOCTEST_CHECK(outdata[4] == 5); + DOCTEST_CHECK(outdata[5] == 6); + DOCTEST_CHECK(outdata[6] == 7); + DOCTEST_CHECK(outdata[7] == 8); } } |