mirror of
https://github.com/slsdetectorgroup/aare.git
synced 2025-06-08 05:30:41 +02:00
53 lines
1.4 KiB
C++
53 lines
1.4 KiB
C++
#include <catch2/catch_test_macros.hpp>
|
|
#include "aare/NumpyFile.hpp"
|
|
#include "aare/NDArray.hpp"
|
|
|
|
#include "test_config.hpp"
|
|
|
|
using aare::NumpyFile;
|
|
using aare::DType;
|
|
TEST_CASE("Read a 1D numpy file with int32 data type"){
|
|
|
|
auto fpath = test_data_path() / "numpy" / "test_1d_int32.npy";
|
|
REQUIRE(std::filesystem::exists(fpath));
|
|
|
|
NumpyFile f(fpath);
|
|
|
|
//we know the file contains 10 elements of np.int32 containing values 0-9
|
|
REQUIRE(f.dtype() == DType::INT32);
|
|
REQUIRE(f.shape() == std::vector<size_t>{10});
|
|
|
|
//use the load function to read the full file into a NDArray
|
|
auto data = f.load<int32_t,1>();
|
|
for(size_t i = 0; i < 10; i++){
|
|
REQUIRE(data(i) == i);
|
|
}
|
|
|
|
}
|
|
|
|
TEST_CASE("Read a 3D numpy file with np.double data type"){
|
|
|
|
auto fpath = test_data_path() / "numpy" / "test_3d_double.npy";
|
|
REQUIRE(std::filesystem::exists(fpath));
|
|
|
|
NumpyFile f(fpath);
|
|
|
|
//we know the file contains 10 elements of np.int32 containing values 0-9
|
|
REQUIRE(f.dtype() == DType::DOUBLE);
|
|
REQUIRE(f.shape() == std::vector<size_t>{3,2,5});
|
|
|
|
//use the load function to read the full file into a NDArray
|
|
//numpy code to generate the array
|
|
// arr2[0,0,0] = 1.0
|
|
// arr2[0,0,1] = 2.0
|
|
// arr2[0,1,0] = 72.0
|
|
// arr2[2,0,4] = 63.0
|
|
|
|
auto data = f.load<double,3>();
|
|
REQUIRE(data(0,0,0) == 1.0);
|
|
REQUIRE(data(0,0,1) == 2.0);
|
|
REQUIRE(data(0,1,0) == 72.0);
|
|
REQUIRE(data(2,0,4) == 63.0);
|
|
|
|
|
|
} |