Allow tests to be called from multiple directories; search for test

files.

Author: BenBrock

test/gtest/coo_test.cpp

@@ -1,20 +1,19 @@
-#include <gtest/gtest.h>
-
-#include <fmt/core.h>
-
+#include "util.hpp"
 #include <binsparse/binsparse.hpp>
-
-inline std::vector file_paths({"1138_bus/1138_bus.mtx",
-                               "chesapeake/chesapeake.mtx",
-                               "mouse_gene/mouse_gene.mtx"});
+#include <filesystem>
+#include <fmt/core.h>
+#include <gtest/gtest.h>
 
 TEST(BinsparseReadWrite, COOFormat) {
   using T = float;
   using I = std::size_t;
 
   std::string binsparse_file = "out.bsp.hdf5";
 
-  for (auto&& file_path : file_paths) {
+  auto base_path = find_prefix(files.front());
+
+  for (auto&& file : files) {
+    auto file_path = base_path + file;
     auto x = binsparse::__detail::mmread<
         T, I, binsparse::__detail::coo_matrix_owning<T, I>>(file_path);
 
@@ -46,4 +45,6 @@ TEST(BinsparseReadWrite, COOFormat) {
     delete matrix_.rowind;
     delete matrix_.colind;
   }
+
+  std::filesystem::remove(binsparse_file);
 }

test/gtest/csr_test.cpp

@@ -1,20 +1,19 @@
-#include <gtest/gtest.h>
-
-#include <fmt/core.h>
-
+#include "util.hpp"
 #include <binsparse/binsparse.hpp>
-
-inline std::vector file_paths({"1138_bus/1138_bus.mtx",
-                               "chesapeake/chesapeake.mtx",
-                               "mouse_gene/mouse_gene.mtx"});
+#include <filesystem>
+#include <fmt/core.h>
+#include <gtest/gtest.h>
 
 TEST(BinsparseReadWrite, CSRFormat) {
   using T = float;
   using I = std::size_t;
 
   std::string binsparse_file = "out.bsp.hdf5";
 
-  for (auto&& file_path : file_paths) {
+  auto base_path = find_prefix(files.front());
+
+  for (auto&& file : files) {
+    auto file_path = base_path + file;
     auto x = binsparse::__detail::mmread<
         T, I, binsparse::__detail::csr_matrix_owning<T, I>>(file_path);
 
@@ -46,4 +45,6 @@ TEST(BinsparseReadWrite, CSRFormat) {
     delete matrix_.row_ptr;
     delete matrix_.colind;
   }
+
+  std::filesystem::remove(binsparse_file);
 }

test/gtest/util.hpp

@@ -0,0 +1,19 @@
+#pragma once
+
+#include <filesystem>
+#include <string>
+#include <vector>
+
+inline std::vector<std::string> files({"data/1138_bus/1138_bus.mtx",
+                                       "data/chesapeake/chesapeake.mtx",
+                                       "data/mouse_gene/mouse_gene.mtx"});
+
+inline std::string find_prefix(std::string file_name) {
+  if (std::filesystem::exists("../../" + file_name)) {
+    return "../../";
+  } else if (std::filesystem::exists("build/" + file_name)) {
+    return "build/";
+  } else {
+    return "";
+  }
+}