mirror of
https://github.com/TREX-CoE/trexio.git
synced 2025-01-03 18:16:22 +01:00
add test to read nucleus_coord from hdf5
This commit is contained in:
parent
92f6a8fa81
commit
472dd5fd91
50
src/test.c
50
src/test.c
@ -7,14 +7,58 @@
|
|||||||
int test_read();
|
int test_read();
|
||||||
int test_write();
|
int test_write();
|
||||||
|
|
||||||
|
int test_h5read();
|
||||||
|
|
||||||
int main() {
|
int main() {
|
||||||
|
test_h5read();
|
||||||
test_write();
|
test_write();
|
||||||
test_read();
|
test_read();
|
||||||
return 0 ;
|
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() {
|
int test_write() {
|
||||||
const char* file_name = "trio_test";
|
const char* file_name = "trio_test";
|
||||||
|
|
||||||
@ -22,7 +66,7 @@ int test_write() {
|
|||||||
trio_exit_code rc;
|
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.};
|
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_t* file = NULL;
|
||||||
trio_exit_code rc;
|
trio_exit_code rc;
|
||||||
|
|
||||||
int64_t num;
|
uint64_t num;
|
||||||
double* charge;
|
double* charge;
|
||||||
double* coord;
|
double* coord;
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user