plevel.sh first test

Author: underwoo

tests/Makefile.am

@@ -29,7 +29,15 @@ else
 skipflag=""
 endif
 
+AM_CFLAGS = $(NETCDF_CFLAGS)
+LDADD = $(NETCDF_LDFLAGS) $(NETCDF_LIBS)
+
+check_PROGRAMS = create_plevel_test_ncfile
+create_plevel_test_ncfile_SOURCES=plevel/create_plevel_test_ncfile.c
+
+
 TESTS_ENVIRONMENT = top_srcdir=$(abs_top_srcdir); export top_srcdir; \
+                    builddir=$(abs_builddir); export builddir; \
                     export skip_mpi=$(skip_MPI); \
                     export skipflag=$(skipflag); \
                     export PKGLIBEXECDIR="$(abs_top_builddir)/src" \
@@ -63,6 +71,7 @@ TESTS = combine-ncc/combine \
         mppnccombine/xt-decomp \
         mppnccombine/xyt-decomp \
         mppnccombine/yt-decomp \
+        plevel/plevel \
         scatter-ncc/scatter_1x1 \
         scatter-ncc/scatter_1x2 \
         scatter-ncc/scatter_2x1

tests/plevel/create_ncfile.c renamed to tests/plevel/create_plevel_test_ncfile.c

@@ -22,7 +22,9 @@ void check_err(int stat, int line) {
 
 int main() {
     int ncid, time_dim, xt_dim, yt_dim, pfull_dim, phalf_dim;
-    int times_id, xt_id, yt_id, phalf_id, pfull_id, pk_id, bk_id, ps_id, temp_id;
+    int times_id, xt_id, yt_id, phalf_id, pfull_id, pk_id, bk_id;
+    int ps_id, temp_id, dummy_id;
+
     int stat;
 
     size_t start_ps[3] = {0, 0, 0};
@@ -61,10 +63,11 @@ int main() {
     check_err(stat, __LINE__);
     stat = nc_def_var(ncid, "bk", NC_FLOAT, 1, &phalf_dim, &bk_id);
     check_err(stat, __LINE__);
-    stat = nc_def_var(ncid, "ps", NC_FLOAT, 3, (int[]){time_dim, yt_dim, xt_dim}, &ps_id);
+    stat = nc_def_var(ncid, "ps", NC_FLOAT, 3, (int[]){time_dim, xt_dim, yt_dim}, &ps_id);
     check_err(stat, __LINE__);
     stat = nc_def_var(ncid, "temp", NC_FLOAT, 4, (int[]){time_dim, pfull_dim, xt_dim, yt_dim}, &temp_id);
     check_err(stat, __LINE__);
+    stat = nc_def_var(ncid, "dummy", NC_FLOAT, 3, (int[]){time_dim, yt_dim, xt_dim}, &dummy_id);
 
     // Assign global attributes
     stat = nc_put_att_text(ncid, NC_GLOBAL, "description", 40, "Sample netCDF file for testing run_timepressure_interp");
@@ -84,7 +87,8 @@ int main() {
     // Allocate memory for sample data and write to file
     double times_data[TIME] = {0};
     float xt_data[NX], yt_data[NY], temp_data[PFULL][NX][NY];
-    float ps_data[NY][NX];
+    float ps_data[NX][NY];
+    float dummy_data[NX][NY];
 
     for (int i = 0; i < NX; i++) xt_data[i] = i;
     for (int i = 0; i < NY; i++) yt_data[i] = i;
@@ -130,10 +134,14 @@ int main() {
             for (int k = 0; k < NY; k++)
                 temp_data[i][j][k] = (rand() % (TEMP_MAX - TEMP_MIN)) + TEMP_MIN;
 
-    for (int i = 0; i < NY; i++)
-        for (int j = 0; j < NX; j++)
+    for (int i = 0; i < NX; i++)
+        for (int j = 0; j < NY; j++)
             ps_data[i][j] = (rand() % (PS_MAX-PS_MIN)) + PS_MIN;
 
+    for (int i = 0; i < NX; i++)
+        for (int j = 0; j < NY; j++)
+            dummy_data[i][j] = i * NX + j;
+
     // Writing data
     stat = nc_put_var_double(ncid, times_id, &times_data[0]);
     check_err(stat, __LINE__);
@@ -156,9 +164,14 @@ int main() {
     check_err(stat, __LINE__);
     stat = nc_put_vara_float(ncid, ps_id,
                              (size_t[3]){0, 0, 0},
-                             (size_t[4]){1, NY, NX},
+                             (size_t[3]){1, NX, NY},
                              &ps_data[0][0]);
     check_err(stat, __LINE__);
+    stat = nc_put_vara_float(ncid, dummy_id,
+                             (size_t[3]){0, 0, 0},
+                             (size_t[3]){1, NX, NY},
+                             &dummy_data[0][0]);
+    check_err(stat, __LINE__);
 
     // Close the file
     stat = nc_close(ncid);

tests/plevel/plevel

@@ -0,0 +1,38 @@
+#!/bin/sh
+
+# ensure the help option prints something and exits successfully
+
+# Copyright (C) 2024 Geophysical Fluid Dynamics Laboratory
+
+# This file is part of the GFDL FRE NetCDF tools package (FRE-NCTools).
+
+# FRE-NCtools is free software: you can redistribute it and/or modify it under
+# the terms of the GNU Lesser General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or (at
+# your option) any later version.
+
+# FRE-NCtools is distributed in the hope that it will be useful, but WITHOUT
+# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
+# for more details.
+
+# You should have received a copy of the GNU Lesser General Public
+# License along with FRE-NCTools.  If not, see
+# <http://www.gnu.org/licenses/>.
+
+if test "$VERBOSE" = yes
+then
+    set -x
+    command -v plevel.sh
+    plevel.sh -V
+fi
+
+. "$srcdir/init.sh"
+
+# Generate the test_input.nc file
+$builddir/create_plevel_test_ncfile || framework_failure_
+
+# Run plevel.sh with default options
+$top_srcdir/src/plevel/plevel.sh -i test_input.nc temp || fail_ plevel.sh failed
+test -e plevel.nc || fail_ plevel.nc file does not exist after running plevel.sh
+ncdump -v temp plevel.nc || fail_ temp variable not found in plevel.sh