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
