88 lines
3.6 KiB
Python
88 lines
3.6 KiB
Python
|
"""V3a: V-TYPES wrappers over BLUE's live alpha kernels — parity + drift guards."""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import math
|
||
|
|
import sys
|
||
|
|
|
||
|
|
sys.path.insert(0, "/mnt/dolphinng5_predict")
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
from hypothesis import given, settings, strategies as st
|
||
|
|
from pydantic import ValidationError
|
||
|
|
|
||
|
|
from prod.clean_arch.violet.alpha_wrappers import (
|
||
|
|
AssetPick, SizeDecision, VioletAssetSelector, VioletBetSizer,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def _sizer(max_leverage: float = 9.0) -> VioletBetSizer:
|
||
|
|
return VioletBetSizer(base_fraction=0.20, min_leverage=0.5, max_leverage=max_leverage)
|
||
|
|
|
||
|
|
|
||
|
|
def _universe(n: int = 4, bars: int = 60) -> dict:
|
||
|
|
return {
|
||
|
|
f"A{i}USDT": [100.0 * (1 + 0.001 * math.sin(0.1 * t) + 0.0005 * i * t) for t in range(bars)]
|
||
|
|
for i in range(n)
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
# ── sizing reproduces the recorded model ──────────────────────────────────────
|
||
|
|
|
||
|
|
def test_notional_fraction_is_fraction_times_conviction():
|
||
|
|
d = _sizer().calculate(capital=69_000.0, vel_div=-0.05, trade_direction=-1)
|
||
|
|
assert d.notional_fraction == pytest.approx(d.fraction * d.conviction_leverage)
|
||
|
|
|
||
|
|
|
||
|
|
def test_extreme_short_signal_saturates_at_max_leverage_9():
|
||
|
|
# vel_div at/under extreme drives conviction to the cap; with max_leverage=9
|
||
|
|
# and base_fraction 0.20, notional_fraction == 1.8 — the recorded our_leverage max.
|
||
|
|
d = _sizer(9.0).calculate(capital=69_000.0, vel_div=-0.20, trade_direction=-1)
|
||
|
|
assert d.conviction_leverage == pytest.approx(9.0)
|
||
|
|
assert d.notional_fraction == pytest.approx(1.8)
|
||
|
|
|
||
|
|
|
||
|
|
def test_max_leverage_parameterization_is_live_not_a_default():
|
||
|
|
# The drift guard: the kernel default is 5.0, blue_parity passes 8.0, live is
|
||
|
|
# 9.0. Passing different caps MUST change the saturated conviction — proving
|
||
|
|
# VIOLET never inherits a hardcoded/drifted value.
|
||
|
|
cap5 = _sizer(5.0).calculate(capital=69_000.0, vel_div=-0.20, trade_direction=-1)
|
||
|
|
cap9 = _sizer(9.0).calculate(capital=69_000.0, vel_div=-0.20, trade_direction=-1)
|
||
|
|
assert cap5.conviction_leverage == pytest.approx(5.0)
|
||
|
|
assert cap9.conviction_leverage == pytest.approx(9.0)
|
||
|
|
assert cap5.conviction_leverage != cap9.conviction_leverage
|
||
|
|
|
||
|
|
|
||
|
|
def test_size_decision_is_typed_and_frozen():
|
||
|
|
d = _sizer().calculate(capital=50_000.0, vel_div=-0.04, trade_direction=-1)
|
||
|
|
assert isinstance(d, SizeDecision)
|
||
|
|
with pytest.raises(ValidationError):
|
||
|
|
d.fraction = 0.5 # frozen
|
||
|
|
|
||
|
|
|
||
|
|
# ── selection returns a typed pick or None ────────────────────────────────────
|
||
|
|
|
||
|
|
def test_selector_returns_typed_short_pick():
|
||
|
|
pick = VioletAssetSelector(lookback_horizon=50).pick(_universe(), regime_direction=-1)
|
||
|
|
if pick is not None:
|
||
|
|
assert isinstance(pick, AssetPick)
|
||
|
|
assert pick.side == "SHORT"
|
||
|
|
|
||
|
|
|
||
|
|
def test_selector_empty_universe_is_none():
|
||
|
|
assert VioletAssetSelector(lookback_horizon=50).pick({}, regime_direction=-1) is None
|
||
|
|
|
||
|
|
|
||
|
|
# ── property: conviction always within the kernel envelope; output finite ─────
|
||
|
|
|
||
|
|
@given(
|
||
|
|
vel_div=st.floats(min_value=-0.5, max_value=0.5, allow_nan=False, allow_infinity=False),
|
||
|
|
cap=st.floats(min_value=1.0, max_value=1e7, allow_nan=False, allow_infinity=False),
|
||
|
|
)
|
||
|
|
@settings(max_examples=60, deadline=None)
|
||
|
|
def test_conviction_within_envelope_and_finite(vel_div, cap):
|
||
|
|
d = _sizer(9.0).calculate(capital=cap, vel_div=vel_div, trade_direction=-1)
|
||
|
|
assert 0.0 <= d.conviction_leverage <= 9.0 + 1e-9
|
||
|
|
assert math.isfinite(d.notional_fraction) and d.notional_fraction >= 0.0
|
||
|
|
assert 0 <= d.bucket_idx <= 3
|