GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-04-20 08:39:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

Allow specifying a temporary directory for fileio outputs.

Browse Source
This commit is contained in:
Antonio Sánchez 2024-01-20 00:55:14 +00:00 committed by Rasmus Munk Larsen
parent 6b6bb9d34e
commit 6163dbe2bc
1 changed files with 22 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
unsupported/test/sparse_extra.cpp
View File

@ -7,6 +7,9 @@
// Public License v. 2.0. If a copy of the MPL was not distributed // Public License v. 2.0. If a copy of the MPL was not distributed
// with this file, You can obtain one at http://mozilla.org/MPL/2.0/. // with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
#include <cstdlib>
#include <string>
#include "sparse.h" #include "sparse.h"
#ifdef min #ifdef min
@ -19,6 +22,16 @@
#include <Eigen/SparseExtra> #include <Eigen/SparseExtra>
// Read from an environment variable TEST_TMPDIR, if available,
// and append the provided filename. Defaults to local directory.
std::string GetTestTempFilename(const char* filename) {
const char* test_tmpdir = std::getenv("TEST_TMPDIR");
if (test_tmpdir == nullptr) {
return std::string(filename);
}
return std::string(test_tmpdir) + std::string("/") + std::string(filename);
}
template <typename SetterType, typename DenseType, typename Scalar, int Options> template <typename SetterType, typename DenseType, typename Scalar, int Options>
bool test_random_setter(SparseMatrix<Scalar, Options>& sm, const DenseType& ref, bool test_random_setter(SparseMatrix<Scalar, Options>& sm, const DenseType& ref,
const std::vector<Vector2i>& nonzeroCoords) { const std::vector<Vector2i>& nonzeroCoords) {
@ -116,8 +129,9 @@ void check_marketio() {
Index cols = internal::random<Index>(1, 100); Index cols = internal::random<Index>(1, 100);
SparseMatrixType m1, m2; SparseMatrixType m1, m2;
m1 = DenseMatrix::Random(rows, cols).sparseView(); m1 = DenseMatrix::Random(rows, cols).sparseView();
saveMarket(m1, "sparse_extra.mtx"); std::string filename = GetTestTempFilename("sparse_extra.mtx");
loadMarket(m2, "sparse_extra.mtx"); saveMarket(m1, filename);
loadMarket(m2, filename);
VERIFY_IS_EQUAL(DenseMatrix(m1), DenseMatrix(m2)); VERIFY_IS_EQUAL(DenseMatrix(m1), DenseMatrix(m2));
} }
@ -126,8 +140,9 @@ void check_marketio_vector() {
Index size = internal::random<Index>(1, 100); Index size = internal::random<Index>(1, 100);
VectorType v1, v2; VectorType v1, v2;
v1 = VectorType::Random(size); v1 = VectorType::Random(size);
saveMarketVector(v1, "vector_extra.mtx"); std::string filename = GetTestTempFilename("vector_extra.mtx");
loadMarketVector(v2, "vector_extra.mtx"); saveMarketVector(v1, filename);
loadMarketVector(v2, filename);
VERIFY_IS_EQUAL(v1, v2); VERIFY_IS_EQUAL(v1, v2);
} }
@ -149,8 +164,9 @@ void check_marketio_dense() {
DenseMatrixType m1, m2; DenseMatrixType m1, m2;
m1 = DenseMatrixType::Random(rows, cols); m1 = DenseMatrixType::Random(rows, cols);
saveMarketDense(m1, "dense_extra.mtx"); std::string filename = GetTestTempFilename("dense_extra.mtx");
loadMarketDense(m2, "dense_extra.mtx"); saveMarketDense(m1, filename);
loadMarketDense(m2, filename);
VERIFY_IS_EQUAL(m1, m2); VERIFY_IS_EQUAL(m1, m2);
} }