[router] Rust e2e test (#2184)

This commit is contained in:
Byron Hsu
2024-11-25 16:02:03 -08:00
committed by GitHub
parent 7f076c2ce6
commit 1f76fc6e3f
5 changed files with 209 additions and 2 deletions

19
rust/py_test/run_suite.py Normal file
View File

@@ -0,0 +1,19 @@
import argparse
import glob
from sglang.test.test_utils import run_unittest_files
if __name__ == "__main__":
arg_parser = argparse.ArgumentParser()
arg_parser.add_argument(
"--timeout-per-file",
type=int,
default=1000,
help="The time limit for running one file in seconds.",
)
args = arg_parser.parse_args()
files = glob.glob("**/test_*.py", recursive=True)
exit_code = run_unittest_files(files, args.timeout_per_file)
exit(exit_code)