Files
vault-dash/tests/test_portfolio_math.py

68 lines
2.3 KiB
Python

from __future__ import annotations
from app.domain.portfolio_math import (
build_alert_context,
portfolio_snapshot_from_config,
strategy_metrics_from_snapshot,
)
from app.models.portfolio import PortfolioConfig
from app.pages.common import strategy_catalog
def test_portfolio_snapshot_from_config_preserves_weight_price_and_margin_values() -> None:
config = PortfolioConfig(entry_price=4400.0, gold_ounces=220.0, entry_basis_mode="weight", loan_amount=145000.0)
snapshot = portfolio_snapshot_from_config(config)
assert snapshot["spot_price"] == 4400.0
assert snapshot["gold_units"] == 220.0
assert snapshot["gold_value"] == 968000.0
assert snapshot["net_equity"] == 823000.0
assert round(snapshot["margin_call_price"], 2) == 878.79
def test_build_alert_context_uses_unit_safe_gold_value_calculation() -> None:
config = PortfolioConfig(entry_price=4400.0, gold_ounces=220.0, entry_basis_mode="weight", loan_amount=145000.0)
context = build_alert_context(
config,
spot_price=4400.0,
source="configured_entry_price",
updated_at="",
)
assert context["gold_units"] == 220.0
assert context["gold_value"] == 968000.0
assert context["net_equity"] == 823000.0
assert round(float(context["margin_call_price"]), 2) == 878.79
def test_strategy_metrics_from_snapshot_preserves_minus_20pct_protective_put_example() -> None:
strategy = next(item for item in strategy_catalog() if item["name"] == "protective_put_atm")
snapshot = {
"gold_value": 215000.0,
"loan_amount": 145000.0,
"net_equity": 70000.0,
"spot_price": 215.0,
"gold_units": 1000.0,
"margin_call_ltv": 0.75,
"margin_call_price": 193.33,
"cash_buffer": 18500.0,
"hedge_budget": 8000.0,
"ltv_ratio": 145000.0 / 215000.0,
}
metrics = strategy_metrics_from_snapshot(strategy, -20, snapshot)
assert metrics["scenario_price"] == 172.0
assert metrics["unhedged_equity"] == 27000.0
assert metrics["hedged_equity"] == 58750.0
assert metrics["waterfall_steps"] == [
("Base equity", 70000.0),
("Spot move", -43000.0),
("Option payoff", 38000.0),
("Call cap", 0.0),
("Hedge cost", -6250.0),
("Net equity", 58750.0),
]