From 5c462058879af33cc36c086cb6f996fa29fcb79f Mon Sep 17 00:00:00 2001
From: Bo Peng <ben.bob@gmail.com>
Date: Mon, 12 Feb 2024 14:36:56 -0600
Subject: [PATCH] Fix run_test script

---
 run_tests.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/run_tests.py b/run_tests.py
index 1e0e6a925..85fecb6eb 100755
--- a/run_tests.py
+++ b/run_tests.py
@@ -15,6 +15,9 @@ def get_testcases():
     for line in output.decode('utf8').splitlines():
         if line.strip().startswith('<Module'):
             cur_module = line.strip().split()[-1].rstrip('>')
+            if not os.path.isfile(cur_module):
+                cur_module = f'test/{cur_module}'
+            assert os.path.isfile(cur_module)
         if line.strip().startswith('<Function'):
             tests.append(cur_module + '::' + line.strip().split()[-1].rstrip('>'))
     return tests