1 files changed, 24 insertions, 3 deletions
diff --git a/tests/use_case/asr/Wav2LetterPostprocessingTest.cc b/tests/use_case/asr/Wav2LetterPostprocessingTest.cc
index 11c4919..9c3d658 100644
--- a/tests/use_case/asr/Wav2LetterPostprocessingTest.cc
+++ b/tests/use_case/asr/Wav2LetterPostprocessingTest.cc
@@ -17,11 +17,23 @@
 #include "Wav2LetterPostprocess.hpp"
 #include "Wav2LetterModel.hpp"
 #include "ClassificationResult.hpp"
+#include "BufAttributes.hpp"
 
 #include <algorithm>
 #include <catch.hpp>
 #include <limits>
 
+namespace arm {
+namespace app {
+    static uint8_t tensorArena[ACTIVATION_BUF_SZ] ACTIVATION_BUF_ATTRIBUTE;
+
+    namespace asr {
+        extern uint8_t* GetModelPointer();
+        extern size_t GetModelLen();
+    } /* namespace asr */
+} /* namespace app */
+} /* namespace arm */
+
 template <typename T>
 static TfLiteTensor GetTestTensor(
                         std::vector<int>&      shape,
@@ -51,7 +63,10 @@ TEST_CASE("Checking return value")
         const uint32_t outputCtxLen = 5;
         arm::app::AsrClassifier classifier;
         arm::app::Wav2LetterModel model;
-        model.Init();
+        model.Init(arm::app::tensorArena,
+                    sizeof(arm::app::tensorArena),
+                    arm::app::asr::GetModelPointer(),
+                    arm::app::asr::GetModelLen());
         std::vector<std::string> dummyLabels = {"a", "b", "$"};
         const uint32_t blankTokenIdx = 2;
         std::vector<arm::app::ClassificationResult> dummyResult;
@@ -71,7 +86,10 @@ TEST_CASE("Checking return value")
         const uint32_t outputCtxLen = 5;
         arm::app::AsrClassifier classifier;
         arm::app::Wav2LetterModel model;
-        model.Init();
+        model.Init(arm::app::tensorArena,
+                    sizeof(arm::app::tensorArena),
+                    arm::app::asr::GetModelPointer(),
+                    arm::app::asr::GetModelLen());
         std::vector<std::string> dummyLabels = {"a", "b", "$"};
         const uint32_t blankTokenIdx = 2;
         std::vector<arm::app::ClassificationResult> dummyResult;
@@ -102,7 +120,10 @@ TEST_CASE("Postprocessing - erasing required elements")
     std::vector<int> tensorShape = {1, 1, nRows, nCols};
     arm::app::AsrClassifier classifier;
     arm::app::Wav2LetterModel model;
-    model.Init();
+    model.Init(arm::app::tensorArena,
+                    sizeof(arm::app::tensorArena),
+                    arm::app::asr::GetModelPointer(),
+                    arm::app::asr::GetModelLen());
     std::vector<std::string> dummyLabels = {"a", "b", "$"};
     std::vector<arm::app::ClassificationResult> dummyResult;