feat(stock-lab): High52WProximity 노드 — 신고가 대비 근접도 룰 점수
This commit is contained in:
32
stock-lab/app/test_screener_nodes_high52w.py
Normal file
32
stock-lab/app/test_screener_nodes_high52w.py
Normal file
@@ -0,0 +1,32 @@
|
||||
import datetime as dt
|
||||
import pandas as pd
|
||||
|
||||
from app.screener.engine import ScreenContext
|
||||
from app.screener.nodes.high52w import High52WProximity
|
||||
from app.screener._test_fixtures import make_master, make_prices, make_flow
|
||||
|
||||
|
||||
def _ctx(master, prices, flow):
|
||||
return ScreenContext(master=master, prices=prices, flow=flow,
|
||||
kospi=pd.Series(dtype=float, name="kospi"),
|
||||
asof=dt.date(2026, 5, 12))
|
||||
|
||||
|
||||
def test_proximity_at_high_returns_100():
|
||||
asof = dt.date(2026, 5, 12)
|
||||
master = make_master(["A"])
|
||||
prices = make_prices(["A"], days=260, asof=asof, trend_pct=0.05)
|
||||
flow = make_flow(["A"], days=260, asof=asof)
|
||||
|
||||
out = High52WProximity().compute(_ctx(master, prices, flow), {"window_days": 252})
|
||||
assert out["A"] >= 95
|
||||
|
||||
|
||||
def test_proximity_below_70pct_returns_0():
|
||||
asof = dt.date(2026, 5, 12)
|
||||
master = make_master(["A"])
|
||||
prices = make_prices(["A"], days=260, asof=asof, start_close=100000, trend_pct=-0.5)
|
||||
flow = make_flow(["A"], days=260, asof=asof)
|
||||
|
||||
out = High52WProximity().compute(_ctx(master, prices, flow), {"window_days": 252})
|
||||
assert out["A"] == 0
|
||||
Reference in New Issue
Block a user