diff --git a/benchmark/testall.py b/benchmark/testall.py index 8b0e7ab..ae6f901 100644 --- a/benchmark/testall.py +++ b/benchmark/testall.py @@ -4,12 +4,13 @@ import time # ================= 配置路径与规则 ================= -CONFIG_DIR = "config" -RUNNERS_DIR = "runners" -TARGET_SCRIPT = "mcoplib_mxbenchmark_ops.py" -OUTPUT_FILE = "teslalloutput.txt" +SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__)) +CONFIG_DIR = os.path.join(SCRIPT_DIR, "config") +RUNNERS_DIR = os.path.join(SCRIPT_DIR, "runners") +TARGET_SCRIPT = os.path.join(SCRIPT_DIR, "mcoplib_mxbenchmark_ops.py") +OUTPUT_FILE = os.path.join(SCRIPT_DIR, "testalloutput.txt") -STATISTICS_DIR = "statistics" +STATISTICS_DIR = os.path.join(SCRIPT_DIR, "statistics") CSV_FILENAME = "mcoplib_ops_performance_C500.csv" CSV_SAVE_PATH = os.path.join(STATISTICS_DIR, CSV_FILENAME) diff --git a/unit_test/test_benchmark_testall_paths.py b/unit_test/test_benchmark_testall_paths.py new file mode 100644 index 0000000..c613843 --- /dev/null +++ b/unit_test/test_benchmark_testall_paths.py @@ -0,0 +1,20 @@ +import os +from pathlib import Path + +from benchmark import testall + + +def test_testall_paths_are_script_relative(): + script_dir = Path(testall.__file__).resolve().parent + + for path in ( + testall.CONFIG_DIR, + testall.RUNNERS_DIR, + testall.TARGET_SCRIPT, + testall.OUTPUT_FILE, + testall.STATISTICS_DIR, + testall.CSV_SAVE_PATH, + ): + assert os.path.isabs(path) + resolved = Path(path).resolve() + assert resolved == script_dir or script_dir in resolved.parents