unit tests: Save_npy_matrix
This commit is contained in:
@@ -37,3 +37,15 @@ TEST(HelloTest, Load_npy_matrix) {
|
||||
EXPECT_DOUBLE_EQ(data[2], expect_data[2]);
|
||||
EXPECT_DOUBLE_EQ(data[3], expect_data[3]);
|
||||
}
|
||||
|
||||
TEST(HelloTest, Save_npy_matrix) {
|
||||
const std::vector<double> data{1, 2, 3, 4, 5, 6};
|
||||
|
||||
npy::npy_data_ptr<double> d;
|
||||
d.data_ptr = data.data();
|
||||
d.shape = {2, 3};
|
||||
d.fortran_order = false;
|
||||
|
||||
const std::string path{"test1/data2.npy"};
|
||||
npy::write_npy(path, d);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user