From f1dfdcace77edde67d822126591121da14f1e744 Mon Sep 17 00:00:00 2001
From: Edoardo Zoni <ezoni@lbl.gov>
Date: Wed, 7 Aug 2024 16:19:38 -0700
Subject: [PATCH] CI: add option to clean up test dirs

---
 .azure-pipelines.yml |  2 +-
 run_test.sh          | 41 +++++++++++++++++++++++++++++++++--------
 2 files changed, 34 insertions(+), 9 deletions(-)

diff --git a/.azure-pipelines.yml b/.azure-pipelines.yml
index 7e4edc2dc36..ad2144212fd 100644
--- a/.azure-pipelines.yml
+++ b/.azure-pipelines.yml
@@ -116,7 +116,7 @@ jobs:
   - bash: |
       set -eu -o pipefail
       df -h
-      ./run_test.sh
+      ./run_test.sh -c
       rm -rf ${WARPX_CI_TMP}
       df -h
     displayName: 'Build & test'
diff --git a/run_test.sh b/run_test.sh
index 4efa86eb36c..ba020a19eb4 100755
--- a/run_test.sh
+++ b/run_test.sh
@@ -23,10 +23,25 @@
 
 set -eu -o pipefail
 
-# Parse command line arguments: if test names are given as command line arguments,
+# Parse command line arguments:
+# - if -c option is given, set clean=1 and call regtest.py with option --rm_testdir
+# in order to remove all subdirectories from each test directory, after each test
+# - if test names are given as command line arguments,
 # store them in variable tests_arg and define new command line argument to call
 # regtest.py with option --tests (works also for single test)
-tests_arg=$*
+clean=0
+while getopts c name
+do
+    case ${name} in
+    c)    clean=1;;
+    esac
+done
+tests_arg=""
+if [ ! -z "${clean}" ]; then
+    tests_arg=${*:2}
+else
+    tests_arg=$*
+fi
 tests_run=${tests_arg:+--tests=${tests_arg}}
 
 # environment options
@@ -80,7 +95,7 @@ curl -sOL https://github.com/openPMD/openPMD-example-datasets/raw/4ba1d257c5b489
 cd -
 
 # Clone the AMReX regression test utility
-git clone https://github.com/AMReX-Codes/regression_testing.git
+git clone -b EZoni_rm_testdir https://github.com/EZoni/regression_testing.git
 
 # Prepare regression tests
 mkdir -p rt-WarpX/WarpX-benchmarks
@@ -93,12 +108,22 @@ cp -r Checksum ../../regression_testing/
 # Run tests
 cd ../../regression_testing/
 echo "cd $PWD"
-# run only tests specified in variable tests_arg (single test or multiple tests)
-if [[ ! -z "${tests_arg}" ]]; then
-  python3 regtest.py ../rt-WarpX/ci-tests.ini --skip_comparison --no_update all "${tests_run}"
-# run all tests (variables tests_arg and tests_run are empty)
+if [ ! -z "${clean}" ]; then
+    # run only tests specified in variable tests_arg (single test or multiple tests)
+    if [ ! -z "${tests_arg}" ]; then
+      python3 regtest.py ../rt-WarpX/ci-tests.ini --rm_testdir --skip_comparison --no_update all "${tests_run}"
+    # run all tests (variables tests_arg and tests_run are empty)
+    else
+      python3 regtest.py ../rt-WarpX/ci-tests.ini --rm_testdir --skip_comparison --no_update all
+    fi
 else
-  python3 regtest.py ../rt-WarpX/ci-tests.ini --skip_comparison --no_update all
+    # run only tests specified in variable tests_arg (single test or multiple tests)
+    if [ ! -z "${tests_arg}" ]; then
+      python3 regtest.py ../rt-WarpX/ci-tests.ini --skip_comparison --no_update all "${tests_run}"
+    # run all tests (variables tests_arg and tests_run are empty)
+    else
+      python3 regtest.py ../rt-WarpX/ci-tests.ini --skip_comparison --no_update all
+    fi
 fi
 
 # clean up python virtual environment