import time
import os
import sys

sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
from kis_trader.engine.tail_engine import run_tail_backtest_rust_experimental
import kis_rust_core

def test_speed():
    # 더미 데이터 생성
    codes_candles = {}
    for i in range(100):
        code = f"KR{i:04d}"
        codes_candles[code] = []
        for j in range(2000): # 종목당 2000개 봉 (대략 2주치 3분봉)
            codes_candles[code].append({
                "candle_time": "20260902120000",
                "open": 1000,
                "high": 1050,
                "low": 950,
                "close": 1020,
                "volume": 10000,
                "rsi": 50.0
            })
            
    params = {}
    
    t0 = time.time()
    run_tail_backtest_rust_experimental(codes_candles, params)
    t1 = time.time()
    
    print(f"Time taken: {t1 - t0:.4f}s")
    
if __name__ == "__main__":
    test_speed()