30 lines
844 B
Python
30 lines
844 B
Python
from gridverse.energi_bridge import EnergiBridge
|
|
|
|
|
|
def test_to_ir_roundtrip_basic():
|
|
payload = {"p": 42, "q": "hello"}
|
|
ir = EnergiBridge.to_ir("LocalProblem", payload)
|
|
assert ir["ir_type"] == "LocalProblem"
|
|
assert ir["ir_payload"] == payload
|
|
|
|
|
|
def test_from_ir_roundtrip_basic():
|
|
ir = {
|
|
"ir_type": "LocalProblem",
|
|
"ir_payload": {"p": 7},
|
|
"metadata": {"source": "test"},
|
|
}
|
|
contract_type, payload = EnergiBridge.from_ir(ir)
|
|
assert contract_type == "LocalProblem"
|
|
assert payload == {"p": 7}
|
|
|
|
|
|
def test_map_gridverse_to_catopt_basic():
|
|
gridverse_contract = {
|
|
"contract_type": "LocalProblem",
|
|
"payload": {"a": 1},
|
|
}
|
|
ir = EnergiBridge.map_gridverse_to_catopt(gridverse_contract)
|
|
assert ir["ir_type"] == "LocalProblem"
|
|
assert ir["ir_payload"] == {"a": 1}
|