[Unit tests] Added more unit tests for all the filters

spirosmaggioros · Dec 26, 2024 · 3d5701d · 3d5701d
1 parent 8d18ae8
commit 3d5701d
Show file tree

Hide file tree

Showing 3 changed files with 48 additions and 0 deletions.
diff --git a/tests/machine_learning/image/average_filter.cc b/tests/machine_learning/image/average_filter.cc
@@ -0,0 +1,16 @@
+#include "../../../src/machine_learning/image/filters/average_filter.h"
+#include "../../../third_party/catch.hpp"
+
+using namespace avg_filter;
+
+TEST_CASE("Testing average filter application") {
+    std::vector<std::vector<int32_t> > image(50, std::vector<int32_t>(50, 42));
+
+    auto resulted = apply_avg_filter(image);
+    REQUIRE(resulted.size() != 0);
+}
+
+TEST_CASE("Testing average filter application with empty image") {
+    std::vector<std::vector<int32_t> > image;
+    CHECK_THROWS(apply_avg_filter(image));
+}
diff --git a/tests/machine_learning/image/gaussian_filter.cc b/tests/machine_learning/image/gaussian_filter.cc
@@ -0,0 +1,16 @@
+#include "../../../src/machine_learning/image/filters/gaussian_filter.h"
+#include "../../../third_party/catch.hpp"
+
+using namespace gaussian_filter;
+
+TEST_CASE("Testing gaussian filter application") {
+    std::vector<std::vector<int32_t> > image(50, std::vector<int32_t>(50, 42));
+
+    auto resulted = apply_gaussian_filter(image);
+    REQUIRE(resulted.size() != 0);
+}
+
+TEST_CASE("Testing gaussian filter application with empty image") {
+    std::vector<std::vector<int32_t> > image;
+    CHECK_THROWS(apply_gaussian_filter(image));
+}
diff --git a/tests/machine_learning/image/median_filter.cc b/tests/machine_learning/image/median_filter.cc
@@ -0,0 +1,16 @@
+#include "../../../src/machine_learning/image/filters/median_filter.h"
+#include "../../../third_party/catch.hpp"
+
+using namespace median_filter;
+
+TEST_CASE("Testing median filter application") {
+    std::vector<std::vector<int32_t> > image(50, std::vector<int32_t>(50, 42));
+
+    auto resulted = apply_median_filter(image);
+    REQUIRE(resulted.size() != 0);
+}
+
+TEST_CASE("Testing median filter application with empty image") {
+    std::vector<std::vector<int32_t> > image;
+    CHECK_THROWS(apply_median_filter(image));
+}