Added support for vectors in debug namespace

spirosmaggioros · Aug 12, 2024 · 6deea62 · 6deea62
1 parent a51b34e
commit 6deea62
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 2 deletions.
diff --git a/src/helpers/debug.h b/src/helpers/debug.h
@@ -3,16 +3,32 @@
 
 #ifdef __cplusplus
 #include <iostream>
+#include <type_traits>
 #endif
 
 namespace DEBUG {
     template <typename T>
-    void print_arguments(T &&t) { // single argument
+    void print_arguments(const T &t) { // single argument
         std::cout << t << ' ';
     }
 
+    template <typename T>
+    void print_arguments(const std::vector<T> &t){ // vectors as arguments
+        for(const T& x : t) {
+            std::cout << x << ' ';
+        }
+    }
+
+    template <class T, typename ...Args>
+    void print_arguments(const std::vector<T> &t, Args&& ...args) { // multiple args
+        for(const T& x : t) {
+            std::cout << x << ' ';
+        }
+        print_arguments(std::forward<Args>(args)...);
+    }
+
     template <typename T, typename ...Args>
-    void print_arguments(T &&t, Args&& ...args) { // multiple arguments
+    void print_arguments(const T &t, Args&& ...args) { // single args
         std::cout << t << ' ';
         print_arguments(std::forward<Args>(args)...);
     }

diff --git a/tests/helpers/debug.cc b/tests/helpers/debug.cc
@@ -18,3 +18,11 @@ TEST_CASE("Testing get args for debug namespace") {
 
     CHECK_NOTHROW(DEBUG::get_args(shortest_path, 0, 2));
 }
+
+TEST_CASE("Testing get args for debug namespace with vectors") {
+    auto it = [&](int a, int b) -> std::vector<int> {
+        return {a, b};
+    };
+    std::vector<int> passed = {2, 4};
+    CHECK_NOTHROW(DEBUG::get_args(it, passed));
+}