1
0
mirror of https://github.com/TREX-CoE/trexio.git synced 2024-12-22 20:35:44 +01:00

add test to read nucleus_coord from hdf5

This commit is contained in:
q-posev 2021-02-23 18:20:14 +01:00
parent 92f6a8fa81
commit 472dd5fd91

View File

@ -7,14 +7,58 @@
int test_read();
int test_write();
int test_h5read();
int main() {
test_h5read();
test_write();
test_read();
return 0 ;
}
int test_h5read() {
const char* file_name = "test.h5";
trio_t* file = NULL;
trio_exit_code rc;
uint64_t num;
//double* charge;
double* coord;
file = trio_create(file_name, TRIO_HDF5);
/*rc = trio_read_nucleus_num(file,&num);
assert (num == 12);
charge = (double*) calloc(num, sizeof(double));
rc = trio_read_nucleus_charge(file,charge);
assert(charge[10] == 1.);*/
num = 4;
coord = (double*) calloc(3*num, sizeof(double));
rc = trio_read_nucleus_coord(file,coord);
for (size_t i=0; i<3*num; i++){
printf("%lf \n", coord[i]);
}
double x = coord[0] - 2.14171677;
assert( x*x < 1000.);
if (rc == TRIO_SUCCESS) {
printf("SUCCESS\n");
} else {
printf("FAILURE\n");
}
trio_close(file);
return 0;
}
int test_write() {
const char* file_name = "trio_test";
@ -22,7 +66,7 @@ int test_write() {
trio_exit_code rc;
int64_t num = 12;
uint64_t num = 12;
double charge[12] = {6., 6., 6., 6., 6., 6., 1., 1., 1., 1., 1., 1.};
@ -64,7 +108,7 @@ int test_read() {
trio_t* file = NULL;
trio_exit_code rc;
int64_t num;
uint64_t num;
double* charge;
double* coord;