Refactor test execution and data handling in func_tests

Refactored the `BaseRunFuncTests` class to simplify and improve test execution and data handling. Replaced `GetTestInput` and `GetTaskPtr` with `GetTestInputData` and `CheckTestOutputData` for cleaner abstraction. Updated error handling and validation logic for better test reliability and maintainability.

Author: allnes

modules/core/util/include/test_util.hpp

@@ -38,10 +38,10 @@ class BaseRunPerfTests : public ::testing::TestWithParam<PerfTestParam<InType, O
   virtual bool CheckTestOutputData(OutType& output_data) = 0;
   virtual InType GetTestInputData() = 0;
 
-  void ExecuteTest(const PerfTestParam<InType, OutType>& perfTestParam) {
-    auto task_getter = std::get<ppc::util::FuncTestParamIndex::kTaskGetter>(perfTestParam);
-    auto test_name = std::get<ppc::util::FuncTestParamIndex::kNameTest>(perfTestParam);
-    auto mode = std::get<ppc::util::FuncTestParamIndex::kTestParams>(perfTestParam);
+  void ExecuteTest(const PerfTestParam<InType, OutType>& perf_test_param) {
+    auto task_getter = std::get<ppc::util::FuncTestParamIndex::kTaskGetter>(perf_test_param);
+    auto test_name = std::get<ppc::util::FuncTestParamIndex::kNameTest>(perf_test_param);
+    auto mode = std::get<ppc::util::FuncTestParamIndex::kTestParams>(perf_test_param);
 
     task_ = task_getter(GetTestInputData());
     ppc::core::Perf perf(task_);
@@ -80,22 +80,21 @@ class BaseRunPerfTests : public ::testing::TestWithParam<PerfTestParam<InType, O
 template <typename InType, typename OutType, typename TestType = void>
 class BaseRunFuncTests : public ::testing::TestWithParam<ppc::util::FuncTestParam<InType, OutType, TestType>> {
  public:
-  virtual void CheckTestOutputData() = 0;
-  ppc::core::TaskPtr<InType, OutType>& GetTaskPtr() { return task; }
-  InType& GetTestInput() { return test_input_; }
+  virtual bool CheckTestOutputData(OutType& output_data) = 0;
+  virtual InType GetTestInputData() = 0;
 
  protected:
-  void ExecuteTest() {
-    ASSERT_TRUE(task->Validation());
-    ASSERT_TRUE(task->PreProcessing());
-    ASSERT_TRUE(task->Run());
-    ASSERT_TRUE(task->PostProcessing());
-    CheckTestOutputData();
+  void ExecuteTest(ppc::util::FuncTestParam<InType, OutType, TestType> test_param) {
+    task_ = std::get<ppc::util::FuncTestParamIndex::kTaskGetter>(test_param)(GetTestInputData());
+    ASSERT_TRUE(task_->Validation());
+    ASSERT_TRUE(task_->PreProcessing());
+    ASSERT_TRUE(task_->Run());
+    ASSERT_TRUE(task_->PostProcessing());
+    ASSERT_TRUE(CheckTestOutputData(task_->GetOutput()));
   }
 
  private:
-  ppc::core::TaskPtr<InType, OutType> task;
-  InType test_input_;
+  ppc::core::TaskPtr<InType, OutType> task_;
 };
 
 }  // namespace ppc::util

tasks/example/tests/func_tests/main.cpp

@@ -20,6 +20,7 @@ using OutType = std::vector<int>;
 using TestType = int;
 
 class NesterovARunFuncTests : public ppc::util::BaseRunFuncTests<InType, OutType, TestType> {
+  InType input_data_;
  public:
   static std::string CustomFuncTestName(
       const ::testing::TestParamInfo<ppc::util::FuncTestParam<InType, OutType, TestType>> &info) {
@@ -30,35 +31,36 @@ class NesterovARunFuncTests : public ppc::util::BaseRunFuncTests<InType, OutType
 
  protected:
   void SetUp() override {
-    PrepareData();
-    GetTaskPtr() = std::get<ppc::util::FuncTestParamIndex::kTaskGetter>(GetParam())(GetTestInput());
-  }
-
-  void PrepareData() {
     int width = -1;
     int height = -1;
     int channels = -1;
     // Read image
     {
       std::string abs_path = ppc::util::GetAbsolutePath("example/tests/data/pic_all.jpg");
       auto *data = stbi_load(abs_path.c_str(), &width, &height, &channels, 0);
-      ASSERT_TRUE(data != nullptr) << "Failed to load image: " << stbi_failure_reason();
+      if (data == nullptr) {
+        throw std::runtime_error("Failed to load image: " + std::string(stbi_failure_reason()));
+      }
       auto img = std::vector<uint8_t>(data, data + (static_cast<ptrdiff_t>(width * height * channels)));
       stbi_image_free(data);
-      ASSERT_EQ(width, height);
+      if(width != height) {
+        throw std::runtime_error("width != height: ");
+      }
     }
 
     const int k_count = (width + height) / std::get<ppc::util::FuncTestParamIndex::kTestParams>(GetParam());
-    GetTestInput() = InType(static_cast<std::vector<int>::size_type>(k_count * k_count), 0);
+    input_data_ = InType(static_cast<std::vector<int>::size_type>(k_count * k_count), 0);
     for (int i = 0; i < k_count; i++) {
-      GetTestInput()[(i * k_count) + i] = 1;
+      input_data_[(i * k_count) + i] = 1;
     }
   }
 
-  void CheckTestOutputData() final { EXPECT_EQ(GetTestInput(), GetTaskPtr()->GetOutput()); }
+  bool CheckTestOutputData(OutType& output_data) final { return input_data_ == output_data; }
+
+  InType GetTestInputData() final { return input_data_; }
 };
 
-TEST_P(NesterovARunFuncTests, MatmulFromPic) { ExecuteTest(); }
+TEST_P(NesterovARunFuncTests, MatmulFromPic) { ExecuteTest(GetParam()); }
 
 #define ADD_FUNC_TASK(TASK)                                                                 \
   std::make_tuple(ppc::core::TaskGetter<TASK, InType>, ppc::util::GetNamespace<TASK>(), 5), \