feat(BT-003): add event preset backtest comparison
This commit is contained in:
@@ -1,6 +1,12 @@
|
||||
"""Backtesting services and historical market-data adapters."""
|
||||
|
||||
from .comparison import EventComparisonService
|
||||
from .historical_provider import SyntheticHistoricalProvider, YFinanceHistoricalPriceSource
|
||||
from .service import BacktestService
|
||||
|
||||
__all__ = ["BacktestService", "SyntheticHistoricalProvider", "YFinanceHistoricalPriceSource"]
|
||||
__all__ = [
|
||||
"BacktestService",
|
||||
"EventComparisonService",
|
||||
"SyntheticHistoricalProvider",
|
||||
"YFinanceHistoricalPriceSource",
|
||||
]
|
||||
|
||||
126
app/services/backtesting/comparison.py
Normal file
126
app/services/backtesting/comparison.py
Normal file
@@ -0,0 +1,126 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from datetime import timedelta
|
||||
|
||||
from app.models.backtest import (
|
||||
BacktestPortfolioState,
|
||||
BacktestScenario,
|
||||
EventComparisonRanking,
|
||||
EventComparisonReport,
|
||||
ProviderRef,
|
||||
TemplateRef,
|
||||
)
|
||||
from app.models.event_preset import EventPreset
|
||||
from app.services.backtesting.historical_provider import SyntheticHistoricalProvider
|
||||
from app.services.backtesting.service import BacktestService
|
||||
from app.services.event_presets import EventPresetService
|
||||
from app.services.strategy_templates import StrategyTemplateService
|
||||
|
||||
|
||||
class EventComparisonService:
|
||||
def __init__(
|
||||
self,
|
||||
provider: SyntheticHistoricalProvider | None = None,
|
||||
template_service: StrategyTemplateService | None = None,
|
||||
event_preset_service: EventPresetService | None = None,
|
||||
backtest_service: BacktestService | None = None,
|
||||
) -> None:
|
||||
self.provider = provider or SyntheticHistoricalProvider()
|
||||
self.template_service = template_service or StrategyTemplateService()
|
||||
self.event_preset_service = event_preset_service or EventPresetService()
|
||||
self.backtest_service = backtest_service or BacktestService(
|
||||
provider=self.provider,
|
||||
template_service=self.template_service,
|
||||
)
|
||||
|
||||
def compare_event(
|
||||
self,
|
||||
*,
|
||||
preset_slug: str,
|
||||
initial_portfolio: BacktestPortfolioState,
|
||||
template_slugs: tuple[str, ...] | None = None,
|
||||
provider_ref: ProviderRef | None = None,
|
||||
) -> EventComparisonReport:
|
||||
preset = self.event_preset_service.get_preset(preset_slug)
|
||||
scenario = self.materialize_scenario(
|
||||
preset,
|
||||
initial_portfolio=initial_portfolio,
|
||||
template_slugs=template_slugs,
|
||||
provider_ref=provider_ref,
|
||||
)
|
||||
run_result = self.backtest_service.run_scenario(scenario)
|
||||
ranked_results = sorted(
|
||||
run_result.template_results,
|
||||
key=lambda result: (
|
||||
result.summary_metrics.margin_call_days_hedged,
|
||||
result.summary_metrics.max_ltv_hedged,
|
||||
result.summary_metrics.total_hedge_cost,
|
||||
-result.summary_metrics.end_value_hedged_net,
|
||||
result.template_slug,
|
||||
),
|
||||
)
|
||||
rankings = tuple(
|
||||
EventComparisonRanking(
|
||||
rank=index,
|
||||
template_slug=result.template_slug,
|
||||
template_name=result.template_name,
|
||||
survived_margin_call=not result.summary_metrics.margin_threshold_breached_hedged,
|
||||
max_ltv_hedged=result.summary_metrics.max_ltv_hedged,
|
||||
hedge_cost=result.summary_metrics.total_hedge_cost,
|
||||
final_equity=result.summary_metrics.end_value_hedged_net,
|
||||
result=result,
|
||||
)
|
||||
for index, result in enumerate(ranked_results, start=1)
|
||||
)
|
||||
return EventComparisonReport(
|
||||
event_preset=preset,
|
||||
scenario=scenario,
|
||||
rankings=rankings,
|
||||
run_result=run_result,
|
||||
)
|
||||
|
||||
def materialize_scenario(
|
||||
self,
|
||||
preset: EventPreset,
|
||||
*,
|
||||
initial_portfolio: BacktestPortfolioState,
|
||||
template_slugs: tuple[str, ...] | None = None,
|
||||
provider_ref: ProviderRef | None = None,
|
||||
) -> BacktestScenario:
|
||||
selected_template_slugs = tuple(template_slugs or preset.scenario_overrides.default_template_slugs)
|
||||
if not selected_template_slugs:
|
||||
raise ValueError("Event comparison requires at least one template slug")
|
||||
|
||||
requested_start = preset.window_start - timedelta(days=preset.scenario_overrides.lookback_days or 0)
|
||||
requested_end = preset.window_end + timedelta(days=preset.scenario_overrides.recovery_days or 0)
|
||||
history = self.provider.load_history(preset.symbol, requested_start, requested_end)
|
||||
if not history:
|
||||
raise ValueError(f"No historical prices found for event preset: {preset.slug}")
|
||||
|
||||
scenario_portfolio = BacktestPortfolioState(
|
||||
currency=initial_portfolio.currency,
|
||||
underlying_units=initial_portfolio.underlying_units,
|
||||
entry_spot=history[0].close,
|
||||
loan_amount=initial_portfolio.loan_amount,
|
||||
margin_call_ltv=initial_portfolio.margin_call_ltv,
|
||||
cash_balance=initial_portfolio.cash_balance,
|
||||
financing_rate=initial_portfolio.financing_rate,
|
||||
)
|
||||
template_refs = tuple(
|
||||
TemplateRef(slug=slug, version=self.template_service.get_template(slug).version)
|
||||
for slug in selected_template_slugs
|
||||
)
|
||||
return BacktestScenario(
|
||||
scenario_id=f"event-{preset.slug}",
|
||||
display_name=preset.display_name,
|
||||
symbol=preset.symbol,
|
||||
start_date=history[0].date,
|
||||
end_date=history[-1].date,
|
||||
initial_portfolio=scenario_portfolio,
|
||||
template_refs=template_refs,
|
||||
provider_ref=provider_ref
|
||||
or ProviderRef(
|
||||
provider_id=self.provider.provider_id,
|
||||
pricing_mode=self.provider.pricing_mode,
|
||||
),
|
||||
)
|
||||
Reference in New Issue
Block a user