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[2], expect_data[2]);
|
||||||
EXPECT_DOUBLE_EQ(data[3], expect_data[3]);
|
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