You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
68 lines
2.0 KiB
68 lines
2.0 KiB
2 years ago
|
#include "medusa/bits/io/CSV.hpp"
|
||
|
|
#include "medusa/bits/io/CSV_Eigen.hpp"
|
||
|
|
|
||
|
|
#include "gtest/gtest.h"
|
||
|
|
|
||
|
|
#include <vector>
|
||
|
|
#include <Eigen/Core>
|
||
|
|
|
||
|
|
namespace mm {
|
||
|
|
|
||
|
|
TEST(IO, CSVRW1d) {
|
||
|
|
std::vector<double> v = {1, 2, 3, -3.41203498123234};
|
||
|
|
CSV::write("test/testdata/v.csv", v);
|
||
|
|
|
||
|
|
std::vector<double> vread = CSV::read("test/testdata/v.csv");
|
||
|
|
EXPECT_EQ(v, vread);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(IO, CSVRW2d) {
|
||
|
|
std::vector<std::vector<double>> M = {{1, 2},
|
||
|
|
{3, -3.4},
|
||
|
|
{-4.5, 4.5}};
|
||
|
|
CSV::write2d("test/testdata/M.csv", M, '!');
|
||
|
|
auto Mread = CSV::read2d("test/testdata/M.csv", '!');
|
||
|
|
EXPECT_EQ(M, Mread);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(IO, CSVEigen1D) {
|
||
|
|
Eigen::VectorXd v(10);
|
||
|
|
v << 1, 3, 4, -2, 3, 5.6, 3.4, -2.3, 4, 5;
|
||
|
|
CSV::write("test/testdata/v2.csv", v);
|
||
|
|
auto vread = CSV::readEigen("test/testdata/v2.csv");
|
||
|
|
|
||
|
|
EXPECT_TRUE((v - vread).isZero(0));
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(IO, CSVEigen2D) {
|
||
|
|
Eigen::MatrixXd M(3, 2); M << 1, 2, 3, 4.3, 7.1, -2.3;
|
||
|
|
CSV::writeEigen("test/testdata/M2.csv", M, ';');
|
||
|
|
|
||
|
|
auto Mread = CSV::readEigen("test/testdata/M2.csv", ';');
|
||
|
|
EXPECT_TRUE((M - Mread).isZero(0));
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(IO, CSVDeath) {
|
||
|
|
EXPECT_DEATH(CSV::read2d("test/testdata/death1.csv"), "Failed parsing .* to double");
|
||
|
|
EXPECT_DEATH(CSV::read2d("test/testdata/death2.csv"),
|
||
|
|
"Not all rows in CSV file have the same number of elements");
|
||
|
|
EXPECT_DEATH(CSV::read2d("test/testdata/death3.csv"), "Error opening CSV file");
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(IO, CSVUsageExample) {
|
||
|
|
/// [CSV usage example]
|
||
|
|
std::vector<double> v = {1, 2.5, -4.5, 12.34};
|
||
|
|
CSV::write("test/testdata/example.csv", v);
|
||
|
|
|
||
|
|
std::vector<double> v2 = CSV::read("test/testdata/example.csv");
|
||
|
|
// v is the same as v2
|
||
|
|
|
||
|
|
// For the following, the CSV_Eigen.hpp header needs to be included.
|
||
|
|
Eigen::MatrixXd M = CSV::readEigen("test/testdata/example.csv");
|
||
|
|
CSV::writeEigen("test/testdata/example.csv", M.transpose()*M);
|
||
|
|
/// [CSV usage example]
|
||
|
|
EXPECT_EQ(v, v2);
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace mm
|