From ad0f5e678f808a1990ceca9f39c34ac3f3815acc Mon Sep 17 00:00:00 2001
From: Gaurav Dhingra <gauravdhingra.gxyd@gmail.com>
Date: Fri, 21 Mar 2025 12:37:24 +0530
Subject: [PATCH] test: improve run_tests.sh script to be less verbose

this makes it clear with colors, which test is starting, which failed
and which test passed
---
 tests/run_tests.sh | 26 +++++++++++++++++++-------
 1 file changed, 19 insertions(+), 7 deletions(-)

diff --git a/tests/run_tests.sh b/tests/run_tests.sh
index 7e57ec3..b1cc88c 100755
--- a/tests/run_tests.sh
+++ b/tests/run_tests.sh
@@ -1,4 +1,11 @@
-set -ex
+#!/bin/bash
+set -e
+
+# Color codes
+RED='\033[0;31m'
+GREEN='\033[0;32m'
+YELLOW='\033[1;33m'
+NC='\033[0m' # No Color
 
 MPIEXEC=${CONDA_PREFIX}/bin/mpiexec
 
@@ -8,20 +15,25 @@ else
   CC=clang
 fi
 
+# Compile C and Fortran sources
 $CC -I$CONDA_PREFIX/include -c ../src/mpi_wrapper.c
 $FC -c ../src/mpi_c_bindings.f90
 $FC -c ../src/mpi.f90
 
 for file in *.f90; do
   filename=$(basename "$file" .f90)
+  echo -e "${YELLOW}Compiling $filename...${NC}"
   $FC -c $file
   $FC mpi_wrapper.o mpi_c_bindings.o mpi.o $filename.o -o $filename -L$CONDA_PREFIX/lib -lmpi -Wl,-rpath,$CONDA_PREFIX/lib
 
-  echo "Running $filename with 1 MPI rank..."
-  ${MPIEXEC} -np 1 $filename
-  echo "Done!"
+  for np in 1 2; do
+    echo -e "${YELLOW}Running $filename with $np MPI ranks...${NC}"
+    if ${MPIEXEC} -np $np ./$filename; then
+      echo -e "${GREEN}Test $filename with $np MPI ranks PASSED!${NC}"
+    else
+      echo -e "${RED}Test $filename with $np MPI ranks FAILED!${NC}"
+      exit 1
+    fi
+  done
 
-  echo "Running $filename with 2 MPI rank..."
-  ${MPIEXEC} -np 2 $filename
-  echo "Done!"
 done