mirror of
https://github.com/TREX-CoE/trexio.git
synced 2024-12-23 04:43:57 +01:00
134 lines
3.1 KiB
C
134 lines
3.1 KiB
C
#include "trexio.h"
|
|
#include <assert.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#define TEST_BACKEND TREXIO_HDF5
|
|
#define TREXIO_FILE "test_str.h5"
|
|
#define RM_COMMAND "rm -rf " TREXIO_FILE
|
|
|
|
static int test_write_str (const char* file_name, const back_end_t backend) {
|
|
|
|
/* Try to write a string attribute (single variable-length string) into the TREXIO file */
|
|
|
|
trexio_t* file = NULL;
|
|
trexio_exit_code rc;
|
|
|
|
// parameters to be written
|
|
const char* sym = "B3U with some comments";
|
|
|
|
/*================= START OF TEST ==================*/
|
|
|
|
// open file in 'write' mode
|
|
file = trexio_open(file_name, 'w', backend, &rc);
|
|
assert (file != NULL);
|
|
|
|
// write string attribute in an empty file
|
|
int max_str_len = 32;
|
|
rc = trexio_write_nucleus_point_group(file, sym, max_str_len);
|
|
assert (rc == TREXIO_SUCCESS);
|
|
|
|
// close current session
|
|
rc = trexio_close(file);
|
|
assert (rc == TREXIO_SUCCESS);
|
|
|
|
/*================= END OF TEST ==================*/
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
static int test_has_str (const char* file_name, const back_end_t backend) {
|
|
|
|
/* Try to check the existence of a string attribute in the TREXIO file */
|
|
|
|
trexio_t* file = NULL;
|
|
trexio_exit_code rc;
|
|
|
|
/*================= START OF TEST ==================*/
|
|
|
|
// open file
|
|
file = trexio_open(file_name, 'r', backend, &rc);
|
|
assert (file != NULL);
|
|
|
|
// check that the previously written string attribute exists
|
|
rc = trexio_has_nucleus_point_group(file);
|
|
assert (rc == TREXIO_SUCCESS);
|
|
|
|
// check that another string attribute does not exist
|
|
rc = trexio_has_mo_type(file);
|
|
assert (rc == TREXIO_HAS_NOT);
|
|
|
|
// close current session
|
|
rc = trexio_close(file);
|
|
assert (rc == TREXIO_SUCCESS);
|
|
|
|
/*================= END OF TEST ==================*/
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
static int test_read_str (const char* file_name, const back_end_t backend) {
|
|
|
|
/* Try to read a string attribute from the TREXIO file */
|
|
|
|
trexio_t* file = NULL;
|
|
trexio_exit_code rc;
|
|
|
|
// parameters to be read
|
|
char* sym;
|
|
|
|
/*================= START OF TEST ==================*/
|
|
|
|
// open file in 'read' mode
|
|
file = trexio_open(file_name, 'r', backend, &rc);
|
|
assert (file != NULL);
|
|
|
|
// read string attribute from the file
|
|
int max_str_len = 32;
|
|
sym = (char*) malloc(max_str_len*sizeof(char));
|
|
|
|
rc = trexio_read_nucleus_point_group(file, sym, max_str_len);
|
|
assert (rc == TREXIO_SUCCESS);
|
|
|
|
char * pch;
|
|
pch = strtok(sym, " ");
|
|
assert (strcmp(pch, "B3U") == 0);
|
|
/* alternative test when 3 symbols are read from the file to sym */
|
|
/*rc = trexio_read_nucleus_point_group(file, sym, 3);
|
|
assert (rc == TREXIO_SUCCESS);
|
|
assert (strcmp(sym, "B3U") == 0 );*/
|
|
free(sym);
|
|
|
|
// close current session
|
|
rc = trexio_close(file);
|
|
assert (rc == TREXIO_SUCCESS);
|
|
|
|
/*================= END OF TEST ==================*/
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
int main(void) {
|
|
|
|
/*============== Test launcher ================*/
|
|
|
|
int rc;
|
|
rc = system(RM_COMMAND);
|
|
assert (rc == 0);
|
|
|
|
test_write_str (TREXIO_FILE, TEST_BACKEND);
|
|
test_has_str (TREXIO_FILE, TEST_BACKEND);
|
|
test_read_str (TREXIO_FILE, TEST_BACKEND);
|
|
|
|
rc = system(RM_COMMAND);
|
|
assert (rc == 0);
|
|
|
|
return 0;
|
|
}
|
|
|
|
|