修正路徑問題

Author: 5000user5000

benchmarks/run_benchmark.py

@@ -1,25 +1,36 @@
 import timeit
 import glob
 import os
+import sys
 import argparse
 import math
 import numpy as np
 
+# 添加專案根目錄到 Python 路徑
+# benchmarks/run_benchmark.py -> 專案根目錄
+SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
+PROJECT_ROOT = os.path.dirname(SCRIPT_DIR)
+sys.path.insert(0, PROJECT_ROOT)
+
 # 嘗試 import，如果失敗則提示
 try:
     import fast_jpeg_decoder as fjd
-except ImportError:
+except ImportError as e:
     print("❌ Error: Could not import 'fast_jpeg_decoder'. Make sure the C++ module is compiled and in path.")
+    print(f"   ImportError: {e}")
+    print(f"   Project root: {PROJECT_ROOT}")
     exit(1)
 
 try:
     from python_implementations import numpy_decoder
-except ImportError:
+except ImportError as e:
     print("❌ Error: Could not import 'numpy_decoder' from 'python_implementations'.")
+    print(f"   ImportError: {e}")
+    print(f"   Project root: {PROJECT_ROOT}")
     exit(1)
 
 # --- Configuration ---
-TEST_DATA_DIR = 'tests/test_data'
+TEST_DATA_DIR = os.path.join(PROJECT_ROOT, 'tests/test_data')
 OUTPUT_DIR = 'output'
 NUMBER_OF_RUNS = 10