From 614f8a5f448b1f7e8298ced2585c178f4d7f435d Mon Sep 17 00:00:00 2001
From: Andrew Kay <git@ajk.me>
Date: Sat, 12 Feb 2022 12:53:48 -0600
Subject: [PATCH] Improve run_tests.sh script

---
 interface2/fpga/tests/run_tests.sh | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/interface2/fpga/tests/run_tests.sh b/interface2/fpga/tests/run_tests.sh
index c11bb2e..97c4430 100755
--- a/interface2/fpga/tests/run_tests.sh
+++ b/interface2/fpga/tests/run_tests.sh
@@ -4,12 +4,18 @@ VVP=vvp
 
 set -o pipefail
 
+if [ $# -eq 0 ]; then
+    TESTS=(*_tb)
+else
+    TESTS=$@
+fi
+
 ANY_FAILURES=0
 
-for TB in *_tb; do
-    echo "Running $TB"
+for TEST in ${TESTS[@]}; do
+    echo "Running $TEST"
 
-    ${VVP} -n $TB | awk "BEGIN{f=0} /^\[FAIL:/{f=1} 1; END{exit(f)}"
+    ${VVP} -n $TEST | awk "BEGIN{f=0} /^\[FAIL:/{f=1} 1; END{exit(f)}"
 
     if [ $? != 0 ]; then
         ANY_FAILURES=1