test_yac_mpi.c

Go to the documentation of this file.

// Copyright (c) 2025 The YAC Authors
//
// SPDX-License-Identifier: BSD-3-Clause
 
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <mpi.h>
#include "tests.h"
#include "yac_mpi_internal.h"
 
int main(void) {
 
  MPI_Init(NULL, NULL);
 
  int comm_rank;
  MPI_Comm_rank(MPI_COMM_WORLD, &comm_rank);
 
  // Test 1: Normal string (incl. empty one)
  {
    char const * test_strings[] = {
      "Hello, YAC!",
      "",
      "This is a longer test string to check packing and unpacking "
      "functionality."
    };
    enum {
      NUM_TEST_STRINGS = sizeof(test_strings) / sizeof(test_strings[0])};
 
    for (size_t str_idx = 0; str_idx < NUM_TEST_STRINGS; ++str_idx) {
      for (int allow_null = 0; allow_null <= 1; ++allow_null) {
 
        const char *test_str = test_strings[str_idx];
        size_t pack_size =
          yac_string_get_pack_size("test", test_str, MPI_COMM_WORLD, allow_null);
        void *buffer = malloc(pack_size);
        int pos = 0;
        yac_string_pack(
          "test", test_str, buffer, (int)pack_size, &pos, MPI_COMM_WORLD,
          allow_null);
 
        pos = 0;
        char *unpacked =
          yac_string_unpack(buffer, (int)pack_size, &pos, MPI_COMM_WORLD);
 
        if (!unpacked || strcmp(unpacked, test_str) != 0) {
          PUT_ERR("Normal string pack/unpack");
        }
        free(unpacked);
        free(buffer);
      } // for allow_null
    } // for str_idx
  }
 
  // Test 2: NULL string with allow_null = 1
  {
    int allow_null = 1;
    size_t pack_size =
    yac_string_get_pack_size("test", NULL, MPI_COMM_WORLD, allow_null);
    void *buffer = malloc(pack_size);
    int pos = 0;
    yac_string_pack(
      "test", NULL, buffer, (int)pack_size, &pos, MPI_COMM_WORLD, allow_null);
 
    pos = 0;
    char *unpacked =
      yac_string_unpack(buffer, (int)pack_size, &pos, MPI_COMM_WORLD);
    if (unpacked != NULL) {
      PUT_ERR("FAILED: NULL string pack/unpack");
    }
    free(buffer);
  }
 
  MPI_Finalize();
  return TEST_EXIT_CODE;
}