diff --git a/google-tests/test1.cpp b/google-tests/test1.cpp index e549c62..6213c4b 100644 --- a/google-tests/test1.cpp +++ b/google-tests/test1.cpp @@ -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 data{1, 2, 3, 4, 5, 6}; + + npy::npy_data_ptr 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); +}