unit tests: Save_npy_matrix

This commit is contained in:
2026-03-01 23:57:18 +01:00
parent 0390fe8b8e
commit 398d05571f

View File

@@ -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);
}