30 lines
910 B
Python
30 lines
910 B
Python
import time
|
|
from gridverse.energi_bridge import EnergiBridge
|
|
|
|
|
|
def test_to_ir_and_from_ir_roundtrip():
|
|
contract_type = "LocalProblem"
|
|
payload = {"id": "lp-1", "name": "test", "variables": {"p": 1}}
|
|
|
|
ir = EnergiBridge.to_ir(contract_type, payload)
|
|
|
|
assert isinstance(ir, dict)
|
|
assert ir.get("ir_type") == contract_type
|
|
assert ir.get("ir_payload") == payload
|
|
assert "metadata" in ir
|
|
|
|
back_type, back_payload = EnergiBridge.from_ir(ir)
|
|
assert back_type == contract_type
|
|
assert back_payload == payload
|
|
|
|
|
|
def test_map_gridverse_to_catopt_basic():
|
|
gridverse_contract = {
|
|
"type": "LocalProblem",
|
|
"payload": {"id": "lp-2", "name": "second"},
|
|
}
|
|
ir = EnergiBridge.map_gridverse_to_catopt(gridverse_contract)
|
|
assert isinstance(ir, dict)
|
|
assert ir.get("ir_type") == "LocalProblem"
|
|
assert ir.get("ir_payload") == gridverse_contract["payload"]
|