refactor func tests in tasks

Author: allnes

tasks/all/example/func_tests/func_all.cpp

@@ -0,0 +1,52 @@
+#include <gtest/gtest.h>
+
+#include <cstddef>
+#include <cstdint>
+#include <memory>
+#include <stb_library.hpp>
+#include <string>
+#include <vector>
+
+#include "all/example/include/ops_all.hpp"
+#include "core/task/include/task.hpp"
+#include "core/util/include/util.hpp"
+
+class NesterovATestTaskAll : public ::testing::TestWithParam<int> {
+ protected:
+  void SetUp() override {
+    width = height = channels = -1;
+    std::string abs_path = ppc::util::GetAbsolutePath("all/example/data/pic_all.jpg");
+    data = stbi_load(abs_path.c_str(), &width, &height, &channels, 0);
+    ASSERT_TRUE(data != nullptr) << "Failed to load image: " << stbi_failure_reason();
+    img = std::vector<uint8_t>(data, data + (width * height * channels));
+    stbi_image_free(data);
+
+    ASSERT_EQ(width, height);
+  }
+
+  int width = -1, height = -1, channels = -1;
+  unsigned char* data = nullptr;
+  std::vector<uint8_t> img;
+};
+
+TEST_P(NesterovATestTaskAll, MatmulFromPic) {
+  int divider = GetParam();
+  const int k_count = (width + height) / divider;
+
+  std::vector<int> in(k_count * k_count, 0);
+  for (int i = 0; i < k_count; i++) {
+    in[(i * k_count) + i] = 1;
+  }
+
+  nesterov_a_test_task_all::TestTaskALL test_task_all(in);
+  ASSERT_TRUE(test_task_all.Validation());
+  test_task_all.PreProcessing();
+  test_task_all.Run();
+  test_task_all.PostProcessing();
+  EXPECT_EQ(in, test_task_all.Get());
+}
+
+INSTANTIATE_TEST_SUITE_P(
+    PicMatrixTests,
+    NesterovATestTaskAll,
+    ::testing::Values(5, 10));

tasks/all/example/func_tests/main.cpp

@@ -11,49 +11,42 @@
 #include "core/util/include/util.hpp"
 #include "mpi/example/include/ops_mpi.hpp"
 
-TEST(nesterov_a_test_task_mpi, test_matmul_50) {
-  constexpr size_t kCount = 50;
-
-  // Create data
-  std::vector<int> in(kCount * kCount, 0);
-
-  for (size_t i = 0; i < kCount; i++) {
-    in[(i * kCount) + i] = 1;
+class NesterovATestTaskMPI : public ::testing::TestWithParam<double> {
+ protected:
+  void SetUp() override {
+    std::ifstream test_file(ppc::util::GetAbsolutePath("mpi/example/data/test.txt"));
+    ASSERT_TRUE(test_file.is_open()) << "Failed to open input file";
+    std::string line;
+    std::getline(test_file, line);
+    test_file.close();
+    base_count = std::stoi(line);
   }
 
-  // Create Task
-  nesterov_a_test_task_mpi::TestTaskMPI test_task_mpi(in);
-  ASSERT_EQ(test_task_mpi.Validation(), true);
-  test_task_mpi.PreProcessing();
-  test_task_mpi.Run();
-  test_task_mpi.PostProcessing();
-
-  EXPECT_EQ(in, test_task_mpi.Get());
-}
-
-TEST(nesterov_a_test_task_mpi, test_matmul_100_from_file) {
-  std::string line;
-  std::ifstream test_file(ppc::util::GetAbsolutePath("mpi/example/data/test.txt"));
-  if (test_file.is_open()) {
-    getline(test_file, line);
+  [[nodiscard]] size_t GetCount() const {
+    return static_cast<size_t>(base_count * GetParam());
   }
-  test_file.close();
 
-  const size_t count = std::stoi(line);
+  int base_count = 0;
+};
 
-  // Create data
-  std::vector<int> in(count * count, 0);
+TEST_P(NesterovATestTaskMPI, MatmulFromFile) {
+  const size_t count = GetCount();
 
-  for (size_t i = 0; i < count; i++) {
+  std::vector<int> in(count * count, 0);
+  for (size_t i = 0; i < count; ++i) {
     in[(i * count) + i] = 1;
   }
 
-  // Create Task
   nesterov_a_test_task_mpi::TestTaskMPI test_task_mpi(in);
-  ASSERT_EQ(test_task_mpi.Validation(), true);
+  ASSERT_TRUE(test_task_mpi.Validation());
   test_task_mpi.PreProcessing();
   test_task_mpi.Run();
   test_task_mpi.PostProcessing();
 
   EXPECT_EQ(in, test_task_mpi.Get());
 }
+
+INSTANTIATE_TEST_SUITE_P(
+    FileMatrixTestsMPI,
+    NesterovATestTaskMPI,
+    ::testing::Values(0.5, 1.0));

tasks/mpi/example/func_tests/main.cpp

@@ -11,47 +11,42 @@
 #include "core/util/include/util.hpp"
 #include "omp/example/include/ops_omp.hpp"
 
-TEST(nesterov_a_test_task_omp, test_matmul_50) {
-  constexpr size_t kCount = 50;
-
-  // Create data
-  std::vector<int> in(kCount * kCount, 0);
-
-  for (size_t i = 0; i < kCount; i++) {
-    in[(i * kCount) + i] = 1;
+class NesterovATestTaskOMP : public ::testing::TestWithParam<double> {
+ protected:
+  void SetUp() override {
+    std::ifstream test_file(ppc::util::GetAbsolutePath("omp/example/data/test.txt"));
+    ASSERT_TRUE(test_file.is_open()) << "Failed to open input file";
+    std::string line;
+    std::getline(test_file, line);
+    test_file.close();
+    base_count = std::stoi(line);
   }
 
-  // Create Task
-  nesterov_a_test_task_omp::TestTaskOpenMP test_task_omp(in);
-  ASSERT_EQ(test_task_omp.Validation(), true);
-  test_task_omp.PreProcessing();
-  test_task_omp.Run();
-  test_task_omp.PostProcessing();
-  EXPECT_EQ(in, test_task_omp.Get());
-}
-
-TEST(nesterov_a_test_task_omp, test_matmul_100_from_file) {
-  std::string line;
-  std::ifstream test_file(ppc::util::GetAbsolutePath("omp/example/data/test.txt"));
-  if (test_file.is_open()) {
-    getline(test_file, line);
+  [[nodiscard]] size_t GetCount() const {
+    return static_cast<size_t>(base_count * GetParam());
   }
-  test_file.close();
 
-  const size_t count = std::stoi(line);
+  int base_count = 0;
+};
 
-  // Create data
-  std::vector<int> in(count * count, 0);
+TEST_P(NesterovATestTaskOMP, MatmulFromFile) {
+  const size_t count = GetCount();
 
-  for (size_t i = 0; i < count; i++) {
+  std::vector<int> in(count * count, 0);
+  for (size_t i = 0; i < count; ++i) {
     in[(i * count) + i] = 1;
   }
 
-  // Create Task
   nesterov_a_test_task_omp::TestTaskOpenMP test_task_omp(in);
-  ASSERT_EQ(test_task_omp.Validation(), true);
+  ASSERT_TRUE(test_task_omp.Validation());
   test_task_omp.PreProcessing();
   test_task_omp.Run();
   test_task_omp.PostProcessing();
+
   EXPECT_EQ(in, test_task_omp.Get());
 }
+
+INSTANTIATE_TEST_SUITE_P(
+    FileMatrixTestsOMP,
+    NesterovATestTaskOMP,
+    ::testing::Values(0.5, 1.0));

tasks/omp/example/func_tests/main.cpp

@@ -11,47 +11,41 @@
 #include "core/util/include/util.hpp"
 #include "seq/example/include/ops_seq.hpp"
 
-TEST(nesterov_a_test_task_seq, test_matmul_50) {
-  constexpr size_t kCount = 50;
-
-  // Create data
-  std::vector<int> in(kCount * kCount, 0);
-
-  for (size_t i = 0; i < kCount; i++) {
-    in[(i * kCount) + i] = 1;
+class NesterovATestTaskSeq : public ::testing::TestWithParam<double> {
+ protected:
+  void SetUp() override {
+    std::ifstream test_file(ppc::util::GetAbsolutePath("seq/example/data/test.txt"));
+    ASSERT_TRUE(test_file.is_open()) << "Failed to open input file";
+    std::string line;
+    std::getline(test_file, line);
+    test_file.close();
+    base_count = std::stoi(line);
   }
 
-  // Create Task
-  nesterov_a_test_task_seq::TestTaskSequential test_task_sequential(in);
-  ASSERT_EQ(test_task_sequential.Validation(), true);
-  test_task_sequential.PreProcessing();
-  test_task_sequential.Run();
-  test_task_sequential.PostProcessing();
-  EXPECT_EQ(in, test_task_sequential.Get());
-}
-
-TEST(nesterov_a_test_task_seq, test_matmul_100_from_file) {
-  std::string line;
-  std::ifstream test_file(ppc::util::GetAbsolutePath("seq/example/data/test.txt"));
-  if (test_file.is_open()) {
-    getline(test_file, line);
+  [[nodiscard]] size_t GetCount() const {
+    return static_cast<size_t>(base_count * GetParam());
   }
-  test_file.close();
 
-  const size_t count = std::stoi(line);
+  int base_count = 0;
+};
 
-  // Create data
-  std::vector<int> in(count * count, 0);
+TEST_P(NesterovATestTaskSeq, MatmulFromFile) {
+  const size_t count = GetCount();
 
+  std::vector<int> in(count * count, 0);
   for (size_t i = 0; i < count; i++) {
     in[(i * count) + i] = 1;
   }
 
-  // Create Task
   nesterov_a_test_task_seq::TestTaskSequential test_task_sequential(in);
-  ASSERT_EQ(test_task_sequential.Validation(), true);
+  ASSERT_TRUE(test_task_sequential.Validation());
   test_task_sequential.PreProcessing();
   test_task_sequential.Run();
   test_task_sequential.PostProcessing();
   EXPECT_EQ(in, test_task_sequential.Get());
 }
+
+INSTANTIATE_TEST_SUITE_P(
+    FileMatrixTests,
+    NesterovATestTaskSeq,
+    ::testing::Values(0.5, 1.0));