diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2021-05-26 15:40:53 +0100 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2021-06-09 14:50:34 +0000 |
commit | 9150bff63a690caa743c471943afe509ebed1044 (patch) | |
tree | f98047d0a3a0e6cf06a4f34e0270a3cc7e3ee8bd /test/DriverTestHelpers.cpp | |
parent | 07648f41d8b1fe9f532fa9c7e8e96a8e3651e59d (diff) | |
download | android-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/DriverTestHelpers.cpp')
-rw-r--r-- | test/DriverTestHelpers.cpp | 26 |
1 files changed, 14 insertions, 12 deletions
diff --git a/test/DriverTestHelpers.cpp b/test/DriverTestHelpers.cpp index 8e8d7bef..44e6e725 100644 --- a/test/DriverTestHelpers.cpp +++ b/test/DriverTestHelpers.cpp @@ -4,7 +4,8 @@ // #include "DriverTestHelpers.hpp" #include <log/log.h> -#include <boost/test/unit_test.hpp> + +#include <doctest/doctest.h> namespace android { @@ -139,10 +140,10 @@ android::sp<V1_0::IPreparedModel> PrepareModelWithStatus(const V1_0::Model& mode driver.prepareModel(model, cb); prepareStatus = cb->GetErrorStatus(); - BOOST_TEST(prepareStatus == expectedStatus); + CHECK((int)prepareStatus == (int)expectedStatus); if (expectedStatus == V1_0::ErrorStatus::NONE) { - BOOST_TEST((cb->GetPreparedModel() != nullptr)); + CHECK((cb->GetPreparedModel() != nullptr)); } return cb->GetPreparedModel(); } @@ -158,10 +159,10 @@ android::sp<V1_0::IPreparedModel> PrepareModelWithStatus(const V1_1::Model& mode driver.prepareModel_1_1(model, V1_1::ExecutionPreference::LOW_POWER, cb); prepareStatus = cb->GetErrorStatus(); - BOOST_TEST(prepareStatus == expectedStatus); + CHECK((int)prepareStatus == (int)expectedStatus); if (expectedStatus == V1_0::ErrorStatus::NONE) { - BOOST_TEST((cb->GetPreparedModel() != nullptr)); + CHECK((cb->GetPreparedModel() != nullptr)); } return cb->GetPreparedModel(); } @@ -184,10 +185,10 @@ android::sp<V1_2::IPreparedModel> PrepareModelWithStatus_1_2(const armnn_driver: driver.prepareModel_1_2(model, V1_1::ExecutionPreference::LOW_POWER, emptyHandle1, emptyHandle2, emptyToken, cb); prepareStatus = cb->GetErrorStatus(); - BOOST_TEST(prepareStatus == expectedStatus); + CHECK((int)prepareStatus == (int)expectedStatus); if (expectedStatus == V1_0::ErrorStatus::NONE) { - BOOST_TEST((cb->GetPreparedModel_1_2() != nullptr)); + CHECK((cb->GetPreparedModel_1_2() != nullptr)); } return cb->GetPreparedModel_1_2(); } @@ -219,7 +220,7 @@ android::sp<V1_3::IPreparedModel> PrepareModelWithStatus_1_3(const armnn_driver: prepareStatus = cb->Get_1_3_ErrorStatus(); if (prepareStatus == V1_3::ErrorStatus::NONE) { - BOOST_TEST((cb->GetPreparedModel_1_3() != nullptr)); + CHECK((cb->GetPreparedModel_1_3() != nullptr)); } return cb->GetPreparedModel_1_3(); } @@ -230,10 +231,10 @@ V1_0::ErrorStatus Execute(android::sp<V1_0::IPreparedModel> preparedModel, const V1_0::Request& request, V1_0::ErrorStatus expectedStatus) { - BOOST_TEST(preparedModel.get() != nullptr); + CHECK(preparedModel.get() != nullptr); android::sp<ExecutionCallback> cb(new ExecutionCallback()); V1_0::ErrorStatus execStatus = preparedModel->execute(request, cb); - BOOST_TEST(execStatus == expectedStatus); + CHECK((int)execStatus == (int)expectedStatus); ALOGI("Execute: waiting for callback to be invoked"); cb->wait(); return execStatus; @@ -242,9 +243,10 @@ V1_0::ErrorStatus Execute(android::sp<V1_0::IPreparedModel> preparedModel, android::sp<ExecutionCallback> ExecuteNoWait(android::sp<V1_0::IPreparedModel> preparedModel, const V1_0::Request& request) { - BOOST_TEST(preparedModel.get() != nullptr); + CHECK(preparedModel.get() != nullptr); android::sp<ExecutionCallback> cb(new ExecutionCallback()); - BOOST_TEST(preparedModel->execute(request, cb) == V1_0::ErrorStatus::NONE); + V1_0::ErrorStatus execStatus = preparedModel->execute(request, cb); + CHECK((int)execStatus == (int)V1_0::ErrorStatus::NONE); ALOGI("ExecuteNoWait: returning callback object"); return cb; } |