feat(tests): 신규 키움 웹소켓 조건검색 및 실시간 조건검색 테스트 추가
변경 사항 ---- - _test_kiwoom_condition_list.py: 키움 웹소켓 조건검색 '목록조회' 기능을 단독으로 테스트하는 스크립트 추가 - _test_kiwoom_condition_realtime.py: 'momentum' 조건식을 실시간으로 등록하고 초기 매칭 종목 리스트 및 실시간 편입/이탈을 수신하는 테스트 스크립트 추가 - _verify_columnar_bitid.py, _verify_shared_e2e_breakout.py, _verify_shared_e2e.py: 공유 메모리 및 dict 간의 데이터 일관성을 검증하는 테스트 추가 영향 ---- - 신규 테스트 스크립트 추가로 키움 웹소켓 API의 기능 검증 및 안정성을 높임 - 기존 기능에 대한 영향 없음 Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
100
kis_trader/backtest/momentum_exit_ab_test.py
Normal file
100
kis_trader/backtest/momentum_exit_ab_test.py
Normal file
@@ -0,0 +1,100 @@
|
||||
#!/usr/bin/env python3
|
||||
"""모멘텀 청산 A/B — 어깨·래칫·트레일·시간컷 (실매 check_sell_signal_momentum_live 동일 우선순위)."""
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
from datetime import datetime
|
||||
|
||||
from database import TradeDB
|
||||
from kis_trader.backtest import momentum_backtest_common as mbc
|
||||
from kis_trader.engine import momentum_engine as me
|
||||
|
||||
START = "2026-05-01"
|
||||
END = "2026-05-31"
|
||||
|
||||
SCENARIOS = [
|
||||
("baseline_DB(전용청산)", {}),
|
||||
("래칫_2:1.5,5:1,8:0.7", {
|
||||
"ratchet_tiers": "2:1.5,5:1,8:0.7",
|
||||
}),
|
||||
("트레일1.5%_무장1%", {
|
||||
"trail_pct": 0.015,
|
||||
"trail_arm_pct": 0.01,
|
||||
}),
|
||||
("시간컷30분", {
|
||||
"max_hold_bars": 30,
|
||||
}),
|
||||
("어깨0.5%_컷0.3%_래칫OFF", {
|
||||
"ratchet_tiers": "",
|
||||
"shoulder_min_high": 0.005,
|
||||
"shoulder_cut_pct": 0.003,
|
||||
}),
|
||||
]
|
||||
|
||||
|
||||
def main() -> None:
|
||||
db = TradeDB()
|
||||
try:
|
||||
start_key, end_key, start_ymd, end_ymd = mbc.date_keys(START, END)
|
||||
candles_by_code, _ = mbc.load_momentum_candles_by_code(db, start_key, end_key)
|
||||
universe_by_slot, _, _, _, _ = mbc.resolve_momentum_universe(
|
||||
start_ymd, end_ymd, use_saved_history=True,
|
||||
)
|
||||
snap = db.get_merged_env_snapshot()
|
||||
fee_rate, sell_tax, slot = mbc.fee_and_slot_from_env_row(snap)
|
||||
base = me.get_momentum_defaults_from_db(db)
|
||||
base["portfolio_mode"] = True
|
||||
base["force_eod_exit"] = False
|
||||
portfolio = mbc.resolve_momentum_portfolio_params(snap, base, slot_money=slot)
|
||||
slot_money = float(portfolio["slot_money"])
|
||||
max_stocks = int(portfolio["max_stocks"])
|
||||
total_budget = float(portfolio["total_budget_krw"])
|
||||
period_days = max(
|
||||
1,
|
||||
(datetime.strptime(END, "%Y-%m-%d") - datetime.strptime(START, "%Y-%m-%d")).days + 1,
|
||||
)
|
||||
|
||||
rows = []
|
||||
for label, patch in SCENARIOS:
|
||||
params = dict(base)
|
||||
params.update(patch)
|
||||
meta: dict = {}
|
||||
trades = mbc.run_momentum_backtest_web_aligned(
|
||||
candles_by_code, params, universe_by_slot,
|
||||
slot_money=slot_money, fee_rate=fee_rate, sell_tax=sell_tax,
|
||||
max_stocks=max_stocks, total_budget_krw=total_budget, meta_out=meta,
|
||||
)
|
||||
stats = mbc.summarize_momentum_trades(
|
||||
trades, total_budget_krw=total_budget, period_days=period_days,
|
||||
)
|
||||
reasons = mbc.count_momentum_sell_reasons(trades)
|
||||
rows.append({
|
||||
"label": label,
|
||||
"pnl": stats["total_pnl"],
|
||||
"win_rate": stats["win_rate"],
|
||||
"trades": stats["total_trades"],
|
||||
"pf": stats["pf"],
|
||||
"sell_reasons": reasons,
|
||||
})
|
||||
print(
|
||||
f"{label:28s} | {stats['total_pnl']:>+10,.0f} | "
|
||||
f"승률 {stats['win_rate']:5.1f}% | {stats['total_trades']:3}건 | "
|
||||
f"PF {stats['pf']:.2f} | {reasons}"
|
||||
)
|
||||
|
||||
out = {
|
||||
"start": START,
|
||||
"end": END,
|
||||
"scenarios": rows,
|
||||
}
|
||||
ts = datetime.now().strftime("%Y%m%d_%H%M%S")
|
||||
path = f"kis_trader/backtest/results/momentum_exit_ab_{ts}.json"
|
||||
with open(path, "w", encoding="utf-8") as f:
|
||||
json.dump(out, f, indent=2, ensure_ascii=False)
|
||||
print(f"\n💾 저장: {path}")
|
||||
finally:
|
||||
db.close()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user