74 lines
2.6 KiB
Python
74 lines
2.6 KiB
Python
|
"""V3e: shadow-decision journal — DDL-shape parity, sink emission, reject-at-source."""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import re
|
||
|
|
import sys
|
||
|
|
from pathlib import Path
|
||
|
|
from types import SimpleNamespace
|
||
|
|
|
||
|
|
sys.path.insert(0, "/mnt/dolphinng5_predict")
|
||
|
|
|
||
|
|
from prod.clean_arch.violet.decision_engine import ShadowDecision
|
||
|
|
from prod.clean_arch.violet.shadow_journal import DecisionRow, TABLE, VioletDecisionJournal
|
||
|
|
|
||
|
|
DDL = Path("/mnt/dolphinng5_predict/prod/clickhouse/violet/22_violet_decisions.sql")
|
||
|
|
|
||
|
|
|
||
|
|
def _ddl_columns() -> set:
|
||
|
|
body = DDL.read_text()
|
||
|
|
inner = body[body.index("("): body.rindex(")")]
|
||
|
|
return set(re.findall(r"`(\w+)`", inner))
|
||
|
|
|
||
|
|
|
||
|
|
def _decision(**over):
|
||
|
|
base = dict(
|
||
|
|
ts_ns=10**12, scan_number=7, asset="BTCUSDT", side="SHORT", vel_div=-0.2,
|
||
|
|
fraction=0.2, conviction_leverage=9.0, notional_fraction=1.8,
|
||
|
|
target_exposure=124200.0, ars_score=1.48, bucket_idx=1, actuated=True,
|
||
|
|
)
|
||
|
|
base.update(over)
|
||
|
|
return ShadowDecision(**base)
|
||
|
|
|
||
|
|
|
||
|
|
def test_decisionrow_fields_equal_ddl_columns():
|
||
|
|
assert set(DecisionRow.model_fields.keys()) == _ddl_columns()
|
||
|
|
|
||
|
|
|
||
|
|
def test_journal_emits_validated_row_to_sink():
|
||
|
|
captured = []
|
||
|
|
j = VioletDecisionJournal(sink=lambda t, r: captured.append((t, r)), session_id="sess1")
|
||
|
|
ok = j.journal(_decision(), mono_ns=999)
|
||
|
|
assert ok and j.rows_emitted == 1
|
||
|
|
table, row = captured[0]
|
||
|
|
assert table == TABLE
|
||
|
|
assert set(row.keys()) == _ddl_columns()
|
||
|
|
assert row["asset"] == "BTCUSDT" and row["actuated"] == 1
|
||
|
|
assert row["session_id"] == "sess1" and row["mono_ns"] == 999
|
||
|
|
|
||
|
|
|
||
|
|
def test_reject_at_source_on_malformed_decision():
|
||
|
|
captured = []
|
||
|
|
j = VioletDecisionJournal(sink=lambda t, r: captured.append(r), session_id="s")
|
||
|
|
# duck-typed bad decision: non-finite vel_div must be rejected before the sink.
|
||
|
|
bad = SimpleNamespace(
|
||
|
|
scan_number=1, asset="BTCUSDT", side="SHORT", vel_div=float("nan"),
|
||
|
|
fraction=0.2, conviction_leverage=9.0, notional_fraction=1.8,
|
||
|
|
target_exposure=1.0, ars_score=1.0, bucket_idx=1, actuated=True,
|
||
|
|
)
|
||
|
|
ok = j.journal(bad, mono_ns=1)
|
||
|
|
assert ok is False
|
||
|
|
assert j.rows_rejected == 1 and j.rows_emitted == 0
|
||
|
|
assert captured == [] # nothing reached the spool
|
||
|
|
|
||
|
|
|
||
|
|
def test_negative_exposure_rejected():
|
||
|
|
j = VioletDecisionJournal(sink=lambda t, r: None, session_id="s")
|
||
|
|
bad = SimpleNamespace(
|
||
|
|
scan_number=1, asset="BTCUSDT", side="SHORT", vel_div=-0.2,
|
||
|
|
fraction=0.2, conviction_leverage=9.0, notional_fraction=1.8,
|
||
|
|
target_exposure=-5.0, ars_score=1.0, bucket_idx=1, actuated=True,
|
||
|
|
)
|
||
|
|
assert j.journal(bad, mono_ns=1) is False
|
||
|
|
assert j.rows_rejected == 1
|