diff options
Diffstat (limited to 'tests/use_case/vww/VisualWakeWordUCTests.cc')
-rw-r--r-- | tests/use_case/vww/VisualWakeWordUCTests.cc | 37 |
1 files changed, 19 insertions, 18 deletions
diff --git a/tests/use_case/vww/VisualWakeWordUCTests.cc b/tests/use_case/vww/VisualWakeWordUCTests.cc index 05a31a4..fe3782b 100644 --- a/tests/use_case/vww/VisualWakeWordUCTests.cc +++ b/tests/use_case/vww/VisualWakeWordUCTests.cc @@ -25,23 +25,24 @@ #include "UseCaseCommonUtils.hpp" namespace arm { - namespace app { - static uint8_t tensorArena[ACTIVATION_BUF_SZ] ACTIVATION_BUF_ATTRIBUTE; - } /* namespace app */ +namespace app { + static uint8_t tensorArena[ACTIVATION_BUF_SZ] ACTIVATION_BUF_ATTRIBUTE; + namespace vww { + extern uint8_t* GetModelPointer(); + extern size_t GetModelLen(); + } /* namespace vww */ +} /* namespace app */ } /* namespace arm */ -extern uint8_t* GetModelPointer(); -extern size_t GetModelLen(); - TEST_CASE("Model info") { arm::app::VisualWakeWordModel model; /* model wrapper object */ /* Load the model */ REQUIRE(model.Init(arm::app::tensorArena, - sizeof(arm::app::tensorArena), - GetModelPointer(), - GetModelLen())); + sizeof(arm::app::tensorArena), + arm::app::vww::GetModelPointer(), + arm::app::vww::GetModelLen())); /* Instantiate application context */ arm::app::ApplicationContext caseContext; @@ -59,9 +60,9 @@ TEST_CASE("Inference by index") /* Load the model */ REQUIRE(model.Init(arm::app::tensorArena, - sizeof(arm::app::tensorArena), - GetModelPointer(), - GetModelLen())); + sizeof(arm::app::tensorArena), + arm::app::vww::GetModelPointer(), + arm::app::vww::GetModelLen())); /* Instantiate application context */ arm::app::ApplicationContext caseContext; @@ -92,9 +93,9 @@ TEST_CASE("Inference run all images") /* Load the model */ REQUIRE(model.Init(arm::app::tensorArena, - sizeof(arm::app::tensorArena), - GetModelPointer(), - GetModelLen())); + sizeof(arm::app::tensorArena), + arm::app::vww::GetModelPointer(), + arm::app::vww::GetModelLen())); /* Instantiate application context */ arm::app::ApplicationContext caseContext; @@ -121,9 +122,9 @@ TEST_CASE("List all images") /* Load the model */ REQUIRE(model.Init(arm::app::tensorArena, - sizeof(arm::app::tensorArena), - GetModelPointer(), - GetModelLen())); + sizeof(arm::app::tensorArena), + arm::app::vww::GetModelPointer(), + arm::app::vww::GetModelLen())); /* Instantiate application context */ arm::app::ApplicationContext caseContext; |