exoroute-cross-venue-order-.../tests/test_energi_bridge_basic.py

18 lines
754 B
Python

import time
from exoroute import LocalProblem, SharedVariables, PlanDelta, EnergiBridge
def test_energi_bridge_map_to_ir_basic():
lp = LocalProblem(id="LP-001", domain="equities", assets=["AAPL", "MSFT"], objective="min_latency", constraints={})
sv = SharedVariables(forecasts={"AAPL": 150.0}, priors={"AAPL": 149.5}, version=1)
pd = PlanDelta(delta={"route": {"AAPL": "optim"}}, timestamp=time.time(), author="tester", contract_id=lp.id)
bridge = EnergiBridge()
ir = bridge.map_to_ir(lp, sv, pd)
# Basic structural checks
assert "object" in ir
assert ir["object"]["type"] == "LocalProblem"
assert ir["morphisms"]["shared_variables"]["forecasts"]["AAPL"] == 150.0
assert ir["plan_delta"]["author"] == "tester"